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

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

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

@@ -96,6 +96,7 @@ class MemoryGraph:
 
         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.
@@ -139,6 +140,7 @@ class MemoryGraph:
 
         return search_results
 
+    @time_perf
     def rerank(self, search_output, query):
         if not self.config.graph_store.config.rerank:
             search_outputs_sequence = [
@@ -485,29 +487,6 @@ class MemoryGraph:
         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": []
-            },
-        }
-
         for item in to_be_added:
             # entities
             source = item["source"]
@@ -574,8 +553,6 @@ class MemoryGraph:
                     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
@@ -620,8 +597,6 @@ class MemoryGraph:
                     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"""
@@ -650,8 +625,6 @@ class MemoryGraph:
                     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
@@ -700,12 +673,8 @@ class MemoryGraph:
                     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.6}/mem0/memory/main.py

@@ -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.6}/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.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"[{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.5 → mem0ai_azure_mysql-0.1.116.6}/pyproject.toml

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