Code Formatting (#1828)

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