@within-7/minto 0.2.0 → 0.3.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/commands/agents/AgentsCommand.js +22 -24
- package/dist/commands/agents/AgentsCommand.js.map +2 -2
- package/dist/commands/context.js +2 -1
- package/dist/commands/context.js.map +2 -2
- package/dist/commands/export.js +2 -1
- package/dist/commands/export.js.map +2 -2
- package/dist/commands/mcp-interactive.js +7 -6
- package/dist/commands/mcp-interactive.js.map +2 -2
- package/dist/commands/model.js +3 -2
- package/dist/commands/model.js.map +2 -2
- package/dist/commands/permissions.js +4 -3
- package/dist/commands/permissions.js.map +2 -2
- package/dist/commands/plugin/AddMarketplaceForm.js +3 -2
- package/dist/commands/plugin/AddMarketplaceForm.js.map +2 -2
- package/dist/commands/plugin/ConfirmDialog.js +2 -1
- package/dist/commands/plugin/ConfirmDialog.js.map +2 -2
- package/dist/commands/plugin/ErrorView.js +2 -1
- package/dist/commands/plugin/ErrorView.js.map +2 -2
- package/dist/commands/plugin/InstalledPluginsByMarketplace.js +5 -4
- package/dist/commands/plugin/InstalledPluginsByMarketplace.js.map +2 -2
- package/dist/commands/plugin/InstalledPluginsManager.js +5 -4
- package/dist/commands/plugin/InstalledPluginsManager.js.map +2 -2
- package/dist/commands/plugin/MainMenu.js +2 -1
- package/dist/commands/plugin/MainMenu.js.map +2 -2
- package/dist/commands/plugin/MarketplaceManager.js +5 -4
- package/dist/commands/plugin/MarketplaceManager.js.map +2 -2
- package/dist/commands/plugin/MarketplaceSelector.js +4 -3
- package/dist/commands/plugin/MarketplaceSelector.js.map +2 -2
- package/dist/commands/plugin/PlaceholderScreen.js +3 -2
- package/dist/commands/plugin/PlaceholderScreen.js.map +2 -2
- package/dist/commands/plugin/PluginBrowser.js +6 -5
- package/dist/commands/plugin/PluginBrowser.js.map +2 -2
- package/dist/commands/plugin/PluginDetailsInstall.js +5 -4
- package/dist/commands/plugin/PluginDetailsInstall.js.map +2 -2
- package/dist/commands/plugin/PluginDetailsManage.js +4 -3
- package/dist/commands/plugin/PluginDetailsManage.js.map +2 -2
- package/dist/commands/plugin.js +16 -15
- package/dist/commands/plugin.js.map +2 -2
- package/dist/commands/sandbox.js +4 -3
- package/dist/commands/sandbox.js.map +2 -2
- package/dist/commands/setup.js +2 -1
- package/dist/commands/setup.js.map +2 -2
- package/dist/commands/status.js +2 -1
- package/dist/commands/status.js.map +2 -2
- package/dist/commands/undo.js +245 -0
- package/dist/commands/undo.js.map +7 -0
- package/dist/commands.js +2 -0
- package/dist/commands.js.map +2 -2
- package/dist/components/AgentThinkingBlock.js +1 -1
- package/dist/components/AgentThinkingBlock.js.map +2 -2
- package/dist/components/AsciiLogo.js +7 -8
- package/dist/components/AsciiLogo.js.map +2 -2
- package/dist/components/AskUserQuestionDialog/AskUserQuestionDialog.js +3 -2
- package/dist/components/AskUserQuestionDialog/AskUserQuestionDialog.js.map +2 -2
- package/dist/components/AskUserQuestionDialog/QuestionView.js +2 -1
- package/dist/components/AskUserQuestionDialog/QuestionView.js.map +2 -2
- package/dist/components/CollapsibleHint.js +2 -1
- package/dist/components/CollapsibleHint.js.map +2 -2
- package/dist/components/Config.js +3 -2
- package/dist/components/Config.js.map +2 -2
- package/dist/components/ConsoleOAuthFlow.js +2 -1
- package/dist/components/ConsoleOAuthFlow.js.map +2 -2
- package/dist/components/Cost.js +2 -1
- package/dist/components/Cost.js.map +2 -2
- package/dist/components/HeaderBar.js +13 -8
- package/dist/components/HeaderBar.js.map +2 -2
- package/dist/components/HistorySearchOverlay.js +4 -3
- package/dist/components/HistorySearchOverlay.js.map +2 -2
- package/dist/components/HotkeyHelpPanel.js +8 -11
- package/dist/components/HotkeyHelpPanel.js.map +2 -2
- package/dist/components/InvalidConfigDialog.js +2 -1
- package/dist/components/InvalidConfigDialog.js.map +2 -2
- package/dist/components/Logo.js +23 -67
- package/dist/components/Logo.js.map +2 -2
- package/dist/components/MCPServerApprovalDialog.js +2 -1
- package/dist/components/MCPServerApprovalDialog.js.map +2 -2
- package/dist/components/MCPServerDialogCopy.js +2 -1
- package/dist/components/MCPServerDialogCopy.js.map +2 -2
- package/dist/components/MCPServerMultiselectDialog.js +2 -1
- package/dist/components/MCPServerMultiselectDialog.js.map +2 -2
- package/dist/components/MessageSelector.js +4 -3
- package/dist/components/MessageSelector.js.map +2 -2
- package/dist/components/ModeIndicator.js +2 -1
- package/dist/components/ModeIndicator.js.map +2 -2
- package/dist/components/ModelConfig.js +4 -3
- package/dist/components/ModelConfig.js.map +2 -2
- package/dist/components/ModelListManager.js +4 -3
- package/dist/components/ModelListManager.js.map +2 -2
- package/dist/components/ModelSelector/ModelSelector.js +26 -13
- package/dist/components/ModelSelector/ModelSelector.js.map +2 -2
- package/dist/components/Onboarding.js +3 -2
- package/dist/components/Onboarding.js.map +2 -2
- package/dist/components/OperationSummary.js +130 -0
- package/dist/components/OperationSummary.js.map +7 -0
- package/dist/components/PromptInput.js +88 -75
- package/dist/components/PromptInput.js.map +2 -2
- package/dist/components/SensitiveFileWarning.js +31 -0
- package/dist/components/SensitiveFileWarning.js.map +7 -0
- package/dist/components/Spinner.js +71 -22
- package/dist/components/Spinner.js.map +2 -2
- package/dist/components/StructuredDiff.js +6 -8
- package/dist/components/StructuredDiff.js.map +2 -2
- package/dist/components/SubagentBlock.js +4 -2
- package/dist/components/SubagentBlock.js.map +2 -2
- package/dist/components/SubagentProgress.js +17 -6
- package/dist/components/SubagentProgress.js.map +2 -2
- package/dist/components/TaskCard.js +14 -11
- package/dist/components/TaskCard.js.map +2 -2
- package/dist/components/TextInput.js +9 -1
- package/dist/components/TextInput.js.map +2 -2
- package/dist/components/TodoPanel.js +44 -26
- package/dist/components/TodoPanel.js.map +2 -2
- package/dist/components/ToolUseLoader.js +2 -2
- package/dist/components/ToolUseLoader.js.map +2 -2
- package/dist/components/TreeConnector.js +4 -3
- package/dist/components/TreeConnector.js.map +2 -2
- package/dist/components/TrustDialog.js +2 -1
- package/dist/components/TrustDialog.js.map +2 -2
- package/dist/components/binary-feedback/BinaryFeedbackView.js +2 -1
- package/dist/components/binary-feedback/BinaryFeedbackView.js.map +2 -2
- package/dist/components/messages/AssistantTextMessage.js +17 -9
- package/dist/components/messages/AssistantTextMessage.js.map +2 -2
- package/dist/components/messages/AssistantToolUseMessage.js +8 -4
- package/dist/components/messages/AssistantToolUseMessage.js.map +2 -2
- package/dist/components/messages/GroupRenderer.js +2 -1
- package/dist/components/messages/GroupRenderer.js.map +2 -2
- package/dist/components/messages/NestedTasksPreview.js +13 -1
- package/dist/components/messages/NestedTasksPreview.js.map +2 -2
- package/dist/components/messages/ParallelTasksGroupView.js +4 -3
- package/dist/components/messages/ParallelTasksGroupView.js.map +2 -2
- package/dist/components/messages/TaskInModuleView.js +35 -15
- package/dist/components/messages/TaskInModuleView.js.map +2 -2
- package/dist/components/messages/TaskOutputContent.js +9 -6
- package/dist/components/messages/TaskOutputContent.js.map +2 -2
- package/dist/components/messages/UserPromptMessage.js +2 -2
- package/dist/components/messages/UserPromptMessage.js.map +2 -2
- package/dist/constants/colors.js +90 -72
- package/dist/constants/colors.js.map +2 -2
- package/dist/constants/prompts.js +22 -1
- package/dist/constants/prompts.js.map +2 -2
- package/dist/constants/toolInputExamples.js +84 -0
- package/dist/constants/toolInputExamples.js.map +7 -0
- package/dist/core/backupManager.js +321 -0
- package/dist/core/backupManager.js.map +7 -0
- package/dist/core/costTracker.js +9 -18
- package/dist/core/costTracker.js.map +2 -2
- package/dist/core/gitAutoCommit.js +287 -0
- package/dist/core/gitAutoCommit.js.map +7 -0
- package/dist/core/index.js +3 -0
- package/dist/core/index.js.map +2 -2
- package/dist/core/operationTracker.js +212 -0
- package/dist/core/operationTracker.js.map +7 -0
- package/dist/core/permissions/rules/allowedToolsRule.js +1 -1
- package/dist/core/permissions/rules/allowedToolsRule.js.map +2 -2
- package/dist/core/permissions/rules/autoEscalationRule.js +5 -0
- package/dist/core/permissions/rules/autoEscalationRule.js.map +2 -2
- package/dist/core/permissions/rules/projectBoundaryRule.js +5 -0
- package/dist/core/permissions/rules/projectBoundaryRule.js.map +2 -2
- package/dist/core/permissions/rules/sensitivePathsRule.js +5 -0
- package/dist/core/permissions/rules/sensitivePathsRule.js.map +2 -2
- package/dist/core/tokenStats.js +9 -0
- package/dist/core/tokenStats.js.map +7 -0
- package/dist/core/tokenStatsManager.js +331 -0
- package/dist/core/tokenStatsManager.js.map +7 -0
- package/dist/entrypoints/cli.js +122 -88
- package/dist/entrypoints/cli.js.map +2 -2
- package/dist/hooks/useAgentTokenStats.js +72 -0
- package/dist/hooks/useAgentTokenStats.js.map +7 -0
- package/dist/hooks/useAgentTranscripts.js +30 -6
- package/dist/hooks/useAgentTranscripts.js.map +2 -2
- package/dist/hooks/useLogMessages.js +12 -1
- package/dist/hooks/useLogMessages.js.map +2 -2
- package/dist/i18n/locales/en.js +6 -5
- package/dist/i18n/locales/en.js.map +2 -2
- package/dist/i18n/locales/zh-CN.js +6 -5
- package/dist/i18n/locales/zh-CN.js.map +2 -2
- package/dist/i18n/types.js.map +1 -1
- package/dist/permissions.js +147 -1
- package/dist/permissions.js.map +2 -2
- package/dist/query.js +78 -4
- package/dist/query.js.map +3 -3
- package/dist/screens/REPL.js +23 -3
- package/dist/screens/REPL.js.map +2 -2
- package/dist/screens/ResumeConversation.js +2 -0
- package/dist/screens/ResumeConversation.js.map +2 -2
- package/dist/services/claude.js +54 -3
- package/dist/services/claude.js.map +2 -2
- package/dist/services/intelligentCompactor.js +1 -1
- package/dist/services/intelligentCompactor.js.map +2 -2
- package/dist/services/mcpClient.js +81 -25
- package/dist/services/mcpClient.js.map +2 -2
- package/dist/services/sandbox/filesystemBoundary.js +58 -17
- package/dist/services/sandbox/filesystemBoundary.js.map +2 -2
- package/dist/services/taskStore.js +205 -0
- package/dist/services/taskStore.js.map +7 -0
- package/dist/tools/AskExpertModelTool/AskExpertModelTool.js +3 -2
- package/dist/tools/AskExpertModelTool/AskExpertModelTool.js.map +2 -2
- package/dist/tools/AskUserQuestionTool/AskUserQuestionTool.js +42 -4
- package/dist/tools/AskUserQuestionTool/AskUserQuestionTool.js.map +2 -2
- package/dist/tools/BashTool/BashTool.js +43 -7
- package/dist/tools/BashTool/BashTool.js.map +2 -2
- package/dist/tools/BashTool/prompt.js +184 -34
- package/dist/tools/BashTool/prompt.js.map +2 -2
- package/dist/tools/FileEditTool/FileEditTool.js +24 -9
- package/dist/tools/FileEditTool/FileEditTool.js.map +2 -2
- package/dist/tools/FileEditTool/prompt.js +10 -4
- package/dist/tools/FileEditTool/prompt.js.map +2 -2
- package/dist/tools/FileEditTool/utils.js +10 -4
- package/dist/tools/FileEditTool/utils.js.map +2 -2
- package/dist/tools/FileReadTool/FileReadTool.js +1 -1
- package/dist/tools/FileReadTool/FileReadTool.js.map +1 -1
- package/dist/tools/FileReadTool/prompt.js +16 -1
- package/dist/tools/FileReadTool/prompt.js.map +2 -2
- package/dist/tools/FileWriteTool/FileWriteTool.js +1 -1
- package/dist/tools/FileWriteTool/FileWriteTool.js.map +1 -1
- package/dist/tools/FileWriteTool/prompt.js +12 -3
- package/dist/tools/FileWriteTool/prompt.js.map +2 -2
- package/dist/tools/GlobTool/prompt.js +12 -1
- package/dist/tools/GlobTool/prompt.js.map +2 -2
- package/dist/tools/GrepTool/GrepTool.js +333 -65
- package/dist/tools/GrepTool/GrepTool.js.map +2 -2
- package/dist/tools/GrepTool/prompt.js +15 -8
- package/dist/tools/GrepTool/prompt.js.map +2 -2
- package/dist/tools/MultiEditTool/prompt.js +5 -3
- package/dist/tools/MultiEditTool/prompt.js.map +2 -2
- package/dist/tools/NotebookEditTool/NotebookEditTool.js +59 -46
- package/dist/tools/NotebookEditTool/NotebookEditTool.js.map +2 -2
- package/dist/tools/NotebookEditTool/prompt.js +1 -1
- package/dist/tools/NotebookEditTool/prompt.js.map +1 -1
- package/dist/tools/PlanModeTool/EnterPlanModeTool.js +3 -2
- package/dist/tools/PlanModeTool/EnterPlanModeTool.js.map +2 -2
- package/dist/tools/PlanModeTool/ExitPlanModeTool.js +3 -2
- package/dist/tools/PlanModeTool/ExitPlanModeTool.js.map +2 -2
- package/dist/tools/PlanModeTool/prompt.js +1 -1
- package/dist/tools/PlanModeTool/prompt.js.map +1 -1
- package/dist/tools/SkillTool/SkillTool.js +4 -3
- package/dist/tools/SkillTool/SkillTool.js.map +2 -2
- package/dist/tools/SkillTool/prompt.js +1 -1
- package/dist/tools/SkillTool/prompt.js.map +1 -1
- package/dist/tools/TaskCreateTool/TaskCreateTool.js +102 -0
- package/dist/tools/TaskCreateTool/TaskCreateTool.js.map +7 -0
- package/dist/tools/TaskCreateTool/prompt.js +47 -0
- package/dist/tools/TaskCreateTool/prompt.js.map +7 -0
- package/dist/tools/TaskGetTool/TaskGetTool.js +115 -0
- package/dist/tools/TaskGetTool/TaskGetTool.js.map +7 -0
- package/dist/tools/TaskGetTool/prompt.js +28 -0
- package/dist/tools/TaskGetTool/prompt.js.map +7 -0
- package/dist/tools/TaskListTool/TaskListTool.js +102 -0
- package/dist/tools/TaskListTool/TaskListTool.js.map +7 -0
- package/dist/tools/TaskListTool/prompt.js +27 -0
- package/dist/tools/TaskListTool/prompt.js.map +7 -0
- package/dist/tools/TaskOutputTool/TaskOutputTool.js +3 -2
- package/dist/tools/TaskOutputTool/TaskOutputTool.js.map +2 -2
- package/dist/tools/TaskStopTool/TaskStopTool.js +150 -0
- package/dist/tools/TaskStopTool/TaskStopTool.js.map +7 -0
- package/dist/tools/TaskStopTool/prompt.js +15 -0
- package/dist/tools/TaskStopTool/prompt.js.map +7 -0
- package/dist/tools/TaskTool/TaskTool.js +49 -1
- package/dist/tools/TaskTool/TaskTool.js.map +2 -2
- package/dist/tools/TaskUpdateTool/TaskUpdateTool.js +134 -0
- package/dist/tools/TaskUpdateTool/TaskUpdateTool.js.map +7 -0
- package/dist/tools/TaskUpdateTool/prompt.js +81 -0
- package/dist/tools/TaskUpdateTool/prompt.js.map +7 -0
- package/dist/tools/URLFetcherTool/prompt.js +1 -1
- package/dist/tools/URLFetcherTool/prompt.js.map +1 -1
- package/dist/tools.js +12 -0
- package/dist/tools.js.map +2 -2
- package/dist/utils/CircuitBreaker.js +242 -0
- package/dist/utils/CircuitBreaker.js.map +7 -0
- package/dist/utils/ask.js +2 -0
- package/dist/utils/ask.js.map +2 -2
- package/dist/utils/config.js +47 -5
- package/dist/utils/config.js.map +2 -2
- package/dist/utils/credentials/CredentialStore.js +1 -0
- package/dist/utils/credentials/CredentialStore.js.map +7 -0
- package/dist/utils/credentials/EncryptedFileStore.js +157 -0
- package/dist/utils/credentials/EncryptedFileStore.js.map +7 -0
- package/dist/utils/credentials/index.js +37 -0
- package/dist/utils/credentials/index.js.map +7 -0
- package/dist/utils/credentials/migration.js +82 -0
- package/dist/utils/credentials/migration.js.map +7 -0
- package/dist/utils/markdown.js +13 -1
- package/dist/utils/markdown.js.map +2 -2
- package/dist/utils/model.js +15 -2
- package/dist/utils/model.js.map +2 -2
- package/dist/utils/permissions/filesystem.js +5 -1
- package/dist/utils/permissions/filesystem.js.map +2 -2
- package/dist/utils/ripgrep.js +53 -1
- package/dist/utils/ripgrep.js.map +2 -2
- package/dist/utils/safePath.js +132 -0
- package/dist/utils/safePath.js.map +7 -0
- package/dist/utils/sensitiveFiles.js +125 -0
- package/dist/utils/sensitiveFiles.js.map +7 -0
- package/dist/utils/taskDisplayUtils.js +9 -9
- package/dist/utils/taskDisplayUtils.js.map +2 -2
- package/dist/utils/terminal.js +12 -0
- package/dist/utils/terminal.js.map +2 -2
- package/dist/utils/theme.js +6 -6
- package/dist/utils/theme.js.map +1 -1
- package/dist/utils/toolRiskClassification.js +207 -0
- package/dist/utils/toolRiskClassification.js.map +7 -0
- package/dist/utils/tooling/safeRender.js +17 -17
- package/dist/utils/tooling/safeRender.js.map +2 -2
- package/dist/version.js +2 -2
- package/dist/version.js.map +1 -1
- package/package.json +22 -28
- package/dist/hooks/useCancelRequest.js +0 -31
- package/dist/hooks/useCancelRequest.js.map +0 -7
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
{
|
|
2
|
+
"version": 3,
|
|
3
|
+
"sources": ["../../../src/tools/TaskListTool/prompt.ts"],
|
|
4
|
+
"sourcesContent": ["export const PROMPT = `Use this tool to list all tasks in the task list.\n\n## When to Use This Tool\n\n- To see what tasks are available to work on (status: 'pending', no owner, not blocked)\n- To check overall progress on the project\n- To find tasks that are blocked and need dependencies resolved\n- After completing a task, to check for newly unblocked work or claim the next available task\n- **Prefer working on tasks in ID order** (lowest ID first) when multiple tasks are available, as earlier tasks often set up context for later ones\n\n## Output\n\nReturns a summary of each task:\n- **id**: Task identifier (use with TaskGet, TaskUpdate)\n- **subject**: Brief description of the task\n- **status**: 'pending', 'in_progress', or 'completed'\n- **owner**: Agent ID if assigned, empty if available\n- **blockedBy**: List of open task IDs that must be resolved first (tasks with blockedBy cannot be claimed until dependencies resolve)\n\nUse TaskGet with a specific task ID to view full details including description and comments.\n`\n\nexport const DESCRIPTION = 'List all tasks in the task list'\n"],
|
|
5
|
+
"mappings": "AAAO,MAAM,SAAS;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAsBf,MAAM,cAAc;",
|
|
6
|
+
"names": []
|
|
7
|
+
}
|
|
@@ -6,6 +6,7 @@ import { BackgroundShellManager } from "../../utils/BackgroundShellManager.js";
|
|
|
6
6
|
import {
|
|
7
7
|
getAgentTranscript
|
|
8
8
|
} from "../../utils/agentTranscripts.js";
|
|
9
|
+
import { SEMANTIC_COLORS } from "../../constants/colors.js";
|
|
9
10
|
const inputSchema = z.strictObject({
|
|
10
11
|
task_id: z.string().describe("The task ID to get output from"),
|
|
11
12
|
block: z.boolean().default(true).optional().describe("Whether to wait for completion"),
|
|
@@ -40,7 +41,7 @@ const TaskOutputTool = {
|
|
|
40
41
|
return `task_id: "${input.task_id}", mode: ${blocking}`;
|
|
41
42
|
},
|
|
42
43
|
renderToolUseRejectedMessage() {
|
|
43
|
-
return /* @__PURE__ */ React.createElement(Box, { paddingLeft: 2 }, /* @__PURE__ */ React.createElement(Text, {
|
|
44
|
+
return /* @__PURE__ */ React.createElement(Box, { paddingLeft: 2 }, /* @__PURE__ */ React.createElement(Text, { color: SEMANTIC_COLORS.dim }, "Task output request rejected"));
|
|
44
45
|
},
|
|
45
46
|
renderToolResultMessage(output, options) {
|
|
46
47
|
if (!output) {
|
|
@@ -60,7 +61,7 @@ const TaskOutputTool = {
|
|
|
60
61
|
return output.output.substring(0, maxLen) + "...";
|
|
61
62
|
};
|
|
62
63
|
const outputPreview = getOutputPreview();
|
|
63
|
-
return /* @__PURE__ */ React.createElement(Box, { flexDirection: "column", paddingLeft: 2 }, /* @__PURE__ */ React.createElement(Box, { flexDirection: "row" }, /* @__PURE__ */ React.createElement(Text, null, output.task_type === "agent" ? "Agent" : "Task", " "), /* @__PURE__ */ React.createElement(Text, { bold: true }, output.task_id), /* @__PURE__ */ React.createElement(Text, null, ": "), /* @__PURE__ */ React.createElement(Text, { color: statusColor }, output.status), output.exitCode !== void 0 && /* @__PURE__ */ React.createElement(Text, {
|
|
64
|
+
return /* @__PURE__ */ React.createElement(Box, { flexDirection: "column", paddingLeft: 2 }, /* @__PURE__ */ React.createElement(Box, { flexDirection: "row" }, /* @__PURE__ */ React.createElement(Text, null, output.task_type === "agent" ? "Agent" : "Task", " "), /* @__PURE__ */ React.createElement(Text, { bold: true }, output.task_id), /* @__PURE__ */ React.createElement(Text, null, ": "), /* @__PURE__ */ React.createElement(Text, { color: statusColor }, output.status), output.exitCode !== void 0 && /* @__PURE__ */ React.createElement(Text, { color: SEMANTIC_COLORS.dim }, " (exit: ", output.exitCode, ")")), output.task_type === "agent" && output.agentType && /* @__PURE__ */ React.createElement(Box, null, /* @__PURE__ */ React.createElement(Text, { color: SEMANTIC_COLORS.dim }, "Type: ", output.agentType), output.messageCount !== void 0 && /* @__PURE__ */ React.createElement(Text, { color: SEMANTIC_COLORS.dim }, " ", "| Messages: ", output.messageCount), output.toolUseCount !== void 0 && /* @__PURE__ */ React.createElement(Text, { color: SEMANTIC_COLORS.dim }, " ", "| Tools: ", output.toolUseCount), output.tokenUsage && verbose && /* @__PURE__ */ React.createElement(Text, { color: SEMANTIC_COLORS.dim }, " ", "| Tokens:", " ", output.tokenUsage.inputTokens + output.tokenUsage.outputTokens)), outputPreview && /* @__PURE__ */ React.createElement(Box, { marginTop: 1 }, /* @__PURE__ */ React.createElement(Text, { color: SEMANTIC_COLORS.dim }, outputPreview)));
|
|
64
65
|
},
|
|
65
66
|
renderResultForAssistant(output) {
|
|
66
67
|
if (!output) {
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
{
|
|
2
2
|
"version": 3,
|
|
3
3
|
"sources": ["../../../src/tools/TaskOutputTool/TaskOutputTool.tsx"],
|
|
4
|
-
"sourcesContent": ["/**\n * Task Output Tool\n *\n * Retrieves output from background tasks (shells, agents, remote sessions).\n * Supports both background shell commands and agent transcripts.\n */\n\nimport { Box, Text } from 'ink'\nimport React from 'react'\nimport { z } from 'zod'\nimport { Tool } from '@tool'\nimport { DESCRIPTION, PROMPT, TOOL_NAME } from './prompt'\nimport { BackgroundShellManager } from '@utils/BackgroundShellManager'\nimport {\n getAgentTranscript,\n listAgentTranscripts,\n type AgentTranscript,\n type AgentStatus,\n} from '@utils/agentTranscripts'\n\nconst inputSchema = z.strictObject({\n task_id: z.string().describe('The task ID to get output from'),\n block: z\n .boolean()\n .default(true)\n .optional()\n .describe('Whether to wait for completion'),\n timeout: z\n .number()\n .min(0)\n .max(600000)\n .default(30000)\n .optional()\n .describe('Max wait time in ms'),\n})\n\ntype Output = {\n task_id: string\n task_type: 'shell' | 'agent'\n status: 'running' | 'completed' | 'failed' | 'interrupted' | 'not_found'\n output: string\n exitCode?: number\n // Agent-specific fields\n agentType?: string\n description?: string\n messageCount?: number\n toolUseCount?: number\n tokenUsage?: {\n inputTokens: number\n outputTokens: number\n }\n}\n\nexport const TaskOutputTool = {\n name: TOOL_NAME,\n async description() {\n return DESCRIPTION\n },\n userFacingName() {\n return 'Task Output'\n },\n inputSchema,\n isReadOnly() {\n return true\n },\n isConcurrencySafe() {\n return true\n },\n async isEnabled() {\n return true\n },\n needsPermissions() {\n return false\n },\n async prompt() {\n return PROMPT\n },\n renderToolUseMessage(input: z.infer<typeof inputSchema>) {\n const blocking = input.block !== false ? 'blocking' : 'non-blocking'\n return `task_id: \"${input.task_id}\", mode: ${blocking}`\n },\n renderToolUseRejectedMessage() {\n return (\n <Box paddingLeft={2}>\n <Text dimColor>Task output request rejected</Text>\n </Box>\n )\n },\n renderToolResultMessage(output: Output, options?: { verbose?: boolean }) {\n if (!output) {\n return (\n <Box paddingLeft={2}>\n <Text>Task output retrieved</Text>\n </Box>\n )\n }\n\n const verbose = options?.verbose ?? false\n const statusColor =\n output.status === 'completed'\n ? 'green'\n : output.status === 'running'\n ? 'yellow'\n : output.status === 'failed' || output.status === 'interrupted'\n ? 'red'\n : 'gray'\n\n // Determine output truncation based on displayMode (via verbose flag)\n // minimal (verbose=false): Show only status line, no output content\n // compact (verbose=false): Show status + truncated output (200 chars)\n // detailed (verbose=true): Show status + full output (up to 2000 chars)\n const getOutputPreview = () => {\n if (!output.output) return null\n\n // For minimal mode, don't show output content at all in the summary\n // The full content is available via the tool result\n if (!verbose) {\n // Compact mode: show truncated output\n const maxLen = 200\n if (output.output.length <= maxLen) return output.output\n return output.output.substring(0, maxLen) + '...'\n }\n\n // Detailed mode: show more output\n const maxLen = 2000\n if (output.output.length <= maxLen) return output.output\n return output.output.substring(0, maxLen) + '...'\n }\n\n const outputPreview = getOutputPreview()\n\n return (\n <Box flexDirection=\"column\" paddingLeft={2}>\n <Box flexDirection=\"row\">\n <Text>{output.task_type === 'agent' ? 'Agent' : 'Task'} </Text>\n <Text bold>{output.task_id}</Text>\n <Text>: </Text>\n <Text color={statusColor}>{output.status}</Text>\n {output.exitCode !== undefined && (\n <Text dimColor> (exit: {output.exitCode})</Text>\n )}\n </Box>\n {output.task_type === 'agent' && output.agentType && (\n <Box>\n <Text dimColor>Type: {output.agentType}</Text>\n {output.messageCount !== undefined && (\n <Text dimColor> | Messages: {output.messageCount}</Text>\n )}\n {output.toolUseCount !== undefined && (\n <Text dimColor> | Tools: {output.toolUseCount}</Text>\n )}\n {output.tokenUsage && verbose && (\n <Text dimColor>\n {' '}\n | Tokens:{' '}\n {output.tokenUsage.inputTokens + output.tokenUsage.outputTokens}\n </Text>\n )}\n </Box>\n )}\n {outputPreview && (\n <Box marginTop={1}>\n <Text dimColor>{outputPreview}</Text>\n </Box>\n )}\n </Box>\n )\n },\n renderResultForAssistant(output: Output) {\n if (!output) {\n return 'Task output retrieved'\n }\n\n const lines = [\n `Task ID: ${output.task_id}`,\n `Task Type: ${output.task_type}`,\n `Status: ${output.status}`,\n ]\n\n if (output.task_type === 'agent') {\n if (output.agentType) lines.push(`Agent Type: ${output.agentType}`)\n if (output.description) lines.push(`Description: ${output.description}`)\n if (output.messageCount !== undefined)\n lines.push(`Messages: ${output.messageCount}`)\n if (output.toolUseCount !== undefined)\n lines.push(`Tool Uses: ${output.toolUseCount}`)\n if (output.tokenUsage) {\n lines.push(\n `Tokens: ${output.tokenUsage.inputTokens + output.tokenUsage.outputTokens} (in: ${output.tokenUsage.inputTokens}, out: ${output.tokenUsage.outputTokens})`,\n )\n }\n }\n\n if (output.exitCode !== undefined) {\n lines.push(`Exit Code: ${output.exitCode}`)\n }\n\n if (output.output) {\n lines.push('', 'Output:', output.output)\n }\n\n return lines.join('\\n')\n },\n async *call(input: z.infer<typeof inputSchema>, _context: any) {\n const { task_id, block = true, timeout = 30000 } = input\n\n // First, check if this is an agent transcript\n const transcript = getAgentTranscript(task_id)\n if (transcript) {\n // Handle agent transcript\n let currentTranscript = transcript\n\n // If blocking and agent is running, poll for completion\n if (block && transcript.status === 'running') {\n const startTime = Date.now()\n const pollInterval = 500 // 500ms\n\n while (Date.now() - startTime < timeout) {\n const updated = getAgentTranscript(task_id)\n if (!updated || updated.status !== 'running') {\n currentTranscript = updated || currentTranscript\n break\n }\n await new Promise(resolve => setTimeout(resolve, pollInterval))\n }\n }\n\n // Extract final text output from transcript messages\n let outputText = ''\n const lastMessage =\n currentTranscript.messages[currentTranscript.messages.length - 1]\n if (lastMessage?.type === 'assistant') {\n const textBlocks = lastMessage.message.content.filter(\n (block: any) => block.type === 'text',\n )\n outputText = textBlocks.map((block: any) => block.text).join('\\n')\n }\n\n const output: Output = {\n task_id,\n task_type: 'agent',\n status: currentTranscript.status as Output['status'],\n output: outputText,\n agentType: currentTranscript.agentType,\n description: currentTranscript.description,\n messageCount: currentTranscript.messages.length,\n toolUseCount: currentTranscript.toolUseCount,\n tokenUsage: currentTranscript.tokenUsage,\n }\n\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n return\n }\n\n // Fall back to background shell\n const manager = BackgroundShellManager.getInstance()\n const shell = manager.get(task_id)\n\n if (!shell) {\n const output: Output = {\n task_id,\n task_type: 'shell',\n status: 'not_found',\n output: `No task found with ID: ${task_id}. This could be a background shell or agent ID.`,\n }\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n return\n }\n\n // If blocking, wait for completion\n if (block && shell.status === 'running') {\n await new Promise<void>((resolve, reject) => {\n const timeoutId = setTimeout(() => {\n manager.removeListener('statusChange', listener)\n resolve()\n }, timeout)\n\n const listener = (shellId: string, status: string) => {\n if (shellId === task_id && status !== 'running') {\n clearTimeout(timeoutId)\n manager.removeListener('statusChange', listener)\n resolve()\n }\n }\n\n manager.on('statusChange', listener)\n })\n }\n\n // Get all output\n const shellOutput = manager.getAllOutput(task_id)\n const currentShell = manager.get(task_id)\n\n const output: Output = {\n task_id,\n task_type: 'shell',\n status:\n currentShell?.status === 'running'\n ? 'running'\n : currentShell?.exitCode === 0\n ? 'completed'\n : 'failed',\n output: shellOutput\n ? [...shellOutput.stdout, ...shellOutput.stderr].join('\\n')\n : '',\n exitCode: currentShell?.exitCode,\n }\n\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n },\n} satisfies Tool<typeof inputSchema, Output>\n"],
|
|
5
|
-
"mappings": "AAOA,SAAS,KAAK,YAAY;AAC1B,OAAO,WAAW;AAClB,SAAS,SAAS;AAElB,SAAS,aAAa,QAAQ,iBAAiB;AAC/C,SAAS,8BAA8B;AACvC;AAAA,EACE;AAAA,OAIK;
|
|
4
|
+
"sourcesContent": ["/**\n * Task Output Tool\n *\n * Retrieves output from background tasks (shells, agents, remote sessions).\n * Supports both background shell commands and agent transcripts.\n */\n\nimport { Box, Text } from 'ink'\nimport React from 'react'\nimport { z } from 'zod'\nimport { Tool } from '@tool'\nimport { DESCRIPTION, PROMPT, TOOL_NAME } from './prompt'\nimport { BackgroundShellManager } from '@utils/BackgroundShellManager'\nimport {\n getAgentTranscript,\n listAgentTranscripts,\n type AgentTranscript,\n type AgentStatus,\n} from '@utils/agentTranscripts'\nimport { SEMANTIC_COLORS } from '@constants/colors'\n\nconst inputSchema = z.strictObject({\n task_id: z.string().describe('The task ID to get output from'),\n block: z\n .boolean()\n .default(true)\n .optional()\n .describe('Whether to wait for completion'),\n timeout: z\n .number()\n .min(0)\n .max(600000)\n .default(30000)\n .optional()\n .describe('Max wait time in ms'),\n})\n\ntype Output = {\n task_id: string\n task_type: 'shell' | 'agent'\n status: 'running' | 'completed' | 'failed' | 'interrupted' | 'not_found'\n output: string\n exitCode?: number\n // Agent-specific fields\n agentType?: string\n description?: string\n messageCount?: number\n toolUseCount?: number\n tokenUsage?: {\n inputTokens: number\n outputTokens: number\n }\n}\n\nexport const TaskOutputTool = {\n name: TOOL_NAME,\n async description() {\n return DESCRIPTION\n },\n userFacingName() {\n return 'Task Output'\n },\n inputSchema,\n isReadOnly() {\n return true\n },\n isConcurrencySafe() {\n return true\n },\n async isEnabled() {\n return true\n },\n needsPermissions() {\n return false\n },\n async prompt() {\n return PROMPT\n },\n renderToolUseMessage(input: z.infer<typeof inputSchema>) {\n const blocking = input.block !== false ? 'blocking' : 'non-blocking'\n return `task_id: \"${input.task_id}\", mode: ${blocking}`\n },\n renderToolUseRejectedMessage() {\n return (\n <Box paddingLeft={2}>\n <Text color={SEMANTIC_COLORS.dim}>Task output request rejected</Text>\n </Box>\n )\n },\n renderToolResultMessage(output: Output, options?: { verbose?: boolean }) {\n if (!output) {\n return (\n <Box paddingLeft={2}>\n <Text>Task output retrieved</Text>\n </Box>\n )\n }\n\n const verbose = options?.verbose ?? false\n const statusColor =\n output.status === 'completed'\n ? 'green'\n : output.status === 'running'\n ? 'yellow'\n : output.status === 'failed' || output.status === 'interrupted'\n ? 'red'\n : 'gray'\n\n // Determine output truncation based on displayMode (via verbose flag)\n // minimal (verbose=false): Show only status line, no output content\n // compact (verbose=false): Show status + truncated output (200 chars)\n // detailed (verbose=true): Show status + full output (up to 2000 chars)\n const getOutputPreview = () => {\n if (!output.output) return null\n\n // For minimal mode, don't show output content at all in the summary\n // The full content is available via the tool result\n if (!verbose) {\n // Compact mode: show truncated output\n const maxLen = 200\n if (output.output.length <= maxLen) return output.output\n return output.output.substring(0, maxLen) + '...'\n }\n\n // Detailed mode: show more output\n const maxLen = 2000\n if (output.output.length <= maxLen) return output.output\n return output.output.substring(0, maxLen) + '...'\n }\n\n const outputPreview = getOutputPreview()\n\n return (\n <Box flexDirection=\"column\" paddingLeft={2}>\n <Box flexDirection=\"row\">\n <Text>{output.task_type === 'agent' ? 'Agent' : 'Task'} </Text>\n <Text bold>{output.task_id}</Text>\n <Text>: </Text>\n <Text color={statusColor}>{output.status}</Text>\n {output.exitCode !== undefined && (\n <Text color={SEMANTIC_COLORS.dim}> (exit: {output.exitCode})</Text>\n )}\n </Box>\n {output.task_type === 'agent' && output.agentType && (\n <Box>\n <Text color={SEMANTIC_COLORS.dim}>Type: {output.agentType}</Text>\n {output.messageCount !== undefined && (\n <Text color={SEMANTIC_COLORS.dim}>\n {' '}\n | Messages: {output.messageCount}\n </Text>\n )}\n {output.toolUseCount !== undefined && (\n <Text color={SEMANTIC_COLORS.dim}>\n {' '}\n | Tools: {output.toolUseCount}\n </Text>\n )}\n {output.tokenUsage && verbose && (\n <Text color={SEMANTIC_COLORS.dim}>\n {' '}\n | Tokens:{' '}\n {output.tokenUsage.inputTokens + output.tokenUsage.outputTokens}\n </Text>\n )}\n </Box>\n )}\n {outputPreview && (\n <Box marginTop={1}>\n <Text color={SEMANTIC_COLORS.dim}>{outputPreview}</Text>\n </Box>\n )}\n </Box>\n )\n },\n renderResultForAssistant(output: Output) {\n if (!output) {\n return 'Task output retrieved'\n }\n\n const lines = [\n `Task ID: ${output.task_id}`,\n `Task Type: ${output.task_type}`,\n `Status: ${output.status}`,\n ]\n\n if (output.task_type === 'agent') {\n if (output.agentType) lines.push(`Agent Type: ${output.agentType}`)\n if (output.description) lines.push(`Description: ${output.description}`)\n if (output.messageCount !== undefined)\n lines.push(`Messages: ${output.messageCount}`)\n if (output.toolUseCount !== undefined)\n lines.push(`Tool Uses: ${output.toolUseCount}`)\n if (output.tokenUsage) {\n lines.push(\n `Tokens: ${output.tokenUsage.inputTokens + output.tokenUsage.outputTokens} (in: ${output.tokenUsage.inputTokens}, out: ${output.tokenUsage.outputTokens})`,\n )\n }\n }\n\n if (output.exitCode !== undefined) {\n lines.push(`Exit Code: ${output.exitCode}`)\n }\n\n if (output.output) {\n lines.push('', 'Output:', output.output)\n }\n\n return lines.join('\\n')\n },\n async *call(input: z.infer<typeof inputSchema>, _context: any) {\n const { task_id, block = true, timeout = 30000 } = input\n\n // First, check if this is an agent transcript\n const transcript = getAgentTranscript(task_id)\n if (transcript) {\n // Handle agent transcript\n let currentTranscript = transcript\n\n // If blocking and agent is running, poll for completion\n if (block && transcript.status === 'running') {\n const startTime = Date.now()\n const pollInterval = 500 // 500ms\n\n while (Date.now() - startTime < timeout) {\n const updated = getAgentTranscript(task_id)\n if (!updated || updated.status !== 'running') {\n currentTranscript = updated || currentTranscript\n break\n }\n await new Promise(resolve => setTimeout(resolve, pollInterval))\n }\n }\n\n // Extract final text output from transcript messages\n let outputText = ''\n const lastMessage =\n currentTranscript.messages[currentTranscript.messages.length - 1]\n if (lastMessage?.type === 'assistant') {\n const textBlocks = lastMessage.message.content.filter(\n (block: any) => block.type === 'text',\n )\n outputText = textBlocks.map((block: any) => block.text).join('\\n')\n }\n\n const output: Output = {\n task_id,\n task_type: 'agent',\n status: currentTranscript.status as Output['status'],\n output: outputText,\n agentType: currentTranscript.agentType,\n description: currentTranscript.description,\n messageCount: currentTranscript.messages.length,\n toolUseCount: currentTranscript.toolUseCount,\n tokenUsage: currentTranscript.tokenUsage,\n }\n\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n return\n }\n\n // Fall back to background shell\n const manager = BackgroundShellManager.getInstance()\n const shell = manager.get(task_id)\n\n if (!shell) {\n const output: Output = {\n task_id,\n task_type: 'shell',\n status: 'not_found',\n output: `No task found with ID: ${task_id}. This could be a background shell or agent ID.`,\n }\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n return\n }\n\n // If blocking, wait for completion\n if (block && shell.status === 'running') {\n await new Promise<void>((resolve, reject) => {\n const timeoutId = setTimeout(() => {\n manager.removeListener('statusChange', listener)\n resolve()\n }, timeout)\n\n const listener = (shellId: string, status: string) => {\n if (shellId === task_id && status !== 'running') {\n clearTimeout(timeoutId)\n manager.removeListener('statusChange', listener)\n resolve()\n }\n }\n\n manager.on('statusChange', listener)\n })\n }\n\n // Get all output\n const shellOutput = manager.getAllOutput(task_id)\n const currentShell = manager.get(task_id)\n\n const output: Output = {\n task_id,\n task_type: 'shell',\n status:\n currentShell?.status === 'running'\n ? 'running'\n : currentShell?.exitCode === 0\n ? 'completed'\n : 'failed',\n output: shellOutput\n ? [...shellOutput.stdout, ...shellOutput.stderr].join('\\n')\n : '',\n exitCode: currentShell?.exitCode,\n }\n\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n },\n} satisfies Tool<typeof inputSchema, Output>\n"],
|
|
5
|
+
"mappings": "AAOA,SAAS,KAAK,YAAY;AAC1B,OAAO,WAAW;AAClB,SAAS,SAAS;AAElB,SAAS,aAAa,QAAQ,iBAAiB;AAC/C,SAAS,8BAA8B;AACvC;AAAA,EACE;AAAA,OAIK;AACP,SAAS,uBAAuB;AAEhC,MAAM,cAAc,EAAE,aAAa;AAAA,EACjC,SAAS,EAAE,OAAO,EAAE,SAAS,gCAAgC;AAAA,EAC7D,OAAO,EACJ,QAAQ,EACR,QAAQ,IAAI,EACZ,SAAS,EACT,SAAS,gCAAgC;AAAA,EAC5C,SAAS,EACN,OAAO,EACP,IAAI,CAAC,EACL,IAAI,GAAM,EACV,QAAQ,GAAK,EACb,SAAS,EACT,SAAS,qBAAqB;AACnC,CAAC;AAmBM,MAAM,iBAAiB;AAAA,EAC5B,MAAM;AAAA,EACN,MAAM,cAAc;AAClB,WAAO;AAAA,EACT;AAAA,EACA,iBAAiB;AACf,WAAO;AAAA,EACT;AAAA,EACA;AAAA,EACA,aAAa;AACX,WAAO;AAAA,EACT;AAAA,EACA,oBAAoB;AAClB,WAAO;AAAA,EACT;AAAA,EACA,MAAM,YAAY;AAChB,WAAO;AAAA,EACT;AAAA,EACA,mBAAmB;AACjB,WAAO;AAAA,EACT;AAAA,EACA,MAAM,SAAS;AACb,WAAO;AAAA,EACT;AAAA,EACA,qBAAqB,OAAoC;AACvD,UAAM,WAAW,MAAM,UAAU,QAAQ,aAAa;AACtD,WAAO,aAAa,MAAM,OAAO,YAAY,QAAQ;AAAA,EACvD;AAAA,EACA,+BAA+B;AAC7B,WACE,oCAAC,OAAI,aAAa,KAChB,oCAAC,QAAK,OAAO,gBAAgB,OAAK,8BAA4B,CAChE;AAAA,EAEJ;AAAA,EACA,wBAAwB,QAAgB,SAAiC;AACvE,QAAI,CAAC,QAAQ;AACX,aACE,oCAAC,OAAI,aAAa,KAChB,oCAAC,YAAK,uBAAqB,CAC7B;AAAA,IAEJ;AAEA,UAAM,UAAU,SAAS,WAAW;AACpC,UAAM,cACJ,OAAO,WAAW,cACd,UACA,OAAO,WAAW,YAChB,WACA,OAAO,WAAW,YAAY,OAAO,WAAW,gBAC9C,QACA;AAMV,UAAM,mBAAmB,MAAM;AAC7B,UAAI,CAAC,OAAO,OAAQ,QAAO;AAI3B,UAAI,CAAC,SAAS;AAEZ,cAAMA,UAAS;AACf,YAAI,OAAO,OAAO,UAAUA,QAAQ,QAAO,OAAO;AAClD,eAAO,OAAO,OAAO,UAAU,GAAGA,OAAM,IAAI;AAAA,MAC9C;AAGA,YAAM,SAAS;AACf,UAAI,OAAO,OAAO,UAAU,OAAQ,QAAO,OAAO;AAClD,aAAO,OAAO,OAAO,UAAU,GAAG,MAAM,IAAI;AAAA,IAC9C;AAEA,UAAM,gBAAgB,iBAAiB;AAEvC,WACE,oCAAC,OAAI,eAAc,UAAS,aAAa,KACvC,oCAAC,OAAI,eAAc,SACjB,oCAAC,YAAM,OAAO,cAAc,UAAU,UAAU,QAAO,GAAC,GACxD,oCAAC,QAAK,MAAI,QAAE,OAAO,OAAQ,GAC3B,oCAAC,YAAK,IAAE,GACR,oCAAC,QAAK,OAAO,eAAc,OAAO,MAAO,GACxC,OAAO,aAAa,UACnB,oCAAC,QAAK,OAAO,gBAAgB,OAAK,YAAS,OAAO,UAAS,GAAC,CAEhE,GACC,OAAO,cAAc,WAAW,OAAO,aACtC,oCAAC,WACC,oCAAC,QAAK,OAAO,gBAAgB,OAAK,UAAO,OAAO,SAAU,GACzD,OAAO,iBAAiB,UACvB,oCAAC,QAAK,OAAO,gBAAgB,OAC1B,KAAI,gBACQ,OAAO,YACtB,GAED,OAAO,iBAAiB,UACvB,oCAAC,QAAK,OAAO,gBAAgB,OAC1B,KAAI,aACK,OAAO,YACnB,GAED,OAAO,cAAc,WACpB,oCAAC,QAAK,OAAO,gBAAgB,OAC1B,KAAI,aACK,KACT,OAAO,WAAW,cAAc,OAAO,WAAW,YACrD,CAEJ,GAED,iBACC,oCAAC,OAAI,WAAW,KACd,oCAAC,QAAK,OAAO,gBAAgB,OAAM,aAAc,CACnD,CAEJ;AAAA,EAEJ;AAAA,EACA,yBAAyB,QAAgB;AACvC,QAAI,CAAC,QAAQ;AACX,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ;AAAA,MACZ,YAAY,OAAO,OAAO;AAAA,MAC1B,cAAc,OAAO,SAAS;AAAA,MAC9B,WAAW,OAAO,MAAM;AAAA,IAC1B;AAEA,QAAI,OAAO,cAAc,SAAS;AAChC,UAAI,OAAO,UAAW,OAAM,KAAK,eAAe,OAAO,SAAS,EAAE;AAClE,UAAI,OAAO,YAAa,OAAM,KAAK,gBAAgB,OAAO,WAAW,EAAE;AACvE,UAAI,OAAO,iBAAiB;AAC1B,cAAM,KAAK,aAAa,OAAO,YAAY,EAAE;AAC/C,UAAI,OAAO,iBAAiB;AAC1B,cAAM,KAAK,cAAc,OAAO,YAAY,EAAE;AAChD,UAAI,OAAO,YAAY;AACrB,cAAM;AAAA,UACJ,WAAW,OAAO,WAAW,cAAc,OAAO,WAAW,YAAY,SAAS,OAAO,WAAW,WAAW,UAAU,OAAO,WAAW,YAAY;AAAA,QACzJ;AAAA,MACF;AAAA,IACF;AAEA,QAAI,OAAO,aAAa,QAAW;AACjC,YAAM,KAAK,cAAc,OAAO,QAAQ,EAAE;AAAA,IAC5C;AAEA,QAAI,OAAO,QAAQ;AACjB,YAAM,KAAK,IAAI,WAAW,OAAO,MAAM;AAAA,IACzC;AAEA,WAAO,MAAM,KAAK,IAAI;AAAA,EACxB;AAAA,EACA,OAAO,KAAK,OAAoC,UAAe;AAC7D,UAAM,EAAE,SAAS,QAAQ,MAAM,UAAU,IAAM,IAAI;AAGnD,UAAM,aAAa,mBAAmB,OAAO;AAC7C,QAAI,YAAY;AAEd,UAAI,oBAAoB;AAGxB,UAAI,SAAS,WAAW,WAAW,WAAW;AAC5C,cAAM,YAAY,KAAK,IAAI;AAC3B,cAAM,eAAe;AAErB,eAAO,KAAK,IAAI,IAAI,YAAY,SAAS;AACvC,gBAAM,UAAU,mBAAmB,OAAO;AAC1C,cAAI,CAAC,WAAW,QAAQ,WAAW,WAAW;AAC5C,gCAAoB,WAAW;AAC/B;AAAA,UACF;AACA,gBAAM,IAAI,QAAQ,aAAW,WAAW,SAAS,YAAY,CAAC;AAAA,QAChE;AAAA,MACF;AAGA,UAAI,aAAa;AACjB,YAAM,cACJ,kBAAkB,SAAS,kBAAkB,SAAS,SAAS,CAAC;AAClE,UAAI,aAAa,SAAS,aAAa;AACrC,cAAM,aAAa,YAAY,QAAQ,QAAQ;AAAA,UAC7C,CAACC,WAAeA,OAAM,SAAS;AAAA,QACjC;AACA,qBAAa,WAAW,IAAI,CAACA,WAAeA,OAAM,IAAI,EAAE,KAAK,IAAI;AAAA,MACnE;AAEA,YAAMC,UAAiB;AAAA,QACrB;AAAA,QACA,WAAW;AAAA,QACX,QAAQ,kBAAkB;AAAA,QAC1B,QAAQ;AAAA,QACR,WAAW,kBAAkB;AAAA,QAC7B,aAAa,kBAAkB;AAAA,QAC/B,cAAc,kBAAkB,SAAS;AAAA,QACzC,cAAc,kBAAkB;AAAA,QAChC,YAAY,kBAAkB;AAAA,MAChC;AAEA,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,MAAMA;AAAA,QACN,oBAAoB,KAAK,yBAAyBA,OAAM;AAAA,MAC1D;AACA;AAAA,IACF;AAGA,UAAM,UAAU,uBAAuB,YAAY;AACnD,UAAM,QAAQ,QAAQ,IAAI,OAAO;AAEjC,QAAI,CAAC,OAAO;AACV,YAAMA,UAAiB;AAAA,QACrB;AAAA,QACA,WAAW;AAAA,QACX,QAAQ;AAAA,QACR,QAAQ,0BAA0B,OAAO;AAAA,MAC3C;AACA,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,MAAMA;AAAA,QACN,oBAAoB,KAAK,yBAAyBA,OAAM;AAAA,MAC1D;AACA;AAAA,IACF;AAGA,QAAI,SAAS,MAAM,WAAW,WAAW;AACvC,YAAM,IAAI,QAAc,CAAC,SAAS,WAAW;AAC3C,cAAM,YAAY,WAAW,MAAM;AACjC,kBAAQ,eAAe,gBAAgB,QAAQ;AAC/C,kBAAQ;AAAA,QACV,GAAG,OAAO;AAEV,cAAM,WAAW,CAAC,SAAiB,WAAmB;AACpD,cAAI,YAAY,WAAW,WAAW,WAAW;AAC/C,yBAAa,SAAS;AACtB,oBAAQ,eAAe,gBAAgB,QAAQ;AAC/C,oBAAQ;AAAA,UACV;AAAA,QACF;AAEA,gBAAQ,GAAG,gBAAgB,QAAQ;AAAA,MACrC,CAAC;AAAA,IACH;AAGA,UAAM,cAAc,QAAQ,aAAa,OAAO;AAChD,UAAM,eAAe,QAAQ,IAAI,OAAO;AAExC,UAAM,SAAiB;AAAA,MACrB;AAAA,MACA,WAAW;AAAA,MACX,QACE,cAAc,WAAW,YACrB,YACA,cAAc,aAAa,IACzB,cACA;AAAA,MACR,QAAQ,cACJ,CAAC,GAAG,YAAY,QAAQ,GAAG,YAAY,MAAM,EAAE,KAAK,IAAI,IACxD;AAAA,MACJ,UAAU,cAAc;AAAA,IAC1B;AAEA,UAAM;AAAA,MACJ,MAAM;AAAA,MACN,MAAM;AAAA,MACN,oBAAoB,KAAK,yBAAyB,MAAM;AAAA,IAC1D;AAAA,EACF;AACF;",
|
|
6
6
|
"names": ["maxLen", "block", "output"]
|
|
7
7
|
}
|
|
@@ -0,0 +1,150 @@
|
|
|
1
|
+
import { Box, Text } from "ink";
|
|
2
|
+
import React from "react";
|
|
3
|
+
import { z } from "zod";
|
|
4
|
+
import { DESCRIPTION, PROMPT, TOOL_NAME } from "./prompt.js";
|
|
5
|
+
import { BackgroundShellManager } from "../../utils/BackgroundShellManager.js";
|
|
6
|
+
import {
|
|
7
|
+
getAgentTranscript,
|
|
8
|
+
interruptAgentTranscript
|
|
9
|
+
} from "../../utils/agentTranscripts.js";
|
|
10
|
+
import { SEMANTIC_COLORS } from "../../constants/colors.js";
|
|
11
|
+
const inputSchema = z.strictObject({
|
|
12
|
+
task_id: z.string().describe("The ID of the background task to stop"),
|
|
13
|
+
shell_id: z.string().optional().describe("Deprecated: use task_id instead")
|
|
14
|
+
});
|
|
15
|
+
const TaskStopTool = {
|
|
16
|
+
name: TOOL_NAME,
|
|
17
|
+
async description() {
|
|
18
|
+
return DESCRIPTION;
|
|
19
|
+
},
|
|
20
|
+
userFacingName() {
|
|
21
|
+
return "Task Stop";
|
|
22
|
+
},
|
|
23
|
+
inputSchema,
|
|
24
|
+
isReadOnly() {
|
|
25
|
+
return false;
|
|
26
|
+
},
|
|
27
|
+
isConcurrencySafe() {
|
|
28
|
+
return true;
|
|
29
|
+
},
|
|
30
|
+
async isEnabled() {
|
|
31
|
+
return true;
|
|
32
|
+
},
|
|
33
|
+
needsPermissions() {
|
|
34
|
+
return false;
|
|
35
|
+
},
|
|
36
|
+
async prompt() {
|
|
37
|
+
return PROMPT;
|
|
38
|
+
},
|
|
39
|
+
renderToolUseMessage(input) {
|
|
40
|
+
const taskId = input.task_id || input.shell_id || "";
|
|
41
|
+
return `task_id: "${taskId}"`;
|
|
42
|
+
},
|
|
43
|
+
renderToolUseRejectedMessage() {
|
|
44
|
+
return /* @__PURE__ */ React.createElement(Box, { paddingLeft: 2 }, /* @__PURE__ */ React.createElement(Text, { color: SEMANTIC_COLORS.dim }, "Task stop request rejected"));
|
|
45
|
+
},
|
|
46
|
+
renderToolResultMessage(output) {
|
|
47
|
+
if (!output) {
|
|
48
|
+
return /* @__PURE__ */ React.createElement(Box, { paddingLeft: 2 }, /* @__PURE__ */ React.createElement(Text, null, "Task stop completed"));
|
|
49
|
+
}
|
|
50
|
+
const statusColor = output.success ? "green" : "red";
|
|
51
|
+
return /* @__PURE__ */ React.createElement(Box, { flexDirection: "column", paddingLeft: 2 }, /* @__PURE__ */ React.createElement(Box, { flexDirection: "row" }, /* @__PURE__ */ React.createElement(Text, null, output.task_type === "agent" ? "Agent" : "Task", " "), /* @__PURE__ */ React.createElement(Text, { bold: true }, output.task_id), /* @__PURE__ */ React.createElement(Text, null, ": "), /* @__PURE__ */ React.createElement(Text, { color: statusColor }, output.success ? "stopped" : "failed")), !output.success && /* @__PURE__ */ React.createElement(Box, { marginTop: 1 }, /* @__PURE__ */ React.createElement(Text, { color: SEMANTIC_COLORS.dim }, output.message)));
|
|
52
|
+
},
|
|
53
|
+
renderResultForAssistant(output) {
|
|
54
|
+
return output.message;
|
|
55
|
+
},
|
|
56
|
+
async *call(input, _context) {
|
|
57
|
+
const taskId = input.task_id || input.shell_id;
|
|
58
|
+
if (!taskId) {
|
|
59
|
+
const output2 = {
|
|
60
|
+
success: false,
|
|
61
|
+
message: "Error: No task_id provided",
|
|
62
|
+
task_id: "",
|
|
63
|
+
task_type: "unknown"
|
|
64
|
+
};
|
|
65
|
+
yield {
|
|
66
|
+
type: "result",
|
|
67
|
+
data: output2,
|
|
68
|
+
resultForAssistant: this.renderResultForAssistant(output2)
|
|
69
|
+
};
|
|
70
|
+
return;
|
|
71
|
+
}
|
|
72
|
+
const transcript = getAgentTranscript(taskId);
|
|
73
|
+
if (transcript) {
|
|
74
|
+
if (transcript.status !== "running") {
|
|
75
|
+
const output3 = {
|
|
76
|
+
success: false,
|
|
77
|
+
message: `Error: Agent '${taskId}' is already ${transcript.status}`,
|
|
78
|
+
task_id: taskId,
|
|
79
|
+
task_type: "agent"
|
|
80
|
+
};
|
|
81
|
+
yield {
|
|
82
|
+
type: "result",
|
|
83
|
+
data: output3,
|
|
84
|
+
resultForAssistant: this.renderResultForAssistant(output3)
|
|
85
|
+
};
|
|
86
|
+
return;
|
|
87
|
+
}
|
|
88
|
+
const updated = interruptAgentTranscript(taskId);
|
|
89
|
+
const success = updated !== null && updated.status === "interrupted";
|
|
90
|
+
const output2 = {
|
|
91
|
+
success,
|
|
92
|
+
message: success ? `Successfully stopped agent '${taskId}'` : `Failed to stop agent '${taskId}'`,
|
|
93
|
+
task_id: taskId,
|
|
94
|
+
task_type: "agent"
|
|
95
|
+
};
|
|
96
|
+
yield {
|
|
97
|
+
type: "result",
|
|
98
|
+
data: output2,
|
|
99
|
+
resultForAssistant: this.renderResultForAssistant(output2)
|
|
100
|
+
};
|
|
101
|
+
return;
|
|
102
|
+
}
|
|
103
|
+
const manager = BackgroundShellManager.getInstance();
|
|
104
|
+
const shell = manager.get(taskId);
|
|
105
|
+
if (!shell) {
|
|
106
|
+
const output2 = {
|
|
107
|
+
success: false,
|
|
108
|
+
message: `Error: No task found with ID '${taskId}'. This could be a background shell or agent ID.`,
|
|
109
|
+
task_id: taskId,
|
|
110
|
+
task_type: "unknown"
|
|
111
|
+
};
|
|
112
|
+
yield {
|
|
113
|
+
type: "result",
|
|
114
|
+
data: output2,
|
|
115
|
+
resultForAssistant: this.renderResultForAssistant(output2)
|
|
116
|
+
};
|
|
117
|
+
return;
|
|
118
|
+
}
|
|
119
|
+
if (shell.status !== "running") {
|
|
120
|
+
const output2 = {
|
|
121
|
+
success: false,
|
|
122
|
+
message: `Error: Task '${taskId}' is already ${shell.status}`,
|
|
123
|
+
task_id: taskId,
|
|
124
|
+
task_type: "shell"
|
|
125
|
+
};
|
|
126
|
+
yield {
|
|
127
|
+
type: "result",
|
|
128
|
+
data: output2,
|
|
129
|
+
resultForAssistant: this.renderResultForAssistant(output2)
|
|
130
|
+
};
|
|
131
|
+
return;
|
|
132
|
+
}
|
|
133
|
+
const killed = manager.kill(taskId);
|
|
134
|
+
const output = {
|
|
135
|
+
success: killed,
|
|
136
|
+
message: killed ? `Successfully stopped background task '${taskId}'` : `Failed to stop background task '${taskId}'`,
|
|
137
|
+
task_id: taskId,
|
|
138
|
+
task_type: "shell"
|
|
139
|
+
};
|
|
140
|
+
yield {
|
|
141
|
+
type: "result",
|
|
142
|
+
data: output,
|
|
143
|
+
resultForAssistant: this.renderResultForAssistant(output)
|
|
144
|
+
};
|
|
145
|
+
}
|
|
146
|
+
};
|
|
147
|
+
export {
|
|
148
|
+
TaskStopTool
|
|
149
|
+
};
|
|
150
|
+
//# sourceMappingURL=TaskStopTool.js.map
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
{
|
|
2
|
+
"version": 3,
|
|
3
|
+
"sources": ["../../../src/tools/TaskStopTool/TaskStopTool.tsx"],
|
|
4
|
+
"sourcesContent": ["/**\n * Task Stop Tool\n *\n * Stops a running background task by its ID.\n * Supports both background shell commands and agent tasks.\n */\n\nimport { Box, Text } from 'ink'\nimport React from 'react'\nimport { z } from 'zod'\nimport { Tool } from '@tool'\nimport { DESCRIPTION, PROMPT, TOOL_NAME } from './prompt'\nimport { BackgroundShellManager } from '@utils/BackgroundShellManager'\nimport {\n getAgentTranscript,\n interruptAgentTranscript,\n} from '@utils/agentTranscripts'\nimport { SEMANTIC_COLORS } from '@constants/colors'\n\nconst inputSchema = z.strictObject({\n task_id: z.string().describe('The ID of the background task to stop'),\n shell_id: z.string().optional().describe('Deprecated: use task_id instead'),\n})\n\ntype Output = {\n success: boolean\n message: string\n task_id: string\n task_type: 'shell' | 'agent' | 'unknown'\n}\n\nexport const TaskStopTool = {\n name: TOOL_NAME,\n async description() {\n return DESCRIPTION\n },\n userFacingName() {\n return 'Task Stop'\n },\n inputSchema,\n isReadOnly() {\n return false\n },\n isConcurrencySafe() {\n return true\n },\n async isEnabled() {\n return true\n },\n needsPermissions() {\n return false // Stopping tasks doesn't require user permission\n },\n async prompt() {\n return PROMPT\n },\n renderToolUseMessage(input: z.infer<typeof inputSchema>) {\n const taskId = input.task_id || input.shell_id || ''\n return `task_id: \"${taskId}\"`\n },\n renderToolUseRejectedMessage() {\n return (\n <Box paddingLeft={2}>\n <Text color={SEMANTIC_COLORS.dim}>Task stop request rejected</Text>\n </Box>\n )\n },\n renderToolResultMessage(output: Output) {\n if (!output) {\n return (\n <Box paddingLeft={2}>\n <Text>Task stop completed</Text>\n </Box>\n )\n }\n\n const statusColor = output.success ? 'green' : 'red'\n\n return (\n <Box flexDirection=\"column\" paddingLeft={2}>\n <Box flexDirection=\"row\">\n <Text>{output.task_type === 'agent' ? 'Agent' : 'Task'} </Text>\n <Text bold>{output.task_id}</Text>\n <Text>: </Text>\n <Text color={statusColor}>\n {output.success ? 'stopped' : 'failed'}\n </Text>\n </Box>\n {!output.success && (\n <Box marginTop={1}>\n <Text color={SEMANTIC_COLORS.dim}>{output.message}</Text>\n </Box>\n )}\n </Box>\n )\n },\n renderResultForAssistant(output: Output) {\n return output.message\n },\n async *call(input: z.infer<typeof inputSchema>, _context: any) {\n // Support deprecated shell_id parameter\n const taskId = input.task_id || input.shell_id\n\n if (!taskId) {\n const output: Output = {\n success: false,\n message: 'Error: No task_id provided',\n task_id: '',\n task_type: 'unknown',\n }\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n return\n }\n\n // First, check if this is an agent transcript\n const transcript = getAgentTranscript(taskId)\n if (transcript) {\n // Handle agent transcript\n if (transcript.status !== 'running') {\n const output: Output = {\n success: false,\n message: `Error: Agent '${taskId}' is already ${transcript.status}`,\n task_id: taskId,\n task_type: 'agent',\n }\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n return\n }\n\n // Interrupt the agent\n const updated = interruptAgentTranscript(taskId)\n const success = updated !== null && updated.status === 'interrupted'\n\n const output: Output = {\n success,\n message: success\n ? `Successfully stopped agent '${taskId}'`\n : `Failed to stop agent '${taskId}'`,\n task_id: taskId,\n task_type: 'agent',\n }\n\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n return\n }\n\n // Fall back to background shell\n const manager = BackgroundShellManager.getInstance()\n const shell = manager.get(taskId)\n\n if (!shell) {\n const output: Output = {\n success: false,\n message: `Error: No task found with ID '${taskId}'. This could be a background shell or agent ID.`,\n task_id: taskId,\n task_type: 'unknown',\n }\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n return\n }\n\n if (shell.status !== 'running') {\n const output: Output = {\n success: false,\n message: `Error: Task '${taskId}' is already ${shell.status}`,\n task_id: taskId,\n task_type: 'shell',\n }\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n return\n }\n\n // Kill the shell\n const killed = manager.kill(taskId)\n\n const output: Output = {\n success: killed,\n message: killed\n ? `Successfully stopped background task '${taskId}'`\n : `Failed to stop background task '${taskId}'`,\n task_id: taskId,\n task_type: 'shell',\n }\n\n yield {\n type: 'result',\n data: output,\n resultForAssistant: this.renderResultForAssistant(output),\n }\n },\n} satisfies Tool<typeof inputSchema, Output>\n"],
|
|
5
|
+
"mappings": "AAOA,SAAS,KAAK,YAAY;AAC1B,OAAO,WAAW;AAClB,SAAS,SAAS;AAElB,SAAS,aAAa,QAAQ,iBAAiB;AAC/C,SAAS,8BAA8B;AACvC;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP,SAAS,uBAAuB;AAEhC,MAAM,cAAc,EAAE,aAAa;AAAA,EACjC,SAAS,EAAE,OAAO,EAAE,SAAS,uCAAuC;AAAA,EACpE,UAAU,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,iCAAiC;AAC5E,CAAC;AASM,MAAM,eAAe;AAAA,EAC1B,MAAM;AAAA,EACN,MAAM,cAAc;AAClB,WAAO;AAAA,EACT;AAAA,EACA,iBAAiB;AACf,WAAO;AAAA,EACT;AAAA,EACA;AAAA,EACA,aAAa;AACX,WAAO;AAAA,EACT;AAAA,EACA,oBAAoB;AAClB,WAAO;AAAA,EACT;AAAA,EACA,MAAM,YAAY;AAChB,WAAO;AAAA,EACT;AAAA,EACA,mBAAmB;AACjB,WAAO;AAAA,EACT;AAAA,EACA,MAAM,SAAS;AACb,WAAO;AAAA,EACT;AAAA,EACA,qBAAqB,OAAoC;AACvD,UAAM,SAAS,MAAM,WAAW,MAAM,YAAY;AAClD,WAAO,aAAa,MAAM;AAAA,EAC5B;AAAA,EACA,+BAA+B;AAC7B,WACE,oCAAC,OAAI,aAAa,KAChB,oCAAC,QAAK,OAAO,gBAAgB,OAAK,4BAA0B,CAC9D;AAAA,EAEJ;AAAA,EACA,wBAAwB,QAAgB;AACtC,QAAI,CAAC,QAAQ;AACX,aACE,oCAAC,OAAI,aAAa,KAChB,oCAAC,YAAK,qBAAmB,CAC3B;AAAA,IAEJ;AAEA,UAAM,cAAc,OAAO,UAAU,UAAU;AAE/C,WACE,oCAAC,OAAI,eAAc,UAAS,aAAa,KACvC,oCAAC,OAAI,eAAc,SACjB,oCAAC,YAAM,OAAO,cAAc,UAAU,UAAU,QAAO,GAAC,GACxD,oCAAC,QAAK,MAAI,QAAE,OAAO,OAAQ,GAC3B,oCAAC,YAAK,IAAE,GACR,oCAAC,QAAK,OAAO,eACV,OAAO,UAAU,YAAY,QAChC,CACF,GACC,CAAC,OAAO,WACP,oCAAC,OAAI,WAAW,KACd,oCAAC,QAAK,OAAO,gBAAgB,OAAM,OAAO,OAAQ,CACpD,CAEJ;AAAA,EAEJ;AAAA,EACA,yBAAyB,QAAgB;AACvC,WAAO,OAAO;AAAA,EAChB;AAAA,EACA,OAAO,KAAK,OAAoC,UAAe;AAE7D,UAAM,SAAS,MAAM,WAAW,MAAM;AAEtC,QAAI,CAAC,QAAQ;AACX,YAAMA,UAAiB;AAAA,QACrB,SAAS;AAAA,QACT,SAAS;AAAA,QACT,SAAS;AAAA,QACT,WAAW;AAAA,MACb;AACA,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,MAAMA;AAAA,QACN,oBAAoB,KAAK,yBAAyBA,OAAM;AAAA,MAC1D;AACA;AAAA,IACF;AAGA,UAAM,aAAa,mBAAmB,MAAM;AAC5C,QAAI,YAAY;AAEd,UAAI,WAAW,WAAW,WAAW;AACnC,cAAMA,UAAiB;AAAA,UACrB,SAAS;AAAA,UACT,SAAS,iBAAiB,MAAM,gBAAgB,WAAW,MAAM;AAAA,UACjE,SAAS;AAAA,UACT,WAAW;AAAA,QACb;AACA,cAAM;AAAA,UACJ,MAAM;AAAA,UACN,MAAMA;AAAA,UACN,oBAAoB,KAAK,yBAAyBA,OAAM;AAAA,QAC1D;AACA;AAAA,MACF;AAGA,YAAM,UAAU,yBAAyB,MAAM;AAC/C,YAAM,UAAU,YAAY,QAAQ,QAAQ,WAAW;AAEvD,YAAMA,UAAiB;AAAA,QACrB;AAAA,QACA,SAAS,UACL,+BAA+B,MAAM,MACrC,yBAAyB,MAAM;AAAA,QACnC,SAAS;AAAA,QACT,WAAW;AAAA,MACb;AAEA,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,MAAMA;AAAA,QACN,oBAAoB,KAAK,yBAAyBA,OAAM;AAAA,MAC1D;AACA;AAAA,IACF;AAGA,UAAM,UAAU,uBAAuB,YAAY;AACnD,UAAM,QAAQ,QAAQ,IAAI,MAAM;AAEhC,QAAI,CAAC,OAAO;AACV,YAAMA,UAAiB;AAAA,QACrB,SAAS;AAAA,QACT,SAAS,iCAAiC,MAAM;AAAA,QAChD,SAAS;AAAA,QACT,WAAW;AAAA,MACb;AACA,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,MAAMA;AAAA,QACN,oBAAoB,KAAK,yBAAyBA,OAAM;AAAA,MAC1D;AACA;AAAA,IACF;AAEA,QAAI,MAAM,WAAW,WAAW;AAC9B,YAAMA,UAAiB;AAAA,QACrB,SAAS;AAAA,QACT,SAAS,gBAAgB,MAAM,gBAAgB,MAAM,MAAM;AAAA,QAC3D,SAAS;AAAA,QACT,WAAW;AAAA,MACb;AACA,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,MAAMA;AAAA,QACN,oBAAoB,KAAK,yBAAyBA,OAAM;AAAA,MAC1D;AACA;AAAA,IACF;AAGA,UAAM,SAAS,QAAQ,KAAK,MAAM;AAElC,UAAM,SAAiB;AAAA,MACrB,SAAS;AAAA,MACT,SAAS,SACL,yCAAyC,MAAM,MAC/C,mCAAmC,MAAM;AAAA,MAC7C,SAAS;AAAA,MACT,WAAW;AAAA,IACb;AAEA,UAAM;AAAA,MACJ,MAAM;AAAA,MACN,MAAM;AAAA,MACN,oBAAoB,KAAK,yBAAyB,MAAM;AAAA,IAC1D;AAAA,EACF;AACF;",
|
|
6
|
+
"names": ["output"]
|
|
7
|
+
}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
const TOOL_NAME = "TaskStop";
|
|
2
|
+
const DESCRIPTION = `Stops a running background task by its ID`;
|
|
3
|
+
const PROMPT = `- Stops a running background task by its ID
|
|
4
|
+
- Takes a task_id parameter identifying the task to stop
|
|
5
|
+
- Returns a success or failure status
|
|
6
|
+
- Use this tool when you need to terminate a long-running task
|
|
7
|
+
- Task IDs can be obtained from BashTool when run_in_background is true, or from Task tool for agent tasks
|
|
8
|
+
- Can only stop tasks that are currently running (not already completed or stopped)
|
|
9
|
+
- Works with both background shells and agent tasks`;
|
|
10
|
+
export {
|
|
11
|
+
DESCRIPTION,
|
|
12
|
+
PROMPT,
|
|
13
|
+
TOOL_NAME
|
|
14
|
+
};
|
|
15
|
+
//# sourceMappingURL=prompt.js.map
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
{
|
|
2
|
+
"version": 3,
|
|
3
|
+
"sources": ["../../../src/tools/TaskStopTool/prompt.ts"],
|
|
4
|
+
"sourcesContent": ["export const TOOL_NAME = 'TaskStop'\n\nexport const DESCRIPTION = `Stops a running background task by its ID`\n\nexport const PROMPT = `- Stops a running background task by its ID\n- Takes a task_id parameter identifying the task to stop\n- Returns a success or failure status\n- Use this tool when you need to terminate a long-running task\n- Task IDs can be obtained from BashTool when run_in_background is true, or from Task tool for agent tasks\n- Can only stop tasks that are currently running (not already completed or stopped)\n- Works with both background shells and agent tasks`\n"],
|
|
5
|
+
"mappings": "AAAO,MAAM,YAAY;AAElB,MAAM,cAAc;AAEpB,MAAM,SAAS;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;",
|
|
6
|
+
"names": []
|
|
7
|
+
}
|
|
@@ -44,17 +44,28 @@ import {
|
|
|
44
44
|
registerToolUseAgent
|
|
45
45
|
} from "../../utils/agentTranscripts.js";
|
|
46
46
|
import { UserFriendlyError } from "../../utils/userFriendlyError.js";
|
|
47
|
+
import {
|
|
48
|
+
pushAgentContext,
|
|
49
|
+
popAgentContext,
|
|
50
|
+
cleanupAgentStreamingState
|
|
51
|
+
} from "../../components/Spinner.js";
|
|
47
52
|
const inputSchema = z.object({
|
|
48
53
|
description: z.string().describe("A short (3-5 word) description of the task"),
|
|
49
54
|
prompt: z.string().describe("The task for the agent to perform"),
|
|
50
55
|
model_name: z.string().optional().describe(
|
|
51
56
|
"Optional: Specific model name to use for this task. If not provided, uses the default task model pointer."
|
|
52
57
|
),
|
|
58
|
+
model: z.enum(["sonnet", "opus", "haiku"]).optional().describe(
|
|
59
|
+
"Optional model hint for this agent. Maps to configured model pointers: sonnet\u2192main, opus\u2192reasoning, haiku\u2192quick. If not specified, uses the task model pointer. Prefer haiku for quick, straightforward tasks."
|
|
60
|
+
),
|
|
53
61
|
subagent_type: z.string().optional().describe(
|
|
54
62
|
'The specialized agent type to use. MUST be an exact name from the available agents list in the tool description. Defaults to "general-purpose" if not specified.'
|
|
55
63
|
),
|
|
56
64
|
resume: z.string().optional().describe(
|
|
57
65
|
"Optional agent ID to resume from. If provided, the agent will continue from the previous execution transcript."
|
|
66
|
+
),
|
|
67
|
+
max_turns: z.number().int().positive().optional().describe(
|
|
68
|
+
"Maximum number of agentic turns (API round-trips) before stopping. Used internally for warmup."
|
|
58
69
|
)
|
|
59
70
|
// run_in_background parameter removed - concurrent execution is automatic
|
|
60
71
|
});
|
|
@@ -69,7 +80,15 @@ const TaskTool = {
|
|
|
69
80
|
return `Launch a specialized agent to handle tasks autonomously. You MUST use an exact agent type name from this list: ${typesList}. Default to "general-purpose" if unsure.`;
|
|
70
81
|
},
|
|
71
82
|
inputSchema,
|
|
72
|
-
async *call({
|
|
83
|
+
async *call({
|
|
84
|
+
description,
|
|
85
|
+
prompt,
|
|
86
|
+
model_name,
|
|
87
|
+
model,
|
|
88
|
+
subagent_type,
|
|
89
|
+
resume,
|
|
90
|
+
max_turns
|
|
91
|
+
}, {
|
|
73
92
|
abortController,
|
|
74
93
|
options: { safeMode = false, forkNumber, messageLogName, verbose },
|
|
75
94
|
readFileTimestamps,
|
|
@@ -142,6 +161,22 @@ ${prompt}`;
|
|
|
142
161
|
}
|
|
143
162
|
toolFilter = agentConfig.tools;
|
|
144
163
|
}
|
|
164
|
+
if (model) {
|
|
165
|
+
const modelPointerMap = {
|
|
166
|
+
sonnet: "main",
|
|
167
|
+
// Use main model for standard tasks
|
|
168
|
+
opus: "reasoning",
|
|
169
|
+
// Use reasoning model for complex tasks
|
|
170
|
+
haiku: "quick"
|
|
171
|
+
// Use quick model for fast tasks
|
|
172
|
+
};
|
|
173
|
+
const pointerType = modelPointerMap[model] || "task";
|
|
174
|
+
const modelManager = getModelManager();
|
|
175
|
+
const resolvedModelName = modelManager.getModelName(pointerType);
|
|
176
|
+
if (resolvedModelName) {
|
|
177
|
+
effectiveModel = resolvedModelName;
|
|
178
|
+
}
|
|
179
|
+
}
|
|
145
180
|
const messages = resumedTranscript ? [...resumedTranscript.messages] : [createUserMessage(effectivePrompt)];
|
|
146
181
|
let tools = await getTaskTools(safeMode);
|
|
147
182
|
if (toolFilter) {
|
|
@@ -243,6 +278,8 @@ IMPORTANT: You are currently running as ${modelToUse}. You do not need to consul
|
|
|
243
278
|
if (temperature !== void 0) {
|
|
244
279
|
queryOptions["temperature"] = temperature;
|
|
245
280
|
}
|
|
281
|
+
pushAgentContext(agentId);
|
|
282
|
+
let turnCount = 0;
|
|
246
283
|
try {
|
|
247
284
|
for await (const message of query(
|
|
248
285
|
messages,
|
|
@@ -269,6 +306,15 @@ IMPORTANT: You are currently running as ${modelToUse}. You do not need to consul
|
|
|
269
306
|
if (message.type !== "assistant") {
|
|
270
307
|
continue;
|
|
271
308
|
}
|
|
309
|
+
turnCount++;
|
|
310
|
+
if (max_turns && turnCount >= max_turns) {
|
|
311
|
+
debugLogger.info("TASK_AGENT_MAX_TURNS_REACHED", {
|
|
312
|
+
agentId,
|
|
313
|
+
turnCount,
|
|
314
|
+
max_turns
|
|
315
|
+
});
|
|
316
|
+
break;
|
|
317
|
+
}
|
|
272
318
|
if (message.message.usage) {
|
|
273
319
|
updateAgentTranscript(agentId, {
|
|
274
320
|
tokenUsage: {
|
|
@@ -413,6 +459,8 @@ ${this.renderResultForAssistant(data)}` : this.renderResultForAssistant(data);
|
|
|
413
459
|
}
|
|
414
460
|
hasYieldedResult = true;
|
|
415
461
|
} finally {
|
|
462
|
+
popAgentContext();
|
|
463
|
+
cleanupAgentStreamingState(agentId);
|
|
416
464
|
if (!hasYieldedResult) {
|
|
417
465
|
interruptAgentTranscript(agentId);
|
|
418
466
|
yield {
|