diff --git a/src/crewai/utilities/embedding_configurator.py b/src/crewai/utilities/embedding_configurator.py index 6c5514739..6b24a9a83 100644 --- a/src/crewai/utilities/embedding_configurator.py +++ b/src/crewai/utilities/embedding_configurator.py @@ -1,5 +1,5 @@ -import os import logging +import os from typing import Any, Dict, Optional, cast # Import chromadb conditionally to handle SQLite3 version errors diff --git a/tests/utilities/test_embedding_configurator.py b/tests/utilities/test_embedding_configurator.py index 434d91805..b48063097 100644 --- a/tests/utilities/test_embedding_configurator.py +++ b/tests/utilities/test_embedding_configurator.py @@ -1,5 +1,6 @@ import unittest -from unittest.mock import patch, MagicMock +from unittest.mock import MagicMock, patch + class TestEmbeddingConfigurator(unittest.TestCase): @patch('crewai.utilities.embedding_configurator.CHROMADB_AVAILABLE', False)