diff --git a/mem0/configs/vector_stores/chroma.py b/mem0/configs/vector_stores/chroma.py
index 625d0fbf..edafb5c2 100644
--- a/mem0/configs/vector_stores/chroma.py
+++ b/mem0/configs/vector_stores/chroma.py
@@ -1,6 +1,6 @@
 from typing import Optional,ClassVar
 
-from pydantic import BaseModel, Field, field_validator, model_validator
+from pydantic import BaseModel, Field, model_validator
 
 class ChromaDbConfig(BaseModel):
     try:
diff --git a/mem0/configs/vector_stores/qdrant.py b/mem0/configs/vector_stores/qdrant.py
index f4aa2fd9..6c261f4f 100644
--- a/mem0/configs/vector_stores/qdrant.py
+++ b/mem0/configs/vector_stores/qdrant.py
@@ -1,6 +1,6 @@
 from typing import Optional,ClassVar
 
-from pydantic import BaseModel, Field, field_validator, model_validator
+from pydantic import BaseModel, Field, model_validator
 
 class QdrantConfig(BaseModel):
     from qdrant_client import QdrantClient 
diff --git a/mem0/memory/main.py b/mem0/memory/main.py
index 951539a8..a673590c 100644
--- a/mem0/memory/main.py
+++ b/mem0/memory/main.py
@@ -1,12 +1,11 @@
 import logging
 import hashlib
-import os
 import uuid
 import pytz
 from datetime import datetime
-from typing import Any, Dict, Optional
+from typing import Any, Dict
 
-from pydantic import BaseModel, Field, ValidationError
+from pydantic import ValidationError
 
 from mem0.llms.utils.tools import (
     ADD_MEMORY_TOOL,
diff --git a/mem0/memory/storage.py b/mem0/memory/storage.py
index ef4de34f..33b1bf76 100644
--- a/mem0/memory/storage.py
+++ b/mem0/memory/storage.py
@@ -1,6 +1,5 @@
 import sqlite3
 import uuid
-from datetime import datetime
 
 
 class SQLiteManager:
diff --git a/mem0/utils/factory.py b/mem0/utils/factory.py
index 2224d29f..a4ff4214 100644
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -17,7 +17,6 @@ class LlmFactory:
         "together": "mem0.llms.together.TogetherLLM",
         "aws_bedrock": "mem0.llms.aws_bedrock.AWSBedrockLLM",
         "litellm": "mem0.llms.litellm.LiteLLM",
-        "ollama": "mem0.llms.ollama.OllamaLLM",
         "azure_openai": "mem0.llms.azure_openai.AzureOpenAILLM",
     }
 
diff --git a/mem0/vector_stores/configs.py b/mem0/vector_stores/configs.py
index 628684a5..0d76705c 100644
--- a/mem0/vector_stores/configs.py
+++ b/mem0/vector_stores/configs.py
@@ -1,4 +1,4 @@
-from typing import Optional, Dict, Type
+from typing import Optional, Dict
 from pydantic import BaseModel, Field, model_validator
 
 class VectorStoreConfig(BaseModel):
diff --git a/tests/test_proxy.py b/tests/test_proxy.py
index 5a3882d9..57d7fb8d 100644
--- a/tests/test_proxy.py
+++ b/tests/test_proxy.py
@@ -92,7 +92,7 @@ def test_completions_create_with_system_message(mock_memory_client, mock_litellm
     mock_memory_client.search.return_value = [{"memory": "Some relevant memory"}]
     mock_litellm.completion.return_value = {"choices": [{"message": {"content": "I'm doing well, thank you!"}}]}
     
-    response = completions.create(
+    completions.create(
         model="gpt-3.5-turbo",
         messages=messages,
         user_id="test_user"