Fix lint issues (#1656)

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:

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 

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,

mem0/memory/storage.py

@@ -1,6 +1,5 @@
 import sqlite3
 import uuid
-from datetime import datetime
 
 
 class SQLiteManager:

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",
     }
 

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):

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"