diff --git a/tests/test_agent_knowledge.py b/tests/test_agent_knowledge.py index aeecfd33e..f8d131619 100644 --- a/tests/test_agent_knowledge.py +++ b/tests/test_agent_knowledge.py @@ -1,12 +1,13 @@ from unittest.mock import MagicMock, patch -import pytest +import pytest from chromadb.api.types import EmbeddingFunction from crewai import Agent, Crew, Task from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource from crewai.process import Process + class MockEmbeddingFunction(EmbeddingFunction): def __call__(self, texts): return [[0.0] * 1536 for _ in texts]