@dexto/orchestration 1.6.0 → 1.6.1

package/dist/index.d.cts

@@ -1,9 +1,615 @@
-export { SignalBus, SignalHandler, SignalPredicate } from './signal-bus.cjs';
-export { TaskRegistry, TaskRegistryConfig } from './task-registry.cjs';
-export { ConditionEngine } from './condition-engine.cjs';
-export { AgentState, ProcessResult, RegisterTaskOptions, Signal, SignalType, Task, TaskEntry, TaskFilter, TaskInfo, TaskResult, TaskStatus, WaitCondition, WaitResult } from './types.cjs';
-export { WaitForInput, WaitForInputSchema, WaitForOutput, createWaitForTool } from './tools/wait-for.cjs';
-export { CheckTaskInput, CheckTaskInputSchema, CheckTaskOutput, createCheckTaskTool } from './tools/check-task.cjs';
-export { ListTasksInput, ListTasksInputSchema, ListTasksOutput, TaskListItem, createListTasksTool } from './tools/list-tasks.cjs';
-import 'zod';
-import '@dexto/core';
+import { z } from 'zod';
+import { Tool } from '@dexto/core';
+
+/**
+ * Orchestration Types
+ *
+ * Core type definitions for the agent orchestration layer.
+ * Defines tasks, signals, wait conditions, and agent states.
+ */
+/**
+ * Task result from agent or process execution
+ */
+interface TaskResult {
+    success: boolean;
+    response?: string;
+    error?: string;
+    agentId?: string;
+    tokenUsage?: {
+        input: number;
+        output: number;
+        total: number;
+    };
+}
+/**
+ * Process result from shell command execution
+ */
+interface ProcessResult {
+    stdout: string;
+    stderr: string;
+    exitCode: number;
+    duration?: number;
+}
+/**
+ * Task types - discriminated union of different background task kinds
+ */
+type Task = {
+    type: 'agent';
+    taskId: string;
+    agentId: string;
+    taskDescription: string;
+    promise: Promise<TaskResult>;
+} | {
+    type: 'process';
+    taskId: string;
+    processId: string;
+    command: string;
+    promise: Promise<ProcessResult>;
+} | {
+    type: 'generic';
+    taskId: string;
+    description: string;
+    promise: Promise<unknown>;
+};
+/**
+ * Task lifecycle status
+ */
+type TaskStatus = 'pending' | 'running' | 'completed' | 'failed' | 'cancelled';
+/**
+ * Entry in the task registry tracking a background task
+ */
+interface TaskEntry {
+    task: Task;
+    status: TaskStatus;
+    startedAt: Date;
+    completedAt?: Date;
+    result?: unknown;
+    error?: string;
+    /** If true, auto-trigger agent turn on completion */
+    notify?: boolean;
+    /** Timeout handle for this task (if set) */
+    timeoutHandle: ReturnType<typeof setTimeout> | undefined;
+}
+/**
+ * Signal types - events that trigger state transitions
+ */
+type Signal = {
+    type: 'task:completed';
+    taskId: string;
+    result: unknown;
+} | {
+    type: 'task:failed';
+    taskId: string;
+    error: string;
+} | {
+    type: 'task:cancelled';
+    taskId: string;
+} | {
+    type: 'timeout';
+    conditionId: string;
+} | {
+    type: 'user:input';
+    content: string;
+    sessionId: string;
+} | {
+    type: 'external';
+    source: string;
+    data: unknown;
+};
+/**
+ * Extract signal type string
+ */
+type SignalType = Signal['type'];
+/**
+ * Wait conditions - composable conditions for suspension
+ */
+type WaitCondition = {
+    type: 'task';
+    taskId: string;
+} | {
+    type: 'any';
+    conditions: WaitCondition[];
+} | {
+    type: 'all';
+    conditions: WaitCondition[];
+} | {
+    type: 'timeout';
+    ms: number;
+    conditionId: string;
+} | {
+    type: 'race';
+    task: WaitCondition;
+    timeout: WaitCondition;
+};
+/**
+ * Agent loop states
+ */
+type AgentState = 'idle' | 'processing' | 'waiting';
+/**
+ * Task info returned by list/check operations (without internal promise)
+ */
+interface TaskInfo {
+    taskId: string;
+    type: Task['type'];
+    status: TaskStatus;
+    startedAt: Date;
+    completedAt?: Date;
+    duration?: number;
+    description?: string;
+    result?: unknown;
+    error?: string;
+    cancelReason?: string;
+}
+/**
+ * Filter options for listing tasks
+ */
+interface TaskFilter {
+    status?: TaskStatus | TaskStatus[];
+    type?: Task['type'];
+}
+/**
+ * Result from waiting on a condition
+ */
+interface WaitResult {
+    signal: Signal;
+    /** For 'all' conditions, contains all signals */
+    allSignals?: Signal[];
+}
+/**
+ * Options for registering a task
+ */
+interface RegisterTaskOptions {
+    /** Auto-trigger agent turn on completion */
+    notify?: boolean;
+    /** Timeout in milliseconds */
+    timeout?: number;
+}
+
+/**
+ * SignalBus
+ *
+ * Event emitter for routing orchestration signals.
+ * Supports typed subscriptions and promise-based waiting.
+ */
+
+/**
+ * Handler function for signal subscriptions
+ */
+type SignalHandler<T extends Signal = Signal> = (signal: T) => void;
+/**
+ * Predicate function for filtering signals
+ */
+type SignalPredicate = (signal: Signal) => boolean;
+/**
+ * SignalBus - Routes signals between orchestration components
+ */
+declare class SignalBus {
+    private emitter;
+    constructor();
+    /**
+     * Emit a signal to all subscribers
+     */
+    emit(signal: Signal): void;
+    /**
+     * Subscribe to signals of a specific type
+     * @returns Unsubscribe function
+     */
+    on<T extends SignalType>(type: T, handler: SignalHandler<Extract<Signal, {
+        type: T;
+    }>>): () => void;
+    /**
+     * Subscribe to all signals
+     * @returns Unsubscribe function
+     */
+    onAny(handler: SignalHandler): () => void;
+    /**
+     * Subscribe to a signal type once
+     */
+    once<T extends SignalType>(type: T, handler: SignalHandler<Extract<Signal, {
+        type: T;
+    }>>): void;
+    /**
+     * Remove a specific handler
+     */
+    off<T extends SignalType>(type: T, handler: SignalHandler<Extract<Signal, {
+        type: T;
+    }>>): void;
+    /**
+     * Wait for a signal matching the predicate
+     * @param predicate Function to test signals
+     * @param timeout Optional timeout in milliseconds
+     * @returns Promise that resolves with the matching signal
+     */
+    waitFor(predicate: SignalPredicate, timeout?: number): Promise<Signal>;
+    /**
+     * Wait for a signal for a specific task
+     */
+    waitForTask(taskId: string, timeout?: number): Promise<Signal>;
+    /**
+     * Wait for any of multiple tasks to complete
+     */
+    waitForAnyTask(taskIds: string[], timeout?: number): Promise<Signal>;
+    /**
+     * Wait for all tasks to complete
+     * @returns Promise that resolves with all signals
+     */
+    waitForAllTasks(taskIds: string[], timeout?: number, resolveInitial?: (taskId: string) => Signal | undefined): Promise<Signal[]>;
+    /**
+     * Remove all listeners
+     */
+    clear(): void;
+}
+
+/**
+ * TaskRegistry
+ *
+ * Tracks all background tasks and their results.
+ * Emits signals on task completion for the ConditionEngine.
+ */
+
+/**
+ * Configuration for TaskRegistry
+ */
+interface TaskRegistryConfig {
+    /** Maximum number of concurrent tasks (default: 20) */
+    maxTasks?: number;
+    /** TTL for completed task results in ms (default: 5 minutes) */
+    resultTTL?: number;
+}
+/**
+ * TaskRegistry - Manages background task lifecycle
+ */
+declare class TaskRegistry {
+    private tasks;
+    private signalBus;
+    private config;
+    constructor(signalBus: SignalBus, config?: TaskRegistryConfig);
+    /**
+     * Generate a unique task ID
+     */
+    private generateTaskId;
+    /**
+     * Get description from task for display
+     */
+    private getTaskDescription;
+    /**
+     * Register a new task and start tracking it
+     * @param task Task to register (must include promise)
+     * @param options Registration options
+     * @returns Task ID
+     */
+    register(task: Task, options?: RegisterTaskOptions): string;
+    /**
+     * Create and register an agent task
+     *
+     * Note: Uses agentId as the taskId so the caller can use the same ID
+     * for wait_for/check_task operations.
+     */
+    registerAgentTask(agentId: string, taskDescription: string, promise: Promise<unknown>, options?: RegisterTaskOptions): string;
+    /**
+     * Create and register a process task
+     */
+    registerProcessTask(processId: string, command: string, promise: Promise<unknown>, options?: RegisterTaskOptions): string;
+    /**
+     * Create and register a generic task
+     */
+    registerGenericTask(description: string, promise: Promise<unknown>, options?: RegisterTaskOptions): string;
+    /**
+     * Called when a task completes successfully
+     */
+    private onTaskComplete;
+    /**
+     * Called when a task fails
+     */
+    private onTaskFailed;
+    /**
+     * Cancel a running task
+     */
+    cancel(taskId: string): void;
+    /**
+     * Get task entry by ID
+     */
+    get(taskId: string): TaskEntry | undefined;
+    /**
+     * Get task status
+     */
+    getStatus(taskId: string): TaskStatus | undefined;
+    /**
+     * Get task result (if completed)
+     */
+    getResult(taskId: string): {
+        status: TaskStatus;
+        result?: unknown;
+        error?: string;
+    } | undefined;
+    /**
+     * Get task info (safe for serialization - no promise)
+     */
+    getInfo(taskId: string): TaskInfo | undefined;
+    /**
+     * List tasks matching filter
+     */
+    list(filter?: TaskFilter): TaskInfo[];
+    /**
+     * Get count of running tasks
+     */
+    getRunningCount(): number;
+    /**
+     * Get tasks that completed with notify=true and haven't been acknowledged
+     */
+    getNotifyPending(): TaskInfo[];
+    /**
+     * Mark notify tasks as acknowledged (clear notify flag)
+     */
+    acknowledgeNotify(taskIds: string[]): void;
+    /**
+     * Clean up old completed tasks
+     */
+    cleanup(olderThan?: Date): number;
+    /**
+     * Clear all tasks
+     */
+    clear(): void;
+    /**
+     * Check if a task exists
+     */
+    has(taskId: string): boolean;
+    /**
+     * Get total task count
+     */
+    get size(): number;
+}
+
+/**
+ * ConditionEngine
+ *
+ * Evaluates wait conditions and resolves when met.
+ * Supports single task, any/all of multiple tasks, timeouts, and races.
+ */
+
+type LoggerLike = {
+    debug: (message: string) => void;
+};
+/**
+ * ConditionEngine - Evaluates composable wait conditions
+ */
+declare class ConditionEngine {
+    private taskRegistry;
+    private signalBus;
+    private logger?;
+    constructor(taskRegistry: TaskRegistry, signalBus: SignalBus, logger?: LoggerLike | undefined);
+    /**
+     * Wait for a condition to be met
+     * @param condition Wait condition to evaluate
+     * @returns Promise resolving to the signal(s) that satisfied the condition
+     */
+    wait(condition: WaitCondition): Promise<WaitResult>;
+    /**
+     * Check if a condition is already satisfied (non-blocking)
+     * @returns WaitResult if satisfied, null if not
+     */
+    check(condition: WaitCondition): WaitResult | null;
+    /**
+     * Check if a single task is completed
+     */
+    private checkTask;
+    /**
+     * Check if any of the conditions is satisfied
+     */
+    private checkAny;
+    /**
+     * Check if all conditions are satisfied
+     */
+    private checkAll;
+    /**
+     * Evaluate a condition asynchronously
+     */
+    private evaluate;
+    /**
+     * Wait for a single task to complete
+     *
+     * Uses subscribe-then-check pattern to avoid race conditions where
+     * the task completes between checking and subscribing.
+     */
+    private evaluateTask;
+    /**
+     * Wait for any of the conditions to be satisfied
+     */
+    private evaluateAny;
+    /**
+     * Wait for all conditions to be satisfied
+     */
+    private evaluateAll;
+    /**
+     * Wait for a timeout
+     */
+    private evaluateTimeout;
+    /**
+     * Race a task condition against a timeout
+     */
+    private evaluateRace;
+    /**
+     * Helper to create a race condition with timeout
+     */
+    static createRaceWithTimeout(taskId: string, timeoutMs: number): WaitCondition;
+    /**
+     * Helper to create an 'any' condition from task IDs
+     */
+    static createAnyTask(taskIds: string[]): WaitCondition;
+    /**
+     * Helper to create an 'all' condition from task IDs
+     */
+    static createAllTasks(taskIds: string[]): WaitCondition;
+}
+
+/**
+ * wait_for Tool
+ *
+ * Suspends agent execution until condition is met using a blocking promise.
+ * TurnExecutor naturally awaits tool execution, so this works seamlessly.
+ */
+
+/**
+ * Input schema for wait_for tool
+ */
+declare const WaitForInputSchema: z.ZodEffects<z.ZodObject<{
+    /** Wait for a single task */
+    taskId: z.ZodOptional<z.ZodString>;
+    /** Wait for multiple tasks */
+    taskIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    /** Mode for multiple tasks: any = first to complete, all = wait for all */
+    mode: z.ZodDefault<z.ZodOptional<z.ZodEnum<["any", "all"]>>>;
+    /** Timeout in milliseconds */
+    timeout: z.ZodOptional<z.ZodNumber>;
+}, "strict", z.ZodTypeAny, {
+    mode: "any" | "all";
+    timeout?: number | undefined;
+    taskId?: string | undefined;
+    taskIds?: string[] | undefined;
+}, {
+    timeout?: number | undefined;
+    taskId?: string | undefined;
+    taskIds?: string[] | undefined;
+    mode?: "any" | "all" | undefined;
+}>, {
+    mode: "any" | "all";
+    timeout?: number | undefined;
+    taskId?: string | undefined;
+    taskIds?: string[] | undefined;
+}, {
+    timeout?: number | undefined;
+    taskId?: string | undefined;
+    taskIds?: string[] | undefined;
+    mode?: "any" | "all" | undefined;
+}>;
+type WaitForInput = z.output<typeof WaitForInputSchema>;
+/**
+ * Output from wait_for tool
+ */
+interface WaitForOutput {
+    /** Task ID that triggered the return (for 'any' mode) */
+    taskId: string;
+    /** Final status */
+    status: 'completed' | 'failed' | 'cancelled' | 'timeout';
+    /** Result if completed */
+    result?: unknown;
+    /** Error if failed */
+    error?: string;
+    /** All results for 'all' mode */
+    allResults?: Array<{
+        taskId: string;
+        status: 'completed' | 'failed' | 'cancelled';
+        result?: unknown;
+        error?: string;
+    }>;
+}
+/**
+ * Create the wait_for tool
+ */
+declare function createWaitForTool(conditionEngine: ConditionEngine): Tool<typeof WaitForInputSchema>;
+
+/**
+ * check_task Tool
+ *
+ * Non-blocking status check for a background task.
+ */
+
+/**
+ * Input schema for check_task tool
+ */
+declare const CheckTaskInputSchema: z.ZodObject<{
+    /** Task ID to check */
+    taskId: z.ZodString;
+}, "strict", z.ZodTypeAny, {
+    taskId: string;
+}, {
+    taskId: string;
+}>;
+type CheckTaskInput = z.output<typeof CheckTaskInputSchema>;
+/**
+ * Output from check_task tool
+ */
+interface CheckTaskOutput {
+    /** Task ID */
+    taskId: string;
+    /** Whether task was found */
+    found: boolean;
+    /** Task type */
+    type?: 'agent' | 'process' | 'generic';
+    /** Current status */
+    status?: 'pending' | 'running' | 'completed' | 'failed' | 'cancelled';
+    /** Description of what the task is doing */
+    description?: string;
+    /** When the task started */
+    startedAt?: string;
+    /** When the task completed (if done) */
+    completedAt?: string;
+    /** Duration in milliseconds (if completed) */
+    duration?: number;
+    /** Result (if completed successfully) */
+    result?: unknown;
+    /** Error message (if failed) */
+    error?: string;
+}
+/**
+ * Create the check_task tool
+ */
+declare function createCheckTaskTool(taskRegistry: TaskRegistry): Tool<typeof CheckTaskInputSchema>;
+
+/**
+ * list_tasks Tool
+ *
+ * List all tracked background tasks with optional filtering.
+ */
+
+/**
+ * Input schema for list_tasks tool
+ */
+declare const ListTasksInputSchema: z.ZodObject<{
+    /** Filter by status */
+    status: z.ZodDefault<z.ZodOptional<z.ZodEnum<["pending", "running", "completed", "failed", "cancelled", "all"]>>>;
+    /** Filter by type */
+    type: z.ZodOptional<z.ZodEnum<["agent", "process", "generic"]>>;
+}, "strict", z.ZodTypeAny, {
+    status: "pending" | "running" | "completed" | "failed" | "cancelled" | "all";
+    type?: "agent" | "process" | "generic" | undefined;
+}, {
+    type?: "agent" | "process" | "generic" | undefined;
+    status?: "pending" | "running" | "completed" | "failed" | "cancelled" | "all" | undefined;
+}>;
+type ListTasksInput = z.output<typeof ListTasksInputSchema>;
+/**
+ * Task info in list output
+ */
+interface TaskListItem {
+    taskId: string;
+    type: 'agent' | 'process' | 'generic';
+    status: TaskStatus;
+    description?: string;
+    startedAt: string;
+    completedAt?: string;
+    duration?: number;
+}
+/**
+ * Output from list_tasks tool
+ */
+interface ListTasksOutput {
+    /** List of tasks matching filter */
+    tasks: TaskListItem[];
+    /** Total count */
+    count: number;
+    /** Counts by status */
+    counts: {
+        running: number;
+        completed: number;
+        failed: number;
+        cancelled: number;
+        pending: number;
+    };
+}
+/**
+ * Create the list_tasks tool
+ */
+declare function createListTasksTool(taskRegistry: TaskRegistry): Tool<typeof ListTasksInputSchema>;
+
+export { type AgentState, type CheckTaskInput, CheckTaskInputSchema, type CheckTaskOutput, ConditionEngine, type ListTasksInput, ListTasksInputSchema, type ListTasksOutput, type ProcessResult, type RegisterTaskOptions, type Signal, SignalBus, type SignalHandler, type SignalPredicate, type SignalType, type Task, type TaskEntry, type TaskFilter, type TaskInfo, type TaskListItem, TaskRegistry, type TaskRegistryConfig, type TaskResult, type TaskStatus, type WaitCondition, type WaitForInput, WaitForInputSchema, type WaitForOutput, type WaitResult, createCheckTaskTool, createListTasksTool, createWaitForTool };

