@adhd/agent-mcp 0.0.4 → 0.0.6

package/src/server.js

@@ -7,8 +7,9 @@ import { logger } from "./logger.js";
 import { agentCreate, agentRead, agentUpdate, agentDelete, agentList, } from "./tools/agent-crud.js";
 import { agentTool, sessionList, sessionClose, sessionClear } from "./tools/session.js";
 import { taskTool, taskList, taskCancel, resultTool } from "./tools/task.js";
+import { usageQuery } from "./tools/usage.js";
 import { ToolError } from "./validation/errors.js";
-import { agentCreateInputSchema, agentReadInputSchema, agentUpdateInputSchema, agentDeleteInputSchema, agentToolInputSchema, sessionListInputSchema, sessionCloseInputSchema, sessionClearInputSchema, taskToolInputSchema, taskListInputSchema, taskCancelInputSchema, resultInputSchema, } from "./validation/index.js";
+import { agentCreateInputSchema, agentReadInputSchema, agentUpdateInputSchema, agentDeleteInputSchema, agentToolInputSchema, sessionListInputSchema, sessionCloseInputSchema, sessionClearInputSchema, taskToolInputSchema, taskListInputSchema, taskCancelInputSchema, resultInputSchema, taskUsageInputSchema, } from "./validation/index.js";
 function toMcpErrorContent(error) {
     let message;
     if (error instanceof ToolError) {
@@ -350,6 +351,16 @@ export function createServer(deps) {
             description: "Clear all messages from a session's context without closing it",
             inputSchema: toMcpInputSchema(sessionClearInputSchema),
         },
+        {
+            name: "usage_query",
+            description: "Query recorded token usage from task_usage by task_id (returns the full delegation subtree), root_task_id, agent_name, or time window",
+            inputSchema: toMcpInputSchema(taskUsageInputSchema),
+        },
+        {
+            name: "guide",
+            description: "Returns a complete guide explaining how to use this server — call this first if you are unsure what to do",
+            inputSchema: { type: "object", properties: {} },
+        },
     ];
     // In-process handler that routes tool calls to the local handlers
     const inProcessHandler = async (toolName, args, ctx) => {
@@ -368,9 +379,17 @@ export function createServer(deps) {
                     selfUrl: deps.selfUrl,
                     inProcessDescriptors,
                     inProcessHandler,
+                    db: deps.db,
                 }, ctx);
             case "result":
-                return resultTool(resultInputSchema.parse(args), { taskStore: deps.taskStore });
+                return resultTool(resultInputSchema.parse(args), {
+                    taskStore: deps.taskStore,
+                    db: deps.db,
+                });
+            case "usage_query":
+                return usageQuery(deps.db, taskUsageInputSchema.parse(args ?? {}));
+            case "guide":
+                return USAGE_GUIDE;
             case "task_list":
                 return taskList(taskListInputSchema.parse(args), { taskStore: deps.taskStore });
             case "task_cancel":
@@ -466,7 +485,12 @@ export function createServer(deps) {
                 inputSchema: toMcpInputSchema(resultInputSchema),
             },
             {
-                name: "usage",
+                name: "usage_query",
+                description: "Query recorded token usage from task_usage by task_id (returns the full delegation subtree), root_task_id, agent_name, or time window. Bare {} returns the most recent rows.",
+                inputSchema: toMcpInputSchema(taskUsageInputSchema),
+            },
+            {
+                name: "guide",
                 description: "Returns a complete guide explaining how to use this server — call this first if you are unsure what to do",
                 inputSchema: { type: "object", properties: {} },
             },
@@ -525,6 +549,7 @@ export function createServer(deps) {
                         selfUrl: deps.selfUrl,
                         inProcessDescriptors,
                         inProcessHandler,
+                        db: deps.db,
                     })
                     // No callerContext — top-level call
                     );
