Fix context manager usage for timeout enforcement

This commit is contained in:
LIghtJUNction
2026-03-21 03:36:32 +08:00
committed by GitHub
parent 37594dd74b
commit 864a6851df

View File

@@ -194,7 +194,7 @@ class Provider(AbstractProvider):
async def test(self, test_timeout: float = 45.0) -> None:
# Use anyio.fail_after to enforce timeout in async context.
# This avoids direct asyncio.wait_for usage inside async functions.
async with anyio.fail_after(test_timeout):
with anyio.fail_after(test_timeout):
await self.text_chat(prompt="REPLY `PONG` ONLY")