@langchain/langgraph-sdk 1.11.1 → 1.11.2
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/client/stream/messages.cjs +54 -17
- package/dist/client/stream/messages.cjs.map +1 -1
- package/dist/client/stream/messages.d.cts +3 -0
- package/dist/client/stream/messages.d.cts.map +1 -1
- package/dist/client/stream/messages.d.ts +3 -0
- package/dist/client/stream/messages.d.ts.map +1 -1
- package/dist/client/stream/messages.js +54 -17
- package/dist/client/stream/messages.js.map +1 -1
- package/dist/stream/assembled-to-message.cjs +42 -8
- package/dist/stream/assembled-to-message.cjs.map +1 -1
- package/dist/stream/assembled-to-message.d.cts +11 -0
- package/dist/stream/assembled-to-message.d.cts.map +1 -1
- package/dist/stream/assembled-to-message.d.ts +11 -0
- package/dist/stream/assembled-to-message.d.ts.map +1 -1
- package/dist/stream/assembled-to-message.js +42 -8
- package/dist/stream/assembled-to-message.js.map +1 -1
- package/package.json +1 -1
|
@@ -89,6 +89,23 @@ function normalizeUsage(usage) {
|
|
|
89
89
|
total_tokens: usage.total_tokens ?? 0
|
|
90
90
|
};
|
|
91
91
|
}
|
|
92
|
+
function readRunId(data) {
|
|
93
|
+
const record = data;
|
|
94
|
+
return typeof record.run_id === "string" ? record.run_id : void 0;
|
|
95
|
+
}
|
|
96
|
+
function readUsage(data) {
|
|
97
|
+
const usage = data.usage;
|
|
98
|
+
return usage != null && typeof usage === "object" && !Array.isArray(usage) ? usage : void 0;
|
|
99
|
+
}
|
|
100
|
+
function readFinishMetadata(data) {
|
|
101
|
+
const record = data;
|
|
102
|
+
const isRecord = (value) => value != null && typeof value === "object" && !Array.isArray(value);
|
|
103
|
+
const merged = {
|
|
104
|
+
...isRecord(record.responseMetadata) ? record.responseMetadata : {},
|
|
105
|
+
...isRecord(record.metadata) ? record.metadata : {}
|
|
106
|
+
};
|
|
107
|
+
return Object.keys(merged).length > 0 ? merged : void 0;
|
|
108
|
+
}
|
|
92
109
|
/**
|
|
93
110
|
* Symbol keys for assembler → StreamingMessage communication.
|
|
94
111
|
* Module-private: invisible to external consumers, accessible to
|
|
@@ -291,10 +308,10 @@ var StreamingMessage = class {
|
|
|
291
308
|
}
|
|
292
309
|
async #assembleMessage() {
|
|
293
310
|
const contentBlocks = [];
|
|
294
|
-
let id;
|
|
295
|
-
let usage;
|
|
296
|
-
let metadata = {};
|
|
297
|
-
let finishReason;
|
|
311
|
+
let id = this.id;
|
|
312
|
+
let usage = normalizeUsage(this.assembled.usage);
|
|
313
|
+
let metadata = { ...this.assembled.finishMetadata ?? {} };
|
|
314
|
+
let finishReason = this.assembled.finishReason;
|
|
298
315
|
for await (const event of this.#events) switch (event.event) {
|
|
299
316
|
case "message-start":
|
|
300
317
|
id = event.id ?? id;
|
|
@@ -311,23 +328,32 @@ var StreamingMessage = class {
|
|
|
311
328
|
case "content-block-finish":
|
|
312
329
|
contentBlocks[event.index] = event.content;
|
|
313
330
|
break;
|
|
314
|
-
case "message-finish":
|
|
331
|
+
case "message-finish": {
|
|
315
332
|
finishReason = event.reason;
|
|
316
333
|
if (event.usage) usage = normalizeUsage(event.usage);
|
|
317
|
-
|
|
334
|
+
const finishMetadata = readFinishMetadata(event);
|
|
335
|
+
if (finishMetadata != null) metadata = {
|
|
318
336
|
...metadata,
|
|
319
|
-
...
|
|
337
|
+
...finishMetadata
|
|
320
338
|
};
|
|
321
339
|
break;
|
|
340
|
+
}
|
|
322
341
|
default: break;
|
|
323
342
|
}
|
|
324
343
|
return new _langchain_core_messages.AIMessage({
|
|
325
344
|
id,
|
|
326
345
|
content: contentBlocks.filter((block) => block != null),
|
|
327
346
|
usage_metadata: usage,
|
|
347
|
+
additional_kwargs: {
|
|
348
|
+
namespace: this.namespace,
|
|
349
|
+
...this.node != null ? { node: this.node } : {},
|
|
350
|
+
...this.assembled.runId != null ? { run_id: this.assembled.runId } : {},
|
|
351
|
+
...this.metadata != null ? { metadata: this.metadata } : {},
|
|
352
|
+
...this.assembled.usage != null ? { usage: this.assembled.usage } : {}
|
|
353
|
+
},
|
|
328
354
|
response_metadata: {
|
|
355
|
+
...finishReason != null ? { finish_reason: finishReason } : {},
|
|
329
356
|
...metadata,
|
|
330
|
-
...finishReason ? { finish_reason: finishReason } : {},
|
|
331
357
|
output_version: "v1"
|
|
332
358
|
}
|
|
333
359
|
});
|
|
@@ -415,8 +441,9 @@ function applyContentDelta(target, delta) {
|
|
|
415
441
|
function messageKeyFor(event) {
|
|
416
442
|
const { namespace, node, data } = event.params;
|
|
417
443
|
const namespaceKey = namespace.join("/");
|
|
444
|
+
const runId = readRunId(data);
|
|
418
445
|
const messageId = data.event === "message-start" ? data.id ?? "" : "";
|
|
419
|
-
return `${namespaceKey}::${node ?? ""}::${messageId}`;
|
|
446
|
+
return `${namespaceKey}::${node ?? ""}::${runId ?? messageId}`;
|
|
420
447
|
}
|
|
421
448
|
function toChatModelStreamEvent(event) {
|
|
422
449
|
return event.params.data;
|
|
@@ -436,6 +463,7 @@ var MessageAssembler = class {
|
|
|
436
463
|
consume(event) {
|
|
437
464
|
const data = event.params.data;
|
|
438
465
|
const namespaceNodeKey = `${event.params.namespace.join("/")}::${event.params.node ?? ""}`;
|
|
466
|
+
const runId = readRunId(data);
|
|
439
467
|
if (data.event === "message-start") {
|
|
440
468
|
const key = messageKeyFor(event);
|
|
441
469
|
this.activeByNamespaceNode.set(namespaceNodeKey, key);
|
|
@@ -443,6 +471,8 @@ var MessageAssembler = class {
|
|
|
443
471
|
id: data.id,
|
|
444
472
|
namespace: [...event.params.namespace],
|
|
445
473
|
node: event.params.node,
|
|
474
|
+
runId,
|
|
475
|
+
usage: readUsage(data),
|
|
446
476
|
metadata: data.metadata,
|
|
447
477
|
blocks: []
|
|
448
478
|
};
|
|
@@ -454,14 +484,18 @@ var MessageAssembler = class {
|
|
|
454
484
|
event
|
|
455
485
|
};
|
|
456
486
|
}
|
|
457
|
-
const
|
|
487
|
+
const runKey = runId != null ? `${namespaceNodeKey}::${runId}` : void 0;
|
|
488
|
+
const fallbackKey = this.activeByNamespaceNode.get(namespaceNodeKey);
|
|
489
|
+
const fallbackMessage = fallbackKey != null ? this.activeMessages.get(fallbackKey) : void 0;
|
|
490
|
+
const activeKey = (runKey != null && this.activeMessages.has(runKey) ? runKey : void 0) ?? (runId == null || fallbackMessage?.runId == null ? fallbackKey : void 0);
|
|
458
491
|
if (!activeKey) {
|
|
459
|
-
const syntheticKey = `${namespaceNodeKey}::`;
|
|
492
|
+
const syntheticKey = runKey ?? `${namespaceNodeKey}::`;
|
|
460
493
|
this.activeByNamespaceNode.set(namespaceNodeKey, syntheticKey);
|
|
461
494
|
const synthetic = {
|
|
462
495
|
id: data.id,
|
|
463
496
|
namespace: [...event.params.namespace],
|
|
464
497
|
node: event.params.node,
|
|
498
|
+
runId,
|
|
465
499
|
blocks: []
|
|
466
500
|
};
|
|
467
501
|
this.activeMessages.set(syntheticKey, synthetic);
|
|
@@ -519,10 +553,10 @@ var MessageAssembler = class {
|
|
|
519
553
|
};
|
|
520
554
|
}
|
|
521
555
|
case "message-finish":
|
|
522
|
-
message.usage = data.usage;
|
|
523
|
-
message.finishMetadata = data
|
|
524
|
-
|
|
525
|
-
this.
|
|
556
|
+
message.usage = readUsage(data) ?? message.usage;
|
|
557
|
+
message.finishMetadata = readFinishMetadata(data);
|
|
558
|
+
message.finishReason = data.reason;
|
|
559
|
+
this.clearActiveMessage(activeKey, namespaceNodeKey);
|
|
526
560
|
this.clearBlockIndexAliases(activeKey);
|
|
527
561
|
return {
|
|
528
562
|
kind: "message-finish",
|
|
@@ -535,8 +569,7 @@ var MessageAssembler = class {
|
|
|
535
569
|
message: data.message,
|
|
536
570
|
code: data.code
|
|
537
571
|
};
|
|
538
|
-
this.
|
|
539
|
-
this.activeByNamespaceNode.delete(namespaceNodeKey);
|
|
572
|
+
this.clearActiveMessage(activeKey, namespaceNodeKey);
|
|
540
573
|
this.clearBlockIndexAliases(activeKey);
|
|
541
574
|
return {
|
|
542
575
|
kind: "message-error",
|
|
@@ -546,6 +579,10 @@ var MessageAssembler = class {
|
|
|
546
579
|
};
|
|
547
580
|
}
|
|
548
581
|
}
|
|
582
|
+
clearActiveMessage(activeKey, namespaceNodeKey) {
|
|
583
|
+
this.activeMessages.delete(activeKey);
|
|
584
|
+
if (this.activeByNamespaceNode.get(namespaceNodeKey) === activeKey) this.activeByNamespaceNode.delete(namespaceNodeKey);
|
|
585
|
+
}
|
|
549
586
|
resolveBlockIndex(activeKey, blocks, protocolIndex, blockType) {
|
|
550
587
|
const current = blocks[protocolIndex];
|
|
551
588
|
if (current == null || current.type === blockType || areCompatibleBlockTypes(current.type, blockType)) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"messages.cjs","names":["#events","MultiCursorBuffer","#textPromise","#reasoningPromise","#resolveText","#resolveReasoning","#textChunks","#textWaiters","#textDone","#reasoningChunks","#reasoningWaiters","#reasoningDone","#usageIterator","#assembleMessage","AIMessage","#assembler","#activeStreaming"],"sources":["../../../src/client/stream/messages.ts"],"sourcesContent":["import {\n AIMessage,\n type ContentBlock as CoreContentBlock,\n type UsageMetadata,\n} from \"@langchain/core/messages\";\nimport type {\n ChatModelStreamEvent,\n ContentBlockDelta as CoreContentBlockDelta,\n} from \"@langchain/core/language_models/event\";\nimport type {\n ContentBlock,\n FinalizedContentBlock,\n MessageMetadata,\n MessagesEvent,\n UsageInfo,\n} from \"@langchain/protocol\";\n\nimport { MultiCursorBuffer } from \"./multi-cursor-buffer.js\";\n\ntype TextContentStream = AsyncIterable<string> &\n PromiseLike<string> & { full: AsyncIterable<string> };\n\ntype UsageMetadataStream = AsyncIterable<UsageMetadata> &\n PromiseLike<UsageMetadata | undefined>;\n\ntype ToolCallsStream = AsyncIterable<CoreContentBlock.Tools.ToolCall> &\n PromiseLike<Array<CoreContentBlock.Tools.ToolCall>> & {\n full: AsyncIterable<Array<CoreContentBlock.Tools.ToolCall>>;\n };\n\nfunction applyCoreContentDelta(\n target: CoreContentBlock,\n delta: CoreContentBlock\n): CoreContentBlock {\n if (target.type !== delta.type) {\n return structuredClone(delta);\n }\n\n switch (delta.type) {\n case \"text\":\n return {\n ...target,\n ...delta,\n text: `${\"text\" in target ? target.text : \"\"}${delta.text}`,\n } as CoreContentBlock;\n case \"reasoning\":\n return {\n ...target,\n ...delta,\n reasoning: `${\"reasoning\" in target ? target.reasoning : \"\"}${delta.reasoning}`,\n } as CoreContentBlock;\n case \"tool_call_chunk\":\n case \"server_tool_call_chunk\": {\n const merged = { ...target, ...delta } as Record<string, unknown>;\n if (delta.id == null && \"id\" in target && target.id != null) {\n merged.id = target.id;\n }\n if (delta.name == null && \"name\" in target && target.name != null) {\n merged.name = target.name;\n }\n merged.args = `${(\"args\" in target ? target.args : \"\") ?? \"\"}${delta.args ?? \"\"}`;\n return merged as unknown as CoreContentBlock;\n }\n default:\n return { ...target, ...delta } as CoreContentBlock;\n }\n}\n\nfunction coreContentBlockFromDelta(\n delta: CoreContentBlockDelta,\n current?: CoreContentBlock\n): CoreContentBlock {\n switch (delta.type) {\n case \"text-delta\":\n return { type: \"text\", text: delta.text } as CoreContentBlock;\n case \"reasoning-delta\":\n return {\n type: \"reasoning\",\n reasoning: delta.reasoning,\n } as CoreContentBlock;\n case \"data-delta\": {\n const merged = { ...(current ?? {}), data: delta.data } as Record<\n string,\n unknown\n >;\n if (delta.encoding) merged.encoding = delta.encoding;\n return merged as unknown as CoreContentBlock;\n }\n case \"block-delta\":\n return delta.fields as CoreContentBlock;\n }\n}\n\nfunction applyCoreEventDelta(\n current: CoreContentBlock | undefined,\n event: Extract<ChatModelStreamEvent, { event: \"content-block-delta\" }> & {\n content?: CoreContentBlock;\n }\n): CoreContentBlock {\n if (event.content) {\n return current\n ? applyCoreContentDelta(current, event.content)\n : event.content;\n }\n\n switch (event.delta.type) {\n case \"text-delta\":\n if (current?.type === \"text\") {\n return {\n ...current,\n text: `${\"text\" in current ? current.text : \"\"}${event.delta.text}`,\n } as CoreContentBlock;\n }\n return coreContentBlockFromDelta(event.delta, current);\n case \"reasoning-delta\":\n if (current?.type === \"reasoning\") {\n return {\n ...current,\n reasoning: `${\"reasoning\" in current ? current.reasoning : \"\"}${event.delta.reasoning}`,\n } as CoreContentBlock;\n }\n return coreContentBlockFromDelta(event.delta, current);\n case \"data-delta\": {\n const merged = { ...(current ?? {}) } as Record<string, unknown>;\n merged.data = `${(merged.data as string | undefined) ?? \"\"}${event.delta.data}`;\n if (event.delta.encoding) merged.encoding = event.delta.encoding;\n return merged as unknown as CoreContentBlock;\n }\n case \"block-delta\":\n return {\n ...(current ?? {}),\n ...event.delta.fields,\n } as CoreContentBlock;\n }\n}\n\nfunction normalizeUsage(\n usage: UsageInfo | Partial<UsageMetadata> | undefined\n): UsageMetadata | undefined {\n if (!usage) return undefined;\n return {\n ...usage,\n input_tokens: usage.input_tokens ?? 0,\n output_tokens: usage.output_tokens ?? 0,\n total_tokens: usage.total_tokens ?? 0,\n };\n}\n\n/**\n * Mutable view of a streamed message as message and content-block events are\n * assembled into a single structure.\n */\nexport interface AssembledMessage {\n id: string;\n namespace: string[];\n blocks: ContentBlock[];\n node?: string;\n usage?: UsageInfo;\n metadata?: MessageMetadata;\n finishMetadata?: Record<string, any>;\n error?: {\n message: string;\n code?: string;\n };\n}\n\n/**\n * Symbol keys for assembler → StreamingMessage communication.\n * Module-private: invisible to external consumers, accessible to\n * {@link StreamingMessageAssembler} within this file.\n */\nconst PUSH_TEXT: unique symbol = Symbol(\"pushText\");\nconst PUSH_REASONING: unique symbol = Symbol(\"pushReasoning\");\nconst PUSH_EVENT: unique symbol = Symbol(\"pushEvent\");\nconst UPDATE_CONTEXT: unique symbol = Symbol(\"updateContext\");\nconst FINISH: unique symbol = Symbol(\"finish\");\nconst ERROR: unique symbol = Symbol(\"error\");\n\n/**\n * Live streaming view of a single message lifecycle, matching the\n * in-process `ChatModelStream` dual-interface pattern.\n *\n * - `text` / `reasoning`: iterate for streaming deltas, or await for\n * the full concatenated string after the message completes.\n * - `usage`: promise that resolves with token usage on message-finish.\n * - `blocks`: the assembled content blocks (updated as deltas arrive).\n *\n * Created by {@link StreamingMessageAssembler} and yielded by\n * the `session.messages` lazy getter.\n */\nexport class StreamingMessage\n implements AsyncIterable<ChatModelStreamEvent>, PromiseLike<AIMessage>\n{\n readonly id: string;\n readonly namespace: string[];\n node: string | undefined;\n readonly metadata: MessageMetadata | undefined;\n readonly assembled: AssembledMessage;\n readonly #events = new MultiCursorBuffer<ChatModelStreamEvent>();\n\n #textChunks: string[] = [];\n #reasoningChunks: string[] = [];\n #textWaiters: Array<() => void> = [];\n #reasoningWaiters: Array<() => void> = [];\n #textDone = false;\n #reasoningDone = false;\n\n #resolveText!: (v: string) => void;\n #resolveReasoning!: (v: string) => void;\n readonly #textPromise: Promise<string>;\n readonly #reasoningPromise: Promise<string>;\n\n constructor(assembled: AssembledMessage) {\n this.id = assembled.id;\n this.assembled = assembled;\n this.namespace = assembled.namespace;\n this.node = assembled.node;\n this.metadata = assembled.metadata;\n this.#textPromise = new Promise<string>((r) => {\n this.#resolveText = r;\n });\n this.#reasoningPromise = new Promise<string>((r) => {\n this.#resolveReasoning = r;\n });\n }\n\n get text(): TextContentStream {\n const chunks = this.#textChunks;\n const waiters = this.#textWaiters;\n const getDone = () => this.#textDone;\n let cursor = 0;\n return {\n [Symbol.asyncIterator]() {\n return {\n async next(): Promise<IteratorResult<string>> {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n };\n },\n then: this.#textPromise.then.bind(this.#textPromise),\n full: {\n async *[Symbol.asyncIterator]() {\n let accumulated = \"\";\n for await (const chunk of {\n [Symbol.asyncIterator]: () =>\n ({\n next: async (): Promise<IteratorResult<string>> => {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n }) satisfies AsyncIterator<string>,\n }) {\n accumulated += chunk;\n yield accumulated;\n }\n },\n },\n };\n }\n\n get reasoning(): TextContentStream {\n const chunks = this.#reasoningChunks;\n const waiters = this.#reasoningWaiters;\n const getDone = () => this.#reasoningDone;\n let cursor = 0;\n return {\n [Symbol.asyncIterator]() {\n return {\n async next(): Promise<IteratorResult<string>> {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n };\n },\n then: this.#reasoningPromise.then.bind(this.#reasoningPromise),\n full: {\n async *[Symbol.asyncIterator]() {\n let accumulated = \"\";\n for await (const chunk of {\n [Symbol.asyncIterator]: () =>\n ({\n next: async (): Promise<IteratorResult<string>> => {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n }) satisfies AsyncIterator<string>,\n }) {\n accumulated += chunk;\n yield accumulated;\n }\n },\n },\n };\n }\n\n get usage(): UsageMetadataStream {\n const promise = (async () => {\n let usage: UsageMetadata | undefined;\n for await (const snapshot of this.#usageIterator()) {\n usage = snapshot;\n }\n return usage;\n })();\n return {\n [Symbol.asyncIterator]: () => this.#usageIterator(),\n then: promise.then.bind(promise),\n };\n }\n\n get toolCalls(): ToolCallsStream {\n const events = this.#events;\n const iterator = async function* () {\n for await (const event of events) {\n if (\n event.event === \"content-block-finish\" &&\n event.content.type === \"tool_call\"\n ) {\n yield event.content as CoreContentBlock.Tools.ToolCall;\n }\n }\n };\n return {\n [Symbol.asyncIterator]: iterator,\n then: async (onfulfilled, onrejected) => {\n try {\n const calls: CoreContentBlock.Tools.ToolCall[] = [];\n for await (const call of iterator()) calls.push(call);\n return onfulfilled ? onfulfilled(calls) : (calls as never);\n } catch (err) {\n if (onrejected) return onrejected(err);\n throw err;\n }\n },\n full: {\n async *[Symbol.asyncIterator]() {\n const calls: CoreContentBlock.Tools.ToolCall[] = [];\n for await (const call of iterator()) {\n calls.push(call);\n yield [...calls];\n }\n },\n },\n };\n }\n\n get output(): PromiseLike<AIMessage> {\n return { then: (onf, onr) => this.#assembleMessage().then(onf, onr) };\n }\n\n get blocks(): ContentBlock[] {\n return this.assembled.blocks;\n }\n\n [Symbol.asyncIterator](): AsyncIterator<ChatModelStreamEvent> {\n return this.#events[Symbol.asyncIterator]();\n }\n\n then<TResult1 = AIMessage, TResult2 = never>(\n onfulfilled?:\n | ((value: AIMessage) => TResult1 | PromiseLike<TResult1>)\n | null,\n onrejected?: ((reason: unknown) => TResult2 | PromiseLike<TResult2>) | null\n ): PromiseLike<TResult1 | TResult2> {\n return this.#assembleMessage().then(onfulfilled, onrejected);\n }\n\n async *#usageIterator(): AsyncGenerator<UsageMetadata> {\n for await (const event of this.#events) {\n if (event.event === \"message-start\" && event.usage) {\n yield normalizeUsage(event.usage)!;\n } else if (event.event === \"message-finish\" && event.usage) {\n yield normalizeUsage(event.usage)!;\n }\n }\n }\n\n async #assembleMessage(): Promise<AIMessage> {\n const contentBlocks: Array<CoreContentBlock | undefined> = [];\n let id: string | undefined;\n let usage: UsageMetadata | undefined;\n let metadata: Record<string, unknown> = {};\n let finishReason: string | undefined;\n\n for await (const event of this.#events) {\n switch (event.event) {\n case \"message-start\":\n id = event.id ?? id;\n if (event.usage) usage = normalizeUsage(event.usage);\n break;\n case \"content-block-start\":\n contentBlocks[event.index] = event.content;\n break;\n case \"content-block-delta\": {\n const current = contentBlocks[event.index];\n contentBlocks[event.index] = applyCoreEventDelta(current, event);\n break;\n }\n case \"content-block-finish\":\n contentBlocks[event.index] = event.content;\n break;\n case \"message-finish\":\n finishReason = event.reason;\n if (event.usage) usage = normalizeUsage(event.usage);\n if (event.responseMetadata) {\n metadata = {\n ...metadata,\n ...event.responseMetadata,\n };\n }\n break;\n default:\n break;\n }\n }\n\n return new AIMessage({\n id,\n content: contentBlocks.filter(\n (block): block is CoreContentBlock => block != null\n ),\n usage_metadata: usage,\n response_metadata: {\n ...metadata,\n ...(finishReason ? { finish_reason: finishReason } : {}),\n output_version: \"v1\" as const,\n },\n });\n }\n\n [PUSH_EVENT](event: ChatModelStreamEvent): void {\n this.#events.push(event);\n }\n\n [UPDATE_CONTEXT](event: MessagesEvent): void {\n this.node = event.params.node ?? this.node;\n }\n\n [PUSH_TEXT](delta: string): void {\n this.#textChunks.push(delta);\n // Wake every caught-up iterator so each one advances its own cursor.\n // Iterators re-check `chunks.length` before delivering, which keeps\n // the cursor the single source of truth for what a consumer has seen.\n const pending = this.#textWaiters.splice(0, this.#textWaiters.length);\n for (const waiter of pending) waiter();\n }\n\n [PUSH_REASONING](delta: string): void {\n this.#reasoningChunks.push(delta);\n const pending = this.#reasoningWaiters.splice(\n 0,\n this.#reasoningWaiters.length\n );\n for (const waiter of pending) waiter();\n }\n\n [FINISH](): void {\n this.#textDone = true;\n this.#reasoningDone = true;\n this.#resolveText(this.#textChunks.join(\"\"));\n this.#resolveReasoning(this.#reasoningChunks.join(\"\"));\n const textPending = this.#textWaiters.splice(0, this.#textWaiters.length);\n for (const waiter of textPending) waiter();\n const reasoningPending = this.#reasoningWaiters.splice(\n 0,\n this.#reasoningWaiters.length\n );\n for (const waiter of reasoningPending) waiter();\n this.#events.close();\n }\n\n [ERROR](): void {\n this[FINISH]();\n }\n}\n\n/**\n * Public view yielded by message projections.\n *\n * `StreamingMessage` is PromiseLike so callers can still `await` a message\n * object directly, but TypeScript applies `Awaited<T>` to values produced by\n * `for await`. Exposing a non-thenable view keeps loop variables typed as the\n * streaming handle instead of as the finalized `AIMessage`.\n */\nexport type StreamingMessageHandle = Omit<StreamingMessage, \"then\">;\n\nexport function toStreamingMessageHandle(\n message: StreamingMessage\n): StreamingMessageHandle {\n return new Proxy(message, {\n get(target, prop) {\n if (prop === \"then\") return undefined;\n const value = Reflect.get(target, prop, target);\n return typeof value === \"function\" ? value.bind(target) : value;\n },\n has(target, prop) {\n if (prop === \"then\") return false;\n return prop in target;\n },\n }) as StreamingMessageHandle;\n}\n\n/**\n * Emitted by `MessageAssembler.consume()` to describe how a message changed in\n * response to a single protocol event.\n */\nexport type MessageAssemblyUpdate =\n | {\n kind: \"message-start\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"content-block-start\" | \"content-block-delta\";\n key: string;\n message: AssembledMessage;\n index: number;\n block: ContentBlock;\n event: MessagesEvent;\n }\n | {\n kind: \"content-block-finish\";\n key: string;\n message: AssembledMessage;\n index: number;\n block: FinalizedContentBlock;\n event: MessagesEvent;\n }\n | {\n kind: \"usage\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"message-finish\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"message-error\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n };\n\nfunction cloneBlock<T extends ContentBlock>(block: T): T {\n return structuredClone(block);\n}\n\nfunction blockFromDelta(\n delta: CoreContentBlockDelta,\n current?: ContentBlock\n): ContentBlock {\n return coreContentBlockFromDelta(\n delta,\n current as unknown as CoreContentBlock | undefined\n ) as unknown as ContentBlock;\n}\n\nfunction applyContentDelta(\n target: ContentBlock,\n delta: ContentBlock\n): ContentBlock {\n if (target.type !== delta.type) {\n return cloneBlock(delta);\n }\n\n switch (delta.type) {\n case \"text\":\n return {\n ...target,\n ...delta,\n text: `${\"text\" in target ? target.text : \"\"}${delta.text}`,\n } as ContentBlock;\n case \"reasoning\":\n return {\n ...target,\n ...delta,\n reasoning: `${\"reasoning\" in target ? target.reasoning : \"\"}${delta.reasoning}`,\n } as ContentBlock;\n case \"tool_call_chunk\":\n case \"server_tool_call_chunk\": {\n // Spread target first, then delta — but preserve target's\n // ``id``/``name`` when the delta explicitly sets them to\n // null/undefined. Some providers (notably Anthropic via the\n // langchain-core compat bridge) only attach the tool-call\n // identifiers to the first ``content-block-start`` chunk; every\n // subsequent ``input_json_delta`` chunk carries ``id=null,\n // name=null``. A naive ``{...target, ...delta}`` spread\n // overwrites the captured identifiers with null, which in turn\n // makes downstream consumers (e.g. ``extractToolCallChunks`` in\n // ``assembled-to-message.ts``) drop the chunk on the floor until\n // the final ``content-block-finish`` event promotes it to a\n // finalized ``tool_call`` — causing tool-call cards to appear\n // all-at-once at the end of the turn instead of incrementally.\n const merged = { ...target, ...delta } as Record<string, unknown>;\n if (delta.id == null && \"id\" in target && target.id != null) {\n merged.id = target.id;\n }\n if (delta.name == null && \"name\" in target && target.name != null) {\n merged.name = target.name;\n }\n merged.args = `${(\"args\" in target ? target.args : \"\") ?? \"\"}${delta.args ?? \"\"}`;\n return merged as unknown as ContentBlock;\n }\n default:\n return {\n ...target,\n ...delta,\n } as ContentBlock;\n }\n}\n\nfunction messageKeyFor(event: MessagesEvent): string {\n const { namespace, node, data } = event.params;\n const namespaceKey = namespace.join(\"/\");\n const messageId = data.event === \"message-start\" ? (data.id ?? \"\") : \"\";\n return `${namespaceKey}::${node ?? \"\"}::${messageId}`;\n}\n\nfunction toChatModelStreamEvent(event: MessagesEvent): ChatModelStreamEvent {\n return event.params.data as unknown as ChatModelStreamEvent;\n}\n\n/**\n * Incrementally assembles `messages` events into complete message objects.\n */\nexport class MessageAssembler {\n private readonly activeMessages = new Map<string, AssembledMessage>();\n private readonly activeByNamespaceNode = new Map<string, string>();\n private readonly blockIndexByProtocolIndexAndType = new Map<string, number>();\n\n /**\n * Applies a single message event and returns the resulting assembly update.\n *\n * @param event - Incoming `messages` event to fold into the assembler state.\n */\n consume(event: MessagesEvent): MessageAssemblyUpdate {\n const data = event.params.data;\n const namespaceNodeKey = `${event.params.namespace.join(\"/\")}::${event.params.node ?? \"\"}`;\n\n if (data.event === \"message-start\") {\n const key = messageKeyFor(event);\n this.activeByNamespaceNode.set(namespaceNodeKey, key);\n const message: AssembledMessage = {\n id: data.id,\n namespace: [...event.params.namespace],\n node: event.params.node,\n metadata: data.metadata,\n blocks: [],\n };\n this.activeMessages.set(key, message);\n return { kind: \"message-start\", key, message, event };\n }\n\n const activeKey = this.activeByNamespaceNode.get(namespaceNodeKey);\n if (!activeKey) {\n // A continuation event (delta/finish/error) arrived without a\n // prior `message-start`. This can happen on late-attaching\n // subscriptions when the server has already trimmed the\n // `message-start` from its replay buffer. Synthesize a minimal\n // active message so the assembler can still fold subsequent\n // events instead of hard-failing the caller.\n const syntheticKey = `${namespaceNodeKey}::`;\n this.activeByNamespaceNode.set(namespaceNodeKey, syntheticKey);\n const synthetic: AssembledMessage = {\n id: data.id,\n namespace: [...event.params.namespace],\n node: event.params.node,\n blocks: [],\n };\n this.activeMessages.set(syntheticKey, synthetic);\n return this.consume(event);\n }\n\n const message = this.activeMessages.get(activeKey);\n if (!message) {\n throw new Error(`No active message state found for key ${activeKey}`);\n }\n\n if ((data as { event?: string }).event === \"usage\") {\n message.usage = (data as { usage?: UsageInfo }).usage;\n return {\n kind: \"usage\",\n key: activeKey,\n message,\n event,\n };\n }\n\n switch (data.event) {\n case \"content-block-start\": {\n message.blocks[data.index] = cloneBlock(data.content);\n this.blockIndexByProtocolIndexAndType.set(\n blockIndexKey(activeKey, data.index, data.content.type),\n data.index\n );\n return {\n kind: \"content-block-start\",\n key: activeKey,\n message,\n index: data.index,\n block: data.content,\n event,\n };\n }\n case \"content-block-delta\": {\n const deltaEvent = data as typeof data & {\n content?: ContentBlock;\n delta?: CoreContentBlockDelta;\n };\n const deltaBlock =\n deltaEvent.content ??\n (deltaEvent.delta != null\n ? blockFromDelta(deltaEvent.delta, message.blocks[data.index])\n : undefined);\n if (deltaBlock == null) {\n throw new Error(\"Received content-block-delta without content\");\n }\n const targetIndex = this.resolveBlockIndex(\n activeKey,\n message.blocks,\n data.index,\n deltaBlock.type\n );\n const current = message.blocks[targetIndex];\n message.blocks[targetIndex] =\n deltaEvent.content != null\n ? current == null\n ? cloneBlock(deltaEvent.content)\n : applyContentDelta(current, deltaEvent.content)\n : (applyCoreEventDelta(\n current as CoreContentBlock | undefined,\n data as unknown as Extract<\n ChatModelStreamEvent,\n { event: \"content-block-delta\" }\n >\n ) as unknown as ContentBlock);\n return {\n kind: \"content-block-delta\",\n key: activeKey,\n message,\n index: targetIndex,\n block: deltaBlock,\n event,\n };\n }\n case \"content-block-finish\": {\n const targetIndex = this.resolveFinishBlockIndex(\n activeKey,\n data.index,\n data.content.type\n );\n message.blocks[targetIndex] = cloneBlock(data.content);\n return {\n kind: \"content-block-finish\",\n key: activeKey,\n message,\n index: targetIndex,\n block: data.content,\n event,\n };\n }\n case \"message-finish\": {\n message.usage = data.usage;\n message.finishMetadata = data.responseMetadata;\n this.activeMessages.delete(activeKey);\n this.activeByNamespaceNode.delete(namespaceNodeKey);\n this.clearBlockIndexAliases(activeKey);\n return {\n kind: \"message-finish\",\n key: activeKey,\n message: structuredClone(message),\n event,\n };\n }\n case \"error\": {\n message.error = { message: data.message, code: data.code };\n this.activeMessages.delete(activeKey);\n this.activeByNamespaceNode.delete(namespaceNodeKey);\n this.clearBlockIndexAliases(activeKey);\n return {\n kind: \"message-error\",\n key: activeKey,\n message: structuredClone(message),\n event,\n };\n }\n }\n }\n\n private resolveBlockIndex(\n activeKey: string,\n blocks: ContentBlock[],\n protocolIndex: number,\n blockType: string\n ): number {\n const current = blocks[protocolIndex];\n if (\n current == null ||\n current.type === blockType ||\n areCompatibleBlockTypes(current.type, blockType)\n ) {\n this.blockIndexByProtocolIndexAndType.set(\n blockIndexKey(activeKey, protocolIndex, blockType),\n protocolIndex\n );\n return protocolIndex;\n }\n\n const key = blockIndexKey(activeKey, protocolIndex, blockType);\n const existing = this.blockIndexByProtocolIndexAndType.get(key);\n if (existing != null) return existing;\n\n const nextIndex = blocks.length;\n this.blockIndexByProtocolIndexAndType.set(key, nextIndex);\n return nextIndex;\n }\n\n private resolveFinishBlockIndex(\n activeKey: string,\n protocolIndex: number,\n blockType: string\n ): number {\n const key = blockIndexKey(activeKey, protocolIndex, blockType);\n const existing = this.blockIndexByProtocolIndexAndType.get(key);\n if (existing != null) return existing;\n\n this.blockIndexByProtocolIndexAndType.set(key, protocolIndex);\n return protocolIndex;\n }\n\n private clearBlockIndexAliases(activeKey: string): void {\n const prefix = `${activeKey}::`;\n for (const key of this.blockIndexByProtocolIndexAndType.keys()) {\n if (key.startsWith(prefix))\n this.blockIndexByProtocolIndexAndType.delete(key);\n }\n }\n}\n\nfunction blockIndexKey(\n activeKey: string,\n protocolIndex: number,\n blockType: string\n): string {\n return `${activeKey}::${protocolIndex}::${blockType}`;\n}\n\nfunction areCompatibleBlockTypes(\n currentType: string,\n nextType: string\n): boolean {\n const toolCallTypes = new Set([\n \"tool_call\",\n \"tool_call_chunk\",\n \"tool_use\",\n \"input_json_delta\",\n ]);\n const serverToolCallTypes = new Set([\n \"server_tool_call\",\n \"server_tool_call_chunk\",\n ]);\n\n return (\n (toolCallTypes.has(currentType) && toolCallTypes.has(nextType)) ||\n (serverToolCallTypes.has(currentType) && serverToolCallTypes.has(nextType))\n );\n}\n\n/**\n * Assembles `messages` events into {@link StreamingMessage} instances\n * with live text/reasoning delta streams, matching the in-process\n * `ChatModelStream` dual-interface pattern.\n */\nexport class StreamingMessageAssembler {\n readonly #assembler = new MessageAssembler();\n readonly #activeStreaming = new Map<string, StreamingMessage>();\n\n /**\n * Folds a single event and returns a new {@link StreamingMessage}\n * when a `message-start` is seen, or `undefined` for continuation\n * events (deltas, finish, error).\n */\n consume(event: MessagesEvent): StreamingMessage | undefined {\n const update = this.#assembler.consume(event);\n if (update == null) return undefined;\n\n switch (update.kind) {\n case \"message-start\": {\n const streaming = new StreamingMessage(update.message);\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n this.#activeStreaming.set(update.key, streaming);\n return streaming;\n }\n case \"content-block-start\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n if (\n streaming &&\n update.block.type === \"text\" &&\n \"text\" in update.block &&\n update.block.text\n ) {\n streaming[PUSH_TEXT](update.block.text);\n }\n if (\n streaming &&\n update.block.type === \"reasoning\" &&\n \"reasoning\" in update.block &&\n update.block.reasoning\n ) {\n streaming[PUSH_REASONING](update.block.reasoning);\n }\n return undefined;\n }\n case \"content-block-delta\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (!streaming) return undefined;\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n if (update.block.type === \"text\" && \"text\" in update.block) {\n streaming[PUSH_TEXT](update.block.text);\n }\n if (update.block.type === \"reasoning\" && \"reasoning\" in update.block) {\n streaming[PUSH_REASONING](update.block.reasoning);\n }\n return undefined;\n }\n case \"content-block-finish\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n return undefined;\n }\n case \"usage\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n return undefined;\n }\n case \"message-finish\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n streaming[FINISH]();\n this.#activeStreaming.delete(update.key);\n }\n return undefined;\n }\n case \"message-error\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n streaming[ERROR]();\n this.#activeStreaming.delete(update.key);\n }\n return undefined;\n }\n }\n }\n}\n"],"mappings":";;;;AA8BA,SAAS,sBACP,QACA,OACkB;AAClB,KAAI,OAAO,SAAS,MAAM,KACxB,QAAO,gBAAgB,MAAM;AAG/B,SAAQ,MAAM,MAAd;EACE,KAAK,OACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,MAAM,GAAG,UAAU,SAAS,OAAO,OAAO,KAAK,MAAM;GACtD;EACH,KAAK,YACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,WAAW,GAAG,eAAe,SAAS,OAAO,YAAY,KAAK,MAAM;GACrE;EACH,KAAK;EACL,KAAK,0BAA0B;GAC7B,MAAM,SAAS;IAAE,GAAG;IAAQ,GAAG;IAAO;AACtC,OAAI,MAAM,MAAM,QAAQ,QAAQ,UAAU,OAAO,MAAM,KACrD,QAAO,KAAK,OAAO;AAErB,OAAI,MAAM,QAAQ,QAAQ,UAAU,UAAU,OAAO,QAAQ,KAC3D,QAAO,OAAO,OAAO;AAEvB,UAAO,OAAO,IAAI,UAAU,SAAS,OAAO,OAAO,OAAO,KAAK,MAAM,QAAQ;AAC7E,UAAO;;EAET,QACE,QAAO;GAAE,GAAG;GAAQ,GAAG;GAAO;;;AAIpC,SAAS,0BACP,OACA,SACkB;AAClB,SAAQ,MAAM,MAAd;EACE,KAAK,aACH,QAAO;GAAE,MAAM;GAAQ,MAAM,MAAM;GAAM;EAC3C,KAAK,kBACH,QAAO;GACL,MAAM;GACN,WAAW,MAAM;GAClB;EACH,KAAK,cAAc;GACjB,MAAM,SAAS;IAAE,GAAI,WAAW,EAAE;IAAG,MAAM,MAAM;IAAM;AAIvD,OAAI,MAAM,SAAU,QAAO,WAAW,MAAM;AAC5C,UAAO;;EAET,KAAK,cACH,QAAO,MAAM;;;AAInB,SAAS,oBACP,SACA,OAGkB;AAClB,KAAI,MAAM,QACR,QAAO,UACH,sBAAsB,SAAS,MAAM,QAAQ,GAC7C,MAAM;AAGZ,SAAQ,MAAM,MAAM,MAApB;EACE,KAAK;AACH,OAAI,SAAS,SAAS,OACpB,QAAO;IACL,GAAG;IACH,MAAM,GAAG,UAAU,UAAU,QAAQ,OAAO,KAAK,MAAM,MAAM;IAC9D;AAEH,UAAO,0BAA0B,MAAM,OAAO,QAAQ;EACxD,KAAK;AACH,OAAI,SAAS,SAAS,YACpB,QAAO;IACL,GAAG;IACH,WAAW,GAAG,eAAe,UAAU,QAAQ,YAAY,KAAK,MAAM,MAAM;IAC7E;AAEH,UAAO,0BAA0B,MAAM,OAAO,QAAQ;EACxD,KAAK,cAAc;GACjB,MAAM,SAAS,EAAE,GAAI,WAAW,EAAE,EAAG;AACrC,UAAO,OAAO,GAAI,OAAO,QAA+B,KAAK,MAAM,MAAM;AACzE,OAAI,MAAM,MAAM,SAAU,QAAO,WAAW,MAAM,MAAM;AACxD,UAAO;;EAET,KAAK,cACH,QAAO;GACL,GAAI,WAAW,EAAE;GACjB,GAAG,MAAM,MAAM;GAChB;;;AAIP,SAAS,eACP,OAC2B;AAC3B,KAAI,CAAC,MAAO,QAAO,KAAA;AACnB,QAAO;EACL,GAAG;EACH,cAAc,MAAM,gBAAgB;EACpC,eAAe,MAAM,iBAAiB;EACtC,cAAc,MAAM,gBAAgB;EACrC;;;;;;;AA0BH,MAAM,YAA2B,OAAO,WAAW;AACnD,MAAM,iBAAgC,OAAO,gBAAgB;AAC7D,MAAM,aAA4B,OAAO,YAAY;AACrD,MAAM,iBAAgC,OAAO,gBAAgB;AAC7D,MAAM,SAAwB,OAAO,SAAS;AAC9C,MAAM,QAAuB,OAAO,QAAQ;;;;;;;;;;;;;AAc5C,IAAa,mBAAb,MAEA;CACE;CACA;CACA;CACA;CACA;CACA,UAAmB,IAAIC,4BAAAA,mBAAyC;CAEhE,cAAwB,EAAE;CAC1B,mBAA6B,EAAE;CAC/B,eAAkC,EAAE;CACpC,oBAAuC,EAAE;CACzC,YAAY;CACZ,iBAAiB;CAEjB;CACA;CACA;CACA;CAEA,YAAY,WAA6B;AACvC,OAAK,KAAK,UAAU;AACpB,OAAK,YAAY;AACjB,OAAK,YAAY,UAAU;AAC3B,OAAK,OAAO,UAAU;AACtB,OAAK,WAAW,UAAU;AAC1B,QAAA,cAAoB,IAAI,SAAiB,MAAM;AAC7C,SAAA,cAAoB;IACpB;AACF,QAAA,mBAAyB,IAAI,SAAiB,MAAM;AAClD,SAAA,mBAAyB;IACzB;;CAGJ,IAAI,OAA0B;EAC5B,MAAM,SAAS,MAAA;EACf,MAAM,UAAU,MAAA;EAChB,MAAM,gBAAgB,MAAA;EACtB,IAAI,SAAS;AACb,SAAO;GACL,CAAC,OAAO,iBAAiB;AACvB,WAAO,EACL,MAAM,OAAwC;AAC5C,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP;;GAEH,MAAM,MAAA,YAAkB,KAAK,KAAK,MAAA,YAAkB;GACpD,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,IAAI,cAAc;AAClB,eAAW,MAAM,SAAS,GACvB,OAAO,uBACL,EACC,MAAM,YAA6C;AACjD,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP,GACJ,EAAE;AACD,oBAAe;AACf,WAAM;;MAGX;GACF;;CAGH,IAAI,YAA+B;EACjC,MAAM,SAAS,MAAA;EACf,MAAM,UAAU,MAAA;EAChB,MAAM,gBAAgB,MAAA;EACtB,IAAI,SAAS;AACb,SAAO;GACL,CAAC,OAAO,iBAAiB;AACvB,WAAO,EACL,MAAM,OAAwC;AAC5C,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP;;GAEH,MAAM,MAAA,iBAAuB,KAAK,KAAK,MAAA,iBAAuB;GAC9D,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,IAAI,cAAc;AAClB,eAAW,MAAM,SAAS,GACvB,OAAO,uBACL,EACC,MAAM,YAA6C;AACjD,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP,GACJ,EAAE;AACD,oBAAe;AACf,WAAM;;MAGX;GACF;;CAGH,IAAI,QAA6B;EAC/B,MAAM,WAAW,YAAY;GAC3B,IAAI;AACJ,cAAW,MAAM,YAAY,MAAA,eAAqB,CAChD,SAAQ;AAEV,UAAO;MACL;AACJ,SAAO;IACJ,OAAO,sBAAsB,MAAA,eAAqB;GACnD,MAAM,QAAQ,KAAK,KAAK,QAAQ;GACjC;;CAGH,IAAI,YAA6B;EAC/B,MAAM,SAAS,MAAA;EACf,MAAM,WAAW,mBAAmB;AAClC,cAAW,MAAM,SAAS,OACxB,KACE,MAAM,UAAU,0BAChB,MAAM,QAAQ,SAAS,YAEvB,OAAM,MAAM;;AAIlB,SAAO;IACJ,OAAO,gBAAgB;GACxB,MAAM,OAAO,aAAa,eAAe;AACvC,QAAI;KACF,MAAM,QAA2C,EAAE;AACnD,gBAAW,MAAM,QAAQ,UAAU,CAAE,OAAM,KAAK,KAAK;AACrD,YAAO,cAAc,YAAY,MAAM,GAAI;aACpC,KAAK;AACZ,SAAI,WAAY,QAAO,WAAW,IAAI;AACtC,WAAM;;;GAGV,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,MAAM,QAA2C,EAAE;AACnD,eAAW,MAAM,QAAQ,UAAU,EAAE;AACnC,WAAM,KAAK,KAAK;AAChB,WAAM,CAAC,GAAG,MAAM;;MAGrB;GACF;;CAGH,IAAI,SAAiC;AACnC,SAAO,EAAE,OAAO,KAAK,QAAQ,MAAA,iBAAuB,CAAC,KAAK,KAAK,IAAI,EAAE;;CAGvE,IAAI,SAAyB;AAC3B,SAAO,KAAK,UAAU;;CAGxB,CAAC,OAAO,iBAAsD;AAC5D,SAAO,MAAA,OAAa,OAAO,gBAAgB;;CAG7C,KACE,aAGA,YACkC;AAClC,SAAO,MAAA,iBAAuB,CAAC,KAAK,aAAa,WAAW;;CAG9D,QAAA,gBAAuD;AACrD,aAAW,MAAM,SAAS,MAAA,OACxB,KAAI,MAAM,UAAU,mBAAmB,MAAM,MAC3C,OAAM,eAAe,MAAM,MAAM;WACxB,MAAM,UAAU,oBAAoB,MAAM,MACnD,OAAM,eAAe,MAAM,MAAM;;CAKvC,OAAA,kBAA6C;EAC3C,MAAM,gBAAqD,EAAE;EAC7D,IAAI;EACJ,IAAI;EACJ,IAAI,WAAoC,EAAE;EAC1C,IAAI;AAEJ,aAAW,MAAM,SAAS,MAAA,OACxB,SAAQ,MAAM,OAAd;GACE,KAAK;AACH,SAAK,MAAM,MAAM;AACjB,QAAI,MAAM,MAAO,SAAQ,eAAe,MAAM,MAAM;AACpD;GACF,KAAK;AACH,kBAAc,MAAM,SAAS,MAAM;AACnC;GACF,KAAK,uBAAuB;IAC1B,MAAM,UAAU,cAAc,MAAM;AACpC,kBAAc,MAAM,SAAS,oBAAoB,SAAS,MAAM;AAChE;;GAEF,KAAK;AACH,kBAAc,MAAM,SAAS,MAAM;AACnC;GACF,KAAK;AACH,mBAAe,MAAM;AACrB,QAAI,MAAM,MAAO,SAAQ,eAAe,MAAM,MAAM;AACpD,QAAI,MAAM,iBACR,YAAW;KACT,GAAG;KACH,GAAG,MAAM;KACV;AAEH;GACF,QACE;;AAIN,SAAO,IAAIa,yBAAAA,UAAU;GACnB;GACA,SAAS,cAAc,QACpB,UAAqC,SAAS,KAChD;GACD,gBAAgB;GAChB,mBAAmB;IACjB,GAAG;IACH,GAAI,eAAe,EAAE,eAAe,cAAc,GAAG,EAAE;IACvD,gBAAgB;IACjB;GACF,CAAC;;CAGJ,CAAC,YAAY,OAAmC;AAC9C,QAAA,OAAa,KAAK,MAAM;;CAG1B,CAAC,gBAAgB,OAA4B;AAC3C,OAAK,OAAO,MAAM,OAAO,QAAQ,KAAK;;CAGxC,CAAC,WAAW,OAAqB;AAC/B,QAAA,WAAiB,KAAK,MAAM;EAI5B,MAAM,UAAU,MAAA,YAAkB,OAAO,GAAG,MAAA,YAAkB,OAAO;AACrE,OAAK,MAAM,UAAU,QAAS,SAAQ;;CAGxC,CAAC,gBAAgB,OAAqB;AACpC,QAAA,gBAAsB,KAAK,MAAM;EACjC,MAAM,UAAU,MAAA,iBAAuB,OACrC,GACA,MAAA,iBAAuB,OACxB;AACD,OAAK,MAAM,UAAU,QAAS,SAAQ;;CAGxC,CAAC,UAAgB;AACf,QAAA,WAAiB;AACjB,QAAA,gBAAsB;AACtB,QAAA,YAAkB,MAAA,WAAiB,KAAK,GAAG,CAAC;AAC5C,QAAA,iBAAuB,MAAA,gBAAsB,KAAK,GAAG,CAAC;EACtD,MAAM,cAAc,MAAA,YAAkB,OAAO,GAAG,MAAA,YAAkB,OAAO;AACzE,OAAK,MAAM,UAAU,YAAa,SAAQ;EAC1C,MAAM,mBAAmB,MAAA,iBAAuB,OAC9C,GACA,MAAA,iBAAuB,OACxB;AACD,OAAK,MAAM,UAAU,iBAAkB,SAAQ;AAC/C,QAAA,OAAa,OAAO;;CAGtB,CAAC,SAAe;AACd,OAAK,SAAS;;;AAclB,SAAgB,yBACd,SACwB;AACxB,QAAO,IAAI,MAAM,SAAS;EACxB,IAAI,QAAQ,MAAM;AAChB,OAAI,SAAS,OAAQ,QAAO,KAAA;GAC5B,MAAM,QAAQ,QAAQ,IAAI,QAAQ,MAAM,OAAO;AAC/C,UAAO,OAAO,UAAU,aAAa,MAAM,KAAK,OAAO,GAAG;;EAE5D,IAAI,QAAQ,MAAM;AAChB,OAAI,SAAS,OAAQ,QAAO;AAC5B,UAAO,QAAQ;;EAElB,CAAC;;AAiDJ,SAAS,WAAmC,OAAa;AACvD,QAAO,gBAAgB,MAAM;;AAG/B,SAAS,eACP,OACA,SACc;AACd,QAAO,0BACL,OACA,QACD;;AAGH,SAAS,kBACP,QACA,OACc;AACd,KAAI,OAAO,SAAS,MAAM,KACxB,QAAO,WAAW,MAAM;AAG1B,SAAQ,MAAM,MAAd;EACE,KAAK,OACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,MAAM,GAAG,UAAU,SAAS,OAAO,OAAO,KAAK,MAAM;GACtD;EACH,KAAK,YACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,WAAW,GAAG,eAAe,SAAS,OAAO,YAAY,KAAK,MAAM;GACrE;EACH,KAAK;EACL,KAAK,0BAA0B;GAc7B,MAAM,SAAS;IAAE,GAAG;IAAQ,GAAG;IAAO;AACtC,OAAI,MAAM,MAAM,QAAQ,QAAQ,UAAU,OAAO,MAAM,KACrD,QAAO,KAAK,OAAO;AAErB,OAAI,MAAM,QAAQ,QAAQ,UAAU,UAAU,OAAO,QAAQ,KAC3D,QAAO,OAAO,OAAO;AAEvB,UAAO,OAAO,IAAI,UAAU,SAAS,OAAO,OAAO,OAAO,KAAK,MAAM,QAAQ;AAC7E,UAAO;;EAET,QACE,QAAO;GACL,GAAG;GACH,GAAG;GACJ;;;AAIP,SAAS,cAAc,OAA8B;CACnD,MAAM,EAAE,WAAW,MAAM,SAAS,MAAM;CACxC,MAAM,eAAe,UAAU,KAAK,IAAI;CACxC,MAAM,YAAY,KAAK,UAAU,kBAAmB,KAAK,MAAM,KAAM;AACrE,QAAO,GAAG,aAAa,IAAI,QAAQ,GAAG,IAAI;;AAG5C,SAAS,uBAAuB,OAA4C;AAC1E,QAAO,MAAM,OAAO;;;;;AAMtB,IAAa,mBAAb,MAA8B;CAC5B,iCAAkC,IAAI,KAA+B;CACrE,wCAAyC,IAAI,KAAqB;CAClE,mDAAoD,IAAI,KAAqB;;;;;;CAO7E,QAAQ,OAA6C;EACnD,MAAM,OAAO,MAAM,OAAO;EAC1B,MAAM,mBAAmB,GAAG,MAAM,OAAO,UAAU,KAAK,IAAI,CAAC,IAAI,MAAM,OAAO,QAAQ;AAEtF,MAAI,KAAK,UAAU,iBAAiB;GAClC,MAAM,MAAM,cAAc,MAAM;AAChC,QAAK,sBAAsB,IAAI,kBAAkB,IAAI;GACrD,MAAM,UAA4B;IAChC,IAAI,KAAK;IACT,WAAW,CAAC,GAAG,MAAM,OAAO,UAAU;IACtC,MAAM,MAAM,OAAO;IACnB,UAAU,KAAK;IACf,QAAQ,EAAE;IACX;AACD,QAAK,eAAe,IAAI,KAAK,QAAQ;AACrC,UAAO;IAAE,MAAM;IAAiB;IAAK;IAAS;IAAO;;EAGvD,MAAM,YAAY,KAAK,sBAAsB,IAAI,iBAAiB;AAClE,MAAI,CAAC,WAAW;GAOd,MAAM,eAAe,GAAG,iBAAiB;AACzC,QAAK,sBAAsB,IAAI,kBAAkB,aAAa;GAC9D,MAAM,YAA8B;IAClC,IAAI,KAAK;IACT,WAAW,CAAC,GAAG,MAAM,OAAO,UAAU;IACtC,MAAM,MAAM,OAAO;IACnB,QAAQ,EAAE;IACX;AACD,QAAK,eAAe,IAAI,cAAc,UAAU;AAChD,UAAO,KAAK,QAAQ,MAAM;;EAG5B,MAAM,UAAU,KAAK,eAAe,IAAI,UAAU;AAClD,MAAI,CAAC,QACH,OAAM,IAAI,MAAM,yCAAyC,YAAY;AAGvE,MAAK,KAA4B,UAAU,SAAS;AAClD,WAAQ,QAAS,KAA+B;AAChD,UAAO;IACL,MAAM;IACN,KAAK;IACL;IACA;IACD;;AAGH,UAAQ,KAAK,OAAb;GACE,KAAK;AACH,YAAQ,OAAO,KAAK,SAAS,WAAW,KAAK,QAAQ;AACrD,SAAK,iCAAiC,IACpC,cAAc,WAAW,KAAK,OAAO,KAAK,QAAQ,KAAK,EACvD,KAAK,MACN;AACD,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO,KAAK;KACZ,OAAO,KAAK;KACZ;KACD;GAEH,KAAK,uBAAuB;IAC1B,MAAM,aAAa;IAInB,MAAM,aACJ,WAAW,YACV,WAAW,SAAS,OACjB,eAAe,WAAW,OAAO,QAAQ,OAAO,KAAK,OAAO,GAC5D,KAAA;AACN,QAAI,cAAc,KAChB,OAAM,IAAI,MAAM,+CAA+C;IAEjE,MAAM,cAAc,KAAK,kBACvB,WACA,QAAQ,QACR,KAAK,OACL,WAAW,KACZ;IACD,MAAM,UAAU,QAAQ,OAAO;AAC/B,YAAQ,OAAO,eACb,WAAW,WAAW,OAClB,WAAW,OACT,WAAW,WAAW,QAAQ,GAC9B,kBAAkB,SAAS,WAAW,QAAQ,GAC/C,oBACC,SACA,KAID;AACP,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO;KACP,OAAO;KACP;KACD;;GAEH,KAAK,wBAAwB;IAC3B,MAAM,cAAc,KAAK,wBACvB,WACA,KAAK,OACL,KAAK,QAAQ,KACd;AACD,YAAQ,OAAO,eAAe,WAAW,KAAK,QAAQ;AACtD,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO;KACP,OAAO,KAAK;KACZ;KACD;;GAEH,KAAK;AACH,YAAQ,QAAQ,KAAK;AACrB,YAAQ,iBAAiB,KAAK;AAC9B,SAAK,eAAe,OAAO,UAAU;AACrC,SAAK,sBAAsB,OAAO,iBAAiB;AACnD,SAAK,uBAAuB,UAAU;AACtC,WAAO;KACL,MAAM;KACN,KAAK;KACL,SAAS,gBAAgB,QAAQ;KACjC;KACD;GAEH,KAAK;AACH,YAAQ,QAAQ;KAAE,SAAS,KAAK;KAAS,MAAM,KAAK;KAAM;AAC1D,SAAK,eAAe,OAAO,UAAU;AACrC,SAAK,sBAAsB,OAAO,iBAAiB;AACnD,SAAK,uBAAuB,UAAU;AACtC,WAAO;KACL,MAAM;KACN,KAAK;KACL,SAAS,gBAAgB,QAAQ;KACjC;KACD;;;CAKP,kBACE,WACA,QACA,eACA,WACQ;EACR,MAAM,UAAU,OAAO;AACvB,MACE,WAAW,QACX,QAAQ,SAAS,aACjB,wBAAwB,QAAQ,MAAM,UAAU,EAChD;AACA,QAAK,iCAAiC,IACpC,cAAc,WAAW,eAAe,UAAU,EAClD,cACD;AACD,UAAO;;EAGT,MAAM,MAAM,cAAc,WAAW,eAAe,UAAU;EAC9D,MAAM,WAAW,KAAK,iCAAiC,IAAI,IAAI;AAC/D,MAAI,YAAY,KAAM,QAAO;EAE7B,MAAM,YAAY,OAAO;AACzB,OAAK,iCAAiC,IAAI,KAAK,UAAU;AACzD,SAAO;;CAGT,wBACE,WACA,eACA,WACQ;EACR,MAAM,MAAM,cAAc,WAAW,eAAe,UAAU;EAC9D,MAAM,WAAW,KAAK,iCAAiC,IAAI,IAAI;AAC/D,MAAI,YAAY,KAAM,QAAO;AAE7B,OAAK,iCAAiC,IAAI,KAAK,cAAc;AAC7D,SAAO;;CAGT,uBAA+B,WAAyB;EACtD,MAAM,SAAS,GAAG,UAAU;AAC5B,OAAK,MAAM,OAAO,KAAK,iCAAiC,MAAM,CAC5D,KAAI,IAAI,WAAW,OAAO,CACxB,MAAK,iCAAiC,OAAO,IAAI;;;AAKzD,SAAS,cACP,WACA,eACA,WACQ;AACR,QAAO,GAAG,UAAU,IAAI,cAAc,IAAI;;AAG5C,SAAS,wBACP,aACA,UACS;CACT,MAAM,gBAAgB,IAAI,IAAI;EAC5B;EACA;EACA;EACA;EACD,CAAC;CACF,MAAM,sBAAsB,IAAI,IAAI,CAClC,oBACA,yBACD,CAAC;AAEF,QACG,cAAc,IAAI,YAAY,IAAI,cAAc,IAAI,SAAS,IAC7D,oBAAoB,IAAI,YAAY,IAAI,oBAAoB,IAAI,SAAS;;;;;;;AAS9E,IAAa,4BAAb,MAAuC;CACrC,aAAsB,IAAI,kBAAkB;CAC5C,mCAA4B,IAAI,KAA+B;;;;;;CAO/D,QAAQ,OAAoD;EAC1D,MAAM,SAAS,MAAA,UAAgB,QAAQ,MAAM;AAC7C,MAAI,UAAU,KAAM,QAAO,KAAA;AAE3B,UAAQ,OAAO,MAAf;GACE,KAAK,iBAAiB;IACpB,MAAM,YAAY,IAAI,iBAAiB,OAAO,QAAQ;AACtD,cAAU,gBAAgB,OAAO,MAAM;AACvC,cAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,UAAA,gBAAsB,IAAI,OAAO,KAAK,UAAU;AAChD,WAAO;;GAET,KAAK,uBAAuB;IAC1B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D,QACE,aACA,OAAO,MAAM,SAAS,UACtB,UAAU,OAAO,SACjB,OAAO,MAAM,KAEb,WAAU,WAAW,OAAO,MAAM,KAAK;AAEzC,QACE,aACA,OAAO,MAAM,SAAS,eACtB,eAAe,OAAO,SACtB,OAAO,MAAM,UAEb,WAAU,gBAAgB,OAAO,MAAM,UAAU;AAEnD;;GAEF,KAAK,uBAAuB;IAC1B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,CAAC,UAAW,QAAO,KAAA;AACvB,cAAU,gBAAgB,OAAO,MAAM;AACvC,cAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,QAAI,OAAO,MAAM,SAAS,UAAU,UAAU,OAAO,MACnD,WAAU,WAAW,OAAO,MAAM,KAAK;AAEzC,QAAI,OAAO,MAAM,SAAS,eAAe,eAAe,OAAO,MAC7D,WAAU,gBAAgB,OAAO,MAAM,UAAU;AAEnD;;GAEF,KAAK,wBAAwB;IAC3B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D;;GAEF,KAAK,SAAS;IACZ,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D;;GAEF,KAAK,kBAAkB;IACrB,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,eAAU,SAAS;AACnB,WAAA,gBAAsB,OAAO,OAAO,IAAI;;AAE1C;;GAEF,KAAK,iBAAiB;IACpB,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,eAAU,QAAQ;AAClB,WAAA,gBAAsB,OAAO,OAAO,IAAI;;AAE1C"}
|
|
1
|
+
{"version":3,"file":"messages.cjs","names":["#events","MultiCursorBuffer","#textPromise","#reasoningPromise","#resolveText","#resolveReasoning","#textChunks","#textWaiters","#textDone","#reasoningChunks","#reasoningWaiters","#reasoningDone","#usageIterator","#assembleMessage","AIMessage","#assembler","#activeStreaming"],"sources":["../../../src/client/stream/messages.ts"],"sourcesContent":["import {\n AIMessage,\n type ContentBlock as CoreContentBlock,\n type UsageMetadata,\n} from \"@langchain/core/messages\";\nimport type {\n ChatModelStreamEvent,\n ContentBlockDelta as CoreContentBlockDelta,\n} from \"@langchain/core/language_models/event\";\nimport type {\n ContentBlock,\n FinalizedContentBlock,\n MessageMetadata,\n MessagesEvent,\n UsageInfo,\n} from \"@langchain/protocol\";\n\nimport { MultiCursorBuffer } from \"./multi-cursor-buffer.js\";\n\ntype TextContentStream = AsyncIterable<string> &\n PromiseLike<string> & { full: AsyncIterable<string> };\n\ntype UsageMetadataStream = AsyncIterable<UsageMetadata> &\n PromiseLike<UsageMetadata | undefined>;\n\ntype ToolCallsStream = AsyncIterable<CoreContentBlock.Tools.ToolCall> &\n PromiseLike<Array<CoreContentBlock.Tools.ToolCall>> & {\n full: AsyncIterable<Array<CoreContentBlock.Tools.ToolCall>>;\n };\n\nfunction applyCoreContentDelta(\n target: CoreContentBlock,\n delta: CoreContentBlock\n): CoreContentBlock {\n if (target.type !== delta.type) {\n return structuredClone(delta);\n }\n\n switch (delta.type) {\n case \"text\":\n return {\n ...target,\n ...delta,\n text: `${\"text\" in target ? target.text : \"\"}${delta.text}`,\n } as CoreContentBlock;\n case \"reasoning\":\n return {\n ...target,\n ...delta,\n reasoning: `${\"reasoning\" in target ? target.reasoning : \"\"}${delta.reasoning}`,\n } as CoreContentBlock;\n case \"tool_call_chunk\":\n case \"server_tool_call_chunk\": {\n const merged = { ...target, ...delta } as Record<string, unknown>;\n if (delta.id == null && \"id\" in target && target.id != null) {\n merged.id = target.id;\n }\n if (delta.name == null && \"name\" in target && target.name != null) {\n merged.name = target.name;\n }\n merged.args = `${(\"args\" in target ? target.args : \"\") ?? \"\"}${delta.args ?? \"\"}`;\n return merged as unknown as CoreContentBlock;\n }\n default:\n return { ...target, ...delta } as CoreContentBlock;\n }\n}\n\nfunction coreContentBlockFromDelta(\n delta: CoreContentBlockDelta,\n current?: CoreContentBlock\n): CoreContentBlock {\n switch (delta.type) {\n case \"text-delta\":\n return { type: \"text\", text: delta.text } as CoreContentBlock;\n case \"reasoning-delta\":\n return {\n type: \"reasoning\",\n reasoning: delta.reasoning,\n } as CoreContentBlock;\n case \"data-delta\": {\n const merged = { ...(current ?? {}), data: delta.data } as Record<\n string,\n unknown\n >;\n if (delta.encoding) merged.encoding = delta.encoding;\n return merged as unknown as CoreContentBlock;\n }\n case \"block-delta\":\n return delta.fields as CoreContentBlock;\n }\n}\n\nfunction applyCoreEventDelta(\n current: CoreContentBlock | undefined,\n event: Extract<ChatModelStreamEvent, { event: \"content-block-delta\" }> & {\n content?: CoreContentBlock;\n }\n): CoreContentBlock {\n if (event.content) {\n return current\n ? applyCoreContentDelta(current, event.content)\n : event.content;\n }\n\n switch (event.delta.type) {\n case \"text-delta\":\n if (current?.type === \"text\") {\n return {\n ...current,\n text: `${\"text\" in current ? current.text : \"\"}${event.delta.text}`,\n } as CoreContentBlock;\n }\n return coreContentBlockFromDelta(event.delta, current);\n case \"reasoning-delta\":\n if (current?.type === \"reasoning\") {\n return {\n ...current,\n reasoning: `${\"reasoning\" in current ? current.reasoning : \"\"}${event.delta.reasoning}`,\n } as CoreContentBlock;\n }\n return coreContentBlockFromDelta(event.delta, current);\n case \"data-delta\": {\n const merged = { ...(current ?? {}) } as Record<string, unknown>;\n merged.data = `${(merged.data as string | undefined) ?? \"\"}${event.delta.data}`;\n if (event.delta.encoding) merged.encoding = event.delta.encoding;\n return merged as unknown as CoreContentBlock;\n }\n case \"block-delta\":\n return {\n ...(current ?? {}),\n ...event.delta.fields,\n } as CoreContentBlock;\n }\n}\n\nfunction normalizeUsage(\n usage: UsageInfo | Partial<UsageMetadata> | undefined\n): UsageMetadata | undefined {\n if (!usage) return undefined;\n return {\n ...usage,\n input_tokens: usage.input_tokens ?? 0,\n output_tokens: usage.output_tokens ?? 0,\n total_tokens: usage.total_tokens ?? 0,\n };\n}\n\nfunction readRunId(data: object): string | undefined {\n const record = data as Record<string, unknown>;\n return typeof record.run_id === \"string\" ? record.run_id : undefined;\n}\n\nfunction readUsage(data: object): UsageInfo | undefined {\n const usage = (data as Record<string, unknown>).usage;\n return usage != null && typeof usage === \"object\" && !Array.isArray(usage)\n ? (usage as UsageInfo)\n : undefined;\n}\n\nfunction readFinishMetadata(data: object): Record<string, unknown> | undefined {\n const record = data as Record<string, unknown>;\n const isRecord = (value: unknown): value is Record<string, unknown> =>\n value != null && typeof value === \"object\" && !Array.isArray(value);\n const merged = {\n ...(isRecord(record.responseMetadata) ? record.responseMetadata : {}),\n ...(isRecord(record.metadata) ? record.metadata : {}),\n };\n return Object.keys(merged).length > 0 ? merged : undefined;\n}\n\n/**\n * Mutable view of a streamed message as message and content-block events are\n * assembled into a single structure.\n */\nexport interface AssembledMessage {\n id: string;\n namespace: string[];\n blocks: ContentBlock[];\n node?: string;\n runId?: string;\n usage?: UsageInfo;\n metadata?: MessageMetadata;\n finishMetadata?: Record<string, any>;\n finishReason?: string;\n error?: {\n message: string;\n code?: string;\n };\n}\n\n/**\n * Symbol keys for assembler → StreamingMessage communication.\n * Module-private: invisible to external consumers, accessible to\n * {@link StreamingMessageAssembler} within this file.\n */\nconst PUSH_TEXT: unique symbol = Symbol(\"pushText\");\nconst PUSH_REASONING: unique symbol = Symbol(\"pushReasoning\");\nconst PUSH_EVENT: unique symbol = Symbol(\"pushEvent\");\nconst UPDATE_CONTEXT: unique symbol = Symbol(\"updateContext\");\nconst FINISH: unique symbol = Symbol(\"finish\");\nconst ERROR: unique symbol = Symbol(\"error\");\n\n/**\n * Live streaming view of a single message lifecycle, matching the\n * in-process `ChatModelStream` dual-interface pattern.\n *\n * - `text` / `reasoning`: iterate for streaming deltas, or await for\n * the full concatenated string after the message completes.\n * - `usage`: promise that resolves with token usage on message-finish.\n * - `blocks`: the assembled content blocks (updated as deltas arrive).\n *\n * Created by {@link StreamingMessageAssembler} and yielded by\n * the `session.messages` lazy getter.\n */\nexport class StreamingMessage\n implements AsyncIterable<ChatModelStreamEvent>, PromiseLike<AIMessage>\n{\n readonly id: string;\n readonly namespace: string[];\n node: string | undefined;\n readonly metadata: MessageMetadata | undefined;\n readonly assembled: AssembledMessage;\n readonly #events = new MultiCursorBuffer<ChatModelStreamEvent>();\n\n #textChunks: string[] = [];\n #reasoningChunks: string[] = [];\n #textWaiters: Array<() => void> = [];\n #reasoningWaiters: Array<() => void> = [];\n #textDone = false;\n #reasoningDone = false;\n\n #resolveText!: (v: string) => void;\n #resolveReasoning!: (v: string) => void;\n readonly #textPromise: Promise<string>;\n readonly #reasoningPromise: Promise<string>;\n\n constructor(assembled: AssembledMessage) {\n this.id = assembled.id;\n this.assembled = assembled;\n this.namespace = assembled.namespace;\n this.node = assembled.node;\n this.metadata = assembled.metadata;\n this.#textPromise = new Promise<string>((r) => {\n this.#resolveText = r;\n });\n this.#reasoningPromise = new Promise<string>((r) => {\n this.#resolveReasoning = r;\n });\n }\n\n get text(): TextContentStream {\n const chunks = this.#textChunks;\n const waiters = this.#textWaiters;\n const getDone = () => this.#textDone;\n let cursor = 0;\n return {\n [Symbol.asyncIterator]() {\n return {\n async next(): Promise<IteratorResult<string>> {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n };\n },\n then: this.#textPromise.then.bind(this.#textPromise),\n full: {\n async *[Symbol.asyncIterator]() {\n let accumulated = \"\";\n for await (const chunk of {\n [Symbol.asyncIterator]: () =>\n ({\n next: async (): Promise<IteratorResult<string>> => {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n }) satisfies AsyncIterator<string>,\n }) {\n accumulated += chunk;\n yield accumulated;\n }\n },\n },\n };\n }\n\n get reasoning(): TextContentStream {\n const chunks = this.#reasoningChunks;\n const waiters = this.#reasoningWaiters;\n const getDone = () => this.#reasoningDone;\n let cursor = 0;\n return {\n [Symbol.asyncIterator]() {\n return {\n async next(): Promise<IteratorResult<string>> {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n };\n },\n then: this.#reasoningPromise.then.bind(this.#reasoningPromise),\n full: {\n async *[Symbol.asyncIterator]() {\n let accumulated = \"\";\n for await (const chunk of {\n [Symbol.asyncIterator]: () =>\n ({\n next: async (): Promise<IteratorResult<string>> => {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n }) satisfies AsyncIterator<string>,\n }) {\n accumulated += chunk;\n yield accumulated;\n }\n },\n },\n };\n }\n\n get usage(): UsageMetadataStream {\n const promise = (async () => {\n let usage: UsageMetadata | undefined;\n for await (const snapshot of this.#usageIterator()) {\n usage = snapshot;\n }\n return usage;\n })();\n return {\n [Symbol.asyncIterator]: () => this.#usageIterator(),\n then: promise.then.bind(promise),\n };\n }\n\n get toolCalls(): ToolCallsStream {\n const events = this.#events;\n const iterator = async function* () {\n for await (const event of events) {\n if (\n event.event === \"content-block-finish\" &&\n event.content.type === \"tool_call\"\n ) {\n yield event.content as CoreContentBlock.Tools.ToolCall;\n }\n }\n };\n return {\n [Symbol.asyncIterator]: iterator,\n then: async (onfulfilled, onrejected) => {\n try {\n const calls: CoreContentBlock.Tools.ToolCall[] = [];\n for await (const call of iterator()) calls.push(call);\n return onfulfilled ? onfulfilled(calls) : (calls as never);\n } catch (err) {\n if (onrejected) return onrejected(err);\n throw err;\n }\n },\n full: {\n async *[Symbol.asyncIterator]() {\n const calls: CoreContentBlock.Tools.ToolCall[] = [];\n for await (const call of iterator()) {\n calls.push(call);\n yield [...calls];\n }\n },\n },\n };\n }\n\n get output(): PromiseLike<AIMessage> {\n return { then: (onf, onr) => this.#assembleMessage().then(onf, onr) };\n }\n\n get blocks(): ContentBlock[] {\n return this.assembled.blocks;\n }\n\n [Symbol.asyncIterator](): AsyncIterator<ChatModelStreamEvent> {\n return this.#events[Symbol.asyncIterator]();\n }\n\n then<TResult1 = AIMessage, TResult2 = never>(\n onfulfilled?:\n | ((value: AIMessage) => TResult1 | PromiseLike<TResult1>)\n | null,\n onrejected?: ((reason: unknown) => TResult2 | PromiseLike<TResult2>) | null\n ): PromiseLike<TResult1 | TResult2> {\n return this.#assembleMessage().then(onfulfilled, onrejected);\n }\n\n async *#usageIterator(): AsyncGenerator<UsageMetadata> {\n for await (const event of this.#events) {\n if (event.event === \"message-start\" && event.usage) {\n yield normalizeUsage(event.usage)!;\n } else if (event.event === \"message-finish\" && event.usage) {\n yield normalizeUsage(event.usage)!;\n }\n }\n }\n\n async #assembleMessage(): Promise<AIMessage> {\n const contentBlocks: Array<CoreContentBlock | undefined> = [];\n let id: string | undefined = this.id;\n let usage = normalizeUsage(this.assembled.usage);\n let metadata: Record<string, unknown> = {\n ...(this.assembled.finishMetadata ?? {}),\n };\n let finishReason: string | undefined = this.assembled.finishReason;\n\n for await (const event of this.#events) {\n switch (event.event) {\n case \"message-start\":\n id = event.id ?? id;\n if (event.usage) usage = normalizeUsage(event.usage);\n break;\n case \"content-block-start\":\n contentBlocks[event.index] = event.content;\n break;\n case \"content-block-delta\": {\n const current = contentBlocks[event.index];\n contentBlocks[event.index] = applyCoreEventDelta(current, event);\n break;\n }\n case \"content-block-finish\":\n contentBlocks[event.index] = event.content;\n break;\n case \"message-finish\": {\n finishReason = event.reason;\n if (event.usage) usage = normalizeUsage(event.usage);\n const finishMetadata = readFinishMetadata(event);\n if (finishMetadata != null) {\n metadata = { ...metadata, ...finishMetadata };\n }\n break;\n }\n default:\n break;\n }\n }\n\n return new AIMessage({\n id,\n content: contentBlocks.filter(\n (block): block is CoreContentBlock => block != null\n ),\n usage_metadata: usage,\n additional_kwargs: {\n namespace: this.namespace,\n ...(this.node != null ? { node: this.node } : {}),\n ...(this.assembled.runId != null\n ? { run_id: this.assembled.runId }\n : {}),\n ...(this.metadata != null ? { metadata: this.metadata } : {}),\n ...(this.assembled.usage != null\n ? { usage: this.assembled.usage }\n : {}),\n },\n response_metadata: {\n ...(finishReason != null ? { finish_reason: finishReason } : {}),\n ...metadata,\n output_version: \"v1\" as const,\n },\n });\n }\n\n [PUSH_EVENT](event: ChatModelStreamEvent): void {\n this.#events.push(event);\n }\n\n [UPDATE_CONTEXT](event: MessagesEvent): void {\n this.node = event.params.node ?? this.node;\n }\n\n [PUSH_TEXT](delta: string): void {\n this.#textChunks.push(delta);\n // Wake every caught-up iterator so each one advances its own cursor.\n // Iterators re-check `chunks.length` before delivering, which keeps\n // the cursor the single source of truth for what a consumer has seen.\n const pending = this.#textWaiters.splice(0, this.#textWaiters.length);\n for (const waiter of pending) waiter();\n }\n\n [PUSH_REASONING](delta: string): void {\n this.#reasoningChunks.push(delta);\n const pending = this.#reasoningWaiters.splice(\n 0,\n this.#reasoningWaiters.length\n );\n for (const waiter of pending) waiter();\n }\n\n [FINISH](): void {\n this.#textDone = true;\n this.#reasoningDone = true;\n this.#resolveText(this.#textChunks.join(\"\"));\n this.#resolveReasoning(this.#reasoningChunks.join(\"\"));\n const textPending = this.#textWaiters.splice(0, this.#textWaiters.length);\n for (const waiter of textPending) waiter();\n const reasoningPending = this.#reasoningWaiters.splice(\n 0,\n this.#reasoningWaiters.length\n );\n for (const waiter of reasoningPending) waiter();\n this.#events.close();\n }\n\n [ERROR](): void {\n this[FINISH]();\n }\n}\n\n/**\n * Public view yielded by message projections.\n *\n * `StreamingMessage` is PromiseLike so callers can still `await` a message\n * object directly, but TypeScript applies `Awaited<T>` to values produced by\n * `for await`. Exposing a non-thenable view keeps loop variables typed as the\n * streaming handle instead of as the finalized `AIMessage`.\n */\nexport type StreamingMessageHandle = Omit<StreamingMessage, \"then\">;\n\nexport function toStreamingMessageHandle(\n message: StreamingMessage\n): StreamingMessageHandle {\n return new Proxy(message, {\n get(target, prop) {\n if (prop === \"then\") return undefined;\n const value = Reflect.get(target, prop, target);\n return typeof value === \"function\" ? value.bind(target) : value;\n },\n has(target, prop) {\n if (prop === \"then\") return false;\n return prop in target;\n },\n }) as StreamingMessageHandle;\n}\n\n/**\n * Emitted by `MessageAssembler.consume()` to describe how a message changed in\n * response to a single protocol event.\n */\nexport type MessageAssemblyUpdate =\n | {\n kind: \"message-start\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"content-block-start\" | \"content-block-delta\";\n key: string;\n message: AssembledMessage;\n index: number;\n block: ContentBlock;\n event: MessagesEvent;\n }\n | {\n kind: \"content-block-finish\";\n key: string;\n message: AssembledMessage;\n index: number;\n block: FinalizedContentBlock;\n event: MessagesEvent;\n }\n | {\n kind: \"usage\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"message-finish\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"message-error\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n };\n\nfunction cloneBlock<T extends ContentBlock>(block: T): T {\n return structuredClone(block);\n}\n\nfunction blockFromDelta(\n delta: CoreContentBlockDelta,\n current?: ContentBlock\n): ContentBlock {\n return coreContentBlockFromDelta(\n delta,\n current as unknown as CoreContentBlock | undefined\n ) as unknown as ContentBlock;\n}\n\nfunction applyContentDelta(\n target: ContentBlock,\n delta: ContentBlock\n): ContentBlock {\n if (target.type !== delta.type) {\n return cloneBlock(delta);\n }\n\n switch (delta.type) {\n case \"text\":\n return {\n ...target,\n ...delta,\n text: `${\"text\" in target ? target.text : \"\"}${delta.text}`,\n } as ContentBlock;\n case \"reasoning\":\n return {\n ...target,\n ...delta,\n reasoning: `${\"reasoning\" in target ? target.reasoning : \"\"}${delta.reasoning}`,\n } as ContentBlock;\n case \"tool_call_chunk\":\n case \"server_tool_call_chunk\": {\n // Spread target first, then delta — but preserve target's\n // ``id``/``name`` when the delta explicitly sets them to\n // null/undefined. Some providers (notably Anthropic via the\n // langchain-core compat bridge) only attach the tool-call\n // identifiers to the first ``content-block-start`` chunk; every\n // subsequent ``input_json_delta`` chunk carries ``id=null,\n // name=null``. A naive ``{...target, ...delta}`` spread\n // overwrites the captured identifiers with null, which in turn\n // makes downstream consumers (e.g. ``extractToolCallChunks`` in\n // ``assembled-to-message.ts``) drop the chunk on the floor until\n // the final ``content-block-finish`` event promotes it to a\n // finalized ``tool_call`` — causing tool-call cards to appear\n // all-at-once at the end of the turn instead of incrementally.\n const merged = { ...target, ...delta } as Record<string, unknown>;\n if (delta.id == null && \"id\" in target && target.id != null) {\n merged.id = target.id;\n }\n if (delta.name == null && \"name\" in target && target.name != null) {\n merged.name = target.name;\n }\n merged.args = `${(\"args\" in target ? target.args : \"\") ?? \"\"}${delta.args ?? \"\"}`;\n return merged as unknown as ContentBlock;\n }\n default:\n return {\n ...target,\n ...delta,\n } as ContentBlock;\n }\n}\n\nfunction messageKeyFor(event: MessagesEvent): string {\n const { namespace, node, data } = event.params;\n const namespaceKey = namespace.join(\"/\");\n const runId = readRunId(data);\n const messageId = data.event === \"message-start\" ? (data.id ?? \"\") : \"\";\n return `${namespaceKey}::${node ?? \"\"}::${runId ?? messageId}`;\n}\n\nfunction toChatModelStreamEvent(event: MessagesEvent): ChatModelStreamEvent {\n return event.params.data as unknown as ChatModelStreamEvent;\n}\n\n/**\n * Incrementally assembles `messages` events into complete message objects.\n */\nexport class MessageAssembler {\n private readonly activeMessages = new Map<string, AssembledMessage>();\n private readonly activeByNamespaceNode = new Map<string, string>();\n private readonly blockIndexByProtocolIndexAndType = new Map<string, number>();\n\n /**\n * Applies a single message event and returns the resulting assembly update.\n *\n * @param event - Incoming `messages` event to fold into the assembler state.\n */\n consume(event: MessagesEvent): MessageAssemblyUpdate {\n const data = event.params.data;\n const namespaceNodeKey = `${event.params.namespace.join(\"/\")}::${event.params.node ?? \"\"}`;\n const runId = readRunId(data);\n\n if (data.event === \"message-start\") {\n const key = messageKeyFor(event);\n this.activeByNamespaceNode.set(namespaceNodeKey, key);\n const message: AssembledMessage = {\n id: data.id,\n namespace: [...event.params.namespace],\n node: event.params.node,\n runId,\n usage: readUsage(data),\n metadata: data.metadata,\n blocks: [],\n };\n this.activeMessages.set(key, message);\n return { kind: \"message-start\", key, message, event };\n }\n\n const runKey = runId != null ? `${namespaceNodeKey}::${runId}` : undefined;\n const fallbackKey = this.activeByNamespaceNode.get(namespaceNodeKey);\n const fallbackMessage =\n fallbackKey != null ? this.activeMessages.get(fallbackKey) : undefined;\n const activeKey =\n (runKey != null && this.activeMessages.has(runKey)\n ? runKey\n : undefined) ??\n (runId == null || fallbackMessage?.runId == null\n ? fallbackKey\n : undefined);\n if (!activeKey) {\n // A continuation event (delta/finish/error) arrived without a\n // prior `message-start`. This can happen on late-attaching\n // subscriptions when the server has already trimmed the\n // `message-start` from its replay buffer. Synthesize a minimal\n // active message so the assembler can still fold subsequent\n // events instead of hard-failing the caller.\n const syntheticKey = runKey ?? `${namespaceNodeKey}::`;\n this.activeByNamespaceNode.set(namespaceNodeKey, syntheticKey);\n const synthetic: AssembledMessage = {\n id: data.id,\n namespace: [...event.params.namespace],\n node: event.params.node,\n runId,\n blocks: [],\n };\n this.activeMessages.set(syntheticKey, synthetic);\n return this.consume(event);\n }\n\n const message = this.activeMessages.get(activeKey);\n if (!message) {\n throw new Error(`No active message state found for key ${activeKey}`);\n }\n\n if ((data as { event?: string }).event === \"usage\") {\n message.usage = (data as { usage?: UsageInfo }).usage;\n return {\n kind: \"usage\",\n key: activeKey,\n message,\n event,\n };\n }\n\n switch (data.event) {\n case \"content-block-start\": {\n message.blocks[data.index] = cloneBlock(data.content);\n this.blockIndexByProtocolIndexAndType.set(\n blockIndexKey(activeKey, data.index, data.content.type),\n data.index\n );\n return {\n kind: \"content-block-start\",\n key: activeKey,\n message,\n index: data.index,\n block: data.content,\n event,\n };\n }\n case \"content-block-delta\": {\n const deltaEvent = data as typeof data & {\n content?: ContentBlock;\n delta?: CoreContentBlockDelta;\n };\n const deltaBlock =\n deltaEvent.content ??\n (deltaEvent.delta != null\n ? blockFromDelta(deltaEvent.delta, message.blocks[data.index])\n : undefined);\n if (deltaBlock == null) {\n throw new Error(\"Received content-block-delta without content\");\n }\n const targetIndex = this.resolveBlockIndex(\n activeKey,\n message.blocks,\n data.index,\n deltaBlock.type\n );\n const current = message.blocks[targetIndex];\n message.blocks[targetIndex] =\n deltaEvent.content != null\n ? current == null\n ? cloneBlock(deltaEvent.content)\n : applyContentDelta(current, deltaEvent.content)\n : (applyCoreEventDelta(\n current as CoreContentBlock | undefined,\n data as unknown as Extract<\n ChatModelStreamEvent,\n { event: \"content-block-delta\" }\n >\n ) as unknown as ContentBlock);\n return {\n kind: \"content-block-delta\",\n key: activeKey,\n message,\n index: targetIndex,\n block: deltaBlock,\n event,\n };\n }\n case \"content-block-finish\": {\n const targetIndex = this.resolveFinishBlockIndex(\n activeKey,\n data.index,\n data.content.type\n );\n message.blocks[targetIndex] = cloneBlock(data.content);\n return {\n kind: \"content-block-finish\",\n key: activeKey,\n message,\n index: targetIndex,\n block: data.content,\n event,\n };\n }\n case \"message-finish\": {\n message.usage = readUsage(data) ?? message.usage;\n message.finishMetadata = readFinishMetadata(data);\n message.finishReason = data.reason;\n this.clearActiveMessage(activeKey, namespaceNodeKey);\n this.clearBlockIndexAliases(activeKey);\n return {\n kind: \"message-finish\",\n key: activeKey,\n message: structuredClone(message),\n event,\n };\n }\n case \"error\": {\n message.error = { message: data.message, code: data.code };\n this.clearActiveMessage(activeKey, namespaceNodeKey);\n this.clearBlockIndexAliases(activeKey);\n return {\n kind: \"message-error\",\n key: activeKey,\n message: structuredClone(message),\n event,\n };\n }\n }\n }\n\n private clearActiveMessage(\n activeKey: string,\n namespaceNodeKey: string\n ): void {\n this.activeMessages.delete(activeKey);\n if (this.activeByNamespaceNode.get(namespaceNodeKey) === activeKey) {\n this.activeByNamespaceNode.delete(namespaceNodeKey);\n }\n }\n\n private resolveBlockIndex(\n activeKey: string,\n blocks: ContentBlock[],\n protocolIndex: number,\n blockType: string\n ): number {\n const current = blocks[protocolIndex];\n if (\n current == null ||\n current.type === blockType ||\n areCompatibleBlockTypes(current.type, blockType)\n ) {\n this.blockIndexByProtocolIndexAndType.set(\n blockIndexKey(activeKey, protocolIndex, blockType),\n protocolIndex\n );\n return protocolIndex;\n }\n\n const key = blockIndexKey(activeKey, protocolIndex, blockType);\n const existing = this.blockIndexByProtocolIndexAndType.get(key);\n if (existing != null) return existing;\n\n const nextIndex = blocks.length;\n this.blockIndexByProtocolIndexAndType.set(key, nextIndex);\n return nextIndex;\n }\n\n private resolveFinishBlockIndex(\n activeKey: string,\n protocolIndex: number,\n blockType: string\n ): number {\n const key = blockIndexKey(activeKey, protocolIndex, blockType);\n const existing = this.blockIndexByProtocolIndexAndType.get(key);\n if (existing != null) return existing;\n\n this.blockIndexByProtocolIndexAndType.set(key, protocolIndex);\n return protocolIndex;\n }\n\n private clearBlockIndexAliases(activeKey: string): void {\n const prefix = `${activeKey}::`;\n for (const key of this.blockIndexByProtocolIndexAndType.keys()) {\n if (key.startsWith(prefix))\n this.blockIndexByProtocolIndexAndType.delete(key);\n }\n }\n}\n\nfunction blockIndexKey(\n activeKey: string,\n protocolIndex: number,\n blockType: string\n): string {\n return `${activeKey}::${protocolIndex}::${blockType}`;\n}\n\nfunction areCompatibleBlockTypes(\n currentType: string,\n nextType: string\n): boolean {\n const toolCallTypes = new Set([\n \"tool_call\",\n \"tool_call_chunk\",\n \"tool_use\",\n \"input_json_delta\",\n ]);\n const serverToolCallTypes = new Set([\n \"server_tool_call\",\n \"server_tool_call_chunk\",\n ]);\n\n return (\n (toolCallTypes.has(currentType) && toolCallTypes.has(nextType)) ||\n (serverToolCallTypes.has(currentType) && serverToolCallTypes.has(nextType))\n );\n}\n\n/**\n * Assembles `messages` events into {@link StreamingMessage} instances\n * with live text/reasoning delta streams, matching the in-process\n * `ChatModelStream` dual-interface pattern.\n */\nexport class StreamingMessageAssembler {\n readonly #assembler = new MessageAssembler();\n readonly #activeStreaming = new Map<string, StreamingMessage>();\n\n /**\n * Folds a single event and returns a new {@link StreamingMessage}\n * when a `message-start` is seen, or `undefined` for continuation\n * events (deltas, finish, error).\n */\n consume(event: MessagesEvent): StreamingMessage | undefined {\n const update = this.#assembler.consume(event);\n if (update == null) return undefined;\n\n switch (update.kind) {\n case \"message-start\": {\n const streaming = new StreamingMessage(update.message);\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n this.#activeStreaming.set(update.key, streaming);\n return streaming;\n }\n case \"content-block-start\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n if (\n streaming &&\n update.block.type === \"text\" &&\n \"text\" in update.block &&\n update.block.text\n ) {\n streaming[PUSH_TEXT](update.block.text);\n }\n if (\n streaming &&\n update.block.type === \"reasoning\" &&\n \"reasoning\" in update.block &&\n update.block.reasoning\n ) {\n streaming[PUSH_REASONING](update.block.reasoning);\n }\n return undefined;\n }\n case \"content-block-delta\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (!streaming) return undefined;\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n if (update.block.type === \"text\" && \"text\" in update.block) {\n streaming[PUSH_TEXT](update.block.text);\n }\n if (update.block.type === \"reasoning\" && \"reasoning\" in update.block) {\n streaming[PUSH_REASONING](update.block.reasoning);\n }\n return undefined;\n }\n case \"content-block-finish\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n return undefined;\n }\n case \"usage\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n return undefined;\n }\n case \"message-finish\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n streaming[FINISH]();\n this.#activeStreaming.delete(update.key);\n }\n return undefined;\n }\n case \"message-error\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n streaming[ERROR]();\n this.#activeStreaming.delete(update.key);\n }\n return undefined;\n }\n }\n }\n}\n"],"mappings":";;;;AA8BA,SAAS,sBACP,QACA,OACkB;AAClB,KAAI,OAAO,SAAS,MAAM,KACxB,QAAO,gBAAgB,MAAM;AAG/B,SAAQ,MAAM,MAAd;EACE,KAAK,OACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,MAAM,GAAG,UAAU,SAAS,OAAO,OAAO,KAAK,MAAM;GACtD;EACH,KAAK,YACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,WAAW,GAAG,eAAe,SAAS,OAAO,YAAY,KAAK,MAAM;GACrE;EACH,KAAK;EACL,KAAK,0BAA0B;GAC7B,MAAM,SAAS;IAAE,GAAG;IAAQ,GAAG;IAAO;AACtC,OAAI,MAAM,MAAM,QAAQ,QAAQ,UAAU,OAAO,MAAM,KACrD,QAAO,KAAK,OAAO;AAErB,OAAI,MAAM,QAAQ,QAAQ,UAAU,UAAU,OAAO,QAAQ,KAC3D,QAAO,OAAO,OAAO;AAEvB,UAAO,OAAO,IAAI,UAAU,SAAS,OAAO,OAAO,OAAO,KAAK,MAAM,QAAQ;AAC7E,UAAO;;EAET,QACE,QAAO;GAAE,GAAG;GAAQ,GAAG;GAAO;;;AAIpC,SAAS,0BACP,OACA,SACkB;AAClB,SAAQ,MAAM,MAAd;EACE,KAAK,aACH,QAAO;GAAE,MAAM;GAAQ,MAAM,MAAM;GAAM;EAC3C,KAAK,kBACH,QAAO;GACL,MAAM;GACN,WAAW,MAAM;GAClB;EACH,KAAK,cAAc;GACjB,MAAM,SAAS;IAAE,GAAI,WAAW,EAAE;IAAG,MAAM,MAAM;IAAM;AAIvD,OAAI,MAAM,SAAU,QAAO,WAAW,MAAM;AAC5C,UAAO;;EAET,KAAK,cACH,QAAO,MAAM;;;AAInB,SAAS,oBACP,SACA,OAGkB;AAClB,KAAI,MAAM,QACR,QAAO,UACH,sBAAsB,SAAS,MAAM,QAAQ,GAC7C,MAAM;AAGZ,SAAQ,MAAM,MAAM,MAApB;EACE,KAAK;AACH,OAAI,SAAS,SAAS,OACpB,QAAO;IACL,GAAG;IACH,MAAM,GAAG,UAAU,UAAU,QAAQ,OAAO,KAAK,MAAM,MAAM;IAC9D;AAEH,UAAO,0BAA0B,MAAM,OAAO,QAAQ;EACxD,KAAK;AACH,OAAI,SAAS,SAAS,YACpB,QAAO;IACL,GAAG;IACH,WAAW,GAAG,eAAe,UAAU,QAAQ,YAAY,KAAK,MAAM,MAAM;IAC7E;AAEH,UAAO,0BAA0B,MAAM,OAAO,QAAQ;EACxD,KAAK,cAAc;GACjB,MAAM,SAAS,EAAE,GAAI,WAAW,EAAE,EAAG;AACrC,UAAO,OAAO,GAAI,OAAO,QAA+B,KAAK,MAAM,MAAM;AACzE,OAAI,MAAM,MAAM,SAAU,QAAO,WAAW,MAAM,MAAM;AACxD,UAAO;;EAET,KAAK,cACH,QAAO;GACL,GAAI,WAAW,EAAE;GACjB,GAAG,MAAM,MAAM;GAChB;;;AAIP,SAAS,eACP,OAC2B;AAC3B,KAAI,CAAC,MAAO,QAAO,KAAA;AACnB,QAAO;EACL,GAAG;EACH,cAAc,MAAM,gBAAgB;EACpC,eAAe,MAAM,iBAAiB;EACtC,cAAc,MAAM,gBAAgB;EACrC;;AAGH,SAAS,UAAU,MAAkC;CACnD,MAAM,SAAS;AACf,QAAO,OAAO,OAAO,WAAW,WAAW,OAAO,SAAS,KAAA;;AAG7D,SAAS,UAAU,MAAqC;CACtD,MAAM,QAAS,KAAiC;AAChD,QAAO,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,GACrE,QACD,KAAA;;AAGN,SAAS,mBAAmB,MAAmD;CAC7E,MAAM,SAAS;CACf,MAAM,YAAY,UAChB,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM;CACrE,MAAM,SAAS;EACb,GAAI,SAAS,OAAO,iBAAiB,GAAG,OAAO,mBAAmB,EAAE;EACpE,GAAI,SAAS,OAAO,SAAS,GAAG,OAAO,WAAW,EAAE;EACrD;AACD,QAAO,OAAO,KAAK,OAAO,CAAC,SAAS,IAAI,SAAS,KAAA;;;;;;;AA4BnD,MAAM,YAA2B,OAAO,WAAW;AACnD,MAAM,iBAAgC,OAAO,gBAAgB;AAC7D,MAAM,aAA4B,OAAO,YAAY;AACrD,MAAM,iBAAgC,OAAO,gBAAgB;AAC7D,MAAM,SAAwB,OAAO,SAAS;AAC9C,MAAM,QAAuB,OAAO,QAAQ;;;;;;;;;;;;;AAc5C,IAAa,mBAAb,MAEA;CACE;CACA;CACA;CACA;CACA;CACA,UAAmB,IAAIC,4BAAAA,mBAAyC;CAEhE,cAAwB,EAAE;CAC1B,mBAA6B,EAAE;CAC/B,eAAkC,EAAE;CACpC,oBAAuC,EAAE;CACzC,YAAY;CACZ,iBAAiB;CAEjB;CACA;CACA;CACA;CAEA,YAAY,WAA6B;AACvC,OAAK,KAAK,UAAU;AACpB,OAAK,YAAY;AACjB,OAAK,YAAY,UAAU;AAC3B,OAAK,OAAO,UAAU;AACtB,OAAK,WAAW,UAAU;AAC1B,QAAA,cAAoB,IAAI,SAAiB,MAAM;AAC7C,SAAA,cAAoB;IACpB;AACF,QAAA,mBAAyB,IAAI,SAAiB,MAAM;AAClD,SAAA,mBAAyB;IACzB;;CAGJ,IAAI,OAA0B;EAC5B,MAAM,SAAS,MAAA;EACf,MAAM,UAAU,MAAA;EAChB,MAAM,gBAAgB,MAAA;EACtB,IAAI,SAAS;AACb,SAAO;GACL,CAAC,OAAO,iBAAiB;AACvB,WAAO,EACL,MAAM,OAAwC;AAC5C,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP;;GAEH,MAAM,MAAA,YAAkB,KAAK,KAAK,MAAA,YAAkB;GACpD,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,IAAI,cAAc;AAClB,eAAW,MAAM,SAAS,GACvB,OAAO,uBACL,EACC,MAAM,YAA6C;AACjD,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP,GACJ,EAAE;AACD,oBAAe;AACf,WAAM;;MAGX;GACF;;CAGH,IAAI,YAA+B;EACjC,MAAM,SAAS,MAAA;EACf,MAAM,UAAU,MAAA;EAChB,MAAM,gBAAgB,MAAA;EACtB,IAAI,SAAS;AACb,SAAO;GACL,CAAC,OAAO,iBAAiB;AACvB,WAAO,EACL,MAAM,OAAwC;AAC5C,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP;;GAEH,MAAM,MAAA,iBAAuB,KAAK,KAAK,MAAA,iBAAuB;GAC9D,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,IAAI,cAAc;AAClB,eAAW,MAAM,SAAS,GACvB,OAAO,uBACL,EACC,MAAM,YAA6C;AACjD,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP,GACJ,EAAE;AACD,oBAAe;AACf,WAAM;;MAGX;GACF;;CAGH,IAAI,QAA6B;EAC/B,MAAM,WAAW,YAAY;GAC3B,IAAI;AACJ,cAAW,MAAM,YAAY,MAAA,eAAqB,CAChD,SAAQ;AAEV,UAAO;MACL;AACJ,SAAO;IACJ,OAAO,sBAAsB,MAAA,eAAqB;GACnD,MAAM,QAAQ,KAAK,KAAK,QAAQ;GACjC;;CAGH,IAAI,YAA6B;EAC/B,MAAM,SAAS,MAAA;EACf,MAAM,WAAW,mBAAmB;AAClC,cAAW,MAAM,SAAS,OACxB,KACE,MAAM,UAAU,0BAChB,MAAM,QAAQ,SAAS,YAEvB,OAAM,MAAM;;AAIlB,SAAO;IACJ,OAAO,gBAAgB;GACxB,MAAM,OAAO,aAAa,eAAe;AACvC,QAAI;KACF,MAAM,QAA2C,EAAE;AACnD,gBAAW,MAAM,QAAQ,UAAU,CAAE,OAAM,KAAK,KAAK;AACrD,YAAO,cAAc,YAAY,MAAM,GAAI;aACpC,KAAK;AACZ,SAAI,WAAY,QAAO,WAAW,IAAI;AACtC,WAAM;;;GAGV,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,MAAM,QAA2C,EAAE;AACnD,eAAW,MAAM,QAAQ,UAAU,EAAE;AACnC,WAAM,KAAK,KAAK;AAChB,WAAM,CAAC,GAAG,MAAM;;MAGrB;GACF;;CAGH,IAAI,SAAiC;AACnC,SAAO,EAAE,OAAO,KAAK,QAAQ,MAAA,iBAAuB,CAAC,KAAK,KAAK,IAAI,EAAE;;CAGvE,IAAI,SAAyB;AAC3B,SAAO,KAAK,UAAU;;CAGxB,CAAC,OAAO,iBAAsD;AAC5D,SAAO,MAAA,OAAa,OAAO,gBAAgB;;CAG7C,KACE,aAGA,YACkC;AAClC,SAAO,MAAA,iBAAuB,CAAC,KAAK,aAAa,WAAW;;CAG9D,QAAA,gBAAuD;AACrD,aAAW,MAAM,SAAS,MAAA,OACxB,KAAI,MAAM,UAAU,mBAAmB,MAAM,MAC3C,OAAM,eAAe,MAAM,MAAM;WACxB,MAAM,UAAU,oBAAoB,MAAM,MACnD,OAAM,eAAe,MAAM,MAAM;;CAKvC,OAAA,kBAA6C;EAC3C,MAAM,gBAAqD,EAAE;EAC7D,IAAI,KAAyB,KAAK;EAClC,IAAI,QAAQ,eAAe,KAAK,UAAU,MAAM;EAChD,IAAI,WAAoC,EACtC,GAAI,KAAK,UAAU,kBAAkB,EAAE,EACxC;EACD,IAAI,eAAmC,KAAK,UAAU;AAEtD,aAAW,MAAM,SAAS,MAAA,OACxB,SAAQ,MAAM,OAAd;GACE,KAAK;AACH,SAAK,MAAM,MAAM;AACjB,QAAI,MAAM,MAAO,SAAQ,eAAe,MAAM,MAAM;AACpD;GACF,KAAK;AACH,kBAAc,MAAM,SAAS,MAAM;AACnC;GACF,KAAK,uBAAuB;IAC1B,MAAM,UAAU,cAAc,MAAM;AACpC,kBAAc,MAAM,SAAS,oBAAoB,SAAS,MAAM;AAChE;;GAEF,KAAK;AACH,kBAAc,MAAM,SAAS,MAAM;AACnC;GACF,KAAK,kBAAkB;AACrB,mBAAe,MAAM;AACrB,QAAI,MAAM,MAAO,SAAQ,eAAe,MAAM,MAAM;IACpD,MAAM,iBAAiB,mBAAmB,MAAM;AAChD,QAAI,kBAAkB,KACpB,YAAW;KAAE,GAAG;KAAU,GAAG;KAAgB;AAE/C;;GAEF,QACE;;AAIN,SAAO,IAAIa,yBAAAA,UAAU;GACnB;GACA,SAAS,cAAc,QACpB,UAAqC,SAAS,KAChD;GACD,gBAAgB;GAChB,mBAAmB;IACjB,WAAW,KAAK;IAChB,GAAI,KAAK,QAAQ,OAAO,EAAE,MAAM,KAAK,MAAM,GAAG,EAAE;IAChD,GAAI,KAAK,UAAU,SAAS,OACxB,EAAE,QAAQ,KAAK,UAAU,OAAO,GAChC,EAAE;IACN,GAAI,KAAK,YAAY,OAAO,EAAE,UAAU,KAAK,UAAU,GAAG,EAAE;IAC5D,GAAI,KAAK,UAAU,SAAS,OACxB,EAAE,OAAO,KAAK,UAAU,OAAO,GAC/B,EAAE;IACP;GACD,mBAAmB;IACjB,GAAI,gBAAgB,OAAO,EAAE,eAAe,cAAc,GAAG,EAAE;IAC/D,GAAG;IACH,gBAAgB;IACjB;GACF,CAAC;;CAGJ,CAAC,YAAY,OAAmC;AAC9C,QAAA,OAAa,KAAK,MAAM;;CAG1B,CAAC,gBAAgB,OAA4B;AAC3C,OAAK,OAAO,MAAM,OAAO,QAAQ,KAAK;;CAGxC,CAAC,WAAW,OAAqB;AAC/B,QAAA,WAAiB,KAAK,MAAM;EAI5B,MAAM,UAAU,MAAA,YAAkB,OAAO,GAAG,MAAA,YAAkB,OAAO;AACrE,OAAK,MAAM,UAAU,QAAS,SAAQ;;CAGxC,CAAC,gBAAgB,OAAqB;AACpC,QAAA,gBAAsB,KAAK,MAAM;EACjC,MAAM,UAAU,MAAA,iBAAuB,OACrC,GACA,MAAA,iBAAuB,OACxB;AACD,OAAK,MAAM,UAAU,QAAS,SAAQ;;CAGxC,CAAC,UAAgB;AACf,QAAA,WAAiB;AACjB,QAAA,gBAAsB;AACtB,QAAA,YAAkB,MAAA,WAAiB,KAAK,GAAG,CAAC;AAC5C,QAAA,iBAAuB,MAAA,gBAAsB,KAAK,GAAG,CAAC;EACtD,MAAM,cAAc,MAAA,YAAkB,OAAO,GAAG,MAAA,YAAkB,OAAO;AACzE,OAAK,MAAM,UAAU,YAAa,SAAQ;EAC1C,MAAM,mBAAmB,MAAA,iBAAuB,OAC9C,GACA,MAAA,iBAAuB,OACxB;AACD,OAAK,MAAM,UAAU,iBAAkB,SAAQ;AAC/C,QAAA,OAAa,OAAO;;CAGtB,CAAC,SAAe;AACd,OAAK,SAAS;;;AAclB,SAAgB,yBACd,SACwB;AACxB,QAAO,IAAI,MAAM,SAAS;EACxB,IAAI,QAAQ,MAAM;AAChB,OAAI,SAAS,OAAQ,QAAO,KAAA;GAC5B,MAAM,QAAQ,QAAQ,IAAI,QAAQ,MAAM,OAAO;AAC/C,UAAO,OAAO,UAAU,aAAa,MAAM,KAAK,OAAO,GAAG;;EAE5D,IAAI,QAAQ,MAAM;AAChB,OAAI,SAAS,OAAQ,QAAO;AAC5B,UAAO,QAAQ;;EAElB,CAAC;;AAiDJ,SAAS,WAAmC,OAAa;AACvD,QAAO,gBAAgB,MAAM;;AAG/B,SAAS,eACP,OACA,SACc;AACd,QAAO,0BACL,OACA,QACD;;AAGH,SAAS,kBACP,QACA,OACc;AACd,KAAI,OAAO,SAAS,MAAM,KACxB,QAAO,WAAW,MAAM;AAG1B,SAAQ,MAAM,MAAd;EACE,KAAK,OACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,MAAM,GAAG,UAAU,SAAS,OAAO,OAAO,KAAK,MAAM;GACtD;EACH,KAAK,YACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,WAAW,GAAG,eAAe,SAAS,OAAO,YAAY,KAAK,MAAM;GACrE;EACH,KAAK;EACL,KAAK,0BAA0B;GAc7B,MAAM,SAAS;IAAE,GAAG;IAAQ,GAAG;IAAO;AACtC,OAAI,MAAM,MAAM,QAAQ,QAAQ,UAAU,OAAO,MAAM,KACrD,QAAO,KAAK,OAAO;AAErB,OAAI,MAAM,QAAQ,QAAQ,UAAU,UAAU,OAAO,QAAQ,KAC3D,QAAO,OAAO,OAAO;AAEvB,UAAO,OAAO,IAAI,UAAU,SAAS,OAAO,OAAO,OAAO,KAAK,MAAM,QAAQ;AAC7E,UAAO;;EAET,QACE,QAAO;GACL,GAAG;GACH,GAAG;GACJ;;;AAIP,SAAS,cAAc,OAA8B;CACnD,MAAM,EAAE,WAAW,MAAM,SAAS,MAAM;CACxC,MAAM,eAAe,UAAU,KAAK,IAAI;CACxC,MAAM,QAAQ,UAAU,KAAK;CAC7B,MAAM,YAAY,KAAK,UAAU,kBAAmB,KAAK,MAAM,KAAM;AACrE,QAAO,GAAG,aAAa,IAAI,QAAQ,GAAG,IAAI,SAAS;;AAGrD,SAAS,uBAAuB,OAA4C;AAC1E,QAAO,MAAM,OAAO;;;;;AAMtB,IAAa,mBAAb,MAA8B;CAC5B,iCAAkC,IAAI,KAA+B;CACrE,wCAAyC,IAAI,KAAqB;CAClE,mDAAoD,IAAI,KAAqB;;;;;;CAO7E,QAAQ,OAA6C;EACnD,MAAM,OAAO,MAAM,OAAO;EAC1B,MAAM,mBAAmB,GAAG,MAAM,OAAO,UAAU,KAAK,IAAI,CAAC,IAAI,MAAM,OAAO,QAAQ;EACtF,MAAM,QAAQ,UAAU,KAAK;AAE7B,MAAI,KAAK,UAAU,iBAAiB;GAClC,MAAM,MAAM,cAAc,MAAM;AAChC,QAAK,sBAAsB,IAAI,kBAAkB,IAAI;GACrD,MAAM,UAA4B;IAChC,IAAI,KAAK;IACT,WAAW,CAAC,GAAG,MAAM,OAAO,UAAU;IACtC,MAAM,MAAM,OAAO;IACnB;IACA,OAAO,UAAU,KAAK;IACtB,UAAU,KAAK;IACf,QAAQ,EAAE;IACX;AACD,QAAK,eAAe,IAAI,KAAK,QAAQ;AACrC,UAAO;IAAE,MAAM;IAAiB;IAAK;IAAS;IAAO;;EAGvD,MAAM,SAAS,SAAS,OAAO,GAAG,iBAAiB,IAAI,UAAU,KAAA;EACjE,MAAM,cAAc,KAAK,sBAAsB,IAAI,iBAAiB;EACpE,MAAM,kBACJ,eAAe,OAAO,KAAK,eAAe,IAAI,YAAY,GAAG,KAAA;EAC/D,MAAM,aACH,UAAU,QAAQ,KAAK,eAAe,IAAI,OAAO,GAC9C,SACA,KAAA,OACH,SAAS,QAAQ,iBAAiB,SAAS,OACxC,cACA,KAAA;AACN,MAAI,CAAC,WAAW;GAOd,MAAM,eAAe,UAAU,GAAG,iBAAiB;AACnD,QAAK,sBAAsB,IAAI,kBAAkB,aAAa;GAC9D,MAAM,YAA8B;IAClC,IAAI,KAAK;IACT,WAAW,CAAC,GAAG,MAAM,OAAO,UAAU;IACtC,MAAM,MAAM,OAAO;IACnB;IACA,QAAQ,EAAE;IACX;AACD,QAAK,eAAe,IAAI,cAAc,UAAU;AAChD,UAAO,KAAK,QAAQ,MAAM;;EAG5B,MAAM,UAAU,KAAK,eAAe,IAAI,UAAU;AAClD,MAAI,CAAC,QACH,OAAM,IAAI,MAAM,yCAAyC,YAAY;AAGvE,MAAK,KAA4B,UAAU,SAAS;AAClD,WAAQ,QAAS,KAA+B;AAChD,UAAO;IACL,MAAM;IACN,KAAK;IACL;IACA;IACD;;AAGH,UAAQ,KAAK,OAAb;GACE,KAAK;AACH,YAAQ,OAAO,KAAK,SAAS,WAAW,KAAK,QAAQ;AACrD,SAAK,iCAAiC,IACpC,cAAc,WAAW,KAAK,OAAO,KAAK,QAAQ,KAAK,EACvD,KAAK,MACN;AACD,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO,KAAK;KACZ,OAAO,KAAK;KACZ;KACD;GAEH,KAAK,uBAAuB;IAC1B,MAAM,aAAa;IAInB,MAAM,aACJ,WAAW,YACV,WAAW,SAAS,OACjB,eAAe,WAAW,OAAO,QAAQ,OAAO,KAAK,OAAO,GAC5D,KAAA;AACN,QAAI,cAAc,KAChB,OAAM,IAAI,MAAM,+CAA+C;IAEjE,MAAM,cAAc,KAAK,kBACvB,WACA,QAAQ,QACR,KAAK,OACL,WAAW,KACZ;IACD,MAAM,UAAU,QAAQ,OAAO;AAC/B,YAAQ,OAAO,eACb,WAAW,WAAW,OAClB,WAAW,OACT,WAAW,WAAW,QAAQ,GAC9B,kBAAkB,SAAS,WAAW,QAAQ,GAC/C,oBACC,SACA,KAID;AACP,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO;KACP,OAAO;KACP;KACD;;GAEH,KAAK,wBAAwB;IAC3B,MAAM,cAAc,KAAK,wBACvB,WACA,KAAK,OACL,KAAK,QAAQ,KACd;AACD,YAAQ,OAAO,eAAe,WAAW,KAAK,QAAQ;AACtD,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO;KACP,OAAO,KAAK;KACZ;KACD;;GAEH,KAAK;AACH,YAAQ,QAAQ,UAAU,KAAK,IAAI,QAAQ;AAC3C,YAAQ,iBAAiB,mBAAmB,KAAK;AACjD,YAAQ,eAAe,KAAK;AAC5B,SAAK,mBAAmB,WAAW,iBAAiB;AACpD,SAAK,uBAAuB,UAAU;AACtC,WAAO;KACL,MAAM;KACN,KAAK;KACL,SAAS,gBAAgB,QAAQ;KACjC;KACD;GAEH,KAAK;AACH,YAAQ,QAAQ;KAAE,SAAS,KAAK;KAAS,MAAM,KAAK;KAAM;AAC1D,SAAK,mBAAmB,WAAW,iBAAiB;AACpD,SAAK,uBAAuB,UAAU;AACtC,WAAO;KACL,MAAM;KACN,KAAK;KACL,SAAS,gBAAgB,QAAQ;KACjC;KACD;;;CAKP,mBACE,WACA,kBACM;AACN,OAAK,eAAe,OAAO,UAAU;AACrC,MAAI,KAAK,sBAAsB,IAAI,iBAAiB,KAAK,UACvD,MAAK,sBAAsB,OAAO,iBAAiB;;CAIvD,kBACE,WACA,QACA,eACA,WACQ;EACR,MAAM,UAAU,OAAO;AACvB,MACE,WAAW,QACX,QAAQ,SAAS,aACjB,wBAAwB,QAAQ,MAAM,UAAU,EAChD;AACA,QAAK,iCAAiC,IACpC,cAAc,WAAW,eAAe,UAAU,EAClD,cACD;AACD,UAAO;;EAGT,MAAM,MAAM,cAAc,WAAW,eAAe,UAAU;EAC9D,MAAM,WAAW,KAAK,iCAAiC,IAAI,IAAI;AAC/D,MAAI,YAAY,KAAM,QAAO;EAE7B,MAAM,YAAY,OAAO;AACzB,OAAK,iCAAiC,IAAI,KAAK,UAAU;AACzD,SAAO;;CAGT,wBACE,WACA,eACA,WACQ;EACR,MAAM,MAAM,cAAc,WAAW,eAAe,UAAU;EAC9D,MAAM,WAAW,KAAK,iCAAiC,IAAI,IAAI;AAC/D,MAAI,YAAY,KAAM,QAAO;AAE7B,OAAK,iCAAiC,IAAI,KAAK,cAAc;AAC7D,SAAO;;CAGT,uBAA+B,WAAyB;EACtD,MAAM,SAAS,GAAG,UAAU;AAC5B,OAAK,MAAM,OAAO,KAAK,iCAAiC,MAAM,CAC5D,KAAI,IAAI,WAAW,OAAO,CACxB,MAAK,iCAAiC,OAAO,IAAI;;;AAKzD,SAAS,cACP,WACA,eACA,WACQ;AACR,QAAO,GAAG,UAAU,IAAI,cAAc,IAAI;;AAG5C,SAAS,wBACP,aACA,UACS;CACT,MAAM,gBAAgB,IAAI,IAAI;EAC5B;EACA;EACA;EACA;EACD,CAAC;CACF,MAAM,sBAAsB,IAAI,IAAI,CAClC,oBACA,yBACD,CAAC;AAEF,QACG,cAAc,IAAI,YAAY,IAAI,cAAc,IAAI,SAAS,IAC7D,oBAAoB,IAAI,YAAY,IAAI,oBAAoB,IAAI,SAAS;;;;;;;AAS9E,IAAa,4BAAb,MAAuC;CACrC,aAAsB,IAAI,kBAAkB;CAC5C,mCAA4B,IAAI,KAA+B;;;;;;CAO/D,QAAQ,OAAoD;EAC1D,MAAM,SAAS,MAAA,UAAgB,QAAQ,MAAM;AAC7C,MAAI,UAAU,KAAM,QAAO,KAAA;AAE3B,UAAQ,OAAO,MAAf;GACE,KAAK,iBAAiB;IACpB,MAAM,YAAY,IAAI,iBAAiB,OAAO,QAAQ;AACtD,cAAU,gBAAgB,OAAO,MAAM;AACvC,cAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,UAAA,gBAAsB,IAAI,OAAO,KAAK,UAAU;AAChD,WAAO;;GAET,KAAK,uBAAuB;IAC1B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D,QACE,aACA,OAAO,MAAM,SAAS,UACtB,UAAU,OAAO,SACjB,OAAO,MAAM,KAEb,WAAU,WAAW,OAAO,MAAM,KAAK;AAEzC,QACE,aACA,OAAO,MAAM,SAAS,eACtB,eAAe,OAAO,SACtB,OAAO,MAAM,UAEb,WAAU,gBAAgB,OAAO,MAAM,UAAU;AAEnD;;GAEF,KAAK,uBAAuB;IAC1B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,CAAC,UAAW,QAAO,KAAA;AACvB,cAAU,gBAAgB,OAAO,MAAM;AACvC,cAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,QAAI,OAAO,MAAM,SAAS,UAAU,UAAU,OAAO,MACnD,WAAU,WAAW,OAAO,MAAM,KAAK;AAEzC,QAAI,OAAO,MAAM,SAAS,eAAe,eAAe,OAAO,MAC7D,WAAU,gBAAgB,OAAO,MAAM,UAAU;AAEnD;;GAEF,KAAK,wBAAwB;IAC3B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D;;GAEF,KAAK,SAAS;IACZ,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D;;GAEF,KAAK,kBAAkB;IACrB,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,eAAU,SAAS;AACnB,WAAA,gBAAsB,OAAO,OAAO,IAAI;;AAE1C;;GAEF,KAAK,iBAAiB;IACpB,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,eAAU,QAAQ;AAClB,WAAA,gBAAsB,OAAO,OAAO,IAAI;;AAE1C"}
|
|
@@ -19,9 +19,11 @@ interface AssembledMessage {
|
|
|
19
19
|
namespace: string[];
|
|
20
20
|
blocks: ContentBlock[];
|
|
21
21
|
node?: string;
|
|
22
|
+
runId?: string;
|
|
22
23
|
usage?: UsageInfo;
|
|
23
24
|
metadata?: MessageMetadata;
|
|
24
25
|
finishMetadata?: Record<string, any>;
|
|
26
|
+
finishReason?: string;
|
|
25
27
|
error?: {
|
|
26
28
|
message: string;
|
|
27
29
|
code?: string;
|
|
@@ -134,6 +136,7 @@ declare class MessageAssembler {
|
|
|
134
136
|
* @param event - Incoming `messages` event to fold into the assembler state.
|
|
135
137
|
*/
|
|
136
138
|
consume(event: MessagesEvent): MessageAssemblyUpdate;
|
|
139
|
+
private clearActiveMessage;
|
|
137
140
|
private resolveBlockIndex;
|
|
138
141
|
private resolveFinishBlockIndex;
|
|
139
142
|
private clearBlockIndexAliases;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"messages.d.cts","names":[],"sources":["../../../src/client/stream/messages.ts"],"mappings":";;;;;KAmBK,iBAAA,GAAoB,aAAA,WACvB,WAAA;EAAwB,IAAA,EAAM,aAAA;AAAA;AAAA,KAE3B,mBAAA,GAAsB,aAAA,CAAc,aAAA,IACvC,WAAA,CAAY,aAAA;AAAA,KAET,eAAA,GAAkB,aAAA,CAAc,cAAA,CAAiB,KAAA,CAAM,QAAA,IAC1D,WAAA,CAAY,KAAA,CAAM,cAAA,CAAiB,KAAA,CAAM,QAAA;EACvC,IAAA,EAAM,aAAA,CAAc,KAAA,CAAM,cAAA,CAAiB,KAAA,CAAM,QAAA;AAAA;;;;;
|
|
1
|
+
{"version":3,"file":"messages.d.cts","names":[],"sources":["../../../src/client/stream/messages.ts"],"mappings":";;;;;KAmBK,iBAAA,GAAoB,aAAA,WACvB,WAAA;EAAwB,IAAA,EAAM,aAAA;AAAA;AAAA,KAE3B,mBAAA,GAAsB,aAAA,CAAc,aAAA,IACvC,WAAA,CAAY,aAAA;AAAA,KAET,eAAA,GAAkB,aAAA,CAAc,cAAA,CAAiB,KAAA,CAAM,QAAA,IAC1D,WAAA,CAAY,KAAA,CAAM,cAAA,CAAiB,KAAA,CAAM,QAAA;EACvC,IAAA,EAAM,aAAA,CAAc,KAAA,CAAM,cAAA,CAAiB,KAAA,CAAM,QAAA;AAAA;;;;;UAoJpC,gBAAA;EACf,EAAA;EACA,SAAA;EACA,MAAA,EAAQ,YAAA;EACR,IAAA;EACA,KAAA;EACA,KAAA,GAAQ,SAAA;EACR,QAAA,GAAW,eAAA;EACX,cAAA,GAAiB,MAAA;EACjB,YAAA;EACA,KAAA;IACE,OAAA;IACA,IAAA;EAAA;AAAA;;;;;;cASE,SAAA;AAAA,cACA,cAAA;AAAA,cACA,UAAA;AAAA,cACA,cAAA;AAAA,cACA,MAAA;AAAA,cACA,KAAA;;;;;;;;;;;;;cAcO,gBAAA,YACA,aAAA,CAAc,oBAAA,GAAuB,WAAA,CAAY,SAAA;EAAA;WAEnD,EAAA;EAAA,SACA,SAAA;EACT,IAAA;EAAA,SACS,QAAA,EAAU,eAAA;EAAA,SACV,SAAA,EAAW,gBAAA;EAepB,WAAA,CAAY,SAAA,EAAW,gBAAA;EAAA,IAcnB,IAAA,CAAA,GAAQ,iBAAA;EAAA,IAqDR,SAAA,CAAA,GAAa,iBAAA;EAAA,IAqDb,KAAA,CAAA,GAAS,mBAAA;EAAA,IAcT,SAAA,CAAA,GAAa,eAAA;EAAA,IAoCb,MAAA,CAAA,GAAU,WAAA,CAAY,SAAA;EAAA,IAItB,MAAA,CAAA,GAAU,YAAA;EAAA,CAIb,MAAA,CAAO,aAAA,KAAkB,aAAA,CAAc,oBAAA;EAIxC,IAAA,YAAgB,SAAA,mBAAA,CACd,WAAA,KACM,KAAA,EAAO,SAAA,KAAc,QAAA,GAAW,WAAA,CAAY,QAAA,WAElD,UAAA,KAAe,MAAA,cAAoB,QAAA,GAAW,WAAA,CAAY,QAAA,YACzD,WAAA,CAAY,QAAA,GAAW,QAAA;EAAA,CA+EzB,UAAA,EAAY,KAAA,EAAO,oBAAA;EAAA,CAInB,cAAA,EAAgB,KAAA,EAAO,aAAA;EAAA,CAIvB,SAAA,EAAW,KAAA;EAAA,CASX,cAAA,EAAgB,KAAA;EAAA,CAShB,MAAA;EAAA,CAeA,KAAA;AAAA;;;;;;;;;KAaS,sBAAA,GAAyB,IAAA,CAAK,gBAAA;;;;;KAsB9B,qBAAA;EAEN,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA,EAAO,aAAA;AAAA;EAGP,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA;EACA,KAAA,EAAO,YAAA;EACP,KAAA,EAAO,aAAA;AAAA;EAGP,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA;EACA,KAAA,EAAO,qBAAA;EACP,KAAA,EAAO,aAAA;AAAA;EAGP,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA,EAAO,aAAA;AAAA;EAGP,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA,EAAO,aAAA;AAAA;EAGP,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA,EAAO,aAAA;AAAA;;;;cAsFA,gBAAA;EAAA,iBACM,cAAA;EAAA,iBACA,qBAAA;EAAA,iBACA,gCAAA;EA9VJ;;;;;EAqWb,OAAA,CAAQ,KAAA,EAAO,aAAA,GAAgB,qBAAA;EAAA,QAuKvB,kBAAA;EAAA,QAUA,iBAAA;EAAA,QA4BA,uBAAA;EAAA,QAaA,sBAAA;AAAA"}
|
|
@@ -19,9 +19,11 @@ interface AssembledMessage {
|
|
|
19
19
|
namespace: string[];
|
|
20
20
|
blocks: ContentBlock$1[];
|
|
21
21
|
node?: string;
|
|
22
|
+
runId?: string;
|
|
22
23
|
usage?: UsageInfo;
|
|
23
24
|
metadata?: MessageMetadata;
|
|
24
25
|
finishMetadata?: Record<string, any>;
|
|
26
|
+
finishReason?: string;
|
|
25
27
|
error?: {
|
|
26
28
|
message: string;
|
|
27
29
|
code?: string;
|
|
@@ -134,6 +136,7 @@ declare class MessageAssembler {
|
|
|
134
136
|
* @param event - Incoming `messages` event to fold into the assembler state.
|
|
135
137
|
*/
|
|
136
138
|
consume(event: MessagesEvent): MessageAssemblyUpdate;
|
|
139
|
+
private clearActiveMessage;
|
|
137
140
|
private resolveBlockIndex;
|
|
138
141
|
private resolveFinishBlockIndex;
|
|
139
142
|
private clearBlockIndexAliases;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"messages.d.ts","names":[],"sources":["../../../src/client/stream/messages.ts"],"mappings":";;;;;KAmBK,iBAAA,GAAoB,aAAA,WACvB,WAAA;EAAwB,IAAA,EAAM,aAAA;AAAA;AAAA,KAE3B,mBAAA,GAAsB,aAAA,CAAc,aAAA,IACvC,WAAA,CAAY,aAAA;AAAA,KAET,eAAA,GAAkB,aAAA,CAAc,YAAA,CAAiB,KAAA,CAAM,QAAA,IAC1D,WAAA,CAAY,KAAA,CAAM,YAAA,CAAiB,KAAA,CAAM,QAAA;EACvC,IAAA,EAAM,aAAA,CAAc,KAAA,CAAM,YAAA,CAAiB,KAAA,CAAM,QAAA;AAAA;;;;;
|
|
1
|
+
{"version":3,"file":"messages.d.ts","names":[],"sources":["../../../src/client/stream/messages.ts"],"mappings":";;;;;KAmBK,iBAAA,GAAoB,aAAA,WACvB,WAAA;EAAwB,IAAA,EAAM,aAAA;AAAA;AAAA,KAE3B,mBAAA,GAAsB,aAAA,CAAc,aAAA,IACvC,WAAA,CAAY,aAAA;AAAA,KAET,eAAA,GAAkB,aAAA,CAAc,YAAA,CAAiB,KAAA,CAAM,QAAA,IAC1D,WAAA,CAAY,KAAA,CAAM,YAAA,CAAiB,KAAA,CAAM,QAAA;EACvC,IAAA,EAAM,aAAA,CAAc,KAAA,CAAM,YAAA,CAAiB,KAAA,CAAM,QAAA;AAAA;;;;;UAoJpC,gBAAA;EACf,EAAA;EACA,SAAA;EACA,MAAA,EAAQ,cAAA;EACR,IAAA;EACA,KAAA;EACA,KAAA,GAAQ,SAAA;EACR,QAAA,GAAW,eAAA;EACX,cAAA,GAAiB,MAAA;EACjB,YAAA;EACA,KAAA;IACE,OAAA;IACA,IAAA;EAAA;AAAA;;;;;;cASE,SAAA;AAAA,cACA,cAAA;AAAA,cACA,UAAA;AAAA,cACA,cAAA;AAAA,cACA,MAAA;AAAA,cACA,KAAA;;;;;;;;;;;;;cAcO,gBAAA,YACA,aAAA,CAAc,oBAAA,GAAuB,WAAA,CAAY,SAAA;EAAA;WAEnD,EAAA;EAAA,SACA,SAAA;EACT,IAAA;EAAA,SACS,QAAA,EAAU,eAAA;EAAA,SACV,SAAA,EAAW,gBAAA;EAepB,WAAA,CAAY,SAAA,EAAW,gBAAA;EAAA,IAcnB,IAAA,CAAA,GAAQ,iBAAA;EAAA,IAqDR,SAAA,CAAA,GAAa,iBAAA;EAAA,IAqDb,KAAA,CAAA,GAAS,mBAAA;EAAA,IAcT,SAAA,CAAA,GAAa,eAAA;EAAA,IAoCb,MAAA,CAAA,GAAU,WAAA,CAAY,SAAA;EAAA,IAItB,MAAA,CAAA,GAAU,cAAA;EAAA,CAIb,MAAA,CAAO,aAAA,KAAkB,aAAA,CAAc,oBAAA;EAIxC,IAAA,YAAgB,SAAA,mBAAA,CACd,WAAA,KACM,KAAA,EAAO,SAAA,KAAc,QAAA,GAAW,WAAA,CAAY,QAAA,WAElD,UAAA,KAAe,MAAA,cAAoB,QAAA,GAAW,WAAA,CAAY,QAAA,YACzD,WAAA,CAAY,QAAA,GAAW,QAAA;EAAA,CA+EzB,UAAA,EAAY,KAAA,EAAO,oBAAA;EAAA,CAInB,cAAA,EAAgB,KAAA,EAAO,aAAA;EAAA,CAIvB,SAAA,EAAW,KAAA;EAAA,CASX,cAAA,EAAgB,KAAA;EAAA,CAShB,MAAA;EAAA,CAeA,KAAA;AAAA;;;;;;;;;KAaS,sBAAA,GAAyB,IAAA,CAAK,gBAAA;;;;;KAsB9B,qBAAA;EAEN,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA,EAAO,aAAA;AAAA;EAGP,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA;EACA,KAAA,EAAO,cAAA;EACP,KAAA,EAAO,aAAA;AAAA;EAGP,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA;EACA,KAAA,EAAO,qBAAA;EACP,KAAA,EAAO,aAAA;AAAA;EAGP,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA,EAAO,aAAA;AAAA;EAGP,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA,EAAO,aAAA;AAAA;EAGP,IAAA;EACA,GAAA;EACA,OAAA,EAAS,gBAAA;EACT,KAAA,EAAO,aAAA;AAAA;;;;cAsFA,gBAAA;EAAA,iBACM,cAAA;EAAA,iBACA,qBAAA;EAAA,iBACA,gCAAA;EA9VJ;;;;;EAqWb,OAAA,CAAQ,KAAA,EAAO,aAAA,GAAgB,qBAAA;EAAA,QAuKvB,kBAAA;EAAA,QAUA,iBAAA;EAAA,QA4BA,uBAAA;EAAA,QAaA,sBAAA;AAAA"}
|
|
@@ -88,6 +88,23 @@ function normalizeUsage(usage) {
|
|
|
88
88
|
total_tokens: usage.total_tokens ?? 0
|
|
89
89
|
};
|
|
90
90
|
}
|
|
91
|
+
function readRunId(data) {
|
|
92
|
+
const record = data;
|
|
93
|
+
return typeof record.run_id === "string" ? record.run_id : void 0;
|
|
94
|
+
}
|
|
95
|
+
function readUsage(data) {
|
|
96
|
+
const usage = data.usage;
|
|
97
|
+
return usage != null && typeof usage === "object" && !Array.isArray(usage) ? usage : void 0;
|
|
98
|
+
}
|
|
99
|
+
function readFinishMetadata(data) {
|
|
100
|
+
const record = data;
|
|
101
|
+
const isRecord = (value) => value != null && typeof value === "object" && !Array.isArray(value);
|
|
102
|
+
const merged = {
|
|
103
|
+
...isRecord(record.responseMetadata) ? record.responseMetadata : {},
|
|
104
|
+
...isRecord(record.metadata) ? record.metadata : {}
|
|
105
|
+
};
|
|
106
|
+
return Object.keys(merged).length > 0 ? merged : void 0;
|
|
107
|
+
}
|
|
91
108
|
/**
|
|
92
109
|
* Symbol keys for assembler → StreamingMessage communication.
|
|
93
110
|
* Module-private: invisible to external consumers, accessible to
|
|
@@ -290,10 +307,10 @@ var StreamingMessage = class {
|
|
|
290
307
|
}
|
|
291
308
|
async #assembleMessage() {
|
|
292
309
|
const contentBlocks = [];
|
|
293
|
-
let id;
|
|
294
|
-
let usage;
|
|
295
|
-
let metadata = {};
|
|
296
|
-
let finishReason;
|
|
310
|
+
let id = this.id;
|
|
311
|
+
let usage = normalizeUsage(this.assembled.usage);
|
|
312
|
+
let metadata = { ...this.assembled.finishMetadata ?? {} };
|
|
313
|
+
let finishReason = this.assembled.finishReason;
|
|
297
314
|
for await (const event of this.#events) switch (event.event) {
|
|
298
315
|
case "message-start":
|
|
299
316
|
id = event.id ?? id;
|
|
@@ -310,23 +327,32 @@ var StreamingMessage = class {
|
|
|
310
327
|
case "content-block-finish":
|
|
311
328
|
contentBlocks[event.index] = event.content;
|
|
312
329
|
break;
|
|
313
|
-
case "message-finish":
|
|
330
|
+
case "message-finish": {
|
|
314
331
|
finishReason = event.reason;
|
|
315
332
|
if (event.usage) usage = normalizeUsage(event.usage);
|
|
316
|
-
|
|
333
|
+
const finishMetadata = readFinishMetadata(event);
|
|
334
|
+
if (finishMetadata != null) metadata = {
|
|
317
335
|
...metadata,
|
|
318
|
-
...
|
|
336
|
+
...finishMetadata
|
|
319
337
|
};
|
|
320
338
|
break;
|
|
339
|
+
}
|
|
321
340
|
default: break;
|
|
322
341
|
}
|
|
323
342
|
return new AIMessage({
|
|
324
343
|
id,
|
|
325
344
|
content: contentBlocks.filter((block) => block != null),
|
|
326
345
|
usage_metadata: usage,
|
|
346
|
+
additional_kwargs: {
|
|
347
|
+
namespace: this.namespace,
|
|
348
|
+
...this.node != null ? { node: this.node } : {},
|
|
349
|
+
...this.assembled.runId != null ? { run_id: this.assembled.runId } : {},
|
|
350
|
+
...this.metadata != null ? { metadata: this.metadata } : {},
|
|
351
|
+
...this.assembled.usage != null ? { usage: this.assembled.usage } : {}
|
|
352
|
+
},
|
|
327
353
|
response_metadata: {
|
|
354
|
+
...finishReason != null ? { finish_reason: finishReason } : {},
|
|
328
355
|
...metadata,
|
|
329
|
-
...finishReason ? { finish_reason: finishReason } : {},
|
|
330
356
|
output_version: "v1"
|
|
331
357
|
}
|
|
332
358
|
});
|
|
@@ -414,8 +440,9 @@ function applyContentDelta(target, delta) {
|
|
|
414
440
|
function messageKeyFor(event) {
|
|
415
441
|
const { namespace, node, data } = event.params;
|
|
416
442
|
const namespaceKey = namespace.join("/");
|
|
443
|
+
const runId = readRunId(data);
|
|
417
444
|
const messageId = data.event === "message-start" ? data.id ?? "" : "";
|
|
418
|
-
return `${namespaceKey}::${node ?? ""}::${messageId}`;
|
|
445
|
+
return `${namespaceKey}::${node ?? ""}::${runId ?? messageId}`;
|
|
419
446
|
}
|
|
420
447
|
function toChatModelStreamEvent(event) {
|
|
421
448
|
return event.params.data;
|
|
@@ -435,6 +462,7 @@ var MessageAssembler = class {
|
|
|
435
462
|
consume(event) {
|
|
436
463
|
const data = event.params.data;
|
|
437
464
|
const namespaceNodeKey = `${event.params.namespace.join("/")}::${event.params.node ?? ""}`;
|
|
465
|
+
const runId = readRunId(data);
|
|
438
466
|
if (data.event === "message-start") {
|
|
439
467
|
const key = messageKeyFor(event);
|
|
440
468
|
this.activeByNamespaceNode.set(namespaceNodeKey, key);
|
|
@@ -442,6 +470,8 @@ var MessageAssembler = class {
|
|
|
442
470
|
id: data.id,
|
|
443
471
|
namespace: [...event.params.namespace],
|
|
444
472
|
node: event.params.node,
|
|
473
|
+
runId,
|
|
474
|
+
usage: readUsage(data),
|
|
445
475
|
metadata: data.metadata,
|
|
446
476
|
blocks: []
|
|
447
477
|
};
|
|
@@ -453,14 +483,18 @@ var MessageAssembler = class {
|
|
|
453
483
|
event
|
|
454
484
|
};
|
|
455
485
|
}
|
|
456
|
-
const
|
|
486
|
+
const runKey = runId != null ? `${namespaceNodeKey}::${runId}` : void 0;
|
|
487
|
+
const fallbackKey = this.activeByNamespaceNode.get(namespaceNodeKey);
|
|
488
|
+
const fallbackMessage = fallbackKey != null ? this.activeMessages.get(fallbackKey) : void 0;
|
|
489
|
+
const activeKey = (runKey != null && this.activeMessages.has(runKey) ? runKey : void 0) ?? (runId == null || fallbackMessage?.runId == null ? fallbackKey : void 0);
|
|
457
490
|
if (!activeKey) {
|
|
458
|
-
const syntheticKey = `${namespaceNodeKey}::`;
|
|
491
|
+
const syntheticKey = runKey ?? `${namespaceNodeKey}::`;
|
|
459
492
|
this.activeByNamespaceNode.set(namespaceNodeKey, syntheticKey);
|
|
460
493
|
const synthetic = {
|
|
461
494
|
id: data.id,
|
|
462
495
|
namespace: [...event.params.namespace],
|
|
463
496
|
node: event.params.node,
|
|
497
|
+
runId,
|
|
464
498
|
blocks: []
|
|
465
499
|
};
|
|
466
500
|
this.activeMessages.set(syntheticKey, synthetic);
|
|
@@ -518,10 +552,10 @@ var MessageAssembler = class {
|
|
|
518
552
|
};
|
|
519
553
|
}
|
|
520
554
|
case "message-finish":
|
|
521
|
-
message.usage = data.usage;
|
|
522
|
-
message.finishMetadata = data
|
|
523
|
-
|
|
524
|
-
this.
|
|
555
|
+
message.usage = readUsage(data) ?? message.usage;
|
|
556
|
+
message.finishMetadata = readFinishMetadata(data);
|
|
557
|
+
message.finishReason = data.reason;
|
|
558
|
+
this.clearActiveMessage(activeKey, namespaceNodeKey);
|
|
525
559
|
this.clearBlockIndexAliases(activeKey);
|
|
526
560
|
return {
|
|
527
561
|
kind: "message-finish",
|
|
@@ -534,8 +568,7 @@ var MessageAssembler = class {
|
|
|
534
568
|
message: data.message,
|
|
535
569
|
code: data.code
|
|
536
570
|
};
|
|
537
|
-
this.
|
|
538
|
-
this.activeByNamespaceNode.delete(namespaceNodeKey);
|
|
571
|
+
this.clearActiveMessage(activeKey, namespaceNodeKey);
|
|
539
572
|
this.clearBlockIndexAliases(activeKey);
|
|
540
573
|
return {
|
|
541
574
|
kind: "message-error",
|
|
@@ -545,6 +578,10 @@ var MessageAssembler = class {
|
|
|
545
578
|
};
|
|
546
579
|
}
|
|
547
580
|
}
|
|
581
|
+
clearActiveMessage(activeKey, namespaceNodeKey) {
|
|
582
|
+
this.activeMessages.delete(activeKey);
|
|
583
|
+
if (this.activeByNamespaceNode.get(namespaceNodeKey) === activeKey) this.activeByNamespaceNode.delete(namespaceNodeKey);
|
|
584
|
+
}
|
|
548
585
|
resolveBlockIndex(activeKey, blocks, protocolIndex, blockType) {
|
|
549
586
|
const current = blocks[protocolIndex];
|
|
550
587
|
if (current == null || current.type === blockType || areCompatibleBlockTypes(current.type, blockType)) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"messages.js","names":["#events","#textPromise","#reasoningPromise","#resolveText","#resolveReasoning","#textChunks","#textWaiters","#textDone","#reasoningChunks","#reasoningWaiters","#reasoningDone","#usageIterator","#assembleMessage","#assembler","#activeStreaming"],"sources":["../../../src/client/stream/messages.ts"],"sourcesContent":["import {\n AIMessage,\n type ContentBlock as CoreContentBlock,\n type UsageMetadata,\n} from \"@langchain/core/messages\";\nimport type {\n ChatModelStreamEvent,\n ContentBlockDelta as CoreContentBlockDelta,\n} from \"@langchain/core/language_models/event\";\nimport type {\n ContentBlock,\n FinalizedContentBlock,\n MessageMetadata,\n MessagesEvent,\n UsageInfo,\n} from \"@langchain/protocol\";\n\nimport { MultiCursorBuffer } from \"./multi-cursor-buffer.js\";\n\ntype TextContentStream = AsyncIterable<string> &\n PromiseLike<string> & { full: AsyncIterable<string> };\n\ntype UsageMetadataStream = AsyncIterable<UsageMetadata> &\n PromiseLike<UsageMetadata | undefined>;\n\ntype ToolCallsStream = AsyncIterable<CoreContentBlock.Tools.ToolCall> &\n PromiseLike<Array<CoreContentBlock.Tools.ToolCall>> & {\n full: AsyncIterable<Array<CoreContentBlock.Tools.ToolCall>>;\n };\n\nfunction applyCoreContentDelta(\n target: CoreContentBlock,\n delta: CoreContentBlock\n): CoreContentBlock {\n if (target.type !== delta.type) {\n return structuredClone(delta);\n }\n\n switch (delta.type) {\n case \"text\":\n return {\n ...target,\n ...delta,\n text: `${\"text\" in target ? target.text : \"\"}${delta.text}`,\n } as CoreContentBlock;\n case \"reasoning\":\n return {\n ...target,\n ...delta,\n reasoning: `${\"reasoning\" in target ? target.reasoning : \"\"}${delta.reasoning}`,\n } as CoreContentBlock;\n case \"tool_call_chunk\":\n case \"server_tool_call_chunk\": {\n const merged = { ...target, ...delta } as Record<string, unknown>;\n if (delta.id == null && \"id\" in target && target.id != null) {\n merged.id = target.id;\n }\n if (delta.name == null && \"name\" in target && target.name != null) {\n merged.name = target.name;\n }\n merged.args = `${(\"args\" in target ? target.args : \"\") ?? \"\"}${delta.args ?? \"\"}`;\n return merged as unknown as CoreContentBlock;\n }\n default:\n return { ...target, ...delta } as CoreContentBlock;\n }\n}\n\nfunction coreContentBlockFromDelta(\n delta: CoreContentBlockDelta,\n current?: CoreContentBlock\n): CoreContentBlock {\n switch (delta.type) {\n case \"text-delta\":\n return { type: \"text\", text: delta.text } as CoreContentBlock;\n case \"reasoning-delta\":\n return {\n type: \"reasoning\",\n reasoning: delta.reasoning,\n } as CoreContentBlock;\n case \"data-delta\": {\n const merged = { ...(current ?? {}), data: delta.data } as Record<\n string,\n unknown\n >;\n if (delta.encoding) merged.encoding = delta.encoding;\n return merged as unknown as CoreContentBlock;\n }\n case \"block-delta\":\n return delta.fields as CoreContentBlock;\n }\n}\n\nfunction applyCoreEventDelta(\n current: CoreContentBlock | undefined,\n event: Extract<ChatModelStreamEvent, { event: \"content-block-delta\" }> & {\n content?: CoreContentBlock;\n }\n): CoreContentBlock {\n if (event.content) {\n return current\n ? applyCoreContentDelta(current, event.content)\n : event.content;\n }\n\n switch (event.delta.type) {\n case \"text-delta\":\n if (current?.type === \"text\") {\n return {\n ...current,\n text: `${\"text\" in current ? current.text : \"\"}${event.delta.text}`,\n } as CoreContentBlock;\n }\n return coreContentBlockFromDelta(event.delta, current);\n case \"reasoning-delta\":\n if (current?.type === \"reasoning\") {\n return {\n ...current,\n reasoning: `${\"reasoning\" in current ? current.reasoning : \"\"}${event.delta.reasoning}`,\n } as CoreContentBlock;\n }\n return coreContentBlockFromDelta(event.delta, current);\n case \"data-delta\": {\n const merged = { ...(current ?? {}) } as Record<string, unknown>;\n merged.data = `${(merged.data as string | undefined) ?? \"\"}${event.delta.data}`;\n if (event.delta.encoding) merged.encoding = event.delta.encoding;\n return merged as unknown as CoreContentBlock;\n }\n case \"block-delta\":\n return {\n ...(current ?? {}),\n ...event.delta.fields,\n } as CoreContentBlock;\n }\n}\n\nfunction normalizeUsage(\n usage: UsageInfo | Partial<UsageMetadata> | undefined\n): UsageMetadata | undefined {\n if (!usage) return undefined;\n return {\n ...usage,\n input_tokens: usage.input_tokens ?? 0,\n output_tokens: usage.output_tokens ?? 0,\n total_tokens: usage.total_tokens ?? 0,\n };\n}\n\n/**\n * Mutable view of a streamed message as message and content-block events are\n * assembled into a single structure.\n */\nexport interface AssembledMessage {\n id: string;\n namespace: string[];\n blocks: ContentBlock[];\n node?: string;\n usage?: UsageInfo;\n metadata?: MessageMetadata;\n finishMetadata?: Record<string, any>;\n error?: {\n message: string;\n code?: string;\n };\n}\n\n/**\n * Symbol keys for assembler → StreamingMessage communication.\n * Module-private: invisible to external consumers, accessible to\n * {@link StreamingMessageAssembler} within this file.\n */\nconst PUSH_TEXT: unique symbol = Symbol(\"pushText\");\nconst PUSH_REASONING: unique symbol = Symbol(\"pushReasoning\");\nconst PUSH_EVENT: unique symbol = Symbol(\"pushEvent\");\nconst UPDATE_CONTEXT: unique symbol = Symbol(\"updateContext\");\nconst FINISH: unique symbol = Symbol(\"finish\");\nconst ERROR: unique symbol = Symbol(\"error\");\n\n/**\n * Live streaming view of a single message lifecycle, matching the\n * in-process `ChatModelStream` dual-interface pattern.\n *\n * - `text` / `reasoning`: iterate for streaming deltas, or await for\n * the full concatenated string after the message completes.\n * - `usage`: promise that resolves with token usage on message-finish.\n * - `blocks`: the assembled content blocks (updated as deltas arrive).\n *\n * Created by {@link StreamingMessageAssembler} and yielded by\n * the `session.messages` lazy getter.\n */\nexport class StreamingMessage\n implements AsyncIterable<ChatModelStreamEvent>, PromiseLike<AIMessage>\n{\n readonly id: string;\n readonly namespace: string[];\n node: string | undefined;\n readonly metadata: MessageMetadata | undefined;\n readonly assembled: AssembledMessage;\n readonly #events = new MultiCursorBuffer<ChatModelStreamEvent>();\n\n #textChunks: string[] = [];\n #reasoningChunks: string[] = [];\n #textWaiters: Array<() => void> = [];\n #reasoningWaiters: Array<() => void> = [];\n #textDone = false;\n #reasoningDone = false;\n\n #resolveText!: (v: string) => void;\n #resolveReasoning!: (v: string) => void;\n readonly #textPromise: Promise<string>;\n readonly #reasoningPromise: Promise<string>;\n\n constructor(assembled: AssembledMessage) {\n this.id = assembled.id;\n this.assembled = assembled;\n this.namespace = assembled.namespace;\n this.node = assembled.node;\n this.metadata = assembled.metadata;\n this.#textPromise = new Promise<string>((r) => {\n this.#resolveText = r;\n });\n this.#reasoningPromise = new Promise<string>((r) => {\n this.#resolveReasoning = r;\n });\n }\n\n get text(): TextContentStream {\n const chunks = this.#textChunks;\n const waiters = this.#textWaiters;\n const getDone = () => this.#textDone;\n let cursor = 0;\n return {\n [Symbol.asyncIterator]() {\n return {\n async next(): Promise<IteratorResult<string>> {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n };\n },\n then: this.#textPromise.then.bind(this.#textPromise),\n full: {\n async *[Symbol.asyncIterator]() {\n let accumulated = \"\";\n for await (const chunk of {\n [Symbol.asyncIterator]: () =>\n ({\n next: async (): Promise<IteratorResult<string>> => {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n }) satisfies AsyncIterator<string>,\n }) {\n accumulated += chunk;\n yield accumulated;\n }\n },\n },\n };\n }\n\n get reasoning(): TextContentStream {\n const chunks = this.#reasoningChunks;\n const waiters = this.#reasoningWaiters;\n const getDone = () => this.#reasoningDone;\n let cursor = 0;\n return {\n [Symbol.asyncIterator]() {\n return {\n async next(): Promise<IteratorResult<string>> {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n };\n },\n then: this.#reasoningPromise.then.bind(this.#reasoningPromise),\n full: {\n async *[Symbol.asyncIterator]() {\n let accumulated = \"\";\n for await (const chunk of {\n [Symbol.asyncIterator]: () =>\n ({\n next: async (): Promise<IteratorResult<string>> => {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n }) satisfies AsyncIterator<string>,\n }) {\n accumulated += chunk;\n yield accumulated;\n }\n },\n },\n };\n }\n\n get usage(): UsageMetadataStream {\n const promise = (async () => {\n let usage: UsageMetadata | undefined;\n for await (const snapshot of this.#usageIterator()) {\n usage = snapshot;\n }\n return usage;\n })();\n return {\n [Symbol.asyncIterator]: () => this.#usageIterator(),\n then: promise.then.bind(promise),\n };\n }\n\n get toolCalls(): ToolCallsStream {\n const events = this.#events;\n const iterator = async function* () {\n for await (const event of events) {\n if (\n event.event === \"content-block-finish\" &&\n event.content.type === \"tool_call\"\n ) {\n yield event.content as CoreContentBlock.Tools.ToolCall;\n }\n }\n };\n return {\n [Symbol.asyncIterator]: iterator,\n then: async (onfulfilled, onrejected) => {\n try {\n const calls: CoreContentBlock.Tools.ToolCall[] = [];\n for await (const call of iterator()) calls.push(call);\n return onfulfilled ? onfulfilled(calls) : (calls as never);\n } catch (err) {\n if (onrejected) return onrejected(err);\n throw err;\n }\n },\n full: {\n async *[Symbol.asyncIterator]() {\n const calls: CoreContentBlock.Tools.ToolCall[] = [];\n for await (const call of iterator()) {\n calls.push(call);\n yield [...calls];\n }\n },\n },\n };\n }\n\n get output(): PromiseLike<AIMessage> {\n return { then: (onf, onr) => this.#assembleMessage().then(onf, onr) };\n }\n\n get blocks(): ContentBlock[] {\n return this.assembled.blocks;\n }\n\n [Symbol.asyncIterator](): AsyncIterator<ChatModelStreamEvent> {\n return this.#events[Symbol.asyncIterator]();\n }\n\n then<TResult1 = AIMessage, TResult2 = never>(\n onfulfilled?:\n | ((value: AIMessage) => TResult1 | PromiseLike<TResult1>)\n | null,\n onrejected?: ((reason: unknown) => TResult2 | PromiseLike<TResult2>) | null\n ): PromiseLike<TResult1 | TResult2> {\n return this.#assembleMessage().then(onfulfilled, onrejected);\n }\n\n async *#usageIterator(): AsyncGenerator<UsageMetadata> {\n for await (const event of this.#events) {\n if (event.event === \"message-start\" && event.usage) {\n yield normalizeUsage(event.usage)!;\n } else if (event.event === \"message-finish\" && event.usage) {\n yield normalizeUsage(event.usage)!;\n }\n }\n }\n\n async #assembleMessage(): Promise<AIMessage> {\n const contentBlocks: Array<CoreContentBlock | undefined> = [];\n let id: string | undefined;\n let usage: UsageMetadata | undefined;\n let metadata: Record<string, unknown> = {};\n let finishReason: string | undefined;\n\n for await (const event of this.#events) {\n switch (event.event) {\n case \"message-start\":\n id = event.id ?? id;\n if (event.usage) usage = normalizeUsage(event.usage);\n break;\n case \"content-block-start\":\n contentBlocks[event.index] = event.content;\n break;\n case \"content-block-delta\": {\n const current = contentBlocks[event.index];\n contentBlocks[event.index] = applyCoreEventDelta(current, event);\n break;\n }\n case \"content-block-finish\":\n contentBlocks[event.index] = event.content;\n break;\n case \"message-finish\":\n finishReason = event.reason;\n if (event.usage) usage = normalizeUsage(event.usage);\n if (event.responseMetadata) {\n metadata = {\n ...metadata,\n ...event.responseMetadata,\n };\n }\n break;\n default:\n break;\n }\n }\n\n return new AIMessage({\n id,\n content: contentBlocks.filter(\n (block): block is CoreContentBlock => block != null\n ),\n usage_metadata: usage,\n response_metadata: {\n ...metadata,\n ...(finishReason ? { finish_reason: finishReason } : {}),\n output_version: \"v1\" as const,\n },\n });\n }\n\n [PUSH_EVENT](event: ChatModelStreamEvent): void {\n this.#events.push(event);\n }\n\n [UPDATE_CONTEXT](event: MessagesEvent): void {\n this.node = event.params.node ?? this.node;\n }\n\n [PUSH_TEXT](delta: string): void {\n this.#textChunks.push(delta);\n // Wake every caught-up iterator so each one advances its own cursor.\n // Iterators re-check `chunks.length` before delivering, which keeps\n // the cursor the single source of truth for what a consumer has seen.\n const pending = this.#textWaiters.splice(0, this.#textWaiters.length);\n for (const waiter of pending) waiter();\n }\n\n [PUSH_REASONING](delta: string): void {\n this.#reasoningChunks.push(delta);\n const pending = this.#reasoningWaiters.splice(\n 0,\n this.#reasoningWaiters.length\n );\n for (const waiter of pending) waiter();\n }\n\n [FINISH](): void {\n this.#textDone = true;\n this.#reasoningDone = true;\n this.#resolveText(this.#textChunks.join(\"\"));\n this.#resolveReasoning(this.#reasoningChunks.join(\"\"));\n const textPending = this.#textWaiters.splice(0, this.#textWaiters.length);\n for (const waiter of textPending) waiter();\n const reasoningPending = this.#reasoningWaiters.splice(\n 0,\n this.#reasoningWaiters.length\n );\n for (const waiter of reasoningPending) waiter();\n this.#events.close();\n }\n\n [ERROR](): void {\n this[FINISH]();\n }\n}\n\n/**\n * Public view yielded by message projections.\n *\n * `StreamingMessage` is PromiseLike so callers can still `await` a message\n * object directly, but TypeScript applies `Awaited<T>` to values produced by\n * `for await`. Exposing a non-thenable view keeps loop variables typed as the\n * streaming handle instead of as the finalized `AIMessage`.\n */\nexport type StreamingMessageHandle = Omit<StreamingMessage, \"then\">;\n\nexport function toStreamingMessageHandle(\n message: StreamingMessage\n): StreamingMessageHandle {\n return new Proxy(message, {\n get(target, prop) {\n if (prop === \"then\") return undefined;\n const value = Reflect.get(target, prop, target);\n return typeof value === \"function\" ? value.bind(target) : value;\n },\n has(target, prop) {\n if (prop === \"then\") return false;\n return prop in target;\n },\n }) as StreamingMessageHandle;\n}\n\n/**\n * Emitted by `MessageAssembler.consume()` to describe how a message changed in\n * response to a single protocol event.\n */\nexport type MessageAssemblyUpdate =\n | {\n kind: \"message-start\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"content-block-start\" | \"content-block-delta\";\n key: string;\n message: AssembledMessage;\n index: number;\n block: ContentBlock;\n event: MessagesEvent;\n }\n | {\n kind: \"content-block-finish\";\n key: string;\n message: AssembledMessage;\n index: number;\n block: FinalizedContentBlock;\n event: MessagesEvent;\n }\n | {\n kind: \"usage\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"message-finish\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"message-error\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n };\n\nfunction cloneBlock<T extends ContentBlock>(block: T): T {\n return structuredClone(block);\n}\n\nfunction blockFromDelta(\n delta: CoreContentBlockDelta,\n current?: ContentBlock\n): ContentBlock {\n return coreContentBlockFromDelta(\n delta,\n current as unknown as CoreContentBlock | undefined\n ) as unknown as ContentBlock;\n}\n\nfunction applyContentDelta(\n target: ContentBlock,\n delta: ContentBlock\n): ContentBlock {\n if (target.type !== delta.type) {\n return cloneBlock(delta);\n }\n\n switch (delta.type) {\n case \"text\":\n return {\n ...target,\n ...delta,\n text: `${\"text\" in target ? target.text : \"\"}${delta.text}`,\n } as ContentBlock;\n case \"reasoning\":\n return {\n ...target,\n ...delta,\n reasoning: `${\"reasoning\" in target ? target.reasoning : \"\"}${delta.reasoning}`,\n } as ContentBlock;\n case \"tool_call_chunk\":\n case \"server_tool_call_chunk\": {\n // Spread target first, then delta — but preserve target's\n // ``id``/``name`` when the delta explicitly sets them to\n // null/undefined. Some providers (notably Anthropic via the\n // langchain-core compat bridge) only attach the tool-call\n // identifiers to the first ``content-block-start`` chunk; every\n // subsequent ``input_json_delta`` chunk carries ``id=null,\n // name=null``. A naive ``{...target, ...delta}`` spread\n // overwrites the captured identifiers with null, which in turn\n // makes downstream consumers (e.g. ``extractToolCallChunks`` in\n // ``assembled-to-message.ts``) drop the chunk on the floor until\n // the final ``content-block-finish`` event promotes it to a\n // finalized ``tool_call`` — causing tool-call cards to appear\n // all-at-once at the end of the turn instead of incrementally.\n const merged = { ...target, ...delta } as Record<string, unknown>;\n if (delta.id == null && \"id\" in target && target.id != null) {\n merged.id = target.id;\n }\n if (delta.name == null && \"name\" in target && target.name != null) {\n merged.name = target.name;\n }\n merged.args = `${(\"args\" in target ? target.args : \"\") ?? \"\"}${delta.args ?? \"\"}`;\n return merged as unknown as ContentBlock;\n }\n default:\n return {\n ...target,\n ...delta,\n } as ContentBlock;\n }\n}\n\nfunction messageKeyFor(event: MessagesEvent): string {\n const { namespace, node, data } = event.params;\n const namespaceKey = namespace.join(\"/\");\n const messageId = data.event === \"message-start\" ? (data.id ?? \"\") : \"\";\n return `${namespaceKey}::${node ?? \"\"}::${messageId}`;\n}\n\nfunction toChatModelStreamEvent(event: MessagesEvent): ChatModelStreamEvent {\n return event.params.data as unknown as ChatModelStreamEvent;\n}\n\n/**\n * Incrementally assembles `messages` events into complete message objects.\n */\nexport class MessageAssembler {\n private readonly activeMessages = new Map<string, AssembledMessage>();\n private readonly activeByNamespaceNode = new Map<string, string>();\n private readonly blockIndexByProtocolIndexAndType = new Map<string, number>();\n\n /**\n * Applies a single message event and returns the resulting assembly update.\n *\n * @param event - Incoming `messages` event to fold into the assembler state.\n */\n consume(event: MessagesEvent): MessageAssemblyUpdate {\n const data = event.params.data;\n const namespaceNodeKey = `${event.params.namespace.join(\"/\")}::${event.params.node ?? \"\"}`;\n\n if (data.event === \"message-start\") {\n const key = messageKeyFor(event);\n this.activeByNamespaceNode.set(namespaceNodeKey, key);\n const message: AssembledMessage = {\n id: data.id,\n namespace: [...event.params.namespace],\n node: event.params.node,\n metadata: data.metadata,\n blocks: [],\n };\n this.activeMessages.set(key, message);\n return { kind: \"message-start\", key, message, event };\n }\n\n const activeKey = this.activeByNamespaceNode.get(namespaceNodeKey);\n if (!activeKey) {\n // A continuation event (delta/finish/error) arrived without a\n // prior `message-start`. This can happen on late-attaching\n // subscriptions when the server has already trimmed the\n // `message-start` from its replay buffer. Synthesize a minimal\n // active message so the assembler can still fold subsequent\n // events instead of hard-failing the caller.\n const syntheticKey = `${namespaceNodeKey}::`;\n this.activeByNamespaceNode.set(namespaceNodeKey, syntheticKey);\n const synthetic: AssembledMessage = {\n id: data.id,\n namespace: [...event.params.namespace],\n node: event.params.node,\n blocks: [],\n };\n this.activeMessages.set(syntheticKey, synthetic);\n return this.consume(event);\n }\n\n const message = this.activeMessages.get(activeKey);\n if (!message) {\n throw new Error(`No active message state found for key ${activeKey}`);\n }\n\n if ((data as { event?: string }).event === \"usage\") {\n message.usage = (data as { usage?: UsageInfo }).usage;\n return {\n kind: \"usage\",\n key: activeKey,\n message,\n event,\n };\n }\n\n switch (data.event) {\n case \"content-block-start\": {\n message.blocks[data.index] = cloneBlock(data.content);\n this.blockIndexByProtocolIndexAndType.set(\n blockIndexKey(activeKey, data.index, data.content.type),\n data.index\n );\n return {\n kind: \"content-block-start\",\n key: activeKey,\n message,\n index: data.index,\n block: data.content,\n event,\n };\n }\n case \"content-block-delta\": {\n const deltaEvent = data as typeof data & {\n content?: ContentBlock;\n delta?: CoreContentBlockDelta;\n };\n const deltaBlock =\n deltaEvent.content ??\n (deltaEvent.delta != null\n ? blockFromDelta(deltaEvent.delta, message.blocks[data.index])\n : undefined);\n if (deltaBlock == null) {\n throw new Error(\"Received content-block-delta without content\");\n }\n const targetIndex = this.resolveBlockIndex(\n activeKey,\n message.blocks,\n data.index,\n deltaBlock.type\n );\n const current = message.blocks[targetIndex];\n message.blocks[targetIndex] =\n deltaEvent.content != null\n ? current == null\n ? cloneBlock(deltaEvent.content)\n : applyContentDelta(current, deltaEvent.content)\n : (applyCoreEventDelta(\n current as CoreContentBlock | undefined,\n data as unknown as Extract<\n ChatModelStreamEvent,\n { event: \"content-block-delta\" }\n >\n ) as unknown as ContentBlock);\n return {\n kind: \"content-block-delta\",\n key: activeKey,\n message,\n index: targetIndex,\n block: deltaBlock,\n event,\n };\n }\n case \"content-block-finish\": {\n const targetIndex = this.resolveFinishBlockIndex(\n activeKey,\n data.index,\n data.content.type\n );\n message.blocks[targetIndex] = cloneBlock(data.content);\n return {\n kind: \"content-block-finish\",\n key: activeKey,\n message,\n index: targetIndex,\n block: data.content,\n event,\n };\n }\n case \"message-finish\": {\n message.usage = data.usage;\n message.finishMetadata = data.responseMetadata;\n this.activeMessages.delete(activeKey);\n this.activeByNamespaceNode.delete(namespaceNodeKey);\n this.clearBlockIndexAliases(activeKey);\n return {\n kind: \"message-finish\",\n key: activeKey,\n message: structuredClone(message),\n event,\n };\n }\n case \"error\": {\n message.error = { message: data.message, code: data.code };\n this.activeMessages.delete(activeKey);\n this.activeByNamespaceNode.delete(namespaceNodeKey);\n this.clearBlockIndexAliases(activeKey);\n return {\n kind: \"message-error\",\n key: activeKey,\n message: structuredClone(message),\n event,\n };\n }\n }\n }\n\n private resolveBlockIndex(\n activeKey: string,\n blocks: ContentBlock[],\n protocolIndex: number,\n blockType: string\n ): number {\n const current = blocks[protocolIndex];\n if (\n current == null ||\n current.type === blockType ||\n areCompatibleBlockTypes(current.type, blockType)\n ) {\n this.blockIndexByProtocolIndexAndType.set(\n blockIndexKey(activeKey, protocolIndex, blockType),\n protocolIndex\n );\n return protocolIndex;\n }\n\n const key = blockIndexKey(activeKey, protocolIndex, blockType);\n const existing = this.blockIndexByProtocolIndexAndType.get(key);\n if (existing != null) return existing;\n\n const nextIndex = blocks.length;\n this.blockIndexByProtocolIndexAndType.set(key, nextIndex);\n return nextIndex;\n }\n\n private resolveFinishBlockIndex(\n activeKey: string,\n protocolIndex: number,\n blockType: string\n ): number {\n const key = blockIndexKey(activeKey, protocolIndex, blockType);\n const existing = this.blockIndexByProtocolIndexAndType.get(key);\n if (existing != null) return existing;\n\n this.blockIndexByProtocolIndexAndType.set(key, protocolIndex);\n return protocolIndex;\n }\n\n private clearBlockIndexAliases(activeKey: string): void {\n const prefix = `${activeKey}::`;\n for (const key of this.blockIndexByProtocolIndexAndType.keys()) {\n if (key.startsWith(prefix))\n this.blockIndexByProtocolIndexAndType.delete(key);\n }\n }\n}\n\nfunction blockIndexKey(\n activeKey: string,\n protocolIndex: number,\n blockType: string\n): string {\n return `${activeKey}::${protocolIndex}::${blockType}`;\n}\n\nfunction areCompatibleBlockTypes(\n currentType: string,\n nextType: string\n): boolean {\n const toolCallTypes = new Set([\n \"tool_call\",\n \"tool_call_chunk\",\n \"tool_use\",\n \"input_json_delta\",\n ]);\n const serverToolCallTypes = new Set([\n \"server_tool_call\",\n \"server_tool_call_chunk\",\n ]);\n\n return (\n (toolCallTypes.has(currentType) && toolCallTypes.has(nextType)) ||\n (serverToolCallTypes.has(currentType) && serverToolCallTypes.has(nextType))\n );\n}\n\n/**\n * Assembles `messages` events into {@link StreamingMessage} instances\n * with live text/reasoning delta streams, matching the in-process\n * `ChatModelStream` dual-interface pattern.\n */\nexport class StreamingMessageAssembler {\n readonly #assembler = new MessageAssembler();\n readonly #activeStreaming = new Map<string, StreamingMessage>();\n\n /**\n * Folds a single event and returns a new {@link StreamingMessage}\n * when a `message-start` is seen, or `undefined` for continuation\n * events (deltas, finish, error).\n */\n consume(event: MessagesEvent): StreamingMessage | undefined {\n const update = this.#assembler.consume(event);\n if (update == null) return undefined;\n\n switch (update.kind) {\n case \"message-start\": {\n const streaming = new StreamingMessage(update.message);\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n this.#activeStreaming.set(update.key, streaming);\n return streaming;\n }\n case \"content-block-start\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n if (\n streaming &&\n update.block.type === \"text\" &&\n \"text\" in update.block &&\n update.block.text\n ) {\n streaming[PUSH_TEXT](update.block.text);\n }\n if (\n streaming &&\n update.block.type === \"reasoning\" &&\n \"reasoning\" in update.block &&\n update.block.reasoning\n ) {\n streaming[PUSH_REASONING](update.block.reasoning);\n }\n return undefined;\n }\n case \"content-block-delta\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (!streaming) return undefined;\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n if (update.block.type === \"text\" && \"text\" in update.block) {\n streaming[PUSH_TEXT](update.block.text);\n }\n if (update.block.type === \"reasoning\" && \"reasoning\" in update.block) {\n streaming[PUSH_REASONING](update.block.reasoning);\n }\n return undefined;\n }\n case \"content-block-finish\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n return undefined;\n }\n case \"usage\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n return undefined;\n }\n case \"message-finish\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n streaming[FINISH]();\n this.#activeStreaming.delete(update.key);\n }\n return undefined;\n }\n case \"message-error\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n streaming[ERROR]();\n this.#activeStreaming.delete(update.key);\n }\n return undefined;\n }\n }\n }\n}\n"],"mappings":";;;AA8BA,SAAS,sBACP,QACA,OACkB;AAClB,KAAI,OAAO,SAAS,MAAM,KACxB,QAAO,gBAAgB,MAAM;AAG/B,SAAQ,MAAM,MAAd;EACE,KAAK,OACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,MAAM,GAAG,UAAU,SAAS,OAAO,OAAO,KAAK,MAAM;GACtD;EACH,KAAK,YACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,WAAW,GAAG,eAAe,SAAS,OAAO,YAAY,KAAK,MAAM;GACrE;EACH,KAAK;EACL,KAAK,0BAA0B;GAC7B,MAAM,SAAS;IAAE,GAAG;IAAQ,GAAG;IAAO;AACtC,OAAI,MAAM,MAAM,QAAQ,QAAQ,UAAU,OAAO,MAAM,KACrD,QAAO,KAAK,OAAO;AAErB,OAAI,MAAM,QAAQ,QAAQ,UAAU,UAAU,OAAO,QAAQ,KAC3D,QAAO,OAAO,OAAO;AAEvB,UAAO,OAAO,IAAI,UAAU,SAAS,OAAO,OAAO,OAAO,KAAK,MAAM,QAAQ;AAC7E,UAAO;;EAET,QACE,QAAO;GAAE,GAAG;GAAQ,GAAG;GAAO;;;AAIpC,SAAS,0BACP,OACA,SACkB;AAClB,SAAQ,MAAM,MAAd;EACE,KAAK,aACH,QAAO;GAAE,MAAM;GAAQ,MAAM,MAAM;GAAM;EAC3C,KAAK,kBACH,QAAO;GACL,MAAM;GACN,WAAW,MAAM;GAClB;EACH,KAAK,cAAc;GACjB,MAAM,SAAS;IAAE,GAAI,WAAW,EAAE;IAAG,MAAM,MAAM;IAAM;AAIvD,OAAI,MAAM,SAAU,QAAO,WAAW,MAAM;AAC5C,UAAO;;EAET,KAAK,cACH,QAAO,MAAM;;;AAInB,SAAS,oBACP,SACA,OAGkB;AAClB,KAAI,MAAM,QACR,QAAO,UACH,sBAAsB,SAAS,MAAM,QAAQ,GAC7C,MAAM;AAGZ,SAAQ,MAAM,MAAM,MAApB;EACE,KAAK;AACH,OAAI,SAAS,SAAS,OACpB,QAAO;IACL,GAAG;IACH,MAAM,GAAG,UAAU,UAAU,QAAQ,OAAO,KAAK,MAAM,MAAM;IAC9D;AAEH,UAAO,0BAA0B,MAAM,OAAO,QAAQ;EACxD,KAAK;AACH,OAAI,SAAS,SAAS,YACpB,QAAO;IACL,GAAG;IACH,WAAW,GAAG,eAAe,UAAU,QAAQ,YAAY,KAAK,MAAM,MAAM;IAC7E;AAEH,UAAO,0BAA0B,MAAM,OAAO,QAAQ;EACxD,KAAK,cAAc;GACjB,MAAM,SAAS,EAAE,GAAI,WAAW,EAAE,EAAG;AACrC,UAAO,OAAO,GAAI,OAAO,QAA+B,KAAK,MAAM,MAAM;AACzE,OAAI,MAAM,MAAM,SAAU,QAAO,WAAW,MAAM,MAAM;AACxD,UAAO;;EAET,KAAK,cACH,QAAO;GACL,GAAI,WAAW,EAAE;GACjB,GAAG,MAAM,MAAM;GAChB;;;AAIP,SAAS,eACP,OAC2B;AAC3B,KAAI,CAAC,MAAO,QAAO,KAAA;AACnB,QAAO;EACL,GAAG;EACH,cAAc,MAAM,gBAAgB;EACpC,eAAe,MAAM,iBAAiB;EACtC,cAAc,MAAM,gBAAgB;EACrC;;;;;;;AA0BH,MAAM,YAA2B,OAAO,WAAW;AACnD,MAAM,iBAAgC,OAAO,gBAAgB;AAC7D,MAAM,aAA4B,OAAO,YAAY;AACrD,MAAM,iBAAgC,OAAO,gBAAgB;AAC7D,MAAM,SAAwB,OAAO,SAAS;AAC9C,MAAM,QAAuB,OAAO,QAAQ;;;;;;;;;;;;;AAc5C,IAAa,mBAAb,MAEA;CACE;CACA;CACA;CACA;CACA;CACA,UAAmB,IAAI,mBAAyC;CAEhE,cAAwB,EAAE;CAC1B,mBAA6B,EAAE;CAC/B,eAAkC,EAAE;CACpC,oBAAuC,EAAE;CACzC,YAAY;CACZ,iBAAiB;CAEjB;CACA;CACA;CACA;CAEA,YAAY,WAA6B;AACvC,OAAK,KAAK,UAAU;AACpB,OAAK,YAAY;AACjB,OAAK,YAAY,UAAU;AAC3B,OAAK,OAAO,UAAU;AACtB,OAAK,WAAW,UAAU;AAC1B,QAAA,cAAoB,IAAI,SAAiB,MAAM;AAC7C,SAAA,cAAoB;IACpB;AACF,QAAA,mBAAyB,IAAI,SAAiB,MAAM;AAClD,SAAA,mBAAyB;IACzB;;CAGJ,IAAI,OAA0B;EAC5B,MAAM,SAAS,MAAA;EACf,MAAM,UAAU,MAAA;EAChB,MAAM,gBAAgB,MAAA;EACtB,IAAI,SAAS;AACb,SAAO;GACL,CAAC,OAAO,iBAAiB;AACvB,WAAO,EACL,MAAM,OAAwC;AAC5C,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP;;GAEH,MAAM,MAAA,YAAkB,KAAK,KAAK,MAAA,YAAkB;GACpD,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,IAAI,cAAc;AAClB,eAAW,MAAM,SAAS,GACvB,OAAO,uBACL,EACC,MAAM,YAA6C;AACjD,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP,GACJ,EAAE;AACD,oBAAe;AACf,WAAM;;MAGX;GACF;;CAGH,IAAI,YAA+B;EACjC,MAAM,SAAS,MAAA;EACf,MAAM,UAAU,MAAA;EAChB,MAAM,gBAAgB,MAAA;EACtB,IAAI,SAAS;AACb,SAAO;GACL,CAAC,OAAO,iBAAiB;AACvB,WAAO,EACL,MAAM,OAAwC;AAC5C,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP;;GAEH,MAAM,MAAA,iBAAuB,KAAK,KAAK,MAAA,iBAAuB;GAC9D,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,IAAI,cAAc;AAClB,eAAW,MAAM,SAAS,GACvB,OAAO,uBACL,EACC,MAAM,YAA6C;AACjD,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP,GACJ,EAAE;AACD,oBAAe;AACf,WAAM;;MAGX;GACF;;CAGH,IAAI,QAA6B;EAC/B,MAAM,WAAW,YAAY;GAC3B,IAAI;AACJ,cAAW,MAAM,YAAY,MAAA,eAAqB,CAChD,SAAQ;AAEV,UAAO;MACL;AACJ,SAAO;IACJ,OAAO,sBAAsB,MAAA,eAAqB;GACnD,MAAM,QAAQ,KAAK,KAAK,QAAQ;GACjC;;CAGH,IAAI,YAA6B;EAC/B,MAAM,SAAS,MAAA;EACf,MAAM,WAAW,mBAAmB;AAClC,cAAW,MAAM,SAAS,OACxB,KACE,MAAM,UAAU,0BAChB,MAAM,QAAQ,SAAS,YAEvB,OAAM,MAAM;;AAIlB,SAAO;IACJ,OAAO,gBAAgB;GACxB,MAAM,OAAO,aAAa,eAAe;AACvC,QAAI;KACF,MAAM,QAA2C,EAAE;AACnD,gBAAW,MAAM,QAAQ,UAAU,CAAE,OAAM,KAAK,KAAK;AACrD,YAAO,cAAc,YAAY,MAAM,GAAI;aACpC,KAAK;AACZ,SAAI,WAAY,QAAO,WAAW,IAAI;AACtC,WAAM;;;GAGV,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,MAAM,QAA2C,EAAE;AACnD,eAAW,MAAM,QAAQ,UAAU,EAAE;AACnC,WAAM,KAAK,KAAK;AAChB,WAAM,CAAC,GAAG,MAAM;;MAGrB;GACF;;CAGH,IAAI,SAAiC;AACnC,SAAO,EAAE,OAAO,KAAK,QAAQ,MAAA,iBAAuB,CAAC,KAAK,KAAK,IAAI,EAAE;;CAGvE,IAAI,SAAyB;AAC3B,SAAO,KAAK,UAAU;;CAGxB,CAAC,OAAO,iBAAsD;AAC5D,SAAO,MAAA,OAAa,OAAO,gBAAgB;;CAG7C,KACE,aAGA,YACkC;AAClC,SAAO,MAAA,iBAAuB,CAAC,KAAK,aAAa,WAAW;;CAG9D,QAAA,gBAAuD;AACrD,aAAW,MAAM,SAAS,MAAA,OACxB,KAAI,MAAM,UAAU,mBAAmB,MAAM,MAC3C,OAAM,eAAe,MAAM,MAAM;WACxB,MAAM,UAAU,oBAAoB,MAAM,MACnD,OAAM,eAAe,MAAM,MAAM;;CAKvC,OAAA,kBAA6C;EAC3C,MAAM,gBAAqD,EAAE;EAC7D,IAAI;EACJ,IAAI;EACJ,IAAI,WAAoC,EAAE;EAC1C,IAAI;AAEJ,aAAW,MAAM,SAAS,MAAA,OACxB,SAAQ,MAAM,OAAd;GACE,KAAK;AACH,SAAK,MAAM,MAAM;AACjB,QAAI,MAAM,MAAO,SAAQ,eAAe,MAAM,MAAM;AACpD;GACF,KAAK;AACH,kBAAc,MAAM,SAAS,MAAM;AACnC;GACF,KAAK,uBAAuB;IAC1B,MAAM,UAAU,cAAc,MAAM;AACpC,kBAAc,MAAM,SAAS,oBAAoB,SAAS,MAAM;AAChE;;GAEF,KAAK;AACH,kBAAc,MAAM,SAAS,MAAM;AACnC;GACF,KAAK;AACH,mBAAe,MAAM;AACrB,QAAI,MAAM,MAAO,SAAQ,eAAe,MAAM,MAAM;AACpD,QAAI,MAAM,iBACR,YAAW;KACT,GAAG;KACH,GAAG,MAAM;KACV;AAEH;GACF,QACE;;AAIN,SAAO,IAAI,UAAU;GACnB;GACA,SAAS,cAAc,QACpB,UAAqC,SAAS,KAChD;GACD,gBAAgB;GAChB,mBAAmB;IACjB,GAAG;IACH,GAAI,eAAe,EAAE,eAAe,cAAc,GAAG,EAAE;IACvD,gBAAgB;IACjB;GACF,CAAC;;CAGJ,CAAC,YAAY,OAAmC;AAC9C,QAAA,OAAa,KAAK,MAAM;;CAG1B,CAAC,gBAAgB,OAA4B;AAC3C,OAAK,OAAO,MAAM,OAAO,QAAQ,KAAK;;CAGxC,CAAC,WAAW,OAAqB;AAC/B,QAAA,WAAiB,KAAK,MAAM;EAI5B,MAAM,UAAU,MAAA,YAAkB,OAAO,GAAG,MAAA,YAAkB,OAAO;AACrE,OAAK,MAAM,UAAU,QAAS,SAAQ;;CAGxC,CAAC,gBAAgB,OAAqB;AACpC,QAAA,gBAAsB,KAAK,MAAM;EACjC,MAAM,UAAU,MAAA,iBAAuB,OACrC,GACA,MAAA,iBAAuB,OACxB;AACD,OAAK,MAAM,UAAU,QAAS,SAAQ;;CAGxC,CAAC,UAAgB;AACf,QAAA,WAAiB;AACjB,QAAA,gBAAsB;AACtB,QAAA,YAAkB,MAAA,WAAiB,KAAK,GAAG,CAAC;AAC5C,QAAA,iBAAuB,MAAA,gBAAsB,KAAK,GAAG,CAAC;EACtD,MAAM,cAAc,MAAA,YAAkB,OAAO,GAAG,MAAA,YAAkB,OAAO;AACzE,OAAK,MAAM,UAAU,YAAa,SAAQ;EAC1C,MAAM,mBAAmB,MAAA,iBAAuB,OAC9C,GACA,MAAA,iBAAuB,OACxB;AACD,OAAK,MAAM,UAAU,iBAAkB,SAAQ;AAC/C,QAAA,OAAa,OAAO;;CAGtB,CAAC,SAAe;AACd,OAAK,SAAS;;;AAclB,SAAgB,yBACd,SACwB;AACxB,QAAO,IAAI,MAAM,SAAS;EACxB,IAAI,QAAQ,MAAM;AAChB,OAAI,SAAS,OAAQ,QAAO,KAAA;GAC5B,MAAM,QAAQ,QAAQ,IAAI,QAAQ,MAAM,OAAO;AAC/C,UAAO,OAAO,UAAU,aAAa,MAAM,KAAK,OAAO,GAAG;;EAE5D,IAAI,QAAQ,MAAM;AAChB,OAAI,SAAS,OAAQ,QAAO;AAC5B,UAAO,QAAQ;;EAElB,CAAC;;AAiDJ,SAAS,WAAmC,OAAa;AACvD,QAAO,gBAAgB,MAAM;;AAG/B,SAAS,eACP,OACA,SACc;AACd,QAAO,0BACL,OACA,QACD;;AAGH,SAAS,kBACP,QACA,OACc;AACd,KAAI,OAAO,SAAS,MAAM,KACxB,QAAO,WAAW,MAAM;AAG1B,SAAQ,MAAM,MAAd;EACE,KAAK,OACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,MAAM,GAAG,UAAU,SAAS,OAAO,OAAO,KAAK,MAAM;GACtD;EACH,KAAK,YACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,WAAW,GAAG,eAAe,SAAS,OAAO,YAAY,KAAK,MAAM;GACrE;EACH,KAAK;EACL,KAAK,0BAA0B;GAc7B,MAAM,SAAS;IAAE,GAAG;IAAQ,GAAG;IAAO;AACtC,OAAI,MAAM,MAAM,QAAQ,QAAQ,UAAU,OAAO,MAAM,KACrD,QAAO,KAAK,OAAO;AAErB,OAAI,MAAM,QAAQ,QAAQ,UAAU,UAAU,OAAO,QAAQ,KAC3D,QAAO,OAAO,OAAO;AAEvB,UAAO,OAAO,IAAI,UAAU,SAAS,OAAO,OAAO,OAAO,KAAK,MAAM,QAAQ;AAC7E,UAAO;;EAET,QACE,QAAO;GACL,GAAG;GACH,GAAG;GACJ;;;AAIP,SAAS,cAAc,OAA8B;CACnD,MAAM,EAAE,WAAW,MAAM,SAAS,MAAM;CACxC,MAAM,eAAe,UAAU,KAAK,IAAI;CACxC,MAAM,YAAY,KAAK,UAAU,kBAAmB,KAAK,MAAM,KAAM;AACrE,QAAO,GAAG,aAAa,IAAI,QAAQ,GAAG,IAAI;;AAG5C,SAAS,uBAAuB,OAA4C;AAC1E,QAAO,MAAM,OAAO;;;;;AAMtB,IAAa,mBAAb,MAA8B;CAC5B,iCAAkC,IAAI,KAA+B;CACrE,wCAAyC,IAAI,KAAqB;CAClE,mDAAoD,IAAI,KAAqB;;;;;;CAO7E,QAAQ,OAA6C;EACnD,MAAM,OAAO,MAAM,OAAO;EAC1B,MAAM,mBAAmB,GAAG,MAAM,OAAO,UAAU,KAAK,IAAI,CAAC,IAAI,MAAM,OAAO,QAAQ;AAEtF,MAAI,KAAK,UAAU,iBAAiB;GAClC,MAAM,MAAM,cAAc,MAAM;AAChC,QAAK,sBAAsB,IAAI,kBAAkB,IAAI;GACrD,MAAM,UAA4B;IAChC,IAAI,KAAK;IACT,WAAW,CAAC,GAAG,MAAM,OAAO,UAAU;IACtC,MAAM,MAAM,OAAO;IACnB,UAAU,KAAK;IACf,QAAQ,EAAE;IACX;AACD,QAAK,eAAe,IAAI,KAAK,QAAQ;AACrC,UAAO;IAAE,MAAM;IAAiB;IAAK;IAAS;IAAO;;EAGvD,MAAM,YAAY,KAAK,sBAAsB,IAAI,iBAAiB;AAClE,MAAI,CAAC,WAAW;GAOd,MAAM,eAAe,GAAG,iBAAiB;AACzC,QAAK,sBAAsB,IAAI,kBAAkB,aAAa;GAC9D,MAAM,YAA8B;IAClC,IAAI,KAAK;IACT,WAAW,CAAC,GAAG,MAAM,OAAO,UAAU;IACtC,MAAM,MAAM,OAAO;IACnB,QAAQ,EAAE;IACX;AACD,QAAK,eAAe,IAAI,cAAc,UAAU;AAChD,UAAO,KAAK,QAAQ,MAAM;;EAG5B,MAAM,UAAU,KAAK,eAAe,IAAI,UAAU;AAClD,MAAI,CAAC,QACH,OAAM,IAAI,MAAM,yCAAyC,YAAY;AAGvE,MAAK,KAA4B,UAAU,SAAS;AAClD,WAAQ,QAAS,KAA+B;AAChD,UAAO;IACL,MAAM;IACN,KAAK;IACL;IACA;IACD;;AAGH,UAAQ,KAAK,OAAb;GACE,KAAK;AACH,YAAQ,OAAO,KAAK,SAAS,WAAW,KAAK,QAAQ;AACrD,SAAK,iCAAiC,IACpC,cAAc,WAAW,KAAK,OAAO,KAAK,QAAQ,KAAK,EACvD,KAAK,MACN;AACD,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO,KAAK;KACZ,OAAO,KAAK;KACZ;KACD;GAEH,KAAK,uBAAuB;IAC1B,MAAM,aAAa;IAInB,MAAM,aACJ,WAAW,YACV,WAAW,SAAS,OACjB,eAAe,WAAW,OAAO,QAAQ,OAAO,KAAK,OAAO,GAC5D,KAAA;AACN,QAAI,cAAc,KAChB,OAAM,IAAI,MAAM,+CAA+C;IAEjE,MAAM,cAAc,KAAK,kBACvB,WACA,QAAQ,QACR,KAAK,OACL,WAAW,KACZ;IACD,MAAM,UAAU,QAAQ,OAAO;AAC/B,YAAQ,OAAO,eACb,WAAW,WAAW,OAClB,WAAW,OACT,WAAW,WAAW,QAAQ,GAC9B,kBAAkB,SAAS,WAAW,QAAQ,GAC/C,oBACC,SACA,KAID;AACP,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO;KACP,OAAO;KACP;KACD;;GAEH,KAAK,wBAAwB;IAC3B,MAAM,cAAc,KAAK,wBACvB,WACA,KAAK,OACL,KAAK,QAAQ,KACd;AACD,YAAQ,OAAO,eAAe,WAAW,KAAK,QAAQ;AACtD,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO;KACP,OAAO,KAAK;KACZ;KACD;;GAEH,KAAK;AACH,YAAQ,QAAQ,KAAK;AACrB,YAAQ,iBAAiB,KAAK;AAC9B,SAAK,eAAe,OAAO,UAAU;AACrC,SAAK,sBAAsB,OAAO,iBAAiB;AACnD,SAAK,uBAAuB,UAAU;AACtC,WAAO;KACL,MAAM;KACN,KAAK;KACL,SAAS,gBAAgB,QAAQ;KACjC;KACD;GAEH,KAAK;AACH,YAAQ,QAAQ;KAAE,SAAS,KAAK;KAAS,MAAM,KAAK;KAAM;AAC1D,SAAK,eAAe,OAAO,UAAU;AACrC,SAAK,sBAAsB,OAAO,iBAAiB;AACnD,SAAK,uBAAuB,UAAU;AACtC,WAAO;KACL,MAAM;KACN,KAAK;KACL,SAAS,gBAAgB,QAAQ;KACjC;KACD;;;CAKP,kBACE,WACA,QACA,eACA,WACQ;EACR,MAAM,UAAU,OAAO;AACvB,MACE,WAAW,QACX,QAAQ,SAAS,aACjB,wBAAwB,QAAQ,MAAM,UAAU,EAChD;AACA,QAAK,iCAAiC,IACpC,cAAc,WAAW,eAAe,UAAU,EAClD,cACD;AACD,UAAO;;EAGT,MAAM,MAAM,cAAc,WAAW,eAAe,UAAU;EAC9D,MAAM,WAAW,KAAK,iCAAiC,IAAI,IAAI;AAC/D,MAAI,YAAY,KAAM,QAAO;EAE7B,MAAM,YAAY,OAAO;AACzB,OAAK,iCAAiC,IAAI,KAAK,UAAU;AACzD,SAAO;;CAGT,wBACE,WACA,eACA,WACQ;EACR,MAAM,MAAM,cAAc,WAAW,eAAe,UAAU;EAC9D,MAAM,WAAW,KAAK,iCAAiC,IAAI,IAAI;AAC/D,MAAI,YAAY,KAAM,QAAO;AAE7B,OAAK,iCAAiC,IAAI,KAAK,cAAc;AAC7D,SAAO;;CAGT,uBAA+B,WAAyB;EACtD,MAAM,SAAS,GAAG,UAAU;AAC5B,OAAK,MAAM,OAAO,KAAK,iCAAiC,MAAM,CAC5D,KAAI,IAAI,WAAW,OAAO,CACxB,MAAK,iCAAiC,OAAO,IAAI;;;AAKzD,SAAS,cACP,WACA,eACA,WACQ;AACR,QAAO,GAAG,UAAU,IAAI,cAAc,IAAI;;AAG5C,SAAS,wBACP,aACA,UACS;CACT,MAAM,gBAAgB,IAAI,IAAI;EAC5B;EACA;EACA;EACA;EACD,CAAC;CACF,MAAM,sBAAsB,IAAI,IAAI,CAClC,oBACA,yBACD,CAAC;AAEF,QACG,cAAc,IAAI,YAAY,IAAI,cAAc,IAAI,SAAS,IAC7D,oBAAoB,IAAI,YAAY,IAAI,oBAAoB,IAAI,SAAS;;;;;;;AAS9E,IAAa,4BAAb,MAAuC;CACrC,aAAsB,IAAI,kBAAkB;CAC5C,mCAA4B,IAAI,KAA+B;;;;;;CAO/D,QAAQ,OAAoD;EAC1D,MAAM,SAAS,MAAA,UAAgB,QAAQ,MAAM;AAC7C,MAAI,UAAU,KAAM,QAAO,KAAA;AAE3B,UAAQ,OAAO,MAAf;GACE,KAAK,iBAAiB;IACpB,MAAM,YAAY,IAAI,iBAAiB,OAAO,QAAQ;AACtD,cAAU,gBAAgB,OAAO,MAAM;AACvC,cAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,UAAA,gBAAsB,IAAI,OAAO,KAAK,UAAU;AAChD,WAAO;;GAET,KAAK,uBAAuB;IAC1B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D,QACE,aACA,OAAO,MAAM,SAAS,UACtB,UAAU,OAAO,SACjB,OAAO,MAAM,KAEb,WAAU,WAAW,OAAO,MAAM,KAAK;AAEzC,QACE,aACA,OAAO,MAAM,SAAS,eACtB,eAAe,OAAO,SACtB,OAAO,MAAM,UAEb,WAAU,gBAAgB,OAAO,MAAM,UAAU;AAEnD;;GAEF,KAAK,uBAAuB;IAC1B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,CAAC,UAAW,QAAO,KAAA;AACvB,cAAU,gBAAgB,OAAO,MAAM;AACvC,cAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,QAAI,OAAO,MAAM,SAAS,UAAU,UAAU,OAAO,MACnD,WAAU,WAAW,OAAO,MAAM,KAAK;AAEzC,QAAI,OAAO,MAAM,SAAS,eAAe,eAAe,OAAO,MAC7D,WAAU,gBAAgB,OAAO,MAAM,UAAU;AAEnD;;GAEF,KAAK,wBAAwB;IAC3B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D;;GAEF,KAAK,SAAS;IACZ,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D;;GAEF,KAAK,kBAAkB;IACrB,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,eAAU,SAAS;AACnB,WAAA,gBAAsB,OAAO,OAAO,IAAI;;AAE1C;;GAEF,KAAK,iBAAiB;IACpB,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,eAAU,QAAQ;AAClB,WAAA,gBAAsB,OAAO,OAAO,IAAI;;AAE1C"}
|
|
1
|
+
{"version":3,"file":"messages.js","names":["#events","#textPromise","#reasoningPromise","#resolveText","#resolveReasoning","#textChunks","#textWaiters","#textDone","#reasoningChunks","#reasoningWaiters","#reasoningDone","#usageIterator","#assembleMessage","#assembler","#activeStreaming"],"sources":["../../../src/client/stream/messages.ts"],"sourcesContent":["import {\n AIMessage,\n type ContentBlock as CoreContentBlock,\n type UsageMetadata,\n} from \"@langchain/core/messages\";\nimport type {\n ChatModelStreamEvent,\n ContentBlockDelta as CoreContentBlockDelta,\n} from \"@langchain/core/language_models/event\";\nimport type {\n ContentBlock,\n FinalizedContentBlock,\n MessageMetadata,\n MessagesEvent,\n UsageInfo,\n} from \"@langchain/protocol\";\n\nimport { MultiCursorBuffer } from \"./multi-cursor-buffer.js\";\n\ntype TextContentStream = AsyncIterable<string> &\n PromiseLike<string> & { full: AsyncIterable<string> };\n\ntype UsageMetadataStream = AsyncIterable<UsageMetadata> &\n PromiseLike<UsageMetadata | undefined>;\n\ntype ToolCallsStream = AsyncIterable<CoreContentBlock.Tools.ToolCall> &\n PromiseLike<Array<CoreContentBlock.Tools.ToolCall>> & {\n full: AsyncIterable<Array<CoreContentBlock.Tools.ToolCall>>;\n };\n\nfunction applyCoreContentDelta(\n target: CoreContentBlock,\n delta: CoreContentBlock\n): CoreContentBlock {\n if (target.type !== delta.type) {\n return structuredClone(delta);\n }\n\n switch (delta.type) {\n case \"text\":\n return {\n ...target,\n ...delta,\n text: `${\"text\" in target ? target.text : \"\"}${delta.text}`,\n } as CoreContentBlock;\n case \"reasoning\":\n return {\n ...target,\n ...delta,\n reasoning: `${\"reasoning\" in target ? target.reasoning : \"\"}${delta.reasoning}`,\n } as CoreContentBlock;\n case \"tool_call_chunk\":\n case \"server_tool_call_chunk\": {\n const merged = { ...target, ...delta } as Record<string, unknown>;\n if (delta.id == null && \"id\" in target && target.id != null) {\n merged.id = target.id;\n }\n if (delta.name == null && \"name\" in target && target.name != null) {\n merged.name = target.name;\n }\n merged.args = `${(\"args\" in target ? target.args : \"\") ?? \"\"}${delta.args ?? \"\"}`;\n return merged as unknown as CoreContentBlock;\n }\n default:\n return { ...target, ...delta } as CoreContentBlock;\n }\n}\n\nfunction coreContentBlockFromDelta(\n delta: CoreContentBlockDelta,\n current?: CoreContentBlock\n): CoreContentBlock {\n switch (delta.type) {\n case \"text-delta\":\n return { type: \"text\", text: delta.text } as CoreContentBlock;\n case \"reasoning-delta\":\n return {\n type: \"reasoning\",\n reasoning: delta.reasoning,\n } as CoreContentBlock;\n case \"data-delta\": {\n const merged = { ...(current ?? {}), data: delta.data } as Record<\n string,\n unknown\n >;\n if (delta.encoding) merged.encoding = delta.encoding;\n return merged as unknown as CoreContentBlock;\n }\n case \"block-delta\":\n return delta.fields as CoreContentBlock;\n }\n}\n\nfunction applyCoreEventDelta(\n current: CoreContentBlock | undefined,\n event: Extract<ChatModelStreamEvent, { event: \"content-block-delta\" }> & {\n content?: CoreContentBlock;\n }\n): CoreContentBlock {\n if (event.content) {\n return current\n ? applyCoreContentDelta(current, event.content)\n : event.content;\n }\n\n switch (event.delta.type) {\n case \"text-delta\":\n if (current?.type === \"text\") {\n return {\n ...current,\n text: `${\"text\" in current ? current.text : \"\"}${event.delta.text}`,\n } as CoreContentBlock;\n }\n return coreContentBlockFromDelta(event.delta, current);\n case \"reasoning-delta\":\n if (current?.type === \"reasoning\") {\n return {\n ...current,\n reasoning: `${\"reasoning\" in current ? current.reasoning : \"\"}${event.delta.reasoning}`,\n } as CoreContentBlock;\n }\n return coreContentBlockFromDelta(event.delta, current);\n case \"data-delta\": {\n const merged = { ...(current ?? {}) } as Record<string, unknown>;\n merged.data = `${(merged.data as string | undefined) ?? \"\"}${event.delta.data}`;\n if (event.delta.encoding) merged.encoding = event.delta.encoding;\n return merged as unknown as CoreContentBlock;\n }\n case \"block-delta\":\n return {\n ...(current ?? {}),\n ...event.delta.fields,\n } as CoreContentBlock;\n }\n}\n\nfunction normalizeUsage(\n usage: UsageInfo | Partial<UsageMetadata> | undefined\n): UsageMetadata | undefined {\n if (!usage) return undefined;\n return {\n ...usage,\n input_tokens: usage.input_tokens ?? 0,\n output_tokens: usage.output_tokens ?? 0,\n total_tokens: usage.total_tokens ?? 0,\n };\n}\n\nfunction readRunId(data: object): string | undefined {\n const record = data as Record<string, unknown>;\n return typeof record.run_id === \"string\" ? record.run_id : undefined;\n}\n\nfunction readUsage(data: object): UsageInfo | undefined {\n const usage = (data as Record<string, unknown>).usage;\n return usage != null && typeof usage === \"object\" && !Array.isArray(usage)\n ? (usage as UsageInfo)\n : undefined;\n}\n\nfunction readFinishMetadata(data: object): Record<string, unknown> | undefined {\n const record = data as Record<string, unknown>;\n const isRecord = (value: unknown): value is Record<string, unknown> =>\n value != null && typeof value === \"object\" && !Array.isArray(value);\n const merged = {\n ...(isRecord(record.responseMetadata) ? record.responseMetadata : {}),\n ...(isRecord(record.metadata) ? record.metadata : {}),\n };\n return Object.keys(merged).length > 0 ? merged : undefined;\n}\n\n/**\n * Mutable view of a streamed message as message and content-block events are\n * assembled into a single structure.\n */\nexport interface AssembledMessage {\n id: string;\n namespace: string[];\n blocks: ContentBlock[];\n node?: string;\n runId?: string;\n usage?: UsageInfo;\n metadata?: MessageMetadata;\n finishMetadata?: Record<string, any>;\n finishReason?: string;\n error?: {\n message: string;\n code?: string;\n };\n}\n\n/**\n * Symbol keys for assembler → StreamingMessage communication.\n * Module-private: invisible to external consumers, accessible to\n * {@link StreamingMessageAssembler} within this file.\n */\nconst PUSH_TEXT: unique symbol = Symbol(\"pushText\");\nconst PUSH_REASONING: unique symbol = Symbol(\"pushReasoning\");\nconst PUSH_EVENT: unique symbol = Symbol(\"pushEvent\");\nconst UPDATE_CONTEXT: unique symbol = Symbol(\"updateContext\");\nconst FINISH: unique symbol = Symbol(\"finish\");\nconst ERROR: unique symbol = Symbol(\"error\");\n\n/**\n * Live streaming view of a single message lifecycle, matching the\n * in-process `ChatModelStream` dual-interface pattern.\n *\n * - `text` / `reasoning`: iterate for streaming deltas, or await for\n * the full concatenated string after the message completes.\n * - `usage`: promise that resolves with token usage on message-finish.\n * - `blocks`: the assembled content blocks (updated as deltas arrive).\n *\n * Created by {@link StreamingMessageAssembler} and yielded by\n * the `session.messages` lazy getter.\n */\nexport class StreamingMessage\n implements AsyncIterable<ChatModelStreamEvent>, PromiseLike<AIMessage>\n{\n readonly id: string;\n readonly namespace: string[];\n node: string | undefined;\n readonly metadata: MessageMetadata | undefined;\n readonly assembled: AssembledMessage;\n readonly #events = new MultiCursorBuffer<ChatModelStreamEvent>();\n\n #textChunks: string[] = [];\n #reasoningChunks: string[] = [];\n #textWaiters: Array<() => void> = [];\n #reasoningWaiters: Array<() => void> = [];\n #textDone = false;\n #reasoningDone = false;\n\n #resolveText!: (v: string) => void;\n #resolveReasoning!: (v: string) => void;\n readonly #textPromise: Promise<string>;\n readonly #reasoningPromise: Promise<string>;\n\n constructor(assembled: AssembledMessage) {\n this.id = assembled.id;\n this.assembled = assembled;\n this.namespace = assembled.namespace;\n this.node = assembled.node;\n this.metadata = assembled.metadata;\n this.#textPromise = new Promise<string>((r) => {\n this.#resolveText = r;\n });\n this.#reasoningPromise = new Promise<string>((r) => {\n this.#resolveReasoning = r;\n });\n }\n\n get text(): TextContentStream {\n const chunks = this.#textChunks;\n const waiters = this.#textWaiters;\n const getDone = () => this.#textDone;\n let cursor = 0;\n return {\n [Symbol.asyncIterator]() {\n return {\n async next(): Promise<IteratorResult<string>> {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n };\n },\n then: this.#textPromise.then.bind(this.#textPromise),\n full: {\n async *[Symbol.asyncIterator]() {\n let accumulated = \"\";\n for await (const chunk of {\n [Symbol.asyncIterator]: () =>\n ({\n next: async (): Promise<IteratorResult<string>> => {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n }) satisfies AsyncIterator<string>,\n }) {\n accumulated += chunk;\n yield accumulated;\n }\n },\n },\n };\n }\n\n get reasoning(): TextContentStream {\n const chunks = this.#reasoningChunks;\n const waiters = this.#reasoningWaiters;\n const getDone = () => this.#reasoningDone;\n let cursor = 0;\n return {\n [Symbol.asyncIterator]() {\n return {\n async next(): Promise<IteratorResult<string>> {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n };\n },\n then: this.#reasoningPromise.then.bind(this.#reasoningPromise),\n full: {\n async *[Symbol.asyncIterator]() {\n let accumulated = \"\";\n for await (const chunk of {\n [Symbol.asyncIterator]: () =>\n ({\n next: async (): Promise<IteratorResult<string>> => {\n while (true) {\n if (cursor < chunks.length) {\n return { done: false, value: chunks[cursor++] };\n }\n if (getDone()) {\n return { done: true, value: undefined };\n }\n await new Promise<void>((resolve) => {\n waiters.push(resolve);\n });\n }\n },\n }) satisfies AsyncIterator<string>,\n }) {\n accumulated += chunk;\n yield accumulated;\n }\n },\n },\n };\n }\n\n get usage(): UsageMetadataStream {\n const promise = (async () => {\n let usage: UsageMetadata | undefined;\n for await (const snapshot of this.#usageIterator()) {\n usage = snapshot;\n }\n return usage;\n })();\n return {\n [Symbol.asyncIterator]: () => this.#usageIterator(),\n then: promise.then.bind(promise),\n };\n }\n\n get toolCalls(): ToolCallsStream {\n const events = this.#events;\n const iterator = async function* () {\n for await (const event of events) {\n if (\n event.event === \"content-block-finish\" &&\n event.content.type === \"tool_call\"\n ) {\n yield event.content as CoreContentBlock.Tools.ToolCall;\n }\n }\n };\n return {\n [Symbol.asyncIterator]: iterator,\n then: async (onfulfilled, onrejected) => {\n try {\n const calls: CoreContentBlock.Tools.ToolCall[] = [];\n for await (const call of iterator()) calls.push(call);\n return onfulfilled ? onfulfilled(calls) : (calls as never);\n } catch (err) {\n if (onrejected) return onrejected(err);\n throw err;\n }\n },\n full: {\n async *[Symbol.asyncIterator]() {\n const calls: CoreContentBlock.Tools.ToolCall[] = [];\n for await (const call of iterator()) {\n calls.push(call);\n yield [...calls];\n }\n },\n },\n };\n }\n\n get output(): PromiseLike<AIMessage> {\n return { then: (onf, onr) => this.#assembleMessage().then(onf, onr) };\n }\n\n get blocks(): ContentBlock[] {\n return this.assembled.blocks;\n }\n\n [Symbol.asyncIterator](): AsyncIterator<ChatModelStreamEvent> {\n return this.#events[Symbol.asyncIterator]();\n }\n\n then<TResult1 = AIMessage, TResult2 = never>(\n onfulfilled?:\n | ((value: AIMessage) => TResult1 | PromiseLike<TResult1>)\n | null,\n onrejected?: ((reason: unknown) => TResult2 | PromiseLike<TResult2>) | null\n ): PromiseLike<TResult1 | TResult2> {\n return this.#assembleMessage().then(onfulfilled, onrejected);\n }\n\n async *#usageIterator(): AsyncGenerator<UsageMetadata> {\n for await (const event of this.#events) {\n if (event.event === \"message-start\" && event.usage) {\n yield normalizeUsage(event.usage)!;\n } else if (event.event === \"message-finish\" && event.usage) {\n yield normalizeUsage(event.usage)!;\n }\n }\n }\n\n async #assembleMessage(): Promise<AIMessage> {\n const contentBlocks: Array<CoreContentBlock | undefined> = [];\n let id: string | undefined = this.id;\n let usage = normalizeUsage(this.assembled.usage);\n let metadata: Record<string, unknown> = {\n ...(this.assembled.finishMetadata ?? {}),\n };\n let finishReason: string | undefined = this.assembled.finishReason;\n\n for await (const event of this.#events) {\n switch (event.event) {\n case \"message-start\":\n id = event.id ?? id;\n if (event.usage) usage = normalizeUsage(event.usage);\n break;\n case \"content-block-start\":\n contentBlocks[event.index] = event.content;\n break;\n case \"content-block-delta\": {\n const current = contentBlocks[event.index];\n contentBlocks[event.index] = applyCoreEventDelta(current, event);\n break;\n }\n case \"content-block-finish\":\n contentBlocks[event.index] = event.content;\n break;\n case \"message-finish\": {\n finishReason = event.reason;\n if (event.usage) usage = normalizeUsage(event.usage);\n const finishMetadata = readFinishMetadata(event);\n if (finishMetadata != null) {\n metadata = { ...metadata, ...finishMetadata };\n }\n break;\n }\n default:\n break;\n }\n }\n\n return new AIMessage({\n id,\n content: contentBlocks.filter(\n (block): block is CoreContentBlock => block != null\n ),\n usage_metadata: usage,\n additional_kwargs: {\n namespace: this.namespace,\n ...(this.node != null ? { node: this.node } : {}),\n ...(this.assembled.runId != null\n ? { run_id: this.assembled.runId }\n : {}),\n ...(this.metadata != null ? { metadata: this.metadata } : {}),\n ...(this.assembled.usage != null\n ? { usage: this.assembled.usage }\n : {}),\n },\n response_metadata: {\n ...(finishReason != null ? { finish_reason: finishReason } : {}),\n ...metadata,\n output_version: \"v1\" as const,\n },\n });\n }\n\n [PUSH_EVENT](event: ChatModelStreamEvent): void {\n this.#events.push(event);\n }\n\n [UPDATE_CONTEXT](event: MessagesEvent): void {\n this.node = event.params.node ?? this.node;\n }\n\n [PUSH_TEXT](delta: string): void {\n this.#textChunks.push(delta);\n // Wake every caught-up iterator so each one advances its own cursor.\n // Iterators re-check `chunks.length` before delivering, which keeps\n // the cursor the single source of truth for what a consumer has seen.\n const pending = this.#textWaiters.splice(0, this.#textWaiters.length);\n for (const waiter of pending) waiter();\n }\n\n [PUSH_REASONING](delta: string): void {\n this.#reasoningChunks.push(delta);\n const pending = this.#reasoningWaiters.splice(\n 0,\n this.#reasoningWaiters.length\n );\n for (const waiter of pending) waiter();\n }\n\n [FINISH](): void {\n this.#textDone = true;\n this.#reasoningDone = true;\n this.#resolveText(this.#textChunks.join(\"\"));\n this.#resolveReasoning(this.#reasoningChunks.join(\"\"));\n const textPending = this.#textWaiters.splice(0, this.#textWaiters.length);\n for (const waiter of textPending) waiter();\n const reasoningPending = this.#reasoningWaiters.splice(\n 0,\n this.#reasoningWaiters.length\n );\n for (const waiter of reasoningPending) waiter();\n this.#events.close();\n }\n\n [ERROR](): void {\n this[FINISH]();\n }\n}\n\n/**\n * Public view yielded by message projections.\n *\n * `StreamingMessage` is PromiseLike so callers can still `await` a message\n * object directly, but TypeScript applies `Awaited<T>` to values produced by\n * `for await`. Exposing a non-thenable view keeps loop variables typed as the\n * streaming handle instead of as the finalized `AIMessage`.\n */\nexport type StreamingMessageHandle = Omit<StreamingMessage, \"then\">;\n\nexport function toStreamingMessageHandle(\n message: StreamingMessage\n): StreamingMessageHandle {\n return new Proxy(message, {\n get(target, prop) {\n if (prop === \"then\") return undefined;\n const value = Reflect.get(target, prop, target);\n return typeof value === \"function\" ? value.bind(target) : value;\n },\n has(target, prop) {\n if (prop === \"then\") return false;\n return prop in target;\n },\n }) as StreamingMessageHandle;\n}\n\n/**\n * Emitted by `MessageAssembler.consume()` to describe how a message changed in\n * response to a single protocol event.\n */\nexport type MessageAssemblyUpdate =\n | {\n kind: \"message-start\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"content-block-start\" | \"content-block-delta\";\n key: string;\n message: AssembledMessage;\n index: number;\n block: ContentBlock;\n event: MessagesEvent;\n }\n | {\n kind: \"content-block-finish\";\n key: string;\n message: AssembledMessage;\n index: number;\n block: FinalizedContentBlock;\n event: MessagesEvent;\n }\n | {\n kind: \"usage\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"message-finish\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n }\n | {\n kind: \"message-error\";\n key: string;\n message: AssembledMessage;\n event: MessagesEvent;\n };\n\nfunction cloneBlock<T extends ContentBlock>(block: T): T {\n return structuredClone(block);\n}\n\nfunction blockFromDelta(\n delta: CoreContentBlockDelta,\n current?: ContentBlock\n): ContentBlock {\n return coreContentBlockFromDelta(\n delta,\n current as unknown as CoreContentBlock | undefined\n ) as unknown as ContentBlock;\n}\n\nfunction applyContentDelta(\n target: ContentBlock,\n delta: ContentBlock\n): ContentBlock {\n if (target.type !== delta.type) {\n return cloneBlock(delta);\n }\n\n switch (delta.type) {\n case \"text\":\n return {\n ...target,\n ...delta,\n text: `${\"text\" in target ? target.text : \"\"}${delta.text}`,\n } as ContentBlock;\n case \"reasoning\":\n return {\n ...target,\n ...delta,\n reasoning: `${\"reasoning\" in target ? target.reasoning : \"\"}${delta.reasoning}`,\n } as ContentBlock;\n case \"tool_call_chunk\":\n case \"server_tool_call_chunk\": {\n // Spread target first, then delta — but preserve target's\n // ``id``/``name`` when the delta explicitly sets them to\n // null/undefined. Some providers (notably Anthropic via the\n // langchain-core compat bridge) only attach the tool-call\n // identifiers to the first ``content-block-start`` chunk; every\n // subsequent ``input_json_delta`` chunk carries ``id=null,\n // name=null``. A naive ``{...target, ...delta}`` spread\n // overwrites the captured identifiers with null, which in turn\n // makes downstream consumers (e.g. ``extractToolCallChunks`` in\n // ``assembled-to-message.ts``) drop the chunk on the floor until\n // the final ``content-block-finish`` event promotes it to a\n // finalized ``tool_call`` — causing tool-call cards to appear\n // all-at-once at the end of the turn instead of incrementally.\n const merged = { ...target, ...delta } as Record<string, unknown>;\n if (delta.id == null && \"id\" in target && target.id != null) {\n merged.id = target.id;\n }\n if (delta.name == null && \"name\" in target && target.name != null) {\n merged.name = target.name;\n }\n merged.args = `${(\"args\" in target ? target.args : \"\") ?? \"\"}${delta.args ?? \"\"}`;\n return merged as unknown as ContentBlock;\n }\n default:\n return {\n ...target,\n ...delta,\n } as ContentBlock;\n }\n}\n\nfunction messageKeyFor(event: MessagesEvent): string {\n const { namespace, node, data } = event.params;\n const namespaceKey = namespace.join(\"/\");\n const runId = readRunId(data);\n const messageId = data.event === \"message-start\" ? (data.id ?? \"\") : \"\";\n return `${namespaceKey}::${node ?? \"\"}::${runId ?? messageId}`;\n}\n\nfunction toChatModelStreamEvent(event: MessagesEvent): ChatModelStreamEvent {\n return event.params.data as unknown as ChatModelStreamEvent;\n}\n\n/**\n * Incrementally assembles `messages` events into complete message objects.\n */\nexport class MessageAssembler {\n private readonly activeMessages = new Map<string, AssembledMessage>();\n private readonly activeByNamespaceNode = new Map<string, string>();\n private readonly blockIndexByProtocolIndexAndType = new Map<string, number>();\n\n /**\n * Applies a single message event and returns the resulting assembly update.\n *\n * @param event - Incoming `messages` event to fold into the assembler state.\n */\n consume(event: MessagesEvent): MessageAssemblyUpdate {\n const data = event.params.data;\n const namespaceNodeKey = `${event.params.namespace.join(\"/\")}::${event.params.node ?? \"\"}`;\n const runId = readRunId(data);\n\n if (data.event === \"message-start\") {\n const key = messageKeyFor(event);\n this.activeByNamespaceNode.set(namespaceNodeKey, key);\n const message: AssembledMessage = {\n id: data.id,\n namespace: [...event.params.namespace],\n node: event.params.node,\n runId,\n usage: readUsage(data),\n metadata: data.metadata,\n blocks: [],\n };\n this.activeMessages.set(key, message);\n return { kind: \"message-start\", key, message, event };\n }\n\n const runKey = runId != null ? `${namespaceNodeKey}::${runId}` : undefined;\n const fallbackKey = this.activeByNamespaceNode.get(namespaceNodeKey);\n const fallbackMessage =\n fallbackKey != null ? this.activeMessages.get(fallbackKey) : undefined;\n const activeKey =\n (runKey != null && this.activeMessages.has(runKey)\n ? runKey\n : undefined) ??\n (runId == null || fallbackMessage?.runId == null\n ? fallbackKey\n : undefined);\n if (!activeKey) {\n // A continuation event (delta/finish/error) arrived without a\n // prior `message-start`. This can happen on late-attaching\n // subscriptions when the server has already trimmed the\n // `message-start` from its replay buffer. Synthesize a minimal\n // active message so the assembler can still fold subsequent\n // events instead of hard-failing the caller.\n const syntheticKey = runKey ?? `${namespaceNodeKey}::`;\n this.activeByNamespaceNode.set(namespaceNodeKey, syntheticKey);\n const synthetic: AssembledMessage = {\n id: data.id,\n namespace: [...event.params.namespace],\n node: event.params.node,\n runId,\n blocks: [],\n };\n this.activeMessages.set(syntheticKey, synthetic);\n return this.consume(event);\n }\n\n const message = this.activeMessages.get(activeKey);\n if (!message) {\n throw new Error(`No active message state found for key ${activeKey}`);\n }\n\n if ((data as { event?: string }).event === \"usage\") {\n message.usage = (data as { usage?: UsageInfo }).usage;\n return {\n kind: \"usage\",\n key: activeKey,\n message,\n event,\n };\n }\n\n switch (data.event) {\n case \"content-block-start\": {\n message.blocks[data.index] = cloneBlock(data.content);\n this.blockIndexByProtocolIndexAndType.set(\n blockIndexKey(activeKey, data.index, data.content.type),\n data.index\n );\n return {\n kind: \"content-block-start\",\n key: activeKey,\n message,\n index: data.index,\n block: data.content,\n event,\n };\n }\n case \"content-block-delta\": {\n const deltaEvent = data as typeof data & {\n content?: ContentBlock;\n delta?: CoreContentBlockDelta;\n };\n const deltaBlock =\n deltaEvent.content ??\n (deltaEvent.delta != null\n ? blockFromDelta(deltaEvent.delta, message.blocks[data.index])\n : undefined);\n if (deltaBlock == null) {\n throw new Error(\"Received content-block-delta without content\");\n }\n const targetIndex = this.resolveBlockIndex(\n activeKey,\n message.blocks,\n data.index,\n deltaBlock.type\n );\n const current = message.blocks[targetIndex];\n message.blocks[targetIndex] =\n deltaEvent.content != null\n ? current == null\n ? cloneBlock(deltaEvent.content)\n : applyContentDelta(current, deltaEvent.content)\n : (applyCoreEventDelta(\n current as CoreContentBlock | undefined,\n data as unknown as Extract<\n ChatModelStreamEvent,\n { event: \"content-block-delta\" }\n >\n ) as unknown as ContentBlock);\n return {\n kind: \"content-block-delta\",\n key: activeKey,\n message,\n index: targetIndex,\n block: deltaBlock,\n event,\n };\n }\n case \"content-block-finish\": {\n const targetIndex = this.resolveFinishBlockIndex(\n activeKey,\n data.index,\n data.content.type\n );\n message.blocks[targetIndex] = cloneBlock(data.content);\n return {\n kind: \"content-block-finish\",\n key: activeKey,\n message,\n index: targetIndex,\n block: data.content,\n event,\n };\n }\n case \"message-finish\": {\n message.usage = readUsage(data) ?? message.usage;\n message.finishMetadata = readFinishMetadata(data);\n message.finishReason = data.reason;\n this.clearActiveMessage(activeKey, namespaceNodeKey);\n this.clearBlockIndexAliases(activeKey);\n return {\n kind: \"message-finish\",\n key: activeKey,\n message: structuredClone(message),\n event,\n };\n }\n case \"error\": {\n message.error = { message: data.message, code: data.code };\n this.clearActiveMessage(activeKey, namespaceNodeKey);\n this.clearBlockIndexAliases(activeKey);\n return {\n kind: \"message-error\",\n key: activeKey,\n message: structuredClone(message),\n event,\n };\n }\n }\n }\n\n private clearActiveMessage(\n activeKey: string,\n namespaceNodeKey: string\n ): void {\n this.activeMessages.delete(activeKey);\n if (this.activeByNamespaceNode.get(namespaceNodeKey) === activeKey) {\n this.activeByNamespaceNode.delete(namespaceNodeKey);\n }\n }\n\n private resolveBlockIndex(\n activeKey: string,\n blocks: ContentBlock[],\n protocolIndex: number,\n blockType: string\n ): number {\n const current = blocks[protocolIndex];\n if (\n current == null ||\n current.type === blockType ||\n areCompatibleBlockTypes(current.type, blockType)\n ) {\n this.blockIndexByProtocolIndexAndType.set(\n blockIndexKey(activeKey, protocolIndex, blockType),\n protocolIndex\n );\n return protocolIndex;\n }\n\n const key = blockIndexKey(activeKey, protocolIndex, blockType);\n const existing = this.blockIndexByProtocolIndexAndType.get(key);\n if (existing != null) return existing;\n\n const nextIndex = blocks.length;\n this.blockIndexByProtocolIndexAndType.set(key, nextIndex);\n return nextIndex;\n }\n\n private resolveFinishBlockIndex(\n activeKey: string,\n protocolIndex: number,\n blockType: string\n ): number {\n const key = blockIndexKey(activeKey, protocolIndex, blockType);\n const existing = this.blockIndexByProtocolIndexAndType.get(key);\n if (existing != null) return existing;\n\n this.blockIndexByProtocolIndexAndType.set(key, protocolIndex);\n return protocolIndex;\n }\n\n private clearBlockIndexAliases(activeKey: string): void {\n const prefix = `${activeKey}::`;\n for (const key of this.blockIndexByProtocolIndexAndType.keys()) {\n if (key.startsWith(prefix))\n this.blockIndexByProtocolIndexAndType.delete(key);\n }\n }\n}\n\nfunction blockIndexKey(\n activeKey: string,\n protocolIndex: number,\n blockType: string\n): string {\n return `${activeKey}::${protocolIndex}::${blockType}`;\n}\n\nfunction areCompatibleBlockTypes(\n currentType: string,\n nextType: string\n): boolean {\n const toolCallTypes = new Set([\n \"tool_call\",\n \"tool_call_chunk\",\n \"tool_use\",\n \"input_json_delta\",\n ]);\n const serverToolCallTypes = new Set([\n \"server_tool_call\",\n \"server_tool_call_chunk\",\n ]);\n\n return (\n (toolCallTypes.has(currentType) && toolCallTypes.has(nextType)) ||\n (serverToolCallTypes.has(currentType) && serverToolCallTypes.has(nextType))\n );\n}\n\n/**\n * Assembles `messages` events into {@link StreamingMessage} instances\n * with live text/reasoning delta streams, matching the in-process\n * `ChatModelStream` dual-interface pattern.\n */\nexport class StreamingMessageAssembler {\n readonly #assembler = new MessageAssembler();\n readonly #activeStreaming = new Map<string, StreamingMessage>();\n\n /**\n * Folds a single event and returns a new {@link StreamingMessage}\n * when a `message-start` is seen, or `undefined` for continuation\n * events (deltas, finish, error).\n */\n consume(event: MessagesEvent): StreamingMessage | undefined {\n const update = this.#assembler.consume(event);\n if (update == null) return undefined;\n\n switch (update.kind) {\n case \"message-start\": {\n const streaming = new StreamingMessage(update.message);\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n this.#activeStreaming.set(update.key, streaming);\n return streaming;\n }\n case \"content-block-start\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n if (\n streaming &&\n update.block.type === \"text\" &&\n \"text\" in update.block &&\n update.block.text\n ) {\n streaming[PUSH_TEXT](update.block.text);\n }\n if (\n streaming &&\n update.block.type === \"reasoning\" &&\n \"reasoning\" in update.block &&\n update.block.reasoning\n ) {\n streaming[PUSH_REASONING](update.block.reasoning);\n }\n return undefined;\n }\n case \"content-block-delta\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (!streaming) return undefined;\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n if (update.block.type === \"text\" && \"text\" in update.block) {\n streaming[PUSH_TEXT](update.block.text);\n }\n if (update.block.type === \"reasoning\" && \"reasoning\" in update.block) {\n streaming[PUSH_REASONING](update.block.reasoning);\n }\n return undefined;\n }\n case \"content-block-finish\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n return undefined;\n }\n case \"usage\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n }\n return undefined;\n }\n case \"message-finish\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n streaming[FINISH]();\n this.#activeStreaming.delete(update.key);\n }\n return undefined;\n }\n case \"message-error\": {\n const streaming = this.#activeStreaming.get(update.key);\n if (streaming) {\n streaming[UPDATE_CONTEXT](update.event);\n streaming[PUSH_EVENT](toChatModelStreamEvent(update.event));\n streaming[ERROR]();\n this.#activeStreaming.delete(update.key);\n }\n return undefined;\n }\n }\n }\n}\n"],"mappings":";;;AA8BA,SAAS,sBACP,QACA,OACkB;AAClB,KAAI,OAAO,SAAS,MAAM,KACxB,QAAO,gBAAgB,MAAM;AAG/B,SAAQ,MAAM,MAAd;EACE,KAAK,OACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,MAAM,GAAG,UAAU,SAAS,OAAO,OAAO,KAAK,MAAM;GACtD;EACH,KAAK,YACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,WAAW,GAAG,eAAe,SAAS,OAAO,YAAY,KAAK,MAAM;GACrE;EACH,KAAK;EACL,KAAK,0BAA0B;GAC7B,MAAM,SAAS;IAAE,GAAG;IAAQ,GAAG;IAAO;AACtC,OAAI,MAAM,MAAM,QAAQ,QAAQ,UAAU,OAAO,MAAM,KACrD,QAAO,KAAK,OAAO;AAErB,OAAI,MAAM,QAAQ,QAAQ,UAAU,UAAU,OAAO,QAAQ,KAC3D,QAAO,OAAO,OAAO;AAEvB,UAAO,OAAO,IAAI,UAAU,SAAS,OAAO,OAAO,OAAO,KAAK,MAAM,QAAQ;AAC7E,UAAO;;EAET,QACE,QAAO;GAAE,GAAG;GAAQ,GAAG;GAAO;;;AAIpC,SAAS,0BACP,OACA,SACkB;AAClB,SAAQ,MAAM,MAAd;EACE,KAAK,aACH,QAAO;GAAE,MAAM;GAAQ,MAAM,MAAM;GAAM;EAC3C,KAAK,kBACH,QAAO;GACL,MAAM;GACN,WAAW,MAAM;GAClB;EACH,KAAK,cAAc;GACjB,MAAM,SAAS;IAAE,GAAI,WAAW,EAAE;IAAG,MAAM,MAAM;IAAM;AAIvD,OAAI,MAAM,SAAU,QAAO,WAAW,MAAM;AAC5C,UAAO;;EAET,KAAK,cACH,QAAO,MAAM;;;AAInB,SAAS,oBACP,SACA,OAGkB;AAClB,KAAI,MAAM,QACR,QAAO,UACH,sBAAsB,SAAS,MAAM,QAAQ,GAC7C,MAAM;AAGZ,SAAQ,MAAM,MAAM,MAApB;EACE,KAAK;AACH,OAAI,SAAS,SAAS,OACpB,QAAO;IACL,GAAG;IACH,MAAM,GAAG,UAAU,UAAU,QAAQ,OAAO,KAAK,MAAM,MAAM;IAC9D;AAEH,UAAO,0BAA0B,MAAM,OAAO,QAAQ;EACxD,KAAK;AACH,OAAI,SAAS,SAAS,YACpB,QAAO;IACL,GAAG;IACH,WAAW,GAAG,eAAe,UAAU,QAAQ,YAAY,KAAK,MAAM,MAAM;IAC7E;AAEH,UAAO,0BAA0B,MAAM,OAAO,QAAQ;EACxD,KAAK,cAAc;GACjB,MAAM,SAAS,EAAE,GAAI,WAAW,EAAE,EAAG;AACrC,UAAO,OAAO,GAAI,OAAO,QAA+B,KAAK,MAAM,MAAM;AACzE,OAAI,MAAM,MAAM,SAAU,QAAO,WAAW,MAAM,MAAM;AACxD,UAAO;;EAET,KAAK,cACH,QAAO;GACL,GAAI,WAAW,EAAE;GACjB,GAAG,MAAM,MAAM;GAChB;;;AAIP,SAAS,eACP,OAC2B;AAC3B,KAAI,CAAC,MAAO,QAAO,KAAA;AACnB,QAAO;EACL,GAAG;EACH,cAAc,MAAM,gBAAgB;EACpC,eAAe,MAAM,iBAAiB;EACtC,cAAc,MAAM,gBAAgB;EACrC;;AAGH,SAAS,UAAU,MAAkC;CACnD,MAAM,SAAS;AACf,QAAO,OAAO,OAAO,WAAW,WAAW,OAAO,SAAS,KAAA;;AAG7D,SAAS,UAAU,MAAqC;CACtD,MAAM,QAAS,KAAiC;AAChD,QAAO,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,GACrE,QACD,KAAA;;AAGN,SAAS,mBAAmB,MAAmD;CAC7E,MAAM,SAAS;CACf,MAAM,YAAY,UAChB,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM;CACrE,MAAM,SAAS;EACb,GAAI,SAAS,OAAO,iBAAiB,GAAG,OAAO,mBAAmB,EAAE;EACpE,GAAI,SAAS,OAAO,SAAS,GAAG,OAAO,WAAW,EAAE;EACrD;AACD,QAAO,OAAO,KAAK,OAAO,CAAC,SAAS,IAAI,SAAS,KAAA;;;;;;;AA4BnD,MAAM,YAA2B,OAAO,WAAW;AACnD,MAAM,iBAAgC,OAAO,gBAAgB;AAC7D,MAAM,aAA4B,OAAO,YAAY;AACrD,MAAM,iBAAgC,OAAO,gBAAgB;AAC7D,MAAM,SAAwB,OAAO,SAAS;AAC9C,MAAM,QAAuB,OAAO,QAAQ;;;;;;;;;;;;;AAc5C,IAAa,mBAAb,MAEA;CACE;CACA;CACA;CACA;CACA;CACA,UAAmB,IAAI,mBAAyC;CAEhE,cAAwB,EAAE;CAC1B,mBAA6B,EAAE;CAC/B,eAAkC,EAAE;CACpC,oBAAuC,EAAE;CACzC,YAAY;CACZ,iBAAiB;CAEjB;CACA;CACA;CACA;CAEA,YAAY,WAA6B;AACvC,OAAK,KAAK,UAAU;AACpB,OAAK,YAAY;AACjB,OAAK,YAAY,UAAU;AAC3B,OAAK,OAAO,UAAU;AACtB,OAAK,WAAW,UAAU;AAC1B,QAAA,cAAoB,IAAI,SAAiB,MAAM;AAC7C,SAAA,cAAoB;IACpB;AACF,QAAA,mBAAyB,IAAI,SAAiB,MAAM;AAClD,SAAA,mBAAyB;IACzB;;CAGJ,IAAI,OAA0B;EAC5B,MAAM,SAAS,MAAA;EACf,MAAM,UAAU,MAAA;EAChB,MAAM,gBAAgB,MAAA;EACtB,IAAI,SAAS;AACb,SAAO;GACL,CAAC,OAAO,iBAAiB;AACvB,WAAO,EACL,MAAM,OAAwC;AAC5C,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP;;GAEH,MAAM,MAAA,YAAkB,KAAK,KAAK,MAAA,YAAkB;GACpD,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,IAAI,cAAc;AAClB,eAAW,MAAM,SAAS,GACvB,OAAO,uBACL,EACC,MAAM,YAA6C;AACjD,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP,GACJ,EAAE;AACD,oBAAe;AACf,WAAM;;MAGX;GACF;;CAGH,IAAI,YAA+B;EACjC,MAAM,SAAS,MAAA;EACf,MAAM,UAAU,MAAA;EAChB,MAAM,gBAAgB,MAAA;EACtB,IAAI,SAAS;AACb,SAAO;GACL,CAAC,OAAO,iBAAiB;AACvB,WAAO,EACL,MAAM,OAAwC;AAC5C,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP;;GAEH,MAAM,MAAA,iBAAuB,KAAK,KAAK,MAAA,iBAAuB;GAC9D,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,IAAI,cAAc;AAClB,eAAW,MAAM,SAAS,GACvB,OAAO,uBACL,EACC,MAAM,YAA6C;AACjD,YAAO,MAAM;AACX,UAAI,SAAS,OAAO,OAClB,QAAO;OAAE,MAAM;OAAO,OAAO,OAAO;OAAW;AAEjD,UAAI,SAAS,CACX,QAAO;OAAE,MAAM;OAAM,OAAO,KAAA;OAAW;AAEzC,YAAM,IAAI,SAAe,YAAY;AACnC,eAAQ,KAAK,QAAQ;QACrB;;OAGP,GACJ,EAAE;AACD,oBAAe;AACf,WAAM;;MAGX;GACF;;CAGH,IAAI,QAA6B;EAC/B,MAAM,WAAW,YAAY;GAC3B,IAAI;AACJ,cAAW,MAAM,YAAY,MAAA,eAAqB,CAChD,SAAQ;AAEV,UAAO;MACL;AACJ,SAAO;IACJ,OAAO,sBAAsB,MAAA,eAAqB;GACnD,MAAM,QAAQ,KAAK,KAAK,QAAQ;GACjC;;CAGH,IAAI,YAA6B;EAC/B,MAAM,SAAS,MAAA;EACf,MAAM,WAAW,mBAAmB;AAClC,cAAW,MAAM,SAAS,OACxB,KACE,MAAM,UAAU,0BAChB,MAAM,QAAQ,SAAS,YAEvB,OAAM,MAAM;;AAIlB,SAAO;IACJ,OAAO,gBAAgB;GACxB,MAAM,OAAO,aAAa,eAAe;AACvC,QAAI;KACF,MAAM,QAA2C,EAAE;AACnD,gBAAW,MAAM,QAAQ,UAAU,CAAE,OAAM,KAAK,KAAK;AACrD,YAAO,cAAc,YAAY,MAAM,GAAI;aACpC,KAAK;AACZ,SAAI,WAAY,QAAO,WAAW,IAAI;AACtC,WAAM;;;GAGV,MAAM,EACJ,QAAQ,OAAO,iBAAiB;IAC9B,MAAM,QAA2C,EAAE;AACnD,eAAW,MAAM,QAAQ,UAAU,EAAE;AACnC,WAAM,KAAK,KAAK;AAChB,WAAM,CAAC,GAAG,MAAM;;MAGrB;GACF;;CAGH,IAAI,SAAiC;AACnC,SAAO,EAAE,OAAO,KAAK,QAAQ,MAAA,iBAAuB,CAAC,KAAK,KAAK,IAAI,EAAE;;CAGvE,IAAI,SAAyB;AAC3B,SAAO,KAAK,UAAU;;CAGxB,CAAC,OAAO,iBAAsD;AAC5D,SAAO,MAAA,OAAa,OAAO,gBAAgB;;CAG7C,KACE,aAGA,YACkC;AAClC,SAAO,MAAA,iBAAuB,CAAC,KAAK,aAAa,WAAW;;CAG9D,QAAA,gBAAuD;AACrD,aAAW,MAAM,SAAS,MAAA,OACxB,KAAI,MAAM,UAAU,mBAAmB,MAAM,MAC3C,OAAM,eAAe,MAAM,MAAM;WACxB,MAAM,UAAU,oBAAoB,MAAM,MACnD,OAAM,eAAe,MAAM,MAAM;;CAKvC,OAAA,kBAA6C;EAC3C,MAAM,gBAAqD,EAAE;EAC7D,IAAI,KAAyB,KAAK;EAClC,IAAI,QAAQ,eAAe,KAAK,UAAU,MAAM;EAChD,IAAI,WAAoC,EACtC,GAAI,KAAK,UAAU,kBAAkB,EAAE,EACxC;EACD,IAAI,eAAmC,KAAK,UAAU;AAEtD,aAAW,MAAM,SAAS,MAAA,OACxB,SAAQ,MAAM,OAAd;GACE,KAAK;AACH,SAAK,MAAM,MAAM;AACjB,QAAI,MAAM,MAAO,SAAQ,eAAe,MAAM,MAAM;AACpD;GACF,KAAK;AACH,kBAAc,MAAM,SAAS,MAAM;AACnC;GACF,KAAK,uBAAuB;IAC1B,MAAM,UAAU,cAAc,MAAM;AACpC,kBAAc,MAAM,SAAS,oBAAoB,SAAS,MAAM;AAChE;;GAEF,KAAK;AACH,kBAAc,MAAM,SAAS,MAAM;AACnC;GACF,KAAK,kBAAkB;AACrB,mBAAe,MAAM;AACrB,QAAI,MAAM,MAAO,SAAQ,eAAe,MAAM,MAAM;IACpD,MAAM,iBAAiB,mBAAmB,MAAM;AAChD,QAAI,kBAAkB,KACpB,YAAW;KAAE,GAAG;KAAU,GAAG;KAAgB;AAE/C;;GAEF,QACE;;AAIN,SAAO,IAAI,UAAU;GACnB;GACA,SAAS,cAAc,QACpB,UAAqC,SAAS,KAChD;GACD,gBAAgB;GAChB,mBAAmB;IACjB,WAAW,KAAK;IAChB,GAAI,KAAK,QAAQ,OAAO,EAAE,MAAM,KAAK,MAAM,GAAG,EAAE;IAChD,GAAI,KAAK,UAAU,SAAS,OACxB,EAAE,QAAQ,KAAK,UAAU,OAAO,GAChC,EAAE;IACN,GAAI,KAAK,YAAY,OAAO,EAAE,UAAU,KAAK,UAAU,GAAG,EAAE;IAC5D,GAAI,KAAK,UAAU,SAAS,OACxB,EAAE,OAAO,KAAK,UAAU,OAAO,GAC/B,EAAE;IACP;GACD,mBAAmB;IACjB,GAAI,gBAAgB,OAAO,EAAE,eAAe,cAAc,GAAG,EAAE;IAC/D,GAAG;IACH,gBAAgB;IACjB;GACF,CAAC;;CAGJ,CAAC,YAAY,OAAmC;AAC9C,QAAA,OAAa,KAAK,MAAM;;CAG1B,CAAC,gBAAgB,OAA4B;AAC3C,OAAK,OAAO,MAAM,OAAO,QAAQ,KAAK;;CAGxC,CAAC,WAAW,OAAqB;AAC/B,QAAA,WAAiB,KAAK,MAAM;EAI5B,MAAM,UAAU,MAAA,YAAkB,OAAO,GAAG,MAAA,YAAkB,OAAO;AACrE,OAAK,MAAM,UAAU,QAAS,SAAQ;;CAGxC,CAAC,gBAAgB,OAAqB;AACpC,QAAA,gBAAsB,KAAK,MAAM;EACjC,MAAM,UAAU,MAAA,iBAAuB,OACrC,GACA,MAAA,iBAAuB,OACxB;AACD,OAAK,MAAM,UAAU,QAAS,SAAQ;;CAGxC,CAAC,UAAgB;AACf,QAAA,WAAiB;AACjB,QAAA,gBAAsB;AACtB,QAAA,YAAkB,MAAA,WAAiB,KAAK,GAAG,CAAC;AAC5C,QAAA,iBAAuB,MAAA,gBAAsB,KAAK,GAAG,CAAC;EACtD,MAAM,cAAc,MAAA,YAAkB,OAAO,GAAG,MAAA,YAAkB,OAAO;AACzE,OAAK,MAAM,UAAU,YAAa,SAAQ;EAC1C,MAAM,mBAAmB,MAAA,iBAAuB,OAC9C,GACA,MAAA,iBAAuB,OACxB;AACD,OAAK,MAAM,UAAU,iBAAkB,SAAQ;AAC/C,QAAA,OAAa,OAAO;;CAGtB,CAAC,SAAe;AACd,OAAK,SAAS;;;AAclB,SAAgB,yBACd,SACwB;AACxB,QAAO,IAAI,MAAM,SAAS;EACxB,IAAI,QAAQ,MAAM;AAChB,OAAI,SAAS,OAAQ,QAAO,KAAA;GAC5B,MAAM,QAAQ,QAAQ,IAAI,QAAQ,MAAM,OAAO;AAC/C,UAAO,OAAO,UAAU,aAAa,MAAM,KAAK,OAAO,GAAG;;EAE5D,IAAI,QAAQ,MAAM;AAChB,OAAI,SAAS,OAAQ,QAAO;AAC5B,UAAO,QAAQ;;EAElB,CAAC;;AAiDJ,SAAS,WAAmC,OAAa;AACvD,QAAO,gBAAgB,MAAM;;AAG/B,SAAS,eACP,OACA,SACc;AACd,QAAO,0BACL,OACA,QACD;;AAGH,SAAS,kBACP,QACA,OACc;AACd,KAAI,OAAO,SAAS,MAAM,KACxB,QAAO,WAAW,MAAM;AAG1B,SAAQ,MAAM,MAAd;EACE,KAAK,OACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,MAAM,GAAG,UAAU,SAAS,OAAO,OAAO,KAAK,MAAM;GACtD;EACH,KAAK,YACH,QAAO;GACL,GAAG;GACH,GAAG;GACH,WAAW,GAAG,eAAe,SAAS,OAAO,YAAY,KAAK,MAAM;GACrE;EACH,KAAK;EACL,KAAK,0BAA0B;GAc7B,MAAM,SAAS;IAAE,GAAG;IAAQ,GAAG;IAAO;AACtC,OAAI,MAAM,MAAM,QAAQ,QAAQ,UAAU,OAAO,MAAM,KACrD,QAAO,KAAK,OAAO;AAErB,OAAI,MAAM,QAAQ,QAAQ,UAAU,UAAU,OAAO,QAAQ,KAC3D,QAAO,OAAO,OAAO;AAEvB,UAAO,OAAO,IAAI,UAAU,SAAS,OAAO,OAAO,OAAO,KAAK,MAAM,QAAQ;AAC7E,UAAO;;EAET,QACE,QAAO;GACL,GAAG;GACH,GAAG;GACJ;;;AAIP,SAAS,cAAc,OAA8B;CACnD,MAAM,EAAE,WAAW,MAAM,SAAS,MAAM;CACxC,MAAM,eAAe,UAAU,KAAK,IAAI;CACxC,MAAM,QAAQ,UAAU,KAAK;CAC7B,MAAM,YAAY,KAAK,UAAU,kBAAmB,KAAK,MAAM,KAAM;AACrE,QAAO,GAAG,aAAa,IAAI,QAAQ,GAAG,IAAI,SAAS;;AAGrD,SAAS,uBAAuB,OAA4C;AAC1E,QAAO,MAAM,OAAO;;;;;AAMtB,IAAa,mBAAb,MAA8B;CAC5B,iCAAkC,IAAI,KAA+B;CACrE,wCAAyC,IAAI,KAAqB;CAClE,mDAAoD,IAAI,KAAqB;;;;;;CAO7E,QAAQ,OAA6C;EACnD,MAAM,OAAO,MAAM,OAAO;EAC1B,MAAM,mBAAmB,GAAG,MAAM,OAAO,UAAU,KAAK,IAAI,CAAC,IAAI,MAAM,OAAO,QAAQ;EACtF,MAAM,QAAQ,UAAU,KAAK;AAE7B,MAAI,KAAK,UAAU,iBAAiB;GAClC,MAAM,MAAM,cAAc,MAAM;AAChC,QAAK,sBAAsB,IAAI,kBAAkB,IAAI;GACrD,MAAM,UAA4B;IAChC,IAAI,KAAK;IACT,WAAW,CAAC,GAAG,MAAM,OAAO,UAAU;IACtC,MAAM,MAAM,OAAO;IACnB;IACA,OAAO,UAAU,KAAK;IACtB,UAAU,KAAK;IACf,QAAQ,EAAE;IACX;AACD,QAAK,eAAe,IAAI,KAAK,QAAQ;AACrC,UAAO;IAAE,MAAM;IAAiB;IAAK;IAAS;IAAO;;EAGvD,MAAM,SAAS,SAAS,OAAO,GAAG,iBAAiB,IAAI,UAAU,KAAA;EACjE,MAAM,cAAc,KAAK,sBAAsB,IAAI,iBAAiB;EACpE,MAAM,kBACJ,eAAe,OAAO,KAAK,eAAe,IAAI,YAAY,GAAG,KAAA;EAC/D,MAAM,aACH,UAAU,QAAQ,KAAK,eAAe,IAAI,OAAO,GAC9C,SACA,KAAA,OACH,SAAS,QAAQ,iBAAiB,SAAS,OACxC,cACA,KAAA;AACN,MAAI,CAAC,WAAW;GAOd,MAAM,eAAe,UAAU,GAAG,iBAAiB;AACnD,QAAK,sBAAsB,IAAI,kBAAkB,aAAa;GAC9D,MAAM,YAA8B;IAClC,IAAI,KAAK;IACT,WAAW,CAAC,GAAG,MAAM,OAAO,UAAU;IACtC,MAAM,MAAM,OAAO;IACnB;IACA,QAAQ,EAAE;IACX;AACD,QAAK,eAAe,IAAI,cAAc,UAAU;AAChD,UAAO,KAAK,QAAQ,MAAM;;EAG5B,MAAM,UAAU,KAAK,eAAe,IAAI,UAAU;AAClD,MAAI,CAAC,QACH,OAAM,IAAI,MAAM,yCAAyC,YAAY;AAGvE,MAAK,KAA4B,UAAU,SAAS;AAClD,WAAQ,QAAS,KAA+B;AAChD,UAAO;IACL,MAAM;IACN,KAAK;IACL;IACA;IACD;;AAGH,UAAQ,KAAK,OAAb;GACE,KAAK;AACH,YAAQ,OAAO,KAAK,SAAS,WAAW,KAAK,QAAQ;AACrD,SAAK,iCAAiC,IACpC,cAAc,WAAW,KAAK,OAAO,KAAK,QAAQ,KAAK,EACvD,KAAK,MACN;AACD,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO,KAAK;KACZ,OAAO,KAAK;KACZ;KACD;GAEH,KAAK,uBAAuB;IAC1B,MAAM,aAAa;IAInB,MAAM,aACJ,WAAW,YACV,WAAW,SAAS,OACjB,eAAe,WAAW,OAAO,QAAQ,OAAO,KAAK,OAAO,GAC5D,KAAA;AACN,QAAI,cAAc,KAChB,OAAM,IAAI,MAAM,+CAA+C;IAEjE,MAAM,cAAc,KAAK,kBACvB,WACA,QAAQ,QACR,KAAK,OACL,WAAW,KACZ;IACD,MAAM,UAAU,QAAQ,OAAO;AAC/B,YAAQ,OAAO,eACb,WAAW,WAAW,OAClB,WAAW,OACT,WAAW,WAAW,QAAQ,GAC9B,kBAAkB,SAAS,WAAW,QAAQ,GAC/C,oBACC,SACA,KAID;AACP,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO;KACP,OAAO;KACP;KACD;;GAEH,KAAK,wBAAwB;IAC3B,MAAM,cAAc,KAAK,wBACvB,WACA,KAAK,OACL,KAAK,QAAQ,KACd;AACD,YAAQ,OAAO,eAAe,WAAW,KAAK,QAAQ;AACtD,WAAO;KACL,MAAM;KACN,KAAK;KACL;KACA,OAAO;KACP,OAAO,KAAK;KACZ;KACD;;GAEH,KAAK;AACH,YAAQ,QAAQ,UAAU,KAAK,IAAI,QAAQ;AAC3C,YAAQ,iBAAiB,mBAAmB,KAAK;AACjD,YAAQ,eAAe,KAAK;AAC5B,SAAK,mBAAmB,WAAW,iBAAiB;AACpD,SAAK,uBAAuB,UAAU;AACtC,WAAO;KACL,MAAM;KACN,KAAK;KACL,SAAS,gBAAgB,QAAQ;KACjC;KACD;GAEH,KAAK;AACH,YAAQ,QAAQ;KAAE,SAAS,KAAK;KAAS,MAAM,KAAK;KAAM;AAC1D,SAAK,mBAAmB,WAAW,iBAAiB;AACpD,SAAK,uBAAuB,UAAU;AACtC,WAAO;KACL,MAAM;KACN,KAAK;KACL,SAAS,gBAAgB,QAAQ;KACjC;KACD;;;CAKP,mBACE,WACA,kBACM;AACN,OAAK,eAAe,OAAO,UAAU;AACrC,MAAI,KAAK,sBAAsB,IAAI,iBAAiB,KAAK,UACvD,MAAK,sBAAsB,OAAO,iBAAiB;;CAIvD,kBACE,WACA,QACA,eACA,WACQ;EACR,MAAM,UAAU,OAAO;AACvB,MACE,WAAW,QACX,QAAQ,SAAS,aACjB,wBAAwB,QAAQ,MAAM,UAAU,EAChD;AACA,QAAK,iCAAiC,IACpC,cAAc,WAAW,eAAe,UAAU,EAClD,cACD;AACD,UAAO;;EAGT,MAAM,MAAM,cAAc,WAAW,eAAe,UAAU;EAC9D,MAAM,WAAW,KAAK,iCAAiC,IAAI,IAAI;AAC/D,MAAI,YAAY,KAAM,QAAO;EAE7B,MAAM,YAAY,OAAO;AACzB,OAAK,iCAAiC,IAAI,KAAK,UAAU;AACzD,SAAO;;CAGT,wBACE,WACA,eACA,WACQ;EACR,MAAM,MAAM,cAAc,WAAW,eAAe,UAAU;EAC9D,MAAM,WAAW,KAAK,iCAAiC,IAAI,IAAI;AAC/D,MAAI,YAAY,KAAM,QAAO;AAE7B,OAAK,iCAAiC,IAAI,KAAK,cAAc;AAC7D,SAAO;;CAGT,uBAA+B,WAAyB;EACtD,MAAM,SAAS,GAAG,UAAU;AAC5B,OAAK,MAAM,OAAO,KAAK,iCAAiC,MAAM,CAC5D,KAAI,IAAI,WAAW,OAAO,CACxB,MAAK,iCAAiC,OAAO,IAAI;;;AAKzD,SAAS,cACP,WACA,eACA,WACQ;AACR,QAAO,GAAG,UAAU,IAAI,cAAc,IAAI;;AAG5C,SAAS,wBACP,aACA,UACS;CACT,MAAM,gBAAgB,IAAI,IAAI;EAC5B;EACA;EACA;EACA;EACD,CAAC;CACF,MAAM,sBAAsB,IAAI,IAAI,CAClC,oBACA,yBACD,CAAC;AAEF,QACG,cAAc,IAAI,YAAY,IAAI,cAAc,IAAI,SAAS,IAC7D,oBAAoB,IAAI,YAAY,IAAI,oBAAoB,IAAI,SAAS;;;;;;;AAS9E,IAAa,4BAAb,MAAuC;CACrC,aAAsB,IAAI,kBAAkB;CAC5C,mCAA4B,IAAI,KAA+B;;;;;;CAO/D,QAAQ,OAAoD;EAC1D,MAAM,SAAS,MAAA,UAAgB,QAAQ,MAAM;AAC7C,MAAI,UAAU,KAAM,QAAO,KAAA;AAE3B,UAAQ,OAAO,MAAf;GACE,KAAK,iBAAiB;IACpB,MAAM,YAAY,IAAI,iBAAiB,OAAO,QAAQ;AACtD,cAAU,gBAAgB,OAAO,MAAM;AACvC,cAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,UAAA,gBAAsB,IAAI,OAAO,KAAK,UAAU;AAChD,WAAO;;GAET,KAAK,uBAAuB;IAC1B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D,QACE,aACA,OAAO,MAAM,SAAS,UACtB,UAAU,OAAO,SACjB,OAAO,MAAM,KAEb,WAAU,WAAW,OAAO,MAAM,KAAK;AAEzC,QACE,aACA,OAAO,MAAM,SAAS,eACtB,eAAe,OAAO,SACtB,OAAO,MAAM,UAEb,WAAU,gBAAgB,OAAO,MAAM,UAAU;AAEnD;;GAEF,KAAK,uBAAuB;IAC1B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,CAAC,UAAW,QAAO,KAAA;AACvB,cAAU,gBAAgB,OAAO,MAAM;AACvC,cAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,QAAI,OAAO,MAAM,SAAS,UAAU,UAAU,OAAO,MACnD,WAAU,WAAW,OAAO,MAAM,KAAK;AAEzC,QAAI,OAAO,MAAM,SAAS,eAAe,eAAe,OAAO,MAC7D,WAAU,gBAAgB,OAAO,MAAM,UAAU;AAEnD;;GAEF,KAAK,wBAAwB;IAC3B,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D;;GAEF,KAAK,SAAS;IACZ,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;;AAE7D;;GAEF,KAAK,kBAAkB;IACrB,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,eAAU,SAAS;AACnB,WAAA,gBAAsB,OAAO,OAAO,IAAI;;AAE1C;;GAEF,KAAK,iBAAiB;IACpB,MAAM,YAAY,MAAA,gBAAsB,IAAI,OAAO,IAAI;AACvD,QAAI,WAAW;AACb,eAAU,gBAAgB,OAAO,MAAM;AACvC,eAAU,YAAY,uBAAuB,OAAO,MAAM,CAAC;AAC3D,eAAU,QAAQ;AAClB,WAAA,gBAAsB,OAAO,OAAO,IAAI;;AAE1C"}
|
|
@@ -26,26 +26,41 @@ let _langchain_core_messages = require("@langchain/core/messages");
|
|
|
26
26
|
* currently-observed blocks.
|
|
27
27
|
*/
|
|
28
28
|
function assembledToBaseMessage(input) {
|
|
29
|
-
const { id, role, blocks, toolCallId, usage } = input;
|
|
29
|
+
const { id, role, blocks, namespace, node, runId, toolCallId, metadata, usage, finishMetadata, finishReason } = input;
|
|
30
30
|
const textContent = extractContentString(blocks);
|
|
31
31
|
const toolCalls = extractToolCalls(blocks);
|
|
32
32
|
const toolCallChunks = extractToolCallChunks(blocks);
|
|
33
|
-
const additionalKwargs =
|
|
33
|
+
const additionalKwargs = {
|
|
34
|
+
...namespace != null ? { namespace } : {},
|
|
35
|
+
...node != null ? { node } : {},
|
|
36
|
+
...runId != null ? { run_id: runId } : {},
|
|
37
|
+
...metadata != null ? { metadata } : {},
|
|
38
|
+
...usage != null ? { usage } : {}
|
|
39
|
+
};
|
|
40
|
+
const hasAdditionalKwargs = Object.keys(additionalKwargs).length > 0;
|
|
41
|
+
const responseMetadata = {
|
|
42
|
+
...finishReason != null ? { finish_reason: finishReason } : {},
|
|
43
|
+
...finishMetadata
|
|
44
|
+
};
|
|
34
45
|
switch (role) {
|
|
35
46
|
case "human": return new _langchain_core_messages.HumanMessage({
|
|
36
47
|
...id != null ? { id } : {},
|
|
37
48
|
content: textContent,
|
|
38
|
-
...
|
|
49
|
+
...hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {},
|
|
50
|
+
...Object.keys(responseMetadata).length > 0 ? { response_metadata: responseMetadata } : {}
|
|
39
51
|
});
|
|
40
52
|
case "system": return new _langchain_core_messages.SystemMessage({
|
|
41
53
|
...id != null ? { id } : {},
|
|
42
54
|
content: textContent,
|
|
43
|
-
...
|
|
55
|
+
...hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {},
|
|
56
|
+
...Object.keys(responseMetadata).length > 0 ? { response_metadata: responseMetadata } : {}
|
|
44
57
|
});
|
|
45
58
|
case "tool": return new _langchain_core_messages.ToolMessage({
|
|
46
59
|
...id != null ? { id } : {},
|
|
47
60
|
content: textContent,
|
|
48
|
-
tool_call_id: toolCallId ?? ""
|
|
61
|
+
tool_call_id: toolCallId ?? "",
|
|
62
|
+
...hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {},
|
|
63
|
+
...Object.keys(responseMetadata).length > 0 ? { response_metadata: responseMetadata } : {}
|
|
49
64
|
});
|
|
50
65
|
default: {
|
|
51
66
|
const payload = {
|
|
@@ -53,8 +68,12 @@ function assembledToBaseMessage(input) {
|
|
|
53
68
|
content: cloneContentBlocks(blocks),
|
|
54
69
|
...toolCalls.length > 0 ? { tool_calls: toolCalls } : {},
|
|
55
70
|
...toolCallChunks.length > 0 ? { tool_call_chunks: toolCallChunks } : {},
|
|
56
|
-
...
|
|
57
|
-
|
|
71
|
+
...hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {},
|
|
72
|
+
usage_metadata: normalizeUsage(usage),
|
|
73
|
+
response_metadata: {
|
|
74
|
+
...responseMetadata,
|
|
75
|
+
output_version: "v1"
|
|
76
|
+
}
|
|
58
77
|
};
|
|
59
78
|
return toolCallChunks.length > 0 ? new _langchain_core_messages.AIMessageChunk(payload) : new _langchain_core_messages.AIMessage(payload);
|
|
60
79
|
}
|
|
@@ -69,10 +88,25 @@ function assembledMessageToBaseMessage(assembled, role, extras = {}) {
|
|
|
69
88
|
id: assembled.id,
|
|
70
89
|
role,
|
|
71
90
|
blocks: assembled.blocks,
|
|
91
|
+
namespace: assembled.namespace,
|
|
92
|
+
node: assembled.node,
|
|
93
|
+
runId: assembled.runId,
|
|
72
94
|
toolCallId: extras.toolCallId,
|
|
73
|
-
|
|
95
|
+
metadata: assembled.metadata,
|
|
96
|
+
usage: assembled.usage,
|
|
97
|
+
finishMetadata: assembled.finishMetadata,
|
|
98
|
+
finishReason: assembled.finishReason
|
|
74
99
|
});
|
|
75
100
|
}
|
|
101
|
+
function normalizeUsage(usage) {
|
|
102
|
+
if (usage == null) return void 0;
|
|
103
|
+
return {
|
|
104
|
+
...usage,
|
|
105
|
+
input_tokens: usage.input_tokens ?? 0,
|
|
106
|
+
output_tokens: usage.output_tokens ?? 0,
|
|
107
|
+
total_tokens: usage.total_tokens ?? 0
|
|
108
|
+
};
|
|
109
|
+
}
|
|
76
110
|
function extractContentString(blocks) {
|
|
77
111
|
let out = "";
|
|
78
112
|
for (const block of blocks) if (block.type === "text" && typeof block.text === "string") out += block.text;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"assembled-to-message.cjs","names":["HumanMessage","SystemMessage","ToolMessage","AIMessageChunk","AIMessage"],"sources":["../../src/stream/assembled-to-message.ts"],"sourcesContent":["/**\n * Convert the protocol-native {@link AssembledMessage} (a namespace +\n * content-block bag) into a class instance from\n * `@langchain/core/messages`.\n *\n * The v2 messages channel carries `role` on the `message-start`\n * event, but `MessageAssembler` drops it by the time it produces an\n * `AssembledMessage`. The selector-side caller therefore captures the\n * role separately (via the assembler's per-update hook, see\n * `projections/messages.ts`) and passes it in.\n *\n * The conversion is intentionally lossless in the limit: when the\n * message finishes we emit the same BaseMessage shape that a\n * `values.messages` entry would round-trip to via\n * {@link tryCoerceMessageLikeToMessage}. Mid-stream, partial blocks\n * (e.g. tool_call_chunk) are folded into the same shape so the UI\n * can render an incrementally-completing message.\n */\nimport {\n AIMessage,\n AIMessageChunk,\n HumanMessage,\n SystemMessage,\n ToolMessage,\n type BaseMessage,\n} from \"@langchain/core/messages\";\nimport type { ContentBlock, MessageRole, UsageInfo } from \"@langchain/protocol\";\nimport type { AssembledMessage } from \"../client/stream/messages.js\";\n\nexport type ExtendedMessageRole = MessageRole | \"tool\";\n\nexport interface AssembledToMessageInput {\n /** Stable message id (from `MessageStartData.id`). */\n id?: string;\n /** Author role captured from the `message-start` event. */\n role: ExtendedMessageRole;\n /** Content blocks assembled so far. */\n blocks: ContentBlock[];\n /** Tool-call id a `role: \"tool\"` message is responding to, if any. */\n toolCallId?: string;\n /** Final-token usage (populated on `message-finish`). */\n usage?: UsageInfo;\n}\n\n/**\n * Produce a `BaseMessage` class instance from an in-progress or\n * finished assembled message. Safe to call repeatedly across deltas —\n * each call returns a new instance whose content reflects the\n * currently-observed blocks.\n */\nexport function assembledToBaseMessage(\n input: AssembledToMessageInput\n): BaseMessage {\n const { id, role, blocks, toolCallId, usage } = input;\n const textContent = extractContentString(blocks);\n const toolCalls = extractToolCalls(blocks);\n const toolCallChunks = extractToolCallChunks(blocks);\n const additionalKwargs =\n usage != null ? ({ usage } as Record<string, unknown>) : undefined;\n\n switch (role) {\n case \"human\":\n return new HumanMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n ...(additionalKwargs != null\n ? { additional_kwargs: additionalKwargs }\n : {}),\n });\n case \"system\":\n return new SystemMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n ...(additionalKwargs != null\n ? { additional_kwargs: additionalKwargs }\n : {}),\n });\n case \"tool\":\n return new ToolMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n tool_call_id: toolCallId ?? \"\",\n });\n case \"ai\":\n default: {\n // Use `AIMessageChunk` whenever tool_call_chunks are present —\n // the concrete `AIMessage` class silently DROPS the\n // `tool_call_chunks` field, which would leave mid-stream tool\n // calls invisible to the UI (it sees an AI message with empty\n // content and no tool calls, rendering a blank bubble until\n // `content-block-finish` finally promotes the chunks to\n // finalized `tool_calls`). The chunk class is assignment-compatible\n // with `BaseMessage` and round-trips through the merge logic.\n const payload = {\n ...(id != null ? { id } : {}),\n content: cloneContentBlocks(blocks),\n ...(toolCalls.length > 0 ? { tool_calls: toolCalls } : {}),\n ...(toolCallChunks.length > 0\n ? { tool_call_chunks: toolCallChunks }\n : {}),\n ...(additionalKwargs != null\n ? { additional_kwargs: additionalKwargs }\n : {}),\n response_metadata: { output_version: \"v1\" as const },\n };\n return toolCallChunks.length > 0\n ? new AIMessageChunk(\n payload as ConstructorParameters<typeof AIMessageChunk>[0]\n )\n : new AIMessage(payload as ConstructorParameters<typeof AIMessage>[0]);\n }\n }\n}\n\n/**\n * Convenience: given the raw assembled message + the role captured\n * from `message-start`, produce a `BaseMessage` with the same id.\n */\nexport function assembledMessageToBaseMessage(\n assembled: AssembledMessage,\n role: ExtendedMessageRole,\n extras: { toolCallId?: string } = {}\n): BaseMessage {\n return assembledToBaseMessage({\n id: assembled.id,\n role,\n blocks: assembled.blocks,\n toolCallId: extras.toolCallId,\n usage: assembled.usage,\n });\n}\n\n// ---------- helpers ----------\n\nfunction extractContentString(blocks: ContentBlock[]): string {\n let out = \"\";\n for (const block of blocks) {\n if (\n block.type === \"text\" &&\n typeof (block as { text?: unknown }).text === \"string\"\n ) {\n out += (block as { text: string }).text;\n }\n }\n return out;\n}\n\nfunction cloneContentBlocks(blocks: ContentBlock[]): ContentBlock[] {\n return blocks.map((block) => structuredClone(block));\n}\n\ninterface LooseToolCall {\n id: string;\n name: string;\n args: Record<string, unknown>;\n type: \"tool_call\";\n}\n\nfunction extractToolCalls(blocks: ContentBlock[]): LooseToolCall[] {\n const out: LooseToolCall[] = [];\n for (const block of blocks) {\n if (block.type !== \"tool_call\" && block.type !== \"tool_use\") continue;\n const tc = block as ToolCallLikeBlock;\n out.push({\n id: tc.id ?? \"\",\n name: tc.name ?? \"\",\n args: normalizeToolCallArgs(tc.args ?? tc.input),\n type: \"tool_call\",\n });\n }\n return out;\n}\n\ninterface ToolCallLikeBlock {\n id?: string;\n name?: string;\n args?: unknown;\n input?: unknown;\n}\n\ninterface LooseToolCallChunk {\n id?: string;\n name?: string;\n args?: string;\n index?: number;\n type: \"tool_call_chunk\";\n}\n\nfunction extractToolCallChunks(blocks: ContentBlock[]): LooseToolCallChunk[] {\n const out: LooseToolCallChunk[] = [];\n for (const block of blocks) {\n if (block.type !== \"tool_call_chunk\") continue;\n const tc = block as {\n id?: string;\n name?: string;\n args?: string;\n index?: number;\n };\n out.push({\n id: tc.id,\n name: tc.name,\n args: tc.args,\n index: tc.index,\n type: \"tool_call_chunk\",\n });\n }\n return out;\n}\n\nfunction normalizeToolCallArgs(value: unknown): Record<string, unknown> {\n if (value != null && typeof value === \"object\" && !Array.isArray(value)) {\n return value as Record<string, unknown>;\n }\n if (typeof value === \"string\" && value.length > 0) {\n try {\n const parsed = JSON.parse(value);\n if (\n parsed != null &&\n typeof parsed === \"object\" &&\n !Array.isArray(parsed)\n ) {\n return parsed as Record<string, unknown>;\n }\n } catch {\n // Partial streaming input is represented via tool_call_chunks.\n }\n }\n return {};\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAkDA,SAAgB,uBACd,OACa;CACb,MAAM,EAAE,IAAI,MAAM,QAAQ,YAAY,UAAU;CAChD,MAAM,cAAc,qBAAqB,OAAO;CAChD,MAAM,YAAY,iBAAiB,OAAO;CAC1C,MAAM,iBAAiB,sBAAsB,OAAO;CACpD,MAAM,mBACJ,SAAS,OAAQ,EAAE,OAAO,GAA+B,KAAA;AAE3D,SAAQ,MAAR;EACE,KAAK,QACH,QAAO,IAAIA,yBAAAA,aAAa;GACtB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,GAAI,oBAAoB,OACpB,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;GACP,CAAC;EACJ,KAAK,SACH,QAAO,IAAIC,yBAAAA,cAAc;GACvB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,GAAI,oBAAoB,OACpB,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;GACP,CAAC;EACJ,KAAK,OACH,QAAO,IAAIC,yBAAAA,YAAY;GACrB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,cAAc,cAAc;GAC7B,CAAC;EAEJ,SAAS;GASP,MAAM,UAAU;IACd,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;IAC5B,SAAS,mBAAmB,OAAO;IACnC,GAAI,UAAU,SAAS,IAAI,EAAE,YAAY,WAAW,GAAG,EAAE;IACzD,GAAI,eAAe,SAAS,IACxB,EAAE,kBAAkB,gBAAgB,GACpC,EAAE;IACN,GAAI,oBAAoB,OACpB,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;IACN,mBAAmB,EAAE,gBAAgB,MAAe;IACrD;AACD,UAAO,eAAe,SAAS,IAC3B,IAAIC,yBAAAA,eACF,QACD,GACD,IAAIC,yBAAAA,UAAU,QAAsD;;;;;;;;AAS9E,SAAgB,8BACd,WACA,MACA,SAAkC,EAAE,EACvB;AACb,QAAO,uBAAuB;EAC5B,IAAI,UAAU;EACd;EACA,QAAQ,UAAU;EAClB,YAAY,OAAO;EACnB,OAAO,UAAU;EAClB,CAAC;;AAKJ,SAAS,qBAAqB,QAAgC;CAC5D,IAAI,MAAM;AACV,MAAK,MAAM,SAAS,OAClB,KACE,MAAM,SAAS,UACf,OAAQ,MAA6B,SAAS,SAE9C,QAAQ,MAA2B;AAGvC,QAAO;;AAGT,SAAS,mBAAmB,QAAwC;AAClE,QAAO,OAAO,KAAK,UAAU,gBAAgB,MAAM,CAAC;;AAUtD,SAAS,iBAAiB,QAAyC;CACjE,MAAM,MAAuB,EAAE;AAC/B,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,MAAM,SAAS,eAAe,MAAM,SAAS,WAAY;EAC7D,MAAM,KAAK;AACX,MAAI,KAAK;GACP,IAAI,GAAG,MAAM;GACb,MAAM,GAAG,QAAQ;GACjB,MAAM,sBAAsB,GAAG,QAAQ,GAAG,MAAM;GAChD,MAAM;GACP,CAAC;;AAEJ,QAAO;;AAkBT,SAAS,sBAAsB,QAA8C;CAC3E,MAAM,MAA4B,EAAE;AACpC,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,MAAM,SAAS,kBAAmB;EACtC,MAAM,KAAK;AAMX,MAAI,KAAK;GACP,IAAI,GAAG;GACP,MAAM,GAAG;GACT,MAAM,GAAG;GACT,OAAO,GAAG;GACV,MAAM;GACP,CAAC;;AAEJ,QAAO;;AAGT,SAAS,sBAAsB,OAAyC;AACtE,KAAI,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,CACrE,QAAO;AAET,KAAI,OAAO,UAAU,YAAY,MAAM,SAAS,EAC9C,KAAI;EACF,MAAM,SAAS,KAAK,MAAM,MAAM;AAChC,MACE,UAAU,QACV,OAAO,WAAW,YAClB,CAAC,MAAM,QAAQ,OAAO,CAEtB,QAAO;SAEH;AAIV,QAAO,EAAE"}
|
|
1
|
+
{"version":3,"file":"assembled-to-message.cjs","names":["HumanMessage","SystemMessage","ToolMessage","AIMessageChunk","AIMessage"],"sources":["../../src/stream/assembled-to-message.ts"],"sourcesContent":["/**\n * Convert the protocol-native {@link AssembledMessage} (a namespace +\n * content-block bag) into a class instance from\n * `@langchain/core/messages`.\n *\n * The v2 messages channel carries `role` on the `message-start`\n * event, but `MessageAssembler` drops it by the time it produces an\n * `AssembledMessage`. The selector-side caller therefore captures the\n * role separately (via the assembler's per-update hook, see\n * `projections/messages.ts`) and passes it in.\n *\n * The conversion is intentionally lossless in the limit: when the\n * message finishes we emit the same BaseMessage shape that a\n * `values.messages` entry would round-trip to via\n * {@link tryCoerceMessageLikeToMessage}. Mid-stream, partial blocks\n * (e.g. tool_call_chunk) are folded into the same shape so the UI\n * can render an incrementally-completing message.\n */\nimport {\n AIMessage,\n AIMessageChunk,\n HumanMessage,\n SystemMessage,\n ToolMessage,\n type BaseMessage,\n type UsageMetadata,\n} from \"@langchain/core/messages\";\nimport type { ContentBlock, MessageRole, UsageInfo } from \"@langchain/protocol\";\nimport type { AssembledMessage } from \"../client/stream/messages.js\";\n\nexport type ExtendedMessageRole = MessageRole | \"tool\";\n\nexport interface AssembledToMessageInput {\n /** Stable message id (from `MessageStartData.id`). */\n id?: string;\n /** Author role captured from the `message-start` event. */\n role: ExtendedMessageRole;\n /** Content blocks assembled so far. */\n blocks: ContentBlock[];\n /** Namespace associated with the streamed message. */\n namespace?: readonly string[];\n /** Graph node associated with the streamed message. */\n node?: string;\n runId?: string;\n /** Tool-call id a `role: \"tool\"` message is responding to, if any. */\n toolCallId?: string;\n /** Provider metadata (populated on `message-start`). */\n metadata?: Readonly<Record<string, unknown>>;\n /** Final-token usage (populated on `message-finish`). */\n usage?: UsageInfo;\n /** Response metadata (populated on `message-finish`). */\n finishMetadata?: Readonly<Record<string, unknown>>;\n /** Finish reason (populated on `message-finish`). */\n finishReason?: string;\n}\n\n/**\n * Produce a `BaseMessage` class instance from an in-progress or\n * finished assembled message. Safe to call repeatedly across deltas —\n * each call returns a new instance whose content reflects the\n * currently-observed blocks.\n */\nexport function assembledToBaseMessage(\n input: AssembledToMessageInput\n): BaseMessage {\n const {\n id,\n role,\n blocks,\n namespace,\n node,\n runId,\n toolCallId,\n metadata,\n usage,\n finishMetadata,\n finishReason,\n } = input;\n const textContent = extractContentString(blocks);\n const toolCalls = extractToolCalls(blocks);\n const toolCallChunks = extractToolCallChunks(blocks);\n const additionalKwargs = {\n ...(namespace != null ? { namespace } : {}),\n ...(node != null ? { node } : {}),\n ...(runId != null ? { run_id: runId } : {}),\n ...(metadata != null ? { metadata } : {}),\n ...(usage != null ? { usage } : {}),\n };\n const hasAdditionalKwargs = Object.keys(additionalKwargs).length > 0;\n const responseMetadata = {\n ...(finishReason != null ? { finish_reason: finishReason } : {}),\n ...finishMetadata,\n };\n\n switch (role) {\n case \"human\":\n return new HumanMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n ...(hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {}),\n ...(Object.keys(responseMetadata).length > 0\n ? { response_metadata: responseMetadata }\n : {}),\n });\n case \"system\":\n return new SystemMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n ...(hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {}),\n ...(Object.keys(responseMetadata).length > 0\n ? { response_metadata: responseMetadata }\n : {}),\n });\n case \"tool\":\n return new ToolMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n tool_call_id: toolCallId ?? \"\",\n ...(hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {}),\n ...(Object.keys(responseMetadata).length > 0\n ? { response_metadata: responseMetadata }\n : {}),\n });\n case \"ai\":\n default: {\n // Use `AIMessageChunk` whenever tool_call_chunks are present —\n // the concrete `AIMessage` class silently DROPS the\n // `tool_call_chunks` field, which would leave mid-stream tool\n // calls invisible to the UI (it sees an AI message with empty\n // content and no tool calls, rendering a blank bubble until\n // `content-block-finish` finally promotes the chunks to\n // finalized `tool_calls`). The chunk class is assignment-compatible\n // with `BaseMessage` and round-trips through the merge logic.\n const payload = {\n ...(id != null ? { id } : {}),\n content: cloneContentBlocks(blocks),\n ...(toolCalls.length > 0 ? { tool_calls: toolCalls } : {}),\n ...(toolCallChunks.length > 0\n ? { tool_call_chunks: toolCallChunks }\n : {}),\n ...(hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {}),\n usage_metadata: normalizeUsage(usage),\n response_metadata: {\n ...responseMetadata,\n output_version: \"v1\" as const,\n },\n };\n return toolCallChunks.length > 0\n ? new AIMessageChunk(\n payload as ConstructorParameters<typeof AIMessageChunk>[0]\n )\n : new AIMessage(payload as ConstructorParameters<typeof AIMessage>[0]);\n }\n }\n}\n\n/**\n * Convenience: given the raw assembled message + the role captured\n * from `message-start`, produce a `BaseMessage` with the same id.\n */\nexport function assembledMessageToBaseMessage(\n assembled: AssembledMessage,\n role: ExtendedMessageRole,\n extras: { toolCallId?: string } = {}\n): BaseMessage {\n return assembledToBaseMessage({\n id: assembled.id,\n role,\n blocks: assembled.blocks,\n namespace: assembled.namespace,\n node: assembled.node,\n runId: assembled.runId,\n toolCallId: extras.toolCallId,\n metadata: assembled.metadata,\n usage: assembled.usage,\n finishMetadata: assembled.finishMetadata,\n finishReason: assembled.finishReason,\n });\n}\n\n// ---------- helpers ----------\n\nfunction normalizeUsage(\n usage: UsageInfo | undefined\n): UsageMetadata | undefined {\n if (usage == null) return undefined;\n return {\n ...usage,\n input_tokens: usage.input_tokens ?? 0,\n output_tokens: usage.output_tokens ?? 0,\n total_tokens: usage.total_tokens ?? 0,\n };\n}\n\nfunction extractContentString(blocks: ContentBlock[]): string {\n let out = \"\";\n for (const block of blocks) {\n if (\n block.type === \"text\" &&\n typeof (block as { text?: unknown }).text === \"string\"\n ) {\n out += (block as { text: string }).text;\n }\n }\n return out;\n}\n\nfunction cloneContentBlocks(blocks: ContentBlock[]): ContentBlock[] {\n return blocks.map((block) => structuredClone(block));\n}\n\ninterface LooseToolCall {\n id: string;\n name: string;\n args: Record<string, unknown>;\n type: \"tool_call\";\n}\n\nfunction extractToolCalls(blocks: ContentBlock[]): LooseToolCall[] {\n const out: LooseToolCall[] = [];\n for (const block of blocks) {\n if (block.type !== \"tool_call\" && block.type !== \"tool_use\") continue;\n const tc = block as ToolCallLikeBlock;\n out.push({\n id: tc.id ?? \"\",\n name: tc.name ?? \"\",\n args: normalizeToolCallArgs(tc.args ?? tc.input),\n type: \"tool_call\",\n });\n }\n return out;\n}\n\ninterface ToolCallLikeBlock {\n id?: string;\n name?: string;\n args?: unknown;\n input?: unknown;\n}\n\ninterface LooseToolCallChunk {\n id?: string;\n name?: string;\n args?: string;\n index?: number;\n type: \"tool_call_chunk\";\n}\n\nfunction extractToolCallChunks(blocks: ContentBlock[]): LooseToolCallChunk[] {\n const out: LooseToolCallChunk[] = [];\n for (const block of blocks) {\n if (block.type !== \"tool_call_chunk\") continue;\n const tc = block as {\n id?: string;\n name?: string;\n args?: string;\n index?: number;\n };\n out.push({\n id: tc.id,\n name: tc.name,\n args: tc.args,\n index: tc.index,\n type: \"tool_call_chunk\",\n });\n }\n return out;\n}\n\nfunction normalizeToolCallArgs(value: unknown): Record<string, unknown> {\n if (value != null && typeof value === \"object\" && !Array.isArray(value)) {\n return value as Record<string, unknown>;\n }\n if (typeof value === \"string\" && value.length > 0) {\n try {\n const parsed = JSON.parse(value);\n if (\n parsed != null &&\n typeof parsed === \"object\" &&\n !Array.isArray(parsed)\n ) {\n return parsed as Record<string, unknown>;\n }\n } catch {\n // Partial streaming input is represented via tool_call_chunks.\n }\n }\n return {};\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AA8DA,SAAgB,uBACd,OACa;CACb,MAAM,EACJ,IACA,MACA,QACA,WACA,MACA,OACA,YACA,UACA,OACA,gBACA,iBACE;CACJ,MAAM,cAAc,qBAAqB,OAAO;CAChD,MAAM,YAAY,iBAAiB,OAAO;CAC1C,MAAM,iBAAiB,sBAAsB,OAAO;CACpD,MAAM,mBAAmB;EACvB,GAAI,aAAa,OAAO,EAAE,WAAW,GAAG,EAAE;EAC1C,GAAI,QAAQ,OAAO,EAAE,MAAM,GAAG,EAAE;EAChC,GAAI,SAAS,OAAO,EAAE,QAAQ,OAAO,GAAG,EAAE;EAC1C,GAAI,YAAY,OAAO,EAAE,UAAU,GAAG,EAAE;EACxC,GAAI,SAAS,OAAO,EAAE,OAAO,GAAG,EAAE;EACnC;CACD,MAAM,sBAAsB,OAAO,KAAK,iBAAiB,CAAC,SAAS;CACnE,MAAM,mBAAmB;EACvB,GAAI,gBAAgB,OAAO,EAAE,eAAe,cAAc,GAAG,EAAE;EAC/D,GAAG;EACJ;AAED,SAAQ,MAAR;EACE,KAAK,QACH,QAAO,IAAIA,yBAAAA,aAAa;GACtB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,GAAI,sBAAsB,EAAE,mBAAmB,kBAAkB,GAAG,EAAE;GACtE,GAAI,OAAO,KAAK,iBAAiB,CAAC,SAAS,IACvC,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;GACP,CAAC;EACJ,KAAK,SACH,QAAO,IAAIC,yBAAAA,cAAc;GACvB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,GAAI,sBAAsB,EAAE,mBAAmB,kBAAkB,GAAG,EAAE;GACtE,GAAI,OAAO,KAAK,iBAAiB,CAAC,SAAS,IACvC,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;GACP,CAAC;EACJ,KAAK,OACH,QAAO,IAAIC,yBAAAA,YAAY;GACrB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,cAAc,cAAc;GAC5B,GAAI,sBAAsB,EAAE,mBAAmB,kBAAkB,GAAG,EAAE;GACtE,GAAI,OAAO,KAAK,iBAAiB,CAAC,SAAS,IACvC,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;GACP,CAAC;EAEJ,SAAS;GASP,MAAM,UAAU;IACd,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;IAC5B,SAAS,mBAAmB,OAAO;IACnC,GAAI,UAAU,SAAS,IAAI,EAAE,YAAY,WAAW,GAAG,EAAE;IACzD,GAAI,eAAe,SAAS,IACxB,EAAE,kBAAkB,gBAAgB,GACpC,EAAE;IACN,GAAI,sBAAsB,EAAE,mBAAmB,kBAAkB,GAAG,EAAE;IACtE,gBAAgB,eAAe,MAAM;IACrC,mBAAmB;KACjB,GAAG;KACH,gBAAgB;KACjB;IACF;AACD,UAAO,eAAe,SAAS,IAC3B,IAAIC,yBAAAA,eACF,QACD,GACD,IAAIC,yBAAAA,UAAU,QAAsD;;;;;;;;AAS9E,SAAgB,8BACd,WACA,MACA,SAAkC,EAAE,EACvB;AACb,QAAO,uBAAuB;EAC5B,IAAI,UAAU;EACd;EACA,QAAQ,UAAU;EAClB,WAAW,UAAU;EACrB,MAAM,UAAU;EAChB,OAAO,UAAU;EACjB,YAAY,OAAO;EACnB,UAAU,UAAU;EACpB,OAAO,UAAU;EACjB,gBAAgB,UAAU;EAC1B,cAAc,UAAU;EACzB,CAAC;;AAKJ,SAAS,eACP,OAC2B;AAC3B,KAAI,SAAS,KAAM,QAAO,KAAA;AAC1B,QAAO;EACL,GAAG;EACH,cAAc,MAAM,gBAAgB;EACpC,eAAe,MAAM,iBAAiB;EACtC,cAAc,MAAM,gBAAgB;EACrC;;AAGH,SAAS,qBAAqB,QAAgC;CAC5D,IAAI,MAAM;AACV,MAAK,MAAM,SAAS,OAClB,KACE,MAAM,SAAS,UACf,OAAQ,MAA6B,SAAS,SAE9C,QAAQ,MAA2B;AAGvC,QAAO;;AAGT,SAAS,mBAAmB,QAAwC;AAClE,QAAO,OAAO,KAAK,UAAU,gBAAgB,MAAM,CAAC;;AAUtD,SAAS,iBAAiB,QAAyC;CACjE,MAAM,MAAuB,EAAE;AAC/B,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,MAAM,SAAS,eAAe,MAAM,SAAS,WAAY;EAC7D,MAAM,KAAK;AACX,MAAI,KAAK;GACP,IAAI,GAAG,MAAM;GACb,MAAM,GAAG,QAAQ;GACjB,MAAM,sBAAsB,GAAG,QAAQ,GAAG,MAAM;GAChD,MAAM;GACP,CAAC;;AAEJ,QAAO;;AAkBT,SAAS,sBAAsB,QAA8C;CAC3E,MAAM,MAA4B,EAAE;AACpC,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,MAAM,SAAS,kBAAmB;EACtC,MAAM,KAAK;AAMX,MAAI,KAAK;GACP,IAAI,GAAG;GACP,MAAM,GAAG;GACT,MAAM,GAAG;GACT,OAAO,GAAG;GACV,MAAM;GACP,CAAC;;AAEJ,QAAO;;AAGT,SAAS,sBAAsB,OAAyC;AACtE,KAAI,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,CACrE,QAAO;AAET,KAAI,OAAO,UAAU,YAAY,MAAM,SAAS,EAC9C,KAAI;EACF,MAAM,SAAS,KAAK,MAAM,MAAM;AAChC,MACE,UAAU,QACV,OAAO,WAAW,YAClB,CAAC,MAAM,QAAQ,OAAO,CAEtB,QAAO;SAEH;AAIV,QAAO,EAAE"}
|
|
@@ -11,10 +11,21 @@ interface AssembledToMessageInput {
|
|
|
11
11
|
role: ExtendedMessageRole;
|
|
12
12
|
/** Content blocks assembled so far. */
|
|
13
13
|
blocks: ContentBlock[];
|
|
14
|
+
/** Namespace associated with the streamed message. */
|
|
15
|
+
namespace?: readonly string[];
|
|
16
|
+
/** Graph node associated with the streamed message. */
|
|
17
|
+
node?: string;
|
|
18
|
+
runId?: string;
|
|
14
19
|
/** Tool-call id a `role: "tool"` message is responding to, if any. */
|
|
15
20
|
toolCallId?: string;
|
|
21
|
+
/** Provider metadata (populated on `message-start`). */
|
|
22
|
+
metadata?: Readonly<Record<string, unknown>>;
|
|
16
23
|
/** Final-token usage (populated on `message-finish`). */
|
|
17
24
|
usage?: UsageInfo;
|
|
25
|
+
/** Response metadata (populated on `message-finish`). */
|
|
26
|
+
finishMetadata?: Readonly<Record<string, unknown>>;
|
|
27
|
+
/** Finish reason (populated on `message-finish`). */
|
|
28
|
+
finishReason?: string;
|
|
18
29
|
}
|
|
19
30
|
/**
|
|
20
31
|
* Produce a `BaseMessage` class instance from an in-progress or
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"assembled-to-message.d.cts","names":[],"sources":["../../src/stream/assembled-to-message.ts"],"mappings":";;;;;
|
|
1
|
+
{"version":3,"file":"assembled-to-message.d.cts","names":[],"sources":["../../src/stream/assembled-to-message.ts"],"mappings":";;;;;KA8BY,mBAAA,GAAsB,WAAA;AAAA,UAEjB,uBAAA;EAIf;EAFA,EAAA;EAIA;EAFA,IAAA,EAAM,mBAAA;EAIN;EAFA,MAAA,EAAQ,YAAA;EAKR;EAHA,SAAA;EAOA;EALA,IAAA;EACA,KAAA;EAMA;EAJA,UAAA;EAMA;EAJA,QAAA,GAAW,QAAA,CAAS,MAAA;EAIM;EAF1B,KAAA,GAAQ,SAAA;EAII;EAFZ,cAAA,GAAiB,QAAA,CAAS,MAAA;EAWZ;EATd,YAAA;AAAA;;;;;;;iBASc,sBAAA,CACd,KAAA,EAAO,uBAAA,GACN,WAAA;;;;;iBAgGa,6BAAA,CACd,SAAA,EAAW,gBAAA,EACX,IAAA,EAAM,mBAAA,EACN,MAAA;EAAU,UAAA;AAAA,IACT,WAAA"}
|
|
@@ -11,10 +11,21 @@ interface AssembledToMessageInput {
|
|
|
11
11
|
role: ExtendedMessageRole;
|
|
12
12
|
/** Content blocks assembled so far. */
|
|
13
13
|
blocks: ContentBlock$1[];
|
|
14
|
+
/** Namespace associated with the streamed message. */
|
|
15
|
+
namespace?: readonly string[];
|
|
16
|
+
/** Graph node associated with the streamed message. */
|
|
17
|
+
node?: string;
|
|
18
|
+
runId?: string;
|
|
14
19
|
/** Tool-call id a `role: "tool"` message is responding to, if any. */
|
|
15
20
|
toolCallId?: string;
|
|
21
|
+
/** Provider metadata (populated on `message-start`). */
|
|
22
|
+
metadata?: Readonly<Record<string, unknown>>;
|
|
16
23
|
/** Final-token usage (populated on `message-finish`). */
|
|
17
24
|
usage?: UsageInfo;
|
|
25
|
+
/** Response metadata (populated on `message-finish`). */
|
|
26
|
+
finishMetadata?: Readonly<Record<string, unknown>>;
|
|
27
|
+
/** Finish reason (populated on `message-finish`). */
|
|
28
|
+
finishReason?: string;
|
|
18
29
|
}
|
|
19
30
|
/**
|
|
20
31
|
* Produce a `BaseMessage` class instance from an in-progress or
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"assembled-to-message.d.ts","names":[],"sources":["../../src/stream/assembled-to-message.ts"],"mappings":";;;;;
|
|
1
|
+
{"version":3,"file":"assembled-to-message.d.ts","names":[],"sources":["../../src/stream/assembled-to-message.ts"],"mappings":";;;;;KA8BY,mBAAA,GAAsB,WAAA;AAAA,UAEjB,uBAAA;EAIf;EAFA,EAAA;EAIA;EAFA,IAAA,EAAM,mBAAA;EAIN;EAFA,MAAA,EAAQ,cAAA;EAKR;EAHA,SAAA;EAOA;EALA,IAAA;EACA,KAAA;EAMA;EAJA,UAAA;EAMA;EAJA,QAAA,GAAW,QAAA,CAAS,MAAA;EAIM;EAF1B,KAAA,GAAQ,SAAA;EAII;EAFZ,cAAA,GAAiB,QAAA,CAAS,MAAA;EAWZ;EATd,YAAA;AAAA;;;;;;;iBASc,sBAAA,CACd,KAAA,EAAO,uBAAA,GACN,WAAA;;;;;iBAgGa,6BAAA,CACd,SAAA,EAAW,gBAAA,EACX,IAAA,EAAM,mBAAA,EACN,MAAA;EAAU,UAAA;AAAA,IACT,WAAA"}
|
|
@@ -25,26 +25,41 @@ import { AIMessage, AIMessageChunk, HumanMessage, SystemMessage, ToolMessage } f
|
|
|
25
25
|
* currently-observed blocks.
|
|
26
26
|
*/
|
|
27
27
|
function assembledToBaseMessage(input) {
|
|
28
|
-
const { id, role, blocks, toolCallId, usage } = input;
|
|
28
|
+
const { id, role, blocks, namespace, node, runId, toolCallId, metadata, usage, finishMetadata, finishReason } = input;
|
|
29
29
|
const textContent = extractContentString(blocks);
|
|
30
30
|
const toolCalls = extractToolCalls(blocks);
|
|
31
31
|
const toolCallChunks = extractToolCallChunks(blocks);
|
|
32
|
-
const additionalKwargs =
|
|
32
|
+
const additionalKwargs = {
|
|
33
|
+
...namespace != null ? { namespace } : {},
|
|
34
|
+
...node != null ? { node } : {},
|
|
35
|
+
...runId != null ? { run_id: runId } : {},
|
|
36
|
+
...metadata != null ? { metadata } : {},
|
|
37
|
+
...usage != null ? { usage } : {}
|
|
38
|
+
};
|
|
39
|
+
const hasAdditionalKwargs = Object.keys(additionalKwargs).length > 0;
|
|
40
|
+
const responseMetadata = {
|
|
41
|
+
...finishReason != null ? { finish_reason: finishReason } : {},
|
|
42
|
+
...finishMetadata
|
|
43
|
+
};
|
|
33
44
|
switch (role) {
|
|
34
45
|
case "human": return new HumanMessage({
|
|
35
46
|
...id != null ? { id } : {},
|
|
36
47
|
content: textContent,
|
|
37
|
-
...
|
|
48
|
+
...hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {},
|
|
49
|
+
...Object.keys(responseMetadata).length > 0 ? { response_metadata: responseMetadata } : {}
|
|
38
50
|
});
|
|
39
51
|
case "system": return new SystemMessage({
|
|
40
52
|
...id != null ? { id } : {},
|
|
41
53
|
content: textContent,
|
|
42
|
-
...
|
|
54
|
+
...hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {},
|
|
55
|
+
...Object.keys(responseMetadata).length > 0 ? { response_metadata: responseMetadata } : {}
|
|
43
56
|
});
|
|
44
57
|
case "tool": return new ToolMessage({
|
|
45
58
|
...id != null ? { id } : {},
|
|
46
59
|
content: textContent,
|
|
47
|
-
tool_call_id: toolCallId ?? ""
|
|
60
|
+
tool_call_id: toolCallId ?? "",
|
|
61
|
+
...hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {},
|
|
62
|
+
...Object.keys(responseMetadata).length > 0 ? { response_metadata: responseMetadata } : {}
|
|
48
63
|
});
|
|
49
64
|
default: {
|
|
50
65
|
const payload = {
|
|
@@ -52,8 +67,12 @@ function assembledToBaseMessage(input) {
|
|
|
52
67
|
content: cloneContentBlocks(blocks),
|
|
53
68
|
...toolCalls.length > 0 ? { tool_calls: toolCalls } : {},
|
|
54
69
|
...toolCallChunks.length > 0 ? { tool_call_chunks: toolCallChunks } : {},
|
|
55
|
-
...
|
|
56
|
-
|
|
70
|
+
...hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {},
|
|
71
|
+
usage_metadata: normalizeUsage(usage),
|
|
72
|
+
response_metadata: {
|
|
73
|
+
...responseMetadata,
|
|
74
|
+
output_version: "v1"
|
|
75
|
+
}
|
|
57
76
|
};
|
|
58
77
|
return toolCallChunks.length > 0 ? new AIMessageChunk(payload) : new AIMessage(payload);
|
|
59
78
|
}
|
|
@@ -68,10 +87,25 @@ function assembledMessageToBaseMessage(assembled, role, extras = {}) {
|
|
|
68
87
|
id: assembled.id,
|
|
69
88
|
role,
|
|
70
89
|
blocks: assembled.blocks,
|
|
90
|
+
namespace: assembled.namespace,
|
|
91
|
+
node: assembled.node,
|
|
92
|
+
runId: assembled.runId,
|
|
71
93
|
toolCallId: extras.toolCallId,
|
|
72
|
-
|
|
94
|
+
metadata: assembled.metadata,
|
|
95
|
+
usage: assembled.usage,
|
|
96
|
+
finishMetadata: assembled.finishMetadata,
|
|
97
|
+
finishReason: assembled.finishReason
|
|
73
98
|
});
|
|
74
99
|
}
|
|
100
|
+
function normalizeUsage(usage) {
|
|
101
|
+
if (usage == null) return void 0;
|
|
102
|
+
return {
|
|
103
|
+
...usage,
|
|
104
|
+
input_tokens: usage.input_tokens ?? 0,
|
|
105
|
+
output_tokens: usage.output_tokens ?? 0,
|
|
106
|
+
total_tokens: usage.total_tokens ?? 0
|
|
107
|
+
};
|
|
108
|
+
}
|
|
75
109
|
function extractContentString(blocks) {
|
|
76
110
|
let out = "";
|
|
77
111
|
for (const block of blocks) if (block.type === "text" && typeof block.text === "string") out += block.text;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"assembled-to-message.js","names":[],"sources":["../../src/stream/assembled-to-message.ts"],"sourcesContent":["/**\n * Convert the protocol-native {@link AssembledMessage} (a namespace +\n * content-block bag) into a class instance from\n * `@langchain/core/messages`.\n *\n * The v2 messages channel carries `role` on the `message-start`\n * event, but `MessageAssembler` drops it by the time it produces an\n * `AssembledMessage`. The selector-side caller therefore captures the\n * role separately (via the assembler's per-update hook, see\n * `projections/messages.ts`) and passes it in.\n *\n * The conversion is intentionally lossless in the limit: when the\n * message finishes we emit the same BaseMessage shape that a\n * `values.messages` entry would round-trip to via\n * {@link tryCoerceMessageLikeToMessage}. Mid-stream, partial blocks\n * (e.g. tool_call_chunk) are folded into the same shape so the UI\n * can render an incrementally-completing message.\n */\nimport {\n AIMessage,\n AIMessageChunk,\n HumanMessage,\n SystemMessage,\n ToolMessage,\n type BaseMessage,\n} from \"@langchain/core/messages\";\nimport type { ContentBlock, MessageRole, UsageInfo } from \"@langchain/protocol\";\nimport type { AssembledMessage } from \"../client/stream/messages.js\";\n\nexport type ExtendedMessageRole = MessageRole | \"tool\";\n\nexport interface AssembledToMessageInput {\n /** Stable message id (from `MessageStartData.id`). */\n id?: string;\n /** Author role captured from the `message-start` event. */\n role: ExtendedMessageRole;\n /** Content blocks assembled so far. */\n blocks: ContentBlock[];\n /** Tool-call id a `role: \"tool\"` message is responding to, if any. */\n toolCallId?: string;\n /** Final-token usage (populated on `message-finish`). */\n usage?: UsageInfo;\n}\n\n/**\n * Produce a `BaseMessage` class instance from an in-progress or\n * finished assembled message. Safe to call repeatedly across deltas —\n * each call returns a new instance whose content reflects the\n * currently-observed blocks.\n */\nexport function assembledToBaseMessage(\n input: AssembledToMessageInput\n): BaseMessage {\n const { id, role, blocks, toolCallId, usage } = input;\n const textContent = extractContentString(blocks);\n const toolCalls = extractToolCalls(blocks);\n const toolCallChunks = extractToolCallChunks(blocks);\n const additionalKwargs =\n usage != null ? ({ usage } as Record<string, unknown>) : undefined;\n\n switch (role) {\n case \"human\":\n return new HumanMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n ...(additionalKwargs != null\n ? { additional_kwargs: additionalKwargs }\n : {}),\n });\n case \"system\":\n return new SystemMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n ...(additionalKwargs != null\n ? { additional_kwargs: additionalKwargs }\n : {}),\n });\n case \"tool\":\n return new ToolMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n tool_call_id: toolCallId ?? \"\",\n });\n case \"ai\":\n default: {\n // Use `AIMessageChunk` whenever tool_call_chunks are present —\n // the concrete `AIMessage` class silently DROPS the\n // `tool_call_chunks` field, which would leave mid-stream tool\n // calls invisible to the UI (it sees an AI message with empty\n // content and no tool calls, rendering a blank bubble until\n // `content-block-finish` finally promotes the chunks to\n // finalized `tool_calls`). The chunk class is assignment-compatible\n // with `BaseMessage` and round-trips through the merge logic.\n const payload = {\n ...(id != null ? { id } : {}),\n content: cloneContentBlocks(blocks),\n ...(toolCalls.length > 0 ? { tool_calls: toolCalls } : {}),\n ...(toolCallChunks.length > 0\n ? { tool_call_chunks: toolCallChunks }\n : {}),\n ...(additionalKwargs != null\n ? { additional_kwargs: additionalKwargs }\n : {}),\n response_metadata: { output_version: \"v1\" as const },\n };\n return toolCallChunks.length > 0\n ? new AIMessageChunk(\n payload as ConstructorParameters<typeof AIMessageChunk>[0]\n )\n : new AIMessage(payload as ConstructorParameters<typeof AIMessage>[0]);\n }\n }\n}\n\n/**\n * Convenience: given the raw assembled message + the role captured\n * from `message-start`, produce a `BaseMessage` with the same id.\n */\nexport function assembledMessageToBaseMessage(\n assembled: AssembledMessage,\n role: ExtendedMessageRole,\n extras: { toolCallId?: string } = {}\n): BaseMessage {\n return assembledToBaseMessage({\n id: assembled.id,\n role,\n blocks: assembled.blocks,\n toolCallId: extras.toolCallId,\n usage: assembled.usage,\n });\n}\n\n// ---------- helpers ----------\n\nfunction extractContentString(blocks: ContentBlock[]): string {\n let out = \"\";\n for (const block of blocks) {\n if (\n block.type === \"text\" &&\n typeof (block as { text?: unknown }).text === \"string\"\n ) {\n out += (block as { text: string }).text;\n }\n }\n return out;\n}\n\nfunction cloneContentBlocks(blocks: ContentBlock[]): ContentBlock[] {\n return blocks.map((block) => structuredClone(block));\n}\n\ninterface LooseToolCall {\n id: string;\n name: string;\n args: Record<string, unknown>;\n type: \"tool_call\";\n}\n\nfunction extractToolCalls(blocks: ContentBlock[]): LooseToolCall[] {\n const out: LooseToolCall[] = [];\n for (const block of blocks) {\n if (block.type !== \"tool_call\" && block.type !== \"tool_use\") continue;\n const tc = block as ToolCallLikeBlock;\n out.push({\n id: tc.id ?? \"\",\n name: tc.name ?? \"\",\n args: normalizeToolCallArgs(tc.args ?? tc.input),\n type: \"tool_call\",\n });\n }\n return out;\n}\n\ninterface ToolCallLikeBlock {\n id?: string;\n name?: string;\n args?: unknown;\n input?: unknown;\n}\n\ninterface LooseToolCallChunk {\n id?: string;\n name?: string;\n args?: string;\n index?: number;\n type: \"tool_call_chunk\";\n}\n\nfunction extractToolCallChunks(blocks: ContentBlock[]): LooseToolCallChunk[] {\n const out: LooseToolCallChunk[] = [];\n for (const block of blocks) {\n if (block.type !== \"tool_call_chunk\") continue;\n const tc = block as {\n id?: string;\n name?: string;\n args?: string;\n index?: number;\n };\n out.push({\n id: tc.id,\n name: tc.name,\n args: tc.args,\n index: tc.index,\n type: \"tool_call_chunk\",\n });\n }\n return out;\n}\n\nfunction normalizeToolCallArgs(value: unknown): Record<string, unknown> {\n if (value != null && typeof value === \"object\" && !Array.isArray(value)) {\n return value as Record<string, unknown>;\n }\n if (typeof value === \"string\" && value.length > 0) {\n try {\n const parsed = JSON.parse(value);\n if (\n parsed != null &&\n typeof parsed === \"object\" &&\n !Array.isArray(parsed)\n ) {\n return parsed as Record<string, unknown>;\n }\n } catch {\n // Partial streaming input is represented via tool_call_chunks.\n }\n }\n return {};\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAkDA,SAAgB,uBACd,OACa;CACb,MAAM,EAAE,IAAI,MAAM,QAAQ,YAAY,UAAU;CAChD,MAAM,cAAc,qBAAqB,OAAO;CAChD,MAAM,YAAY,iBAAiB,OAAO;CAC1C,MAAM,iBAAiB,sBAAsB,OAAO;CACpD,MAAM,mBACJ,SAAS,OAAQ,EAAE,OAAO,GAA+B,KAAA;AAE3D,SAAQ,MAAR;EACE,KAAK,QACH,QAAO,IAAI,aAAa;GACtB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,GAAI,oBAAoB,OACpB,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;GACP,CAAC;EACJ,KAAK,SACH,QAAO,IAAI,cAAc;GACvB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,GAAI,oBAAoB,OACpB,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;GACP,CAAC;EACJ,KAAK,OACH,QAAO,IAAI,YAAY;GACrB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,cAAc,cAAc;GAC7B,CAAC;EAEJ,SAAS;GASP,MAAM,UAAU;IACd,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;IAC5B,SAAS,mBAAmB,OAAO;IACnC,GAAI,UAAU,SAAS,IAAI,EAAE,YAAY,WAAW,GAAG,EAAE;IACzD,GAAI,eAAe,SAAS,IACxB,EAAE,kBAAkB,gBAAgB,GACpC,EAAE;IACN,GAAI,oBAAoB,OACpB,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;IACN,mBAAmB,EAAE,gBAAgB,MAAe;IACrD;AACD,UAAO,eAAe,SAAS,IAC3B,IAAI,eACF,QACD,GACD,IAAI,UAAU,QAAsD;;;;;;;;AAS9E,SAAgB,8BACd,WACA,MACA,SAAkC,EAAE,EACvB;AACb,QAAO,uBAAuB;EAC5B,IAAI,UAAU;EACd;EACA,QAAQ,UAAU;EAClB,YAAY,OAAO;EACnB,OAAO,UAAU;EAClB,CAAC;;AAKJ,SAAS,qBAAqB,QAAgC;CAC5D,IAAI,MAAM;AACV,MAAK,MAAM,SAAS,OAClB,KACE,MAAM,SAAS,UACf,OAAQ,MAA6B,SAAS,SAE9C,QAAQ,MAA2B;AAGvC,QAAO;;AAGT,SAAS,mBAAmB,QAAwC;AAClE,QAAO,OAAO,KAAK,UAAU,gBAAgB,MAAM,CAAC;;AAUtD,SAAS,iBAAiB,QAAyC;CACjE,MAAM,MAAuB,EAAE;AAC/B,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,MAAM,SAAS,eAAe,MAAM,SAAS,WAAY;EAC7D,MAAM,KAAK;AACX,MAAI,KAAK;GACP,IAAI,GAAG,MAAM;GACb,MAAM,GAAG,QAAQ;GACjB,MAAM,sBAAsB,GAAG,QAAQ,GAAG,MAAM;GAChD,MAAM;GACP,CAAC;;AAEJ,QAAO;;AAkBT,SAAS,sBAAsB,QAA8C;CAC3E,MAAM,MAA4B,EAAE;AACpC,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,MAAM,SAAS,kBAAmB;EACtC,MAAM,KAAK;AAMX,MAAI,KAAK;GACP,IAAI,GAAG;GACP,MAAM,GAAG;GACT,MAAM,GAAG;GACT,OAAO,GAAG;GACV,MAAM;GACP,CAAC;;AAEJ,QAAO;;AAGT,SAAS,sBAAsB,OAAyC;AACtE,KAAI,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,CACrE,QAAO;AAET,KAAI,OAAO,UAAU,YAAY,MAAM,SAAS,EAC9C,KAAI;EACF,MAAM,SAAS,KAAK,MAAM,MAAM;AAChC,MACE,UAAU,QACV,OAAO,WAAW,YAClB,CAAC,MAAM,QAAQ,OAAO,CAEtB,QAAO;SAEH;AAIV,QAAO,EAAE"}
|
|
1
|
+
{"version":3,"file":"assembled-to-message.js","names":[],"sources":["../../src/stream/assembled-to-message.ts"],"sourcesContent":["/**\n * Convert the protocol-native {@link AssembledMessage} (a namespace +\n * content-block bag) into a class instance from\n * `@langchain/core/messages`.\n *\n * The v2 messages channel carries `role` on the `message-start`\n * event, but `MessageAssembler` drops it by the time it produces an\n * `AssembledMessage`. The selector-side caller therefore captures the\n * role separately (via the assembler's per-update hook, see\n * `projections/messages.ts`) and passes it in.\n *\n * The conversion is intentionally lossless in the limit: when the\n * message finishes we emit the same BaseMessage shape that a\n * `values.messages` entry would round-trip to via\n * {@link tryCoerceMessageLikeToMessage}. Mid-stream, partial blocks\n * (e.g. tool_call_chunk) are folded into the same shape so the UI\n * can render an incrementally-completing message.\n */\nimport {\n AIMessage,\n AIMessageChunk,\n HumanMessage,\n SystemMessage,\n ToolMessage,\n type BaseMessage,\n type UsageMetadata,\n} from \"@langchain/core/messages\";\nimport type { ContentBlock, MessageRole, UsageInfo } from \"@langchain/protocol\";\nimport type { AssembledMessage } from \"../client/stream/messages.js\";\n\nexport type ExtendedMessageRole = MessageRole | \"tool\";\n\nexport interface AssembledToMessageInput {\n /** Stable message id (from `MessageStartData.id`). */\n id?: string;\n /** Author role captured from the `message-start` event. */\n role: ExtendedMessageRole;\n /** Content blocks assembled so far. */\n blocks: ContentBlock[];\n /** Namespace associated with the streamed message. */\n namespace?: readonly string[];\n /** Graph node associated with the streamed message. */\n node?: string;\n runId?: string;\n /** Tool-call id a `role: \"tool\"` message is responding to, if any. */\n toolCallId?: string;\n /** Provider metadata (populated on `message-start`). */\n metadata?: Readonly<Record<string, unknown>>;\n /** Final-token usage (populated on `message-finish`). */\n usage?: UsageInfo;\n /** Response metadata (populated on `message-finish`). */\n finishMetadata?: Readonly<Record<string, unknown>>;\n /** Finish reason (populated on `message-finish`). */\n finishReason?: string;\n}\n\n/**\n * Produce a `BaseMessage` class instance from an in-progress or\n * finished assembled message. Safe to call repeatedly across deltas —\n * each call returns a new instance whose content reflects the\n * currently-observed blocks.\n */\nexport function assembledToBaseMessage(\n input: AssembledToMessageInput\n): BaseMessage {\n const {\n id,\n role,\n blocks,\n namespace,\n node,\n runId,\n toolCallId,\n metadata,\n usage,\n finishMetadata,\n finishReason,\n } = input;\n const textContent = extractContentString(blocks);\n const toolCalls = extractToolCalls(blocks);\n const toolCallChunks = extractToolCallChunks(blocks);\n const additionalKwargs = {\n ...(namespace != null ? { namespace } : {}),\n ...(node != null ? { node } : {}),\n ...(runId != null ? { run_id: runId } : {}),\n ...(metadata != null ? { metadata } : {}),\n ...(usage != null ? { usage } : {}),\n };\n const hasAdditionalKwargs = Object.keys(additionalKwargs).length > 0;\n const responseMetadata = {\n ...(finishReason != null ? { finish_reason: finishReason } : {}),\n ...finishMetadata,\n };\n\n switch (role) {\n case \"human\":\n return new HumanMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n ...(hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {}),\n ...(Object.keys(responseMetadata).length > 0\n ? { response_metadata: responseMetadata }\n : {}),\n });\n case \"system\":\n return new SystemMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n ...(hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {}),\n ...(Object.keys(responseMetadata).length > 0\n ? { response_metadata: responseMetadata }\n : {}),\n });\n case \"tool\":\n return new ToolMessage({\n ...(id != null ? { id } : {}),\n content: textContent,\n tool_call_id: toolCallId ?? \"\",\n ...(hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {}),\n ...(Object.keys(responseMetadata).length > 0\n ? { response_metadata: responseMetadata }\n : {}),\n });\n case \"ai\":\n default: {\n // Use `AIMessageChunk` whenever tool_call_chunks are present —\n // the concrete `AIMessage` class silently DROPS the\n // `tool_call_chunks` field, which would leave mid-stream tool\n // calls invisible to the UI (it sees an AI message with empty\n // content and no tool calls, rendering a blank bubble until\n // `content-block-finish` finally promotes the chunks to\n // finalized `tool_calls`). The chunk class is assignment-compatible\n // with `BaseMessage` and round-trips through the merge logic.\n const payload = {\n ...(id != null ? { id } : {}),\n content: cloneContentBlocks(blocks),\n ...(toolCalls.length > 0 ? { tool_calls: toolCalls } : {}),\n ...(toolCallChunks.length > 0\n ? { tool_call_chunks: toolCallChunks }\n : {}),\n ...(hasAdditionalKwargs ? { additional_kwargs: additionalKwargs } : {}),\n usage_metadata: normalizeUsage(usage),\n response_metadata: {\n ...responseMetadata,\n output_version: \"v1\" as const,\n },\n };\n return toolCallChunks.length > 0\n ? new AIMessageChunk(\n payload as ConstructorParameters<typeof AIMessageChunk>[0]\n )\n : new AIMessage(payload as ConstructorParameters<typeof AIMessage>[0]);\n }\n }\n}\n\n/**\n * Convenience: given the raw assembled message + the role captured\n * from `message-start`, produce a `BaseMessage` with the same id.\n */\nexport function assembledMessageToBaseMessage(\n assembled: AssembledMessage,\n role: ExtendedMessageRole,\n extras: { toolCallId?: string } = {}\n): BaseMessage {\n return assembledToBaseMessage({\n id: assembled.id,\n role,\n blocks: assembled.blocks,\n namespace: assembled.namespace,\n node: assembled.node,\n runId: assembled.runId,\n toolCallId: extras.toolCallId,\n metadata: assembled.metadata,\n usage: assembled.usage,\n finishMetadata: assembled.finishMetadata,\n finishReason: assembled.finishReason,\n });\n}\n\n// ---------- helpers ----------\n\nfunction normalizeUsage(\n usage: UsageInfo | undefined\n): UsageMetadata | undefined {\n if (usage == null) return undefined;\n return {\n ...usage,\n input_tokens: usage.input_tokens ?? 0,\n output_tokens: usage.output_tokens ?? 0,\n total_tokens: usage.total_tokens ?? 0,\n };\n}\n\nfunction extractContentString(blocks: ContentBlock[]): string {\n let out = \"\";\n for (const block of blocks) {\n if (\n block.type === \"text\" &&\n typeof (block as { text?: unknown }).text === \"string\"\n ) {\n out += (block as { text: string }).text;\n }\n }\n return out;\n}\n\nfunction cloneContentBlocks(blocks: ContentBlock[]): ContentBlock[] {\n return blocks.map((block) => structuredClone(block));\n}\n\ninterface LooseToolCall {\n id: string;\n name: string;\n args: Record<string, unknown>;\n type: \"tool_call\";\n}\n\nfunction extractToolCalls(blocks: ContentBlock[]): LooseToolCall[] {\n const out: LooseToolCall[] = [];\n for (const block of blocks) {\n if (block.type !== \"tool_call\" && block.type !== \"tool_use\") continue;\n const tc = block as ToolCallLikeBlock;\n out.push({\n id: tc.id ?? \"\",\n name: tc.name ?? \"\",\n args: normalizeToolCallArgs(tc.args ?? tc.input),\n type: \"tool_call\",\n });\n }\n return out;\n}\n\ninterface ToolCallLikeBlock {\n id?: string;\n name?: string;\n args?: unknown;\n input?: unknown;\n}\n\ninterface LooseToolCallChunk {\n id?: string;\n name?: string;\n args?: string;\n index?: number;\n type: \"tool_call_chunk\";\n}\n\nfunction extractToolCallChunks(blocks: ContentBlock[]): LooseToolCallChunk[] {\n const out: LooseToolCallChunk[] = [];\n for (const block of blocks) {\n if (block.type !== \"tool_call_chunk\") continue;\n const tc = block as {\n id?: string;\n name?: string;\n args?: string;\n index?: number;\n };\n out.push({\n id: tc.id,\n name: tc.name,\n args: tc.args,\n index: tc.index,\n type: \"tool_call_chunk\",\n });\n }\n return out;\n}\n\nfunction normalizeToolCallArgs(value: unknown): Record<string, unknown> {\n if (value != null && typeof value === \"object\" && !Array.isArray(value)) {\n return value as Record<string, unknown>;\n }\n if (typeof value === \"string\" && value.length > 0) {\n try {\n const parsed = JSON.parse(value);\n if (\n parsed != null &&\n typeof parsed === \"object\" &&\n !Array.isArray(parsed)\n ) {\n return parsed as Record<string, unknown>;\n }\n } catch {\n // Partial streaming input is represented via tool_call_chunks.\n }\n }\n return {};\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AA8DA,SAAgB,uBACd,OACa;CACb,MAAM,EACJ,IACA,MACA,QACA,WACA,MACA,OACA,YACA,UACA,OACA,gBACA,iBACE;CACJ,MAAM,cAAc,qBAAqB,OAAO;CAChD,MAAM,YAAY,iBAAiB,OAAO;CAC1C,MAAM,iBAAiB,sBAAsB,OAAO;CACpD,MAAM,mBAAmB;EACvB,GAAI,aAAa,OAAO,EAAE,WAAW,GAAG,EAAE;EAC1C,GAAI,QAAQ,OAAO,EAAE,MAAM,GAAG,EAAE;EAChC,GAAI,SAAS,OAAO,EAAE,QAAQ,OAAO,GAAG,EAAE;EAC1C,GAAI,YAAY,OAAO,EAAE,UAAU,GAAG,EAAE;EACxC,GAAI,SAAS,OAAO,EAAE,OAAO,GAAG,EAAE;EACnC;CACD,MAAM,sBAAsB,OAAO,KAAK,iBAAiB,CAAC,SAAS;CACnE,MAAM,mBAAmB;EACvB,GAAI,gBAAgB,OAAO,EAAE,eAAe,cAAc,GAAG,EAAE;EAC/D,GAAG;EACJ;AAED,SAAQ,MAAR;EACE,KAAK,QACH,QAAO,IAAI,aAAa;GACtB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,GAAI,sBAAsB,EAAE,mBAAmB,kBAAkB,GAAG,EAAE;GACtE,GAAI,OAAO,KAAK,iBAAiB,CAAC,SAAS,IACvC,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;GACP,CAAC;EACJ,KAAK,SACH,QAAO,IAAI,cAAc;GACvB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,GAAI,sBAAsB,EAAE,mBAAmB,kBAAkB,GAAG,EAAE;GACtE,GAAI,OAAO,KAAK,iBAAiB,CAAC,SAAS,IACvC,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;GACP,CAAC;EACJ,KAAK,OACH,QAAO,IAAI,YAAY;GACrB,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;GAC5B,SAAS;GACT,cAAc,cAAc;GAC5B,GAAI,sBAAsB,EAAE,mBAAmB,kBAAkB,GAAG,EAAE;GACtE,GAAI,OAAO,KAAK,iBAAiB,CAAC,SAAS,IACvC,EAAE,mBAAmB,kBAAkB,GACvC,EAAE;GACP,CAAC;EAEJ,SAAS;GASP,MAAM,UAAU;IACd,GAAI,MAAM,OAAO,EAAE,IAAI,GAAG,EAAE;IAC5B,SAAS,mBAAmB,OAAO;IACnC,GAAI,UAAU,SAAS,IAAI,EAAE,YAAY,WAAW,GAAG,EAAE;IACzD,GAAI,eAAe,SAAS,IACxB,EAAE,kBAAkB,gBAAgB,GACpC,EAAE;IACN,GAAI,sBAAsB,EAAE,mBAAmB,kBAAkB,GAAG,EAAE;IACtE,gBAAgB,eAAe,MAAM;IACrC,mBAAmB;KACjB,GAAG;KACH,gBAAgB;KACjB;IACF;AACD,UAAO,eAAe,SAAS,IAC3B,IAAI,eACF,QACD,GACD,IAAI,UAAU,QAAsD;;;;;;;;AAS9E,SAAgB,8BACd,WACA,MACA,SAAkC,EAAE,EACvB;AACb,QAAO,uBAAuB;EAC5B,IAAI,UAAU;EACd;EACA,QAAQ,UAAU;EAClB,WAAW,UAAU;EACrB,MAAM,UAAU;EAChB,OAAO,UAAU;EACjB,YAAY,OAAO;EACnB,UAAU,UAAU;EACpB,OAAO,UAAU;EACjB,gBAAgB,UAAU;EAC1B,cAAc,UAAU;EACzB,CAAC;;AAKJ,SAAS,eACP,OAC2B;AAC3B,KAAI,SAAS,KAAM,QAAO,KAAA;AAC1B,QAAO;EACL,GAAG;EACH,cAAc,MAAM,gBAAgB;EACpC,eAAe,MAAM,iBAAiB;EACtC,cAAc,MAAM,gBAAgB;EACrC;;AAGH,SAAS,qBAAqB,QAAgC;CAC5D,IAAI,MAAM;AACV,MAAK,MAAM,SAAS,OAClB,KACE,MAAM,SAAS,UACf,OAAQ,MAA6B,SAAS,SAE9C,QAAQ,MAA2B;AAGvC,QAAO;;AAGT,SAAS,mBAAmB,QAAwC;AAClE,QAAO,OAAO,KAAK,UAAU,gBAAgB,MAAM,CAAC;;AAUtD,SAAS,iBAAiB,QAAyC;CACjE,MAAM,MAAuB,EAAE;AAC/B,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,MAAM,SAAS,eAAe,MAAM,SAAS,WAAY;EAC7D,MAAM,KAAK;AACX,MAAI,KAAK;GACP,IAAI,GAAG,MAAM;GACb,MAAM,GAAG,QAAQ;GACjB,MAAM,sBAAsB,GAAG,QAAQ,GAAG,MAAM;GAChD,MAAM;GACP,CAAC;;AAEJ,QAAO;;AAkBT,SAAS,sBAAsB,QAA8C;CAC3E,MAAM,MAA4B,EAAE;AACpC,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,MAAM,SAAS,kBAAmB;EACtC,MAAM,KAAK;AAMX,MAAI,KAAK;GACP,IAAI,GAAG;GACP,MAAM,GAAG;GACT,MAAM,GAAG;GACT,OAAO,GAAG;GACV,MAAM;GACP,CAAC;;AAEJ,QAAO;;AAGT,SAAS,sBAAsB,OAAyC;AACtE,KAAI,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,CACrE,QAAO;AAET,KAAI,OAAO,UAAU,YAAY,MAAM,SAAS,EAC9C,KAAI;EACF,MAAM,SAAS,KAAK,MAAM,MAAM;AAChC,MACE,UAAU,QACV,OAAO,WAAW,YAClB,CAAC,MAAM,QAAQ,OAAO,CAEtB,QAAO;SAEH;AAIV,QAAO,EAAE"}
|