Refactor OpenAI provider to improve error handling and default model usage

2026-06-05 22:50:18 +00:00 · 2024-10-30 09:12:14 -04:00
parent 7d89af37f1
commit b47f04c557
1 changed files with 5 additions and 2 deletions
@@ -60,7 +60,10 @@ class OpenAI(BaseProvider):
        ]

        response = self.structured_client.chat.completions.create(
-            messages=messages, model=llm_model, response_model=response_model, **kwargs
+            messages=messages,
+            model=llm_model or self.DEFAULT_MODEL,
+            response_model=response_model,
+            **kwargs,
        )
        return response

@@ -70,7 +73,7 @@ class OpenAI(BaseProvider):
        ]

        response = self.client.chat.completions.create(
-            messages=messages, model=llm_model, **kwargs
+            messages=messages, model=llm_model or self.DEFAULT_MODEL, **kwargs
        )

        return response.choices[0].message.content