@adhd/agent-mcp 2.0.0 → 2.1.0

package/src/tools/task.js

@@ -1,406 +0,0 @@
-import { logger } from "../logger.js";
-import { config } from "../config.js";
-import { resolveHitl } from "../engine/orchestrator.js";
-import { createProvider } from "../providers/factory.js";
-import { McpClientRegistry as McpClientRegistryCtor } from "../clients/registry.js";
-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: runs a prompt against an agent without creating a
- * session row or persisting any messages. Persists a tasks row + task_events +
- * task_usage so the run is observable via task_list / result / metrics.
- */
-async function runEphemeralTask(input, deps, callerContext) {
-    const agentDefinition = deps.agentStore.read(input.agent_name);
-    const taskId = generateId();
-    // Use a synthetic session ID only as an in-memory identifier for message
-    // threading within this run — it is never written to the DB.
-    const ephemeralSessionId = generateId();
-    const rootTaskId = callerContext
-        ? (callerContext.rootTaskId ?? callerContext.taskId)
-        : undefined;
-    // Persist the tasks row so this run is observable (task_list, result tool, metrics).
-    // session_id is NULL and is_ephemeral=1 — no sessions row is created.
-    deps.taskStore.create({
-        id: taskId,
-        sessionId: null,
-        isEphemeral: true,
-        prompt: input.prompt,
-        parentTaskId: callerContext?.taskId,
-        recursionDepth: (callerContext?.recursionDepth ?? -1) + 1,
-    });
-    const executionContext = {
-        taskId,
-        sessionId: ephemeralSessionId,
-        agentName: agentDefinition.name,
-        agentDefinition,
-        callingAgentName: callerContext?.agentName,
-        parentTaskId: callerContext?.taskId,
-        rootTaskId: rootTaskId ?? undefined,
-        recursionDepth: (callerContext?.recursionDepth ?? -1) + 1,
-        toolCallCount: 0,
-    };
-    const provider = createProvider(agentDefinition.provider, agentDefinition.mcpServers);
-    const userMessage = {
-        id: generateId(),
-        sessionId: ephemeralSessionId,
-        role: "user",
-        content: input.prompt,
-        createdAt: nowIso(),
-    };
-    const messages = agentDefinition.systemPrompt
-        ? [
-            {
-                id: generateId(),
-                sessionId: ephemeralSessionId,
-                role: "system",
-                content: agentDefinition.systemPrompt,
-                createdAt: nowIso(),
-            },
-            userMessage,
-        ]
-        : [userMessage];
-    // No messages are persisted — noopSessionStore swallows appendMessage calls.
-    const noopSessionStore = {
-        // eslint-disable-next-line @typescript-eslint/no-empty-function
-        appendMessage: async () => { },
-    };
-    const registry = new McpClientRegistryCtor(
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    agentDefinition.mcpServers, deps.selfUrl, deps.inProcessDescriptors, deps.inProcessHandler, executionContext);
-    const controller = new AbortController();
-    deps.taskStore.registerCancellation(taskId, controller);
-    try {
-        await deps.orchestrator.run({
-            executionContext,
-            messages,
-            registry,
-            provider,
-            policy: deps.policy,
-            taskStore: deps.taskStore, // REAL store — events + status persisted
-            sessionStore: noopSessionStore,
-            signal: controller.signal,
-            taskId,
-            hooks: deps.hooks,
-            isEphemeral: true, // forbids request_human_input
-        });
-    }
-    catch {
-        // Orchestrator already updated status via deps.taskStore
-    }
-    finally {
-        deps.taskStore.unregisterCancellation(taskId);
-    }
-    const finalTask = deps.taskStore.read(taskId);
-    const usage = buildTaskUsageReport(deps.db, taskId);
-    return {
-        task_id: taskId,
-        status: finalTask.status,
-        result: finalTask.result,
-        usage,
-    };
-}
-/**
- * `task` tool — runs a prompt against a session's agent (session mode) or runs a
- * one-shot ephemeral task without persisting any context (agent_name mode).
- *
- * Session mode mandatory order (per plan Gap 17):
- *  1. Validate session exists and is active
- *  2. Load snapshotted AgentDefinition
- *  3. Create Task row
- *  4. Build ExecutionContext
- *  5. Create AbortController
- *  6. Register cancellation
- *  7. Build provider
- *  8. Build per-task registry
- *  9. Run orchestrator (background or sync)
- */
-export async function taskTool(input, deps, callerContext) {
-    if ("agent_name" in input) {
-        return runEphemeralTask(input, deps, callerContext);
-    }
-    // 1. Validate session
-    const session = deps.sessionStore.read(input.session_id);
-    if (session.status !== "active") {
-        throw new ToolError("SESSION_CLOSED", `Session '${input.session_id}' is closed`);
-    }
-    // 2. Load snapshotted agent definition
-    const agentDefinition = deps.sessionStore.getAgentDefinition(input.session_id);
-    // 3. Create task row
-    // Validate no dependency cycle before inserting the row.
-    // [inv:cycle-check-synchronous] — throws ToolError("VALIDATION_ERROR") on cycle.
-    const dependsOn = input.depends_on ?? [];
-    const onUpstreamFailure = input
-        .on_upstream_failure;
-    // Generate the id up front so the cycle check validates the SAME id that is
-    // inserted below (TaskStore.create reuses input.id when provided).
-    const newTaskId = generateId();
-    if (dependsOn.length > 0) {
-        deps.dagEngine.validateNoCycle(newTaskId, dependsOn);
-    }
-    const task = deps.taskStore.create({
-        id: newTaskId,
-        sessionId: input.session_id,
-        prompt: input.prompt,
-        parentTaskId: callerContext?.taskId,
-        recursionDepth: (callerContext?.recursionDepth ?? -1) + 1,
-        dependsOn: dependsOn.length > 0 ? dependsOn : undefined,
-        onUpstreamFailure,
-    });
-    // 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,
-        sessionId: input.session_id,
-        agentName: agentDefinition.name,
-        agentDefinition,
-        callingAgentName: callerContext?.agentName,
-        parentTaskId: callerContext?.taskId,
-        rootTaskId: rootTaskId ?? undefined,
-        recursionDepth: (callerContext?.recursionDepth ?? -1) + 1,
-        toolCallCount: 0,
-    };
-    // 5. Create AbortController
-    const controller = new AbortController();
-    // 6. Register cancellation
-    deps.taskStore.registerCancellation(task.id, controller);
-    // 7. Build provider
-    const provider = createProvider(agentDefinition.provider, agentDefinition.mcpServers);
-    // Build initial messages
-    const existingMessages = deps.sessionStore.getMessages(input.session_id);
-    const userMessage = {
-        id: generateId(),
-        sessionId: input.session_id,
-        role: "user",
-        content: input.prompt,
-        createdAt: nowIso(),
-    };
-    await deps.sessionStore.appendMessage(input.session_id, userMessage);
-    const messages = [...existingMessages, userMessage];
-    // Add system message if the agent has one
-    const allMessages = agentDefinition.systemPrompt
-        ? [
-            {
-                id: generateId(),
-                sessionId: input.session_id,
-                role: "system",
-                content: agentDefinition.systemPrompt,
-                createdAt: nowIso(),
-            },
-            ...messages,
-        ]
-        : messages;
-    // 8. Build per-task registry
-    const registry = new McpClientRegistryCtor(
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    agentDefinition.mcpServers, deps.selfUrl, deps.inProcessDescriptors, deps.inProcessHandler, executionContext);
-    // 9. Run orchestrator
-    const runTask = async () => {
-        try {
-            await deps.orchestrator.run({
-                executionContext,
-                messages: allMessages,
-                registry,
-                provider,
-                policy: deps.policy,
-                taskStore: deps.taskStore,
-                sessionStore: deps.sessionStore,
-                signal: controller.signal,
-                taskId: task.id,
-                hooks: deps.hooks,
-            });
-        }
-        finally {
-            // [inv:dispatch-on-completion] — dispatchReady fires on every terminal
-            // event (completed, failed, cancelled) so dependent waiting tasks are
-            // evaluated regardless of how this task ended.
-            await deps.dagEngine.dispatchReady(task.id);
-        }
-    };
-    // Compute stream_url when streaming is requested for a backgrounded task.
-    // A synchronous task is already complete by the time the caller gets the
-    // response, so an SSE URL for it would only ever hit the terminal-on-connect
-    // path and close immediately — omit it.
-    const sseBaseUrl = config.sse.baseUrl;
-    const streamUrl = input.stream && input.background
-        ? `${sseBaseUrl}/tasks/${task.id}/stream`
-        : undefined;
-    if (input.background) {
-        // Enqueue for background execution — return immediately
-        deps.queue.enqueue(task.id, runTask);
-        logger.info({ taskId: task.id, sessionId: input.session_id }, "Task enqueued for background execution");
-        const response = {
-            task_id: task.id,
-            status: "pending",
-        };
-        if (streamUrl) {
-            response.stream_url = streamUrl;
-        }
-        return response;
-    }
-    else {
-        // Synchronous execution — wait for completion
-        try {
-            await runTask();
-        }
-        catch (error) {
-            // Orchestrator already updated the task status
-            // Return the current task state
-        }
-        const finalTask = deps.taskStore.read(task.id);
-        const usage = buildTaskUsageReport(deps.db, finalTask.id);
-        const response = {
-            task_id: finalTask.id,
-            status: finalTask.status,
-            result: finalTask.result,
-            usage,
-        };
-        if (streamUrl) {
-            response.stream_url = streamUrl;
-        }
-        return response;
-    }
-}
-/**
- * Re-enqueue an existing task row that is already in "pending" status.
- *
- * Called by DagEngine.dispatchFn (built in index.ts) when a waiting task
- * transitions to pending after all its dependencies complete. The task row
- * already exists in the DB — this function only builds the runtime context
- * (executionContext, messages, provider, registry) and enqueues it.
- *
- * Also used on server startup to re-enqueue tasks orphaned by a crash between
- * DagEngine's DB update and queue.enqueue().
- */
-export async function enqueueExistingTask(taskId, deps) {
-    const task = deps.taskStore.read(taskId);
-    // Ephemeral tasks have no session row and no in-memory provider context.
-    // If the process restarted, the orchestrator context is gone and the task
-    // cannot be resumed. Mark it failed so it doesn't linger as pending.
-    if (task.isEphemeral || !task.sessionId) {
-        logger.warn({ taskId, isEphemeral: task.isEphemeral }, "enqueueExistingTask: skipping ephemeral task — context lost on restart");
-        try {
-            deps.taskStore.updateStatus(taskId, "failed", {
-                error: "Ephemeral task context lost on server restart; create a new task.",
-            });
-        }
-        catch {
-            // Already in a terminal state — ignore
-        }
-        return;
-    }
-    const session = deps.sessionStore.read(task.sessionId);
-    if (session.status !== "active") {
-        logger.warn({ taskId, sessionId: task.sessionId }, "enqueueExistingTask: session is not active, skipping dispatch");
-        return;
-    }
-    const agentDefinition = deps.sessionStore.getAgentDefinition(task.sessionId);
-    const executionContext = {
-        taskId,
-        sessionId: task.sessionId,
-        agentName: agentDefinition.name,
-        agentDefinition,
-        recursionDepth: task.recursionDepth,
-        toolCallCount: 0,
-        inputs: task.inputs ?? undefined,
-    };
-    const controller = new AbortController();
-    deps.taskStore.registerCancellation(taskId, controller);
-    const provider = createProvider(agentDefinition.provider, agentDefinition.mcpServers);
-    const existingMessages = deps.sessionStore.getMessages(task.sessionId);
-    const userMessage = {
-        id: generateId(),
-        sessionId: task.sessionId,
-        role: "user",
-        content: task.prompt,
-        createdAt: nowIso(),
-    };
-    const messages = [...existingMessages, userMessage];
-    const allMessages = agentDefinition.systemPrompt
-        ? [
-            {
-                id: generateId(),
-                sessionId: task.sessionId,
-                role: "system",
-                content: agentDefinition.systemPrompt,
-                createdAt: nowIso(),
-            },
-            ...messages,
-        ]
-        : messages;
-    const registry = new McpClientRegistryCtor(
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    agentDefinition.mcpServers, deps.selfUrl, deps.inProcessDescriptors, deps.inProcessHandler, executionContext);
-    deps.queue.enqueue(taskId, async () => {
-        try {
-            await deps.orchestrator.run({
-                executionContext,
-                messages: allMessages,
-                registry,
-                provider,
-                policy: deps.policy,
-                taskStore: deps.taskStore,
-                sessionStore: deps.sessionStore,
-                signal: controller.signal,
-                taskId,
-                hooks: deps.hooks,
-            });
-        }
-        finally {
-            // [inv:dispatch-on-completion] — dispatchReady fires on every terminal event
-            await deps.dagEngine.dispatchReady(taskId);
-        }
-    });
-}
-export function taskList(input, deps) {
-    return deps.taskStore.list(input);
-}
-export function taskCancel(input, deps) {
-    const task = deps.taskStore.read(input.task_id); // throws TASK_NOT_FOUND
-    const cancellableStatuses = ["pending", "running", "awaiting_input"];
-    if (!cancellableStatuses.includes(task.status)) {
-        throw new ToolError("TASK_NOT_CANCELLABLE", `Task '${input.task_id}' has status '${task.status}' and cannot be cancelled`);
-    }
-    deps.taskStore.cancel(input.task_id);
-    return { success: true };
-}
-/**
- * `task_resume` tool — resumes a suspended `awaiting_input` task by providing
- * the `userInput` that the orchestrator is waiting for.
- *
- * The caller must supply the `resumeToken` that was written to the DB when the
- * task was suspended. If the process restarted while the task was suspended the
- * in-memory resolver no longer exists; the task is auto-failed and
- * `TASK_NOT_RESUMABLE` is thrown so the caller knows not to retry.
- */
-export async function taskResume(input, deps) {
-    const task = deps.taskStore.read(input.taskId); // throws TASK_NOT_FOUND
-    if (task.status !== "awaiting_input") {
-        throw new ToolError("VALIDATION_ERROR", `Task '${input.taskId}' is not awaiting input (status: ${task.status})`);
-    }
-    if (task.resumeToken !== input.resumeToken) {
-        throw new ToolError("VALIDATION_ERROR", "Invalid resumeToken");
-    }
-    const resolved = resolveHitl(input.taskId, input.userInput);
-    if (!resolved) {
-        // Process restarted — in-memory resolver is gone. Auto-fail the task so it
-        // doesn't remain stranded in awaiting_input with no escape path.
-        deps.taskStore.updateStatus(input.taskId, "failed", {
-            error: "Task could not be resumed: server restarted while task was suspended. Create a new task.",
-        });
-        throw new ToolError("TASK_NOT_RESUMABLE", `Task '${input.taskId}' has no active suspension (process restarted; task has been failed)`);
-    }
-    return { success: true, taskId: input.taskId };
-}
-export function resultTool(input, deps) {
-    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 +0,0 @@
-{"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;AACtC,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAItC,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAGxD,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;AAEhD,OAAO,EAAE,oBAAoB,EAAiB,MAAM,YAAY,CAAC;AAyBjE;;;;GAIG;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,yEAAyE;IACzE,6DAA6D;IAC7D,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,qFAAqF;IACrF,sEAAsE;IACtE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;QAClB,EAAE,EAAE,MAAM;QACV,SAAS,EAAE,IAAI;QACf,WAAW,EAAE,IAAI;QACjB,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,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,6EAA6E;IAC7E,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;IACzC,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAExD,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,IAAI,CAAC,SAAS,EAAG,yCAAyC;YACrE,YAAY,EAAE,gBAAgB;YAC9B,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,MAAM;YACN,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,WAAW,EAAE,IAAI,EAAY,8BAA8B;SAC9D,CAAC,CAAC;IACP,CAAC;IAAC,MAAM,CAAC;QACL,yDAAyD;IAC7D,CAAC;YAAS,CAAC;QACP,IAAI,CAAC,SAAS,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC9C,MAAM,KAAK,GAAG,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IAEpD,OAAO;QACH,OAAO,EAAE,MAAM;QACf,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,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,yDAAyD;IACzD,iFAAiF;IACjF,MAAM,SAAS,GAAI,KAAmC,CAAC,UAAU,IAAI,EAAE,CAAC;IACxE,MAAM,iBAAiB,GAAI,KAAmD;SACzE,mBAAmB,CAAC;IACzB,4EAA4E;IAC5E,mEAAmE;IACnE,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;IAC/B,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IACzD,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;QAC/B,EAAE,EAAE,SAAS;QACb,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;QACzD,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;QACvD,iBAAiB;KACpB,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,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;gBACxB,gBAAgB;gBAChB,QAAQ,EAAE,WAAW;gBACrB,QAAQ;gBACR,QAAQ;gBACR,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,MAAM,EAAE,UAAU,CAAC,MAAM;gBACzB,MAAM,EAAE,IAAI,CAAC,EAAE;gBACf,KAAK,EAAE,IAAI,CAAC,KAAK;aACpB,CAAC,CAAC;QACP,CAAC;gBAAS,CAAC;YACP,uEAAuE;YACvE,sEAAsE;YACtE,+CAA+C;YAC/C,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChD,CAAC;IACL,CAAC,CAAC;IAEF,0EAA0E;IAC1E,yEAAyE;IACzE,6EAA6E;IAC7E,wCAAwC;IACxC,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC;IACtC,MAAM,SAAS,GACX,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,UAAU;QAC5B,CAAC,CAAC,GAAG,UAAU,UAAU,IAAI,CAAC,EAAE,SAAS;QACzC,CAAC,CAAC,SAAS,CAAC;IAEpB,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,MAAM,QAAQ,GAAmB;YAC7B,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,MAAM,EAAE,SAAS;SACpB,CAAC;QACF,IAAI,SAAS,EAAE,CAAC;YACZ,QAAQ,CAAC,UAAU,GAAG,SAAS,CAAC;QACpC,CAAC;QACD,OAAO,QAAQ,CAAC;IACpB,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,MAAM,QAAQ,GAAmB;YAC7B,OAAO,EAAE,SAAS,CAAC,EAAE;YACrB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,KAAK;SACR,CAAC;QACF,IAAI,SAAS,EAAE,CAAC;YACZ,QAAQ,CAAC,UAAU,GAAG,SAAS,CAAC;QACpC,CAAC;QACD,OAAO,QAAQ,CAAC;IACpB,CAAC;AACL,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,MAAc,EAAE,IAAc;IACpE,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAEzC,yEAAyE;IACzE,0EAA0E;IAC1E,qEAAqE;IACrE,IAAI,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACtC,MAAM,CAAC,IAAI,CACP,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,EACzC,wEAAwE,CAC3E,CAAC;QACF,IAAI,CAAC;YACD,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,MAAM,EAAE,QAAQ,EAAE;gBAC1C,KAAK,EAAE,mEAAmE;aAC7E,CAAC,CAAC;QACP,CAAC;QAAC,MAAM,CAAC;YACL,uCAAuC;QAC3C,CAAC;QACD,OAAO;IACX,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAEvD,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,CACP,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,EACrC,+DAA+D,CAClE,CAAC;QACF,OAAO;IACX,CAAC;IAED,MAAM,eAAe,GAAG,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAE7E,MAAM,gBAAgB,GAAqB;QACvC,MAAM;QACN,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,SAAS,EAAE,eAAe,CAAC,IAAI;QAC/B,eAAe;QACf,cAAc,EAAE,IAAI,CAAC,cAAc;QACnC,aAAa,EAAE,CAAC;QAChB,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,SAAS;KACnC,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAExD,MAAM,QAAQ,GAAG,cAAc,CAAC,eAAe,CAAC,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IAEtF,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACvE,MAAM,WAAW,GAAG;QAChB,EAAE,EAAE,UAAU,EAAE;QAChB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,IAAI,EAAE,MAAe;QACrB,OAAO,EAAE,IAAI,CAAC,MAAM;QACpB,SAAS,EAAE,MAAM,EAAE;KACtB,CAAC;IACF,MAAM,QAAQ,GAAG,CAAC,GAAG,gBAAgB,EAAE,WAAW,CAAC,CAAC;IAEpD,MAAM,WAAW,GAAG,eAAe,CAAC,YAAY;QAC5C,CAAC,CAAC;YACI;gBACI,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,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,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,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE;QAClC,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;gBACxB,gBAAgB;gBAChB,QAAQ,EAAE,WAAW;gBACrB,QAAQ;gBACR,QAAQ;gBACR,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,MAAM,EAAE,UAAU,CAAC,MAAM;gBACzB,MAAM;gBACN,KAAK,EAAE,IAAI,CAAC,KAAK;aACpB,CAAC,CAAC;QACP,CAAC;gBAAS,CAAC;YACP,6EAA6E;YAC7E,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC/C,CAAC;IACL,CAAC,CAAC,CAAC;AACP,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,EAAE,gBAAgB,CAAU,CAAC;IAC9E,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;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC5B,KAAiE,EACjE,IAAiC;IAEjC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,wBAAwB;IAExE,IAAI,IAAI,CAAC,MAAM,KAAK,gBAAgB,EAAE,CAAC;QACnC,MAAM,IAAI,SAAS,CACf,kBAAkB,EAClB,SAAS,KAAK,CAAC,MAAM,oCAAoC,IAAI,CAAC,MAAM,GAAG,CAC1E,CAAC;IACN,CAAC;IAED,IAAI,IAAI,CAAC,WAAW,KAAK,KAAK,CAAC,WAAW,EAAE,CAAC;QACzC,MAAM,IAAI,SAAS,CAAC,kBAAkB,EAAE,qBAAqB,CAAC,CAAC;IACnE,CAAC;IAED,MAAM,QAAQ,GAAG,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;IAC5D,IAAI,CAAC,QAAQ,EAAE,CAAC;QACZ,2EAA2E;QAC3E,iEAAiE;QACjE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,MAAM,EAAE,QAAQ,EAAE;YAChD,KAAK,EAAE,0FAA0F;SACpG,CAAC,CAAC;QACH,MAAM,IAAI,SAAS,CACf,oBAAoB,EACpB,SAAS,KAAK,CAAC,MAAM,sEAAsE,CAC9F,CAAC;IACN,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;AACnD,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

@@ -1,60 +0,0 @@
-import type { BetterSQLite3Database } from "drizzle-orm/better-sqlite3";
-import { taskUsageTable } from "../db/schema.js";
-import type { GroupedUsageRow, 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[];
-    /**
-     * Populated only when `group_by` is set in the input. Each entry aggregates
-     * all matching tasks along the chosen dimension (agent, model, or provider).
-     * `rows` is empty when `groups` is present.
-     */
-    groups?: GroupedUsageRow[];
-    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

@@ -1 +0,0 @@
-{"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,EAAc,MAAM,iBAAiB,CAAC;AAC7D,OAAO,KAAK,EACV,eAAe,EACf,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;;;;OAIG;IACH,MAAM,CAAC,EAAE,eAAe,EAAE,CAAC;IAC3B,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,CA8FhF;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,oBAAoB,CAClC,EAAE,EAAE,QAAQ,EACZ,MAAM,EAAE,MAAM,GACb,eAAe,GAAG,SAAS,CAkB7B"}

package/src/tools/usage.js

@@ -1,174 +0,0 @@
-import { and, desc, eq, gte, or, sql } 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, tasksTable } 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 whereClause = filters.length > 0 ? and(...filters) : undefined;
-    const limit = input?.limit ?? 50;
-    // ── Grouped aggregation path ──────────────────────────────────────────────
-    if (input?.group_by) {
-        // Map the group_by enum to the actual column reference.
-        let groupCol;
-        if (input.group_by === "model") {
-            groupCol = taskUsageTable.model;
-        }
-        else if (input.group_by === "provider") {
-            groupCol = taskUsageTable.providerType;
-        }
-        else {
-            groupCol = taskUsageTable.agentName; // "agent"
-        }
-        const groups = db
-            .select({
-            key: groupCol,
-            taskCount: sql `count(*)`,
-            completedCount: sql `sum(case when ${tasksTable.status} = 'completed' then 1 else 0 end)`,
-            failedCount: sql `sum(case when ${tasksTable.status} = 'failed' then 1 else 0 end)`,
-            cancelledCount: sql `sum(case when ${tasksTable.status} = 'cancelled' then 1 else 0 end)`,
-            inputTokens: sql `sum(${taskUsageTable.inputTokens})`,
-            outputTokens: sql `sum(${taskUsageTable.outputTokens})`,
-            toolCallCount: sql `sum(${taskUsageTable.toolCallCount})`,
-            modelCalls: sql `sum(${taskUsageTable.modelCalls})`,
-            avgLatencyMs: sql `avg(case when ${taskUsageTable.latencyMs} > 0 then ${taskUsageTable.latencyMs} else null end)`,
-            cacheReadTokens: sql `sum(${taskUsageTable.cacheReadTokens})`,
-            cacheCreationTokens: sql `sum(${taskUsageTable.cacheCreationTokens})`,
-        })
-            .from(taskUsageTable)
-            .leftJoin(tasksTable, eq(taskUsageTable.taskId, tasksTable.id))
-            .where(whereClause)
-            .groupBy(groupCol)
-            .orderBy(desc(sql `sum(${taskUsageTable.inputTokens} + ${taskUsageTable.outputTokens})`))
-            .limit(limit)
-            .all();
-        const totalInputTokens = groups.reduce((n, r) => n + r.inputTokens, 0);
-        const totalOutputTokens = groups.reduce((n, r) => n + r.outputTokens, 0);
-        const totalToolCalls = groups.reduce((n, r) => n + r.toolCallCount, 0);
-        const totalModelCalls = groups.reduce((n, r) => n + r.modelCalls, 0);
-        const taskCount = groups.reduce((n, r) => n + r.taskCount, 0);
-        return {
-            rows: [],
-            groups,
-            summary: { totalInputTokens, totalOutputTokens, totalToolCalls, totalModelCalls, taskCount },
-        };
-    }
-    // ── Raw rows path (original behaviour) ───────────────────────────────────
-    const rows = db
-        .select()
-        .from(taskUsageTable)
-        .where(whereClause)
-        .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