mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
fix: Sort imports in test_agent_knowledge.py using ruff
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -1,12 +1,13 @@
|
|||||||
from unittest.mock import MagicMock, patch
|
from unittest.mock import MagicMock, patch
|
||||||
import pytest
|
|
||||||
|
|
||||||
|
import pytest
|
||||||
from chromadb.api.types import EmbeddingFunction
|
from chromadb.api.types import EmbeddingFunction
|
||||||
|
|
||||||
from crewai import Agent, Crew, Task
|
from crewai import Agent, Crew, Task
|
||||||
from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource
|
from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource
|
||||||
from crewai.process import Process
|
from crewai.process import Process
|
||||||
|
|
||||||
|
|
||||||
class MockEmbeddingFunction(EmbeddingFunction):
|
class MockEmbeddingFunction(EmbeddingFunction):
|
||||||
def __call__(self, texts):
|
def __call__(self, texts):
|
||||||
return [[0.0] * 1536 for _ in texts]
|
return [[0.0] * 1536 for _ in texts]
|
||||||
|
|||||||
Reference in New Issue
Block a user