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 c60652bab..8ac05f5dc 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,5 +1,4 @@ import logging -from typing import List, Optional from chromadb.api.types import Documents, EmbeddingFunction, Embeddings @@ -9,7 +8,7 @@ logger = logging.getLogger(__name__) class VoyageAIEmbeddingFunction(EmbeddingFunction[Documents]): def __init__(self, api_key: str, model_name: str = "voyage-3"): try: - import voyageai + import voyageai # noqa: F401 except ImportError: raise ValueError( "The voyageai python package is not installed. Please install it with `pip install voyageai`" diff --git a/tests/utilities/test_embedding_configurator.py b/tests/utilities/test_embedding_configurator.py index a0446ab5a..26de07bef 100644 --- a/tests/utilities/test_embedding_configurator.py +++ b/tests/utilities/test_embedding_configurator.py @@ -1,5 +1,3 @@ -import os -import pytest from unittest.mock import patch, MagicMock from crewai.utilities.embedding_configurator import EmbeddingConfigurator