From 080af0ec5386f6b2d392e5587760dbf7344e4dec Mon Sep 17 00:00:00 2001 From: ccurme Date: Tue, 7 May 2024 10:25:55 -0400 Subject: [PATCH] langchain: sync -> async methods in OpenAI assistants (#21378) --- libs/langchain/langchain/agents/openai_assistant/base.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/langchain/langchain/agents/openai_assistant/base.py b/libs/langchain/langchain/agents/openai_assistant/base.py index a7571debf62..0266536f949 100644 --- a/libs/langchain/langchain/agents/openai_assistant/base.py +++ b/libs/langchain/langchain/agents/openai_assistant/base.py @@ -434,7 +434,7 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]): ], "metadata": input.get("thread_metadata"), } - run = await self._create_thread_and_run(input, thread) + run = await self._acreate_thread_and_run(input, thread) # Starting a new run in an existing thread. elif "run_id" not in input: _ = await self.async_client.beta.threads.messages.create( @@ -444,14 +444,14 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]): file_ids=input.get("file_ids", []), metadata=input.get("message_metadata"), ) - run = await self._create_run(input) + run = await self._acreate_run(input) # Submitting tool outputs to an existing run, outside the AgentExecutor # framework. else: run = await self.async_client.beta.threads.runs.submit_tool_outputs( **input ) - run = await self._wait_for_run(run.id, run.thread_id) + run = await self._await_for_run(run.id, run.thread_id) except BaseException as e: run_manager.on_chain_error(e) raise e