ruff

2025-12-11 20:09:06 +01:00 · 2025-12-11 20:09:06 +01:00 · 4a99138daf
commit 4a99138daf
parent 926168b6bd
1 changed files with 1 additions and 0 deletions
--- a/cognee/tests/test_search_db.py
+++ b/cognee/tests/test_search_db.py
@ -31,6 +31,7 @@ async def cleanup_litellm_clients():
    # Cleanup LiteLLM async clients to prevent RuntimeWarning about unawaited coroutine
    try:
        import litellm
+
        if hasattr(litellm, "close_litellm_async_clients"):
            await litellm.close_litellm_async_clients()
    except Exception: