refactor: v2 search and update examples (#3508)

This commit is contained in:
Parshva Daftari
2025-09-26 22:55:53 +05:30
committed by GitHub
parent ed5a1e9fc6
commit 135883935f
16 changed files with 21 additions and 27 deletions
+2 -2
View File
@@ -65,7 +65,7 @@ Create functions to handle context retrieval, response generation, and addition
def retrieve_context(query: str, user_id: str) -> List[Dict]:
"""Retrieve relevant context from Mem0"""
try:
memories = mem0.search(query, user_id=user_id, output_format='v1.1')
memories = mem0.search(query, user_id=user_id)
memory_list = memories['results']
serialized_memories = ' '.join([mem["memory"] for mem in memory_list])
@@ -107,7 +107,7 @@ def save_interaction(user_id: str, user_input: str, assistant_response: str):
"content": assistant_response
}
]
result = mem0.add(interaction, user_id=user_id, output_format='v1.1')
result = mem0.add(interaction, user_id=user_id)
print(f"Memory saved successfully: {len(result.get('results', []))} memories added")
except Exception as e:
print(f"Error saving interaction: {e}")