@@ -533,8 +558,13 @@ export function createServer(deps) {
                 case "task_cancel":
                     return toMcpContent(taskCancel(taskCancelInputSchema.parse(args), { taskStore: deps.taskStore }));
                 case "result":
-                    return toMcpContent(resultTool(resultInputSchema.parse(args), { taskStore: deps.taskStore }));
-                case "usage":
+                    return toMcpContent(resultTool(resultInputSchema.parse(args), {
+                        taskStore: deps.taskStore,
+                        db: deps.db,
+                    }));
+                case "usage_query":
+                    return toMcpContent(usageQuery(deps.db, taskUsageInputSchema.parse(args ?? {})));
+                case "guide":
                     return toMcpContent(USAGE_GUIDE);
                 default:
                     return toMcpErrorContent(new ToolError("VALIDATION_ERROR", `Unknown tool: ${name}`));

package/src/server.js.map

@@ -1 +1 @@
-{"version":3,"file":"server.js","sourceRoot":"","sources":["../../../../../packages/ai/agent-mcp/src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACnG,OAAO,EACH,qBAAqB,EACrB,sBAAsB,GACzB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAUrC,OAAO,EACH,WAAW,EACX,SAAS,EACT,WAAW,EACX,WAAW,EACX,SAAS,GACZ,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACxF,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7E,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EACH,sBAAsB,EACtB,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,sBAAsB,EACtB,uBAAuB,EACvB,uBAAuB,EACvB,mBAAmB,EACnB,mBAAmB,EACnB,qBAAqB,EACrB,iBAAiB,GACpB,MAAM,uBAAuB,CAAC;AAa/B,SAAS,iBAAiB,CAAC,KAAc;IACrC,IAAI,OAAe,CAAC;IACpB,IAAI,KAAK,YAAY,SAAS,EAAE,CAAC;QAC7B,OAAO,GAAG,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;IACjD,CAAC;SAAM,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAChC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC5B,CAAC;SAAM,CAAC;QACJ,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO;QACH,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;KACtD,CAAC;AACN,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAChC,OAAO;QACH,OAAO,EAAE;YACL;gBACI,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;aACvC;SACJ;KACJ,CAAC;AACN,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,SAAS,gBAAgB,CAAC,MAAoB;IAC1C,MAAM,UAAU,GAAG,CAAC,CAAC,YAAY,CAAC,MAAM,CAA4B,CAAC;IAErE,4EAA4E;IAC5E,4CAA4C;IAC5C,IAAI,UAAU,CAAC,MAAM,CAAC,KAAK,QAAQ,EAAE,CAAC;QAClC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,UAAU,CAAC;QAC/C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,qEAAqE;IACrE,oEAAoE;IACpE,MAAM,QAAQ,GACT,UAAU,CAAC,OAAO,CAA2C;QAC7D,UAAU,CAAC,OAAO,CAA2C,CAAC;IAEnE,IAAI,QAAQ,EAAE,CAAC;QACX,MAAM,gBAAgB,GAA4B,EAAE,CAAC;QACrD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;YACpC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9D,MAAM,CAAC,MAAM,CAAC,gBAAgB,EAAE,KAAgC,CAAC,CAAC;YACtE,CAAC;QACL,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC;IAC5D,CAAC;IAED,2CAA2C;IAC3C,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;AAC9C,CAAC;AAED,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAsOnB,CAAC,IAAI,EAAE,CAAC;AAET,MAAM,UAAU,YAAY,CAAC,IAAgB;IACzC,MAAM,MAAM,GAAG,IAAI,MAAM,CACrB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,EACvC,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAClC,CAAC;IAEF,wEAAwE;IACxE,8DAA8D;IAC9D,6CAA6C;IAC7C,MAAM,oBAAoB,GAA8B;QACpD;YACI,IAAI,EAAE,OAAO;YACb,WAAW,EAAE,yCAAyC;YACtD,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;SACtD;QACD;YACI,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,gCAAgC;YAC7C,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;SACrD;QACD;YACI,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,0BAA0B;YACvC,WAAW,EAAE,gBAAgB,CAAC,iBAAiB,CAAC;SACnD;QACD;YACI,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,YAAY;YACzB,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;SACrD;QACD;YACI,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,uBAAuB;YACpC,WAAW,EAAE,gBAAgB,CAAC,qBAAqB,CAAC;SACvD;QACD;YACI,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,eAAe;YAC5B,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;SACxD;QACD;YACI,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,iBAAiB;YAC9B,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;SACzD;QACD;YACI,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,gEAAgE;YAC7E,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;SACzD;KACJ,CAAC;IAEF,kEAAkE;IAClE,MAAM,gBAAgB,GAAyB,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QACzE,QAAQ,QAAQ,EAAE,CAAC;YACf,KAAK,OAAO;gBACR,OAAO,SAAS,CACZ,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAChC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,EACrF,GAAG,CACN,CAAC;YACN,KAAK,MAAM;gBACP,OAAO,QAAQ,CACX,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAC/B;oBACI,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,oBAAoB;oBACpB,gBAAgB;iBACnB,EACD,GAAG,CACN,CAAC;YACN,KAAK,QAAQ;gBACT,OAAO,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACpF,KAAK,WAAW;gBACZ,OAAO,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACpF,KAAK,aAAa;gBACd,OAAO,UAAU,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACxF,KAAK,cAAc;gBACf,OAAO,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACnD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP,KAAK,eAAe;gBAChB,OAAO,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACrD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP,KAAK,eAAe;gBAChB,OAAO,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACrD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP;gBACI,MAAM,IAAI,SAAS,CAAC,kBAAkB,EAAE,4BAA4B,QAAQ,EAAE,CAAC,CAAC;QACxF,CAAC;IACL,CAAC,CAAC;IAEF,yEAAyE;IACzE,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAC1D,KAAK,EAAE;YACH;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,sCAAsC;gBACnD,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,wCAAwC;gBACrD,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;aACtD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,mCAAmC;gBAChD,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;aAClD;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,kDAAkD;gBAC/D,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;aACtD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,eAAe;gBAC5B,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,yBAAyB;gBACtC,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;aACzD;YACD;gBACI,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,gEAAgE;gBAC7E,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;aACzD;YACD;gBACI,IAAI,EAAE,MAAM;gBACZ,WAAW,EAAE,wKAAwK;gBACrL,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;aACrD;YACD;gBACI,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,YAAY;gBACzB,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;aACrD;YACD;gBACI,IAAI,EAAE,aAAa;gBACnB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,qBAAqB,CAAC;aACvD;YACD;gBACI,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,4CAA4C;gBACzD,WAAW,EAAE,gBAAgB,CAAC,iBAAiB,CAAC;aACnD;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,2GAA2G;gBACxH,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;aAClD;SACJ;KACJ,CAAC,CAAC,CAAC;IAEJ,yEAAyE;IACzE,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAC,OAAO,EAAC,EAAE;QAC5D,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,IAAI,CAAC;YACD,QAAQ,IAAI,EAAE,CAAC;gBACX,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,YAAY;oBACb,OAAO,YAAY,CAAC,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAEtG,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,YAAY;oBACb,OAAO,YAAY,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1E,KAAK,OAAO;oBACR,OAAO,YAAY,CACf,MAAM,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC;oBACF,uCAAuC;qBAC1C,CAAC;gBAEN,KAAK,cAAc;oBACf,OAAO,YAAY,CACf,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC5C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,eAAe;oBAChB,OAAO,YAAY,CACf,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,eAAe;oBAChB,OAAO,YAAY,CACf,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,MAAM;oBACP,OAAO,YAAY,CACf,MAAM,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC5C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,SAAS,EAAE,IAAI,CAAC,SAAS;wBACzB,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;wBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;wBACrB,oBAAoB;wBACpB,gBAAgB;qBACnB,CAAC;oBACF,oCAAoC;qBACvC,CAAC;gBAEN,KAAK,WAAW;oBACZ,OAAO,YAAY,CAAC,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;gBAElG,KAAK,aAAa;oBACd,OAAO,YAAY,CAAC,UAAU,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;gBAEtG,KAAK,QAAQ;oBACT,OAAO,YAAY,CAAC,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;gBAElG,KAAK,OAAO;oBACR,OAAO,YAAY,CAAC,WAAW,CAAC,CAAC;gBAErC;oBACI,OAAO,iBAAiB,CAAC,IAAI,SAAS,CAAC,kBAAkB,EAAE,iBAAiB,IAAI,EAAE,CAAC,CAAC,CAAC;YAC7F,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,IAAgB;IAI9C,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC;IACtD,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,MAAM,EAAE,EAAE,CAAC,CAAC;IAEzD,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QACxB,MAAM,cAAc,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAClD,MAAM,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAErC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,oBAAoB,CAAC,CAAC;QAE1D,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;SACJ,CAAC;IACN,CAAC;IAED,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;QACvB,8DAA8D;QAC9D,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QAErE,MAAM,aAAa,GAAG,IAAI,6BAA6B,CAAC;YACpD,kBAAkB,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE;SAChD,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;YACnD,MAAM,aAAa,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAEpC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;gBACzB,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,oBAAoB,CAAC,CAAC;gBAC/D,OAAO,EAAE,CAAC;YACd,CAAC,CAAC,CAAC;YACH,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,oBAAoB,IAAI,EAAE,CAAC;QAC3C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;gBACrB,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBACxC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBAC7D,CAAC,CAAC,CAAC;YACP,CAAC;YACD,UAAU;SACb,CAAC;IACN,CAAC;IAED,gBAAgB;IAChB,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;QACtB,mFAAmF;QACnF,wEAAwE;QACxE,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,qEAAqE,CAAC,CAAC;QAClG,MAAM,cAAc,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAClD,MAAM,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAErC,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;SACJ,CAAC;IACN,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,sBAAsB,SAAS,EAAE,CAAC,CAAC;AACvD,CAAC"}
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../../../../packages/ai/agent-mcp/src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACnG,OAAO,EACH,qBAAqB,EACrB,sBAAsB,GACzB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAUrC,OAAO,EACH,WAAW,EACX,SAAS,EACT,WAAW,EACX,WAAW,EACX,SAAS,GACZ,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACxF,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7E,OAAO,EAAE,UAAU,EAAiB,MAAM,kBAAkB,CAAC;AAC7D,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EACH,sBAAsB,EACtB,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,sBAAsB,EACtB,uBAAuB,EACvB,uBAAuB,EACvB,mBAAmB,EACnB,mBAAmB,EACnB,qBAAqB,EACrB,iBAAiB,EACjB,oBAAoB,GACvB,MAAM,uBAAuB,CAAC;AAe/B,SAAS,iBAAiB,CAAC,KAAc;IACrC,IAAI,OAAe,CAAC;IACpB,IAAI,KAAK,YAAY,SAAS,EAAE,CAAC;QAC7B,OAAO,GAAG,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;IACjD,CAAC;SAAM,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAChC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC5B,CAAC;SAAM,CAAC;QACJ,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO;QACH,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;KACtD,CAAC;AACN,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAChC,OAAO;QACH,OAAO,EAAE;YACL;gBACI,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;aACvC;SACJ;KACJ,CAAC;AACN,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,SAAS,gBAAgB,CAAC,MAAoB;IAC1C,MAAM,UAAU,GAAG,CAAC,CAAC,YAAY,CAAC,MAAM,CAA4B,CAAC;IAErE,4EAA4E;IAC5E,4CAA4C;IAC5C,IAAI,UAAU,CAAC,MAAM,CAAC,KAAK,QAAQ,EAAE,CAAC;QAClC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,UAAU,CAAC;QAC/C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,qEAAqE;IACrE,oEAAoE;IACpE,MAAM,QAAQ,GACT,UAAU,CAAC,OAAO,CAA2C;QAC7D,UAAU,CAAC,OAAO,CAA2C,CAAC;IAEnE,IAAI,QAAQ,EAAE,CAAC;QACX,MAAM,gBAAgB,GAA4B,EAAE,CAAC;QACrD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;YACpC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9D,MAAM,CAAC,MAAM,CAAC,gBAAgB,EAAE,KAAgC,CAAC,CAAC;YACtE,CAAC;QACL,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC;IAC5D,CAAC;IAED,2CAA2C;IAC3C,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;AAC9C,CAAC;AAED,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAsOnB,CAAC,IAAI,EAAE,CAAC;AAET,MAAM,UAAU,YAAY,CAAC,IAAgB;IACzC,MAAM,MAAM,GAAG,IAAI,MAAM,CACrB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,EACvC,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAClC,CAAC;IAEF,wEAAwE;IACxE,8DAA8D;IAC9D,6CAA6C;IAC7C,MAAM,oBAAoB,GAA8B;QACpD;YACI,IAAI,EAAE,OAAO;YACb,WAAW,EAAE,yCAAyC;YACtD,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;SACtD;QACD;YACI,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,gCAAgC;YAC7C,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;SACrD;QACD;YACI,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,0BAA0B;YACvC,WAAW,EAAE,gBAAgB,CAAC,iBAAiB,CAAC;SACnD;QACD;YACI,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,YAAY;YACzB,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;SACrD;QACD;YACI,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,uBAAuB;YACpC,WAAW,EAAE,gBAAgB,CAAC,qBAAqB,CAAC;SACvD;QACD;YACI,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,eAAe;YAC5B,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;SACxD;QACD;YACI,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,iBAAiB;YAC9B,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;SACzD;QACD;YACI,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,gEAAgE;YAC7E,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;SACzD;QACD;YACI,IAAI,EAAE,aAAa;YACnB,WAAW,EACP,uIAAuI;YAC3I,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;SACtD;QACD;YACI,IAAI,EAAE,OAAO;YACb,WAAW,EACP,2GAA2G;YAC/G,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;SAClD;KACJ,CAAC;IAEF,kEAAkE;IAClE,MAAM,gBAAgB,GAAyB,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QACzE,QAAQ,QAAQ,EAAE,CAAC;YACf,KAAK,OAAO;gBACR,OAAO,SAAS,CACZ,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAChC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,EACrF,GAAG,CACN,CAAC;YACN,KAAK,MAAM;gBACP,OAAO,QAAQ,CACX,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAC/B;oBACI,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,oBAAoB;oBACpB,gBAAgB;oBAChB,EAAE,EAAE,IAAI,CAAC,EAAE;iBACd,EACD,GAAG,CACN,CAAC;YACN,KAAK,QAAQ;gBACT,OAAO,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBAC7C,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,EAAE,EAAE,IAAI,CAAC,EAAE;iBACd,CAAC,CAAC;YACP,KAAK,aAAa;gBACd,OAAO,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE,oBAAoB,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC;YACvE,KAAK,OAAO;gBACR,OAAO,WAAW,CAAC;YACvB,KAAK,WAAW;gBACZ,OAAO,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACpF,KAAK,aAAa;gBACd,OAAO,UAAU,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACxF,KAAK,cAAc;gBACf,OAAO,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACnD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP,KAAK,eAAe;gBAChB,OAAO,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACrD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP,KAAK,eAAe;gBAChB,OAAO,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACrD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP;gBACI,MAAM,IAAI,SAAS,CAAC,kBAAkB,EAAE,4BAA4B,QAAQ,EAAE,CAAC,CAAC;QACxF,CAAC;IACL,CAAC,CAAC;IAEF,yEAAyE;IACzE,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAC1D,KAAK,EAAE;YACH;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,sCAAsC;gBACnD,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,wCAAwC;gBACrD,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;aACtD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,mCAAmC;gBAChD,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;aAClD;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,kDAAkD;gBAC/D,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;aACtD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,eAAe;gBAC5B,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,yBAAyB;gBACtC,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;aACzD;YACD;gBACI,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,gEAAgE;gBAC7E,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;aACzD;YACD;gBACI,IAAI,EAAE,MAAM;gBACZ,WAAW,EAAE,wKAAwK;gBACrL,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;aACrD;YACD;gBACI,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,YAAY;gBACzB,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;aACrD;YACD;gBACI,IAAI,EAAE,aAAa;gBACnB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,qBAAqB,CAAC;aACvD;YACD;gBACI,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,4CAA4C;gBACzD,WAAW,EAAE,gBAAgB,CAAC,iBAAiB,CAAC;aACnD;YACD;gBACI,IAAI,EAAE,aAAa;gBACnB,WAAW,EACP,8KAA8K;gBAClL,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;aACtD;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,2GAA2G;gBACxH,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;aAClD;SACJ;KACJ,CAAC,CAAC,CAAC;IAEJ,yEAAyE;IACzE,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAC,OAAO,EAAC,EAAE;QAC5D,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,IAAI,CAAC;YACD,QAAQ,IAAI,EAAE,CAAC;gBACX,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,YAAY;oBACb,OAAO,YAAY,CAAC,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAEtG,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,YAAY;oBACb,OAAO,YAAY,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1E,KAAK,OAAO;oBACR,OAAO,YAAY,CACf,MAAM,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC;oBACF,uCAAuC;qBAC1C,CAAC;gBAEN,KAAK,cAAc;oBACf,OAAO,YAAY,CACf,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC5C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,eAAe;oBAChB,OAAO,YAAY,CACf,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,eAAe;oBAChB,OAAO,YAAY,CACf,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,MAAM;oBACP,OAAO,YAAY,CACf,MAAM,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC5C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,SAAS,EAAE,IAAI,CAAC,SAAS;wBACzB,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;wBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;wBACrB,oBAAoB;wBACpB,gBAAgB;wBAChB,EAAE,EAAE,IAAI,CAAC,EAAE;qBACd,CAAC;oBACF,oCAAoC;qBACvC,CAAC;gBAEN,KAAK,WAAW;oBACZ,OAAO,YAAY,CAAC,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;gBAElG,KAAK,aAAa;oBACd,OAAO,YAAY,CAAC,UAAU,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;gBAEtG,KAAK,QAAQ;oBACT,OAAO,YAAY,CACf,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBACtC,SAAS,EAAE,IAAI,CAAC,SAAS;wBACzB,EAAE,EAAE,IAAI,CAAC,EAAE;qBACd,CAAC,CACL,CAAC;gBAEN,KAAK,aAAa;oBACd,OAAO,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE,oBAAoB,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;gBAErF,KAAK,OAAO;oBACR,OAAO,YAAY,CAAC,WAAW,CAAC,CAAC;gBAErC;oBACI,OAAO,iBAAiB,CAAC,IAAI,SAAS,CAAC,kBAAkB,EAAE,iBAAiB,IAAI,EAAE,CAAC,CAAC,CAAC;YAC7F,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,IAAgB;IAI9C,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC;IACtD,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,MAAM,EAAE,EAAE,CAAC,CAAC;IAEzD,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QACxB,MAAM,cAAc,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAClD,MAAM,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAErC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,oBAAoB,CAAC,CAAC;QAE1D,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;SACJ,CAAC;IACN,CAAC;IAED,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;QACvB,8DAA8D;QAC9D,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QAErE,MAAM,aAAa,GAAG,IAAI,6BAA6B,CAAC;YACpD,kBAAkB,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE;SAChD,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;YACnD,MAAM,aAAa,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAEpC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;gBACzB,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,oBAAoB,CAAC,CAAC;gBAC/D,OAAO,EAAE,CAAC;YACd,CAAC,CAAC,CAAC;YACH,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,oBAAoB,IAAI,EAAE,CAAC;QAC3C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;gBACrB,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBACxC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBAC7D,CAAC,CAAC,CAAC;YACP,CAAC;YACD,UAAU;SACb,CAAC;IACN,CAAC;IAED,gBAAgB;IAChB,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;QACtB,mFAAmF;QACnF,wEAAwE;QACxE,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,qEAAqE,CAAC,CAAC;QAClG,MAAM,cAAc,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAClD,MAAM,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAErC,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;SACJ,CAAC;IACN,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,sBAAsB,SAAS,EAAE,CAAC,CAAC;AACvD,CAAC"}

package/src/store/session-store.d.ts

@@ -22,4 +22,16 @@ export declare class SessionStore {
     appendMessage(sessionId: string, message: Message): void;
     getMessages(sessionId: string): Message[];
 }
+/**
+ * Estimates the token count for a set of messages using a 4-chars-per-token heuristic.
+ */
+export declare function estimateTokens(messages: Message[]): number;
+/**
+ * Returns a windowed view of messages that fits within tokenLimit estimated tokens.
+ * System messages are always preserved. Oldest non-system messages are dropped first.
+ * Returns the original array unchanged if tokenLimit <= 0 or the array already fits.
+ *
+ * See [inv:window-messages] in docs/plan/0.0.6/contexts/_shared.md.
+ */
+export declare function windowMessages(messages: Message[], tokenLimit: number): Message[];
 //# sourceMappingURL=session-store.d.ts.map

package/src/store/session-store.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"session-store.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/store/session-store.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAIxE,OAAO,KAAK,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAKlG,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAE3D,qBAAa,YAAY;IAEjB,OAAO,CAAC,QAAQ,CAAC,EAAE;IACnB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;gBADN,EAAE,EAAE,qBAAqB,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,EAChD,KAAK,CAAC,EAAE,aAAa,YAAA;IAG1C,MAAM,CAAC,KAAK,EAAE;QACV,SAAS,EAAE,MAAM,CAAC;QAClB,eAAe,EAAE,eAAe,CAAC;KACpC,GAAG,OAAO;IAwBX,IAAI,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAyBzB;;;;OAIG;IACH,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,eAAe;IAiBtD,IAAI,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,EAAE;IAiCxC,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAyB1B,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IAmBxC,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IAgBxD,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,EAAE;CAmB5C"}
+{"version":3,"file":"session-store.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/store/session-store.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAIxE,OAAO,KAAK,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAKlG,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAE3D,qBAAa,YAAY;IAEjB,OAAO,CAAC,QAAQ,CAAC,EAAE;IACnB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;gBADN,EAAE,EAAE,qBAAqB,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,EAChD,KAAK,CAAC,EAAE,aAAa,YAAA;IAG1C,MAAM,CAAC,KAAK,EAAE;QACV,SAAS,EAAE,MAAM,CAAC;QAClB,eAAe,EAAE,eAAe,CAAC;KACpC,GAAG,OAAO;IAwBX,IAAI,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAyBzB;;;;OAIG;IACH,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,eAAe;IAiBtD,IAAI,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,EAAE;IAiCxC,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAyB1B,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IAmBxC,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IAgBxD,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,EAAE;CAmB5C;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAW1D;AAED;;;;;;GAMG;AACH,wBAAgB,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,EAAE,CAuBjF"}

package/src/store/session-store.js

@@ -153,4 +153,44 @@ export class SessionStore {
         }));
     }
 }
