mem0ai-azure-mysql 0.1.116.5__tar.gz → 0.1.116.7__tar.gz

{mem0ai_azure_mysql-0.1.116.5 → mem0ai_azure_mysql-0.1.116.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mem0ai-azure-mysql
-Version: 0.1.116.5
+Version: 0.1.116.7
 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.5 → mem0ai_azure_mysql-0.1.116.7}/mem0/memory/graph_memory.py

@@ -76,7 +76,7 @@ class MemoryGraph:
         self.threshold = self.config.graph_store.config.similarity_threshold
         self.top_k = self.config.graph_store.config.top_k
 
-    def add(self, data, filters):
+    def add(self, data, metadata, filters):
         """
         Adds data to the graph.
 
@@ -92,10 +92,11 @@ class MemoryGraph:
         # TODO: Batch queries with APOC plugin
         # TODO: Add more filter support
         deleted_entities = self._delete_entities(to_be_deleted, filters)
-        added_entities = self._add_entities(to_be_added, filters, entity_type_map)
+        added_entities = self._add_entities(to_be_added, metadata, filters, entity_type_map)
 
         return {"deleted_entities": deleted_entities, "added_entities": added_entities}
 
+    @time_perf
     def search(self, query, filters, limit=100):
         """
         Search for memories and related graph data.
@@ -128,6 +129,7 @@ class MemoryGraph:
                     "relationship": item[1],
                     "destination": item[2],
                     "score": item[3] if len(item) > 3 else 0.0,
+                    "metadata": item[4] if len(item) > 4 else {},
                 }
             )
 
@@ -139,15 +141,12 @@ class MemoryGraph:
 
         return search_results
 
+    @time_perf
     def rerank(self, search_output, query):
         if not self.config.graph_store.config.rerank:
-            search_outputs_sequence = [
-                [item["source"], item["relationship"], item["destination"]] for item in search_output
+            return [
+                [item["source"], item["relationship"], item["destination"], 0.0, item.get("metadata", {})] for item in search_output
             ]
-            bm25 = BM25Okapi(search_outputs_sequence)
-
-            tokenized_query = query.split(" ")
-            return bm25.get_top_n(tokenized_query, search_outputs_sequence, n=5)
         
         # Only cohere is supported for now
         rerank_config = self.config.graph_store.config.rerank.config
@@ -170,7 +169,8 @@ class MemoryGraph:
                 original_output["source"],
                 original_output["relationship"],
                 original_output["destination"],
-                result.relevance_score
+                result.relevance_score,
+                original_output["metadata"],
             ])
         return reranked_results
 
@@ -336,14 +336,14 @@ class MemoryGraph:
         CALL {{
             WITH n, row
             MATCH (n)-[r]->(m {self.node_label} {{{node_props_str}}})
-            RETURN n.name AS source, elementId(n) AS source_id, type(r) AS relationship, elementId(r) AS relation_id, m.name AS destination, elementId(m) AS destination_id
+            RETURN n.name AS source, elementId(n) AS source_id, n.note_ids AS source_note_ids, type(r) AS relationship, elementId(r) AS relation_id, m.name AS destination, elementId(m) AS destination_id, m.note_ids AS destination_note_ids
             UNION
             WITH n, row
             MATCH (n)<-[r]-(m {self.node_label} {{{node_props_str}}})
-            RETURN m.name AS source, elementId(m) AS source_id, type(r) AS relationship, elementId(r) AS relation_id, n.name AS destination, elementId(n) AS destination_id
+            RETURN m.name AS source, elementId(m) AS source_id, m.note_ids AS source_note_ids, type(r) AS relationship, elementId(r) AS relation_id, n.name AS destination, elementId(n) AS destination_id, n.note_ids AS destination_note_ids
         }}
