Drop patch

This commit is contained in:
Brandon Hancock
2024-12-12 13:43:36 -05:00
parent 1ba5c49ebe
commit de96f56c95
3 changed files with 25 additions and 336 deletions

View File

@@ -46,4 +46,5 @@ class BaseKnowledgeSource(BaseModel, ABC):
Save the documents to the storage.
This method should be called after the chunks and embeddings are generated.
"""
print("CHUNKS: ", self.chunks)
self.storage.save(self.chunks)

View File

@@ -1592,18 +1592,18 @@ def test_agent_execute_task_with_ollama():
assert "AI" in result or "artificial intelligence" in result.lower()
@pytest.mark.vcr(filter_headers=["authorization"])
# @pytest.mark.vcr(filter_headers=["authorization"])
def test_agent_with_knowledge_sources():
# Create a knowledge source with some content
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.sources = [string_source]
mock_knowledge_instance.query.return_value = [{"content": content}]
# with patch(
# "crewai.knowledge.storage.knowledge_storage.KnowledgeStorage"
# ) as MockKnowledge:
# mock_knowledge_instance = MockKnowledge.return_value
# mock_knowledge_instance.sources = [string_source]
# mock_knowledge_instance.query.return_value = [{"content": content}]
agent = Agent(
role="Information Agent",

File diff suppressed because one or more lines are too long