fix double argument issue

Significant-Gravitas · May 31, 2024 · cd9a5d7 · cd9a5d7
1 parent 58c755a
commit cd9a5d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/forge/forge/llm/providers/_openai_base.py b/forge/forge/llm/providers/_openai_base.py
@@ -340,11 +340,11 @@ async def _create_chat_completion(
             int: Number of prompt tokens used
             int: Number of completion tokens used
         """
+        completion_kwargs["model"] = completion_kwargs.get("model") or model
 
         @self._retry_api_request
         async def _create_chat_completion_with_retry() -> ChatCompletion:
             return await self._client.chat.completions.create(
-                model=model,
                 **completion_kwargs,  # type: ignore
             )