-        WITH distinct source, source_id, relationship, relation_id, destination, destination_id, similarity
-        RETURN source, source_id, relationship, relation_id, destination, destination_id, similarity
+        WITH distinct source, source_id, relationship, relation_id, destination, destination_id, similarity, source_note_ids, destination_note_ids
+        RETURN source, source_id, relationship, relation_id, destination, destination_id, similarity, {{note_ids: source_note_ids + destination_note_ids}} AS metadata
         ORDER BY similarity DESC
         LIMIT {limit}
         """
@@ -371,6 +371,12 @@ class MemoryGraph:
             key = (rel["source"], rel["relationship"], rel["destination"])
             if key not in seen:
                 seen.add(key)
+
+                original_metadata = rel.get("metadata", {})
+                rel["metadata"] = {
+                    "note_ids": list(set(original_metadata.get("note_ids", []))),
+                }
+
                 unique_relations.append(rel)
 
         return unique_relations
@@ -469,7 +475,7 @@ class MemoryGraph:
         return results
 
     @time_perf
-    def _add_entities(self, to_be_added, filters, entity_type_map):
+    def _add_entities(self, to_be_added, metadata, filters, entity_type_map):
         """Add the new entities to the graph. Merge the nodes if they already exist."""
         user_id = filters["user_id"]
         agent_id = filters.get("agent_id", None)
@@ -484,29 +490,9 @@ class MemoryGraph:
         entities_to_embed = list(entities_to_embed)
         embeddings = self.embedding_model.embed_in_batch(entities_to_embed)
         embedding_map = {entity: embedding for entity, embedding in zip(entities_to_embed, embeddings)}
-
-        query_stats = {
-                1: 0, 2: 0, 3: 0, 4: 0
-            }
     
-        queries = {
-            "only_source_found": {
-                "template": "",
-                "rows": []
-            },
-            "only_destination_found": {
-                "template": "",
-                "rows": []
-            },
-            "both_found": {
-                "template": "",
-                "rows": []
-            },
-            "none_found": {
-                "template": "",
-                "rows": []
-            },
-        }
+        # metadata
+        note_ids = metadata.pop("note_ids", [])
 
         for item in to_be_added:
             # entities
@@ -544,14 +530,17 @@ class MemoryGraph:
                 MATCH (source)
                 WHERE elementId(source) = $source_id
                 SET source.mentions = coalesce(source.mentions, 0) + 1
+                SET source.note_ids = coalesce(source.note_ids, []) + $note_ids
                 WITH source
                 MERGE (destination {destination_label} {{{merge_props_str}}})
                 ON CREATE SET
                     destination.created = timestamp(),
-                    destination.mentions = 1
+                    destination.mentions = 1,
+                    destination.note_ids = $note_ids
                     {destination_extra_set}
                 ON MATCH SET
-                    destination.mentions = coalesce(destination.mentions, 0) + 1
+                    destination.mentions = coalesce(destination.mentions, 0) + 1,
+                    destination.note_ids = coalesce(destination.note_ids, []) + $note_ids
                 WITH source, destination
                 CALL db.create.setNodeVectorProperty(destination, 'embedding', $destination_embedding)
                 WITH source, destination
@@ -569,13 +558,12 @@ class MemoryGraph:
                     "destination_name": destination,
                     "destination_embedding": dest_embedding,
                     "user_id": user_id,
+                    "note_ids": note_ids,
                 }
                 if agent_id:
                     params["agent_id"] = agent_id
                 if run_id:
                     params["run_id"] = run_id
-                
-                query_stats[1] += 1
 
             elif destination_node_search_result and not source_node_search_result:
                 # Build source MERGE properties
@@ -590,14 +578,17 @@ class MemoryGraph:
                 MATCH (destination)
                 WHERE elementId(destination) = $destination_id
                 SET destination.mentions = coalesce(destination.mentions, 0) + 1
+                SET destination.note_ids = coalesce(destination.note_ids, []) + $note_ids
                 WITH destination
                 MERGE (source {source_label} {{{merge_props_str}}})
                 ON CREATE SET
                     source.created = timestamp(),
-                    source.mentions = 1
+                    source.mentions = 1,
+                    source.note_ids = $note_ids
                     {source_extra_set}
                 ON MATCH SET
-                    source.mentions = coalesce(source.mentions, 0) + 1
+                    source.mentions = coalesce(source.mentions, 0) + 1,
+                    source.note_ids = coalesce(source.note_ids, []) + $note_ids
                 WITH source, destination
                 CALL db.create.setNodeVectorProperty(source, 'embedding', $source_embedding)
                 WITH source, destination
@@ -615,23 +606,24 @@ class MemoryGraph:
                     "source_name": source,
                     "source_embedding": source_embedding,
                     "user_id": user_id,
+                    "note_ids": note_ids,
                 }
                 if agent_id:
                     params["agent_id"] = agent_id
                 if run_id:
                     params["run_id"] = run_id
-                
-                query_stats[2] += 1
 
             elif source_node_search_result and destination_node_search_result:
                 cypher = f"""
                 MATCH (source)
                 WHERE elementId(source) = $source_id
                 SET source.mentions = coalesce(source.mentions, 0) + 1
+                SET source.note_ids = coalesce(source.note_ids, []) + $note_ids
                 WITH source
                 MATCH (destination)
                 WHERE elementId(destination) = $destination_id
                 SET destination.mentions = coalesce(destination.mentions, 0) + 1
+                SET destination.note_ids = coalesce(destination.note_ids, []) + $note_ids
                 MERGE (source)-[r:{relationship}]->(destination)
                 ON CREATE SET 
                     r.created_at = timestamp(),
@@ -645,13 +637,12 @@ class MemoryGraph:
                     "source_id": source_node_search_result[0]["elementId(source_candidate)"],
                     "destination_id": destination_node_search_result[0]["elementId(destination_candidate)"],
                     "user_id": user_id,
+                    "note_ids": note_ids,
                 }
                 if agent_id:
                     params["agent_id"] = agent_id
                 if run_id:
                     params["run_id"] = run_id
