From b0aa915dea19cc875bee3bee9241563db113fbbc Mon Sep 17 00:00:00 2001 From: pm390 <56439961+pm390@users.noreply.github.com> Date: Mon, 15 Jul 2024 20:14:39 +0200 Subject: [PATCH] community[patch]: use asyncio.sleep instead of sleep in OpenAI Assistant async (#24275) **Description:** Implemented async sleep using asyncio instead of synchronous sleep in openAI Assistants **Issue:** 24194 **Dependencies:** asyncio **Twitter handle:** pietromald60939 --- libs/langchain/langchain/agents/openai_assistant/base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/libs/langchain/langchain/agents/openai_assistant/base.py b/libs/langchain/langchain/agents/openai_assistant/base.py index 2db4625ec6b..c592942c721 100644 --- a/libs/langchain/langchain/agents/openai_assistant/base.py +++ b/libs/langchain/langchain/agents/openai_assistant/base.py @@ -1,5 +1,6 @@ from __future__ import annotations +import asyncio import json from json import JSONDecodeError from time import sleep @@ -742,5 +743,5 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]): ) in_progress = run.status in ("in_progress", "queued") if in_progress: - sleep(self.check_every_ms / 1000) + await asyncio.sleep(self.check_every_ms / 1000) return run