@vybestack/llxprt-code-tools 0.10.0-nightly.260613.1adad3b34

package/dist/src/tools/check-async-tasks.js

@@ -0,0 +1,241 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * @plan PLAN-20260130-ASYNCTASK.P12
+ * @plan PLAN-20260130-ASYNCTASK.P14
+ * @requirement REQ-ASYNC-005
+ */
+import { BaseDeclarativeTool, BaseToolInvocation, Kind, } from './tools.js';
+import { ToolErrorType } from '../types/tool-error.js';
+function resolveAsyncTaskService(dependenciesOrService) {
+    if ('getTaskStatus' in dependenciesOrService) {
+        return dependenciesOrService;
+    }
+    return dependenciesOrService.getAsyncTaskService?.();
+}
+function getTaskStartTime(task) {
+    return task.launchedAt ?? Date.now();
+}
+function getSubagentName(task) {
+    return task.subagentName ?? task.name ?? task.id;
+}
+function getGoalPrompt(task) {
+    return task.goalPrompt ?? '';
+}
+class CheckAsyncTasksInvocation extends BaseToolInvocation {
+    taskService;
+    constructor(params, taskService, messageBus) {
+        super(params, messageBus);
+        this.taskService = taskService;
+    }
+    getDescription() {
+        if (this.params.task_id) {
+            return `Check status of async task '${this.params.task_id}'`;
+        }
+        return 'List all async tasks';
+    }
+    async execute() {
+        if (this.params.task_id) {
+            return this.executePeek(this.params.task_id);
+        }
+        return this.executeList();
+    }
+    executeList() {
+        const tasks = this.taskService.getTaskStatus();
+        if (tasks.length === 0) {
+            return {
+                llmContent: 'No async tasks.',
+                returnDisplay: 'No async tasks are currently running or completed.',
+                metadata: { count: 0 },
+            };
+        }
+        const running = tasks.filter((t) => t.status === 'running');
+        const completed = tasks.filter((t) => t.status === 'completed');
+        const failed = tasks.filter((t) => t.status === 'failed');
+        const cancelled = tasks.filter((t) => t.status === 'cancelled');
+        const lines = [];
+        lines.push('Async Tasks Summary:');
+        lines.push(`- Running: ${running.length}`);
+        lines.push(`- Completed: ${completed.length}`);
+        lines.push(`- Failed: ${failed.length}`);
+        lines.push(`- Cancelled: ${cancelled.length}`);
+        lines.push('');
+        lines.push('Details:');
+        for (const task of tasks) {
+            const statusIcon = this.statusIcon(task.status);
+            const duration = this.formatDuration(task.launchedAt, task.completedAt);
+            lines.push(`${statusIcon} [${task.id}] - ${task.status} (${duration})`);
+        }
+        const llmContent = lines.join('\n');
+        const displayLines = tasks.map((t) => {
+            const icon = this.statusIcon(t.status);
+            return `${icon} **${t.id}** - ${t.status}`;
+        });
+        return {
+            llmContent,
+            returnDisplay: displayLines.join('\n'),
+            metadata: {
+                count: tasks.length,
+                running: running.length,
+                completed: completed.length,
+                failed: failed.length,
+                cancelled: cancelled.length,
+            },
+        };
+    }
+    executePeek(taskId) {
+        const task = this.taskService.getTask?.(taskId);
+        if (task) {
+            return this.formatTaskDetails(task);
+        }
+        const { task: prefixTask, candidates } = this.taskService.getTaskByPrefix?.(taskId) ?? {
+            task: undefined,
+            candidates: [],
+        };
+        if (prefixTask) {
+            return this.formatTaskDetails(prefixTask);
+        }
+        if (candidates && candidates.length > 0) {
+            const candidateList = candidates.map((c) => `- ${c.id}`).join('\n');
+            return {
+                llmContent: `Ambiguous task ID prefix '${taskId}'. Candidates:\n${candidateList}`,
+                returnDisplay: `Ambiguous prefix. Did you mean:\n${candidateList}`,
+                error: {
+                    message: 'Ambiguous task ID',
+                    type: ToolErrorType.INVALID_TOOL_PARAMS,
+                },
+            };
+        }
+        return {
+            llmContent: `No async task found with ID or prefix '${taskId}'.`,
+            returnDisplay: `Task not found: ${taskId}`,
+            error: {
+                message: 'Task not found',
+                type: ToolErrorType.INVALID_TOOL_PARAMS,
+            },
+        };
+    }
+    formatTaskDetails(task) {
+        const details = {
+            id: task.id,
+            subagentName: getSubagentName(task),
+            goalPrompt: getGoalPrompt(task),
+            status: task.status,
+            launchedAt: new Date(getTaskStartTime(task)).toISOString(),
+            duration: this.formatDuration(task.launchedAt, task.completedAt),
+        };
+        if (task.completedAt !== undefined &&
+            task.completedAt !== 0 &&
+            !Number.isNaN(task.completedAt)) {
+            details.completedAt = new Date(task.completedAt).toISOString();
+        }
+        const output = task.output;
+        if (output !== undefined && output !== '') {
+            details.output = task.output;
+        }
+        if (task.error !== undefined && task.error !== '') {
+            details.error = task.error;
+        }
+        return {
+            llmContent: JSON.stringify(details, null, 2),
+            returnDisplay: this.formatTaskDisplay(task),
+            metadata: details,
+        };
+    }
+    formatDuration(startTime, endTime) {
+        const end = endTime ?? Date.now();
+        const durationMs = end - (startTime ?? end);
+        const seconds = Math.floor(durationMs / 1000);
+        if (seconds < 60) {
+            return `${seconds}s`;
+        }
+        const minutes = Math.floor(seconds / 60);
+        const remainingSeconds = seconds % 60;
+        if (minutes < 60) {
+            return `${minutes}m ${remainingSeconds}s`;
+        }
+        const hours = Math.floor(minutes / 60);
+        const remainingMinutes = minutes % 60;
+        return `${hours}h ${remainingMinutes}m`;
+    }
+    formatTaskDisplay(task) {
+        const lines = [];
+        const icon = this.statusIcon(task.status);
+        const goalPrompt = getGoalPrompt(task);
+        lines.push(`${icon} **${getSubagentName(task)}**`);
+        lines.push(`ID: \`${task.id}\``);
+        lines.push(`Status: ${task.status}`);
+        lines.push(`Goal: ${goalPrompt.substring(0, 100)}${goalPrompt.length > 100 ? '...' : ''}`);
+        lines.push(`Duration: ${this.formatDuration(task.launchedAt, task.completedAt)}`);
+        const output = task.output;
+        if (output?.emitted_vars && Object.keys(output.emitted_vars).length > 0) {
+            lines.push('Emitted variables:');
+            for (const [key, value] of Object.entries(output.emitted_vars)) {
+                const valueStr = String(value);
+                const truncated = valueStr.substring(0, 50);
+                const suffix = valueStr.length > 50 ? '...' : '';
+                lines.push(`  - ${key}: ${truncated}${suffix}`);
+            }
+        }
+        if (task.error) {
+            lines.push(`Error: ${task.error}`);
+        }
+        return lines.join('\n');
+    }
+    statusIcon(status) {
+        if (status === 'completed') {
+            return '[OK]';
+        }
+        if (status === 'failed') {
+            return '[FAILED]';
+        }
+        return '';
+    }
+}
+/**
+ * Tool that allows querying the status of async tasks.
+ * Two modes:
+ * - List mode (no task_id): Shows summary of all tasks
+ * - Peek mode (with task_id): Shows detailed info for specific task
+ */
+export class CheckAsyncTasksTool extends BaseDeclarativeTool {
+    dependencies;
+    static Name = 'check_async_tasks';
+    constructor(dependencies) {
+        super(CheckAsyncTasksTool.Name, 'Check Async Tasks', 'Check the status of background async tasks. Call with no arguments to list all tasks, ' +
+            'or provide a task_id (or prefix) to get detailed info about a specific task.', Kind.Think, {
+            type: 'object',
+            additionalProperties: false,
+            properties: {
+                task_id: {
+                    type: 'string',
+                    description: 'Optional task ID or unique prefix to get details for a specific task.',
+                },
+            },
+        });
+        this.dependencies = dependencies;
+    }
+    createInvocation(params, messageBus) {
+        const service = resolveAsyncTaskService(this.dependencies);
+        if (!service) {
+            throw new Error('AsyncTaskManager service is unavailable. Please configure async tasks before invoking this tool.');
+        }
+        return new CheckAsyncTasksInvocation(params, service, messageBus);
+    }
+    async execute(params) {
+        const service = resolveAsyncTaskService(this.dependencies);
+        if (!service) {
+            throw new Error('AsyncTaskManager service is unavailable. Please configure async tasks before invoking this tool.');
+        }
+        return new CheckAsyncTasksInvocation(params, service, {
+            requestConfirmation: async () => undefined,
+        }).execute();
+    }
+    validateToolParamValues() {
+        return null;
+    }
+}
+//# sourceMappingURL=check-async-tasks.js.map

