From b47f04c557cda4dc3613ce4e04bd8d070fb15bd4 Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Wed, 30 Oct 2024 09:12:14 -0400 Subject: [PATCH] Refactor OpenAI provider to improve error handling and default model usage --- simplemind/providers/openai.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/simplemind/providers/openai.py b/simplemind/providers/openai.py index cfa8471..3895096 100644 --- a/simplemind/providers/openai.py +++ b/simplemind/providers/openai.py @@ -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