andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 16:18:30 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix linting issues in context-aware knowledge search tests

Browse Source 
- Remove unused imports: MagicMock, Knowledge, KnowledgeConfig
- Remove unused result variables in test functions
- Addresses lint failures in CI

Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
Devin AI
2025-08-17 17:31:16 +00:00
parent 6f139dff06
commit ac7390b287
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_context_aware_knowledge_search.py
View File

@@ -1,12 +1,10 @@
"""Test context-aware knowledge search functionality.""" """Test context-aware knowledge search functionality."""
import pytest import pytest
from unittest.mock import patch, MagicMock from unittest.mock import patch
from crewai import Agent, Task, Crew, LLM from crewai import Agent, Task, Crew, LLM
from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource
from crewai.knowledge.knowledge import Knowledge
from crewai.knowledge.knowledge_config import KnowledgeConfig
@pytest.mark.vcr(filter_headers=["authorization"]) @pytest.mark.vcr(filter_headers=["authorization"])
@@ -48,7 +46,7 @@ def test_knowledge_search_with_context():
with patch.object(writer, '_get_knowledge_search_query') as mock_search: with patch.object(writer, '_get_knowledge_search_query') as mock_search:
mock_search.return_value = "mocked query" mock_search.return_value = "mocked query"
result = crew.kickoff() crew.kickoff()
mock_search.assert_called_once() mock_search.assert_called_once()
call_args = mock_search.call_args call_args = mock_search.call_args
@@ -81,7 +79,7 @@ def test_knowledge_search_without_context():
with patch.object(agent, '_get_knowledge_search_query') as mock_search: with patch.object(agent, '_get_knowledge_search_query') as mock_search:
mock_search.return_value = "mocked query" mock_search.return_value = "mocked query"
result = crew.kickoff() crew.kickoff()
mock_search.assert_called_once() mock_search.assert_called_once()
call_args = mock_search.call_args call_args = mock_search.call_args
@@ -153,7 +151,7 @@ def test_knowledge_search_query_template_with_context():
with patch.object(agent.llm, 'call') as mock_call: with patch.object(agent.llm, 'call') as mock_call:
mock_call.return_value = "mocked response" mock_call.return_value = "mocked response"
result = agent._get_knowledge_search_query(task_prompt, context) agent._get_knowledge_search_query(task_prompt, context)
mock_call.assert_called_once() mock_call.assert_called_once()
call_args = mock_call.call_args[0][0] call_args = mock_call.call_args[0][0]
@@ -178,7 +176,7 @@ def test_knowledge_search_query_template_without_context():
with patch.object(agent.llm, 'call') as mock_call: with patch.object(agent.llm, 'call') as mock_call:
mock_call.return_value = "mocked response" mock_call.return_value = "mocked response"
result = agent._get_knowledge_search_query(task_prompt) agent._get_knowledge_search_query(task_prompt)
mock_call.assert_called_once() mock_call.assert_called_once()
call_args = mock_call.call_args[0][0] call_args = mock_call.call_args[0][0]
@@ -213,7 +211,7 @@ def test_structured_context_integration():
with patch.object(agent.llm, 'call') as mock_call: with patch.object(agent.llm, 'call') as mock_call:
mock_call.return_value = "Check getUserStatus method and URS-01 error" mock_call.return_value = "Check getUserStatus method and URS-01 error"
result = agent._get_knowledge_search_query(task_prompt, structured_context) agent._get_knowledge_search_query(task_prompt, structured_context)
mock_call.assert_called_once() mock_call.assert_called_once()
call_args = mock_call.call_args[0][0] call_args = mock_call.call_args[0][0]