mem0ai-azure-mysql 0.1.116.2__py3-none-any.whl → 0.1.116.3__py3-none-any.whl

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.dist-info → mem0ai_azure_mysql-0.1.116.3.dist-info}/METADATA

@@ -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.dist-info → mem0ai_azure_mysql-0.1.116.3.dist-info}/RECORD

@@ -88,7 +88,7 @@ mem0/llms/vllm.py,sha256=vVrz7uGjNqVgX3YNBLWNWA4ctEVK9sZxxaOOKYRQQdM,3802
 mem0/llms/xai.py,sha256=pUdxDH0-p1fVct66IMMkMl3Rgcdm1d4ko2aOT6feYWU,1764
 mem0/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mem0/memory/base.py,sha256=uwk-rL60CbErBDt9-bMVK4laALtc4wd2oeVskVzxd3c,1315
-mem0/memory/graph_memory.py,sha256=xjFqkXa0lhAgHlIwdsMu11jT34dZG81VOCmlflbaqTk,32019
+mem0/memory/graph_memory.py,sha256=45OxECfg-wj17nRSWJ0RN5vim91skY3SPCO4jwK9opM,32605
 mem0/memory/kuzu_memory.py,sha256=ootVNrq9KLw3uV-MdCuC-ZjAHiafWOi0AVpjkcGDFOI,30338
 mem0/memory/main.py,sha256=8BmnRKIeRxO5hZ9tivEDjv087JYYpbP1EtLR0H_Syn0,76818
 mem0/memory/memgraph_memory.py,sha256=Kg8sk-YSCK2bR9qHcRWPkuBqMugHznD34qLpQjjgZ1E,27870
@@ -120,7 +120,7 @@ mem0/vector_stores/supabase.py,sha256=RCegadYW4Q8CT_NSeCywUVW01TLJQT4mDzeStNHEJH
 mem0/vector_stores/upstash_vector.py,sha256=dAIef-bIB6AE_dVR8UpVWZUPQs-wLhqfxEHyyXO5kQU,9073
 mem0/vector_stores/vertex_ai_vector_search.py,sha256=RaaPLTTbTPftlg7nm9WTROOhurSxqnAyx-g3tYoxv0I,24659
 mem0/vector_stores/weaviate.py,sha256=v91CA79Z_zJBohwaXlOW6jtudDlGq0kYKQFHMHGqUy4,11833
-mem0ai_azure_mysql-0.1.116.2.data/data/README.md,sha256=1kzUMONek69u8WrI7zxwFevzYBYiI7-1NTLjSl8_6kg,1235
-mem0ai_azure_mysql-0.1.116.2.dist-info/METADATA,sha256=Vv4AcU-LaxIyGcrHm63MSZpGN0rTgJitYsn0ITdyAgg,4155
-mem0ai_azure_mysql-0.1.116.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mem0ai_azure_mysql-0.1.116.2.dist-info/RECORD,,
+mem0ai_azure_mysql-0.1.116.3.data/data/README.md,sha256=1kzUMONek69u8WrI7zxwFevzYBYiI7-1NTLjSl8_6kg,1235
+mem0ai_azure_mysql-0.1.116.3.dist-info/METADATA,sha256=TbxK0ZR_1-XLew5KMypKxALUt_XaPh4Uv7TfyCFJg2o,4155
+mem0ai_azure_mysql-0.1.116.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mem0ai_azure_mysql-0.1.116.3.dist-info/RECORD,,