@tanstack/ai 0.57.0 → 0.59.0
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/esm/activities/chat/agents/define-agent.d.ts +81 -0
- package/dist/esm/activities/chat/agents/define-agent.js +34 -0
- package/dist/esm/activities/chat/agents/define-agent.js.map +1 -0
- package/dist/esm/activities/chat/agents/route.d.ts +53 -0
- package/dist/esm/activities/chat/agents/route.js +59 -0
- package/dist/esm/activities/chat/agents/route.js.map +1 -0
- package/dist/esm/activities/chat/agents/spawn.d.ts +124 -0
- package/dist/esm/activities/chat/agents/spawn.js +490 -0
- package/dist/esm/activities/chat/agents/spawn.js.map +1 -0
- package/dist/esm/activities/chat/agents/turn.d.ts +36 -0
- package/dist/esm/activities/chat/agents/turn.js +78 -0
- package/dist/esm/activities/chat/agents/turn.js.map +1 -0
- package/dist/esm/activities/chat/index.d.ts +13 -3
- package/dist/esm/activities/chat/index.js +345 -19
- package/dist/esm/activities/chat/index.js.map +1 -1
- package/dist/esm/activities/chat/messages.d.ts +7 -1
- package/dist/esm/activities/chat/messages.js +94 -18
- package/dist/esm/activities/chat/messages.js.map +1 -1
- package/dist/esm/activities/chat/middleware/run-store.d.ts +43 -7
- package/dist/esm/activities/chat/middleware/run-store.js +8 -1
- package/dist/esm/activities/chat/middleware/run-store.js.map +1 -1
- package/dist/esm/activities/chat/middleware/types.d.ts +50 -3
- package/dist/esm/activities/chat/middleware/types.js.map +1 -1
- package/dist/esm/activities/chat/stream/processor.d.ts +46 -0
- package/dist/esm/activities/chat/stream/processor.js +280 -7
- package/dist/esm/activities/chat/stream/processor.js.map +1 -1
- package/dist/esm/activities/chat/tools/tool-calls.d.ts +17 -3
- package/dist/esm/activities/chat/tools/tool-calls.js +56 -7
- package/dist/esm/activities/chat/tools/tool-calls.js.map +1 -1
- package/dist/esm/activities/generateAudio/index.js +1 -1
- package/dist/esm/activities/generateImage/index.js +1 -1
- package/dist/esm/activities/generateLiveVideo/index.js +1 -1
- package/dist/esm/activities/generateSpeech/index.js +1 -1
- package/dist/esm/activities/generateTranscription/index.js +1 -1
- package/dist/esm/activities/generateVoice/index.js +1 -1
- package/dist/esm/activities/generateWorld/index.js +1 -1
- package/dist/esm/activities/index.d.ts +3 -0
- package/dist/esm/activities/index.js +5 -3
- package/dist/esm/activities/summarize/index.js +1 -1
- package/dist/esm/client.d.ts +5 -36
- package/dist/esm/client.js +4 -37
- package/dist/esm/client.js.map +1 -1
- package/dist/esm/index.d.ts +5 -0
- package/dist/esm/index.js +7 -4
- package/dist/esm/middlewares/content-guard.js.map +1 -1
- package/dist/esm/stream-to-response.js +12 -6
- package/dist/esm/stream-to-response.js.map +1 -1
- package/dist/esm/strip-to-spec-middleware.js +2 -1
- package/dist/esm/strip-to-spec-middleware.js.map +1 -1
- package/dist/esm/types.d.ts +128 -89
- package/dist/esm/utilities/adapter-yield-chunk.d.ts +5 -1
- package/dist/esm/utilities/ag-ui-usage.d.ts +9 -9
- package/dist/esm/utilities/ag-ui-usage.js +66 -3
- package/dist/esm/utilities/ag-ui-usage.js.map +1 -1
- package/dist/esm/utilities/ag-ui-wire.js +56 -0
- package/dist/esm/utilities/ag-ui-wire.js.map +1 -1
- package/dist/esm/utilities/durability-batch.d.ts +8 -0
- package/dist/esm/utilities/durability-batch.js +45 -0
- package/dist/esm/utilities/durability-batch.js.map +1 -0
- package/dist/esm/utilities/normalize-stream-chunk.js +7 -2
- package/dist/esm/utilities/normalize-stream-chunk.js.map +1 -1
- package/dist/esm/utilities/spec-event-keys.js +13 -8
- package/dist/esm/utilities/spec-event-keys.js.map +1 -1
- package/dist/esm/utilities/subagent-wire.d.ts +36 -0
- package/dist/esm/utilities/subagent-wire.js +131 -0
- package/dist/esm/utilities/subagent-wire.js.map +1 -0
- package/package.json +4 -4
- package/skills/ai-core/adapter-configuration/references/grok-adapter.md +1 -1
- package/skills/ai-core/media-generation/SKILL.md +2 -2
- package/skills/ai-core/middleware/SKILL.md +7 -4
- package/src/activities/chat/agents/define-agent.ts +121 -0
- package/src/activities/chat/agents/route.ts +115 -0
- package/src/activities/chat/agents/spawn.ts +806 -0
- package/src/activities/chat/agents/turn.ts +151 -0
- package/src/activities/chat/index.ts +540 -16
- package/src/activities/chat/messages.ts +125 -15
- package/src/activities/chat/middleware/run-store.ts +56 -7
- package/src/activities/chat/middleware/types.ts +55 -2
- package/src/activities/chat/stream/processor.ts +428 -8
- package/src/activities/chat/tools/tool-calls.ts +92 -16
- package/src/activities/index.ts +15 -0
- package/src/client.ts +22 -35
- package/src/index.ts +23 -0
- package/src/middlewares/content-guard.ts +7 -5
- package/src/stream-to-response.ts +16 -6
- package/src/strip-to-spec-middleware.ts +2 -1
- package/src/types.ts +177 -95
- package/src/utilities/adapter-yield-chunk.ts +10 -2
- package/src/utilities/ag-ui-usage.test.ts +38 -0
- package/src/utilities/ag-ui-usage.ts +98 -11
- package/src/utilities/ag-ui-wire.ts +74 -0
- package/src/utilities/durability-batch.ts +48 -0
- package/src/utilities/normalize-stream-chunk.ts +10 -2
- package/src/utilities/spec-event-keys.ts +34 -7
- package/src/utilities/subagent-wire.ts +184 -0
|
@@ -0,0 +1,490 @@
|
|
|
1
|
+
import { EventType } from "../../../types.js";
|
|
2
|
+
import { addTokenUsage, isTanstackUsage, rebuildTokenUsage, toSpecTokenUsage } from "../../../utilities/ag-ui-usage.js";
|
|
3
|
+
import { tanstackMetadata, withTanstackMetadata } from "../../../utilities/merge-metadata.js";
|
|
4
|
+
import { INTERRUPT_BINDING_METADATA_KEY } from "../../../interrupt-resume.js";
|
|
5
|
+
import { EMIT_STREAM_CHUNK, SUBAGENT_TOOL } from "../tools/tool-calls.js";
|
|
6
|
+
//#region src/activities/chat/agents/spawn.ts
|
|
7
|
+
var SUBAGENT_STARTED = EventType.SUBAGENT_STARTED;
|
|
8
|
+
var SUBAGENT_FINISHED = EventType.SUBAGENT_FINISHED;
|
|
9
|
+
var SUBAGENT_ERROR = EventType.SUBAGENT_ERROR;
|
|
10
|
+
function createSubagentSink() {
|
|
11
|
+
return {
|
|
12
|
+
interrupts: [],
|
|
13
|
+
usage: []
|
|
14
|
+
};
|
|
15
|
+
}
|
|
16
|
+
function createSubagentId() {
|
|
17
|
+
return `subagent-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
|
|
18
|
+
}
|
|
19
|
+
function childRunId(parentRunId, subagentRunId) {
|
|
20
|
+
return `${parentRunId}:${subagentRunId}`;
|
|
21
|
+
}
|
|
22
|
+
/**
|
|
23
|
+
* Bind child interrupts to the parent run. The client resumes the parent run,
|
|
24
|
+
* so each binding must name that run. The resumed child then validates with
|
|
25
|
+
* the parent's interrupted run id.
|
|
26
|
+
*/
|
|
27
|
+
function rebindInterrupts(interrupts, runId) {
|
|
28
|
+
return interrupts.map((interrupt) => {
|
|
29
|
+
const binding = interrupt.metadata?.[INTERRUPT_BINDING_METADATA_KEY];
|
|
30
|
+
if (typeof binding !== "object" || binding === null) return interrupt;
|
|
31
|
+
return {
|
|
32
|
+
...interrupt,
|
|
33
|
+
metadata: {
|
|
34
|
+
...interrupt.metadata,
|
|
35
|
+
[INTERRUPT_BINDING_METADATA_KEY]: {
|
|
36
|
+
...binding,
|
|
37
|
+
interruptedRunId: runId,
|
|
38
|
+
generation: 0
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
};
|
|
42
|
+
});
|
|
43
|
+
}
|
|
44
|
+
function createAbortError() {
|
|
45
|
+
const error = /* @__PURE__ */ new Error("Aborted");
|
|
46
|
+
error.name = "AbortError";
|
|
47
|
+
return error;
|
|
48
|
+
}
|
|
49
|
+
function isAbortError(error, signal) {
|
|
50
|
+
if (signal?.aborted) return true;
|
|
51
|
+
return error instanceof Error && (error.name === "AbortError" || error.message === "Aborted");
|
|
52
|
+
}
|
|
53
|
+
function stoppedEvent(subagentRunId) {
|
|
54
|
+
return {
|
|
55
|
+
type: SUBAGENT_ERROR,
|
|
56
|
+
subagentRunId,
|
|
57
|
+
message: "Stopped",
|
|
58
|
+
timestamp: Date.now()
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
function childThreadId(sandbox, parentThreadId, name) {
|
|
62
|
+
return sandbox === "inherit" ? parentThreadId : `${parentThreadId}:${name}`;
|
|
63
|
+
}
|
|
64
|
+
function linkAbort(parent) {
|
|
65
|
+
const controller = new AbortController();
|
|
66
|
+
if (!parent) return {
|
|
67
|
+
controller,
|
|
68
|
+
dispose: () => {}
|
|
69
|
+
};
|
|
70
|
+
if (parent.aborted) {
|
|
71
|
+
controller.abort();
|
|
72
|
+
return {
|
|
73
|
+
controller,
|
|
74
|
+
dispose: () => {}
|
|
75
|
+
};
|
|
76
|
+
}
|
|
77
|
+
const onAbort = () => controller.abort();
|
|
78
|
+
parent.addEventListener("abort", onAbort, { once: true });
|
|
79
|
+
return {
|
|
80
|
+
controller,
|
|
81
|
+
dispose: () => parent.removeEventListener("abort", onAbort)
|
|
82
|
+
};
|
|
83
|
+
}
|
|
84
|
+
function orAbort(promise, signal) {
|
|
85
|
+
if (!signal) return promise;
|
|
86
|
+
if (signal.aborted) return Promise.reject(createAbortError());
|
|
87
|
+
return new Promise((resolve, reject) => {
|
|
88
|
+
const onAbort = () => reject(createAbortError());
|
|
89
|
+
signal.addEventListener("abort", onAbort, { once: true });
|
|
90
|
+
promise.then((value) => {
|
|
91
|
+
signal.removeEventListener("abort", onAbort);
|
|
92
|
+
resolve(value);
|
|
93
|
+
}, (error) => {
|
|
94
|
+
signal.removeEventListener("abort", onAbort);
|
|
95
|
+
reject(error);
|
|
96
|
+
});
|
|
97
|
+
});
|
|
98
|
+
}
|
|
99
|
+
function agentByName(agents, name) {
|
|
100
|
+
const agent = agents.find((entry) => entry.name === name);
|
|
101
|
+
if (!agent) throw new Error(`Unknown subagent: ${name}`);
|
|
102
|
+
return agent;
|
|
103
|
+
}
|
|
104
|
+
function openAgentStream(entry, bag, ctx, sink, parentToolCallId) {
|
|
105
|
+
const agent = agentByName(bag.agents, entry.name);
|
|
106
|
+
const resume = entry.resume;
|
|
107
|
+
const subagentRunId = resume?.subagentRunId ?? createSubagentId();
|
|
108
|
+
if (resume !== void 0 && ctx.interruptedRunId === void 0) throw new Error(`Subagent "${entry.name}" has interrupt answers, but the run has no parentRunId. Pass the interrupted run id as parentRunId.`);
|
|
109
|
+
const resumed = resume !== void 0 ? {
|
|
110
|
+
messages: [...ctx.messages, ...resume.messages],
|
|
111
|
+
parentRunId: ctx.interruptedRunId,
|
|
112
|
+
resume: resume.entries
|
|
113
|
+
} : void 0;
|
|
114
|
+
return spawnAgentStream(agent, {
|
|
115
|
+
messages: resumed?.messages ?? ctx.messages,
|
|
116
|
+
...ctx.abortSignal ? { abortSignal: ctx.abortSignal } : {},
|
|
117
|
+
threadId: childThreadId(bag.sandbox, ctx.threadId, entry.name),
|
|
118
|
+
runId: childRunId(ctx.parentRunId, subagentRunId),
|
|
119
|
+
parentRunId: resumed?.parentRunId ?? ctx.parentRunId,
|
|
120
|
+
subagentRunId,
|
|
121
|
+
...resumed ? { resume: resumed.resume } : {}
|
|
122
|
+
}, sink, parentToolCallId);
|
|
123
|
+
}
|
|
124
|
+
var ROUTER_PICK_ERROR = "subagents.router must return main, a name, a list of names, { names, order }, or { steps }.";
|
|
125
|
+
function assertOrder(order) {
|
|
126
|
+
if (order !== void 0 && order !== "parallel" && order !== "sequence") throw new Error("subagents.router order must be parallel or sequence.");
|
|
127
|
+
}
|
|
128
|
+
function normalizeNames(names, agents) {
|
|
129
|
+
if (names.length === 0) throw new Error(ROUTER_PICK_ERROR);
|
|
130
|
+
const hasMain = names.includes("main");
|
|
131
|
+
if (hasMain && names.length > 1) throw new Error("Do not mix main into a subagent list.");
|
|
132
|
+
if (hasMain) return ["main"];
|
|
133
|
+
for (const name of names) agentByName(agents, name);
|
|
134
|
+
return [...names];
|
|
135
|
+
}
|
|
136
|
+
function isStringList(pick) {
|
|
137
|
+
return Array.isArray(pick);
|
|
138
|
+
}
|
|
139
|
+
function normalizeRouterPick(pick, agents) {
|
|
140
|
+
if (pick === "main" || typeof pick === "string") return { steps: [{ names: normalizeNames([pick], agents) }] };
|
|
141
|
+
if (isStringList(pick)) return { steps: [{ names: normalizeNames(pick, agents) }] };
|
|
142
|
+
if ("steps" in pick) {
|
|
143
|
+
if (pick.steps.length === 0) throw new Error(ROUTER_PICK_ERROR);
|
|
144
|
+
const steps = pick.steps.map((step) => {
|
|
145
|
+
assertOrder(step.order);
|
|
146
|
+
const names = normalizeNames(step.names, agents);
|
|
147
|
+
return step.order === void 0 ? { names } : {
|
|
148
|
+
names,
|
|
149
|
+
order: step.order
|
|
150
|
+
};
|
|
151
|
+
});
|
|
152
|
+
const flat = steps.flatMap((step) => step.names);
|
|
153
|
+
if (flat.includes("main") && flat.length > 1) throw new Error("Do not mix main into a subagent list.");
|
|
154
|
+
return { steps };
|
|
155
|
+
}
|
|
156
|
+
assertOrder(pick.order);
|
|
157
|
+
const names = normalizeNames(pick.names, agents);
|
|
158
|
+
return { steps: [pick.order === void 0 ? { names } : {
|
|
159
|
+
names,
|
|
160
|
+
order: pick.order
|
|
161
|
+
}] };
|
|
162
|
+
}
|
|
163
|
+
/**
|
|
164
|
+
* Tag a child chunk with its subagent. A chunk that a nested child already
|
|
165
|
+
* tagged keeps its own id, and a nested child's start names this child as
|
|
166
|
+
* its parent.
|
|
167
|
+
*/
|
|
168
|
+
function attributeChunk(chunk, subagentRunId) {
|
|
169
|
+
if (chunk.type === SUBAGENT_STARTED) return chunk.parentSubagentRunId !== void 0 ? chunk : {
|
|
170
|
+
...chunk,
|
|
171
|
+
parentSubagentRunId: subagentRunId
|
|
172
|
+
};
|
|
173
|
+
if (chunk.type === SUBAGENT_FINISHED || chunk.type === SUBAGENT_ERROR) return chunk;
|
|
174
|
+
if ("subagentRunId" in chunk && typeof chunk.subagentRunId === "string") return chunk;
|
|
175
|
+
return {
|
|
176
|
+
...chunk,
|
|
177
|
+
subagentRunId
|
|
178
|
+
};
|
|
179
|
+
}
|
|
180
|
+
function runUsage(chunk) {
|
|
181
|
+
if (chunk?.type !== EventType.RUN_FINISHED) return [];
|
|
182
|
+
if (Array.isArray(chunk.usage)) return chunk.usage;
|
|
183
|
+
return isTanstackUsage(chunk.usage) ? toSpecTokenUsage(chunk.usage).usage : [];
|
|
184
|
+
}
|
|
185
|
+
/** The full usage of a run: token counts plus cost and the other fields. */
|
|
186
|
+
function fullUsage(chunk) {
|
|
187
|
+
if (chunk?.type !== EventType.RUN_FINISHED) return void 0;
|
|
188
|
+
return rebuildTokenUsage(chunk.usage, tanstackMetadata(chunk)?.usage);
|
|
189
|
+
}
|
|
190
|
+
/** Add a finished child run's usage to the sink. */
|
|
191
|
+
function collectUsage(sink, finished) {
|
|
192
|
+
sink.usage.push(...runUsage(finished));
|
|
193
|
+
const full = fullUsage(finished);
|
|
194
|
+
if (full) sink.total = sink.total ? addTokenUsage(sink.total, full) : full;
|
|
195
|
+
}
|
|
196
|
+
/**
|
|
197
|
+
* Put the children's usage on a parent terminal. `usage[]` keeps one entry per
|
|
198
|
+
* model call. `metadata.tanstack.usage` holds the summed cost and the other
|
|
199
|
+
* TanStack fields, so `fromSpecTokenUsage` reads the full total. Empties the
|
|
200
|
+
* sink, so the next parent terminal does not count it again.
|
|
201
|
+
*
|
|
202
|
+
* `RUN_ERROR` is accepted too: a turn that failed still spent whatever its
|
|
203
|
+
* children spent. Such a chunk carries no usage of its own, so `runUsage` and
|
|
204
|
+
* `fullUsage` return empty for it and the children's total stands alone.
|
|
205
|
+
*/
|
|
206
|
+
function withChildUsage(chunk, sink) {
|
|
207
|
+
if (sink.usage.length === 0 && !sink.total) return chunk;
|
|
208
|
+
const own = fullUsage(chunk);
|
|
209
|
+
const total = own && sink.total ? addTokenUsage(own, sink.total) : own ?? sink.total;
|
|
210
|
+
const usage = [...runUsage(chunk), ...sink.usage.splice(0)];
|
|
211
|
+
sink.total = void 0;
|
|
212
|
+
const leftover = total ? toSpecTokenUsage(total).leftover : void 0;
|
|
213
|
+
const next = {
|
|
214
|
+
...chunk,
|
|
215
|
+
usage
|
|
216
|
+
};
|
|
217
|
+
if (!leftover) return next;
|
|
218
|
+
return withTanstackMetadata(next, { usage: leftover });
|
|
219
|
+
}
|
|
220
|
+
async function* spawnAgentStream(agent, ctx, sink, parentToolCallId) {
|
|
221
|
+
const id = ctx.subagentRunId;
|
|
222
|
+
yield {
|
|
223
|
+
type: SUBAGENT_STARTED,
|
|
224
|
+
subagentRunId: id,
|
|
225
|
+
name: agent.name,
|
|
226
|
+
description: agent.description,
|
|
227
|
+
...ctx.parentSubagentRunId !== void 0 ? { parentSubagentRunId: ctx.parentSubagentRunId } : {},
|
|
228
|
+
...parentToolCallId !== void 0 ? { parentToolCallId } : {},
|
|
229
|
+
timestamp: Date.now()
|
|
230
|
+
};
|
|
231
|
+
let iterator;
|
|
232
|
+
let finished;
|
|
233
|
+
try {
|
|
234
|
+
if (ctx.abortSignal?.aborted) {
|
|
235
|
+
yield stoppedEvent(id);
|
|
236
|
+
return;
|
|
237
|
+
}
|
|
238
|
+
iterator = (await orAbort(Promise.resolve(agent.run(ctx)), ctx.abortSignal))[Symbol.asyncIterator]();
|
|
239
|
+
while (true) {
|
|
240
|
+
if (ctx.abortSignal?.aborted) {
|
|
241
|
+
yield stoppedEvent(id);
|
|
242
|
+
return;
|
|
243
|
+
}
|
|
244
|
+
const result = await orAbort(iterator.next(), ctx.abortSignal);
|
|
245
|
+
if (result.done) break;
|
|
246
|
+
const chunk = result.value;
|
|
247
|
+
if (chunk.type === EventType.RUN_STARTED || chunk.type === EventType.MESSAGES_SNAPSHOT) continue;
|
|
248
|
+
if (chunk.type === EventType.RUN_FINISHED) {
|
|
249
|
+
if (sink) collectUsage(sink, chunk);
|
|
250
|
+
finished = chunk;
|
|
251
|
+
continue;
|
|
252
|
+
}
|
|
253
|
+
if (chunk.type === EventType.RUN_ERROR) {
|
|
254
|
+
yield {
|
|
255
|
+
type: SUBAGENT_ERROR,
|
|
256
|
+
subagentRunId: id,
|
|
257
|
+
message: chunk.message || "Subagent failed",
|
|
258
|
+
...chunk.code ? { code: chunk.code } : {},
|
|
259
|
+
timestamp: Date.now()
|
|
260
|
+
};
|
|
261
|
+
return;
|
|
262
|
+
}
|
|
263
|
+
yield attributeChunk(chunk, id);
|
|
264
|
+
}
|
|
265
|
+
if (ctx.abortSignal?.aborted) {
|
|
266
|
+
yield stoppedEvent(id);
|
|
267
|
+
return;
|
|
268
|
+
}
|
|
269
|
+
const outcome = finished?.type === EventType.RUN_FINISHED ? finished.outcome : void 0;
|
|
270
|
+
if (outcome?.type === "cancelled") {
|
|
271
|
+
yield stoppedEvent(id);
|
|
272
|
+
return;
|
|
273
|
+
}
|
|
274
|
+
if (outcome?.type === "interrupt") {
|
|
275
|
+
const interrupts = outcome.interrupts.map((interrupt) => interrupt.subagentRunId ? interrupt : {
|
|
276
|
+
...interrupt,
|
|
277
|
+
subagentRunId: id
|
|
278
|
+
});
|
|
279
|
+
sink?.interrupts.push(...interrupts);
|
|
280
|
+
yield {
|
|
281
|
+
type: SUBAGENT_FINISHED,
|
|
282
|
+
subagentRunId: id,
|
|
283
|
+
outcome: {
|
|
284
|
+
type: "suspended",
|
|
285
|
+
interruptIds: interrupts.filter((interrupt) => interrupt.subagentRunId === id).map((interrupt) => interrupt.id)
|
|
286
|
+
},
|
|
287
|
+
timestamp: Date.now()
|
|
288
|
+
};
|
|
289
|
+
return;
|
|
290
|
+
}
|
|
291
|
+
const result = finished?.type === EventType.RUN_FINISHED ? finished.result : void 0;
|
|
292
|
+
yield {
|
|
293
|
+
type: SUBAGENT_FINISHED,
|
|
294
|
+
subagentRunId: id,
|
|
295
|
+
...result !== void 0 ? { result } : {},
|
|
296
|
+
timestamp: Date.now()
|
|
297
|
+
};
|
|
298
|
+
} catch (error) {
|
|
299
|
+
yield {
|
|
300
|
+
type: SUBAGENT_ERROR,
|
|
301
|
+
subagentRunId: id,
|
|
302
|
+
message: isAbortError(error, ctx.abortSignal) ? "Stopped" : error instanceof Error ? error.message : String(error),
|
|
303
|
+
timestamp: Date.now()
|
|
304
|
+
};
|
|
305
|
+
} finally {
|
|
306
|
+
try {
|
|
307
|
+
await iterator?.return?.();
|
|
308
|
+
} catch {}
|
|
309
|
+
}
|
|
310
|
+
}
|
|
311
|
+
async function* mergeAgentStreams(streams) {
|
|
312
|
+
const readers = streams.map((stream) => {
|
|
313
|
+
const iterator = stream[Symbol.asyncIterator]();
|
|
314
|
+
return {
|
|
315
|
+
iterator,
|
|
316
|
+
next: iterator.next()
|
|
317
|
+
};
|
|
318
|
+
});
|
|
319
|
+
try {
|
|
320
|
+
while (readers.length > 0) {
|
|
321
|
+
const indexed = readers.map((reader, index) => reader.next.then((result) => ({
|
|
322
|
+
index,
|
|
323
|
+
result,
|
|
324
|
+
reader
|
|
325
|
+
})));
|
|
326
|
+
const winner = await Promise.race(indexed);
|
|
327
|
+
if (winner.result.done) {
|
|
328
|
+
readers.splice(winner.index, 1);
|
|
329
|
+
continue;
|
|
330
|
+
}
|
|
331
|
+
yield winner.result.value;
|
|
332
|
+
winner.reader.next = winner.reader.iterator.next();
|
|
333
|
+
}
|
|
334
|
+
} finally {
|
|
335
|
+
for (const reader of readers) reader.iterator.return?.().catch(() => {});
|
|
336
|
+
}
|
|
337
|
+
}
|
|
338
|
+
/** True when a child in these chunks failed or stopped for outside input. */
|
|
339
|
+
function stopsSequence(chunks, id) {
|
|
340
|
+
return chunks.some((chunk) => chunk.type === SUBAGENT_ERROR && chunk.subagentRunId === id || chunk.type === SUBAGENT_FINISHED && chunk.subagentRunId === id && chunk.outcome?.type === "suspended");
|
|
341
|
+
}
|
|
342
|
+
async function* spawnNamedAgents(entries, bag, ctx, sink) {
|
|
343
|
+
if (bag.sandbox === "inherit" && entries.length > 1) throw new Error("subagents.sandbox 'inherit' cannot start two children in one turn");
|
|
344
|
+
const group = linkAbort(ctx.abortSignal);
|
|
345
|
+
const groupCtx = {
|
|
346
|
+
...ctx,
|
|
347
|
+
abortSignal: group.controller.signal
|
|
348
|
+
};
|
|
349
|
+
try {
|
|
350
|
+
if (bag.order === "sequence") {
|
|
351
|
+
let messages = ctx.messages;
|
|
352
|
+
for (const entry of entries) {
|
|
353
|
+
const chunks = [];
|
|
354
|
+
let id;
|
|
355
|
+
for await (const chunk of openAgentStream(entry, bag, {
|
|
356
|
+
...groupCtx,
|
|
357
|
+
messages
|
|
358
|
+
}, sink)) {
|
|
359
|
+
if (chunk.type === SUBAGENT_STARTED && id === void 0) id = chunk.subagentRunId;
|
|
360
|
+
chunks.push(chunk);
|
|
361
|
+
yield chunk;
|
|
362
|
+
}
|
|
363
|
+
if (id !== void 0 && stopsSequence(chunks, id)) return;
|
|
364
|
+
const text = [entry.resume?.text, collectNamedText(chunks, [entry.name])].filter((part) => part !== void 0 && part !== "").join("");
|
|
365
|
+
if (text) messages = [...messages, {
|
|
366
|
+
role: "assistant",
|
|
367
|
+
content: text
|
|
368
|
+
}];
|
|
369
|
+
}
|
|
370
|
+
return;
|
|
371
|
+
}
|
|
372
|
+
const streams = entries.map((entry) => openAgentStream(entry, bag, groupCtx, sink));
|
|
373
|
+
const onlyStream = streams.length === 1 ? streams[0] : void 0;
|
|
374
|
+
if (onlyStream) {
|
|
375
|
+
yield* onlyStream;
|
|
376
|
+
return;
|
|
377
|
+
}
|
|
378
|
+
yield* mergeAgentStreams(streams);
|
|
379
|
+
} finally {
|
|
380
|
+
group.controller.abort();
|
|
381
|
+
group.dispose();
|
|
382
|
+
}
|
|
383
|
+
}
|
|
384
|
+
/**
|
|
385
|
+
* Text of the named direct children, in `names` order. Text from nested
|
|
386
|
+
* children stays out: their chunks carry their own id.
|
|
387
|
+
*/
|
|
388
|
+
function collectNamedText(chunks, names) {
|
|
389
|
+
const nameByRunId = /* @__PURE__ */ new Map();
|
|
390
|
+
const textByName = /* @__PURE__ */ new Map();
|
|
391
|
+
for (const chunk of chunks) {
|
|
392
|
+
if (chunk.type === SUBAGENT_STARTED) {
|
|
393
|
+
if (chunk.parentSubagentRunId === void 0) nameByRunId.set(chunk.subagentRunId, chunk.name);
|
|
394
|
+
continue;
|
|
395
|
+
}
|
|
396
|
+
if (chunk.type !== EventType.TEXT_MESSAGE_CONTENT) continue;
|
|
397
|
+
if (!("subagentRunId" in chunk) || typeof chunk.subagentRunId !== "string") continue;
|
|
398
|
+
const name = nameByRunId.get(chunk.subagentRunId);
|
|
399
|
+
if (!name) continue;
|
|
400
|
+
textByName.set(name, `${textByName.get(name) ?? ""}${chunk.delta}`);
|
|
401
|
+
}
|
|
402
|
+
return names.map((name) => textByName.get(name)?.trim() ?? "").filter((text) => text.length > 0).join("\n\n");
|
|
403
|
+
}
|
|
404
|
+
/**
|
|
405
|
+
* The parent conversation up to the message that carries this tool call, with
|
|
406
|
+
* that message's tool calls removed. Its string text stays; array content is
|
|
407
|
+
* dropped.
|
|
408
|
+
*/
|
|
409
|
+
function messagesBeforeCall(messages, toolCallId) {
|
|
410
|
+
const index = messages.findIndex((message) => message.toolCalls?.some((call) => call.id === toolCallId));
|
|
411
|
+
if (index === -1) return [...messages];
|
|
412
|
+
const host = messages[index];
|
|
413
|
+
const kept = messages.slice(0, index);
|
|
414
|
+
if (host && typeof host.content === "string" && host.content !== "") {
|
|
415
|
+
const { toolCalls: _calls, ...text } = host;
|
|
416
|
+
kept.push(text);
|
|
417
|
+
}
|
|
418
|
+
return kept;
|
|
419
|
+
}
|
|
420
|
+
/**
|
|
421
|
+
* Record the parent messages when the model calls a subagent tool, so the
|
|
422
|
+
* child reads the conversation as it is at that call.
|
|
423
|
+
*/
|
|
424
|
+
function subagentCallMessages(names) {
|
|
425
|
+
const byCall = /* @__PURE__ */ new Map();
|
|
426
|
+
return {
|
|
427
|
+
middleware: {
|
|
428
|
+
name: "subagent-call-messages",
|
|
429
|
+
onBeforeToolCall(ctx, hook) {
|
|
430
|
+
if (!names.has(hook.toolName)) return void 0;
|
|
431
|
+
byCall.set(hook.toolCallId, messagesBeforeCall(ctx.messages, hook.toolCallId));
|
|
432
|
+
}
|
|
433
|
+
},
|
|
434
|
+
messagesFor: (toolCallId) => toolCallId === void 0 ? void 0 : byCall.get(toolCallId)
|
|
435
|
+
};
|
|
436
|
+
}
|
|
437
|
+
function createSyntheticSubagentTools(bag, parent) {
|
|
438
|
+
return bag.agents.map((agent) => ({
|
|
439
|
+
name: agent.name,
|
|
440
|
+
description: agent.description,
|
|
441
|
+
[SUBAGENT_TOOL]: true,
|
|
442
|
+
execute: async (_input, context) => {
|
|
443
|
+
const toolContext = context;
|
|
444
|
+
const toolCallId = toolContext?.toolCallId;
|
|
445
|
+
const suspended = parent.turn?.children.find((child) => child.status === "suspended" && child.parentToolCallId !== void 0 && child.parentToolCallId === toolCallId);
|
|
446
|
+
const entry = suspended ? {
|
|
447
|
+
name: agent.name,
|
|
448
|
+
resume: {
|
|
449
|
+
subagentRunId: suspended.subagentRunId,
|
|
450
|
+
messages: suspended.messages,
|
|
451
|
+
entries: suspended.resume,
|
|
452
|
+
text: suspended.text
|
|
453
|
+
}
|
|
454
|
+
} : { name: agent.name };
|
|
455
|
+
const sink = createSubagentSink();
|
|
456
|
+
const link = linkAbort(parent.abortSignal);
|
|
457
|
+
let subagentRunId = suspended?.subagentRunId ?? "";
|
|
458
|
+
let text = suspended?.text ?? "";
|
|
459
|
+
let error;
|
|
460
|
+
try {
|
|
461
|
+
for await (const chunk of openAgentStream(entry, bag, {
|
|
462
|
+
messages: parent.messagesFor?.(toolCallId) ?? parent.messages,
|
|
463
|
+
abortSignal: link.controller.signal,
|
|
464
|
+
threadId: parent.threadId,
|
|
465
|
+
parentRunId: parent.runId,
|
|
466
|
+
...parent.interruptedRunId !== void 0 ? { interruptedRunId: parent.interruptedRunId } : {}
|
|
467
|
+
}, sink, toolCallId)) {
|
|
468
|
+
if (chunk.type === SUBAGENT_STARTED && subagentRunId === "") subagentRunId = chunk.subagentRunId;
|
|
469
|
+
if (chunk.type === EventType.TEXT_MESSAGE_CONTENT && "subagentRunId" in chunk && chunk.subagentRunId === subagentRunId) text += chunk.delta;
|
|
470
|
+
if (chunk.type === SUBAGENT_ERROR && chunk.subagentRunId === subagentRunId) error = chunk.message;
|
|
471
|
+
toolContext?.[EMIT_STREAM_CHUNK]?.(chunk);
|
|
472
|
+
}
|
|
473
|
+
} finally {
|
|
474
|
+
link.dispose();
|
|
475
|
+
}
|
|
476
|
+
parent.sink.usage.push(...sink.usage);
|
|
477
|
+
if (sink.total) parent.sink.total = parent.sink.total ? addTokenUsage(parent.sink.total, sink.total) : sink.total;
|
|
478
|
+
return {
|
|
479
|
+
subagentRunId,
|
|
480
|
+
text,
|
|
481
|
+
...error !== void 0 ? { error } : {},
|
|
482
|
+
...sink.interrupts.length > 0 ? { interrupts: sink.interrupts } : {}
|
|
483
|
+
};
|
|
484
|
+
}
|
|
485
|
+
}));
|
|
486
|
+
}
|
|
487
|
+
//#endregion
|
|
488
|
+
export { SUBAGENT_ERROR, SUBAGENT_FINISHED, SUBAGENT_STARTED, collectNamedText, collectUsage, createSubagentId, createSubagentSink, createSyntheticSubagentTools, normalizeRouterPick, rebindInterrupts, spawnAgentStream, spawnNamedAgents, subagentCallMessages, withChildUsage };
|
|
489
|
+
|
|
490
|
+
//# sourceMappingURL=spawn.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"spawn.js","names":[],"sources":["../../../../../src/activities/chat/agents/spawn.ts"],"sourcesContent":["import { EventType } from '../../../types'\nimport {\n addTokenUsage,\n isTanstackUsage,\n rebuildTokenUsage,\n toSpecTokenUsage,\n} from '../../../utilities/ag-ui-usage'\nimport {\n tanstackMetadata,\n withTanstackMetadata,\n} from '../../../utilities/merge-metadata'\nimport { INTERRUPT_BINDING_METADATA_KEY } from '../../../interrupt-resume'\nimport { EMIT_STREAM_CHUNK, SUBAGENT_TOOL } from '../tools/tool-calls'\nimport type { SubagentToolOutcome } from '../tools/tool-calls'\nimport type { SpecTokenUsage } from '../../../utilities/ag-ui-usage'\nimport type {\n Interrupt,\n ModelMessage,\n RunAgentResumeItem,\n StreamChunk,\n SubagentErrorEvent,\n SubagentFinishedEvent,\n SubagentStartedEvent,\n TokenUsage,\n Tool,\n UIMessage,\n} from '../../../types'\nimport type { DefinedAgent, SubagentRunContext } from './define-agent'\nimport type { ChatMiddleware } from '../middleware/types'\nimport type { SubagentTurn } from './turn'\n\nexport const SUBAGENT_STARTED = EventType.SUBAGENT_STARTED\nexport const SUBAGENT_FINISHED = EventType.SUBAGENT_FINISHED\nexport const SUBAGENT_ERROR = EventType.SUBAGENT_ERROR\n\nexport type SubagentOrder = 'parallel' | 'sequence'\n\nexport interface SubagentRouterPlan {\n names: ReadonlyArray<string>\n /** Overrides `subagents.order` for this turn. */\n order?: SubagentOrder\n}\n\nexport interface SubagentStep {\n names: ReadonlyArray<string>\n /** Overrides `subagents.order` for this step. */\n order?: SubagentOrder\n}\n\nexport interface SubagentStepsPlan {\n steps: ReadonlyArray<SubagentStep>\n}\n\nexport type SubagentRouterPick =\n | 'main'\n | string\n | ReadonlyArray<string>\n | SubagentRouterPlan\n | SubagentStepsPlan\n\nexport interface SubagentsBag<\n TAgents extends ReadonlyArray<DefinedAgent> = ReadonlyArray<DefinedAgent>,\n> {\n agents: TAgents\n router?: (ctx: {\n messages: SubagentRunContext['messages']\n agents: NoInfer<TAgents>\n abortSignal?: AbortSignal\n }) => SubagentRouterPick | Promise<SubagentRouterPick>\n strategy?: 'exclusive' | 'handoff'\n /**\n * How a router list runs. `parallel` starts every name together.\n * `sequence` runs each name after the previous one finishes, and passes\n * that child's text to the next child.\n */\n order?: 'parallel' | 'sequence'\n sandbox?: 'own' | 'inherit'\n}\n\n/** What the children of one parent run left behind for the parent terminal. */\nexport interface SubagentSink {\n interrupts: Array<Interrupt>\n /** One AG-UI entry per child model call. */\n usage: Array<SpecTokenUsage>\n /** Summed full usage of the children, including cost. */\n total?: TokenUsage\n}\n\nexport function createSubagentSink(): SubagentSink {\n return { interrupts: [], usage: [] }\n}\n\n/** One child to start, or a suspended child to continue. */\nexport interface SpawnEntry {\n name: string\n resume?: {\n subagentRunId: string\n /** The child's own messages from the interrupted run. */\n messages: Array<UIMessage | ModelMessage>\n entries: Array<RunAgentResumeItem>\n /** Text the child wrote before it stopped. */\n text: string\n }\n}\n\ninterface SpawnContext {\n messages: SubagentRunContext['messages']\n abortSignal?: AbortSignal\n threadId: string\n /** The parent chat run. */\n parentRunId: string\n /** The interrupted parent run, on a resume. */\n interruptedRunId?: string\n}\n\nexport function createSubagentId() {\n return `subagent-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`\n}\n\nfunction childRunId(parentRunId: string, subagentRunId: string) {\n return `${parentRunId}:${subagentRunId}`\n}\n\n/**\n * Bind child interrupts to the parent run. The client resumes the parent run,\n * so each binding must name that run. The resumed child then validates with\n * the parent's interrupted run id.\n */\nexport function rebindInterrupts(\n interrupts: ReadonlyArray<Interrupt>,\n runId: string,\n): Array<Interrupt> {\n return interrupts.map((interrupt) => {\n const binding = interrupt.metadata?.[INTERRUPT_BINDING_METADATA_KEY]\n if (typeof binding !== 'object' || binding === null) return interrupt\n return {\n ...interrupt,\n metadata: {\n ...interrupt.metadata,\n [INTERRUPT_BINDING_METADATA_KEY]: {\n ...binding,\n interruptedRunId: runId,\n generation: 0,\n },\n },\n }\n })\n}\n\nfunction createAbortError() {\n const error = new Error('Aborted')\n error.name = 'AbortError'\n return error\n}\n\nfunction isAbortError(error: unknown, signal?: AbortSignal) {\n if (signal?.aborted) return true\n return (\n error instanceof Error &&\n (error.name === 'AbortError' || error.message === 'Aborted')\n )\n}\n\nfunction stoppedEvent(subagentRunId: string) {\n return {\n type: SUBAGENT_ERROR,\n subagentRunId,\n message: 'Stopped',\n timestamp: Date.now(),\n } satisfies SubagentErrorEvent\n}\n\nfunction childThreadId(\n sandbox: SubagentsBag['sandbox'],\n parentThreadId: string,\n name: string,\n) {\n return sandbox === 'inherit' ? parentThreadId : `${parentThreadId}:${name}`\n}\n\nfunction linkAbort(parent?: AbortSignal) {\n const controller = new AbortController()\n if (!parent) return { controller, dispose: () => {} }\n if (parent.aborted) {\n controller.abort()\n return { controller, dispose: () => {} }\n }\n const onAbort = () => controller.abort()\n parent.addEventListener('abort', onAbort, { once: true })\n return {\n controller,\n dispose: () => parent.removeEventListener('abort', onAbort),\n }\n}\n\nfunction orAbort<T>(promise: Promise<T>, signal?: AbortSignal) {\n if (!signal) return promise\n if (signal.aborted) return Promise.reject(createAbortError())\n return new Promise<T>((resolve, reject) => {\n const onAbort = () => reject(createAbortError())\n signal.addEventListener('abort', onAbort, { once: true })\n promise.then(\n (value) => {\n signal.removeEventListener('abort', onAbort)\n resolve(value)\n },\n (error: unknown) => {\n signal.removeEventListener('abort', onAbort)\n reject(error)\n },\n )\n })\n}\n\nfunction agentByName(agents: ReadonlyArray<DefinedAgent>, name: string) {\n const agent = agents.find((entry) => entry.name === name)\n if (!agent) {\n throw new Error(`Unknown subagent: ${name}`)\n }\n return agent\n}\n\nfunction openAgentStream(\n entry: SpawnEntry,\n bag: SubagentsBag,\n ctx: SpawnContext,\n sink?: SubagentSink,\n parentToolCallId?: string,\n) {\n const agent = agentByName(bag.agents, entry.name)\n const resume = entry.resume\n const subagentRunId = resume?.subagentRunId ?? createSubagentId()\n // The parent binds child interrupts to its own run (see rebindInterrupts),\n // so the resumed child continues from the interrupted parent run id.\n if (resume !== undefined && ctx.interruptedRunId === undefined) {\n throw new Error(\n `Subagent \"${entry.name}\" has interrupt answers, but the run has no parentRunId. Pass the interrupted run id as parentRunId.`,\n )\n }\n const resumed =\n resume !== undefined\n ? {\n messages: [...ctx.messages, ...resume.messages],\n parentRunId: ctx.interruptedRunId,\n resume: resume.entries,\n }\n : undefined\n return spawnAgentStream(\n agent,\n {\n messages: resumed?.messages ?? ctx.messages,\n ...(ctx.abortSignal ? { abortSignal: ctx.abortSignal } : {}),\n threadId: childThreadId(bag.sandbox, ctx.threadId, entry.name),\n runId: childRunId(ctx.parentRunId, subagentRunId),\n parentRunId: resumed?.parentRunId ?? ctx.parentRunId,\n subagentRunId,\n ...(resumed ? { resume: resumed.resume } : {}),\n },\n sink,\n parentToolCallId,\n )\n}\n\nconst ROUTER_PICK_ERROR =\n 'subagents.router must return main, a name, a list of names, { names, order }, or { steps }.'\n\nfunction assertOrder(order: SubagentOrder | undefined) {\n if (order !== undefined && order !== 'parallel' && order !== 'sequence') {\n throw new Error('subagents.router order must be parallel or sequence.')\n }\n}\n\nfunction normalizeNames(\n names: ReadonlyArray<string>,\n agents: ReadonlyArray<DefinedAgent>,\n): ReadonlyArray<string> {\n if (names.length === 0) throw new Error(ROUTER_PICK_ERROR)\n const hasMain = names.includes('main')\n if (hasMain && names.length > 1) {\n throw new Error('Do not mix main into a subagent list.')\n }\n if (hasMain) return ['main']\n for (const name of names) agentByName(agents, name)\n return [...names]\n}\n\nfunction isStringList(pick: SubagentRouterPick): pick is ReadonlyArray<string> {\n return Array.isArray(pick)\n}\n\nexport function normalizeRouterPick(\n pick: SubagentRouterPick,\n agents: ReadonlyArray<DefinedAgent>,\n): { steps: ReadonlyArray<SubagentStep> } {\n if (pick === 'main' || typeof pick === 'string') {\n return { steps: [{ names: normalizeNames([pick], agents) }] }\n }\n if (isStringList(pick)) {\n return { steps: [{ names: normalizeNames(pick, agents) }] }\n }\n if ('steps' in pick) {\n if (pick.steps.length === 0) throw new Error(ROUTER_PICK_ERROR)\n const steps = pick.steps.map((step) => {\n assertOrder(step.order)\n const names = normalizeNames(step.names, agents)\n return step.order === undefined ? { names } : { names, order: step.order }\n })\n const flat = steps.flatMap((step) => step.names)\n if (flat.includes('main') && flat.length > 1) {\n throw new Error('Do not mix main into a subagent list.')\n }\n return { steps }\n }\n assertOrder(pick.order)\n const names = normalizeNames(pick.names, agents)\n return {\n steps: [\n pick.order === undefined ? { names } : { names, order: pick.order },\n ],\n }\n}\n\n/**\n * Tag a child chunk with its subagent. A chunk that a nested child already\n * tagged keeps its own id, and a nested child's start names this child as\n * its parent.\n */\nfunction attributeChunk(\n chunk: StreamChunk,\n subagentRunId: string,\n): StreamChunk {\n if (chunk.type === SUBAGENT_STARTED) {\n return chunk.parentSubagentRunId !== undefined\n ? chunk\n : { ...chunk, parentSubagentRunId: subagentRunId }\n }\n if (chunk.type === SUBAGENT_FINISHED || chunk.type === SUBAGENT_ERROR) {\n return chunk\n }\n if ('subagentRunId' in chunk && typeof chunk.subagentRunId === 'string') {\n return chunk\n }\n // RUN_* and MESSAGES_SNAPSHOT never get here (spawnAgentStream drops them).\n // Everything else is tagged.\n return { ...chunk, subagentRunId } as StreamChunk\n}\n\nfunction runUsage(chunk: StreamChunk | undefined): Array<SpecTokenUsage> {\n if (chunk?.type !== EventType.RUN_FINISHED) return []\n if (Array.isArray(chunk.usage)) return chunk.usage\n return isTanstackUsage(chunk.usage) ? toSpecTokenUsage(chunk.usage).usage : []\n}\n\n/** The full usage of a run: token counts plus cost and the other fields. */\nfunction fullUsage(chunk: StreamChunk | undefined): TokenUsage | undefined {\n if (chunk?.type !== EventType.RUN_FINISHED) return undefined\n return rebuildTokenUsage(chunk.usage, tanstackMetadata(chunk)?.usage)\n}\n\n/** Add a finished child run's usage to the sink. */\nexport function collectUsage(sink: SubagentSink, finished?: StreamChunk) {\n sink.usage.push(...runUsage(finished))\n const full = fullUsage(finished)\n if (full) sink.total = sink.total ? addTokenUsage(sink.total, full) : full\n}\n\n/** A parent run's last chunk: it completed, or it failed. */\ntype ParentTerminal = Extract<\n StreamChunk,\n { type: 'RUN_FINISHED' | 'RUN_ERROR' }\n>\n\n/**\n * Put the children's usage on a parent terminal. `usage[]` keeps one entry per\n * model call. `metadata.tanstack.usage` holds the summed cost and the other\n * TanStack fields, so `fromSpecTokenUsage` reads the full total. Empties the\n * sink, so the next parent terminal does not count it again.\n *\n * `RUN_ERROR` is accepted too: a turn that failed still spent whatever its\n * children spent. Such a chunk carries no usage of its own, so `runUsage` and\n * `fullUsage` return empty for it and the children's total stands alone.\n */\nexport function withChildUsage(\n chunk: ParentTerminal,\n sink: SubagentSink,\n): ParentTerminal {\n if (sink.usage.length === 0 && !sink.total) return chunk\n const own = fullUsage(chunk)\n const total =\n own && sink.total ? addTokenUsage(own, sink.total) : (own ?? sink.total)\n const usage = [...runUsage(chunk), ...sink.usage.splice(0)]\n sink.total = undefined\n const leftover = total ? toSpecTokenUsage(total).leftover : undefined\n const next = { ...chunk, usage }\n if (!leftover) return next\n // `withTanstackMetadata` runs the value through `Omit`, which collapses this\n // union and widens `type` back to `RUN_FINISHED | RUN_ERROR`. It only adds a\n // metadata key, so the runtime shape is the input's: restore that type.\n return withTanstackMetadata(next, { usage: leftover }) as ParentTerminal\n}\n\nexport async function* spawnAgentStream(\n agent: DefinedAgent,\n ctx: SubagentRunContext,\n sink?: SubagentSink,\n parentToolCallId?: string,\n): AsyncIterable<StreamChunk> {\n const id = ctx.subagentRunId\n yield {\n type: SUBAGENT_STARTED,\n subagentRunId: id,\n name: agent.name,\n description: agent.description,\n ...(ctx.parentSubagentRunId !== undefined\n ? { parentSubagentRunId: ctx.parentSubagentRunId }\n : {}),\n ...(parentToolCallId !== undefined ? { parentToolCallId } : {}),\n timestamp: Date.now(),\n } satisfies SubagentStartedEvent\n\n let iterator: AsyncIterator<StreamChunk> | undefined\n let finished: StreamChunk | undefined\n try {\n if (ctx.abortSignal?.aborted) {\n yield stoppedEvent(id)\n return\n }\n const stream = await orAbort(\n Promise.resolve(agent.run(ctx)),\n ctx.abortSignal,\n )\n iterator = stream[Symbol.asyncIterator]()\n while (true) {\n if (ctx.abortSignal?.aborted) {\n yield stoppedEvent(id)\n return\n }\n const result = await orAbort(iterator.next(), ctx.abortSignal)\n if (result.done) break\n const chunk = result.value\n // Run-scoped events describe the child run. The SUBAGENT_* events carry\n // that information for the parent stream.\n if (\n chunk.type === EventType.RUN_STARTED ||\n chunk.type === EventType.MESSAGES_SNAPSHOT\n ) {\n continue\n }\n if (chunk.type === EventType.RUN_FINISHED) {\n // The engine yields one RUN_FINISHED per model call. Add each one.\n if (sink) collectUsage(sink, chunk)\n finished = chunk\n continue\n }\n if (chunk.type === EventType.RUN_ERROR) {\n yield {\n type: SUBAGENT_ERROR,\n subagentRunId: id,\n message: chunk.message || 'Subagent failed',\n ...(chunk.code ? { code: chunk.code } : {}),\n timestamp: Date.now(),\n } satisfies SubagentErrorEvent\n return\n }\n yield attributeChunk(chunk, id)\n }\n if (ctx.abortSignal?.aborted) {\n yield stoppedEvent(id)\n return\n }\n const outcome =\n finished?.type === EventType.RUN_FINISHED ? finished.outcome : undefined\n if (outcome?.type === 'cancelled') {\n yield stoppedEvent(id)\n return\n }\n if (outcome?.type === 'interrupt') {\n const interrupts = outcome.interrupts.map((interrupt) =>\n interrupt.subagentRunId\n ? interrupt\n : { ...interrupt, subagentRunId: id },\n )\n sink?.interrupts.push(...interrupts)\n yield {\n type: SUBAGENT_FINISHED,\n subagentRunId: id,\n outcome: {\n type: 'suspended',\n interruptIds: interrupts\n .filter((interrupt) => interrupt.subagentRunId === id)\n .map((interrupt) => interrupt.id),\n },\n timestamp: Date.now(),\n } satisfies SubagentFinishedEvent\n return\n }\n const result =\n finished?.type === EventType.RUN_FINISHED ? finished.result : undefined\n yield {\n type: SUBAGENT_FINISHED,\n subagentRunId: id,\n ...(result !== undefined ? { result } : {}),\n timestamp: Date.now(),\n } satisfies SubagentFinishedEvent\n } catch (error) {\n yield {\n type: SUBAGENT_ERROR,\n subagentRunId: id,\n message: isAbortError(error, ctx.abortSignal)\n ? 'Stopped'\n : error instanceof Error\n ? error.message\n : String(error),\n timestamp: Date.now(),\n } satisfies SubagentErrorEvent\n } finally {\n try {\n await iterator?.return?.()\n } catch {\n // Child stream may already be closed or aborted.\n }\n }\n}\n\nasync function* mergeAgentStreams(streams: Array<AsyncIterable<StreamChunk>>) {\n const readers = streams.map((stream) => {\n const iterator = stream[Symbol.asyncIterator]()\n return {\n iterator,\n next: iterator.next(),\n }\n })\n\n try {\n while (readers.length > 0) {\n const indexed = readers.map((reader, index) =>\n reader.next.then((result) => ({ index, result, reader })),\n )\n const winner = await Promise.race(indexed)\n if (winner.result.done) {\n readers.splice(winner.index, 1)\n continue\n }\n yield winner.result.value\n winner.reader.next = winner.reader.iterator.next()\n }\n } finally {\n // The reader stopped early. Close every child so its finally runs.\n for (const reader of readers) {\n void reader.iterator.return?.().catch(() => {})\n }\n }\n}\n\n/** True when a child in these chunks failed or stopped for outside input. */\nfunction stopsSequence(chunks: ReadonlyArray<StreamChunk>, id: string) {\n return chunks.some(\n (chunk) =>\n (chunk.type === SUBAGENT_ERROR && chunk.subagentRunId === id) ||\n (chunk.type === SUBAGENT_FINISHED &&\n chunk.subagentRunId === id &&\n chunk.outcome?.type === 'suspended'),\n )\n}\n\nexport async function* spawnNamedAgents(\n entries: ReadonlyArray<SpawnEntry>,\n bag: SubagentsBag,\n ctx: SpawnContext,\n sink?: SubagentSink,\n) {\n if (bag.sandbox === 'inherit' && entries.length > 1) {\n throw new Error(\n \"subagents.sandbox 'inherit' cannot start two children in one turn\",\n )\n }\n // One signal for the group. The finally stops every child that still runs\n // when the reader stops early.\n const group = linkAbort(ctx.abortSignal)\n const groupCtx = { ...ctx, abortSignal: group.controller.signal }\n try {\n if (bag.order === 'sequence') {\n let messages = ctx.messages\n for (const entry of entries) {\n const chunks: Array<StreamChunk> = []\n let id: string | undefined\n for await (const chunk of openAgentStream(\n entry,\n bag,\n { ...groupCtx, messages },\n sink,\n )) {\n if (chunk.type === SUBAGENT_STARTED && id === undefined) {\n id = chunk.subagentRunId\n }\n chunks.push(chunk)\n yield chunk\n }\n if (id !== undefined && stopsSequence(chunks, id)) return\n const text = [\n entry.resume?.text,\n collectNamedText(chunks, [entry.name]),\n ]\n .filter((part) => part !== undefined && part !== '')\n .join('')\n if (text) {\n messages = [...messages, { role: 'assistant', content: text }]\n }\n }\n return\n }\n const streams = entries.map((entry) =>\n openAgentStream(entry, bag, groupCtx, sink),\n )\n const onlyStream = streams.length === 1 ? streams[0] : undefined\n if (onlyStream) {\n yield* onlyStream\n return\n }\n yield* mergeAgentStreams(streams)\n } finally {\n group.controller.abort()\n group.dispose()\n }\n}\n\n/**\n * Text of the named direct children, in `names` order. Text from nested\n * children stays out: their chunks carry their own id.\n */\nexport function collectNamedText(\n chunks: Array<StreamChunk>,\n names: ReadonlyArray<string>,\n) {\n const nameByRunId = new Map<string, string>()\n const textByName = new Map<string, string>()\n for (const chunk of chunks) {\n if (chunk.type === SUBAGENT_STARTED) {\n if (chunk.parentSubagentRunId === undefined) {\n nameByRunId.set(chunk.subagentRunId, chunk.name)\n }\n continue\n }\n if (chunk.type !== EventType.TEXT_MESSAGE_CONTENT) continue\n if (!('subagentRunId' in chunk) || typeof chunk.subagentRunId !== 'string')\n continue\n const name = nameByRunId.get(chunk.subagentRunId)\n if (!name) continue\n textByName.set(name, `${textByName.get(name) ?? ''}${chunk.delta}`)\n }\n return names\n .map((name) => textByName.get(name)?.trim() ?? '')\n .filter((text) => text.length > 0)\n .join('\\n\\n')\n}\n\n/**\n * The parent conversation up to the message that carries this tool call, with\n * that message's tool calls removed. Its string text stays; array content is\n * dropped.\n */\nfunction messagesBeforeCall(\n messages: ReadonlyArray<ModelMessage>,\n toolCallId: string,\n): Array<ModelMessage> {\n const index = messages.findIndex((message) =>\n message.toolCalls?.some((call) => call.id === toolCallId),\n )\n if (index === -1) return [...messages]\n const host = messages[index]\n const kept = messages.slice(0, index)\n if (host && typeof host.content === 'string' && host.content !== '') {\n const { toolCalls: _calls, ...text } = host\n void _calls\n kept.push(text)\n }\n return kept\n}\n\n/**\n * Record the parent messages when the model calls a subagent tool, so the\n * child reads the conversation as it is at that call.\n */\nexport function subagentCallMessages(names: ReadonlySet<string>) {\n const byCall = new Map<string, Array<ModelMessage>>()\n const middleware: ChatMiddleware = {\n name: 'subagent-call-messages',\n onBeforeToolCall(ctx, hook) {\n if (!names.has(hook.toolName)) return undefined\n byCall.set(\n hook.toolCallId,\n messagesBeforeCall(ctx.messages, hook.toolCallId),\n )\n return undefined\n },\n }\n return {\n middleware,\n messagesFor: (toolCallId: string | undefined) =>\n toolCallId === undefined ? undefined : byCall.get(toolCallId),\n }\n}\n\nexport function createSyntheticSubagentTools(\n bag: SubagentsBag,\n parent: {\n /** Messages the parent run started with. Used when no call was recorded. */\n messages: SubagentRunContext['messages']\n /** The parent messages at a tool call. See subagentCallMessages. */\n messagesFor?: (\n toolCallId: string | undefined,\n ) => SubagentRunContext['messages'] | undefined\n threadId: string\n runId: string\n interruptedRunId?: string\n abortSignal?: AbortSignal\n turn?: SubagentTurn\n sink: SubagentSink\n },\n): Array<Tool> {\n return bag.agents.map((agent) => ({\n name: agent.name,\n description: agent.description,\n [SUBAGENT_TOOL]: true,\n execute: async (_input: unknown, context?: unknown) => {\n const toolContext = context as\n | {\n toolCallId?: string\n [EMIT_STREAM_CHUNK]?: (chunk: StreamChunk) => void\n }\n | undefined\n const toolCallId = toolContext?.toolCallId\n const suspended = parent.turn?.children.find(\n (child) =>\n child.status === 'suspended' &&\n child.parentToolCallId !== undefined &&\n child.parentToolCallId === toolCallId,\n )\n const entry: SpawnEntry = suspended\n ? {\n name: agent.name,\n resume: {\n subagentRunId: suspended.subagentRunId,\n messages: suspended.messages,\n entries: suspended.resume,\n text: suspended.text,\n },\n }\n : { name: agent.name }\n const sink = createSubagentSink()\n const link = linkAbort(parent.abortSignal)\n let subagentRunId = suspended?.subagentRunId ?? ''\n let text = suspended?.text ?? ''\n let error: string | undefined\n try {\n for await (const chunk of openAgentStream(\n entry,\n bag,\n {\n messages: parent.messagesFor?.(toolCallId) ?? parent.messages,\n abortSignal: link.controller.signal,\n threadId: parent.threadId,\n parentRunId: parent.runId,\n ...(parent.interruptedRunId !== undefined\n ? { interruptedRunId: parent.interruptedRunId }\n : {}),\n },\n sink,\n toolCallId,\n )) {\n if (chunk.type === SUBAGENT_STARTED && subagentRunId === '') {\n subagentRunId = chunk.subagentRunId\n }\n if (\n chunk.type === EventType.TEXT_MESSAGE_CONTENT &&\n 'subagentRunId' in chunk &&\n chunk.subagentRunId === subagentRunId\n ) {\n text += chunk.delta\n }\n if (\n chunk.type === SUBAGENT_ERROR &&\n chunk.subagentRunId === subagentRunId\n ) {\n error = chunk.message\n }\n toolContext?.[EMIT_STREAM_CHUNK]?.(chunk)\n }\n } finally {\n link.dispose()\n }\n parent.sink.usage.push(...sink.usage)\n if (sink.total) {\n parent.sink.total = parent.sink.total\n ? addTokenUsage(parent.sink.total, sink.total)\n : sink.total\n }\n return {\n subagentRunId,\n text,\n ...(error !== undefined ? { error } : {}),\n ...(sink.interrupts.length > 0 ? { interrupts: sink.interrupts } : {}),\n } satisfies SubagentToolOutcome\n },\n }))\n}\n"],"mappings":";;;;;;AA+BA,IAAa,mBAAmB,UAAU;AAC1C,IAAa,oBAAoB,UAAU;AAC3C,IAAa,iBAAiB,UAAU;AAuDxC,SAAgB,qBAAmC;CACjD,OAAO;EAAE,YAAY,CAAC;EAAG,OAAO,CAAC;CAAE;AACrC;AAyBA,SAAgB,mBAAmB;CACjC,OAAO,YAAY,KAAK,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC;AACxE;AAEA,SAAS,WAAW,aAAqB,eAAuB;CAC9D,OAAO,GAAG,YAAY,GAAG;AAC3B;;;;;;AAOA,SAAgB,iBACd,YACA,OACkB;CAClB,OAAO,WAAW,KAAK,cAAc;EACnC,MAAM,UAAU,UAAU,WAAW;EACrC,IAAI,OAAO,YAAY,YAAY,YAAY,MAAM,OAAO;EAC5D,OAAO;GACL,GAAG;GACH,UAAU;IACR,GAAG,UAAU;KACZ,iCAAiC;KAChC,GAAG;KACH,kBAAkB;KAClB,YAAY;IACd;GACF;EACF;CACF,CAAC;AACH;AAEA,SAAS,mBAAmB;CAC1B,MAAM,wBAAQ,IAAI,MAAM,SAAS;CACjC,MAAM,OAAO;CACb,OAAO;AACT;AAEA,SAAS,aAAa,OAAgB,QAAsB;CAC1D,IAAI,QAAQ,SAAS,OAAO;CAC5B,OACE,iBAAiB,UAChB,MAAM,SAAS,gBAAgB,MAAM,YAAY;AAEtD;AAEA,SAAS,aAAa,eAAuB;CAC3C,OAAO;EACL,MAAM;EACN;EACA,SAAS;EACT,WAAW,KAAK,IAAI;CACtB;AACF;AAEA,SAAS,cACP,SACA,gBACA,MACA;CACA,OAAO,YAAY,YAAY,iBAAiB,GAAG,eAAe,GAAG;AACvE;AAEA,SAAS,UAAU,QAAsB;CACvC,MAAM,aAAa,IAAI,gBAAgB;CACvC,IAAI,CAAC,QAAQ,OAAO;EAAE;EAAY,eAAe,CAAC;CAAE;CACpD,IAAI,OAAO,SAAS;EAClB,WAAW,MAAM;EACjB,OAAO;GAAE;GAAY,eAAe,CAAC;EAAE;CACzC;CACA,MAAM,gBAAgB,WAAW,MAAM;CACvC,OAAO,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;CACxD,OAAO;EACL;EACA,eAAe,OAAO,oBAAoB,SAAS,OAAO;CAC5D;AACF;AAEA,SAAS,QAAW,SAAqB,QAAsB;CAC7D,IAAI,CAAC,QAAQ,OAAO;CACpB,IAAI,OAAO,SAAS,OAAO,QAAQ,OAAO,iBAAiB,CAAC;CAC5D,OAAO,IAAI,SAAY,SAAS,WAAW;EACzC,MAAM,gBAAgB,OAAO,iBAAiB,CAAC;EAC/C,OAAO,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;EACxD,QAAQ,MACL,UAAU;GACT,OAAO,oBAAoB,SAAS,OAAO;GAC3C,QAAQ,KAAK;EACf,IACC,UAAmB;GAClB,OAAO,oBAAoB,SAAS,OAAO;GAC3C,OAAO,KAAK;EACd,CACF;CACF,CAAC;AACH;AAEA,SAAS,YAAY,QAAqC,MAAc;CACtE,MAAM,QAAQ,OAAO,MAAM,UAAU,MAAM,SAAS,IAAI;CACxD,IAAI,CAAC,OACH,MAAM,IAAI,MAAM,qBAAqB,MAAM;CAE7C,OAAO;AACT;AAEA,SAAS,gBACP,OACA,KACA,KACA,MACA,kBACA;CACA,MAAM,QAAQ,YAAY,IAAI,QAAQ,MAAM,IAAI;CAChD,MAAM,SAAS,MAAM;CACrB,MAAM,gBAAgB,QAAQ,iBAAiB,iBAAiB;CAGhE,IAAI,WAAW,KAAA,KAAa,IAAI,qBAAqB,KAAA,GACnD,MAAM,IAAI,MACR,aAAa,MAAM,KAAK,qGAC1B;CAEF,MAAM,UACJ,WAAW,KAAA,IACP;EACE,UAAU,CAAC,GAAG,IAAI,UAAU,GAAG,OAAO,QAAQ;EAC9C,aAAa,IAAI;EACjB,QAAQ,OAAO;CACjB,IACA,KAAA;CACN,OAAO,iBACL,OACA;EACE,UAAU,SAAS,YAAY,IAAI;EACnC,GAAI,IAAI,cAAc,EAAE,aAAa,IAAI,YAAY,IAAI,CAAC;EAC1D,UAAU,cAAc,IAAI,SAAS,IAAI,UAAU,MAAM,IAAI;EAC7D,OAAO,WAAW,IAAI,aAAa,aAAa;EAChD,aAAa,SAAS,eAAe,IAAI;EACzC;EACA,GAAI,UAAU,EAAE,QAAQ,QAAQ,OAAO,IAAI,CAAC;CAC9C,GACA,MACA,gBACF;AACF;AAEA,IAAM,oBACJ;AAEF,SAAS,YAAY,OAAkC;CACrD,IAAI,UAAU,KAAA,KAAa,UAAU,cAAc,UAAU,YAC3D,MAAM,IAAI,MAAM,sDAAsD;AAE1E;AAEA,SAAS,eACP,OACA,QACuB;CACvB,IAAI,MAAM,WAAW,GAAG,MAAM,IAAI,MAAM,iBAAiB;CACzD,MAAM,UAAU,MAAM,SAAS,MAAM;CACrC,IAAI,WAAW,MAAM,SAAS,GAC5B,MAAM,IAAI,MAAM,uCAAuC;CAEzD,IAAI,SAAS,OAAO,CAAC,MAAM;CAC3B,KAAK,MAAM,QAAQ,OAAO,YAAY,QAAQ,IAAI;CAClD,OAAO,CAAC,GAAG,KAAK;AAClB;AAEA,SAAS,aAAa,MAAyD;CAC7E,OAAO,MAAM,QAAQ,IAAI;AAC3B;AAEA,SAAgB,oBACd,MACA,QACwC;CACxC,IAAI,SAAS,UAAU,OAAO,SAAS,UACrC,OAAO,EAAE,OAAO,CAAC,EAAE,OAAO,eAAe,CAAC,IAAI,GAAG,MAAM,EAAE,CAAC,EAAE;CAE9D,IAAI,aAAa,IAAI,GACnB,OAAO,EAAE,OAAO,CAAC,EAAE,OAAO,eAAe,MAAM,MAAM,EAAE,CAAC,EAAE;CAE5D,IAAI,WAAW,MAAM;EACnB,IAAI,KAAK,MAAM,WAAW,GAAG,MAAM,IAAI,MAAM,iBAAiB;EAC9D,MAAM,QAAQ,KAAK,MAAM,KAAK,SAAS;GACrC,YAAY,KAAK,KAAK;GACtB,MAAM,QAAQ,eAAe,KAAK,OAAO,MAAM;GAC/C,OAAO,KAAK,UAAU,KAAA,IAAY,EAAE,MAAM,IAAI;IAAE;IAAO,OAAO,KAAK;GAAM;EAC3E,CAAC;EACD,MAAM,OAAO,MAAM,SAAS,SAAS,KAAK,KAAK;EAC/C,IAAI,KAAK,SAAS,MAAM,KAAK,KAAK,SAAS,GACzC,MAAM,IAAI,MAAM,uCAAuC;EAEzD,OAAO,EAAE,MAAM;CACjB;CACA,YAAY,KAAK,KAAK;CACtB,MAAM,QAAQ,eAAe,KAAK,OAAO,MAAM;CAC/C,OAAO,EACL,OAAO,CACL,KAAK,UAAU,KAAA,IAAY,EAAE,MAAM,IAAI;EAAE;EAAO,OAAO,KAAK;CAAM,CACpE,EACF;AACF;;;;;;AAOA,SAAS,eACP,OACA,eACa;CACb,IAAI,MAAM,SAAS,kBACjB,OAAO,MAAM,wBAAwB,KAAA,IACjC,QACA;EAAE,GAAG;EAAO,qBAAqB;CAAc;CAErD,IAAI,MAAM,SAAS,qBAAqB,MAAM,SAAS,gBACrD,OAAO;CAET,IAAI,mBAAmB,SAAS,OAAO,MAAM,kBAAkB,UAC7D,OAAO;CAIT,OAAO;EAAE,GAAG;EAAO;CAAc;AACnC;AAEA,SAAS,SAAS,OAAuD;CACvE,IAAI,OAAO,SAAS,UAAU,cAAc,OAAO,CAAC;CACpD,IAAI,MAAM,QAAQ,MAAM,KAAK,GAAG,OAAO,MAAM;CAC7C,OAAO,gBAAgB,MAAM,KAAK,IAAI,iBAAiB,MAAM,KAAK,CAAC,CAAC,QAAQ,CAAC;AAC/E;;AAGA,SAAS,UAAU,OAAwD;CACzE,IAAI,OAAO,SAAS,UAAU,cAAc,OAAO,KAAA;CACnD,OAAO,kBAAkB,MAAM,OAAO,iBAAiB,KAAK,CAAC,EAAE,KAAK;AACtE;;AAGA,SAAgB,aAAa,MAAoB,UAAwB;CACvE,KAAK,MAAM,KAAK,GAAG,SAAS,QAAQ,CAAC;CACrC,MAAM,OAAO,UAAU,QAAQ;CAC/B,IAAI,MAAM,KAAK,QAAQ,KAAK,QAAQ,cAAc,KAAK,OAAO,IAAI,IAAI;AACxE;;;;;;;;;;;AAkBA,SAAgB,eACd,OACA,MACgB;CAChB,IAAI,KAAK,MAAM,WAAW,KAAK,CAAC,KAAK,OAAO,OAAO;CACnD,MAAM,MAAM,UAAU,KAAK;CAC3B,MAAM,QACJ,OAAO,KAAK,QAAQ,cAAc,KAAK,KAAK,KAAK,IAAK,OAAO,KAAK;CACpE,MAAM,QAAQ,CAAC,GAAG,SAAS,KAAK,GAAG,GAAG,KAAK,MAAM,OAAO,CAAC,CAAC;CAC1D,KAAK,QAAQ,KAAA;CACb,MAAM,WAAW,QAAQ,iBAAiB,KAAK,CAAC,CAAC,WAAW,KAAA;CAC5D,MAAM,OAAO;EAAE,GAAG;EAAO;CAAM;CAC/B,IAAI,CAAC,UAAU,OAAO;CAItB,OAAO,qBAAqB,MAAM,EAAE,OAAO,SAAS,CAAC;AACvD;AAEA,gBAAuB,iBACrB,OACA,KACA,MACA,kBAC4B;CAC5B,MAAM,KAAK,IAAI;CACf,MAAM;EACJ,MAAM;EACN,eAAe;EACf,MAAM,MAAM;EACZ,aAAa,MAAM;EACnB,GAAI,IAAI,wBAAwB,KAAA,IAC5B,EAAE,qBAAqB,IAAI,oBAAoB,IAC/C,CAAC;EACL,GAAI,qBAAqB,KAAA,IAAY,EAAE,iBAAiB,IAAI,CAAC;EAC7D,WAAW,KAAK,IAAI;CACtB;CAEA,IAAI;CACJ,IAAI;CACJ,IAAI;EACF,IAAI,IAAI,aAAa,SAAS;GAC5B,MAAM,aAAa,EAAE;GACrB;EACF;EAKA,YAAW,MAJU,QACnB,QAAQ,QAAQ,MAAM,IAAI,GAAG,CAAC,GAC9B,IAAI,WACN,EAAA,CACkB,OAAO,cAAc,CAAC;EACxC,OAAO,MAAM;GACX,IAAI,IAAI,aAAa,SAAS;IAC5B,MAAM,aAAa,EAAE;IACrB;GACF;GACA,MAAM,SAAS,MAAM,QAAQ,SAAS,KAAK,GAAG,IAAI,WAAW;GAC7D,IAAI,OAAO,MAAM;GACjB,MAAM,QAAQ,OAAO;GAGrB,IACE,MAAM,SAAS,UAAU,eACzB,MAAM,SAAS,UAAU,mBAEzB;GAEF,IAAI,MAAM,SAAS,UAAU,cAAc;IAEzC,IAAI,MAAM,aAAa,MAAM,KAAK;IAClC,WAAW;IACX;GACF;GACA,IAAI,MAAM,SAAS,UAAU,WAAW;IACtC,MAAM;KACJ,MAAM;KACN,eAAe;KACf,SAAS,MAAM,WAAW;KAC1B,GAAI,MAAM,OAAO,EAAE,MAAM,MAAM,KAAK,IAAI,CAAC;KACzC,WAAW,KAAK,IAAI;IACtB;IACA;GACF;GACA,MAAM,eAAe,OAAO,EAAE;EAChC;EACA,IAAI,IAAI,aAAa,SAAS;GAC5B,MAAM,aAAa,EAAE;GACrB;EACF;EACA,MAAM,UACJ,UAAU,SAAS,UAAU,eAAe,SAAS,UAAU,KAAA;EACjE,IAAI,SAAS,SAAS,aAAa;GACjC,MAAM,aAAa,EAAE;GACrB;EACF;EACA,IAAI,SAAS,SAAS,aAAa;GACjC,MAAM,aAAa,QAAQ,WAAW,KAAK,cACzC,UAAU,gBACN,YACA;IAAE,GAAG;IAAW,eAAe;GAAG,CACxC;GACA,MAAM,WAAW,KAAK,GAAG,UAAU;GACnC,MAAM;IACJ,MAAM;IACN,eAAe;IACf,SAAS;KACP,MAAM;KACN,cAAc,WACX,QAAQ,cAAc,UAAU,kBAAkB,EAAE,CAAC,CACrD,KAAK,cAAc,UAAU,EAAE;IACpC;IACA,WAAW,KAAK,IAAI;GACtB;GACA;EACF;EACA,MAAM,SACJ,UAAU,SAAS,UAAU,eAAe,SAAS,SAAS,KAAA;EAChE,MAAM;GACJ,MAAM;GACN,eAAe;GACf,GAAI,WAAW,KAAA,IAAY,EAAE,OAAO,IAAI,CAAC;GACzC,WAAW,KAAK,IAAI;EACtB;CACF,SAAS,OAAO;EACd,MAAM;GACJ,MAAM;GACN,eAAe;GACf,SAAS,aAAa,OAAO,IAAI,WAAW,IACxC,YACA,iBAAiB,QACf,MAAM,UACN,OAAO,KAAK;GAClB,WAAW,KAAK,IAAI;EACtB;CACF,UAAU;EACR,IAAI;GACF,MAAM,UAAU,SAAS;EAC3B,QAAQ,CAER;CACF;AACF;AAEA,gBAAgB,kBAAkB,SAA4C;CAC5E,MAAM,UAAU,QAAQ,KAAK,WAAW;EACtC,MAAM,WAAW,OAAO,OAAO,cAAc,CAAC;EAC9C,OAAO;GACL;GACA,MAAM,SAAS,KAAK;EACtB;CACF,CAAC;CAED,IAAI;EACF,OAAO,QAAQ,SAAS,GAAG;GACzB,MAAM,UAAU,QAAQ,KAAK,QAAQ,UACnC,OAAO,KAAK,MAAM,YAAY;IAAE;IAAO;IAAQ;GAAO,EAAE,CAC1D;GACA,MAAM,SAAS,MAAM,QAAQ,KAAK,OAAO;GACzC,IAAI,OAAO,OAAO,MAAM;IACtB,QAAQ,OAAO,OAAO,OAAO,CAAC;IAC9B;GACF;GACA,MAAM,OAAO,OAAO;GACpB,OAAO,OAAO,OAAO,OAAO,OAAO,SAAS,KAAK;EACnD;CACF,UAAU;EAER,KAAK,MAAM,UAAU,SACnB,OAAY,SAAS,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC;CAElD;AACF;;AAGA,SAAS,cAAc,QAAoC,IAAY;CACrE,OAAO,OAAO,MACX,UACE,MAAM,SAAS,kBAAkB,MAAM,kBAAkB,MACzD,MAAM,SAAS,qBACd,MAAM,kBAAkB,MACxB,MAAM,SAAS,SAAS,WAC9B;AACF;AAEA,gBAAuB,iBACrB,SACA,KACA,KACA,MACA;CACA,IAAI,IAAI,YAAY,aAAa,QAAQ,SAAS,GAChD,MAAM,IAAI,MACR,mEACF;CAIF,MAAM,QAAQ,UAAU,IAAI,WAAW;CACvC,MAAM,WAAW;EAAE,GAAG;EAAK,aAAa,MAAM,WAAW;CAAO;CAChE,IAAI;EACF,IAAI,IAAI,UAAU,YAAY;GAC5B,IAAI,WAAW,IAAI;GACnB,KAAK,MAAM,SAAS,SAAS;IAC3B,MAAM,SAA6B,CAAC;IACpC,IAAI;IACJ,WAAW,MAAM,SAAS,gBACxB,OACA,KACA;KAAE,GAAG;KAAU;IAAS,GACxB,IACF,GAAG;KACD,IAAI,MAAM,SAAS,oBAAoB,OAAO,KAAA,GAC5C,KAAK,MAAM;KAEb,OAAO,KAAK,KAAK;KACjB,MAAM;IACR;IACA,IAAI,OAAO,KAAA,KAAa,cAAc,QAAQ,EAAE,GAAG;IACnD,MAAM,OAAO,CACX,MAAM,QAAQ,MACd,iBAAiB,QAAQ,CAAC,MAAM,IAAI,CAAC,CACvC,CAAC,CACE,QAAQ,SAAS,SAAS,KAAA,KAAa,SAAS,EAAE,CAAC,CACnD,KAAK,EAAE;IACV,IAAI,MACF,WAAW,CAAC,GAAG,UAAU;KAAE,MAAM;KAAa,SAAS;IAAK,CAAC;GAEjE;GACA;EACF;EACA,MAAM,UAAU,QAAQ,KAAK,UAC3B,gBAAgB,OAAO,KAAK,UAAU,IAAI,CAC5C;EACA,MAAM,aAAa,QAAQ,WAAW,IAAI,QAAQ,KAAK,KAAA;EACvD,IAAI,YAAY;GACd,OAAO;GACP;EACF;EACA,OAAO,kBAAkB,OAAO;CAClC,UAAU;EACR,MAAM,WAAW,MAAM;EACvB,MAAM,QAAQ;CAChB;AACF;;;;;AAMA,SAAgB,iBACd,QACA,OACA;CACA,MAAM,8BAAc,IAAI,IAAoB;CAC5C,MAAM,6BAAa,IAAI,IAAoB;CAC3C,KAAK,MAAM,SAAS,QAAQ;EAC1B,IAAI,MAAM,SAAS,kBAAkB;GACnC,IAAI,MAAM,wBAAwB,KAAA,GAChC,YAAY,IAAI,MAAM,eAAe,MAAM,IAAI;GAEjD;EACF;EACA,IAAI,MAAM,SAAS,UAAU,sBAAsB;EACnD,IAAI,EAAE,mBAAmB,UAAU,OAAO,MAAM,kBAAkB,UAChE;EACF,MAAM,OAAO,YAAY,IAAI,MAAM,aAAa;EAChD,IAAI,CAAC,MAAM;EACX,WAAW,IAAI,MAAM,GAAG,WAAW,IAAI,IAAI,KAAK,KAAK,MAAM,OAAO;CACpE;CACA,OAAO,MACJ,KAAK,SAAS,WAAW,IAAI,IAAI,CAAC,EAAE,KAAK,KAAK,EAAE,CAAC,CACjD,QAAQ,SAAS,KAAK,SAAS,CAAC,CAAC,CACjC,KAAK,MAAM;AAChB;;;;;;AAOA,SAAS,mBACP,UACA,YACqB;CACrB,MAAM,QAAQ,SAAS,WAAW,YAChC,QAAQ,WAAW,MAAM,SAAS,KAAK,OAAO,UAAU,CAC1D;CACA,IAAI,UAAU,IAAI,OAAO,CAAC,GAAG,QAAQ;CACrC,MAAM,OAAO,SAAS;CACtB,MAAM,OAAO,SAAS,MAAM,GAAG,KAAK;CACpC,IAAI,QAAQ,OAAO,KAAK,YAAY,YAAY,KAAK,YAAY,IAAI;EACnE,MAAM,EAAE,WAAW,QAAQ,GAAG,SAAS;EAEvC,KAAK,KAAK,IAAI;CAChB;CACA,OAAO;AACT;;;;;AAMA,SAAgB,qBAAqB,OAA4B;CAC/D,MAAM,yBAAS,IAAI,IAAiC;CAYpD,OAAO;EACL,YAAA;GAXA,MAAM;GACN,iBAAiB,KAAK,MAAM;IAC1B,IAAI,CAAC,MAAM,IAAI,KAAK,QAAQ,GAAG,OAAO,KAAA;IACtC,OAAO,IACL,KAAK,YACL,mBAAmB,IAAI,UAAU,KAAK,UAAU,CAClD;GAEF;EAGA;EACA,cAAc,eACZ,eAAe,KAAA,IAAY,KAAA,IAAY,OAAO,IAAI,UAAU;CAChE;AACF;AAEA,SAAgB,6BACd,KACA,QAca;CACb,OAAO,IAAI,OAAO,KAAK,WAAW;EAChC,MAAM,MAAM;EACZ,aAAa,MAAM;GAClB,gBAAgB;EACjB,SAAS,OAAO,QAAiB,YAAsB;GACrD,MAAM,cAAc;GAMpB,MAAM,aAAa,aAAa;GAChC,MAAM,YAAY,OAAO,MAAM,SAAS,MACrC,UACC,MAAM,WAAW,eACjB,MAAM,qBAAqB,KAAA,KAC3B,MAAM,qBAAqB,UAC/B;GACA,MAAM,QAAoB,YACtB;IACE,MAAM,MAAM;IACZ,QAAQ;KACN,eAAe,UAAU;KACzB,UAAU,UAAU;KACpB,SAAS,UAAU;KACnB,MAAM,UAAU;IAClB;GACF,IACA,EAAE,MAAM,MAAM,KAAK;GACvB,MAAM,OAAO,mBAAmB;GAChC,MAAM,OAAO,UAAU,OAAO,WAAW;GACzC,IAAI,gBAAgB,WAAW,iBAAiB;GAChD,IAAI,OAAO,WAAW,QAAQ;GAC9B,IAAI;GACJ,IAAI;IACF,WAAW,MAAM,SAAS,gBACxB,OACA,KACA;KACE,UAAU,OAAO,cAAc,UAAU,KAAK,OAAO;KACrD,aAAa,KAAK,WAAW;KAC7B,UAAU,OAAO;KACjB,aAAa,OAAO;KACpB,GAAI,OAAO,qBAAqB,KAAA,IAC5B,EAAE,kBAAkB,OAAO,iBAAiB,IAC5C,CAAC;IACP,GACA,MACA,UACF,GAAG;KACD,IAAI,MAAM,SAAS,oBAAoB,kBAAkB,IACvD,gBAAgB,MAAM;KAExB,IACE,MAAM,SAAS,UAAU,wBACzB,mBAAmB,SACnB,MAAM,kBAAkB,eAExB,QAAQ,MAAM;KAEhB,IACE,MAAM,SAAS,kBACf,MAAM,kBAAkB,eAExB,QAAQ,MAAM;KAEhB,cAAc,kBAAkB,GAAG,KAAK;IAC1C;GACF,UAAU;IACR,KAAK,QAAQ;GACf;GACA,OAAO,KAAK,MAAM,KAAK,GAAG,KAAK,KAAK;GACpC,IAAI,KAAK,OACP,OAAO,KAAK,QAAQ,OAAO,KAAK,QAC5B,cAAc,OAAO,KAAK,OAAO,KAAK,KAAK,IAC3C,KAAK;GAEX,OAAO;IACL;IACA;IACA,GAAI,UAAU,KAAA,IAAY,EAAE,MAAM,IAAI,CAAC;IACvC,GAAI,KAAK,WAAW,SAAS,IAAI,EAAE,YAAY,KAAK,WAAW,IAAI,CAAC;GACtE;EACF;CACF,EAAE;AACJ"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import { ModelMessage, RunAgentResumeItem, SubagentStatus, UIMessage } from '../../../types.js';
|
|
2
|
+
type AnyMessage = UIMessage | ModelMessage;
|
|
3
|
+
export interface SubagentTurnChild {
|
|
4
|
+
subagentRunId: string;
|
|
5
|
+
name: string;
|
|
6
|
+
status: SubagentStatus;
|
|
7
|
+
parentToolCallId?: string;
|
|
8
|
+
/** The child's own messages from the earlier run. */
|
|
9
|
+
messages: Array<AnyMessage>;
|
|
10
|
+
/** Text the child wrote. */
|
|
11
|
+
text: string;
|
|
12
|
+
/** Resume entries for this child and its nested children. */
|
|
13
|
+
resume: Array<RunAgentResumeItem>;
|
|
14
|
+
}
|
|
15
|
+
/** The interrupted turn that a resume continues. */
|
|
16
|
+
export interface SubagentTurn {
|
|
17
|
+
/**
|
|
18
|
+
* Messages up to and including the last top-level user message. Child wire
|
|
19
|
+
* messages do not count.
|
|
20
|
+
*/
|
|
21
|
+
before: Array<AnyMessage>;
|
|
22
|
+
/** Direct children of the parent in that turn. */
|
|
23
|
+
children: Array<SubagentTurnChild>;
|
|
24
|
+
/** Resume entries that no child owns. They belong to the parent run. */
|
|
25
|
+
rest: Array<RunAgentResumeItem>;
|
|
26
|
+
/** The router plan of the earlier run, from the children's metadata. */
|
|
27
|
+
plan?: unknown;
|
|
28
|
+
}
|
|
29
|
+
/** Key of the router plan in `SUBAGENT_STARTED` metadata (`metadata.tanstack`). */
|
|
30
|
+
export declare const SUBAGENT_PLAN_KEY = "subagentPlan";
|
|
31
|
+
/**
|
|
32
|
+
* Find the children of the trailing assistant turn that `resume` answers.
|
|
33
|
+
* Returns undefined when no child owns a resume entry.
|
|
34
|
+
*/
|
|
35
|
+
export declare function readSubagentTurn(messages: ReadonlyArray<AnyMessage>, resume: ReadonlyArray<RunAgentResumeItem> | undefined): SubagentTurn | undefined;
|
|
36
|
+
export {};
|