github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-27 00:48:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain: sync -> async methods in OpenAI assistants (#21378)

Browse Source
This commit is contained in:
ccurme 2024-05-07 10:25:55 -04:00 committed by GitHub
parent ad3fd44a7f
commit 080af0ec53
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/langchain/langchain/agents/openai_assistant/base.py
View File

@ -434,7 +434,7 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]):
], ],
"metadata": input.get("thread_metadata"), "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. # Starting a new run in an existing thread.
elif "run_id" not in input: elif "run_id" not in input:
_ = await self.async_client.beta.threads.messages.create( _ = await self.async_client.beta.threads.messages.create(
@ -444,14 +444,14 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]):
file_ids=input.get("file_ids", []), file_ids=input.get("file_ids", []),
metadata=input.get("message_metadata"), 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 # Submitting tool outputs to an existing run, outside the AgentExecutor
# framework. # framework.
else: else:
run = await self.async_client.beta.threads.runs.submit_tool_outputs( run = await self.async_client.beta.threads.runs.submit_tool_outputs(
**input **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: except BaseException as e:
run_manager.on_chain_error(e) run_manager.on_chain_error(e)
raise e raise e