Compare commits
2 Commits
f97cf9a551
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
| b69efe9482 | |||
| 4da384e68f |
+23
-8
@@ -284,12 +284,16 @@ class Mem0LocalMemoryProvider(MemoryProvider):
|
||||
)
|
||||
|
||||
def _format_search_results(self, results: List[Dict]) -> str:
|
||||
"""Format search results into a bullet list string."""
|
||||
lines = [
|
||||
r.get("text") or r.get("memory", "")
|
||||
for r in results
|
||||
if r.get("text") or r.get("memory")
|
||||
]
|
||||
"""Format search results into a bullet list string with IDs."""
|
||||
lines = []
|
||||
for r in results:
|
||||
text = r.get("text") or r.get("memory", "")
|
||||
if text:
|
||||
mem_id = r.get("id", "")
|
||||
if mem_id:
|
||||
lines.append(f"[{mem_id}] {text}")
|
||||
else:
|
||||
lines.append(text)
|
||||
return "\n".join(f"- {line}" for line in lines) if lines else ""
|
||||
|
||||
def initialize(self, session_id: str, **kwargs) -> None:
|
||||
@@ -471,7 +475,15 @@ class Mem0LocalMemoryProvider(MemoryProvider):
|
||||
self._record_success()
|
||||
if not memories:
|
||||
return json.dumps({"result": "No memories stored yet."})
|
||||
lines = [m.get("text", "") for m in memories if m.get("text")]
|
||||
lines = []
|
||||
for m in memories:
|
||||
text = m.get("text", "")
|
||||
if text:
|
||||
mem_id = m.get("id", "")
|
||||
if mem_id:
|
||||
lines.append(f"[{mem_id}] {text}")
|
||||
else:
|
||||
lines.append(text)
|
||||
return json.dumps({"result": "\n".join(lines), "count": len(lines)})
|
||||
except Exception as e:
|
||||
self._record_failure()
|
||||
@@ -493,9 +505,12 @@ class Mem0LocalMemoryProvider(MemoryProvider):
|
||||
if not results:
|
||||
return json.dumps({"result": "No relevant memories found."})
|
||||
items = [
|
||||
{"memory": r.get("text", ""), "score": r.get("score", 0)}
|
||||
{"id": r.get("id", ""), "memory": r.get("text", ""), "score": r.get("score", 0)}
|
||||
for r in results
|
||||
if r.get("text")
|
||||
]
|
||||
if not items:
|
||||
return json.dumps({"result": "No relevant memories found."})
|
||||
return json.dumps({"results": items, "count": len(items)})
|
||||
except Exception as e:
|
||||
self._record_failure()
|
||||
|
||||
Reference in New Issue
Block a user