mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
Fix import sorting with ruff
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -108,8 +108,10 @@ class EmbeddingConfigurator:
|
|||||||
from chromadb.utils.embedding_functions.google_embedding_function import (
|
from chromadb.utils.embedding_functions.google_embedding_function import (
|
||||||
GoogleVertexEmbeddingFunction,
|
GoogleVertexEmbeddingFunction,
|
||||||
)
|
)
|
||||||
|
|
||||||
from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction
|
from crewai.utilities.embedding_functions import (
|
||||||
|
FixedGoogleVertexEmbeddingFunction,
|
||||||
|
)
|
||||||
|
|
||||||
return FixedGoogleVertexEmbeddingFunction(
|
return FixedGoogleVertexEmbeddingFunction(
|
||||||
model_name=model_name,
|
model_name=model_name,
|
||||||
|
|||||||
@@ -1,11 +1,11 @@
|
|||||||
from typing import Any, List, Optional
|
from typing import Any, List, Optional
|
||||||
|
from urllib.parse import parse_qs, urlencode, urlparse, urlunparse
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
from chromadb import Documents, Embeddings
|
from chromadb import Documents, Embeddings
|
||||||
from chromadb.utils.embedding_functions.google_embedding_function import (
|
from chromadb.utils.embedding_functions.google_embedding_function import (
|
||||||
GoogleVertexEmbeddingFunction,
|
GoogleVertexEmbeddingFunction,
|
||||||
)
|
)
|
||||||
from urllib.parse import parse_qs, urlencode, urlparse, urlunparse
|
|
||||||
|
|
||||||
|
|
||||||
class FixedGoogleVertexEmbeddingFunction(GoogleVertexEmbeddingFunction):
|
class FixedGoogleVertexEmbeddingFunction(GoogleVertexEmbeddingFunction):
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
import pytest
|
|
||||||
from unittest.mock import MagicMock, patch
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
from crewai.utilities.embedding_configurator import EmbeddingConfigurator
|
from crewai.utilities.embedding_configurator import EmbeddingConfigurator
|
||||||
from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction
|
from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
import requests
|
import requests
|
||||||
from unittest.mock import MagicMock, patch
|
|
||||||
|
|
||||||
from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction
|
from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user