diff --git a/api/utils/api_utils.py b/api/utils/api_utils.py index 6518e9c61..abc69973b 100644 --- a/api/utils/api_utils.py +++ b/api/utils/api_utils.py @@ -688,12 +688,7 @@ async def is_strong_enough(chat_model, embedding_model): if chat_model: res = await asyncio.wait_for( - asyncio.to_thread( - chat_model.chat, - "Nothing special.", - [{"role": "user", "content": "Are you strong enough!?"}], - {} - ), + chat_model.async_chat("Nothing special.", [{"role": "user", "content": "Are you strong enough!?"}]), timeout=30 ) if "**ERROR**" in res: diff --git a/pyproject.toml b/pyproject.toml index 6578b78d9..137582ef9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,7 +14,7 @@ dependencies = [ "azure-storage-file-datalake==12.16.0", "anthropic==0.34.1", "arxiv==2.1.3", - "aspose-slide==24.7.0; platform_machine == 'x86_64' or (sys_platform == 'darwin' and platform_machine == 'arm64')", + "aspose-slides==24.7.0; platform_machine == 'x86_64' or (sys_platform == 'darwin' and platform_machine == 'arm64')", "atlassian-python-api==4.0.7", "beartype>=0.20.0,<1.0.0", "bio==1.7.1",