OpenAI error message (#1354)

evidentlyai · Oct 24, 2024 · 35038bf · 35038bf
1 parent 15bfe24
commit 35038bf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/evidently/utils/llm/wrapper.py b/src/evidently/utils/llm/wrapper.py
@@ -194,8 +194,8 @@ async def complete(self, messages: List[LLMMessage]) -> str:
         messages = [{"role": msg.role, "content": msg.content} for msg in messages]
         try:
             response = await self.client.chat.completions.create(model=self.model, messages=messages)  # type: ignore[arg-type]
-        except openai.OpenAIError as e:
-            raise LLMRequestError("Failed to call OpenAI complete API") from e
+        except openai.APIError as e:
+            raise LLMRequestError(f"Failed to call OpenAI complete API: {e.message}") from e
         content = response.choices[0].message.content
         assert content is not None  # todo: better error
         return content