feat(agent/core): Add max_output_tokens parameter to `create_chat_c…

…ompletion` interface

autogpts/autogpt/autogpt/core/resource/model_providers/openai.py

@@ -416,12 +416,17 @@ async def create_chat_completion(
         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)
 

autogpts/autogpt/autogpt/core/resource/model_providers/schema.py

@@ -357,6 +357,7 @@ async def create_chat_completion(
         model_name: str,
         completion_parser: Callable[[AssistantChatMessage], _T] = lambda _: None,
         functions: Optional[list[CompletionModelFunction]] = None,
+        max_output_tokens: Optional[int] = None,
         **kwargs,
     ) -> ChatModelResponse[_T]:
         ...

autogpts/autogpt/autogpt/processing/text.py

@@ -160,7 +160,7 @@ async def _process_text(
             model_prompt=prompt.messages,
             model_name=model,
             temperature=0.5,
-            max_tokens=max_result_tokens,
+            max_output_tokens=max_result_tokens,
             completion_parser=lambda s: (
                 extract_list_from_json(s.content) if output_type is not str else None
             ),