ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter3/user-memory/test_json_cards_string_payload.py

@@ -0,0 +1,37 @@
+"""Regression test for JSON cards tool payload parsing."""
+import os
+import sys
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from agent import UserMemoryAgent
+from config import MemoryMode
+
+
+def test_json_cards_add_memory_parses_stringified_json():
+    captured = {}
+
+    class FakeMemoryManager:
+        def add_memory(self, content, session_id, **kwargs):
+            captured["content"] = content
+            captured["session_id"] = session_id
+            return "banking.first_national.checking_account_number"
+
+    agent = UserMemoryAgent.__new__(UserMemoryAgent)
+    agent.config = type("Cfg", (), {"memory_mode": MemoryMode.JSON_CARDS})()
+    agent.memory_manager = FakeMemoryManager()
+    agent.session_id = "session-test"
+
+    result = agent._tool_add_memory(
+        content='{"category": "banking", "subcategory": "first_national", "key": "checking_account_number", "value": "4429853327"}'
+    )
+
+    assert captured["session_id"] == "session-test"
+    assert captured["content"] == {
+        "category": "banking",
+        "subcategory": "first_national",
+        "key": "checking_account_number",
+        "value": "4429853327",
+    }
+    assert result["success"] is True
+    assert result["memory_id"] == "banking.first_national.checking_account_number"