diff options
Diffstat (limited to 'autogpts/autogpt/autogpt/core/resource/model_providers/openai.py')
-rw-r--r-- | autogpts/autogpt/autogpt/core/resource/model_providers/openai.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py b/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py index 91fe09d9d..2ebb56638 100644 --- a/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py +++ b/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py @@ -416,12 +416,17 @@ class OpenAIProvider( model_name: OpenAIModelName, completion_parser: Callable[[AssistantChatMessage], _T] = lambda _: None, functions: Optional[list[CompletionModelFunction]] = None, + max_output_tokens: Optional[int] = None, **kwargs, ) -> ChatModelResponse[_T]: """Create a completion using the OpenAI API.""" openai_messages, completion_kwargs = self._get_chat_completion_args( - model_prompt, model_name, functions, **kwargs + model_prompt=model_prompt, + model_name=model_name, + functions=functions, + max_tokens=max_output_tokens, + **kwargs, ) tool_calls_compat_mode = bool(functions and "tools" not in completion_kwargs) |