+/**
+ * Estimates the token count for a set of messages using a 4-chars-per-token heuristic.
+ */
+export function estimateTokens(messages) {
+    return Math.ceil(messages.reduce((sum, m) => {
+        return (sum +
+            (m.content?.length ?? 0) +
+            (m.toolCalls ? JSON.stringify(m.toolCalls).length : 0) +
+            (m.toolResults ? JSON.stringify(m.toolResults).length : 0));
+    }, 0) / 4);
+}
+/**
+ * Returns a windowed view of messages that fits within tokenLimit estimated tokens.
+ * System messages are always preserved. Oldest non-system messages are dropped first.
+ * Returns the original array unchanged if tokenLimit <= 0 or the array already fits.
+ *
+ * See [inv:window-messages] in docs/plan/0.0.6/contexts/_shared.md.
+ */
+export function windowMessages(messages, tokenLimit) {
+    if (tokenLimit <= 0)
+        return messages;
+    if (estimateTokens(messages) <= tokenLimit)
+        return messages;
+    const systemMessages = messages.filter(m => m.role === "system");
+    const nonSystemMessages = messages.filter(m => m.role !== "system");
+    const systemBudget = estimateTokens(systemMessages);
+    const remaining = Math.max(0, tokenLimit - systemBudget);
+    const selected = [];
+    let used = 0;
+    // Walk newest-to-oldest; include at least one message even if it exceeds budget
+    for (let i = nonSystemMessages.length - 1; i >= 0; i--) {
+        const msg = nonSystemMessages[i];
+        const cost = estimateTokens([msg]);
+        if (used + cost > remaining && selected.length > 0)
+            break;
+        selected.unshift(msg);
+        used += cost;
+    }
+    return [...systemMessages, ...selected];
+}
 //# sourceMappingURL=session-store.js.map

package/src/store/session-store.js.map

