ai: Properly forward non-successful responses from non-streaming chat (…

…#7440)
geldata · Jun 24, 2024 · d142189 · d142189
1 parent 16193df
commit d142189
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/edb/server/protocol/ai_ext.py b/edb/server/protocol/ai_ext.py
@@ -646,6 +646,13 @@ async def _start_openai_like_chat(
             }
         )
 
+        if result.status_code >= 400:
+            await result.aread()
+            raise AIProviderError(
+                f"API call to generate chat completions failed with status "
+                f"{result.status_code}: {result.text}"
+            )
+
         response.status = http.HTTPStatus.OK
         response_text = result.json()["choices"][0]["message"]["content"]
         response.content_type = b'application/json'
@@ -795,6 +802,13 @@ async def _start_anthropic_chat(
             }
         )
 
+        if result.status_code >= 400:
+            await result.aread()
+            raise AIProviderError(
+                f"API call to generate chat completions failed with status "
+                f"{result.status_code}: {result.text}"
+            )
+
         response.status = http.HTTPStatus.OK
         response.content_type = b'application/json'
         response_text = result.json()["content"][0]["text"]