Add audit and memory UI

2026-05-20 23:24:12 +08:00
parent 61cdd8bbaf
commit 3d815d3ae3
6 changed files with 532 additions and 4 deletions
@@ -0,0 +1,61 @@
+from fastapi.testclient import TestClient
+
+from duck_core.api import create_app
+from duck_core.memory.store import MemoryStore
+
+
+def test_memory_api_stores_workspace_scoped_notes(tmp_path, monkeypatch):
+    monkeypatch.setenv("DUCK_DB_PATH", str(tmp_path / "duck.sqlite3"))
+    client = TestClient(create_app())
+
+    first = client.post(
+        "/v1/memory",
+        json={
+            "text": "User prefers concise Russian answers.",
+            "workspace": "/tmp/project-a",
+            "conversation_id": "chat_a",
+            "memory_type": "preference",
+            "importance": 0.8,
+        },
+    ).json()
+    client.post(
+        "/v1/memory",
+        json={
+            "text": "Different workspace note.",
+            "workspace": "/tmp/project-b",
+            "conversation_id": "chat_b",
+        },
+    )
+
+    listed = client.get("/v1/memory", params={"workspace": "/tmp/project-a"}).json()
+    search = client.get(
+        "/v1/memory/search",
+        params={"q": "concise Russian", "workspace": "/tmp/project-a"},
+    ).json()
+
+    assert first["memory_id"].startswith("mem_")
+    assert [item["text"] for item in listed["results"]] == [
+        "User prefers concise Russian answers."
+    ]
+    assert search["results"][0]["memory_id"] == first["memory_id"]
+    assert search["results"][0]["memory_type"] == "preference"
+    assert "Different workspace note." not in str(search)
+
+
+async def test_memory_store_searches_text_and_metadata(tmp_path):
+    store = MemoryStore(str(tmp_path / "duck.sqlite3"))
+    await store.init()
+    await store.add(
+        text="RX580 should use Vulkan builds.",
+        workspace="/tmp/duck",
+        conversation_id="chat_runtime",
+        memory_type="fact",
+        importance=0.9,
+        metadata={"topic": "gpu"},
+    )
+
+    results = await store.search("vulkan", workspace="/tmp/duck")
+
+    assert len(results) == 1
+    assert results[0].workspace == "/tmp/duck"
+    assert results[0].metadata["topic"] == "gpu"