Code Formatting (#1828)

mem0/__init__.py

@@ -2,5 +2,5 @@ import importlib.metadata
 
 __version__ = importlib.metadata.version("mem0ai")
 
-from mem0.memory.main import Memory  # noqa
 from mem0.client.main import MemoryClient  # noqa
+from mem0.memory.main import Memory  # noqa

mem0/configs/base.py

@@ -1,12 +1,13 @@
 import os
 from typing import Any, Dict, Optional
+
 from pydantic import BaseModel, Field
 
-from mem0.memory.setup import mem0_dir
-from mem0.vector_stores.configs import VectorStoreConfig
-from mem0.llms.configs import LlmConfig
 from mem0.embeddings.configs import EmbedderConfig
 from mem0.graphs.configs import GraphStoreConfig
+from mem0.llms.configs import LlmConfig
+from mem0.memory.setup import mem0_dir
+from mem0.vector_stores.configs import VectorStoreConfig
 
 
 class MemoryItem(BaseModel):

mem0/configs/embeddings/base.py

@@ -1,9 +1,10 @@
 from abc import ABC
-from mem0.configs.base import AzureConfig
-from typing import Optional, Union, Dict
+from typing import Dict, Optional, Union
 
 import httpx
 
+from mem0.configs.base import AzureConfig
+
 
 class BaseEmbedderConfig(ABC):
     """

mem0/configs/llms/base.py

@@ -1,9 +1,10 @@
 from abc import ABC
-from mem0.configs.base import AzureConfig
-from typing import Optional, Union, Dict
+from typing import Dict, Optional, Union
 
 import httpx
 
+from mem0.configs.base import AzureConfig
+
 
 class BaseLlmConfig(ABC):
     """

mem0/configs/vector_stores/chroma.py

@@ -1,6 +1,6 @@
 import subprocess
 import sys
-from typing import Optional, ClassVar, Dict, Any
+from typing import Any, ClassVar, Dict, Optional
 
 from pydantic import BaseModel, Field, model_validator
 

mem0/configs/vector_stores/pgvector.py

@@ -1,4 +1,4 @@
-from typing import Optional, Dict, Any
+from typing import Any, Dict, Optional
 
 from pydantic import BaseModel, Field, model_validator
 

mem0/configs/vector_stores/qdrant.py

@@ -1,5 +1,6 @@
+from typing import Any, ClassVar, Dict, Optional
+
 from pydantic import BaseModel, Field, model_validator
-from typing import Optional, ClassVar, Dict, Any
 
 
 class QdrantConfig(BaseModel):

mem0/embeddings/base.py

@@ -1,5 +1,5 @@
-from typing import Optional
 from abc import ABC, abstractmethod
+from typing import Optional
 
 from mem0.configs.embeddings.base import BaseEmbedderConfig
 

mem0/graphs/configs.py

@@ -1,7 +1,10 @@
 from typing import Optional
+
 from pydantic import BaseModel, Field, field_validator, model_validator
+
 from mem0.llms.configs import LlmConfig
 
+
 class Neo4jConfig(BaseModel):
     url: Optional[str] = Field(None, description="Host address for the graph database")
     username: Optional[str] = Field(None, description="Username for the graph database")

mem0/llms/anthropic.py

@@ -1,7 +1,4 @@
-import subprocess
-import sys
 import os
-import json
 from typing import Dict, List, Optional
 
 try:
@@ -9,8 +6,8 @@ try:
 except ImportError:
    raise ImportError("The 'anthropic' library is required. Please install it using 'pip install anthropic'.")
 
-from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.base import LLMBase
 
 
 class AnthropicLLM(LLMBase):

mem0/llms/aws_bedrock.py

@@ -1,16 +1,14 @@
-import subprocess
-import sys
-import os
 import json
-from typing import Dict, List, Optional, Any
+import os
+from typing import Any, Dict, List, Optional
 
 try:
     import boto3
 except ImportError:
     raise ImportError("The 'boto3' library is required. Please install it using 'pip install boto3'.")
 
-from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.base import LLMBase
 
 
 class AWSBedrockLLM(LLMBase):

mem0/llms/azure_openai.py

@@ -1,11 +1,11 @@
-import os
 import json
+import os
 from typing import Dict, List, Optional
 
 from openai import AzureOpenAI
 
-from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.base import LLMBase
 
 
 class AzureOpenAILLM(LLMBase):

mem0/llms/base.py

@@ -1,5 +1,5 @@
-from typing import Optional
 from abc import ABC, abstractmethod
+from typing import Optional
 
 from mem0.configs.llms.base import BaseLlmConfig
 

mem0/llms/groq.py

@@ -1,7 +1,5 @@
-import subprocess
-import sys
-import os
 import json
+import os
 from typing import Dict, List, Optional
 
 try:
@@ -9,8 +7,8 @@ try:
 except ImportError:
     raise ImportError("The 'groq' library is required. Please install it using 'pip install groq'.")
 
-from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.base import LLMBase
 
 
 class GroqLLM(LLMBase):

mem0/llms/litellm.py

@@ -1,5 +1,3 @@
-import subprocess
-import sys
 import json
 from typing import Dict, List, Optional
 
@@ -8,8 +6,8 @@ try:
 except ImportError:
     raise ImportError("The 'litellm' library is required. Please install it using 'pip install litellm'.")
 
-from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.base import LLMBase
 
 
 class LiteLLM(LLMBase):

mem0/llms/ollama.py

@@ -1,5 +1,3 @@
-import subprocess
-import sys
 from typing import Dict, List, Optional
 
 try:
@@ -7,8 +5,8 @@ try:
 except ImportError:
     raise ImportError("The 'ollama' library is required. Please install it using 'pip install ollama'.")
 
-from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.base import LLMBase
 
 
 class OllamaLLM(LLMBase):

mem0/llms/openai.py

@@ -1,11 +1,11 @@
-import os
 import json
+import os
 from typing import Dict, List, Optional
 
 from openai import OpenAI
 
-from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.base import LLMBase
 
 
 class OpenAILLM(LLMBase):

mem0/llms/openai_structured.py

@@ -1,10 +1,11 @@
-import os, json
+import json
+import os
 from typing import Dict, List, Optional
 
 from openai import OpenAI
 
-from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.base import LLMBase
 
 
 class OpenAIStructuredLLM(LLMBase):

mem0/llms/together.py

@@ -1,7 +1,5 @@
-import subprocess
-import sys
-import os
 import json
+import os
 from typing import Dict, List, Optional
 
 try:
@@ -9,8 +7,8 @@ try:
 except ImportError:
     raise ImportError("The 'together' library is required. Please install it using 'pip install together'.")
 
-from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig
+from mem0.llms.base import LLMBase
 
 
 class TogetherLLM(LLMBase):

mem0/memory/graph_memory.py

@@ -1,10 +1,17 @@
-import json
 import logging
+
 from langchain_community.graphs import Neo4jGraph
 from rank_bm25 import BM25Okapi
-from mem0.utils.factory import LlmFactory, EmbedderFactory
-from mem0.graphs.utils import get_update_memory_messages, EXTRACT_ENTITIES_PROMPT
-from mem0.graphs.tools import UPDATE_MEMORY_TOOL_GRAPH, ADD_MEMORY_TOOL_GRAPH, NOOP_TOOL, ADD_MESSAGE_TOOL, SEARCH_TOOL
+
+from mem0.graphs.tools import (
+    ADD_MEMORY_TOOL_GRAPH,
+    ADD_MESSAGE_TOOL,
+    NOOP_TOOL,
+    SEARCH_TOOL,
+    UPDATE_MEMORY_TOOL_GRAPH,
+)
+from mem0.graphs.utils import EXTRACT_ENTITIES_PROMPT, get_update_memory_messages
+from mem0.utils.factory import EmbedderFactory, LlmFactory
 
 logger = logging.getLogger(__name__)
 

mem0/memory/main.py

@@ -1,22 +1,24 @@
-import logging
+import concurrent
 import hashlib
-import uuid
-import pytz
 import json
+import logging
+import threading
+import uuid
+import warnings
 from datetime import datetime
 from typing import Any, Dict
-import warnings
+
+import pytz
 from pydantic import ValidationError
+
+from mem0.configs.base import MemoryConfig, MemoryItem
+from mem0.configs.prompts import get_update_memory_messages
 from mem0.memory.base import MemoryBase
 from mem0.memory.setup import setup_config
 from mem0.memory.storage import SQLiteManager
 from mem0.memory.telemetry import capture_event
 from mem0.memory.utils import get_fact_retrieval_messages, parse_messages
-from mem0.configs.prompts import get_update_memory_messages
-from mem0.utils.factory import LlmFactory, EmbedderFactory, VectorStoreFactory
-from mem0.configs.base import MemoryItem, MemoryConfig
-import threading
-import concurrent
+from mem0.utils.factory import EmbedderFactory, LlmFactory, VectorStoreFactory
 
 # Setup user config
 setup_config()
@@ -154,7 +156,7 @@ class Memory(MemoryBase):
                 logging.info(resp)
                 try:
                     if resp["event"] == "ADD":
-                        memory_id = self._create_memory(data=resp["text"], metadata=metadata)
+                        self._create_memory(data=resp["text"], metadata=metadata)
                     elif resp["event"] == "UPDATE":
                         self._update_memory(memory_id=resp["id"], data=resp["text"], metadata=metadata)
                     elif resp["event"] == "DELETE":
@@ -175,7 +177,7 @@ class Memory(MemoryBase):
             else:
                 self.graph.user_id = "USER"
             data = "\n".join([msg["content"] for msg in messages if "content" in msg and msg["role"] != "system"])
-            added_entities = self.graph.add(data, filters)
+            self.graph.add(data, filters)
 
     def get(self, memory_id):
         """

mem0/memory/telemetry.py

@@ -1,3 +1,4 @@
+import logging
 import platform
 import sys
 
@@ -5,6 +6,8 @@ from posthog import Posthog
 
 from mem0.memory.setup import get_user_id, setup_config
 
+logging.getLogger('posthog').setLevel(logging.CRITICAL + 1)
+logging.getLogger('urllib3').setLevel(logging.CRITICAL + 1)
 
 class AnonymousTelemetry:
     def __init__(self, project_api_key, host):

mem0/proxy/main.py

@@ -1,9 +1,10 @@
 import logging
 import subprocess
 import sys
-import httpx
-from typing import Optional, List, Union
 import threading
+from typing import List, Optional, Union
+
+import httpx
 
 try:
     import litellm
@@ -20,9 +21,9 @@ except ImportError:
         raise ImportError("The required 'litellm' library is not installed.")
         sys.exit(1)
 
-from mem0.memory.telemetry import capture_client_event
 from mem0 import Memory, MemoryClient
 from mem0.configs.prompts import MEMORY_ANSWER_PROMPT
+from mem0.memory.telemetry import capture_client_event
 
 logger = logging.getLogger(__name__)
 

mem0/utils/factory.py

@@ -1,7 +1,7 @@
 import importlib
 
-from mem0.configs.llms.base import BaseLlmConfig
 from mem0.configs.embeddings.base import BaseEmbedderConfig
+from mem0.configs.llms.base import BaseLlmConfig
 
 
 def load_class(class_type):

mem0/vector_stores/chroma.py

@@ -1,7 +1,5 @@
-import subprocess
-import sys
 import logging
-from typing import Optional, List, Dict
+from typing import Dict, List, Optional
 
 from pydantic import BaseModel
 

mem0/vector_stores/configs.py

@@ -1,4 +1,5 @@
-from typing import Optional, Dict
+from typing import Dict, Optional
+
 from pydantic import BaseModel, Field, model_validator
 
 

mem0/vector_stores/pgvector.py

@@ -1,8 +1,7 @@
-import subprocess
-import sys
 import json
 import logging
-from typing import Optional, List
+from typing import List, Optional
+
 from pydantic import BaseModel
 
 try:

mem0/vector_stores/qdrant.py

@@ -1,6 +1,6 @@
+import logging
 import os
 import shutil
-import logging
 
 from qdrant_client import QdrantClient
 from qdrant_client.models import (