Formatting and version bump -> 0.1.82 (#2507)

Author: Dev-Khant

mem0/memory/graph_memory.py

@@ -167,7 +167,9 @@ def _retrieve_nodes_from_data(self, data, filters):
             for item in search_results["tool_calls"][0]["arguments"]["entities"]:
                 entity_type_map[item["entity"]] = item["entity_type"]
         except Exception as e:
-            logger.exception(f"Error in search tool: {e}, llm_provider={self.llm_provider}, search_results={search_results}")
+            logger.exception(
+                f"Error in search tool: {e}, llm_provider={self.llm_provider}, search_results={search_results}"
+            )
 
         entity_type_map = {k.lower().replace(" ", "_"): v.lower().replace(" ", "_") for k, v in entity_type_map.items()}
         logger.debug(f"Entity type map: {entity_type_map}")

mem0/memory/main.py

@@ -12,14 +12,12 @@
 
 from mem0.configs.base import MemoryConfig, MemoryItem
 from mem0.configs.enums import MemoryType
-from mem0.configs.prompts import (PROCEDURAL_MEMORY_SYSTEM_PROMPT,
-                                  get_update_memory_messages)
+from mem0.configs.prompts import PROCEDURAL_MEMORY_SYSTEM_PROMPT, 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,
-                               parse_vision_messages, remove_code_blocks)
+from mem0.memory.utils import get_fact_retrieval_messages, parse_messages, parse_vision_messages, remove_code_blocks
 from mem0.utils.factory import EmbedderFactory, LlmFactory, VectorStoreFactory
 
 # Setup user config
@@ -638,7 +636,9 @@ def _create_procedural_memory(self, messages, metadata=None, llm=None, prompt=No
         try:
             from langchain_core.messages.utils import convert_to_messages  # type: ignore
         except Exception:
-            logger.error("Import error while loading langchain-core. Please install 'langchain-core' to use procedural memory.")
+            logger.error(
+                "Import error while loading langchain-core. Please install 'langchain-core' to use procedural memory."
+            )
             raise
 
         logger.info("Creating procedural memory")

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mem0ai"
-version = "0.1.81"
+version = "0.1.82"
 description = "Long-term memory for AI Agents"
 authors = ["Mem0 <[email protected]>"]
 exclude = [
@@ -30,7 +30,6 @@ sqlalchemy = "^2.0.31"
 langchain-neo4j = "^0.4.0"
 neo4j = "^5.23.1"
 rank-bm25 = "^0.2.2"
-azure-search-documents = "^11.5.0"
 psycopg2-binary = "^2.9.10"
 
 [tool.poetry.extras]