Properly return API errors

njbbaer · May 5, 2024 · c9449c2 · c9449c2
1 parent 38a2f0c
commit c9449c2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/api_client.py b/src/api_client.py
@@ -38,4 +38,7 @@ async def call_api(self, messages, parameters):
             url = f"{self.api_url}/v1/chat/completions"
             response = await client.post(url, headers=headers, json=body)
             response.raise_for_status()
-            return response.json()
+            response_json = response.json()
+            if "error" in response_json:
+                raise Exception(response_json["error"]["message"])
+            return response_json