From 50059c7120dac0e17999e72e0af2b293af3b12be Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Fri, 25 Apr 2025 20:58:24 +0000 Subject: [PATCH] Fix import sorting with ruff Co-Authored-By: Joe Moura --- src/crewai/utilities/embedding_configurator.py | 6 ++++-- src/crewai/utilities/embedding_functions.py | 2 +- tests/utilities/test_embedding_configurator.py | 3 ++- tests/utilities/test_embedding_functions.py | 3 ++- 4 files changed, 9 insertions(+), 5 deletions(-) diff --git a/src/crewai/utilities/embedding_configurator.py b/src/crewai/utilities/embedding_configurator.py index 31b31a836..25e7ada04 100644 --- a/src/crewai/utilities/embedding_configurator.py +++ b/src/crewai/utilities/embedding_configurator.py @@ -108,8 +108,10 @@ class EmbeddingConfigurator: from chromadb.utils.embedding_functions.google_embedding_function import ( GoogleVertexEmbeddingFunction, ) - - from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction + + from crewai.utilities.embedding_functions import ( + FixedGoogleVertexEmbeddingFunction, + ) return FixedGoogleVertexEmbeddingFunction( model_name=model_name, diff --git a/src/crewai/utilities/embedding_functions.py b/src/crewai/utilities/embedding_functions.py index a0b43629f..07c247b7b 100644 --- a/src/crewai/utilities/embedding_functions.py +++ b/src/crewai/utilities/embedding_functions.py @@ -1,11 +1,11 @@ from typing import Any, List, Optional +from urllib.parse import parse_qs, urlencode, urlparse, urlunparse import requests from chromadb import Documents, Embeddings from chromadb.utils.embedding_functions.google_embedding_function import ( GoogleVertexEmbeddingFunction, ) -from urllib.parse import parse_qs, urlencode, urlparse, urlunparse class FixedGoogleVertexEmbeddingFunction(GoogleVertexEmbeddingFunction): diff --git a/tests/utilities/test_embedding_configurator.py b/tests/utilities/test_embedding_configurator.py index 4a25ba2eb..ca8cc5e28 100644 --- a/tests/utilities/test_embedding_configurator.py +++ b/tests/utilities/test_embedding_configurator.py @@ -1,6 +1,7 @@ -import pytest from unittest.mock import MagicMock, patch +import pytest + from crewai.utilities.embedding_configurator import EmbeddingConfigurator from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction diff --git a/tests/utilities/test_embedding_functions.py b/tests/utilities/test_embedding_functions.py index 89039ac31..aa6d08851 100644 --- a/tests/utilities/test_embedding_functions.py +++ b/tests/utilities/test_embedding_functions.py @@ -1,6 +1,7 @@ +from unittest.mock import MagicMock, patch + import pytest import requests -from unittest.mock import MagicMock, patch from crewai.utilities.embedding_functions import FixedGoogleVertexEmbeddingFunction