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

fix test

Browse Source
This commit is contained in:
Lorenze Jay
2025-01-24 14:25:20 -08:00
parent 79aaab99c4
commit ab8d56de4f
1 changed files with 22 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/agent_test.py
View File

@@ -1606,6 +1606,13 @@ def test_agent_with_knowledge_sources():
def test_agent_with_knowledge_sources_works_with_test(): def test_agent_with_knowledge_sources_works_with_test():
content = "Brandon's favorite color is red and he likes Mexican food." content = "Brandon's favorite color is red and he likes Mexican food."
string_source = StringKnowledgeSource(content=content) string_source = StringKnowledgeSource(content=content)
with patch(
"crewai.knowledge.storage.knowledge_storage.KnowledgeStorage"
) as MockKnowledge:
mock_knowledge_instance = MockKnowledge.return_value
mock_knowledge_instance.sources = [string_source]
agent = Agent( agent = Agent(
role="test role", role="test role",
goal="test goal", goal="test goal",
@@ -1617,6 +1624,7 @@ def test_agent_with_knowledge_sources_works_with_test():
agent.create_agent_executor() agent.create_agent_executor()
agent_copy = agent.copy() agent_copy = agent.copy()
# Test that all properties were copied correctly
assert agent_copy.role == agent.role assert agent_copy.role == agent.role
assert agent_copy.goal == agent.goal assert agent_copy.goal == agent.goal
assert agent_copy.backstory == agent.backstory assert agent_copy.backstory == agent.backstory