Fix lint and type errors

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-04-25 20:56:22 +00:00
parent 1f2def2cbe
commit 335f1dfdf8
4 changed files with 8 additions and 6 deletions

View File

@@ -108,6 +108,7 @@ class EmbeddingConfigurator:
from chromadb.utils.embedding_functions.google_embedding_function import (
GoogleVertexEmbeddingFunction,
)
from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction
return FixedGoogleVertexEmbeddingFunction(

View File

@@ -1,10 +1,11 @@
from typing import List, Any
from typing import Any, List, Optional
import requests
from chromadb import Documents, Embeddings
from chromadb.utils.embedding_functions.google_embedding_function import (
GoogleVertexEmbeddingFunction,
)
import requests
from urllib.parse import urlparse, parse_qs, urlencode, urlunparse
from urllib.parse import parse_qs, urlencode, urlparse, urlunparse
class FixedGoogleVertexEmbeddingFunction(GoogleVertexEmbeddingFunction):
@@ -17,7 +18,7 @@ class FixedGoogleVertexEmbeddingFunction(GoogleVertexEmbeddingFunction):
def __init__(self,
model_name: str = "textembedding-gecko",
api_key: str = None,
api_key: Optional[str] = None,
**kwargs: Any):
super().__init__(model_name=model_name, api_key=api_key, **kwargs)

View File

@@ -1,5 +1,5 @@
import pytest
from unittest.mock import patch, MagicMock
from unittest.mock import MagicMock, patch
from crewai.utilities.embedding_configurator import EmbeddingConfigurator
from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction

View File

@@ -1,6 +1,6 @@
import pytest
import requests
from unittest.mock import patch, MagicMock
from unittest.mock import MagicMock, patch
from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction