ZacharyZhang-NY
3d792417cf
remember / supersede / forget now reach the one Python writer from a terminal, from MCP, and from HTTP. The server, not the caller, decides who owns a memory. - cli: remember, supersede, forget. --holder falls back to settings.default_holder until the holder registry lands. Timestamps are parsed once with one clear message. - ingestion: import pymupdf as fitz. The deprecated fitz import printed a warning to stdout, so every CLI command's JSON answer was unparseable. - mcp: remember, supersede_memory and forget_memory forward to the Python writer over HTTP (ELY_EYE_API, default http://127.0.0.1:8765). SQLite stays read-only to the Go side, so the write rules live in one place. - mcp reads follow schema v2: atoms.time became ingested_at, and reads go through v_atoms_current, so a superseded memory stops being findable. holder, origin, scenario and event_time now show up in fetch_atom and search_atoms. - scripts/mcp_write_smoke.sh proves the loop end to end: remember, search, supersede, forget, then not found. Evidence: 92 pytest passed; go build, go vet and go test clean; mypy clean on the 10 touched source files with a fresh cache; ruff clean except a pre-existing B008 at cli.py:292, identical at HEAD.
119 lines
3.9 KiB
Bash
Executable File
119 lines
3.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Prove the MCP write tools reach the Python writer over HTTP, end to end.
|
|
set -euo pipefail
|
|
|
|
root="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
home="$(mktemp -d)"
|
|
port="${PORT:-8799}"
|
|
export ELY_EYE_HOME="$home"
|
|
export ELY_EYE_API="http://127.0.0.1:${port}"
|
|
|
|
python_bin="$root/.venv/bin/python"
|
|
[ -x "$python_bin" ] || python_bin="python3"
|
|
|
|
api=""
|
|
cleanup() {
|
|
if [ -n "$api" ]; then kill "$api" 2>/dev/null || true; fi
|
|
rm -rf "$home"
|
|
}
|
|
trap cleanup EXIT
|
|
|
|
"$python_bin" -m ely_eye.cli serve --port "$port" >"$home/api.log" 2>&1 &
|
|
api=$!
|
|
|
|
ready=0
|
|
for _ in $(seq 1 80); do
|
|
if curl -sf "http://127.0.0.1:${port}/api/status" >/dev/null 2>&1; then ready=1; break; fi
|
|
sleep 0.5
|
|
done
|
|
if [ "$ready" -ne 1 ]; then
|
|
echo "FAIL: the Ely-Eye API did not start"
|
|
cat "$home/api.log"
|
|
exit 1
|
|
fi
|
|
|
|
(cd "$root/ely-eye-mcp" && go build -o "$home/ely-eye-mcp" ./cmd/ely-eye-mcp)
|
|
|
|
"$python_bin" - "$home/ely-eye-mcp" <<'PY'
|
|
import json
|
|
import subprocess
|
|
import sys
|
|
|
|
binary = sys.argv[1]
|
|
proc = subprocess.Popen(
|
|
[binary, "server"],
|
|
stdin=subprocess.PIPE,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.DEVNULL,
|
|
text=True,
|
|
)
|
|
|
|
|
|
def send(message):
|
|
proc.stdin.write(json.dumps(message, ensure_ascii=False) + "\n")
|
|
proc.stdin.flush()
|
|
|
|
|
|
def call(id_, method, params):
|
|
send({"jsonrpc": "2.0", "id": id_, "method": method, "params": params})
|
|
while True:
|
|
line = proc.stdout.readline()
|
|
if not line:
|
|
raise SystemExit(f"FAIL: the MCP server closed the pipe at request {id_}")
|
|
answer = json.loads(line)
|
|
if answer.get("id") == id_:
|
|
if "error" in answer:
|
|
raise SystemExit(f"FAIL: {method} returned {answer['error']}")
|
|
return answer["result"]
|
|
|
|
|
|
def tool(name, arguments):
|
|
result = call(3, "tools/call", {"name": name, "arguments": arguments})
|
|
if result.get("isError"):
|
|
raise SystemExit(f"FAIL: {name} returned an error: {result['content']}")
|
|
return result["structuredContent"]
|
|
|
|
|
|
call(1, "initialize", {
|
|
"protocolVersion": "2025-06-18",
|
|
"capabilities": {},
|
|
"clientInfo": {"name": "ely-eye-smoke", "version": "0"},
|
|
})
|
|
send({"jsonrpc": "2.0", "method": "notifications/initialized"})
|
|
|
|
names = [item["name"] for item in call(2, "tools/list", {})["tools"]]
|
|
for required in ("remember", "supersede_memory", "forget_memory"):
|
|
if required not in names:
|
|
raise SystemExit(f"FAIL: {required} is not registered; tools are {names}")
|
|
|
|
statement = "keys are in the hall drawer"
|
|
remembered = tool("remember", {"text": statement, "scenario": "home"})["atom"]
|
|
if not remembered["holder_id"]:
|
|
raise SystemExit(f"FAIL: the server stamped no holder: {remembered}")
|
|
atom_id = remembered["atom_id"]
|
|
|
|
hits = tool("search_atoms", {"query": statement})["hits"]
|
|
if not hits or hits[0]["atom"]["atom_id"] != atom_id:
|
|
raise SystemExit(f"FAIL: the new memory is not retrievable: {hits}")
|
|
if hits[0]["atom"].get("scenario") != "home":
|
|
raise SystemExit(f"FAIL: the search hit lost the scenario: {hits[0]['atom']}")
|
|
|
|
replacement = "keys are on the coffee table"
|
|
newer = tool("supersede_memory", {"atom_id": atom_id, "text": replacement})["atom"]
|
|
if newer["supersedes"] != [atom_id]:
|
|
raise SystemExit(f"FAIL: the replacement does not point back: {newer}")
|
|
|
|
after = tool("search_atoms", {"query": statement})["hits"]
|
|
if any(hit["atom"]["atom_id"] == atom_id for hit in after):
|
|
raise SystemExit(f"FAIL: the superseded memory is still retrievable: {after}")
|
|
|
|
report = tool("forget_memory", {"atom_id": atom_id})["report"]
|
|
if report["atoms_deleted"] != 1:
|
|
raise SystemExit(f"FAIL: forget deleted {report['atoms_deleted']} atoms, want 1")
|
|
gone = call(3, "tools/call", {"name": "fetch_atom", "arguments": {"atom_id": atom_id}})
|
|
if not gone.get("isError") or "not found" not in gone["content"][0]["text"]:
|
|
raise SystemExit(f"FAIL: the forgotten memory is still readable: {gone}")
|
|
|
|
print("PASS: remember, supersede, search and forget work through MCP")
|
|
PY
|