refactor: rename tools to LLM-friendly names
mem0_profile → mem0_list_all (clearer intent) mem0_conclude → mem0_save_memory (self-explanatory)
This commit is contained in:
+6
-6
@@ -75,7 +75,7 @@ def _load_config() -> dict:
|
|||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
|
|
||||||
PROFILE_SCHEMA = {
|
PROFILE_SCHEMA = {
|
||||||
"name": "mem0_profile",
|
"name": "mem0_list_all",
|
||||||
"description": (
|
"description": (
|
||||||
"Retrieve all stored memories about the user — preferences, facts, "
|
"Retrieve all stored memories about the user — preferences, facts, "
|
||||||
"project context. Fast, no reranking. Use at conversation start."
|
"project context. Fast, no reranking. Use at conversation start."
|
||||||
@@ -107,7 +107,7 @@ SEARCH_SCHEMA = {
|
|||||||
}
|
}
|
||||||
|
|
||||||
CONCLUDE_SCHEMA = {
|
CONCLUDE_SCHEMA = {
|
||||||
"name": "mem0_conclude",
|
"name": "mem0_save_memory",
|
||||||
"description": (
|
"description": (
|
||||||
"Store a durable fact about the user. Stored verbatim (no LLM extraction). "
|
"Store a durable fact about the user. Stored verbatim (no LLM extraction). "
|
||||||
"Use for explicit preferences, corrections, or decisions."
|
"Use for explicit preferences, corrections, or decisions."
|
||||||
@@ -310,8 +310,8 @@ class Mem0LocalMemoryProvider(MemoryProvider):
|
|||||||
return (
|
return (
|
||||||
"# Mem0 Memory (Local)\n"
|
"# Mem0 Memory (Local)\n"
|
||||||
f"Active. User: {self._user_id}.\n"
|
f"Active. User: {self._user_id}.\n"
|
||||||
"Use mem0_search to find memories, mem0_conclude to store facts, "
|
"Use mem0_search to find memories, mem0_save_memory to store facts, "
|
||||||
"mem0_profile for a full overview.\n"
|
"mem0_list_all for a full overview.\n"
|
||||||
"\n"
|
"\n"
|
||||||
"## Memory Context Format\n"
|
"## Memory Context Format\n"
|
||||||
"Retrieved memories are injected via the <mem0_context> XML tag. "
|
"Retrieved memories are injected via the <mem0_context> XML tag. "
|
||||||
@@ -465,7 +465,7 @@ class Mem0LocalMemoryProvider(MemoryProvider):
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
return tool_error(str(e))
|
return tool_error(str(e))
|
||||||
|
|
||||||
if tool_name == "mem0_profile":
|
if tool_name == "mem0_list_all":
|
||||||
try:
|
try:
|
||||||
memories = client.get_all(user_id=self._user_id)
|
memories = client.get_all(user_id=self._user_id)
|
||||||
self._record_success()
|
self._record_success()
|
||||||
@@ -501,7 +501,7 @@ class Mem0LocalMemoryProvider(MemoryProvider):
|
|||||||
self._record_failure()
|
self._record_failure()
|
||||||
return tool_error(f"Search failed: {e}")
|
return tool_error(f"Search failed: {e}")
|
||||||
|
|
||||||
elif tool_name == "mem0_conclude":
|
elif tool_name == "mem0_save_memory":
|
||||||
conclusion = args.get("conclusion", "")
|
conclusion = args.get("conclusion", "")
|
||||||
if not conclusion:
|
if not conclusion:
|
||||||
return tool_error("Missing required parameter: conclusion")
|
return tool_error("Missing required parameter: conclusion")
|
||||||
|
|||||||
+2
-2
@@ -17,9 +17,9 @@ requires_env:
|
|||||||
description: "Min similarity score % to include memory 0-100 (default: 60)"
|
description: "Min similarity score % to include memory 0-100 (default: 60)"
|
||||||
|
|
||||||
provides_tools:
|
provides_tools:
|
||||||
- mem0_profile
|
- mem0_list_all
|
||||||
- mem0_search
|
- mem0_search
|
||||||
- mem0_conclude
|
- mem0_save_memory
|
||||||
- mem0_delete
|
- mem0_delete
|
||||||
|
|
||||||
pip_dependencies:
|
pip_dependencies:
|
||||||
|
|||||||
Reference in New Issue
Block a user