package/dist/src/tools/check-async-tasks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"check-async-tasks.js","sourceRoot":"","sources":["../../../src/tools/check-async-tasks.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;GAIG;AAEH,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,IAAI,GAEL,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAUvD,SAAS,uBAAuB,CAC9B,qBAA0E;IAE1E,IAAI,eAAe,IAAI,qBAAqB,EAAE,CAAC;QAC7C,OAAO,qBAAqB,CAAC;IAC/B,CAAC;IACD,OAAO,qBAAqB,CAAC,mBAAmB,EAAE,EAAE,CAAC;AACvD,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAmB;IAC3C,OAAO,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;AACvC,CAAC;AAED,SAAS,eAAe,CAAC,IAAmB;IAC1C,OAAO,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;AACnD,CAAC;AAED,SAAS,aAAa,CAAC,IAAmB;IACxC,OAAO,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;AAC/B,CAAC;AAED,MAAM,yBAA0B,SAAQ,kBAGvC;IACkB,WAAW,CAAoB;IAEhD,YACE,MAA6B,EAC7B,WAA8B,EAC9B,UAA2B;QAE3B,KAAK,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAC1B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAEQ,cAAc;QACrB,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACxB,OAAO,+BAA+B,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,CAAC;QAC/D,CAAC;QACD,OAAO,sBAAsB,CAAC;IAChC,CAAC;IAEQ,KAAK,CAAC,OAAO;QACpB,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC;QAED,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC;IAEO,WAAW;QACjB,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC;QAE/C,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO;gBACL,UAAU,EAAE,iBAAiB;gBAC7B,aAAa,EAAE,oDAAoD;gBACnE,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE;aACvB,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;QAC5D,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC;QAC1D,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC;QAEhE,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACnC,KAAK,CAAC,IAAI,CAAC,cAAc,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC3C,KAAK,CAAC,IAAI,CAAC,gBAAgB,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/C,KAAK,CAAC,IAAI,CAAC,aAAa,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QACzC,KAAK,CAAC,IAAI,CAAC,gBAAgB,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/C,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAEvB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YACxE,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,KAAK,IAAI,CAAC,EAAE,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,GAAG,CAAC,CAAC;QAC1E,CAAC;QAED,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,YAAY,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACnC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YACvC,OAAO,GAAG,IAAI,MAAM,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,MAAM,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,UAAU;YACV,aAAa,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;YACtC,QAAQ,EAAE;gBACR,KAAK,EAAE,KAAK,CAAC,MAAM;gBACnB,OAAO,EAAE,OAAO,CAAC,MAAM;gBACvB,SAAS,EAAE,SAAS,CAAC,MAAM;gBAC3B,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,SAAS,EAAE,SAAS,CAAC,MAAM;aAC5B;SACF,CAAC;IACJ,CAAC;IAEO,WAAW,CAAC,MAAc;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;QAEhD,IAAI,IAAI,EAAE,CAAC;YACT,OAAO,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC;QAED,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,eAAe,EAAE,CACzE,MAAM,CACP,IAAI;YACH,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxC,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEpE,OAAO;gBACL,UAAU,EAAE,6BAA6B,MAAM,mBAAmB,aAAa,EAAE;gBACjF,aAAa,EAAE,oCAAoC,aAAa,EAAE;gBAClE,KAAK,EAAE;oBACL,OAAO,EAAE,mBAAmB;oBAC5B,IAAI,EAAE,aAAa,CAAC,mBAAmB;iBACxC;aACF,CAAC;QACJ,CAAC;QAED,OAAO;YACL,UAAU,EAAE,0CAA0C,MAAM,IAAI;YAChE,aAAa,EAAE,mBAAmB,MAAM,EAAE;YAC1C,KAAK,EAAE;gBACL,OAAO,EAAE,gBAAgB;gBACzB,IAAI,EAAE,aAAa,CAAC,mBAAmB;aACxC;SACF,CAAC;IACJ,CAAC;IAEO,iBAAiB,CAAC,IAAmB;QAC3C,MAAM,OAAO,GAA4B;YACvC,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,YAAY,EAAE,eAAe,CAAC,IAAI,CAAC;YACnC,UAAU,EAAE,aAAa,CAAC,IAAI,CAAC;YAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,UAAU,EAAE,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,WAAW,EAAE;YAC1D,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC;SACjE,CAAC;QAEF,IACE,IAAI,CAAC,WAAW,KAAK,SAAS;YAC9B,IAAI,CAAC,WAAW,KAAK,CAAC;YACtB,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,EAC/B,CAAC;YACD,OAAO,CAAC,WAAW,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;QACjE,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAiB,CAAC;QACtC,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC1C,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC/B,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,KAAK,EAAE,EAAE,CAAC;YAClD,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QAC7B,CAAC;QAED,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5C,aAAa,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC;YAC3C,QAAQ,EAAE,OAAO;SAClB,CAAC;IACJ,CAAC;IAEO,cAAc,CACpB,SAA6B,EAC7B,OAAgB;QAEhB,MAAM,GAAG,GAAG,OAAO,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QAClC,MAAM,UAAU,GAAG,GAAG,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC;QAE9C,IAAI,OAAO,GAAG,EAAE,EAAE,CAAC;YACjB,OAAO,GAAG,OAAO,GAAG,CAAC;QACvB,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC;QACzC,MAAM,gBAAgB,GAAG,OAAO,GAAG,EAAE,CAAC;QAEtC,IAAI,OAAO,GAAG,EAAE,EAAE,CAAC;YACjB,OAAO,GAAG,OAAO,KAAK,gBAAgB,GAAG,CAAC;QAC5C,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC;QACvC,MAAM,gBAAgB,GAAG,OAAO,GAAG,EAAE,CAAC;QACtC,OAAO,GAAG,KAAK,KAAK,gBAAgB,GAAG,CAAC;IAC1C,CAAC;IAEO,iBAAiB,CAAC,IAAmB;QAC3C,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;QAEvC,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,MAAM,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACrC,KAAK,CAAC,IAAI,CACR,SAAS,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,UAAU,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAC/E,CAAC;QACF,KAAK,CAAC,IAAI,CACR,aAAa,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,EAAE,CACtE,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,CAAC,MAAoD,CAAC;QACzE,IAAI,MAAM,EAAE,YAAY,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxE,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACjC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;gBAC/D,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC/B,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC5C,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBACjD,KAAK,CAAC,IAAI,CAAC,OAAO,GAAG,KAAK,SAAS,GAAG,MAAM,EAAE,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,KAAK,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAEO,UAAU,CAAC,MAA+B;QAChD,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;YAC3B,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACxB,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;CACF;AAED;;;;;GAKG;AACH,MAAM,OAAO,mBAAoB,SAAQ,mBAGxC;IAIoB;IAHnB,MAAM,CAAU,IAAI,GAAG,mBAAmB,CAAC;IAE3C,YACmB,YAEI;QAErB,KAAK,CACH,mBAAmB,CAAC,IAAI,EACxB,mBAAmB,EACnB,wFAAwF;YACtF,8EAA8E,EAChF,IAAI,CAAC,KAAK,EACV;YACE,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE;gBACV,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EACT,uEAAuE;iBAC1E;aACF;SACF,CACF,CAAC;QArBe,iBAAY,GAAZ,YAAY,CAER;IAoBvB,CAAC;IAEkB,gBAAgB,CACjC,MAA6B,EAC7B,UAA2B;QAE3B,MAAM,OAAO,GAAG,uBAAuB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAE3D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CACb,kGAAkG,CACnG,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,yBAAyB,CAAC,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAA6B;QACzC,MAAM,OAAO,GAAG,uBAAuB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC3D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CACb,kGAAkG,CACnG,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,yBAAyB,CAAC,MAAM,EAAE,OAAO,EAAE;YACpD,mBAAmB,EAAE,KAAK,IAAI,EAAE,CAAC,SAAS;SAC3C,CAAC,CAAC,OAAO,EAAE,CAAC;IACf,CAAC;IAEkB,uBAAuB;QACxC,OAAO,IAAI,CAAC;IACd,CAAC"}