-                
-                query_stats[3] += 1
 
             else:
                 # Build dynamic MERGE props for both source and destination
@@ -669,7 +660,8 @@ class MemoryGraph:
                 cypher = f"""
                 MERGE (source {source_label} {{{source_props_str}}})
                 ON CREATE SET source.created = timestamp(),
-                            source.mentions = 1
+                            source.mentions = 1,
+                            source.note_ids = $note_ids
                             {source_extra_set}
                 ON MATCH SET source.mentions = coalesce(source.mentions, 0) + 1
                 WITH source
@@ -677,7 +669,8 @@ class MemoryGraph:
                 WITH source
                 MERGE (destination {destination_label} {{{dest_props_str}}})
                 ON CREATE SET destination.created = timestamp(),
-                            destination.mentions = 1
+                            destination.mentions = 1,
+                            destination.note_ids = $note_ids
                             {destination_extra_set}
                 ON MATCH SET destination.mentions = coalesce(destination.mentions, 0) + 1
                 WITH source, destination
@@ -695,17 +688,14 @@ class MemoryGraph:
                     "source_embedding": source_embedding,
                     "dest_embedding": dest_embedding,
                     "user_id": user_id,
+                    "note_ids": note_ids,
                 }
                 if agent_id:
                     params["agent_id"] = agent_id
                 if run_id:
                     params["run_id"] = run_id
-                
-                query_stats[4] += 1
-
             result = self.graph.query(cypher, params=params)
             results.append(result)
-        logger.warning(f"Add entities query stats: {query_stats}")
         return results
 
     def _remove_spaces_from_entities(self, entity_list):

{mem0ai_azure_mysql-0.1.116.5 → mem0ai_azure_mysql-0.1.116.7}/mem0/memory/main.py

@@ -258,7 +258,7 @@ class Memory(MemoryBase):
 
         with concurrent.futures.ThreadPoolExecutor() as executor:
             future1 = executor.submit(self._add_to_vector_store, messages, processed_metadata, effective_filters, infer)
-            future2 = executor.submit(self._add_to_graph, messages, effective_filters)
+            future2 = executor.submit(self._add_to_graph, messages, processed_metadata, effective_filters)
 
             concurrent.futures.wait([future1, future2])
 
@@ -454,14 +454,14 @@ class Memory(MemoryBase):
         return returned_memories
 
     @time_perf
-    def _add_to_graph(self, messages, filters):
+    def _add_to_graph(self, messages, metadata, filters):
         added_entities = []
         if self.enable_graph:
             if filters.get("user_id") is None:
                 filters["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)
+            added_entities = self.graph.add(data, metadata, filters)
 
         return added_entities
 
@@ -1100,7 +1100,7 @@ class AsyncMemory(MemoryBase):
         vector_store_task = asyncio.create_task(
             self._add_to_vector_store(messages, processed_metadata, effective_filters, infer)
         )
-        graph_task = asyncio.create_task(self._add_to_graph(messages, effective_filters))
+        graph_task = asyncio.create_task(self._add_to_graph(messages, processed_metadata, effective_filters))
 
         vector_store_result, graph_result = await asyncio.gather(vector_store_task, graph_task)
 
@@ -1308,14 +1308,14 @@ class AsyncMemory(MemoryBase):
         return returned_memories
 
     @time_perf
-    async def _add_to_graph(self, messages, filters):
+    async def _add_to_graph(self, messages, metadata, filters):
         added_entities = []
         if self.enable_graph:
             if filters.get("user_id") is None:
                 filters["user_id"] = "user"
 
             data = "\n".join([msg["content"] for msg in messages if "content" in msg and msg["role"] != "system"])
-            added_entities = await asyncio.to_thread(self.graph.add, data, filters)
+            added_entities = await asyncio.to_thread(self.graph.add, data, metadata, filters)
 
         return added_entities
 
@@ -1553,6 +1553,7 @@ class AsyncMemory(MemoryBase):
         else:
             return {"results": original_memories}
 
+    @time_perf
     async def _search_vector_store(self, query, filters, limit, threshold: Optional[float] = None):
         embeddings = await asyncio.to_thread(self.embedding_model.embed, query, "search")
         memories = await asyncio.to_thread(

{mem0ai_azure_mysql-0.1.116.5 → mem0ai_azure_mysql-0.1.116.7}/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"[{func.__name__}] executed in {elapsed:.6f} seconds")
+            logger.warning(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"[{func.__name__}] executed in {elapsed:.6f} seconds")
+            logger.warning(f"[Mem0 time perf][{func.__name__}] executed in {elapsed:.6f} seconds")
             return result
         return sync_wrapper

{mem0ai_azure_mysql-0.1.116.5 → mem0ai_azure_mysql-0.1.116.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "mem0ai-azure-mysql"
-version = "0.1.116.5"
+version = "0.1.116.7"
 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"