@@ -1 +1 @@
-{"version":3,"file":"session-store.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/store/session-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,aAAa,CAAC;AAItC,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,OAAO,EAAE,qBAAqB,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAC9E,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAGhD,MAAM,OAAO,YAAY;IAEA;IACA;IAFrB,YACqB,EAAgD,EAChD,KAAqB;QADrB,OAAE,GAAF,EAAE,CAA8C;QAChD,UAAK,GAAL,KAAK,CAAgB;IACvC,CAAC;IAEJ,MAAM,CAAC,KAGN;QACG,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;QACrB,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;QAExB,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC;YACjC,EAAE;YACF,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,YAAY,EAAE,KAAK,CAAC,eAAe,CAAC,OAAO;YAC3C,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,eAAe,CAAC;YAChD,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACjB,CAAC,CAAC,GAAG,EAAE,CAAC;QAET,MAAM,CAAC,IAAI,CACP,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,EAC7C,iBAAiB,CACpB,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC9B,KAAK,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,iBAAiB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACtD,OAAO,OAAO,CAAC;IACnB,CAAC;IAED,IAAI,CAAC,EAAU;QACX,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aACd,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;aAC/B,GAAG,EAAE,CAAC;QAEX,IAAI,CAAC,GAAG,EAAE,CAAC;YACP,MAAM,IAAI,SAAS,CACf,mBAAmB,EACnB,YAAY,EAAE,aAAa,CAC9B,CAAC;QACN,CAAC;QAED,OAAO,aAAa,CAAC,KAAK,CAAC;YACvB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,YAAY,EAAE,GAAG,CAAC,YAAY;YAC9B,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,SAAS;SACtC,CAAC,CAAC;IACP,CAAC;IAED;;;;OAIG;IACH,kBAAkB,CAAC,SAAiB;QAChC,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aACd,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;aACtC,GAAG,EAAE,CAAC;QAEX,IAAI,CAAC,GAAG,EAAE,CAAC;YACP,MAAM,IAAI,SAAS,CACf,mBAAmB,EACnB,YAAY,SAAS,aAAa,CACrC,CAAC;QACN,CAAC;QAED,OAAO,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IAClE,CAAC;IAED,IAAI,CAAC,KAAuB;QACxB,MAAM,UAAU,GAAG,EAAE,CAAC;QAEtB,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YAClB,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACf,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,IAAI,GACN,UAAU,CAAC,MAAM,GAAG,CAAC;YACjB,CAAC,CAAC,IAAI,CAAC,EAAE;iBACJ,MAAM,EAAE;iBACR,IAAI,CAAC,aAAa,CAAC;iBACnB,KAAK,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;iBACzB,GAAG,EAAE;YACV,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,GAAG,EAAE,CAAC;QAErD,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAClB,aAAa,CAAC,KAAK,CAAC;YAChB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,YAAY,EAAE,GAAG,CAAC,YAAY;YAC9B,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,SAAS;SACtC,CAAC,CACL,CAAC;IACN,CAAC;IAED,KAAK,CAAC,EAAU;QACZ,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,sCAAsC;QAErE,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,IAAI,SAAS,CACf,gBAAgB,EAChB,YAAY,EAAE,qBAAqB,CACtC,CAAC;QACN,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;QACrB,IAAI,CAAC,EAAE;aACF,MAAM,CAAC,aAAa,CAAC;aACrB,GAAG,CAAC;YACD,MAAM,EAAE,QAAQ;YAChB,QAAQ,EAAE,GAAG;YACb,SAAS,EAAE,GAAG;SACjB,CAAC;aACD,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;aAC/B,GAAG,EAAE,CAAC;QAEX,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,gBAAgB,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC;IAED,aAAa,CAAC,SAAiB;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,sCAAsC;QAE5E,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,IAAI,SAAS,CACf,gBAAgB,EAChB,YAAY,SAAS,mCAAmC,CAC3D,CAAC;QACN,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE;aACjB,MAAM,CAAC,aAAa,CAAC;aACrB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;aAC7C,GAAG,EAAE,CAAC;QAEX,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,yBAAyB,CAAC,CAAC;QAC/E,OAAO,MAAM,CAAC,OAAO,CAAC;IAC1B,CAAC;IAED,aAAa,CAAC,SAAiB,EAAE,OAAgB;QAC7C,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC;YACjC,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,SAAS;YACT,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,IAAI;YAChC,SAAS,EAAE,OAAO,CAAC,SAAS;gBACxB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC;gBACnC,CAAC,CAAC,IAAI;YACV,WAAW,EAAE,OAAO,CAAC,WAAW;gBAC5B,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC;gBACrC,CAAC,CAAC,IAAI;YACV,SAAS,EAAE,OAAO,CAAC,SAAS;SAC/B,CAAC,CAAC,GAAG,EAAE,CAAC;IACb,CAAC;IAED,WAAW,CAAC,SAAiB;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACf,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;aAC7C,GAAG,EAAE,CAAC;QAEX,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACpB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,IAAI,EAAE,GAAG,CAAC,IAAuB;YACjC,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,SAAS;YACjC,SAAS,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS;YAChE,WAAW,EAAE,GAAG,CAAC,WAAW;gBACxB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC;gBAC7B,CAAC,CAAC,SAAS;YACf,SAAS,EAAE,GAAG,CAAC,SAAS;SAC3B,CAAC,CAAC,CAAC;IACR,CAAC;CACJ"}
+{"version":3,"file":"session-store.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/store/session-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,aAAa,CAAC;AAItC,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,OAAO,EAAE,qBAAqB,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAC9E,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAGhD,MAAM,OAAO,YAAY;IAEA;IACA;IAFrB,YACqB,EAAgD,EAChD,KAAqB;QADrB,OAAE,GAAF,EAAE,CAA8C;QAChD,UAAK,GAAL,KAAK,CAAgB;IACvC,CAAC;IAEJ,MAAM,CAAC,KAGN;QACG,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;QACrB,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;QAExB,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC;YACjC,EAAE;YACF,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,YAAY,EAAE,KAAK,CAAC,eAAe,CAAC,OAAO;YAC3C,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,eAAe,CAAC;YAChD,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACjB,CAAC,CAAC,GAAG,EAAE,CAAC;QAET,MAAM,CAAC,IAAI,CACP,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,EAC7C,iBAAiB,CACpB,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC9B,KAAK,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,iBAAiB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACtD,OAAO,OAAO,CAAC;IACnB,CAAC;IAED,IAAI,CAAC,EAAU;QACX,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aACd,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;aAC/B,GAAG,EAAE,CAAC;QAEX,IAAI,CAAC,GAAG,EAAE,CAAC;YACP,MAAM,IAAI,SAAS,CACf,mBAAmB,EACnB,YAAY,EAAE,aAAa,CAC9B,CAAC;QACN,CAAC;QAED,OAAO,aAAa,CAAC,KAAK,CAAC;YACvB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,YAAY,EAAE,GAAG,CAAC,YAAY;YAC9B,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,SAAS;SACtC,CAAC,CAAC;IACP,CAAC;IAED;;;;OAIG;IACH,kBAAkB,CAAC,SAAiB;QAChC,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aACd,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;aACtC,GAAG,EAAE,CAAC;QAEX,IAAI,CAAC,GAAG,EAAE,CAAC;YACP,MAAM,IAAI,SAAS,CACf,mBAAmB,EACnB,YAAY,SAAS,aAAa,CACrC,CAAC;QACN,CAAC;QAED,OAAO,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IAClE,CAAC;IAED,IAAI,CAAC,KAAuB;QACxB,MAAM,UAAU,GAAG,EAAE,CAAC;QAEtB,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YAClB,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACf,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,IAAI,GACN,UAAU,CAAC,MAAM,GAAG,CAAC;YACjB,CAAC,CAAC,IAAI,CAAC,EAAE;iBACJ,MAAM,EAAE;iBACR,IAAI,CAAC,aAAa,CAAC;iBACnB,KAAK,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;iBACzB,GAAG,EAAE;YACV,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,GAAG,EAAE,CAAC;QAErD,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAClB,aAAa,CAAC,KAAK,CAAC;YAChB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,YAAY,EAAE,GAAG,CAAC,YAAY;YAC9B,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,SAAS;SACtC,CAAC,CACL,CAAC;IACN,CAAC;IAED,KAAK,CAAC,EAAU;QACZ,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,sCAAsC;QAErE,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,IAAI,SAAS,CACf,gBAAgB,EAChB,YAAY,EAAE,qBAAqB,CACtC,CAAC;QACN,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;QACrB,IAAI,CAAC,EAAE;aACF,MAAM,CAAC,aAAa,CAAC;aACrB,GAAG,CAAC;YACD,MAAM,EAAE,QAAQ;YAChB,QAAQ,EAAE,GAAG;YACb,SAAS,EAAE,GAAG;SACjB,CAAC;aACD,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;aAC/B,GAAG,EAAE,CAAC;QAEX,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,gBAAgB,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC;IAED,aAAa,CAAC,SAAiB;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,sCAAsC;QAE5E,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,IAAI,SAAS,CACf,gBAAgB,EAChB,YAAY,SAAS,mCAAmC,CAC3D,CAAC;QACN,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE;aACjB,MAAM,CAAC,aAAa,CAAC;aACrB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;aAC7C,GAAG,EAAE,CAAC;QAEX,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,yBAAyB,CAAC,CAAC;QAC/E,OAAO,MAAM,CAAC,OAAO,CAAC;IAC1B,CAAC;IAED,aAAa,CAAC,SAAiB,EAAE,OAAgB;QAC7C,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC;YACjC,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,SAAS;YACT,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,IAAI;YAChC,SAAS,EAAE,OAAO,CAAC,SAAS;gBACxB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC;gBACnC,CAAC,CAAC,IAAI;YACV,WAAW,EAAE,OAAO,CAAC,WAAW;gBAC5B,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC;gBACrC,CAAC,CAAC,IAAI;YACV,SAAS,EAAE,OAAO,CAAC,SAAS;SAC/B,CAAC,CAAC,GAAG,EAAE,CAAC;IACb,CAAC;IAED,WAAW,CAAC,SAAiB;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACf,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;aAC7C,GAAG,EAAE,CAAC;QAEX,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACpB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,IAAI,EAAE,GAAG,CAAC,IAAuB;YACjC,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,SAAS;YACjC,SAAS,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS;YAChE,WAAW,EAAE,GAAG,CAAC,WAAW;gBACxB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC;gBAC7B,CAAC,CAAC,SAAS;YACf,SAAS,EAAE,GAAG,CAAC,SAAS;SAC3B,CAAC,CAAC,CAAC;IACR,CAAC;CACJ;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,QAAmB;IAC9C,OAAO,IAAI,CAAC,IAAI,CACZ,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;QACvB,OAAO,CACH,GAAG;YACH,CAAC,CAAC,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC;YACxB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YACtD,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAC7D,CAAC;IACN,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CACZ,CAAC;AACN,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,cAAc,CAAC,QAAmB,EAAE,UAAkB;IAClE,IAAI,UAAU,IAAI,CAAC;QAAE,OAAO,QAAQ,CAAC;IACrC,IAAI,cAAc,CAAC,QAAQ,CAAC,IAAI,UAAU;QAAE,OAAO,QAAQ,CAAC;IAE5D,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;IACjE,MAAM,iBAAiB,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;IAEpE,MAAM,YAAY,GAAG,cAAc,CAAC,cAAc,CAAC,CAAC;IACpD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,GAAG,YAAY,CAAC,CAAC;IAEzD,MAAM,QAAQ,GAAc,EAAE,CAAC;IAC/B,IAAI,IAAI,GAAG,CAAC,CAAC;IAEb,gFAAgF;IAChF,KAAK,IAAI,CAAC,GAAG,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACrD,MAAM,GAAG,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,IAAI,GAAG,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACnC,IAAI,IAAI,GAAG,IAAI,GAAG,SAAS,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM;QAC1D,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACtB,IAAI,IAAI,IAAI,CAAC;IACjB,CAAC;IAED,OAAO,CAAC,GAAG,cAAc,EAAE,GAAG,QAAQ,CAAC,CAAC;AAC5C,CAAC"}

package/src/tools/task.d.ts

@@ -5,8 +5,9 @@ import type { InProcessToolDescriptor, InProcessToolHandler } from "../clients/i
 import type { AgentStore } from "../store/agent-store.js";
 import type { SessionStore } from "../store/session-store.js";
 import type { TaskStore } from "../store/task-store.js";
-import type { ExecutionContext, ResultInput, Task, TaskCancelInput, TaskListInput, TaskToolInput, TaskToolOutput } from "../validation/index.js";
+import type { ExecutionContext, ResultInput, Task, TaskCancelInput, TaskListInput, TaskToolInput, TaskToolOutput, TaskUsageReport } from "../validation/index.js";
 import type { IHookRegistry } from "@adhd/agent-mcp-types";
