eclipse-score/mcp-servers/maintain-working-memory

wm.initialize_session(
    goal="Refactor auth module to async",
    subgoals=[
        "Identify all auth functions",
        "Map dependencies on sync calls",
        "Implement async versions",
        "Update call sites",
    ],
    assumptions=["auth.py is the only auth module"],
)
# Returns {"session_id": "...", "setup": {"ok": ..., ...}}

# After each tool call, record (COMPRESS to 1-2 sentences)
graph_result = wm.query_graph("All functions in lib/auth.py")
# Returns {"query": "...", "matches": [...], "setup": {...}}

# Add assumptions (especially ones to verify)
wm.initialize_session(
    goal="Track assumptions",
    subgoals=[],
    assumptions={
        "auth.py is only auth module": "high",
        "No external code imports auth": "medium",
    },
)
# Before risky decisions, check for unverified ones
unverified = wm.get_unverified_assumptions()
if unverified:
    print(f"WARNING: {len(unverified)} unverified assumptions")
    for a in unverified:
        if a.confidence in ["low", "medium"]:
            print(
                f"  - {a.key} (confidence: {a.confidence}) - VERIFY BEFORE PROCEEDING"
            )

# Document what you decide and why (with reversibility)
decision_result = wm.record_decision(
    decision="Implement async in auth.py first, then update callers",
    reason=[
        "Minimizes risk of partial refactoring",
        "Only 3 call sites to update",
        "Can test refactored functions independently",
    ],
    reversible=True,
    grounded_nodes=["auth.py", "auth_module"],
)
# Returns {"unresolved_nodes": []}
# Grounded nodes accept repo-relative source paths or canonical graph node IDs.
# Non-empty unresolved_nodes entries do not contribute to retrieval; restate
# them as real paths or graph IDs.

# After each major step, update your plan
wm.record_outcome(
    task="Map all call sites of authenticate()",
    verdict="pass",
    coverage=0.35,
    surfaced_nodes=["authenticate"],
    missing_nodes=[],
)

# Get current state
state = wm.get_working_memory()
print(f"Entries: {len(state)}")
# Check for potential issues
unverified = wm.get_unverified_assumptions()
if unverified:
    print(f"WARNING: {len(unverified)} unverified assumptions")