mem0ai-azure-mysql 0.1.116.8__tar.gz → 0.1.116.10__tar.gz

{mem0ai_azure_mysql-0.1.116.8 → mem0ai_azure_mysql-0.1.116.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mem0ai-azure-mysql
-Version: 0.1.116.8
+Version: 0.1.116.10
 Summary: Long-term memory for AI Agents with Azure DefaultAzureCredential authentication and MySQL history database support
 Requires-Python: <4.0,>=3.9
 Requires-Dist: azure-identity>=1.23.1

{mem0ai_azure_mysql-0.1.116.8 → mem0ai_azure_mysql-0.1.116.10}/mem0/memory/graph_memory.py

@@ -374,7 +374,7 @@ class MemoryGraph:
 
                 original_metadata = rel.get("metadata", {})
                 rel["metadata"] = {
-                    "note_ids": list(set(original_metadata.get("note_ids", []))),
+                    "note_ids": list(set(original_metadata.get("note_ids") or [])),
                 }
 
                 unique_relations.append(rel)

{mem0ai_azure_mysql-0.1.116.8 → mem0ai_azure_mysql-0.1.116.10}/mem0/memory/main.py

@@ -113,6 +113,9 @@ def _build_filters_and_metadata(
     if resolved_actor_id:
         effective_query_filters["actor_id"] = resolved_actor_id
 
+    # set default source
+    base_metadata_template['source'] = 'vector'
+
     return base_metadata_template, effective_query_filters
 
 
@@ -1050,6 +1053,7 @@ class AsyncMemory(MemoryBase):
         memory_type: Optional[str] = None,
         prompt: Optional[str] = None,
         llm=None,
+        add_graph_results_to_vector_store: bool = False,
     ):
         """
         Create a new memory asynchronously.
@@ -1065,6 +1069,7 @@ class AsyncMemory(MemoryBase):
                                          Pass "procedural_memory" to create procedural memories.
             prompt (str, optional): Prompt to use for the memory creation. Defaults to None.
             llm (BaseChatModel, optional): LLM class to use for generating procedural memories. Defaults to None. Useful when user is using LangChain ChatModel.
+            add_graph_results_to_vector_store (bool, optional): Whether to add graph results to vector store. Defaults to False.
         Returns:
             dict: A dictionary containing the result of the memory addition operation.
         """
@@ -1104,6 +1109,10 @@ class AsyncMemory(MemoryBase):
 
         vector_store_result, graph_result = await asyncio.gather(vector_store_task, graph_task)
 
+        if add_graph_results_to_vector_store:
+            vector_store_result_from_graph = await self._add_graph_results_to_vector_store(graph_result, processed_metadata, effective_filters, infer)
+            vector_store_result.extend(vector_store_result_from_graph)
+
         if self.api_version == "v1.0":
             warnings.warn(
                 "The current add API output format is deprecated. "
@@ -1122,6 +1131,31 @@ class AsyncMemory(MemoryBase):
 
         return {"results": vector_store_result}
 
+    async def _add_graph_results_to_vector_store(self, graph_results, metadata, filter, infer):
+        if not graph_results:
+            return []
+        
+        added_entities = graph_results.get("added_entities", [])
+        messages = []
+        for entities in added_entities:
+            for entity in entities:
+                source = entity["source"]
+                # Avoid duplicate user_id data in memory
+                if source.startswith("user_id:"):
+                    source = ""
+                relationship = entity["relationship"]
+                target = entity["target"]
+                entity_data = f"{source} {relationship} {target}"
+                message = {"role": "user", "content": entity_data}
+                messages.append(message)
+
+        if messages:
+            copied_metadata = deepcopy(metadata) if metadata else {}
+            copied_metadata['source'] = 'graph'
+            return await self._add_to_vector_store(messages, copied_metadata, filter, infer)
+    
+        return []
+
     @time_perf
     async def _add_to_vector_store(
         self,
@@ -1484,6 +1518,7 @@ class AsyncMemory(MemoryBase):
         limit: int = 100,
         filters: Optional[Dict[str, Any]] = None,
         threshold: Optional[float] = None,
+        use_graph: Optional[bool] = None,
     ):
         """
         Searches for memories based on a query
@@ -1501,6 +1536,8 @@ class AsyncMemory(MemoryBase):
                   and potentially "relations" if graph store is enabled.
                   Example for v1.1+: `{"results": [{"id": "...", "memory": "...", "score": 0.8, ...}]}`
         """
+        if use_graph is None:
+            use_graph = self.enable_graph
 
         _, effective_filters = _build_filters_and_metadata(
             user_id=user_id, agent_id=agent_id, run_id=run_id, input_filters=filters
@@ -1526,7 +1563,7 @@ class AsyncMemory(MemoryBase):
         vector_store_task = asyncio.create_task(self._search_vector_store(query, effective_filters, limit, threshold))
 
         graph_task = None
-        if self.enable_graph:
+        if use_graph:
             if hasattr(self.graph.search, "__await__"):  # Check if graph search is async
                 graph_task = asyncio.create_task(self.graph.search(query, effective_filters, limit))
             else:
@@ -1538,7 +1575,7 @@ class AsyncMemory(MemoryBase):
             original_memories = await vector_store_task
             graph_entities = None
 
-        if self.enable_graph:
+        if use_graph:
             return {"results": original_memories, "relations": graph_entities}
 
         if self.api_version == "v1.0":

{mem0ai_azure_mysql-0.1.116.8 → mem0ai_azure_mysql-0.1.116.10}/mem0/memory/utils.py

@@ -197,7 +197,7 @@ def time_perf(func):
             result = await func(*args, **kwargs)
             end = time.perf_counter()
             elapsed = end - start
-            logger.warning(f"[Mem0 time perf][{func.__name__}] executed in {elapsed:.6f} seconds")
+            logger.debug(f"[Mem0 time perf][{func.__name__}] executed in {elapsed:.6f} seconds")
             return result
         return async_wrapper
     else:
@@ -207,6 +207,6 @@ def time_perf(func):
             result = func(*args, **kwargs)
             end = time.perf_counter()
             elapsed = end - start
-            logger.warning(f"[Mem0 time perf][{func.__name__}] executed in {elapsed:.6f} seconds")
+            logger.debug(f"[Mem0 time perf][{func.__name__}] executed in {elapsed:.6f} seconds")
             return result
         return sync_wrapper

{mem0ai_azure_mysql-0.1.116.8 → mem0ai_azure_mysql-0.1.116.10}/mem0/vector_stores/vertex_ai_vector_search.py

@@ -18,7 +18,6 @@ from mem0.configs.vector_stores.vertex_ai_vector_search import (
 from mem0.vector_stores.base import VectorStoreBase
 
 # Configure logging
-logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
 

{mem0ai_azure_mysql-0.1.116.8 → mem0ai_azure_mysql-0.1.116.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "mem0ai-azure-mysql"
-version = "0.1.116.8"
+version = "0.1.116.10"
 description = "Long-term memory for AI Agents with Azure DefaultAzureCredential authentication and MySQL history database support"
 readme = "README.md"
 requires-python = ">=3.9,<4.0"