gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update test_search_db.py

Browse source
This commit is contained in:
hajdul88 2025-12-11 20:12:00 +01:00
parent 4a99138daf
commit d349283ce3
1 changed files with 14 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
cognee/tests/test_search_db.py
View file

@ -24,18 +24,26 @@ from collections import Counter
logger = get_logger() logger = get_logger()
@pytest_asyncio.fixture(autouse=True) @pytest_asyncio.fixture(scope="function", autouse=True)
async def cleanup_litellm_clients(): async def cleanup_litellm_clients():
"""Fixture to properly cleanup LiteLLM async clients after each test.""" """Fixture to properly cleanup LiteLLM async clients after each test.
This prevents RuntimeWarning in Python 3.10 about unawaited coroutine
'close_litellm_async_clients' during event loop cleanup.
"""
yield yield
# Cleanup LiteLLM async clients to prevent RuntimeWarning about unawaited coroutine # Cleanup LiteLLM async clients before event loop closes
try: try:
import litellm import litellm
if hasattr(litellm, "close_litellm_async_clients"): if hasattr(litellm, "close_litellm_async_clients"):
await litellm.close_litellm_async_clients() # Ensure we await the cleanup coroutine
except Exception: cleanup_coro = litellm.close_litellm_async_clients()
pass # LiteLLM might not be available or already cleaned up if cleanup_coro is not None:
await cleanup_coro
except (RuntimeError, Exception):
# Event loop might already be closing, ignore the error
pass
async def setup_test_environment(): async def setup_test_environment():