diff --git a/mcp-wrapper/src/tools.ts b/mcp-wrapper/src/tools.ts index 6201e74..aa68a99 100644 --- a/mcp-wrapper/src/tools.ts +++ b/mcp-wrapper/src/tools.ts @@ -192,10 +192,16 @@ export const toolSchemas: Record = { "Retrieve the current session context payload (identity, session handle, " + "topic cluster, rich club). Call at session start to load relevant " + "background memory. Returns JSON with l0, l1, l2, rich_club, and " + - "compact_handle fields.", + "compact_handle fields. Optional session_id parameter to query a specific session.", inputSchema: { type: "object", - properties: {}, + properties: { + session_id: { + type: "string", + description: + "Optional session id to query. When omitted, uses the wrapper's current session.", + }, + }, }, }, profile_get_set: { @@ -361,8 +367,10 @@ export async function invokeTool( return bridge.call("memory_capture", args); case "memory_consolidate": return bridge.call("memory_consolidate", args); - case "memory_session_context": - return bridge.call("session_start_payload", {}); + case "memory_session_context": { + const sessionId = (args.session_id as string) || null; + return bridge.call("session_start_payload", sessionId ? { session_id: sessionId } : {}); + } case "profile_get_set": { const op = args.operation as string; if (op === "get") {