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:
|
def _format_search_results(self, results: List[Dict]) -> str:
|
||||||
"""Format search results into a bullet list string."""
|
"""Format search results into a bullet list string with IDs."""
|
||||||
lines = [
|
lines = []
|
||||||
r.get("text") or r.get("memory", "")
|
for r in results:
|
||||||
for r in results
|
text = r.get("text") or r.get("memory", "")
|
||||||
if 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 ""
|
return "\n".join(f"- {line}" for line in lines) if lines else ""
|
||||||
|
|
||||||
def initialize(self, session_id: str, **kwargs) -> None:
|
def initialize(self, session_id: str, **kwargs) -> None:
|
||||||
@@ -471,7 +475,15 @@ class Mem0LocalMemoryProvider(MemoryProvider):
|
|||||||
self._record_success()
|
self._record_success()
|
||||||
if not memories:
|
if not memories:
|
||||||
return json.dumps({"result": "No memories stored yet."})
|
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)})
|
return json.dumps({"result": "\n".join(lines), "count": len(lines)})
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self._record_failure()
|
self._record_failure()
|
||||||
@@ -493,9 +505,12 @@ class Mem0LocalMemoryProvider(MemoryProvider):
|
|||||||
if not results:
|
if not results:
|
||||||
return json.dumps({"result": "No relevant memories found."})
|
return json.dumps({"result": "No relevant memories found."})
|
||||||
items = [
|
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
|
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)})
|
return json.dumps({"results": items, "count": len(items)})
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self._record_failure()
|
self._record_failure()
|
||||||
|
|||||||
Reference in New Issue
Block a user