fix again

This commit is contained in:
Lorenze Jay
2025-01-24 15:20:45 -08:00
parent 4008ba74f8
commit 65b6ff1cc7
2 changed files with 7 additions and 210 deletions

View File

@@ -1603,16 +1603,22 @@ def test_agent_with_knowledge_sources():
assert "red" in result.raw.lower()
@pytest.mark.vcr(filter_headers=["authorization"])
def test_agent_with_knowledge_sources_works_with_copy():
content = "Brandon's favorite color is red and he likes Mexican food."
string_source = StringKnowledgeSource(content=content)
# Create a mock for KnowledgeStorage
with patch(
"crewai.knowledge.storage.knowledge_storage.KnowledgeStorage"
) as MockKnowledge:
# Set up the mock instance
mock_knowledge_instance = MockKnowledge.return_value
mock_knowledge_instance.sources = [string_source]
mock_knowledge_instance.query.return_value = [{"content": content}]
# Mock the save method specifically
mock_knowledge_instance.save.return_value = None
# Mock initialize_knowledge_storage to avoid connection issues
mock_knowledge_instance.initialize_knowledge_storage.return_value = None
agent = Agent(
role="Information Agent",
@@ -1628,9 +1634,6 @@ def test_agent_with_knowledge_sources_works_with_copy():
) as mock_agent_copy:
mock_agent_copy.return_value = agent_copy
assert agent_copy.role == agent.role
assert agent_copy.goal == agent.goal
assert agent_copy.backstory == agent.backstory
assert agent_copy.knowledge_sources == agent.knowledge_sources
assert isinstance(agent_copy.llm, LLM)

File diff suppressed because one or more lines are too long