fixed test

This commit is contained in:
Lorenze Jay
2025-01-24 14:43:49 -08:00
parent 849908c7ea
commit c6d8c75869

View File

@@ -1602,28 +1602,24 @@ def test_agent_with_knowledge_sources():
assert "red" in result.raw.lower()
def test_agent_with_knowledge_sources_works_with_test():
@patch("crewai.knowledge.storage.knowledge_storage.KnowledgeStorage")
def test_agent_with_knowledge_sources_works_with_copy(
MockKnowledgeStorage,
):
content = "Brandon's favorite color is red and he likes Mexican food."
string_source = StringKnowledgeSource(content=content)
with patch(
"crewai.knowledge.storage.knowledge_storage.KnowledgeStorage"
) as MockKnowledge:
mock_knowledge_instance = MockKnowledge.return_value
mock_knowledge_instance = MockKnowledgeStorage.return_value
mock_knowledge_instance.sources = [string_source]
mock_knowledge_instance.query.return_value = [{"content": content}]
agent = Agent(
role="test role",
goal="test goal",
backstory="test backstory",
llm=LLM(model="gpt-4o-mini"),
role="Information Agent",
goal="Provide information based on knowledge sources",
backstory="You have access to specific knowledge sources.",
knowledge_sources=[string_source],
)
agent.create_agent_executor()
agent_copy = agent.copy()
# Test that all properties were copied correctly
assert agent_copy.role == agent.role
assert agent_copy.goal == agent.goal
assert agent_copy.backstory == agent.backstory