From f36418a5b0dd0fa2662478ec34b02196ca956187 Mon Sep 17 00:00:00 2001 From: Frederico Wu Date: Tue, 19 Mar 2024 00:34:03 -0400 Subject: [PATCH] langchain: creating assistants with file_ids (#19199) Changing OpenAIAssistantRunnable.create_assistant to send the `file_ids` parameter to openai.beta.assistants.create Co-authored-by: Frederico Wu --- libs/langchain/langchain/agents/openai_assistant/base.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/libs/langchain/langchain/agents/openai_assistant/base.py b/libs/langchain/langchain/agents/openai_assistant/base.py index 99620322d8d..e7cb100ae50 100644 --- a/libs/langchain/langchain/agents/openai_assistant/base.py +++ b/libs/langchain/langchain/agents/openai_assistant/base.py @@ -212,6 +212,7 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]): instructions=instructions, tools=[convert_to_openai_tool(tool) for tool in tools], # type: ignore model=model, + file_ids=kwargs.get("file_ids"), ) return cls(assistant_id=assistant.id, client=client, **kwargs) @@ -333,6 +334,7 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]): instructions=instructions, tools=openai_tools, # type: ignore model=model, + file_ids=kwargs.get("file_ids"), ) return cls(assistant_id=assistant.id, async_client=async_client, **kwargs)