package/dist/tools/check-task.cjs

@@ -23,6 +23,7 @@ __export(check_task_exports, {
 });
 module.exports = __toCommonJS(check_task_exports);
 var import_zod = require("zod");
+var import_core = require("@dexto/core");
 const CheckTaskInputSchema = import_zod.z.object({
   /** Task ID to check */
   taskId: import_zod.z.string().describe("Task ID to check status of")
@@ -30,9 +31,14 @@ const CheckTaskInputSchema = import_zod.z.object({
 function createCheckTaskTool(taskRegistry) {
   return {
     id: "check_task",
-    displayName: "Check Task",
     description: "Check the status of a background task. Returns immediately without waiting. Use this to poll task status or check if a task is done.",
     inputSchema: CheckTaskInputSchema,
+    presentation: {
+      describeHeader: (input) => (0, import_core.createLocalToolCallHeader)({
+        title: "Check Task",
+        argsText: (0, import_core.truncateForHeader)(input.taskId, 80)
+      })
+    },
     execute: async (input, _context) => {
       const info = taskRegistry.getInfo(input.taskId);
       if (!info) {

package/dist/tools/check-task.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"check-task.d.ts","sourceRoot":"","sources":["../../src/tools/check-task.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAExD;;GAEG;AACH,eAAO,MAAM,oBAAoB;IAEzB,uBAAuB;;;;;;EAGlB,CAAC;AAEd,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAEnE;;GAEG;AACH,MAAM,WAAW,eAAe;IAC5B,cAAc;IACd,MAAM,EAAE,MAAM,CAAC;IACf,6BAA6B;IAC7B,KAAK,EAAE,OAAO,CAAC;IACf,gBAAgB;IAChB,IAAI,CAAC,EAAE,OAAO,GAAG,SAAS,GAAG,SAAS,CAAC;IACvC,qBAAqB;IACrB,MAAM,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,WAAW,CAAC;IACtE,4CAA4C;IAC5C,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,4BAA4B;IAC5B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,wCAAwC;IACxC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,8CAA8C;IAC9C,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,yCAAyC;IACzC,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,gCAAgC;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,YAAY,EAAE,YAAY,GAAG,IAAI,CAAC,OAAO,oBAAoB,CAAC,CAmCjG"}
+{"version":3,"file":"check-task.d.ts","sourceRoot":"","sources":["../../src/tools/check-task.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAExD;;GAEG;AACH,eAAO,MAAM,oBAAoB;IAEzB,uBAAuB;;;;;;EAGlB,CAAC;AAEd,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAEnE;;GAEG;AACH,MAAM,WAAW,eAAe;IAC5B,cAAc;IACd,MAAM,EAAE,MAAM,CAAC;IACf,6BAA6B;IAC7B,KAAK,EAAE,OAAO,CAAC;IACf,gBAAgB;IAChB,IAAI,CAAC,EAAE,OAAO,GAAG,SAAS,GAAG,SAAS,CAAC;IACvC,qBAAqB;IACrB,MAAM,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,WAAW,CAAC;IACtE,4CAA4C;IAC5C,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,4BAA4B;IAC5B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,wCAAwC;IACxC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,8CAA8C;IAC9C,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,yCAAyC;IACzC,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,gCAAgC;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,YAAY,EAAE,YAAY,GAAG,IAAI,CAAC,OAAO,oBAAoB,CAAC,CAyCjG"}

package/dist/tools/check-task.js

@@ -1,4 +1,5 @@
 import { z } from "zod";
+import { createLocalToolCallHeader, truncateForHeader } from "@dexto/core";
 const CheckTaskInputSchema = z.object({
   /** Task ID to check */
   taskId: z.string().describe("Task ID to check status of")
@@ -6,9 +7,14 @@ const CheckTaskInputSchema = z.object({
 function createCheckTaskTool(taskRegistry) {
   return {
     id: "check_task",
-    displayName: "Check Task",
     description: "Check the status of a background task. Returns immediately without waiting. Use this to poll task status or check if a task is done.",
     inputSchema: CheckTaskInputSchema,
+    presentation: {
+      describeHeader: (input) => createLocalToolCallHeader({
+        title: "Check Task",
+        argsText: truncateForHeader(input.taskId, 80)
+      })
+    },
     execute: async (input, _context) => {
       const info = taskRegistry.getInfo(input.taskId);
       if (!info) {

package/dist/tools/list-tasks.cjs

@@ -23,6 +23,7 @@ __export(list_tasks_exports, {
 });
 module.exports = __toCommonJS(list_tasks_exports);
 var import_zod = require("zod");
+var import_core = require("@dexto/core");
 const ListTasksInputSchema = import_zod.z.object({
   /** Filter by status */
   status: import_zod.z.enum(["pending", "running", "completed", "failed", "cancelled", "all"]).optional().default("all").describe("Filter tasks by status"),
@@ -32,9 +33,20 @@ const ListTasksInputSchema = import_zod.z.object({
 function createListTasksTool(taskRegistry) {
   return {
     id: "list_tasks",
-    displayName: "List Tasks",
     description: "List all background tasks with optional filtering by status or type. Returns task information and counts.",
     inputSchema: ListTasksInputSchema,
+    presentation: {
+      describeHeader: (input) => {
+        const bits = [];
+        if (input.status && input.status !== "all") bits.push(`status=${input.status}`);
+        if (input.type) bits.push(`type=${input.type}`);
+        const argsText = bits.length > 0 ? (0, import_core.truncateForHeader)(bits.join(", "), 120) : void 0;
+        return (0, import_core.createLocalToolCallHeader)({
+          title: "List Tasks",
+          ...argsText ? { argsText } : {}
+        });
+      }
+    },
     execute: async (input, _context) => {
       const filter = {};
       if (input.status && input.status !== "all") {