Code Formatting (#1828)

tests/llms/test_azure_openai.py

@@ -1,10 +1,11 @@
 
+from unittest.mock import Mock, patch
+
 import httpx
 import pytest
-from unittest.mock import Mock, patch
 
-from mem0.llms.azure_openai import AzureOpenAILLM
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.azure_openai import AzureOpenAILLM
 
 MODEL = "gpt-4o" # or your custom deployment name 
 TEMPERATURE = 0.7

tests/llms/test_groq.py

@@ -1,7 +1,10 @@
-import pytest
 from unittest.mock import Mock, patch
-from mem0.llms.groq import GroqLLM
+
+import pytest
+
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.groq import GroqLLM
+
 
 @pytest.fixture
 def mock_groq_client():

tests/llms/test_litellm.py

@@ -1,8 +1,10 @@
-import pytest
 from unittest.mock import Mock, patch
 
-from mem0.llms import litellm
+import pytest
+
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms import litellm
+
 
 @pytest.fixture
 def mock_litellm():

tests/llms/test_ollama.py

@@ -1,9 +1,12 @@
-import pytest
 from unittest.mock import Mock, patch
-from mem0.llms.ollama import OllamaLLM
+
+import pytest
+
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.ollama import OllamaLLM
 from mem0.llms.utils.tools import ADD_MEMORY_TOOL
 
+
 @pytest.fixture
 def mock_ollama_client():
     with patch('mem0.llms.ollama.Client') as mock_ollama:

tests/llms/test_openai.py

@@ -1,7 +1,10 @@
-import pytest
 from unittest.mock import Mock, patch
-from mem0.llms.openai import OpenAILLM
+
+import pytest
+
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.openai import OpenAILLM
+
 
 @pytest.fixture
 def mock_openai_client():

tests/llms/test_together.py

@@ -1,7 +1,10 @@
-import pytest
 from unittest.mock import Mock, patch
-from mem0.llms.together import TogetherLLM
+
+import pytest
+
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.together import TogetherLLM
+
 
 @pytest.fixture
 def mock_together_client():