This commit is contained in:
Kevin Hu 2025-12-12 20:11:25 +08:00
parent 4f0a9a8027
commit 07d41247e8
2 changed files with 2 additions and 7 deletions

View file

@ -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:

View file

@ -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",