@markusylisiurunen/tau 0.3.32 → 0.3.33
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/README.md +9 -14
- package/dist/core/agent/agent_runtime.js +1505 -0
- package/dist/core/agent/agent_runtime.js.map +1 -0
- package/dist/core/agent/events.js +2 -0
- package/dist/core/agent/events.js.map +1 -0
- package/dist/core/commands/registry.js +0 -42
- package/dist/core/commands/registry.js.map +1 -1
- package/dist/core/config/content_loader.js +3 -1
- package/dist/core/config/content_loader.js.map +1 -1
- package/dist/core/index.js +1 -1
- package/dist/core/index.js.map +1 -1
- package/dist/core/models/catalog.js +7 -1
- package/dist/core/models/catalog.js.map +1 -1
- package/dist/core/models/tau_model_overrides.js +10 -0
- package/dist/core/models/tau_model_overrides.js.map +1 -0
- package/dist/core/modes/index.js.map +1 -1
- package/dist/core/personas.js +2 -1
- package/dist/core/personas.js.map +1 -1
- package/dist/core/runtime/agent_model.js +50 -0
- package/dist/core/runtime/agent_model.js.map +1 -0
- package/dist/core/runtime/chat_runtime.js +173 -63
- package/dist/core/runtime/chat_runtime.js.map +1 -1
- package/dist/core/runtime/model_executor.js +2 -0
- package/dist/core/runtime/model_executor.js.map +1 -0
- package/dist/core/runtime/model_sampler.js +21 -0
- package/dist/core/runtime/model_sampler.js.map +1 -0
- package/dist/core/session/compaction.js +1 -1
- package/dist/core/session/compaction.js.map +1 -1
- package/dist/core/session/direct_bash.js.map +1 -1
- package/dist/core/session/message_accumulator.js +3 -5
- package/dist/core/session/message_accumulator.js.map +1 -1
- package/dist/core/session/runner.js +197 -98
- package/dist/core/session/runner.js.map +1 -1
- package/dist/core/subagents/agent_supervisor.js +337 -0
- package/dist/core/subagents/agent_supervisor.js.map +1 -0
- package/dist/core/tools/activity.js +2 -0
- package/dist/core/tools/activity.js.map +1 -0
- package/dist/core/tools/bash.js +64 -67
- package/dist/core/tools/bash.js.map +1 -1
- package/dist/core/tools/catalog.js +46 -24
- package/dist/core/tools/catalog.js.map +1 -1
- package/dist/core/tools/code_mode.js +65 -58
- package/dist/core/tools/code_mode.js.map +1 -1
- package/dist/core/tools/edit.js +11 -12
- package/dist/core/tools/edit.js.map +1 -1
- package/dist/core/tools/nook.js +3 -2
- package/dist/core/tools/nook.js.map +1 -1
- package/dist/core/tools/registry.js +27 -13
- package/dist/core/tools/registry.js.map +1 -1
- package/dist/core/tools/send_input_to_agent.js +75 -90
- package/dist/core/tools/send_input_to_agent.js.map +1 -1
- package/dist/core/tools/spawn_agent.js +95 -100
- package/dist/core/tools/spawn_agent.js.map +1 -1
- package/dist/core/tools/terminate_agent.js +75 -82
- package/dist/core/tools/terminate_agent.js.map +1 -1
- package/dist/core/tools/view_image.js +11 -16
- package/dist/core/tools/view_image.js.map +1 -1
- package/dist/core/tools/wait_for_agents.js +61 -68
- package/dist/core/tools/wait_for_agents.js.map +1 -1
- package/dist/core/tools/web.js +3 -3
- package/dist/core/tools/web.js.map +1 -1
- package/dist/core/tools/write.js +10 -11
- package/dist/core/tools/write.js.map +1 -1
- package/dist/core/usage/logs.js.map +1 -1
- package/dist/core/utils/subagent_utils.js.map +1 -1
- package/dist/core/version.js +1 -1
- package/dist/execution/execution_environment.js.map +1 -1
- package/dist/execution/tool_backend_execution_environment.js +1 -7
- package/dist/execution/tool_backend_execution_environment.js.map +1 -1
- package/dist/host/client_tool_broker.js +26 -20
- package/dist/host/client_tool_broker.js.map +1 -1
- package/dist/host/hosted_ephemeral_agent_session.js +104 -167
- package/dist/host/hosted_ephemeral_agent_session.js.map +1 -1
- package/dist/host/local_session_host.js +310 -197
- package/dist/host/local_session_host.js.map +1 -1
- package/dist/host/session_host.js.map +1 -1
- package/dist/host/session_protocol_handler.js +121 -101
- package/dist/host/session_protocol_handler.js.map +1 -1
- package/dist/main.js +10 -2
- package/dist/main.js.map +1 -1
- package/dist/protocol/index.d.ts +1 -1
- package/dist/protocol/index.js.map +1 -1
- package/dist/protocol/session_protocol.d.ts +19 -21
- package/dist/protocol/session_protocol.js +150 -56
- package/dist/protocol/session_protocol.js.map +1 -1
- package/dist/sdk/index.d.ts +1 -1
- package/dist/sdk/index.js.map +1 -1
- package/dist/sdk/session.js +4 -21
- package/dist/sdk/session.js.map +1 -1
- package/dist/sdk/types.d.ts +2 -7
- package/dist/store/file_session_store.js +13 -9
- package/dist/store/file_session_store.js.map +1 -1
- package/dist/store/session_snapshot_migrations.js +133 -0
- package/dist/store/session_snapshot_migrations.js.map +1 -0
- package/dist/tui/chat_controller/command_hints.js +0 -6
- package/dist/tui/chat_controller/command_hints.js.map +1 -1
- package/dist/tui/chat_view.js +19 -12
- package/dist/tui/chat_view.js.map +1 -1
- package/dist/tui/session_chat_app.js +0 -8
- package/dist/tui/session_chat_app.js.map +1 -1
- package/dist/tui/session_chat_controller.js +94 -115
- package/dist/tui/session_chat_controller.js.map +1 -1
- package/dist/tui/tool_ui_router.js +12 -226
- package/dist/tui/tool_ui_router.js.map +1 -1
- package/dist/tui/ui/bash_execution.js +3 -3
- package/dist/tui/ui/bash_execution.js.map +1 -1
- package/dist/tui/ui/chat_message_model.js +2 -2
- package/dist/tui/ui/chat_message_model.js.map +1 -1
- package/dist/tui/ui/subagent_panel.js +39 -2
- package/dist/tui/ui/subagent_panel.js.map +1 -1
- package/dist/tui/ui/tool_ui_model.js +2 -0
- package/dist/tui/ui/tool_ui_model.js.map +1 -0
- package/dist/tui/ui/tool_ui_registry.js +256 -31
- package/dist/tui/ui/tool_ui_registry.js.map +1 -1
- package/package.json +1 -1
- package/dist/core/events/types.js +0 -2
- package/dist/core/events/types.js.map +0 -1
- package/dist/core/modes/mode_adapter.js +0 -2
- package/dist/core/modes/mode_adapter.js.map +0 -1
- package/dist/core/runtime/conversation_turn_runtime.js +0 -88
- package/dist/core/runtime/conversation_turn_runtime.js.map +0 -1
- package/dist/core/session/core_session.js +0 -86
- package/dist/core/session/core_session.js.map +0 -1
- package/dist/core/session/pruning.js +0 -587
- package/dist/core/session/pruning.js.map +0 -1
- package/dist/core/session/session_engine.js +0 -1246
- package/dist/core/session/session_engine.js.map +0 -1
- package/dist/core/subagents/control_plane.js +0 -339
- package/dist/core/subagents/control_plane.js.map +0 -1
- package/dist/core/subagents/subagent_engine.js +0 -258
- package/dist/core/subagents/subagent_engine.js.map +0 -1
|
@@ -1,8 +1,7 @@
|
|
|
1
1
|
import { Type } from "typebox";
|
|
2
2
|
import { z } from "zod";
|
|
3
|
-
import { createToolError, createToolResult } from "../utils/messages.js";
|
|
4
3
|
import { parseToolArgs } from "../utils/zod.js";
|
|
5
|
-
import {
|
|
4
|
+
import { createTextToolOutcome, executeTool, } from "./registry.js";
|
|
6
5
|
import { buildSubagentUiText, formatSubagentStatusLine } from "./subagent_ui.js";
|
|
7
6
|
import { TOOL_NAME_TERMINATE_AGENT } from "./tool_names.js";
|
|
8
7
|
const TERMINATE_AGENT_DESCRIPTION = [
|
|
@@ -31,10 +30,10 @@ function formatTerminateResult(result) {
|
|
|
31
30
|
};
|
|
32
31
|
return JSON.stringify(payload, null, 2);
|
|
33
32
|
}
|
|
34
|
-
function formatTerminateOutput(result) {
|
|
33
|
+
function formatTerminateOutput(result, succeeded) {
|
|
35
34
|
const finalText = result.finalText?.trimEnd() ?? "";
|
|
36
35
|
const body = finalText.trim().length > 0 ? finalText : "";
|
|
37
|
-
if (
|
|
36
|
+
if (!succeeded) {
|
|
38
37
|
const errorLine = result.error ? `Error: ${result.error}` : `Status: ${result.status}`;
|
|
39
38
|
return body ? `${errorLine}\n${body}` : errorLine;
|
|
40
39
|
}
|
|
@@ -48,15 +47,16 @@ function getTerminateAgentDisplayTarget(raw) {
|
|
|
48
47
|
const parsedArgs = parseToolArgs(terminateArgsSchema, raw);
|
|
49
48
|
return parsedArgs.ok ? parsedArgs.data.id : "(invalid arguments)";
|
|
50
49
|
}
|
|
51
|
-
export function createTerminateAgentToolDefinition() {
|
|
50
|
+
export function createTerminateAgentToolDefinition(supervisor) {
|
|
52
51
|
return {
|
|
53
52
|
schema: TERMINATE_AGENT_TOOL,
|
|
54
|
-
|
|
55
|
-
async
|
|
53
|
+
describe: (toolCall) => ({ headerTarget: getTerminateAgentDisplayTarget(toolCall.arguments) }),
|
|
54
|
+
async execute(toolCall, context) {
|
|
55
|
+
const { signal } = context;
|
|
56
56
|
let id = "";
|
|
57
57
|
const headerTarget = getTerminateAgentDisplayTarget(toolCall.arguments);
|
|
58
58
|
const blocked = (reason) => {
|
|
59
|
-
const
|
|
59
|
+
const outcome = createTextToolOutcome(reason, "blocked");
|
|
60
60
|
const uiEvent = {
|
|
61
61
|
type: "terminate_agent_blocked",
|
|
62
62
|
toolCallId: toolCall.id,
|
|
@@ -64,83 +64,23 @@ export function createTerminateAgentToolDefinition() {
|
|
|
64
64
|
headerTarget,
|
|
65
65
|
reason,
|
|
66
66
|
};
|
|
67
|
-
return {
|
|
67
|
+
return { content: outcome.content, outcome: outcome.outcome, uiEvent };
|
|
68
68
|
};
|
|
69
69
|
const parsedArgs = parseToolArgs(terminateArgsSchema, toolCall.arguments);
|
|
70
70
|
if (!parsedArgs.ok) {
|
|
71
|
-
return
|
|
71
|
+
return executeTool(context, () => blocked(`Invalid arguments: ${parsedArgs.error}`));
|
|
72
72
|
}
|
|
73
73
|
({ id } = parsedArgs.data);
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
startedUiEvent: {
|
|
80
|
-
type: "terminate_agent_started",
|
|
81
|
-
toolCallId: toolCall.id,
|
|
82
|
-
agentId: id,
|
|
83
|
-
headerTarget,
|
|
84
|
-
},
|
|
85
|
-
run: (async () => {
|
|
86
|
-
try {
|
|
87
|
-
const result = await controlPlane.terminate(id, signal);
|
|
88
|
-
if (!result) {
|
|
89
|
-
const message = `Unknown subagent ID '${id}'.`;
|
|
90
|
-
const uiText = buildSubagentUiText({
|
|
91
|
-
output: message,
|
|
92
|
-
statusText: "error",
|
|
93
|
-
maxOutputLines: 16,
|
|
94
|
-
fullText: message,
|
|
95
|
-
});
|
|
96
|
-
const uiEvent = {
|
|
97
|
-
type: "terminate_agent_finished",
|
|
98
|
-
toolCallId: toolCall.id,
|
|
99
|
-
agentId: id,
|
|
100
|
-
headerTarget,
|
|
101
|
-
status: "error",
|
|
102
|
-
message,
|
|
103
|
-
uiText,
|
|
104
|
-
};
|
|
105
|
-
const toolResult = createToolError(toolCall, message);
|
|
106
|
-
return { toolResult, uiEvent };
|
|
107
|
-
}
|
|
108
|
-
const resultText = formatTerminateResult(result);
|
|
109
|
-
const succeeded = result.status === "success";
|
|
110
|
-
const baseStatusText = formatSubagentStatusLine({
|
|
111
|
-
costTotal: result.costTotal,
|
|
112
|
-
durationMs: getTerminateDurationMs(result),
|
|
113
|
-
});
|
|
114
|
-
const statusText = succeeded
|
|
115
|
-
? baseStatusText
|
|
116
|
-
: `${baseStatusText} · Status ${result.status}`;
|
|
117
|
-
const uiText = buildSubagentUiText({
|
|
118
|
-
output: formatTerminateOutput(result),
|
|
119
|
-
statusText,
|
|
120
|
-
maxOutputLines: 16,
|
|
121
|
-
fullText: resultText,
|
|
122
|
-
});
|
|
123
|
-
const uiEvent = {
|
|
124
|
-
type: "terminate_agent_finished",
|
|
125
|
-
toolCallId: toolCall.id,
|
|
126
|
-
agentId: id,
|
|
127
|
-
headerTarget,
|
|
128
|
-
status: succeeded ? "success" : "error",
|
|
129
|
-
finalStatus: result.status,
|
|
130
|
-
message: succeeded ? undefined : `Subagent finished with status ${result.status}.`,
|
|
131
|
-
uiText,
|
|
132
|
-
};
|
|
133
|
-
const toolResult = createToolResult(toolCall, resultText, !succeeded);
|
|
134
|
-
return { toolResult, uiEvent };
|
|
135
|
-
}
|
|
136
|
-
catch (error) {
|
|
137
|
-
const message = error instanceof Error ? error.message : String(error);
|
|
138
|
-
const reason = message.trim() || "The terminate_agent request failed.";
|
|
74
|
+
return executeTool(context, async () => {
|
|
75
|
+
try {
|
|
76
|
+
const result = await supervisor.terminate(id, signal);
|
|
77
|
+
if (!result) {
|
|
78
|
+
const message = `Unknown subagent ID '${id}'.`;
|
|
139
79
|
const uiText = buildSubagentUiText({
|
|
140
|
-
output:
|
|
80
|
+
output: message,
|
|
141
81
|
statusText: "error",
|
|
142
82
|
maxOutputLines: 16,
|
|
143
|
-
fullText:
|
|
83
|
+
fullText: message,
|
|
144
84
|
});
|
|
145
85
|
const uiEvent = {
|
|
146
86
|
type: "terminate_agent_finished",
|
|
@@ -148,14 +88,67 @@ export function createTerminateAgentToolDefinition() {
|
|
|
148
88
|
agentId: id,
|
|
149
89
|
headerTarget,
|
|
150
90
|
status: "error",
|
|
151
|
-
message
|
|
91
|
+
message,
|
|
152
92
|
uiText,
|
|
153
93
|
};
|
|
154
|
-
const
|
|
155
|
-
return {
|
|
94
|
+
const outcome = createTextToolOutcome(message, "blocked");
|
|
95
|
+
return { content: outcome.content, outcome: outcome.outcome, uiEvent };
|
|
156
96
|
}
|
|
157
|
-
|
|
158
|
-
|
|
97
|
+
const resultText = formatTerminateResult(result);
|
|
98
|
+
const succeeded = result.status === "success" || result.status === "aborted";
|
|
99
|
+
const baseStatusText = formatSubagentStatusLine({
|
|
100
|
+
costTotal: result.costTotal,
|
|
101
|
+
durationMs: getTerminateDurationMs(result),
|
|
102
|
+
});
|
|
103
|
+
const statusText = succeeded
|
|
104
|
+
? baseStatusText
|
|
105
|
+
: `${baseStatusText} · Status ${result.status}`;
|
|
106
|
+
const uiText = buildSubagentUiText({
|
|
107
|
+
output: formatTerminateOutput(result, succeeded),
|
|
108
|
+
statusText,
|
|
109
|
+
maxOutputLines: 16,
|
|
110
|
+
fullText: resultText,
|
|
111
|
+
});
|
|
112
|
+
const uiEvent = {
|
|
113
|
+
type: "terminate_agent_finished",
|
|
114
|
+
toolCallId: toolCall.id,
|
|
115
|
+
agentId: id,
|
|
116
|
+
headerTarget,
|
|
117
|
+
status: succeeded ? "success" : "error",
|
|
118
|
+
finalStatus: result.status,
|
|
119
|
+
message: succeeded ? undefined : `Subagent finished with status ${result.status}.`,
|
|
120
|
+
uiText,
|
|
121
|
+
};
|
|
122
|
+
const outcome = createTextToolOutcome(resultText, succeeded ? "succeeded" : "failed");
|
|
123
|
+
return { content: outcome.content, outcome: outcome.outcome, uiEvent };
|
|
124
|
+
}
|
|
125
|
+
catch (error) {
|
|
126
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
127
|
+
const reason = message.trim() || "The terminate_agent request failed.";
|
|
128
|
+
const uiText = buildSubagentUiText({
|
|
129
|
+
output: reason,
|
|
130
|
+
statusText: "error",
|
|
131
|
+
maxOutputLines: 16,
|
|
132
|
+
fullText: reason,
|
|
133
|
+
});
|
|
134
|
+
const uiEvent = {
|
|
135
|
+
type: "terminate_agent_finished",
|
|
136
|
+
toolCallId: toolCall.id,
|
|
137
|
+
agentId: id,
|
|
138
|
+
headerTarget,
|
|
139
|
+
status: "error",
|
|
140
|
+
message: reason,
|
|
141
|
+
uiText,
|
|
142
|
+
};
|
|
143
|
+
const outcome = createTextToolOutcome(reason, signal.aborted ? "cancelled" : "failed");
|
|
144
|
+
return { content: outcome.content, outcome: outcome.outcome, uiEvent };
|
|
145
|
+
}
|
|
146
|
+
}, {
|
|
147
|
+
type: "terminate_agent_started",
|
|
148
|
+
toolCallId: toolCall.id,
|
|
149
|
+
agentId: id,
|
|
150
|
+
headerTarget,
|
|
151
|
+
});
|
|
159
152
|
},
|
|
160
153
|
};
|
|
161
154
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"terminate_agent.js","sourceRoot":"","sources":["../../../src/core/tools/terminate_agent.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"terminate_agent.js","sourceRoot":"","sources":["../../../src/core/tools/terminate_agent.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,OAAO,EAEL,qBAAqB,EACrB,WAAW,GAIZ,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AACjF,OAAO,EAAE,yBAAyB,EAAE,MAAM,iBAAiB,CAAC;AAE5D,MAAM,2BAA2B,GAAG;IAClC,wEAAwE;IACxE,mEAAmE;CACpE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAEZ,MAAM,8BAA8B,GAAG,2BAA2B,CAAC;AAEnE,MAAM,CAAC,MAAM,oBAAoB,GAAS;IACxC,IAAI,EAAE,yBAAyB;IAC/B,WAAW,EAAE,2BAA2B;IACxC,UAAU,EAAE,IAAI,CAAC,MAAM,CACrB;QACE,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,8BAA8B,EAAE,CAAC;KACjE,EACD,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAChC;CACF,CAAC;AAEF,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;CAC7B,CAAC,CAAC;AAEH,SAAS,qBAAqB,CAAC,MAAsB;IACnD,MAAM,OAAO,GAAG;QACd,EAAE,EAAE,MAAM,CAAC,EAAE;QACb,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;KACpB,CAAC;IAEF,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAC1C,CAAC;AAED,SAAS,qBAAqB,CAAC,MAAsB,EAAE,SAAkB;IACvE,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IACpD,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;IAE1D,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,WAAW,MAAM,CAAC,MAAM,EAAE,CAAC;QACvF,OAAO,IAAI,CAAC,CAAC,CAAC,GAAG,SAAS,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACpD,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,sBAAsB,CAAC,MAAsB;IACpD,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;IACnD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,8BAA8B,CAAC,GAAY;IAClD,MAAM,UAAU,GAAG,aAAa,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC;IAC3D,OAAO,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,qBAAqB,CAAC;AACpE,CAAC;AAED,MAAM,UAAU,kCAAkC,CAAC,UAA2B;IAC5E,OAAO;QACL,MAAM,EAAE,oBAAoB;QAC5B,QAAQ,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,8BAA8B,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9F,KAAK,CAAC,OAAO,CACX,QAAkB,EAClB,OAA6B;YAE7B,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;YAC3B,IAAI,EAAE,GAAG,EAAE,CAAC;YACZ,MAAM,YAAY,GAAG,8BAA8B,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YAExE,MAAM,OAAO,GAAG,CAAC,MAAc,EAA6B,EAAE;gBAC5D,MAAM,OAAO,GAAG,qBAAqB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;gBACzD,MAAM,OAAO,GAAiB;oBAC5B,IAAI,EAAE,yBAAyB;oBAC/B,UAAU,EAAE,QAAQ,CAAC,EAAE;oBACvB,OAAO,EAAE,EAAE,IAAI,SAAS;oBACxB,YAAY;oBACZ,MAAM;iBACP,CAAC;gBACF,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;YACzE,CAAC,CAAC;YAEF,MAAM,UAAU,GAAG,aAAa,CAAC,mBAAmB,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YAC1E,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,CAAC;gBACnB,OAAO,WAAW,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,sBAAsB,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YACvF,CAAC;YAED,CAAC,EAAE,EAAE,EAAE,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;YAE3B,OAAO,WAAW,CAChB,OAAO,EACP,KAAK,IAAwC,EAAE;gBAC7C,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;oBACtD,IAAI,CAAC,MAAM,EAAE,CAAC;wBACZ,MAAM,OAAO,GAAG,wBAAwB,EAAE,IAAI,CAAC;wBAC/C,MAAM,MAAM,GAAG,mBAAmB,CAAC;4BACjC,MAAM,EAAE,OAAO;4BACf,UAAU,EAAE,OAAO;4BACnB,cAAc,EAAE,EAAE;4BAClB,QAAQ,EAAE,OAAO;yBAClB,CAAC,CAAC;wBACH,MAAM,OAAO,GAAiB;4BAC5B,IAAI,EAAE,0BAA0B;4BAChC,UAAU,EAAE,QAAQ,CAAC,EAAE;4BACvB,OAAO,EAAE,EAAE;4BACX,YAAY;4BACZ,MAAM,EAAE,OAAO;4BACf,OAAO;4BACP,MAAM;yBACP,CAAC;wBACF,MAAM,OAAO,GAAG,qBAAqB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;wBAC1D,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;oBACzE,CAAC;oBAED,MAAM,UAAU,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;oBACjD,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,CAAC;oBAC7E,MAAM,cAAc,GAAG,wBAAwB,CAAC;wBAC9C,SAAS,EAAE,MAAM,CAAC,SAAS;wBAC3B,UAAU,EAAE,sBAAsB,CAAC,MAAM,CAAC;qBAC3C,CAAC,CAAC;oBACH,MAAM,UAAU,GAAG,SAAS;wBAC1B,CAAC,CAAC,cAAc;wBAChB,CAAC,CAAC,GAAG,cAAc,aAAa,MAAM,CAAC,MAAM,EAAE,CAAC;oBAClD,MAAM,MAAM,GAAG,mBAAmB,CAAC;wBACjC,MAAM,EAAE,qBAAqB,CAAC,MAAM,EAAE,SAAS,CAAC;wBAChD,UAAU;wBACV,cAAc,EAAE,EAAE;wBAClB,QAAQ,EAAE,UAAU;qBACrB,CAAC,CAAC;oBACH,MAAM,OAAO,GAAiB;wBAC5B,IAAI,EAAE,0BAA0B;wBAChC,UAAU,EAAE,QAAQ,CAAC,EAAE;wBACvB,OAAO,EAAE,EAAE;wBACX,YAAY;wBACZ,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;wBACvC,WAAW,EAAE,MAAM,CAAC,MAAM;wBAC1B,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,iCAAiC,MAAM,CAAC,MAAM,GAAG;wBAClF,MAAM;qBACP,CAAC;oBACF,MAAM,OAAO,GAAG,qBAAqB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;oBACtF,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;gBACzE,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACvE,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,EAAE,IAAI,qCAAqC,CAAC;oBACvE,MAAM,MAAM,GAAG,mBAAmB,CAAC;wBACjC,MAAM,EAAE,MAAM;wBACd,UAAU,EAAE,OAAO;wBACnB,cAAc,EAAE,EAAE;wBAClB,QAAQ,EAAE,MAAM;qBACjB,CAAC,CAAC;oBACH,MAAM,OAAO,GAAiB;wBAC5B,IAAI,EAAE,0BAA0B;wBAChC,UAAU,EAAE,QAAQ,CAAC,EAAE;wBACvB,OAAO,EAAE,EAAE;wBACX,YAAY;wBACZ,MAAM,EAAE,OAAO;wBACf,OAAO,EAAE,MAAM;wBACf,MAAM;qBACP,CAAC;oBACF,MAAM,OAAO,GAAG,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;oBACvF,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;gBACzE,CAAC;YACH,CAAC,EACD;gBACE,IAAI,EAAE,yBAAyB;gBAC/B,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,OAAO,EAAE,EAAE;gBACX,YAAY;aACb,CACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -2,10 +2,9 @@ import { fileTypeFromBuffer } from "file-type";
|
|
|
2
2
|
import sharp from "sharp";
|
|
3
3
|
import { Type } from "typebox";
|
|
4
4
|
import { z } from "zod";
|
|
5
|
-
import { createToolError } from "../utils/messages.js";
|
|
6
5
|
import { formatBytes } from "../utils/truncate.js";
|
|
7
6
|
import { formatZodError } from "../utils/zod.js";
|
|
8
|
-
import {
|
|
7
|
+
import { createTextToolOutcome, executeTool, } from "./registry.js";
|
|
9
8
|
import { TOOL_NAME_VIEW_IMAGE } from "./tool_names.js";
|
|
10
9
|
const VIEW_IMAGE_DESCRIPTION = [
|
|
11
10
|
"View an image file and return it to the model.",
|
|
@@ -159,14 +158,14 @@ function buildViewImageUiText(args) {
|
|
|
159
158
|
export function createViewImageToolDefinition(backend) {
|
|
160
159
|
return {
|
|
161
160
|
schema: VIEW_IMAGE_TOOL,
|
|
162
|
-
|
|
163
|
-
async
|
|
164
|
-
return
|
|
161
|
+
describe: (toolCall) => ({ headerTarget: getViewImageDisplayTarget(toolCall.arguments) }),
|
|
162
|
+
async execute(toolCall, context) {
|
|
163
|
+
return executeTool(context, async () => {
|
|
165
164
|
const parsedArgs = parseViewImageArgs(toolCall.arguments);
|
|
166
165
|
const path = parsedArgs.ok ? parsedArgs.data.path : "";
|
|
167
166
|
const headerTarget = getViewImageDisplayTarget(toolCall.arguments);
|
|
168
|
-
const blocked = (reason) => {
|
|
169
|
-
const
|
|
167
|
+
const blocked = (reason, semanticOutcome = "blocked") => {
|
|
168
|
+
const outcome = createTextToolOutcome(reason, semanticOutcome);
|
|
170
169
|
const uiEvent = {
|
|
171
170
|
type: "view_image_blocked",
|
|
172
171
|
toolCallId: toolCall.id,
|
|
@@ -174,7 +173,7 @@ export function createViewImageToolDefinition(backend) {
|
|
|
174
173
|
headerTarget,
|
|
175
174
|
reason,
|
|
176
175
|
};
|
|
177
|
-
return {
|
|
176
|
+
return { content: outcome.content, outcome: outcome.outcome, uiEvent };
|
|
178
177
|
};
|
|
179
178
|
if (!parsedArgs.ok) {
|
|
180
179
|
return blocked(`Invalid arguments: ${parsedArgs.error}`);
|
|
@@ -191,16 +190,12 @@ export function createViewImageToolDefinition(backend) {
|
|
|
191
190
|
const encodedImage = await prepareImageForModel(content, mimeType);
|
|
192
191
|
const data = encodedImage.content.toString("base64");
|
|
193
192
|
const resultText = `Viewed ${resolvedPath} (${encodedImage.mimeType})`;
|
|
194
|
-
const
|
|
195
|
-
role: "toolResult",
|
|
196
|
-
toolCallId: toolCall.id,
|
|
197
|
-
toolName: toolCall.name,
|
|
193
|
+
const outcome = {
|
|
198
194
|
content: [
|
|
199
195
|
{ type: "text", text: resultText },
|
|
200
196
|
{ type: "image", data, mimeType: encodedImage.mimeType },
|
|
201
197
|
],
|
|
202
|
-
|
|
203
|
-
timestamp: Date.now(),
|
|
198
|
+
outcome: "succeeded",
|
|
204
199
|
};
|
|
205
200
|
const uiText = buildViewImageUiText({
|
|
206
201
|
mimeType: encodedImage.mimeType,
|
|
@@ -215,11 +210,11 @@ export function createViewImageToolDefinition(backend) {
|
|
|
215
210
|
bytes: encodedImage.content.byteLength,
|
|
216
211
|
uiText,
|
|
217
212
|
};
|
|
218
|
-
return {
|
|
213
|
+
return { content: outcome.content, outcome: outcome.outcome, uiEvent };
|
|
219
214
|
}
|
|
220
215
|
catch (e) {
|
|
221
216
|
const errorMessage = e instanceof Error ? e.message : String(e);
|
|
222
|
-
return blocked(`Tool view_image failed: ${errorMessage}
|
|
217
|
+
return blocked(`Tool view_image failed: ${errorMessage}`, "failed");
|
|
223
218
|
}
|
|
224
219
|
});
|
|
225
220
|
},
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"view_image.js","sourceRoot":"","sources":["../../../src/core/tools/view_image.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAC/C,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"view_image.js","sourceRoot":"","sources":["../../../src/core/tools/view_image.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAC/C,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAGjD,OAAO,EAEL,qBAAqB,EACrB,WAAW,GAIZ,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAEvD,MAAM,sBAAsB,GAAG;IAC7B,gDAAgD;IAChD,qFAAqF;CACtF,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAEZ,MAAM,2BAA2B,GAAG,iCAAiC,CAAC;AAEtE,MAAM,yBAAyB,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;AACnD,MAAM,0BAA0B,GAAG,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC;AACrD,MAAM,2BAA2B,GAAG,IAAI,CAAC;AACzC,MAAM,0BAA0B,GAAG;IACjC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG;CAChE,CAAC;AACX,MAAM,8BAA8B,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAU,CAAC;AACrE,MAAM,qBAAqB,GAAG,CAAC,YAAY,EAAE,WAAW,EAAE,YAAY,CAAU,CAAC;AAEjF,MAAM,CAAC,MAAM,eAAe,GAAS;IACnC,IAAI,EAAE,oBAAoB;IAC1B,WAAW,EAAE,sBAAsB;IACnC,UAAU,EAAE,IAAI,CAAC,MAAM,CACrB;QACE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,2BAA2B,EAAE,CAAC;KAChE,EACD,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAChC;CACF,CAAC;AAEF,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;CAC/B,CAAC,CAAC;AAEH,SAAS,kBAAkB,CACzB,GAAY;IAEZ,MAAM,MAAM,GAAG,mBAAmB,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAClD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;IAC5D,CAAC;IACD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;AACzC,CAAC;AAED,SAAS,yBAAyB,CAAC,GAAY;IAC7C,MAAM,UAAU,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC;IAC3C,OAAO,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,qBAAqB,CAAC;AACtE,CAAC;AAuBD,SAAS,oBAAoB,CAAC,QAA4B;IACxD,OAAO,QAAQ,CAAC,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,QAA8B,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;AAC3F,CAAC;AAED,SAAS,kBAAkB,CAAC,YAAoB;IAC9C,MAAM,IAAI,GAAG,IAAI,GAAG,CAAS,CAAC,YAAY,CAAC,CAAC,CAAC;IAC7C,KAAK,MAAM,IAAI,IAAI,0BAA0B,EAAE,CAAC;QAC9C,IAAI,IAAI,GAAG,YAAY,EAAE,CAAC;YACxB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,CAAC,GAAG,IAAI,CAAC,CAAC;AACnB,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAqC;IAClE,OAAO,8BAA8B,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;AAClF,CAAC;AAED,SAAS,gBAAgB,CAAC,IAGzB;IACC,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAC1C,MAAM,KAAK,GAAsB,EAAE,CAAC;IAEpC,IAAI,cAAc,KAAK,WAAW,EAAE,CAAC;QACnC,KAAK,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;IACxC,CAAC;SAAM,CAAC;QACN,KAAK,CAAC,IAAI,CAAC,GAAG,qBAAqB,CAAC,cAAc,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,IAAI,cAAc,KAAK,YAAY,EAAE,CAAC;QACpC,KAAK,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QACrE,KAAK,CAAC,IAAI,CAAC,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC,CAAC;IACrD,CAAC;IAED,IAAI,CAAC,QAAQ,IAAI,cAAc,KAAK,YAAY,EAAE,CAAC;QACjD,KAAK,CAAC,IAAI,CAAC,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC,CAAC;IACrD,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,KAAK,UAAU,oBAAoB,CAAC,IAInC;IACC,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;IAC7C,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;QACrC,KAAK,EAAE,YAAY;QACnB,MAAM,EAAE,YAAY;QACpB,GAAG,EAAE,QAAQ;QACb,kBAAkB,EAAE,IAAI;KACzB,CAAC,CAAC;IAEH,IAAI,IAAI,CAAC,QAAQ,KAAK,YAAY,EAAE,CAAC;QACnC,OAAO;YACL,QAAQ,EAAE,YAAY;YACtB,OAAO,EAAE,MAAM,QAAQ,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;SACnE,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,QAAQ,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO;YACL,QAAQ,EAAE,WAAW;YACrB,OAAO,EAAE,MAAM,QAAQ,CAAC,GAAG,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE;SAChE,CAAC;IACJ,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,YAAY;QACtB,OAAO,EAAE,MAAM,QAAQ;aACpB,IAAI,CAAC;YACJ,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC;aACD,QAAQ,EAAE;KACd,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,oBAAoB,CACjC,OAAe,EACf,cAAkC;IAElC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;IACjD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;IAC7B,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;IAE/B,IAAI,CAAC,KAAK,IAAI,CAAC,MAAM,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;IACtD,CAAC;IAED,IACE,KAAK,IAAI,2BAA2B;QACpC,MAAM,IAAI,2BAA2B;QACrC,OAAO,CAAC,UAAU,IAAI,0BAA0B,EAChD,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC;IAC/C,CAAC;IAED,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,2BAA2B,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;IACpF,MAAM,aAAa,GAAG,kBAAkB,CAAC,YAAY,CAAC,CAAC;IACvD,MAAM,WAAW,GAAG,gBAAgB,CAAC;QACnC,cAAc;QACd,QAAQ,EAAE,QAAQ,CAAC,QAAQ,IAAI,KAAK;KACrC,CAAC,CAAC;IAEH,IAAI,QAAkC,CAAC;IAEvC,KAAK,MAAM,YAAY,IAAI,aAAa,EAAE,CAAC;QACzC,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAG,MAAM,oBAAoB,CAAC;gBAC3C,OAAO;gBACP,YAAY,EAAE,YAAY;gBAC1B,IAAI;aACL,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,IAAI,SAAS,CAAC,OAAO,CAAC,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;gBAC5E,QAAQ,GAAG,SAAS,CAAC;YACvB,CAAC;YAED,IAAI,SAAS,CAAC,OAAO,CAAC,UAAU,IAAI,0BAA0B,EAAE,CAAC;gBAC/D,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,eAAe,GAAG,WAAW,CAAC,0BAA0B,CAAC,CAAC;IAChE,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CACb,oCAAoC,eAAe,0BAA0B,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAC1H,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,oCAAoC,eAAe,GAAG,CAAC,CAAC;AAC1E,CAAC;AAED,SAAS,oBAAoB,CAAC,IAA4C;IACxE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IACpC,MAAM,eAAe,GAAG,QAAQ,CAAC,OAAO,EAAE,CAAC;IAC3C,MAAM,SAAS,GAAG,eAAe;QAC/B,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;IAEzB,OAAO;QACL,YAAY,EAAE,EAAE;QAChB,UAAU,EAAE,QAAQ;QACpB,SAAS;KACV,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,OAA6B;IACzE,OAAO;QACL,MAAM,EAAE,eAAe;QACvB,QAAQ,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,yBAAyB,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACzF,KAAK,CAAC,OAAO,CACX,QAAkB,EAClB,OAA6B;YAE7B,OAAO,WAAW,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE;gBACrC,MAAM,UAAU,GAAG,kBAAkB,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBAC1D,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvD,MAAM,YAAY,GAAG,yBAAyB,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBAEnE,MAAM,OAAO,GAAG,CACd,MAAc,EACd,eAAe,GAAoC,SAAS,EACjC,EAAE;oBAC7B,MAAM,OAAO,GAAG,qBAAqB,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;oBAC/D,MAAM,OAAO,GAAiB;wBAC5B,IAAI,EAAE,oBAAoB;wBAC1B,UAAU,EAAE,QAAQ,CAAC,EAAE;wBACvB,IAAI,EAAE,IAAI,IAAI,gBAAgB;wBAC9B,YAAY;wBACZ,MAAM;qBACP,CAAC;oBACF,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;gBACzE,CAAC,CAAC;gBAEF,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,CAAC;oBACnB,OAAO,OAAO,CAAC,sBAAsB,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;gBAC3D,CAAC;gBAED,IAAI,CAAC;oBACH,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,IAAI,EAAE;wBACzE,QAAQ,EAAE,yBAAyB;qBACpC,CAAC,CAAC;oBAEH,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,OAAO,CAAC,CAAC;oBACnD,MAAM,QAAQ,GAAG,QAAQ,EAAE,IAAI,CAAC;oBAChC,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,EAAE,CAAC;wBACpC,OAAO,OAAO,CACZ,gDAAgD,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CACpF,CAAC;oBACJ,CAAC;oBAED,MAAM,YAAY,GAAG,MAAM,oBAAoB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;oBACnE,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;oBACrD,MAAM,UAAU,GAAG,UAAU,YAAY,KAAK,YAAY,CAAC,QAAQ,GAAG,CAAC;oBACvE,MAAM,OAAO,GAAyB;wBACpC,OAAO,EAAE;4BACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE;4BAClC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,CAAC,QAAQ,EAAE;yBACzD;wBACD,OAAO,EAAE,WAAW;qBACrB,CAAC;oBAEF,MAAM,MAAM,GAAG,oBAAoB,CAAC;wBAClC,QAAQ,EAAE,YAAY,CAAC,QAAQ;wBAC/B,QAAQ,EAAE,UAAU;qBACrB,CAAC,CAAC;oBACH,MAAM,OAAO,GAAiB;wBAC5B,IAAI,EAAE,oBAAoB;wBAC1B,UAAU,EAAE,QAAQ,CAAC,EAAE;wBACvB,IAAI,EAAE,YAAY;wBAClB,YAAY,EAAE,YAAY;wBAC1B,QAAQ,EAAE,YAAY,CAAC,QAAQ;wBAC/B,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,UAAU;wBACtC,MAAM;qBACP,CAAC;oBAEF,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;gBACzE,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,YAAY,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;oBAChE,OAAO,OAAO,CAAC,2BAA2B,YAAY,EAAE,EAAE,QAAQ,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -1,9 +1,8 @@
|
|
|
1
1
|
import { Type } from "typebox";
|
|
2
2
|
import { z } from "zod";
|
|
3
|
-
import { createToolError, createToolResult } from "../utils/messages.js";
|
|
4
3
|
import { truncateForTokens } from "../utils/truncate.js";
|
|
5
4
|
import { parseToolArgs } from "../utils/zod.js";
|
|
6
|
-
import {
|
|
5
|
+
import { createTextToolOutcome, executeTool, } from "./registry.js";
|
|
7
6
|
import { buildSubagentUiText, formatSubagentStatusLine } from "./subagent_ui.js";
|
|
8
7
|
import { TOOL_NAME_WAIT_FOR_AGENTS } from "./tool_names.js";
|
|
9
8
|
const WAIT_FOR_AGENTS_DESCRIPTION = [
|
|
@@ -95,15 +94,16 @@ function getWaitForAgentsDisplayTarget(raw) {
|
|
|
95
94
|
const parsedArgs = parseToolArgs(waitArgsSchema, raw);
|
|
96
95
|
return parsedArgs.ok ? formatAgentIdsDisplayTarget(parsedArgs.data.ids) : "(invalid arguments)";
|
|
97
96
|
}
|
|
98
|
-
export function createWaitForAgentsToolDefinition() {
|
|
97
|
+
export function createWaitForAgentsToolDefinition(supervisor) {
|
|
99
98
|
return {
|
|
100
99
|
schema: WAIT_FOR_AGENTS_TOOL,
|
|
101
|
-
|
|
102
|
-
async
|
|
100
|
+
describe: (toolCall) => ({ headerTarget: getWaitForAgentsDisplayTarget(toolCall.arguments) }),
|
|
101
|
+
async execute(toolCall, context) {
|
|
102
|
+
const { signal } = context;
|
|
103
103
|
let ids = [];
|
|
104
104
|
const headerTarget = getWaitForAgentsDisplayTarget(toolCall.arguments);
|
|
105
105
|
const blocked = (reason) => {
|
|
106
|
-
const
|
|
106
|
+
const outcome = createTextToolOutcome(reason, "blocked");
|
|
107
107
|
const uiEvent = {
|
|
108
108
|
type: "wait_for_agents_blocked",
|
|
109
109
|
toolCallId: toolCall.id,
|
|
@@ -111,11 +111,11 @@ export function createWaitForAgentsToolDefinition() {
|
|
|
111
111
|
headerTarget,
|
|
112
112
|
reason,
|
|
113
113
|
};
|
|
114
|
-
return {
|
|
114
|
+
return { content: outcome.content, outcome: outcome.outcome, uiEvent };
|
|
115
115
|
};
|
|
116
116
|
const parsedArgs = parseToolArgs(waitArgsSchema, toolCall.arguments);
|
|
117
117
|
if (!parsedArgs.ok) {
|
|
118
|
-
return
|
|
118
|
+
return executeTool(context, () => blocked(`Invalid arguments: ${parsedArgs.error}`));
|
|
119
119
|
}
|
|
120
120
|
({ ids } = parsedArgs.data);
|
|
121
121
|
const deduped = [];
|
|
@@ -127,66 +127,59 @@ export function createWaitForAgentsToolDefinition() {
|
|
|
127
127
|
}
|
|
128
128
|
}
|
|
129
129
|
const dedupedTarget = formatAgentIdsDisplayTarget(deduped);
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
|
|
141
|
-
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
148
|
-
|
|
149
|
-
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
|
|
159
|
-
|
|
160
|
-
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
175
|
-
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
uiText,
|
|
184
|
-
};
|
|
185
|
-
const toolResult = createToolError(toolCall, reason);
|
|
186
|
-
return { toolResult, uiEvent };
|
|
187
|
-
}
|
|
188
|
-
})(),
|
|
189
|
-
};
|
|
130
|
+
return executeTool(context, async () => {
|
|
131
|
+
try {
|
|
132
|
+
const results = await supervisor.waitForAgents(deduped, signal);
|
|
133
|
+
const resultText = formatWaitOutput(results, deduped);
|
|
134
|
+
const outputText = formatWaitOutput(results, deduped, WAIT_FOR_AGENTS_OUTPUT_MAX_TOKENS);
|
|
135
|
+
const hasFailures = results.some((result) => result.status !== "success");
|
|
136
|
+
const statusText = formatSubagentStatusLine({
|
|
137
|
+
costTotal: getWaitCostTotal(results),
|
|
138
|
+
durationMs: getWaitDurationMs(results),
|
|
139
|
+
});
|
|
140
|
+
const uiText = buildSubagentUiText({
|
|
141
|
+
output: outputText,
|
|
142
|
+
statusText,
|
|
143
|
+
fullText: resultText,
|
|
144
|
+
});
|
|
145
|
+
const uiEvent = {
|
|
146
|
+
type: "wait_for_agents_finished",
|
|
147
|
+
toolCallId: toolCall.id,
|
|
148
|
+
agentIds: deduped,
|
|
149
|
+
headerTarget: dedupedTarget,
|
|
150
|
+
status: hasFailures ? "error" : "success",
|
|
151
|
+
message: hasFailures ? "One or more subagents reported errors." : undefined,
|
|
152
|
+
uiText,
|
|
153
|
+
};
|
|
154
|
+
const outcome = createTextToolOutcome(resultText, hasFailures ? "failed" : "succeeded");
|
|
155
|
+
return { content: outcome.content, outcome: outcome.outcome, uiEvent };
|
|
156
|
+
}
|
|
157
|
+
catch (error) {
|
|
158
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
159
|
+
const reason = message.trim() || "The wait_for_agents request failed.";
|
|
160
|
+
const uiText = buildSubagentUiText({
|
|
161
|
+
output: reason,
|
|
162
|
+
statusText: "error",
|
|
163
|
+
fullText: reason,
|
|
164
|
+
});
|
|
165
|
+
const uiEvent = {
|
|
166
|
+
type: "wait_for_agents_finished",
|
|
167
|
+
toolCallId: toolCall.id,
|
|
168
|
+
agentIds: deduped,
|
|
169
|
+
headerTarget: dedupedTarget,
|
|
170
|
+
status: "error",
|
|
171
|
+
message: reason,
|
|
172
|
+
uiText,
|
|
173
|
+
};
|
|
174
|
+
const outcome = createTextToolOutcome(reason, signal.aborted ? "cancelled" : "failed");
|
|
175
|
+
return { content: outcome.content, outcome: outcome.outcome, uiEvent };
|
|
176
|
+
}
|
|
177
|
+
}, {
|
|
178
|
+
type: "wait_for_agents_started",
|
|
179
|
+
toolCallId: toolCall.id,
|
|
180
|
+
agentIds: deduped,
|
|
181
|
+
headerTarget: dedupedTarget,
|
|
182
|
+
});
|
|
190
183
|
},
|
|
191
184
|
};
|
|
192
185
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"wait_for_agents.js","sourceRoot":"","sources":["../../../src/core/tools/wait_for_agents.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"wait_for_agents.js","sourceRoot":"","sources":["../../../src/core/tools/wait_for_agents.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,OAAO,EAEL,qBAAqB,EACrB,WAAW,GAIZ,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AACjF,OAAO,EAAE,yBAAyB,EAAE,MAAM,iBAAiB,CAAC;AAE5D,MAAM,2BAA2B,GAAG;IAClC,gHAAgH;IAChH,2DAA2D;CAC5D,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAEZ,MAAM,+BAA+B,GAAG,mCAAmC,CAAC;AAC5E,MAAM,iCAAiC,GAAG,GAAG,CAAC;AAE9C,MAAM,CAAC,MAAM,oBAAoB,GAAS;IACxC,IAAI,EAAE,yBAAyB;IAC/B,WAAW,EAAE,2BAA2B;IACxC,UAAU,EAAE,IAAI,CAAC,MAAM,CACrB;QACE,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,+BAA+B,EAAE,CAAC,EAAE;YAC7E,QAAQ,EAAE,CAAC;SACZ,CAAC;KACH,EACD,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAChC;CACF,CAAC;AAEF,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9B,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;CAC9C,CAAC,CAAC;AAEH,SAAS,iBAAiB,CAAC,MAAsB;IAC/C,MAAM,SAAS,GACb,MAAM,CAAC,MAAM,KAAK,SAAS;QACzB,CAAC,CAAC,MAAM,CAAC,KAAK;YACZ,CAAC,CAAC,UAAU,MAAM,CAAC,KAAK,EAAE;YAC1B,CAAC,CAAC,WAAW,MAAM,CAAC,MAAM,EAAE;QAC9B,CAAC,CAAC,SAAS,CAAC;IAChB,MAAM,SAAS,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC;IACjG,OAAO,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,yBAAyB,CAAC,MAAsB;IACvD,MAAM,MAAM,GAAG,KAAK,MAAM,CAAC,EAAE,IAAI,CAAC;IAClC,MAAM,IAAI,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;IACvC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QACjB,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IACD,OAAO,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;AACvC,CAAC;AAED,SAAS,8BAA8B,CAAC,MAAsB,EAAE,SAAiB;IAC/E,MAAM,MAAM,GAAG,KAAK,MAAM,CAAC,EAAE,IAAI,CAAC;IAClC,MAAM,IAAI,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;IACvC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QACjB,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IACD,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IAC7F,MAAM,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACzD,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC;AAChC,CAAC;AAED,SAAS,gBAAgB,CACvB,OAAyB,EACzB,YAAsB,EACtB,oBAA6B;IAE7B,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACvD,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,CAAC;IAC3C,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;IAEvE,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1B,MAAM,CAAC,IAAI,CACT,aAAa,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,oBAAoB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI;YAC9E,yFAAyF,CAC5F,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;QAChC,IAAI,KAAK,GAAG,CAAC;YAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,KAAK,GAAG,oBAAoB;YAChC,CAAC,CAAC,8BAA8B,CAAC,MAAM,EAAE,oBAAoB,CAAC;YAC9D,CAAC,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC;QACtC,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;IACxB,CAAC,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAyB;IAClD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC3C,IAAI,QAAQ,GAAG,MAAM,CAAC,iBAAiB,CAAC;IACxC,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;QAChD,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QACnD,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACxC,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,GAAG,QAAQ,CAAC,CAAC;AACxC,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAyB;IACjD,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;AACpE,CAAC;AAED,SAAS,2BAA2B,CAAC,GAAa;IAChD,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACxB,CAAC;AAED,SAAS,6BAA6B,CAAC,GAAY;IACjD,MAAM,UAAU,GAAG,aAAa,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;IACtD,OAAO,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,2BAA2B,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC;AAClG,CAAC;AAED,MAAM,UAAU,iCAAiC,CAAC,UAA2B;IAC3E,OAAO;QACL,MAAM,EAAE,oBAAoB;QAC5B,QAAQ,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,6BAA6B,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC7F,KAAK,CAAC,OAAO,CACX,QAAkB,EAClB,OAA6B;YAE7B,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;YAC3B,IAAI,GAAG,GAAa,EAAE,CAAC;YACvB,MAAM,YAAY,GAAG,6BAA6B,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,OAAO,GAAG,CAAC,MAAc,EAA6B,EAAE;gBAC5D,MAAM,OAAO,GAAG,qBAAqB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;gBACzD,MAAM,OAAO,GAAiB;oBAC5B,IAAI,EAAE,yBAAyB;oBAC/B,UAAU,EAAE,QAAQ,CAAC,EAAE;oBACvB,QAAQ,EAAE,GAAG;oBACb,YAAY;oBACZ,MAAM;iBACP,CAAC;gBACF,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;YACzE,CAAC,CAAC;YAEF,MAAM,UAAU,GAAG,aAAa,CAAC,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YACrE,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,CAAC;gBACnB,OAAO,WAAW,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,sBAAsB,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YACvF,CAAC;YAED,CAAC,EAAE,GAAG,EAAE,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;YAE5B,MAAM,OAAO,GAAa,EAAE,CAAC;YAC7B,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;YAC/B,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;gBACrB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBAClB,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBACb,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACnB,CAAC;YACH,CAAC;YACD,MAAM,aAAa,GAAG,2BAA2B,CAAC,OAAO,CAAC,CAAC;YAE3D,OAAO,WAAW,CAChB,OAAO,EACP,KAAK,IAAwC,EAAE;gBAC7C,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;oBAChE,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;oBACtD,MAAM,UAAU,GAAG,gBAAgB,CACjC,OAAO,EACP,OAAO,EACP,iCAAiC,CAClC,CAAC;oBACF,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;oBAC1E,MAAM,UAAU,GAAG,wBAAwB,CAAC;wBAC1C,SAAS,EAAE,gBAAgB,CAAC,OAAO,CAAC;wBACpC,UAAU,EAAE,iBAAiB,CAAC,OAAO,CAAC;qBACvC,CAAC,CAAC;oBACH,MAAM,MAAM,GAAG,mBAAmB,CAAC;wBACjC,MAAM,EAAE,UAAU;wBAClB,UAAU;wBACV,QAAQ,EAAE,UAAU;qBACrB,CAAC,CAAC;oBACH,MAAM,OAAO,GAAiB;wBAC5B,IAAI,EAAE,0BAA0B;wBAChC,UAAU,EAAE,QAAQ,CAAC,EAAE;wBACvB,QAAQ,EAAE,OAAO;wBACjB,YAAY,EAAE,aAAa;wBAC3B,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;wBACzC,OAAO,EAAE,WAAW,CAAC,CAAC,CAAC,wCAAwC,CAAC,CAAC,CAAC,SAAS;wBAC3E,MAAM;qBACP,CAAC;oBACF,MAAM,OAAO,GAAG,qBAAqB,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;oBACxF,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;gBACzE,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACvE,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,EAAE,IAAI,qCAAqC,CAAC;oBACvE,MAAM,MAAM,GAAG,mBAAmB,CAAC;wBACjC,MAAM,EAAE,MAAM;wBACd,UAAU,EAAE,OAAO;wBACnB,QAAQ,EAAE,MAAM;qBACjB,CAAC,CAAC;oBACH,MAAM,OAAO,GAAiB;wBAC5B,IAAI,EAAE,0BAA0B;wBAChC,UAAU,EAAE,QAAQ,CAAC,EAAE;wBACvB,QAAQ,EAAE,OAAO;wBACjB,YAAY,EAAE,aAAa;wBAC3B,MAAM,EAAE,OAAO;wBACf,OAAO,EAAE,MAAM;wBACf,MAAM;qBACP,CAAC;oBACF,MAAM,OAAO,GAAG,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;oBACvF,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;gBACzE,CAAC;YACH,CAAC,EACD;gBACE,IAAI,EAAE,yBAAyB;gBAC/B,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,QAAQ,EAAE,OAAO;gBACjB,YAAY,EAAE,aAAa;aAC5B,CACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}
|
package/dist/core/tools/web.js
CHANGED
|
@@ -382,15 +382,15 @@ const defaultDeps = {
|
|
|
382
382
|
createExaClient,
|
|
383
383
|
discover: discoverAgentContent,
|
|
384
384
|
};
|
|
385
|
-
export function createWebToolDefinition(backend, deps = defaultDeps) {
|
|
385
|
+
export function createWebToolDefinition(backend, config, deps = defaultDeps) {
|
|
386
386
|
const timeoutMs = deps.timeoutMs ?? WEB_CODE_MODE_TIMEOUT_MS;
|
|
387
387
|
const implementation = {
|
|
388
388
|
schema: WEB_TOOL,
|
|
389
389
|
outputPolicy: { maxTokens: WEB_CODE_MODE_OUTPUT_TOKENS },
|
|
390
390
|
timeoutMs,
|
|
391
391
|
parseArguments: parseWebArguments,
|
|
392
|
-
execute: async ({ code,
|
|
393
|
-
const apiKey = getExaApiKey(
|
|
392
|
+
execute: async ({ code, signal, backend: executionBackend }) => {
|
|
393
|
+
const apiKey = getExaApiKey(config);
|
|
394
394
|
const exa = apiKey ? deps.createExaClient(apiKey) : undefined;
|
|
395
395
|
return executeWebProgram(code, exa, deps, executionBackend, signal, timeoutMs);
|
|
396
396
|
},
|