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

ruff

Browse source
This commit is contained in:
hajdul88 2025-12-11 13:13:12 +01:00
parent ca593bba15
commit 4d0dca0b6c
2 changed files with 8 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
cognee/tests/unit/modules/retrieval/conversation_history_test.py
View file

@ -158,9 +158,7 @@ class TestConversationHistoryUtils:
"""Test save_conversation_history returns False when user_id is None."""
session_user.set(None)
with patch(
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
with patch("cognee.modules.retrieval.utils.session_cache.CacheConfig") as MockCacheConfig:
mock_config = MagicMock()
mock_config.caching = True
MockCacheConfig.return_value = mock_config
@ -183,9 +181,7 @@ class TestConversationHistoryUtils:
user = create_mock_user()
session_user.set(user)
with patch(
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
with patch("cognee.modules.retrieval.utils.session_cache.CacheConfig") as MockCacheConfig:
mock_config = MagicMock()
mock_config.caching = False
MockCacheConfig.return_value = mock_config
@ -305,9 +301,7 @@ class TestConversationHistoryUtils:
"""Test get_conversation_history returns empty string when user_id is None."""
session_user.set(None)
with patch(
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
with patch("cognee.modules.retrieval.utils.session_cache.CacheConfig") as MockCacheConfig:
mock_config = MagicMock()
mock_config.caching = True
MockCacheConfig.return_value = mock_config
@ -326,9 +320,7 @@ class TestConversationHistoryUtils:
user = create_mock_user()
session_user.set(user)
with patch(
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
with patch("cognee.modules.retrieval.utils.session_cache.CacheConfig") as MockCacheConfig:
mock_config = MagicMock()
mock_config.caching = False
MockCacheConfig.return_value = mock_config
@ -367,9 +359,7 @@ class TestConversationHistoryUtils:
await get_conversation_history(session_id=None)
mock_cache.get_latest_qa.assert_called_once_with(
str(user.id), "default_session"
)
mock_cache.get_latest_qa.assert_called_once_with(str(user.id), "default_session")
@pytest.mark.asyncio
async def test_get_conversation_history_cache_engine_none(self):
@ -406,9 +396,7 @@ class TestConversationHistoryUtils:
from cognee.infrastructure.databases.exceptions import CacheConnectionError
mock_cache = create_mock_cache_engine([])
mock_cache.get_latest_qa = AsyncMock(
side_effect=CacheConnectionError("Connection failed")
)
mock_cache.get_latest_qa = AsyncMock(side_effect=CacheConnectionError("Connection failed"))
cache_module = importlib.import_module(
"cognee.infrastructure.databases.cache.get_cache_engine"

3
cognee/tests/unit/modules/retrieval/test_completion.py
View file

@ -227,7 +227,8 @@ class TestGenerateCompletion:
)
mock_render.assert_called_once_with(
"user_prompt.txt", {"question": "What is AI?", "context": "AI is artificial intelligence"}
"user_prompt.txt",
{"question": "What is AI?", "context": "AI is artificial intelligence"},
)