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

WIP: test check with prints

Browse Source
This commit is contained in:
Lorenze Jay
2025-01-27 12:24:13 -08:00
parent adec0892fa
commit 1cc9c981e4
9 changed files with 76 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/agent_test.py
View File

@@ -1618,17 +1618,22 @@ def test_agent_with_knowledge_sources_works_with_copy():
mock_knowledge_instance.query.return_value = [{"content": content}]
mock_knowledge_instance.save.return_value = None
mock_knowledge_instance.initialize_knowledge_storage.return_value = None
mock_knowledge_instance.collection_name = "test_collection"
# mock_knowledge_instance.embedder = {
# "provider": "openai",
# "config": {"model_name": "text-embedding-3-small", "api_key": "123"},
# }
agent = Agent(
role="Information Agent",
goal="Provide information based on knowledge sources",
backstory="You have access to specific knowledge sources.",
llm=LLM(model="gpt-4o-mini", api_key="123"),
llm=LLM(model="gpt-4o-mini"),
knowledge_sources=[string_source],
embedder_config={
"provider": "openai",
"config": {"model_name": "text-embedding-3-small", "api_key": "123"},
},
# embedder={
# "provider": "openai",
# "config": {"model_name": "text-embedding-3-small", "api_key": "123"},
# },
)
# Actually call copy instead of mocking it