diff --git a/tests/test_agent_knowledge.py b/tests/test_agent_knowledge.py index ec471b654..85f5c5ef0 100644 --- a/tests/test_agent_knowledge.py +++ b/tests/test_agent_knowledge.py @@ -1,9 +1,10 @@ +import pytest +from unittest.mock import MagicMock, patch + 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 -import pytest -from unittest.mock import MagicMock, patch class MockEmbeddingFunction(EmbeddingFunction): def __call__(self, texts):