From 0c4bdbf37967a165287cca59f3f72fa659020e8b Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 14 May 2025 19:20:29 +0000 Subject: [PATCH] Fix lint errors: Remove unused imports Co-Authored-By: Joe Moura --- .../utils/embedding_functions/voyageai_embedding_function.py | 3 +-- tests/utilities/test_embedding_configurator.py | 2 -- 2 files changed, 1 insertion(+), 4 deletions(-) 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