diff --git a/forge/forge/llm/providers/_openai_base.py b/forge/forge/llm/providers/_openai_base.py index 7c307f97ecc5..852876286b3a 100644 --- 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 )