fix: require memory schema fields by action

2026-05-03 18:38:23 +08:00
parent ae1f058b3c
commit 5b24c0fa85
2 changed files with 62 additions and 0 deletions
--- a/tools/memory_tool.py
+++ b/tools/memory_tool.py
@@ -560,6 +560,29 @@ MEMORY_SCHEMA = {
            },
        },
        "required": ["action", "target"],
+        "allOf": [
+            {
+                "if": {
+                    "properties": {"action": {"const": "add"}},
+                    "required": ["action"],
+                },
+                "then": {"required": ["content"]},
+            },
+            {
+                "if": {
+                    "properties": {"action": {"const": "replace"}},
+                    "required": ["action"],
+                },
+                "then": {"required": ["old_text", "content"]},
+            },
+            {
+                "if": {
+                    "properties": {"action": {"const": "remove"}},
+                    "required": ["action"],
+                },
+                "then": {"required": ["old_text"]},
+            },
+        ],
    },
 }