+import { type Database } from "./usage.js";
 export interface TaskDeps {
     agentStore: AgentStore;
     sessionStore: SessionStore;
@@ -18,6 +19,11 @@ export interface TaskDeps {
     selfUrl: string | undefined;
     inProcessDescriptors: InProcessToolDescriptor[];
     inProcessHandler: InProcessToolHandler;
+    /**
+     * Drizzle DB handle — used to enrich `task` / `result` responses with the
+     * task's token-usage rollup (direct + subtree). See [dod.2].
+     */
+    db: Database;
 }
 /**
  * `task` tool — runs a prompt against a session's agent (session mode) or runs a
@@ -39,5 +45,7 @@ export declare function taskList(input: TaskListInput, deps: Pick<TaskDeps, "tas
 export declare function taskCancel(input: TaskCancelInput, deps: Pick<TaskDeps, "taskStore">): {
     success: true;
 };
-export declare function resultTool(input: ResultInput, deps: Pick<TaskDeps, "taskStore">): Task;
+export declare function resultTool(input: ResultInput, deps: Pick<TaskDeps, "taskStore" | "db">): Task & {
+    usage?: TaskUsageReport;
+};
 //# sourceMappingURL=task.d.ts.map

package/src/tools/task.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"task.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/task.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAE9F,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAExD,OAAO,KAAK,EACR,gBAAgB,EAChB,WAAW,EACX,IAAI,EACJ,eAAe,EACf,aAAa,EAEb,aAAa,EACb,cAAc,EACjB,MAAM,wBAAwB,CAAC;AAIhC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAE3D,MAAM,WAAW,QAAQ;IACrB,UAAU,EAAE,UAAU,CAAC;IACvB,YAAY,EAAE,YAAY,CAAC;IAC3B,SAAS,EAAE,SAAS,CAAC;IACrB,YAAY,EAAE,YAAY,CAAC;IAC3B,KAAK,EAAE,eAAe,CAAC;IACvB,MAAM,EAAE,YAAY,CAAC;IACrB,KAAK,EAAE,aAAa,CAAC;IACrB,OAAO,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,oBAAoB,EAAE,uBAAuB,EAAE,CAAC;IAChD,gBAAgB,EAAE,oBAAoB,CAAC;CAC1C;AAyGD;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,QAAQ,CAC1B,KAAK,EAAE,aAAa,EACpB,IAAI,EAAE,QAAQ,EACd,aAAa,CAAC,EAAE,gBAAgB,GACjC,OAAO,CAAC,cAAc,CAAC,CA4HzB;AAED,wBAAgB,QAAQ,CAAC,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,IAAI,EAAE,CAExF;AAED,wBAAgB,UAAU,CAAC,KAAK,EAAE,eAAe,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG;IAAE,OAAO,EAAE,IAAI,CAAA;CAAE,CAavG;AAED,wBAAgB,UAAU,CAAC,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,IAAI,CAEtF"}
+{"version":3,"file":"task.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/task.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAE9F,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAExD,OAAO,KAAK,EACR,gBAAgB,EAChB,WAAW,EACX,IAAI,EACJ,eAAe,EACf,aAAa,EAEb,aAAa,EACb,cAAc,EACd,eAAe,EAClB,MAAM,wBAAwB,CAAC;AAIhC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAwB,KAAK,QAAQ,EAAE,MAAM,YAAY,CAAC;AAEjE,MAAM,WAAW,QAAQ;IACrB,UAAU,EAAE,UAAU,CAAC;IACvB,YAAY,EAAE,YAAY,CAAC;IAC3B,SAAS,EAAE,SAAS,CAAC;IACrB,YAAY,EAAE,YAAY,CAAC;IAC3B,KAAK,EAAE,eAAe,CAAC;IACvB,MAAM,EAAE,YAAY,CAAC;IACrB,KAAK,EAAE,aAAa,CAAC;IACrB,OAAO,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,oBAAoB,EAAE,uBAAuB,EAAE,CAAC;IAChD,gBAAgB,EAAE,oBAAoB,CAAC;IACvC;;;OAGG;IACH,EAAE,EAAE,QAAQ,CAAC;CAChB;AAgHD;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,QAAQ,CAC1B,KAAK,EAAE,aAAa,EACpB,IAAI,EAAE,QAAQ,EACd,aAAa,CAAC,EAAE,gBAAgB,GACjC,OAAO,CAAC,cAAc,CAAC,CAqIzB;AAED,wBAAgB,QAAQ,CAAC,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,IAAI,EAAE,CAExF;AAED,wBAAgB,UAAU,CAAC,KAAK,EAAE,eAAe,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG;IAAE,OAAO,EAAE,IAAI,CAAA;CAAE,CAavG;AAED,wBAAgB,UAAU,CACtB,KAAK,EAAE,WAAW,EAClB,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,GAAG,IAAI,CAAC,GACzC,IAAI,GAAG;IAAE,KAAK,CAAC,EAAE,eAAe,CAAA;CAAE,CAIpC"}

package/src/tools/task.js

@@ -4,6 +4,7 @@ import { McpClientRegistry as McpClientRegistryCtor } from "../clients/registry.
 import { ToolError } from "../validation/errors.js";
 import { generateId } from "../utils/ids.js";
 import { nowIso } from "../utils/timestamps.js";
+import { buildTaskUsageReport } from "./usage.js";
 /**
  * One-shot ephemeral execution: loads the agent definition but creates no session
  * row and persists no messages. Always synchronous.
@@ -12,6 +13,9 @@ async function runEphemeralTask(input, deps, callerContext) {
     const agentDefinition = deps.agentStore.read(input.agent_name);
     const taskId = generateId();
     const ephemeralSessionId = generateId();
+    const rootTaskId = callerContext
+        ? (callerContext.rootTaskId ?? callerContext.taskId)
+        : undefined;
     const executionContext = {
         taskId,
         sessionId: ephemeralSessionId,
@@ -19,6 +23,7 @@ async function runEphemeralTask(input, deps, callerContext) {
         agentDefinition,
         callingAgentName: callerContext?.agentName,
         parentTaskId: callerContext?.taskId,
+        rootTaskId: rootTaskId ?? undefined,
         recursionDepth: (callerContext?.recursionDepth ?? -1) + 1,
         toolCallCount: 0,
     };
@@ -82,10 +87,12 @@ async function runEphemeralTask(input, deps, callerContext) {
     catch {
         // Orchestrator already captured status via captureTaskStore
     }
+    const usage = buildTaskUsageReport(deps.db, taskId);
     return {
         task_id: taskId,
         status: capturedStatus,
         result: capturedResult,
+        usage,
     };
 }
 /**
@@ -121,6 +128,11 @@ export async function taskTool(input, deps, callerContext) {
         parentTaskId: callerContext?.taskId,
         recursionDepth: (callerContext?.recursionDepth ?? -1) + 1,
     });
+    // Derive rootTaskId at creation time from the in-memory callerContext chain.
+    // A DB walk is deliberately avoided — ephemeral tasks have no tasks row.
+    const rootTaskId = callerContext
+        ? (callerContext.rootTaskId ?? callerContext.taskId)
+        : undefined;
     // 4. Build execution context
     const executionContext = {
         taskId: task.id,
@@ -129,6 +141,7 @@ export async function taskTool(input, deps, callerContext) {
         agentDefinition,
         callingAgentName: callerContext?.agentName,
         parentTaskId: callerContext?.taskId,
+        rootTaskId: rootTaskId ?? undefined,
         recursionDepth: (callerContext?.recursionDepth ?? -1) + 1,
         toolCallCount: 0,
     };
@@ -200,10 +213,12 @@ export async function taskTool(input, deps, callerContext) {
             // Return the current task state
         }
         const finalTask = deps.taskStore.read(task.id);
+        const usage = buildTaskUsageReport(deps.db, finalTask.id);
         return {
             task_id: finalTask.id,
             status: finalTask.status,
             result: finalTask.result,
+            usage,
         };
     }
 }
@@ -220,6 +235,8 @@ export function taskCancel(input, deps) {
     return { success: true };
 }
 export function resultTool(input, deps) {
-    return deps.taskStore.read(input.task_id); // throws TASK_NOT_FOUND
+    const task = deps.taskStore.read(input.task_id); // throws TASK_NOT_FOUND
+    const usage = buildTaskUsageReport(deps.db, task.id);
+    return { ...task, usage };
 }
 //# sourceMappingURL=task.js.map

package/src/tools/task.js.map

@@ -1 +1 @@
-{"version":3,"file":"task.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/task.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAKtC,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAIzD,OAAO,EAAE,iBAAiB,IAAI,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAWpF,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAgBhD;;;GAGG;AACH,KAAK,UAAU,gBAAgB,CAC3B,KAA6C,EAC7C,IAAc,EACd,aAAgC;IAEhC,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAE/D,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,kBAAkB,GAAG,UAAU,EAAE,CAAC;IAExC,MAAM,gBAAgB,GAAqB;QACvC,MAAM;QACN,SAAS,EAAE,kBAAkB;QAC7B,SAAS,EAAE,eAAe,CAAC,IAAI;QAC/B,eAAe;QACf,gBAAgB,EAAE,aAAa,EAAE,SAAS;QAC1C,YAAY,EAAE,aAAa,EAAE,MAAM;QACnC,cAAc,EAAE,CAAC,aAAa,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;QACzD,aAAa,EAAE,CAAC;KACnB,CAAC;IAEF,MAAM,QAAQ,GAAG,cAAc,CAAC,eAAe,CAAC,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IAEtF,MAAM,WAAW,GAAG;QAChB,EAAE,EAAE,UAAU,EAAE;QAChB,SAAS,EAAE,kBAAkB;QAC7B,IAAI,EAAE,MAAe;QACrB,OAAO,EAAE,KAAK,CAAC,MAAM;QACrB,SAAS,EAAE,MAAM,EAAE;KACtB,CAAC;IACF,MAAM,QAAQ,GAAG,eAAe,CAAC,YAAY;QACzC,CAAC,CAAC;YACI;gBACI,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,kBAAkB;gBAC7B,IAAI,EAAE,QAAiB;gBACvB,OAAO,EAAE,eAAe,CAAC,YAAY;gBACrC,SAAS,EAAE,MAAM,EAAE;aACtB;YACD,WAAW;SACd;QACH,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;IAEpB,sDAAsD;IACtD,IAAI,cAAc,GAAe,SAAS,CAAC;IAC3C,IAAI,cAAkC,CAAC;IAEvC,MAAM,gBAAgB,GAAG;QACrB,YAAY,EAAE,CAAC,GAAW,EAAE,MAAkB,EAAE,MAA4B,EAAE,EAAE;YAC5E,cAAc,GAAG,MAAM,CAAC;YACxB,IAAI,MAAM,EAAE,MAAM,KAAK,SAAS;gBAAE,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC;YACjE,OAAO,EAAU,CAAC;QACtB,CAAC;QACD,gEAAgE;QAChE,WAAW,EAAE,GAAG,EAAE,GAAE,CAAC;QACrB,gEAAgE;QAChE,sBAAsB,EAAE,GAAG,EAAE,GAAE,CAAC;KACX,CAAC;IAE1B,MAAM,gBAAgB,GAAG;QACrB,gEAAgE;QAChE,aAAa,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;KACL,CAAC;IAE7B,MAAM,QAAQ,GAAG,IAAI,qBAAqB;IACtC,8DAA8D;IAC9D,eAAe,CAAC,UAAiB,EACjC,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,gBAAgB,EACrB,gBAAgB,CACnB,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IAEzC,IAAI,CAAC;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;YACxB,gBAAgB;YAChB,QAAQ;YACR,QAAQ;YACR,QAAQ;YACR,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,gBAAgB;YAC3B,YAAY,EAAE,gBAAgB;YAC9B,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,MAAM;YACN,KAAK,EAAE,IAAI,CAAC,KAAK;SACpB,CAAC,CAAC;IACP,CAAC;IAAC,MAAM,CAAC;QACL,4DAA4D;IAChE,CAAC;IAED,OAAO;QACH,OAAO,EAAE,MAAM;QACf,MAAM,EAAE,cAAc;QACtB,MAAM,EAAE,cAAc;KACzB,CAAC;AACN,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC1B,KAAoB,EACpB,IAAc,EACd,aAAgC;IAEhC,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;QACxB,OAAO,gBAAgB,CAAC,KAAK,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;IACxD,CAAC;IAED,sBAAsB;IACtB,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACzD,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,IAAI,SAAS,CAAC,gBAAgB,EAAE,YAAY,KAAK,CAAC,UAAU,aAAa,CAAC,CAAC;IACrF,CAAC;IAED,uCAAuC;IACvC,MAAM,eAAe,GAAG,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAE/E,qBAAqB;IACrB,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;QAC/B,SAAS,EAAE,KAAK,CAAC,UAAU;QAC3B,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,YAAY,EAAE,aAAa,EAAE,MAAM;QACnC,cAAc,EAAE,CAAC,aAAa,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;KAC5D,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,gBAAgB,GAAqB;QACvC,MAAM,EAAE,IAAI,CAAC,EAAE;QACf,SAAS,EAAE,KAAK,CAAC,UAAU;QAC3B,SAAS,EAAE,eAAe,CAAC,IAAI;QAC/B,eAAe;QACf,gBAAgB,EAAE,aAAa,EAAE,SAAS;QAC1C,YAAY,EAAE,aAAa,EAAE,MAAM;QACnC,cAAc,EAAE,CAAC,aAAa,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;QACzD,aAAa,EAAE,CAAC;KACnB,CAAC;IAEF,4BAA4B;IAC5B,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IAEzC,2BAA2B;IAC3B,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAEzD,oBAAoB;IACpB,MAAM,QAAQ,GAAG,cAAc,CAAC,eAAe,CAAC,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IAEtF,yBAAyB;IACzB,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACzE,MAAM,WAAW,GAAG;QAChB,EAAE,EAAE,UAAU,EAAE;QAChB,SAAS,EAAE,KAAK,CAAC,UAAU;QAC3B,IAAI,EAAE,MAAe;QACrB,OAAO,EAAE,KAAK,CAAC,MAAM;QACrB,SAAS,EAAE,MAAM,EAAE;KACtB,CAAC;IACF,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,KAAK,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;IACrE,MAAM,QAAQ,GAAG,CAAC,GAAG,gBAAgB,EAAE,WAAW,CAAC,CAAC;IAEpD,0CAA0C;IAC1C,MAAM,WAAW,GAAG,eAAe,CAAC,YAAY;QAC5C,CAAC,CAAC;YACI;gBACI,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,KAAK,CAAC,UAAU;gBAC3B,IAAI,EAAE,QAAiB;gBACvB,OAAO,EAAE,eAAe,CAAC,YAAY;gBACrC,SAAS,EAAE,MAAM,EAAE;aACtB;YACD,GAAG,QAAQ;SACd;QACH,CAAC,CAAC,QAAQ,CAAC;IAEf,6BAA6B;IAC7B,MAAM,QAAQ,GAAG,IAAI,qBAAqB;IACtC,8DAA8D;IAC9D,eAAe,CAAC,UAAiB,EACjC,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,gBAAgB,EACrB,gBAAgB,CACnB,CAAC;IAEF,sBAAsB;IACtB,MAAM,OAAO,GAAG,KAAK,IAAmB,EAAE;QACtC,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;YACxB,gBAAgB;YAChB,QAAQ,EAAE,WAAW;YACrB,QAAQ;YACR,QAAQ;YACR,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,KAAK,EAAE,IAAI,CAAC,KAAK;SACpB,CAAC,CAAC;IACP,CAAC,CAAC;IAEF,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACnB,wDAAwD;QACxD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAErC,MAAM,CAAC,IAAI,CACP,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,UAAU,EAAE,EAChD,wCAAwC,CAC3C,CAAC;QAEF,OAAO;YACH,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,MAAM,EAAE,SAAS;SACpB,CAAC;IACN,CAAC;SAAM,CAAC;QACJ,8CAA8C;QAC9C,IAAI,CAAC;YACD,MAAM,OAAO,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,+CAA+C;YAC/C,gCAAgC;QACpC,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/C,OAAO;YACH,OAAO,EAAE,SAAS,CAAC,EAAE;YACrB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,MAAM,EAAE,SAAS,CAAC,MAAM;SAC3B,CAAC;IACN,CAAC;AACL,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,KAAoB,EAAE,IAAiC;IAC5E,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,KAAsB,EAAE,IAAiC;IAChF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB;IAEzE,MAAM,mBAAmB,GAAG,CAAC,SAAS,EAAE,SAAS,CAAU,CAAC;IAC5D,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,MAA4C,CAAC,EAAE,CAAC;QACnF,MAAM,IAAI,SAAS,CACf,sBAAsB,EACtB,SAAS,KAAK,CAAC,OAAO,iBAAiB,IAAI,CAAC,MAAM,2BAA2B,CAChF,CAAC;IACN,CAAC;IAED,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACrC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,KAAkB,EAAE,IAAiC;IAC5E,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB;AACvE,CAAC"}
+{"version":3,"file":"task.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/task.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAKtC,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAIzD,OAAO,EAAE,iBAAiB,IAAI,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAYpF,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAEhD,OAAO,EAAE,oBAAoB,EAAiB,MAAM,YAAY,CAAC;AAoBjE;;;GAGG;AACH,KAAK,UAAU,gBAAgB,CAC3B,KAA6C,EAC7C,IAAc,EACd,aAAgC;IAEhC,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAE/D,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,kBAAkB,GAAG,UAAU,EAAE,CAAC;IACxC,MAAM,UAAU,GAAG,aAAa;QAC5B,CAAC,CAAC,CAAC,aAAa,CAAC,UAAU,IAAI,aAAa,CAAC,MAAM,CAAC;QACpD,CAAC,CAAC,SAAS,CAAC;IAEhB,MAAM,gBAAgB,GAAqB;QACvC,MAAM;QACN,SAAS,EAAE,kBAAkB;QAC7B,SAAS,EAAE,eAAe,CAAC,IAAI;QAC/B,eAAe;QACf,gBAAgB,EAAE,aAAa,EAAE,SAAS;QAC1C,YAAY,EAAE,aAAa,EAAE,MAAM;QACnC,UAAU,EAAE,UAAU,IAAI,SAAS;QACnC,cAAc,EAAE,CAAC,aAAa,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;QACzD,aAAa,EAAE,CAAC;KACnB,CAAC;IAEF,MAAM,QAAQ,GAAG,cAAc,CAAC,eAAe,CAAC,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IAEtF,MAAM,WAAW,GAAG;QAChB,EAAE,EAAE,UAAU,EAAE;QAChB,SAAS,EAAE,kBAAkB;QAC7B,IAAI,EAAE,MAAe;QACrB,OAAO,EAAE,KAAK,CAAC,MAAM;QACrB,SAAS,EAAE,MAAM,EAAE;KACtB,CAAC;IACF,MAAM,QAAQ,GAAG,eAAe,CAAC,YAAY;QACzC,CAAC,CAAC;YACI;gBACI,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,kBAAkB;gBAC7B,IAAI,EAAE,QAAiB;gBACvB,OAAO,EAAE,eAAe,CAAC,YAAY;gBACrC,SAAS,EAAE,MAAM,EAAE;aACtB;YACD,WAAW;SACd;QACH,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;IAEpB,sDAAsD;IACtD,IAAI,cAAc,GAAe,SAAS,CAAC;IAC3C,IAAI,cAAkC,CAAC;IAEvC,MAAM,gBAAgB,GAAG;QACrB,YAAY,EAAE,CAAC,GAAW,EAAE,MAAkB,EAAE,MAA4B,EAAE,EAAE;YAC5E,cAAc,GAAG,MAAM,CAAC;YACxB,IAAI,MAAM,EAAE,MAAM,KAAK,SAAS;gBAAE,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC;YACjE,OAAO,EAAU,CAAC;QACtB,CAAC;QACD,gEAAgE;QAChE,WAAW,EAAE,GAAG,EAAE,GAAE,CAAC;QACrB,gEAAgE;QAChE,sBAAsB,EAAE,GAAG,EAAE,GAAE,CAAC;KACX,CAAC;IAE1B,MAAM,gBAAgB,GAAG;QACrB,gEAAgE;QAChE,aAAa,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;KACL,CAAC;IAE7B,MAAM,QAAQ,GAAG,IAAI,qBAAqB;IACtC,8DAA8D;IAC9D,eAAe,CAAC,UAAiB,EACjC,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,gBAAgB,EACrB,gBAAgB,CACnB,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IAEzC,IAAI,CAAC;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;YACxB,gBAAgB;YAChB,QAAQ;YACR,QAAQ;YACR,QAAQ;YACR,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,gBAAgB;YAC3B,YAAY,EAAE,gBAAgB;YAC9B,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,MAAM;YACN,KAAK,EAAE,IAAI,CAAC,KAAK;SACpB,CAAC,CAAC;IACP,CAAC;IAAC,MAAM,CAAC;QACL,4DAA4D;IAChE,CAAC;IAED,MAAM,KAAK,GAAG,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IAEpD,OAAO;QACH,OAAO,EAAE,MAAM;QACf,MAAM,EAAE,cAAc;QACtB,MAAM,EAAE,cAAc;QACtB,KAAK;KACR,CAAC;AACN,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC1B,KAAoB,EACpB,IAAc,EACd,aAAgC;IAEhC,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;QACxB,OAAO,gBAAgB,CAAC,KAAK,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;IACxD,CAAC;IAED,sBAAsB;IACtB,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACzD,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,IAAI,SAAS,CAAC,gBAAgB,EAAE,YAAY,KAAK,CAAC,UAAU,aAAa,CAAC,CAAC;IACrF,CAAC;IAED,uCAAuC;IACvC,MAAM,eAAe,GAAG,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAE/E,qBAAqB;IACrB,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;QAC/B,SAAS,EAAE,KAAK,CAAC,UAAU;QAC3B,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,YAAY,EAAE,aAAa,EAAE,MAAM;QACnC,cAAc,EAAE,CAAC,aAAa,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;KAC5D,CAAC,CAAC;IAEH,6EAA6E;IAC7E,yEAAyE;IACzE,MAAM,UAAU,GAAG,aAAa;QAC5B,CAAC,CAAC,CAAC,aAAa,CAAC,UAAU,IAAI,aAAa,CAAC,MAAM,CAAC;QACpD,CAAC,CAAC,SAAS,CAAC;IAEhB,6BAA6B;IAC7B,MAAM,gBAAgB,GAAqB;QACvC,MAAM,EAAE,IAAI,CAAC,EAAE;QACf,SAAS,EAAE,KAAK,CAAC,UAAU;QAC3B,SAAS,EAAE,eAAe,CAAC,IAAI;QAC/B,eAAe;QACf,gBAAgB,EAAE,aAAa,EAAE,SAAS;QAC1C,YAAY,EAAE,aAAa,EAAE,MAAM;QACnC,UAAU,EAAE,UAAU,IAAI,SAAS;QACnC,cAAc,EAAE,CAAC,aAAa,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;QACzD,aAAa,EAAE,CAAC;KACnB,CAAC;IAEF,4BAA4B;IAC5B,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IAEzC,2BAA2B;IAC3B,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAEzD,oBAAoB;IACpB,MAAM,QAAQ,GAAG,cAAc,CAAC,eAAe,CAAC,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IAEtF,yBAAyB;IACzB,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACzE,MAAM,WAAW,GAAG;QAChB,EAAE,EAAE,UAAU,EAAE;QAChB,SAAS,EAAE,KAAK,CAAC,UAAU;QAC3B,IAAI,EAAE,MAAe;QACrB,OAAO,EAAE,KAAK,CAAC,MAAM;QACrB,SAAS,EAAE,MAAM,EAAE;KACtB,CAAC;IACF,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,KAAK,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;IACrE,MAAM,QAAQ,GAAG,CAAC,GAAG,gBAAgB,EAAE,WAAW,CAAC,CAAC;IAEpD,0CAA0C;IAC1C,MAAM,WAAW,GAAG,eAAe,CAAC,YAAY;QAC5C,CAAC,CAAC;YACI;gBACI,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,KAAK,CAAC,UAAU;gBAC3B,IAAI,EAAE,QAAiB;gBACvB,OAAO,EAAE,eAAe,CAAC,YAAY;gBACrC,SAAS,EAAE,MAAM,EAAE;aACtB;YACD,GAAG,QAAQ;SACd;QACH,CAAC,CAAC,QAAQ,CAAC;IAEf,6BAA6B;IAC7B,MAAM,QAAQ,GAAG,IAAI,qBAAqB;IACtC,8DAA8D;IAC9D,eAAe,CAAC,UAAiB,EACjC,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,gBAAgB,EACrB,gBAAgB,CACnB,CAAC;IAEF,sBAAsB;IACtB,MAAM,OAAO,GAAG,KAAK,IAAmB,EAAE;QACtC,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;YACxB,gBAAgB;YAChB,QAAQ,EAAE,WAAW;YACrB,QAAQ;YACR,QAAQ;YACR,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,KAAK,EAAE,IAAI,CAAC,KAAK;SACpB,CAAC,CAAC;IACP,CAAC,CAAC;IAEF,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACnB,wDAAwD;QACxD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAErC,MAAM,CAAC,IAAI,CACP,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,UAAU,EAAE,EAChD,wCAAwC,CAC3C,CAAC;QAEF,OAAO;YACH,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,MAAM,EAAE,SAAS;SACpB,CAAC;IACN,CAAC;SAAM,CAAC;QACJ,8CAA8C;QAC9C,IAAI,CAAC;YACD,MAAM,OAAO,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,+CAA+C;YAC/C,gCAAgC;QACpC,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,KAAK,GAAG,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QAC1D,OAAO;YACH,OAAO,EAAE,SAAS,CAAC,EAAE;YACrB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,KAAK;SACR,CAAC;IACN,CAAC;AACL,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,KAAoB,EAAE,IAAiC;IAC5E,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,KAAsB,EAAE,IAAiC;IAChF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB;IAEzE,MAAM,mBAAmB,GAAG,CAAC,SAAS,EAAE,SAAS,CAAU,CAAC;IAC5D,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,MAA4C,CAAC,EAAE,CAAC;QACnF,MAAM,IAAI,SAAS,CACf,sBAAsB,EACtB,SAAS,KAAK,CAAC,OAAO,iBAAiB,IAAI,CAAC,MAAM,2BAA2B,CAChF,CAAC;IACN,CAAC;IAED,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACrC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,UAAU,CACtB,KAAkB,EAClB,IAAwC;IAExC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB;IACzE,MAAM,KAAK,GAAG,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IACrD,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;AAC9B,CAAC"}

package/src/tools/usage.d.ts

@@ -0,0 +1,54 @@
+import type { BetterSQLite3Database } from "drizzle-orm/better-sqlite3";
+import { taskUsageTable } from "../db/schema.js";
+import type { TaskUsageInput, TaskUsageReport } from "../validation/usage.js";
+/**
+ * Drizzle database handle. Identical to what the stores and UsagePlugin receive.
+ */
+export type Database = BetterSQLite3Database<any>;
+/** A single row from the `task_usage` table, as returned by drizzle select. */
+export type TaskUsageRow = typeof taskUsageTable.$inferSelect;
+export interface UsageQueryResult {
+    rows: TaskUsageRow[];
+    summary: {
+        totalInputTokens: number;
+        totalOutputTokens: number;
+        totalToolCalls: number;
+        totalModelCalls: number;
+        taskCount: number;
+    };
+}
+/**
+ * `usage_query` tool implementation.
+ *
+ * Queries `task_usage` with optional filters. Every filter is optional; a bare
+ * `{}` (or `undefined`) returns the most recent rows, capped at `limit`.
+ *
+ * - `task_id`: returns that row AND the full delegation subtree
+ *   (`task_id = ? OR root_task_id = ?`).
+ * - `root_task_id`: returns the root row plus every descendant
+ *   (`task_id = ? OR root_task_id = ?`).
+ * - `agent_name`, `since`: plain equality / lower-bound filters.
+ * - `include_incomplete` (default false): when false, only `is_complete = 1`
+ *   rows are returned.
+ *
+ * Returns `{ rows: [], summary: { ...zeros, taskCount: 0 } }` when the table
+ * is empty or nothing matches.
+ */
+export declare function usageQuery(db: Database, input: TaskUsageInput): UsageQueryResult;
+/**
+ * Builds a {@link TaskUsageReport} for enriching a `task` / `result` response.
+ *
+ * - `direct`: aggregates `task_usage WHERE task_id = ?` (this task's own model
+ *   calls only).
+ * - `subtree`: aggregates `task_usage WHERE task_id = ? OR root_task_id = ?`
+ *   (this task plus every sub-task that recorded it as its root).
+ * - `taskCount`: number of distinct rows folded into the subtree.
+ *
+ * Returns `undefined` when no `task_usage` row exists for the task — i.e. the
+ * task ran zero model calls (still running, or cancelled before its first
+ * model call). See `[shape:TaskUsageReport]`.
+ *
+ * claudecli rows are present with zero token counts and are returned as-is.
+ */
+export declare function buildTaskUsageReport(db: Database, taskId: string): TaskUsageReport | undefined;
+//# sourceMappingURL=usage.d.ts.map

package/src/tools/usage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/usage.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAExE,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EACV,cAAc,EACd,eAAe,EAEhB,MAAM,wBAAwB,CAAC;AAEhC;;GAEG;AAEH,MAAM,MAAM,QAAQ,GAAG,qBAAqB,CAAC,GAAG,CAAC,CAAC;AAElD,+EAA+E;AAC/E,MAAM,MAAM,YAAY,GAAG,OAAO,cAAc,CAAC,YAAY,CAAC;AAE9D,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,YAAY,EAAE,CAAC;IACrB,OAAO,EAAE;QACP,gBAAgB,EAAE,MAAM,CAAC;QACzB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,cAAc,EAAE,MAAM,CAAC;QACvB,eAAe,EAAE,MAAM,CAAC;QACxB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AA8BD;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,UAAU,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,cAAc,GAAG,gBAAgB,CA4ChF;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,oBAAoB,CAClC,EAAE,EAAE,QAAQ,EACZ,MAAM,EAAE,MAAM,GACb,eAAe,GAAG,SAAS,CAkB7B"}

package/src/tools/usage.js

@@ -0,0 +1,126 @@
+import { and, desc, eq, gte, or } from "drizzle-orm";
+const SEVERITY = { length: 3, tool_calls: 2, stop: 1, unknown: 0 };
+function mostSevereStr(a, b) {
+    if (!a && !b)
+        return undefined;
+    const sa = SEVERITY[a ?? ""] ?? 0;
+    const sb = SEVERITY[b ?? ""] ?? 0;
+    return sa >= sb ? a : b;
+}
+import { taskUsageTable } from "../db/schema.js";
+/**
+ * Folds a set of `task_usage` rows into a single {@link UsageSummary}.
+ *
+ * claudecli rows carry zeros for token counts — they are summed as-is, so a
+ * caller seeing `inputTokens === 0` must not conclude "zero tokens consumed";
+ * it may mean the provider does not report token counts. See
+ * `[inv:claudecli-undefined]`.
+ */
+function summarise(rows) {
+    return rows.reduce((acc, row) => ({
+        inputTokens: acc.inputTokens + (row.inputTokens ?? 0),
+        outputTokens: acc.outputTokens + (row.outputTokens ?? 0),
+        modelCalls: acc.modelCalls + (row.modelCalls ?? 0),
+        toolCallCount: acc.toolCallCount + (row.toolCallCount ?? 0),
+        latencyMs: acc.latencyMs + (row.latencyMs ?? 0),
+        stopReason: mostSevereStr(acc.stopReason, row.stopReason ?? undefined),
+    }), {
+        inputTokens: 0,
+        outputTokens: 0,
+        modelCalls: 0,
+        toolCallCount: 0,
+        latencyMs: 0,
+    });
+}
+/**
+ * `usage_query` tool implementation.
+ *
+ * Queries `task_usage` with optional filters. Every filter is optional; a bare
+ * `{}` (or `undefined`) returns the most recent rows, capped at `limit`.
+ *
+ * - `task_id`: returns that row AND the full delegation subtree
+ *   (`task_id = ? OR root_task_id = ?`).
+ * - `root_task_id`: returns the root row plus every descendant
+ *   (`task_id = ? OR root_task_id = ?`).
+ * - `agent_name`, `since`: plain equality / lower-bound filters.
+ * - `include_incomplete` (default false): when false, only `is_complete = 1`
+ *   rows are returned.
+ *
+ * Returns `{ rows: [], summary: { ...zeros, taskCount: 0 } }` when the table
+ * is empty or nothing matches.
+ */
+export function usageQuery(db, input) {
+    const filters = [];
+    // task_id and root_task_id both expand to a subtree match.
+    if (input?.task_id) {
+        const id = input.task_id;
+        const subtree = or(eq(taskUsageTable.taskId, id), eq(taskUsageTable.rootTaskId, id));
+        if (subtree)
+            filters.push(subtree);
+    }
+    if (input?.root_task_id) {
+        const id = input.root_task_id;
+        const subtree = or(eq(taskUsageTable.taskId, id), eq(taskUsageTable.rootTaskId, id));
+        if (subtree)
+            filters.push(subtree);
+    }
+    if (input?.agent_name) {
+        filters.push(eq(taskUsageTable.agentName, input.agent_name));
+    }
+    if (input?.since) {
+        filters.push(gte(taskUsageTable.createdAt, input.since));
+    }
+    if (!input?.include_incomplete) {
+        filters.push(eq(taskUsageTable.isComplete, 1));
+    }
+    const limit = input?.limit ?? 50;
+    const rows = db
+        .select()
+        .from(taskUsageTable)
+        .where(filters.length > 0 ? and(...filters) : undefined)
+        .orderBy(desc(taskUsageTable.createdAt))
+        .limit(limit)
+        .all();
+    return {
+        rows,
+        summary: {
+            totalInputTokens: rows.reduce((n, r) => n + (r.inputTokens ?? 0), 0),
+            totalOutputTokens: rows.reduce((n, r) => n + (r.outputTokens ?? 0), 0),
+            totalToolCalls: rows.reduce((n, r) => n + (r.toolCallCount ?? 0), 0),
+            totalModelCalls: rows.reduce((n, r) => n + (r.modelCalls ?? 0), 0),
+            taskCount: rows.length,
+        },
+    };
+}
+/**
+ * Builds a {@link TaskUsageReport} for enriching a `task` / `result` response.
+ *
+ * - `direct`: aggregates `task_usage WHERE task_id = ?` (this task's own model
+ *   calls only).
+ * - `subtree`: aggregates `task_usage WHERE task_id = ? OR root_task_id = ?`
+ *   (this task plus every sub-task that recorded it as its root).
+ * - `taskCount`: number of distinct rows folded into the subtree.
+ *
+ * Returns `undefined` when no `task_usage` row exists for the task — i.e. the
+ * task ran zero model calls (still running, or cancelled before its first
+ * model call). See `[shape:TaskUsageReport]`.
+ *
+ * claudecli rows are present with zero token counts and are returned as-is.
+ */
+export function buildTaskUsageReport(db, taskId) {
+    const subtreeRows = db
+        .select()
+        .from(taskUsageTable)
+        .where(or(eq(taskUsageTable.taskId, taskId), eq(taskUsageTable.rootTaskId, taskId)))
+        .all();
+    if (subtreeRows.length === 0) {
+        return undefined;
+    }
+    const directRows = subtreeRows.filter(row => row.taskId === taskId);
+    return {
+        direct: summarise(directRows),
+        subtree: summarise(subtreeRows),
+        taskCount: subtreeRows.length,
+    };
+}
+//# sourceMappingURL=usage.js.map

package/src/tools/usage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/usage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAY,MAAM,aAAa,CAAC;AAE/D,MAAM,QAAQ,GAA2B,EAAE,MAAM,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;AAE3F,SAAS,aAAa,CAAC,CAAqB,EAAE,CAAqB;IAC/D,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;QAAE,OAAO,SAAS,CAAC;IAC/B,MAAM,EAAE,GAAG,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,EAAE,GAAG,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC;IAClC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC5B,CAAC;AAID,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AA2BjD;;;;;;;GAOG;AACH,SAAS,SAAS,CAAC,IAAoB;IACrC,OAAO,IAAI,CAAC,MAAM,CAChB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QACb,WAAW,EAAE,GAAG,CAAC,WAAW,GAAG,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC;QACrD,YAAY,EAAE,GAAG,CAAC,YAAY,GAAG,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,CAAC;QACxD,UAAU,EAAE,GAAG,CAAC,UAAU,GAAG,CAAC,GAAG,CAAC,UAAU,IAAI,CAAC,CAAC;QAClD,aAAa,EAAE,GAAG,CAAC,aAAa,GAAG,CAAC,GAAG,CAAC,aAAa,IAAI,CAAC,CAAC;QAC3D,SAAS,EAAE,GAAG,CAAC,SAAS,GAAG,CAAC,GAAG,CAAC,SAAS,IAAI,CAAC,CAAC;QAC/C,UAAU,EAAE,aAAa,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,UAAU,IAAI,SAAS,CAAC;KACvE,CAAC,EACF;QACE,WAAW,EAAE,CAAC;QACd,YAAY,EAAE,CAAC;QACf,UAAU,EAAE,CAAC;QACb,aAAa,EAAE,CAAC;QAChB,SAAS,EAAE,CAAC;KACb,CACF,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,UAAU,UAAU,CAAC,EAAY,EAAE,KAAqB;IAC5D,MAAM,OAAO,GAAU,EAAE,CAAC;IAE1B,2DAA2D;IAC3D,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;QACnB,MAAM,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC;QACzB,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC;QACrF,IAAI,OAAO;YAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IACD,IAAI,KAAK,EAAE,YAAY,EAAE,CAAC;QACxB,MAAM,EAAE,GAAG,KAAK,CAAC,YAAY,CAAC;QAC9B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC;QACrF,IAAI,OAAO;YAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IACD,IAAI,KAAK,EAAE,UAAU,EAAE,CAAC;QACtB,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,cAAc,CAAC,SAAS,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;IAC/D,CAAC;IACD,IAAI,KAAK,EAAE,KAAK,EAAE,CAAC;QACjB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3D,CAAC;IACD,IAAI,CAAC,KAAK,EAAE,kBAAkB,EAAE,CAAC;QAC/B,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,KAAK,GAAG,KAAK,EAAE,KAAK,IAAI,EAAE,CAAC;IAEjC,MAAM,IAAI,GAAG,EAAE;SACZ,MAAM,EAAE;SACR,IAAI,CAAC,cAAc,CAAC;SACpB,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;SACvD,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;SACvC,KAAK,CAAC,KAAK,CAAC;SACZ,GAAG,EAAE,CAAC;IAET,OAAO;QACL,IAAI;QACJ,OAAO,EAAE;YACP,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;YACpE,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;YACtE,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,aAAa,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;YACpE,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;YAClE,SAAS,EAAE,IAAI,CAAC,MAAM;SACvB;KACF,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,oBAAoB,CAClC,EAAY,EACZ,MAAc;IAEd,MAAM,WAAW,GAAG,EAAE;SACnB,MAAM,EAAE;SACR,IAAI,CAAC,cAAc,CAAC;SACpB,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;SACnF,GAAG,EAAE,CAAC;IAET,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;IAEpE,OAAO;QACL,MAAM,EAAE,SAAS,CAAC,UAAU,CAAC;QAC7B,OAAO,EAAE,SAAS,CAAC,WAAW,CAAC;QAC/B,SAAS,EAAE,WAAW,CAAC,MAAM;KAC9B,CAAC;AACJ,CAAC"}

package/src/validation/agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/validation/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,eAAO,MAAM,iBAAiB;;;;;iBAK5B,CAAC;AAsDH,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2BAK/B,CAAC;AAEH,YAAY,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAG5D,eAAO,MAAM,sBAAsB;;iBAKjC,CAAC;AAEH,YAAY,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAG9D,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAahC,CAAC;AAEH,YAAY,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAI7D,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAIjC,CAAC;AAaH,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAGjC,CAAC;AAEH,eAAO,MAAM,oBAAoB;;iBAE/B,CAAC;AAEH,eAAO,MAAM,sBAAsB;;iBAEjC,CAAC;AAEH,eAAO,MAAM,oBAAoB,+CAA0B,CAAC;AAE5D,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAClE,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC"}
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/validation/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,eAAO,MAAM,iBAAiB;;;;;iBAK5B,CAAC;AAgEH,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2BAK/B,CAAC;AAEH,YAAY,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAG5D,eAAO,MAAM,sBAAsB;;iBAKjC,CAAC;AAEH,YAAY,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAG9D,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAahC,CAAC;AAEH,YAAY,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAI7D,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAIjC,CAAC;AAaH,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAGjC,CAAC;AAEH,eAAO,MAAM,oBAAoB;;iBAE/B,CAAC;AAEH,eAAO,MAAM,sBAAsB;;iBAEjC,CAAC;AAEH,eAAO,MAAM,oBAAoB,+CAA0B,CAAC;AAE5D,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAClE,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC"}

package/src/validation/agent.js

@@ -20,11 +20,20 @@ const anthropicProviderSchema = z.object({
     timeoutMs: z.number().int().positive().optional(),
     retryConfig: retryConfigSchema.optional(),
 });
+// OpenAI-compatible servers require /vN in the base URL — the SDK appends
+// /chat/completions directly, so http://host:1234 → 404 every time.
+const versionedBaseUrlSchema = z
+    .string()
+    .url()
+    .refine((url) => /\/v\d+\/?$/.test(url), {
+    message: 'baseURL must include the API version path (e.g. "http://localhost:1234/v1")',
+})
+    .optional();
 const openaiProviderSchema = z.object({
     type: z.literal("openai"),
     model: z.string(),
     apiKeyEnv: z.string().optional(),
-    baseURL: z.string().url().optional(),
+    baseURL: versionedBaseUrlSchema,
     temperature: z.number().min(0).max(1).optional(),
     maxTokens: z.number().int().positive().optional(),
     timeoutMs: z.number().int().positive().optional(),
@@ -34,7 +43,7 @@ const lmstudioProviderSchema = z.object({
     type: z.literal("lmstudio"),
     model: z.string(),
     apiKeyEnv: z.string().optional(),
-    baseURL: z.string().url().optional(),
+    baseURL: versionedBaseUrlSchema,
     temperature: z.number().min(0).max(1).optional(),
     maxTokens: z.number().int().positive().optional(),
     timeoutMs: z.number().int().positive().optional(),