mem0ai-azure-mysql 0.1.116.2__tar.gz → 0.1.116.3__tar.gz

{mem0ai_azure_mysql-0.1.116.2 → mem0ai_azure_mysql-0.1.116.3}/PKG-INFO

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

@@ -114,6 +114,8 @@ class MemoryGraph:
         logger.info(f"Searching for query in neo4j: {query} with filters: {filters}")
         entity_type_map = self._retrieve_nodes_from_data(query, filters)
         search_output = self._search_graph_db(node_list=list(entity_type_map.keys()), filters=filters)
+        print(f'entity_type_map: {entity_type_map}')
+        print(f'search_output: {search_output}')
 
         if not search_output:
             return []
@@ -156,12 +158,14 @@ class MemoryGraph:
             base_url=rerank_config.get("base_url"),
         )
         docs = [f'{item["source"]} {item["relationship"]} {item["destination"]}' for item in search_output]
+        print(f'docs: {docs}')
         response = co.rerank(
             model=rerank_config.get("model"),
             query=query,
             documents=docs,
             top_n=self.top_k,
         )
+        print(f'response.results: {response.results}')
 
         reranked_results = []
         for result in response.results:
@@ -172,6 +176,7 @@ class MemoryGraph:
                 original_output["destination"],
                 result.relevance_score
             ])
+        print(f'reranked_results: {reranked_results}')
         return reranked_results
 
     def delete_all(self, filters):
@@ -362,7 +367,16 @@ class MemoryGraph:
             ans = self.graph.query(cypher_query, params=params)
             result_relations.extend(ans)
 
-        return result_relations
+        # Remove duplicates, key is (source, relationship, destination)
+        seen = set()
+        unique_relations = []
+        for rel in result_relations:
+            key = (rel["source"], rel["relationship"], rel["destination"])
+            if key not in seen:
+                seen.add(key)
+                unique_relations.append(rel)
+
+        return unique_relations
 
     def _get_delete_entities_from_search_output(self, search_output, data, filters):
         """Get the entities to be deleted from the search output."""

{mem0ai_azure_mysql-0.1.116.2 → mem0ai_azure_mysql-0.1.116.3}/pyproject.toml

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