From 528ab0c4108ec0cdea03ce4e04a7acface9a9824 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Sun, 9 Feb 2025 23:39:44 +0000
Subject: [PATCH] style: Fix import sorting in test_memory_reset.py

Co-Authored-By: Joe Moura <joao@crewai.com>
---
 tests/memory/test_memory_reset.py | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/tests/memory/test_memory_reset.py b/tests/memory/test_memory_reset.py
index 12ec6fc35..eadab6b58 100644
--- a/tests/memory/test_memory_reset.py
+++ b/tests/memory/test_memory_reset.py
@@ -13,6 +13,7 @@ from crewai.utilities.exceptions.embedding_exceptions import (
     EmbeddingProviderError,
 )
 
+
 @pytest.fixture
 def temp_db_dir() -> Generator[Path, None, None]:
     """Create a temporary directory for test databases."""
@@ -20,32 +21,36 @@ def temp_db_dir() -> Generator[Path, None, None]:
         path = Path(tmpdir)
         yield path
 
+
 def test_memory_reset_with_openai(temp_db_dir):
     """Test memory reset with default OpenAI provider."""
     os.environ["OPENAI_API_KEY"] = "test-key"
     memory = ShortTermMemory(path=str(temp_db_dir))
     memory.reset()  # Should work with OpenAI as default
 
+
 def test_memory_reset_with_ollama(temp_db_dir):
     """Test memory reset with Ollama provider."""
     os.environ["CREWAI_EMBEDDING_PROVIDER"] = "ollama"
     memory = ShortTermMemory(path=str(temp_db_dir))
     memory.reset()  # Should not raise any OpenAI-related errors
 
+
 def test_memory_reset_with_custom_provider(temp_db_dir):
     """Test memory reset with custom embedding provider."""
+
     class CustomEmbedder(EmbeddingFunction):
         def __call__(self, input: Documents) -> Embeddings:
             if isinstance(input, str):
                 input = [input]
             return [[0.5] * 10] * len(input)
-            
+
     memory = ShortTermMemory(
-        path=str(temp_db_dir),
-        embedder_config={"provider": CustomEmbedder()}
+        path=str(temp_db_dir), embedder_config={"provider": CustomEmbedder()}
     )
     memory.reset()  # Should work with custom embedder
 
+
 def test_memory_reset_with_invalid_provider(temp_db_dir):
     """Test memory reset with invalid provider raises appropriate error."""
     os.environ["CREWAI_EMBEDDING_PROVIDER"] = "invalid_provider"
@@ -54,6 +59,7 @@ def test_memory_reset_with_invalid_provider(temp_db_dir):
         memory.reset()
     assert "Unsupported embedding provider" in str(exc_info.value)
 
+
 def test_memory_reset_with_missing_api_key(temp_db_dir):
     """Test memory reset with missing API key raises appropriate error."""
     os.environ.pop("OPENAI_API_KEY", None)  # Ensure key is not set
@@ -63,17 +69,18 @@ def test_memory_reset_with_missing_api_key(temp_db_dir):
         memory.reset()
     assert "openai api key" in str(exc_info.value).lower()
 
+
 def test_memory_reset_cleans_up_files(temp_db_dir):
     """Test that memory reset properly cleans up database files."""
+
     class TestEmbedder(EmbeddingFunction):
         def __call__(self, input: Documents) -> Embeddings:
             if isinstance(input, str):
                 input = [input]
             return [[0.5] * 10] * len(input)
-            
+
     memory = ShortTermMemory(
-        path=str(temp_db_dir),
-        embedder_config={"provider": TestEmbedder()}
+        path=str(temp_db_dir), embedder_config={"provider": TestEmbedder()}
     )
     memory.save("test memory", {"test": "metadata"})
     assert any(temp_db_dir.iterdir())  # Directory should have files