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

fix linting

Browse source
This commit is contained in:
hajdul88 2025-12-11 13:18:25 +01:00
parent 4d0dca0b6c
commit 303fcd1803
1 changed files with 0 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -8,7 +8,6 @@ class TestGenerateCompletion:
async def test_generate_completion_with_system_prompt(self): async def test_generate_completion_with_system_prompt(self):
"""Test generate_completion with provided system_prompt.""" """Test generate_completion with provided system_prompt."""
mock_llm_response = "Generated answer" mock_llm_response = "Generated answer"
mock_llm_gateway = AsyncMock(return_value=mock_llm_response)
with ( with (
patch( patch(
@ -42,7 +41,6 @@ class TestGenerateCompletion:
async def test_generate_completion_without_system_prompt(self): async def test_generate_completion_without_system_prompt(self):
"""Test generate_completion reads system_prompt from file when not provided.""" """Test generate_completion reads system_prompt from file when not provided."""
mock_llm_response = "Generated answer" mock_llm_response = "Generated answer"
mock_llm_gateway = AsyncMock(return_value=mock_llm_response)
with ( with (
patch( patch(
@ -79,7 +77,6 @@ class TestGenerateCompletion:
async def test_generate_completion_with_conversation_history(self): async def test_generate_completion_with_conversation_history(self):
"""Test generate_completion includes conversation_history in system_prompt.""" """Test generate_completion includes conversation_history in system_prompt."""
mock_llm_response = "Generated answer" mock_llm_response = "Generated answer"
mock_llm_gateway = AsyncMock(return_value=mock_llm_response)
with ( with (
patch( patch(
@ -122,7 +119,6 @@ class TestGenerateCompletion:
async def test_generate_completion_with_conversation_history_and_custom_system_prompt(self): async def test_generate_completion_with_conversation_history_and_custom_system_prompt(self):
"""Test generate_completion includes conversation_history with custom system_prompt.""" """Test generate_completion includes conversation_history with custom system_prompt."""
mock_llm_response = "Generated answer" mock_llm_response = "Generated answer"
mock_llm_gateway = AsyncMock(return_value=mock_llm_response)
with ( with (
patch( patch(
@ -163,7 +159,6 @@ class TestGenerateCompletion:
"""Test generate_completion with custom response_model.""" """Test generate_completion with custom response_model."""
mock_response_model = MagicMock() mock_response_model = MagicMock()
mock_llm_response = {"answer": "Generated answer"} mock_llm_response = {"answer": "Generated answer"}
mock_llm_gateway = AsyncMock(return_value=mock_llm_response)
with ( with (
patch( patch(
@ -237,7 +232,6 @@ class TestSummarizeText:
async def test_summarize_text_with_system_prompt(self): async def test_summarize_text_with_system_prompt(self):
"""Test summarize_text with provided system_prompt.""" """Test summarize_text with provided system_prompt."""
mock_llm_response = "Summary text" mock_llm_response = "Summary text"
mock_llm_gateway = AsyncMock(return_value=mock_llm_response)
with patch( with patch(
"cognee.modules.retrieval.utils.completion.LLMGateway.acreate_structured_output", "cognee.modules.retrieval.utils.completion.LLMGateway.acreate_structured_output",
@ -263,7 +257,6 @@ class TestSummarizeText:
async def test_summarize_text_without_system_prompt(self): async def test_summarize_text_without_system_prompt(self):
"""Test summarize_text reads system_prompt from file when not provided.""" """Test summarize_text reads system_prompt from file when not provided."""
mock_llm_response = "Summary text" mock_llm_response = "Summary text"
mock_llm_gateway = AsyncMock(return_value=mock_llm_response)
with ( with (
patch( patch(
@ -294,7 +287,6 @@ class TestSummarizeText:
async def test_summarize_text_default_prompt_path(self): async def test_summarize_text_default_prompt_path(self):
"""Test summarize_text uses default prompt path when not provided.""" """Test summarize_text uses default prompt path when not provided."""
mock_llm_response = "Summary text" mock_llm_response = "Summary text"
mock_llm_gateway = AsyncMock(return_value=mock_llm_response)
with ( with (
patch( patch(
@ -323,7 +315,6 @@ class TestSummarizeText:
async def test_summarize_text_custom_prompt_path(self): async def test_summarize_text_custom_prompt_path(self):
"""Test summarize_text uses custom prompt path when provided.""" """Test summarize_text uses custom prompt path when provided."""
mock_llm_response = "Summary text" mock_llm_response = "Summary text"
mock_llm_gateway = AsyncMock(return_value=mock_llm_response)
with ( with (
patch( patch(