package/dist/src/tools/codesearch.d.ts

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Portions of this code are derived from opencode (https://github.com/sst/opencode)
+ * Copyright (c) 2025 opencode
+ * Licensed under the MIT License.
+ */
+import type { ISettingsService, IToolKeyStorage } from '../interfaces/index.js';
+import { BaseDeclarativeTool, type ToolInvocation, type ToolResult } from './tools.js';
+export interface CodeSearchToolParams {
+    query: string;
+    tokensNum?: number;
+}
+export interface CodeSearchToolDependencies {
+    keyStorage?: Pick<IToolKeyStorage, 'resolveKey'>;
+    settingsService?: Pick<ISettingsService, 'getSetting' | 'getSettingsService'>;
+}
+export declare class CodeSearchTool extends BaseDeclarativeTool<CodeSearchToolParams, ToolResult> {
+    private readonly dependencies;
+    static readonly Name = "codesearch";
+    constructor(dependencies?: CodeSearchToolDependencies);
+    protected createInvocation(params: CodeSearchToolParams, messageBus?: import('../interfaces/index.js').IToolMessageBus): ToolInvocation<CodeSearchToolParams, ToolResult>;
+}

package/dist/src/tools/codesearch.js

@@ -0,0 +1,153 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Portions of this code are derived from opencode (https://github.com/sst/opencode)
+ * Copyright (c) 2025 opencode
+ * Licensed under the MIT License.
+ */
+import fetch from 'node-fetch';
+import { ToolErrorType } from '../types/tool-error.js';
+import { ensureJsonSafe } from '../utils/unicodeUtils.js';
+import { BaseDeclarativeTool, BaseToolInvocation, Kind, } from './tools.js';
+const API_CONFIG = {
+    BASE_URL: 'https://mcp.exa.ai',
+    ENDPOINTS: {
+        CONTEXT: '/mcp',
+    },
+};
+const TOKEN_LIMITS = {
+    DEFAULT: 5000,
+    MIN: 1000,
+    MAX: 50000,
+};
+export class CodeSearchTool extends BaseDeclarativeTool {
+    dependencies;
+    static Name = 'codesearch';
+    constructor(dependencies = {}) {
+        super(CodeSearchTool.Name, 'CodeSearch', 'Search for relevant code snippets, APIs, Libraries, and SDKs documentation. Use this to find examples and usage patterns.', Kind.Search, {
+            type: 'object',
+            properties: {
+                query: {
+                    type: 'string',
+                    description: "Search query to find relevant context. For example, 'React useState hook examples', 'Python pandas dataframe filtering'.",
+                },
+                tokensNum: {
+                    type: 'number',
+                    description: 'Number of tokens to return (1000-50000). Default is 5000 tokens.',
+                    default: TOKEN_LIMITS.DEFAULT,
+                    minimum: TOKEN_LIMITS.MIN,
+                    maximum: TOKEN_LIMITS.MAX,
+                },
+            },
+            required: ['query'],
+        });
+        this.dependencies = dependencies;
+    }
+    createInvocation(params, messageBus) {
+        return new CodeSearchToolInvocation(this.dependencies, params, messageBus);
+    }
+}
+class CodeSearchToolInvocation extends BaseToolInvocation {
+    dependencies;
+    constructor(dependencies, params, messageBus) {
+        super(params, messageBus);
+        this.dependencies = dependencies;
+    }
+    getDescription() {
+        return `Search code for: ${this.params.query}`;
+    }
+    async buildEndpointUrl() {
+        const baseUrl = `${API_CONFIG.BASE_URL}${API_CONFIG.ENDPOINTS.CONTEXT}`;
+        const key = await this.dependencies.keyStorage?.resolveKey('exa');
+        if (key !== undefined && key !== null) {
+            return `${baseUrl}?exaApiKey=${encodeURIComponent(key)}`;
+        }
+        return baseUrl;
+    }
+    async execute(signal, _updateOutput) {
+        const codeRequest = {
+            jsonrpc: '2.0',
+            id: 1,
+            method: 'tools/call',
+            params: {
+                name: 'get_code_context_exa',
+                arguments: {
+                    query: this.params.query,
+                    tokensNum: this.getEffectiveTokensNum(),
+                },
+            },
+        };
+        try {
+            const headers = {
+                accept: 'application/json, text/event-stream',
+                'content-type': 'application/json',
+            };
+            const endpointUrl = await this.buildEndpointUrl();
+            const response = await fetch(endpointUrl, {
+                method: 'POST',
+                headers,
+                body: JSON.stringify(codeRequest),
+                signal,
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                throw new Error(`Code search error (${response.status}): ${errorText}`);
+            }
+            const responseText = await response.text();
+            const lines = responseText.split('\n');
+            for (const line of lines) {
+                if (line.startsWith('data: ')) {
+                    try {
+                        const data = JSON.parse(line.substring(6));
+                        if (data.result?.content !== undefined &&
+                            data.result.content.length > 0) {
+                            const content = ensureJsonSafe(data.result.content[0].text);
+                            return {
+                                llmContent: content,
+                                returnDisplay: content,
+                            };
+                        }
+                    }
+                    catch {
+                        // Ignore parse errors for intermediate lines
+                    }
+                }
+            }
+            return {
+                llmContent: 'No code snippets or documentation found. Please try a different query.',
+                returnDisplay: 'No results found.',
+            };
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            return {
+                llmContent: `Error performing code search: ${errorMessage}`,
+                returnDisplay: `Error: ${errorMessage}`,
+                error: {
+                    message: errorMessage,
+                    type: ToolErrorType.SEARCH_ERROR,
+                },
+            };
+        }
+    }
+    getEffectiveTokensNum() {
+        const settingMaxTokens = this.getSettingMaxTokens();
+        let tokens = this.params.tokensNum ?? TOKEN_LIMITS.DEFAULT;
+        if (settingMaxTokens !== undefined) {
+            tokens = Math.min(tokens, settingMaxTokens);
+        }
+        return Math.max(TOKEN_LIMITS.MIN, Math.min(tokens, TOKEN_LIMITS.MAX));
+    }
+    getSettingMaxTokens() {
+        const directValue = this.dependencies.settingsService?.getSetting('tool-output-max-tokens');
+        if (typeof directValue === 'number')
+            return directValue;
+        const nestedValue = this.dependencies.settingsService
+            ?.getSettingsService()
+            .get?.('tool-output-max-tokens');
+        return typeof nestedValue === 'number' ? nestedValue : undefined;
+    }
+}
+//# sourceMappingURL=codesearch.js.map

package/dist/src/tools/codesearch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codesearch.js","sourceRoot":"","sources":["../../../src/tools/codesearch.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,MAAM,YAAY,CAAC;AAG/B,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,IAAI,GAGL,MAAM,YAAY,CAAC;AAEpB,MAAM,UAAU,GAAG;IACjB,QAAQ,EAAE,oBAAoB;IAC9B,SAAS,EAAE;QACT,OAAO,EAAE,MAAM;KAChB;CACO,CAAC;AAEX,MAAM,YAAY,GAAG;IACnB,OAAO,EAAE,IAAI;IACb,GAAG,EAAE,IAAI;IACT,GAAG,EAAE,KAAK;CACF,CAAC;AAmCX,MAAM,OAAO,cAAe,SAAQ,mBAGnC;IAG8B;IAF7B,MAAM,CAAU,IAAI,GAAG,YAAY,CAAC;IAEpC,YAA6B,eAA2C,EAAE;QACxE,KAAK,CACH,cAAc,CAAC,IAAI,EACnB,YAAY,EACZ,2HAA2H,EAC3H,IAAI,CAAC,MAAM,EACX;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EACT,0HAA0H;iBAC7H;gBACD,SAAS,EAAE;oBACT,IAAI,EAAE,QAAQ;oBACd,WAAW,EACT,kEAAkE;oBACpE,OAAO,EAAE,YAAY,CAAC,OAAO;oBAC7B,OAAO,EAAE,YAAY,CAAC,GAAG;oBACzB,OAAO,EAAE,YAAY,CAAC,GAAG;iBAC1B;aACF;YACD,QAAQ,EAAE,CAAC,OAAO,CAAC;SACpB,CACF,CAAC;QAzByB,iBAAY,GAAZ,YAAY,CAAiC;IA0B1E,CAAC;IAES,gBAAgB,CACxB,MAA4B,EAC5B,UAA6D;QAE7D,OAAO,IAAI,wBAAwB,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;IAC7E,CAAC;;AAGH,MAAM,wBAAyB,SAAQ,kBAGtC;IAEoB;IADnB,YACmB,YAAwC,EACzD,MAA4B,EAC5B,UAA6D;QAE7D,KAAK,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAJT,iBAAY,GAAZ,YAAY,CAA4B;IAK3D,CAAC;IAED,cAAc;QACZ,OAAO,oBAAoB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,MAAM,OAAO,GAAG,GAAG,UAAU,CAAC,QAAQ,GAAG,UAAU,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QACxE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC;QAClE,IAAI,GAAG,KAAK,SAAS,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;YACtC,OAAO,GAAG,OAAO,cAAc,kBAAkB,CAAC,GAAG,CAAC,EAAE,CAAC;QAC3D,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,OAAO,CACX,MAAmB,EACnB,aAAwC;QAExC,MAAM,WAAW,GAAmB;YAClC,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,CAAC;YACL,MAAM,EAAE,YAAY;YACpB,MAAM,EAAE;gBACN,IAAI,EAAE,sBAAsB;gBAC5B,SAAS,EAAE;oBACT,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;oBACxB,SAAS,EAAE,IAAI,CAAC,qBAAqB,EAAE;iBACxC;aACF;SACF,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,OAAO,GAA2B;gBACtC,MAAM,EAAE,qCAAqC;gBAC7C,cAAc,EAAE,kBAAkB;aACnC,CAAC;YAEF,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAClD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,WAAW,EAAE;gBACxC,MAAM,EAAE,MAAM;gBACd,OAAO;gBACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;gBACjC,MAAM;aACP,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CAAC,sBAAsB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;YAC1E,CAAC;YAED,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC9B,IAAI,CAAC;wBACH,MAAM,IAAI,GAAoB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;wBAC5D,IACE,IAAI,CAAC,MAAM,EAAE,OAAO,KAAK,SAAS;4BAClC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAC9B,CAAC;4BACD,MAAM,OAAO,GAAG,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;4BAC5D,OAAO;gCACL,UAAU,EAAE,OAAO;gCACnB,aAAa,EAAE,OAAO;6BACvB,CAAC;wBACJ,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,6CAA6C;oBAC/C,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO;gBACL,UAAU,EACR,wEAAwE;gBAC1E,aAAa,EAAE,mBAAmB;aACnC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzD,OAAO;gBACL,UAAU,EAAE,iCAAiC,YAAY,EAAE;gBAC3D,aAAa,EAAE,UAAU,YAAY,EAAE;gBACvC,KAAK,EAAE;oBACL,OAAO,EAAE,YAAY;oBACrB,IAAI,EAAE,aAAa,CAAC,YAAY;iBACjC;aACF,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,qBAAqB;QAC3B,MAAM,gBAAgB,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACpD,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,YAAY,CAAC,OAAO,CAAC;QAE3D,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACnC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;QAC9C,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC;IACxE,CAAC;IAEO,mBAAmB;QACzB,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,UAAU,CAC/D,wBAAwB,CACzB,CAAC;QACF,IAAI,OAAO,WAAW,KAAK,QAAQ;YAAE,OAAO,WAAW,CAAC;QAExD,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe;YACnD,EAAE,kBAAkB,EAAE;aACrB,GAAG,EAAE,CAAC,wBAAwB,CAAC,CAAC;QACnC,OAAO,OAAO,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;IACnE,CAAC;CACF"}

package/dist/src/tools/delete_line_range.d.ts

@@ -0,0 +1,41 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseDeclarativeTool, type ToolInvocation, type ToolResult } from './tools.js';
+import type { IIdeService, ILspService, IToolHost, IToolMessageBus } from '../interfaces/index.js';
+/**
+ * Parameters for the DeleteLineRange tool
+ */
+export interface DeleteLineRangeToolParams {
+    /**
+     * The absolute path to the file to modify
+     */
+    absolute_path?: string;
+    /**
+     * Alternative parameter name for absolute_path (for backward compatibility)
+     */
+    file_path?: string;
+    /**
+     * The 1-based line number to start deleting from (inclusive)
+     */
+    start_line: number;
+    /**
+     * The 1-based line number to end deleting at (inclusive)
+     */
+    end_line: number;
+}
+/**
+ * Implementation of the DeleteLineRange tool logic
+ */
+export declare class DeleteLineRangeTool extends BaseDeclarativeTool<DeleteLineRangeToolParams, ToolResult> {
+    private readonly host;
+    private readonly ideService?;
+    private readonly lspService?;
+    static readonly Name: string;
+    constructor(host: IToolHost, ideService?: IIdeService | undefined, lspService?: ILspService | undefined);
+    protected validateToolParamValues(params: DeleteLineRangeToolParams): string | null;
+    protected createInvocation(params: DeleteLineRangeToolParams, messageBus: IToolMessageBus): ToolInvocation<DeleteLineRangeToolParams, ToolResult>;
+    execute(params: DeleteLineRangeToolParams, signal?: AbortSignal): Promise<ToolResult>;
+}