diff --git a/libs/community/langchain_community/agents/openai_assistant/base.py b/libs/community/langchain_community/agents/openai_assistant/base.py index ff3e670caae..f180be7c72f 100644 --- a/libs/community/langchain_community/agents/openai_assistant/base.py +++ b/libs/community/langchain_community/agents/openai_assistant/base.py @@ -227,8 +227,10 @@ class OpenAIAssistantV2Runnable(OpenAIAssistantRunnable): tools: Sequence[Union[BaseTool, dict]], model: str, *, + model_kwargs: dict[str, float] = {}, client: Optional[Union[openai.OpenAI, openai.AzureOpenAI]] = None, tool_resources: Optional[Union[AssistantToolResources, dict, NotGiven]] = None, + extra_body: Optional[object] = None, **kwargs: Any, ) -> OpenAIAssistantRunnable: """Create an OpenAI Assistant and instantiate the Runnable. @@ -241,6 +243,9 @@ class OpenAIAssistantV2Runnable(OpenAIAssistantRunnable): model: Assistant model to use. client: OpenAI or AzureOpenAI client. Will create default OpenAI client (Assistant v2) if not specified. + model_kwargs: Additional model arguments. Only available for temperature + and top_p parameters. + extra_body: Additional body parameters to be passed to the assistant. Returns: OpenAIAssistantRunnable configured to run using the created assistant. @@ -257,6 +262,8 @@ class OpenAIAssistantV2Runnable(OpenAIAssistantRunnable): tools=[_get_assistants_tool(tool) for tool in tools], # type: ignore tool_resources=tool_resources, # type: ignore[arg-type] model=model, + extra_body=extra_body, + **model_kwargs, ) return cls(assistant_id=assistant.id, client=client, **kwargs)