diff --git a/src/crewai/knowledge/embedder/chromadb/utils/embedding_functions/voyageai_embedding_function.py b/src/crewai/knowledge/embedder/chromadb/utils/embedding_functions/voyageai_embedding_function.py index 921b84ba1..516426e7f 100644 --- a/src/crewai/knowledge/embedder/chromadb/utils/embedding_functions/voyageai_embedding_function.py +++ b/src/crewai/knowledge/embedder/chromadb/utils/embedding_functions/voyageai_embedding_function.py @@ -1,7 +1,7 @@ import logging from typing import List, Union -from chromadb.api.types import Documents, EmbeddingFunction, Embeddings +from chromadb.api.types import Documents, EmbeddingFunction logger = logging.getLogger(__name__) diff --git a/tests/utilities/test_embedding_configurator.py b/tests/utilities/test_embedding_configurator.py index 6891ec5da..8d75fee45 100644 --- a/tests/utilities/test_embedding_configurator.py +++ b/tests/utilities/test_embedding_configurator.py @@ -2,7 +2,6 @@ import pytest from unittest.mock import patch, MagicMock from crewai.utilities.embedding_configurator import EmbeddingConfigurator -from crewai.knowledge.embedder.chromadb.utils.embedding_functions.voyageai_embedding_function import VoyageAIEmbeddingFunction class TestEmbeddingConfigurator: