@code-yeongyu/senpi 2026.7.20 → 2026.7.22

package/node_modules/@earendil-works/pi-ai/dist/tool-call-middleware/stream-wrapper-shared.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-wrapper-shared.js","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper-shared.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,6BAA6B,EAAE,4BAA4B,EAAE,MAAM,8BAA8B,CAAC;AAC3G,OAAO,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAY3E,SAAS,iBAAiB,CAAC,KAAsD;IAChF,OAAO,KAAK,EAAE,IAAI,KAAK,UAAU,IAAI,aAAa,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ,CAAC;AACtG,CAAC;AAED,SAAS,WAAW,CAAC,KAAY;IAChC,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;AACtE,CAAC;AAED,wEAAwE;AACxE,MAAM,OAAO,uBAAuB;IAYnC,YACC,MAAmC,EACnC,MAAwB,EACxB,OAAO,GAAmC,EAAE;QAZrC,0BAAqB,GAAkB,IAAI,CAAC;QACnC,+BAA0B,GAAG,IAAI,GAAG,EAAyB,CAAC;QAC9D,mCAA8B,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC3D,+BAA0B,GAAG,IAAI,GAAG,EAAkB,CAAC;QACvD,yBAAoB,GAAiC,EAAE,CAAC;QAUxE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,sBAAsB,GAAG,OAAO,CAAC,sBAAsB,KAAK,IAAI,CAAC;QACtE,IAAI,CAAC,OAAO,GAAG,6BAA6B,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACjH,IAAI,CAAC,QAAQ,GAAG,IAAI,wBAAwB,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACjG,CAAC;IAED,IAAI,CAAC,MAAwB;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,4BAA4B,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;IAC5G,CAAC;IAED,gBAAgB,CAAC,UAAsC;QACtD,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,4BAA4B,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACjH,CAAC;IAED,SAAS,CAAC,YAAoB,EAAE,WAAiD;QAChF,IAAI,CAAC,qBAAqB,GAAG,YAAY,CAAC;QAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;QAC/C,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACjC,MAAM,KAAK,GACV,WAAW,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;YAC1F,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjC,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;YAC9D,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QACnE,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QACzD,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1F,OAAO,UAAU,CAAC;IACnB,CAAC;IAED,aAAa,CAAC,YAAoB,EAAE,MAAwB;QAC3D,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,oBAAoB,CAAC,YAAoB,EAAE,KAAa,EAAE,MAAwB;QACjF,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,cAAc,CAAC,YAAoB,EAAE,OAAe,EAAE,MAAwB;QAC7E,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,mBAAmB,CAAC,MAAoC;QACvD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,MAAM,kBAAkB,GAAe,EAAE,CAAC;QAC1C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC5B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACpB,KAAK,MAAM;oBACV,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC1B,MAAM;gBACP,KAAK,gBAAgB;oBACpB,WAAW,GAAG,IAAI,CAAC;oBACnB,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;oBAC1B,MAAM;gBACP,KAAK,gBAAgB;oBACpB,WAAW,GAAG,IAAI,CAAC;oBACnB,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC;oBACvD,MAAM;gBACP,KAAK,cAAc,EAAE,CAAC;oBACrB,WAAW,GAAG,IAAI,CAAC;oBACnB,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;oBAC1C,IAAI,SAAS;wBAAE,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBAClD,MAAM;gBACP,CAAC;gBACD;oBACC,WAAW,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACF,CAAC;QACD,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC;IAC5C,CAAC;IAED,UAAU;QACT,MAAM,iBAAiB,GACtB,IAAI,CAAC,qBAAqB,IAAI,IAAI;YACjC,CAAC,CAAC,IAAI;YACN,CAAC,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC;gBACjE,CAAC,IAAI,CAAC,sBAAsB;oBAC3B,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC;oBACrE,CAAC,CAAC,IAAI,CAAC;gBACR,IAAI,CAAC,CAAC;QACT,IAAI,iBAAiB,IAAI,IAAI,EAAE,CAAC;YAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YACtD,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;oBAChB,IAAI,EAAE,UAAU;oBAChB,YAAY,EAAE,iBAAiB;oBAC/B,OAAO,EAAE,KAAK,CAAC,IAAI;oBACnB,OAAO,EAAE,IAAI,CAAC,OAAO;iBACrB,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QACD,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,yBAAyB;QACxB,KAAK,MAAM,CAAC,YAAY,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACpE,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC;gBAAE,SAAS;YACxC,MAAM,aAAa,GAAa;gBAC/B,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,UAAU,EAAE,IAAI;gBAChB,YAAY,EAAE,0CAA0C;aACxD,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;YACnD,KAAK,MAAM,CAAC,WAAW,EAAE,kBAAkB,CAAC,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAC;gBACjF,IAAI,kBAAkB,KAAK,YAAY;oBAAE,IAAI,CAAC,0BAA0B,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC9F,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1G,CAAC;IACF,CAAC;IAED,QAAQ,CAAC,WAA6B,EAAE,WAAoB;QAC3D,MAAM,YAAY,GAAG,6BAA6B,CACjD,WAAW,EACX,IAAI,CAAC,OAAO,CAAC,OAAO,EACpB,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,sBAAsB,CAC3B,CAAC;QACF,IACC,CAAC,WAAW,IAAI,IAAI,CAAC,2BAA2B,EAAE,CAAC;YACnD,CAAC,YAAY,CAAC,UAAU,KAAK,MAAM,IAAI,YAAY,CAAC,UAAU,KAAK,QAAQ,CAAC,EAC3E,CAAC;YACF,YAAY,CAAC,UAAU,GAAG,SAAS,CAAC;QACrC,CAAC;QACD,OAAO,YAAY,CAAC;IACrB,CAAC;IAED,2BAA2B;QAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC;IACrG,CAAC;IAEO,QAAQ,CAAC,IAAY;QAC5B,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI;YAAE,OAAO;QAC/C,IAAI,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,IAAI,CAAC;QAC3F,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;YAC1B,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YAC3C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;YAC9E,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;QACnF,CAAC;aAAM,CAAC;YACP,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YACjD,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM;gBAAE,OAAO;YACnC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5F,CAAC;IAEO,aAAa,CAAC,KAA6D;QAClF,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI,EAAE,CAAC;YACxC,MAAM,SAAS,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YAClF,MAAM,KAAK,GAAG,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC9E,IACC,SAAS,KAAK,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC;gBAC7C,KAAK,EAAE,IAAI,KAAK,MAAM;gBACtB,KAAK,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBACvB,KAAK,CAAC,aAAa,KAAK,SAAS,EAChC,CAAC;gBACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YAC5B,CAAC;QACF,CAAC;QACD,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;QACjD,MAAM,QAAQ,GAAoB;YACjC,IAAI,EAAE,UAAU;YAChB,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,EAAE;YACb,WAAW,EAAE,EAAE;SACf,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpC,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;QAC/D,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI;YAAE,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;QAC9G,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IACnF,CAAC;IAEO,cAAc,CAAC,WAAmB,EAAE,KAAa;QACxD,MAAM,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACtE,IAAI,YAAY,IAAI,IAAI;YAAE,OAAO;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QACjD,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC;YAAE,OAAO;QACtC,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC;QAC3B,KAAK,CAAC,SAAS,GAAG,kBAAkB,CAA0B,KAAK,CAAC,WAAW,CAAC,CAAC;QACjF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAC1F,CAAC;IAEO,WAAW,CAAC,KAA2D;QAC9E,MAAM,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACtE,IAAI,YAAY,IAAI,IAAI;YAAE,OAAO,SAAS,CAAC;QAC3C,MAAM,aAAa,GAAa;YAC/B,IAAI,EAAE,UAAU;YAChB,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,GAAG,CAAC,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1D,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;SACjF,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;QACnD,IAAI,CAAC,0BAA0B,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QACzG,OAAO,aAAa,CAAC;IACtB,CAAC;CACD","sourcesContent":["import type { AssistantMessage, AssistantMessageEventStream, TextContent, ToolCall } from \"../types.ts\";\nimport type { AssistantMessageDiagnostic } from \"../utils/diagnostics.ts\";\nimport { parseStreamingJson } from \"../utils/json-parse.ts\";\nimport { cloneAssistantMessageMetadata, syncAssistantMessageMetadata } from \"./stream-message-metadata.ts\";\nimport { StreamThinkingProjection } from \"./stream-thinking-projection.ts\";\nimport type { StreamParserEvent } from \"./types.ts\";\n\ntype PartialToolCall = ToolCall & { partialJson: string };\n\nexport type ParserProjectionResult = {\n\treadonly sawToolCall: boolean;\n\treadonly completedToolCalls: readonly ToolCall[];\n};\n\ntype StreamMessageProjectionOptions = { readonly preserveSourceMetadata?: boolean };\n\nfunction isPartialToolCall(block: AssistantMessage[\"content\"][number] | undefined): block is PartialToolCall {\n\treturn block?.type === \"toolCall\" && \"partialJson\" in block && typeof block.partialJson === \"string\";\n}\n\nfunction assertNever(value: never): never {\n\tthrow new Error(`Unexpected parser event: ${JSON.stringify(value)}`);\n}\n\n/** Projects parser events into the canonical outer assistant stream. */\nexport class StreamMessageProjection {\n\treadonly message: AssistantMessage;\n\tprivate readonly stream: AssistantMessageEventStream;\n\tprivate currentInnerTextIndex: number | null = null;\n\tprivate readonly textBlockIndexByInnerIndex = new Map<number, number | null>();\n\tprivate readonly lastTextBlockIndexByInnerIndex = new Map<number, number>();\n\tprivate readonly toolCallIndexByParserIndex = new Map<number, number>();\n\tprivate readonly projectedDiagnostics: AssistantMessageDiagnostic[] = [];\n\tprivate readonly thinking: StreamThinkingProjection;\n\tprivate readonly preserveSourceMetadata: boolean;\n\tprivate source: AssistantMessage;\n\n\tconstructor(\n\t\tstream: AssistantMessageEventStream,\n\t\tsource: AssistantMessage,\n\t\toptions: StreamMessageProjectionOptions = {},\n\t) {\n\t\tthis.stream = stream;\n\t\tthis.source = source;\n\t\tthis.preserveSourceMetadata = options.preserveSourceMetadata === true;\n\t\tthis.message = cloneAssistantMessageMetadata(source, [], this.projectedDiagnostics, this.preserveSourceMetadata);\n\t\tthis.thinking = new StreamThinkingProjection(stream, this.message, this.preserveSourceMetadata);\n\t}\n\n\tsync(source: AssistantMessage): void {\n\t\tthis.source = source;\n\t\tsyncAssistantMessageMetadata(this.message, source, this.projectedDiagnostics, this.preserveSourceMetadata);\n\t}\n\n\tappendDiagnostic(diagnostic: AssistantMessageDiagnostic): void {\n\t\tthis.projectedDiagnostics.push(diagnostic);\n\t\tsyncAssistantMessageMetadata(this.message, this.source, this.projectedDiagnostics, this.preserveSourceMetadata);\n\t}\n\n\tstartText(contentIndex: number, sourceBlock?: AssistantMessage[\"content\"][number]): number {\n\t\tthis.currentInnerTextIndex = contentIndex;\n\t\tconst outerIndex = this.message.content.length;\n\t\tif (this.preserveSourceMetadata) {\n\t\t\tconst block: TextContent =\n\t\t\t\tsourceBlock?.type === \"text\" ? { ...sourceBlock, text: \"\" } : { type: \"text\", text: \"\" };\n\t\t\tthis.message.content.push(block);\n\t\t\tthis.textBlockIndexByInnerIndex.set(contentIndex, outerIndex);\n\t\t\tthis.lastTextBlockIndexByInnerIndex.set(contentIndex, outerIndex);\n\t\t} else {\n\t\t\tthis.textBlockIndexByInnerIndex.set(contentIndex, null);\n\t\t}\n\t\tthis.stream.push({ type: \"text_start\", contentIndex: outerIndex, partial: this.message });\n\t\treturn outerIndex;\n\t}\n\n\tstartThinking(contentIndex: number, source: AssistantMessage): number {\n\t\treturn this.thinking.start(contentIndex, source);\n\t}\n\n\tprojectThinkingDelta(contentIndex: number, delta: string, source: AssistantMessage): void {\n\t\tthis.thinking.delta(contentIndex, delta, source);\n\t}\n\n\tfinishThinking(contentIndex: number, content: string, source: AssistantMessage): void {\n\t\tthis.thinking.end(contentIndex, content, source);\n\t}\n\n\tprojectParserEvents(events: readonly StreamParserEvent[]): ParserProjectionResult {\n\t\tlet sawToolCall = false;\n\t\tconst completedToolCalls: ToolCall[] = [];\n\t\tfor (const event of events) {\n\t\t\tswitch (event.type) {\n\t\t\t\tcase \"text\":\n\t\t\t\t\tthis.pushText(event.text);\n\t\t\t\t\tbreak;\n\t\t\t\tcase \"toolcall_start\":\n\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\tthis.startToolCall(event);\n\t\t\t\t\tbreak;\n\t\t\t\tcase \"toolcall_delta\":\n\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\tthis.updateToolCall(event.index, event.argumentsDelta);\n\t\t\t\t\tbreak;\n\t\t\t\tcase \"toolcall_end\": {\n\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\tconst completed = this.endToolCall(event);\n\t\t\t\t\tif (completed) completedToolCalls.push(completed);\n\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t\tdefault:\n\t\t\t\t\tassertNever(event);\n\t\t\t}\n\t\t}\n\t\treturn { sawToolCall, completedToolCalls };\n\t}\n\n\tfinishText(): void {\n\t\tconst outerContentIndex =\n\t\t\tthis.currentInnerTextIndex == null\n\t\t\t\t? null\n\t\t\t\t: (this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex) ??\n\t\t\t\t\t(this.preserveSourceMetadata\n\t\t\t\t\t\t? this.lastTextBlockIndexByInnerIndex.get(this.currentInnerTextIndex)\n\t\t\t\t\t\t: null) ??\n\t\t\t\t\tnull);\n\t\tif (outerContentIndex != null) {\n\t\t\tconst block = this.message.content[outerContentIndex];\n\t\t\tif (block?.type === \"text\") {\n\t\t\t\tthis.stream.push({\n\t\t\t\t\ttype: \"text_end\",\n\t\t\t\t\tcontentIndex: outerContentIndex,\n\t\t\t\t\tcontent: block.text,\n\t\t\t\t\tpartial: this.message,\n\t\t\t\t});\n\t\t\t}\n\t\t}\n\t\tthis.currentInnerTextIndex = null;\n\t}\n\n\tfinalizeDanglingToolCalls(): void {\n\t\tfor (const [contentIndex, block] of this.message.content.entries()) {\n\t\t\tif (!isPartialToolCall(block)) continue;\n\t\t\tconst finalToolCall: ToolCall = {\n\t\t\t\ttype: \"toolCall\",\n\t\t\t\tid: block.id,\n\t\t\t\tname: block.name,\n\t\t\t\targuments: block.arguments,\n\t\t\t\tincomplete: true,\n\t\t\t\terrorMessage: \"Tool call stream ended before completion\",\n\t\t\t};\n\t\t\tthis.message.content[contentIndex] = finalToolCall;\n\t\t\tfor (const [parserIndex, mappedContentIndex] of this.toolCallIndexByParserIndex) {\n\t\t\t\tif (mappedContentIndex === contentIndex) this.toolCallIndexByParserIndex.delete(parserIndex);\n\t\t\t}\n\t\t\tthis.stream.push({ type: \"toolcall_end\", contentIndex, toolCall: finalToolCall, partial: this.message });\n\t\t}\n\t}\n\n\tfinalize(doneMessage: AssistantMessage, sawToolCall: boolean): AssistantMessage {\n\t\tconst finalMessage = cloneAssistantMessageMetadata(\n\t\t\tdoneMessage,\n\t\t\tthis.message.content,\n\t\t\tthis.projectedDiagnostics,\n\t\t\tthis.preserveSourceMetadata,\n\t\t);\n\t\tif (\n\t\t\t(sawToolCall || this.hasFinalizedToolCallContent()) &&\n\t\t\t(finalMessage.stopReason === \"stop\" || finalMessage.stopReason === \"length\")\n\t\t) {\n\t\t\tfinalMessage.stopReason = \"toolUse\";\n\t\t}\n\t\treturn finalMessage;\n\t}\n\n\thasFinalizedToolCallContent(): boolean {\n\t\treturn this.message.content.some((block) => block.type === \"toolCall\" && !isPartialToolCall(block));\n\t}\n\n\tprivate pushText(text: string): void {\n\t\tif (this.currentInnerTextIndex == null) return;\n\t\tlet contentIndex = this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex) ?? null;\n\t\tif (contentIndex == null) {\n\t\t\tcontentIndex = this.message.content.length;\n\t\t\tthis.message.content.push({ type: \"text\", text });\n\t\t\tthis.textBlockIndexByInnerIndex.set(this.currentInnerTextIndex, contentIndex);\n\t\t\tthis.lastTextBlockIndexByInnerIndex.set(this.currentInnerTextIndex, contentIndex);\n\t\t} else {\n\t\t\tconst block = this.message.content[contentIndex];\n\t\t\tif (block?.type !== \"text\") return;\n\t\t\tblock.text += text;\n\t\t}\n\t\tthis.stream.push({ type: \"text_delta\", contentIndex, delta: text, partial: this.message });\n\t}\n\n\tprivate startToolCall(event: Extract<StreamParserEvent, { type: \"toolcall_start\" }>): void {\n\t\tif (this.currentInnerTextIndex != null) {\n\t\t\tconst textIndex = this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex);\n\t\t\tconst block = textIndex == null ? undefined : this.message.content[textIndex];\n\t\t\tif (\n\t\t\t\ttextIndex === this.message.content.length - 1 &&\n\t\t\t\tblock?.type === \"text\" &&\n\t\t\t\tblock.text.length === 0 &&\n\t\t\t\tblock.textSignature === undefined\n\t\t\t) {\n\t\t\t\tthis.message.content.pop();\n\t\t\t}\n\t\t}\n\t\tconst contentIndex = this.message.content.length;\n\t\tconst toolCall: PartialToolCall = {\n\t\t\ttype: \"toolCall\",\n\t\t\tid: event.id,\n\t\t\tname: event.name,\n\t\t\targuments: {},\n\t\t\tpartialJson: \"\",\n\t\t};\n\t\tthis.message.content.push(toolCall);\n\t\tthis.toolCallIndexByParserIndex.set(event.index, contentIndex);\n\t\tif (this.currentInnerTextIndex != null) this.textBlockIndexByInnerIndex.set(this.currentInnerTextIndex, null);\n\t\tthis.stream.push({ type: \"toolcall_start\", contentIndex, partial: this.message });\n\t}\n\n\tprivate updateToolCall(parserIndex: number, delta: string): void {\n\t\tconst contentIndex = this.toolCallIndexByParserIndex.get(parserIndex);\n\t\tif (contentIndex == null) return;\n\t\tconst block = this.message.content[contentIndex];\n\t\tif (!isPartialToolCall(block)) return;\n\t\tblock.partialJson += delta;\n\t\tblock.arguments = parseStreamingJson<Record<string, unknown>>(block.partialJson);\n\t\tthis.stream.push({ type: \"toolcall_delta\", contentIndex, delta, partial: this.message });\n\t}\n\n\tprivate endToolCall(event: Extract<StreamParserEvent, { type: \"toolcall_end\" }>): ToolCall | undefined {\n\t\tconst contentIndex = this.toolCallIndexByParserIndex.get(event.index);\n\t\tif (contentIndex == null) return undefined;\n\t\tconst finalToolCall: ToolCall = {\n\t\t\ttype: \"toolCall\",\n\t\t\tid: event.id,\n\t\t\tname: event.name,\n\t\t\targuments: event.arguments,\n\t\t\t...(event.incomplete === true ? { incomplete: true } : {}),\n\t\t\t...(event.errorMessage === undefined ? {} : { errorMessage: event.errorMessage }),\n\t\t};\n\t\tthis.message.content[contentIndex] = finalToolCall;\n\t\tthis.toolCallIndexByParserIndex.delete(event.index);\n\t\tthis.stream.push({ type: \"toolcall_end\", contentIndex, toolCall: finalToolCall, partial: this.message });\n\t\treturn finalToolCall;\n\t}\n}\n"]}

package/node_modules/@earendil-works/pi-ai/dist/tool-call-middleware/stream-wrapper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stream-wrapper.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAEX,2BAA2B,EAE3B,IAAI,EAGJ,MAAM,aAAa,CAAC;AAGrB,OAAO,KAAK,EAAqB,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAkStE,wBAAgB,gCAAgC,CAC/C,WAAW,EAAE,2BAA2B,EACxC,QAAQ,EAAE,gBAAgB,EAC1B,KAAK,EAAE,IAAI,EAAE,GACX,2BAA2B,CAoN7B"}
+{"version":3,"file":"stream-wrapper.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAoB,2BAA2B,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAGvF,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,wBAAgB,gCAAgC,CAC/C,WAAW,EAAE,2BAA2B,EACxC,QAAQ,EAAE,gBAAgB,EAC1B,KAAK,EAAE,IAAI,EAAE,GACX,2BAA2B,CAkH7B"}

package/node_modules/@earendil-works/pi-ai/dist/tool-call-middleware/stream-wrapper.js

@@ -1,394 +1,118 @@
 import { AssistantMessageEventStream as AssistantMessageEventStreamImpl } from "../utils/event-stream.js";
-import { parseStreamingJson } from "../utils/json-parse.js";
-function isPartialToolCall(block) {
-    return block?.type === "toolCall" && typeof block.partialJson === "string";
-}
-function cloneUsage(usage) {
-    return {
-        input: usage.input,
-        output: usage.output,
-        cacheRead: usage.cacheRead,
-        cacheWrite: usage.cacheWrite,
-        totalTokens: usage.totalTokens,
-        cost: {
-            input: usage.cost.input,
-            output: usage.cost.output,
-            cacheRead: usage.cost.cacheRead,
-            cacheWrite: usage.cost.cacheWrite,
-            total: usage.cost.total,
-        },
-    };
-}
-function createOuterMessage(message) {
-    return {
-        role: "assistant",
-        api: message.api,
-        provider: message.provider,
-        model: message.model,
-        responseId: message.responseId,
-        content: [],
-        usage: cloneUsage(message.usage),
-        stopReason: message.stopReason,
-        errorMessage: message.errorMessage,
-        timestamp: message.timestamp,
-    };
-}
-function syncOuterMetadata(target, source) {
-    target.api = source.api;
-    target.provider = source.provider;
-    target.model = source.model;
-    target.responseId = source.responseId;
-    target.usage = cloneUsage(source.usage);
-    target.stopReason = source.stopReason;
-    target.errorMessage = source.errorMessage;
-    target.timestamp = source.timestamp;
-}
-function pushParserEvent(parserEvent, outerStream, outerMessage, textBlockIndexByInnerIndex, toolCallIndexByParserIndex, currentInnerTextIndex) {
-    switch (parserEvent.type) {
-        case "text": {
-            if (currentInnerTextIndex == null) {
-                return;
-            }
-            let textBlockIndex = textBlockIndexByInnerIndex.get(currentInnerTextIndex) ?? null;
-            if (textBlockIndex == null) {
-                textBlockIndex = outerMessage.content.length;
-                outerMessage.content.push({ type: "text", text: parserEvent.text });
-                textBlockIndexByInnerIndex.set(currentInnerTextIndex, textBlockIndex);
-            }
-            else {
-                const block = outerMessage.content[textBlockIndex];
-                if (block?.type !== "text") {
-                    return;
-                }
-                block.text += parserEvent.text;
-            }
-            outerStream.push({
-                type: "text_delta",
-                contentIndex: textBlockIndex,
-                delta: parserEvent.text,
-                partial: outerMessage,
-            });
-            return;
-        }
-        case "toolcall_start": {
-            const contentIndex = outerMessage.content.length;
-            const toolCall = {
-                type: "toolCall",
-                id: parserEvent.id,
-                name: parserEvent.name,
-                arguments: {},
-                partialJson: "",
-            };
-            outerMessage.content.push(toolCall);
-            toolCallIndexByParserIndex.set(parserEvent.index, contentIndex);
-            if (currentInnerTextIndex != null) {
-                textBlockIndexByInnerIndex.set(currentInnerTextIndex, null);
-            }
-            outerStream.push({
-                type: "toolcall_start",
-                contentIndex,
-                partial: outerMessage,
-            });
-            return;
-        }
-        case "toolcall_delta": {
-            const contentIndex = toolCallIndexByParserIndex.get(parserEvent.index);
-            if (contentIndex == null) {
-                return;
-            }
-            const block = outerMessage.content[contentIndex];
-            if (!isPartialToolCall(block)) {
-                return;
-            }
-            block.partialJson += parserEvent.argumentsDelta;
-            block.arguments = parseStreamingJson(block.partialJson);
-            outerStream.push({
-                type: "toolcall_delta",
-                contentIndex,
-                delta: parserEvent.argumentsDelta,
-                partial: outerMessage,
-            });
-            return;
-        }
-        case "toolcall_end": {
-            const contentIndex = toolCallIndexByParserIndex.get(parserEvent.index);
-            if (contentIndex == null) {
-                return;
-            }
-            const block = outerMessage.content[contentIndex];
-            if (block?.type !== "toolCall") {
-                return;
-            }
-            const finalToolCall = {
-                type: "toolCall",
-                id: parserEvent.id,
-                name: parserEvent.name,
-                arguments: parserEvent.arguments,
-                ...(parserEvent.incomplete === true ? { incomplete: true } : {}),
-                ...(parserEvent.errorMessage === undefined ? {} : { errorMessage: parserEvent.errorMessage }),
-            };
-            outerMessage.content[contentIndex] = finalToolCall;
-            toolCallIndexByParserIndex.delete(parserEvent.index);
-            outerStream.push({
-                type: "toolcall_end",
-                contentIndex,
-                toolCall: finalToolCall,
-                partial: outerMessage,
-            });
-        }
-    }
-}
-function processParserEvents(parserEvents, outerStream, outerMessage, textBlockIndexByInnerIndex, toolCallIndexByParserIndex, currentInnerTextIndex) {
-    let emittedToolCall = false;
-    for (const parserEvent of parserEvents) {
-        if (parserEvent.type === "toolcall_start" ||
-            parserEvent.type === "toolcall_delta" ||
-            parserEvent.type === "toolcall_end") {
-            emittedToolCall = true;
-        }
-        pushParserEvent(parserEvent, outerStream, outerMessage, textBlockIndexByInnerIndex, toolCallIndexByParserIndex, currentInnerTextIndex);
-    }
-    return emittedToolCall;
-}
-function finalizeMessage(outerMessage, doneMessage, sawToolCall) {
-    const finalMessage = {
-        ...createOuterMessage(doneMessage),
-        content: outerMessage.content,
-    };
-    if ((sawToolCall || hasFinalizedToolCallContent(finalMessage)) &&
-        (finalMessage.stopReason === "stop" || finalMessage.stopReason === "length")) {
-        finalMessage.stopReason = "toolUse";
-    }
-    return finalMessage;
-}
-function hasFinalizedToolCallContent(message) {
-    return message.content.some((block) => {
-        const hasFinalized = block.type === "toolCall" && typeof block.partialJson !== "string";
-        return hasFinalized;
-    });
-}
-function finalizeDanglingPartialToolCalls(outerStream, outerMessage, toolCallIndexByParserIndex) {
-    for (const [contentIndex, block] of outerMessage.content.entries()) {
-        if (!isPartialToolCall(block)) {
-            continue;
-        }
-        const finalToolCall = {
-            type: "toolCall",
-            id: block.id,
-            name: block.name,
-            arguments: block.arguments,
-            incomplete: true,
-            errorMessage: "Tool call stream ended before completion",
-        };
-        outerMessage.content[contentIndex] = finalToolCall;
-        for (const [parserIndex, mappedContentIndex] of toolCallIndexByParserIndex) {
-            if (mappedContentIndex === contentIndex) {
-                toolCallIndexByParserIndex.delete(parserIndex);
-            }
-        }
-        outerStream.push({
-            type: "toolcall_end",
-            contentIndex,
-            toolCall: finalToolCall,
-            partial: outerMessage,
-        });
-    }
-}
-function finalizePendingTextBlock(outerStream, outerMessage, parser, textBlockIndexByInnerIndex, toolCallIndexByParserIndex, currentInnerTextIndex, sawToolCall) {
-    const nextSawToolCall = processParserEvents(parser.finish(), outerStream, outerMessage, textBlockIndexByInnerIndex, toolCallIndexByParserIndex, currentInnerTextIndex) || sawToolCall;
-    const outerContentIndex = currentInnerTextIndex == null ? null : (textBlockIndexByInnerIndex.get(currentInnerTextIndex) ?? null);
-    if (outerContentIndex != null) {
-        const outerBlock = outerMessage.content[outerContentIndex];
-        if (outerBlock?.type === "text") {
-            outerStream.push({
-                type: "text_end",
-                contentIndex: outerContentIndex,
-                content: outerBlock.text,
-                partial: outerMessage,
-            });
-        }
-    }
-    return {
-        currentInnerTextIndex: null,
-        sawToolCall: nextSawToolCall,
-    };
-}
+import { StreamMessageProjection } from "./stream-wrapper-shared.js";
 export function wrapStreamWithToolCallMiddleware(innerStream, protocol, tools) {
     const outerStream = new AssistantMessageEventStreamImpl();
     const parser = protocol.createStreamParser(tools);
     void (async () => {
-        let outerMessage = null;
-        let currentInnerTextIndex = null;
+        let projection = null;
         let sawToolCall = false;
         let parserHasPendingInput = false;
-        const textBlockIndexByInnerIndex = new Map();
-        const thinkingBlockIndexByInnerIndex = new Map();
-        const toolCallIndexByParserIndex = new Map();
-        const flushParserAtTermination = () => {
-            if (!outerMessage || !parserHasPendingInput) {
+        const flushParser = () => {
+            if (!projection || !parserHasPendingInput)
                 return;
-            }
-            ({ currentInnerTextIndex, sawToolCall } = finalizePendingTextBlock(outerStream, outerMessage, parser, textBlockIndexByInnerIndex, toolCallIndexByParserIndex, currentInnerTextIndex, sawToolCall));
+            const result = projection.projectParserEvents(parser.finish());
+            sawToolCall = sawToolCall || result.sawToolCall;
+            projection.finishText();
             parserHasPendingInput = false;
         };
-        const finalizeDanglingToolCallsAtTermination = () => {
-            if (outerMessage) {
-                finalizeDanglingPartialToolCalls(outerStream, outerMessage, toolCallIndexByParserIndex);
-            }
-        };
         try {
             for await (const event of innerStream) {
                 switch (event.type) {
-                    case "start": {
-                        outerMessage = createOuterMessage(event.partial);
-                        outerStream.push({ type: "start", partial: outerMessage });
+                    case "start":
+                        projection = new StreamMessageProjection(outerStream, event.partial);
+                        outerStream.push({ type: "start", partial: projection.message });
                         break;
-                    }
-                    case "text_start": {
-                        if (outerMessage) {
-                            syncOuterMetadata(outerMessage, event.partial);
-                        }
-                        currentInnerTextIndex = event.contentIndex;
-                        textBlockIndexByInnerIndex.set(event.contentIndex, null);
-                        if (outerMessage) {
-                            outerStream.push({ ...event, partial: outerMessage });
-                        }
+                    case "text_start":
+                        if (!projection)
+                            break;
+                        projection.sync(event.partial);
+                        projection.startText(event.contentIndex, event.partial.content[event.contentIndex]);
                         break;
-                    }
                     case "text_delta": {
-                        if (!outerMessage) {
+                        if (!projection)
                             break;
-                        }
-                        syncOuterMetadata(outerMessage, event.partial);
-                        if (event.delta.length > 0) {
+                        projection.sync(event.partial);
+                        if (event.delta.length > 0)
                             parserHasPendingInput = true;
-                        }
-                        sawToolCall =
-                            processParserEvents(parser.feed(event.delta), outerStream, outerMessage, textBlockIndexByInnerIndex, toolCallIndexByParserIndex, currentInnerTextIndex) || sawToolCall;
+                        const result = projection.projectParserEvents(parser.feed(event.delta));
+                        sawToolCall = sawToolCall || result.sawToolCall;
                         break;
                     }
-                    case "text_end": {
-                        if (!outerMessage) {
+                    case "text_end":
+                        if (!projection)
                             break;
-                        }
-                        syncOuterMetadata(outerMessage, event.partial);
-                        if (parserHasPendingInput) {
-                            ({ currentInnerTextIndex, sawToolCall } = finalizePendingTextBlock(outerStream, outerMessage, parser, textBlockIndexByInnerIndex, toolCallIndexByParserIndex, currentInnerTextIndex, sawToolCall));
-                            parserHasPendingInput = false;
-                        }
+                        projection.sync(event.partial);
+                        flushParser();
                         break;
-                    }
-                    case "thinking_start": {
-                        if (!outerMessage) {
+                    case "thinking_start":
+                        if (!projection)
                             break;
-                        }
-                        syncOuterMetadata(outerMessage, event.partial);
-                        const contentIndex = outerMessage.content.length;
-                        const thinkingBlock = { type: "thinking", thinking: "" };
-                        outerMessage.content.push(thinkingBlock);
-                        thinkingBlockIndexByInnerIndex.set(event.contentIndex, contentIndex);
-                        outerStream.push({ type: "thinking_start", contentIndex, partial: outerMessage });
+                        projection.sync(event.partial);
+                        projection.startThinking(event.contentIndex, event.partial);
                         break;
-                    }
-                    case "thinking_delta": {
-                        if (!outerMessage) {
-                            break;
-                        }
-                        syncOuterMetadata(outerMessage, event.partial);
-                        const contentIndex = thinkingBlockIndexByInnerIndex.get(event.contentIndex);
-                        if (contentIndex == null) {
-                            break;
-                        }
-                        const block = outerMessage.content[contentIndex];
-                        if (block?.type !== "thinking") {
+                    case "thinking_delta":
+                        if (!projection)
                             break;
-                        }
-                        block.thinking += event.delta;
-                        outerStream.push({
-                            type: "thinking_delta",
-                            contentIndex,
-                            delta: event.delta,
-                            partial: outerMessage,
-                        });
+                        projection.sync(event.partial);
+                        projection.projectThinkingDelta(event.contentIndex, event.delta, event.partial);
                         break;
-                    }
-                    case "thinking_end": {
-                        if (!outerMessage) {
-                            break;
-                        }
-                        syncOuterMetadata(outerMessage, event.partial);
-                        const contentIndex = thinkingBlockIndexByInnerIndex.get(event.contentIndex);
-                        if (contentIndex == null) {
-                            break;
-                        }
-                        const block = outerMessage.content[contentIndex];
-                        if (block?.type !== "thinking") {
+                    case "thinking_end":
+                        if (!projection)
                             break;
-                        }
-                        block.thinking = event.content;
-                        outerStream.push({
-                            type: "thinking_end",
-                            contentIndex,
-                            content: event.content,
-                            partial: outerMessage,
-                        });
+                        projection.sync(event.partial);
+                        projection.finishThinking(event.contentIndex, event.content, event.partial);
                         break;
-                    }
                     case "done": {
-                        if (!outerMessage) {
-                            outerMessage = createOuterMessage(event.message);
-                        }
-                        flushParserAtTermination();
-                        finalizeDanglingToolCallsAtTermination();
-                        const finalMessage = finalizeMessage(outerMessage, event.message, sawToolCall);
-                        const finalizedToolCall = sawToolCall || hasFinalizedToolCallContent(finalMessage);
-                        const finalReason = finalizedToolCall && (event.reason === "stop" || event.reason === "length")
-                            ? "toolUse"
-                            : event.reason;
-                        outerStream.push({ type: "done", reason: finalReason, message: finalMessage });
+                        projection ??= new StreamMessageProjection(outerStream, event.message);
+                        flushParser();
+                        projection.finalizeDanglingToolCalls();
+                        const message = projection.finalize(event.message, sawToolCall);
+                        const recovered = sawToolCall || projection.hasFinalizedToolCallContent();
+                        const reason = recovered && (event.reason === "stop" || event.reason === "length") ? "toolUse" : event.reason;
+                        outerStream.push({ type: "done", reason, message });
                         outerStream.end();
                         return;
                     }
                     case "error": {
-                        if (!outerMessage) {
+                        if (!projection) {
                             outerStream.push(event);
                             outerStream.end(event.error);
                             return;
                         }
-                        syncOuterMetadata(outerMessage, event.error);
-                        flushParserAtTermination();
-                        finalizeDanglingToolCallsAtTermination();
-                        const finalErrorMessage = finalizeMessage(outerMessage, event.error, sawToolCall);
-                        if (hasFinalizedToolCallContent(finalErrorMessage)) {
-                            const recoveredMessage = {
-                                ...finalErrorMessage,
-                                stopReason: "toolUse",
-                            };
-                            outerStream.push({ type: "done", reason: "toolUse", message: recoveredMessage });
-                            outerStream.end(recoveredMessage);
+                        projection.sync(event.error);
+                        flushParser();
+                        projection.finalizeDanglingToolCalls();
+                        const message = projection.finalize(event.error, sawToolCall);
+                        if (projection.hasFinalizedToolCallContent()) {
+                            const recovered = { ...message, stopReason: "toolUse" };
+                            outerStream.push({ type: "done", reason: "toolUse", message: recovered });
+                            outerStream.end(recovered);
                             return;
                         }
-                        outerStream.push({ type: "error", reason: event.reason, error: finalErrorMessage });
-                        outerStream.end(finalErrorMessage);
+                        outerStream.push({ type: "error", reason: event.reason, error: message });
+                        outerStream.end(message);
                         return;
                     }
+                    case "toolcall_start":
+                    case "toolcall_delta":
+                    case "toolcall_end":
+                        break;
                 }
             }
-            const finalInnerMessage = await innerStream.result();
-            if (!outerMessage) {
-                outerMessage = createOuterMessage(finalInnerMessage);
-            }
-            flushParserAtTermination();
-            finalizeDanglingToolCallsAtTermination();
-            outerStream.end(finalizeMessage(outerMessage, finalInnerMessage, sawToolCall));
+            const innerMessage = await innerStream.result();
+            projection ??= new StreamMessageProjection(outerStream, innerMessage);
+            flushParser();
+            projection.finalizeDanglingToolCalls();
+            outerStream.end(projection.finalize(innerMessage, sawToolCall));
         }
         catch (error) {
-            flushParserAtTermination();
-            finalizeDanglingToolCallsAtTermination();
-            const fallbackMessage = outerMessage ?? createOuterMessage(await innerStream.result());
-            fallbackMessage.stopReason = "error";
-            fallbackMessage.errorMessage = error instanceof Error ? error.message : JSON.stringify(error);
-            outerStream.push({ type: "error", reason: "error", error: fallbackMessage });
+            flushParser();
+            projection ??= new StreamMessageProjection(outerStream, await innerStream.result());
+            projection.finalizeDanglingToolCalls();
+            const fallback = projection.message;
+            fallback.stopReason = "error";
+            fallback.errorMessage = error instanceof Error ? error.message : JSON.stringify(error);
+            outerStream.push({ type: "error", reason: "error", error: fallback });
             outerStream.end();
         }
     })();

package/node_modules/@earendil-works/pi-ai/dist/tool-call-middleware/stream-wrapper.js.map

@@ -1 +1 @@
-{"version":3,"file":"stream-wrapper.js","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,2BAA2B,IAAI,+BAA+B,EAAE,MAAM,0BAA0B,CAAC;AAC1G,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAO5D,SAAS,iBAAiB,CAAC,KAAsD;IAChF,OAAO,KAAK,EAAE,IAAI,KAAK,UAAU,IAAI,OAAQ,KAAyB,CAAC,WAAW,KAAK,QAAQ,CAAC;AACjG,CAAC;AAED,SAAS,UAAU,CAAC,KAAY;IAC/B,OAAO;QACN,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,UAAU,EAAE,KAAK,CAAC,UAAU;QAC5B,WAAW,EAAE,KAAK,CAAC,WAAW;QAC9B,IAAI,EAAE;YACL,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK;YACvB,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM;YACzB,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS;YAC/B,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU;YACjC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK;SACvB;KACD,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,OAAyB;IACpD,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,GAAG,EAAE,OAAO,CAAC,GAAG;QAChB,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,OAAO,EAAE,EAAE;QACX,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC;QAChC,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,YAAY,EAAE,OAAO,CAAC,YAAY;QAClC,SAAS,EAAE,OAAO,CAAC,SAAS;KAC5B,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAwB,EAAE,MAAwB;IAC5E,MAAM,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;IACxB,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;IAClC,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IAC5B,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;IACtC,MAAM,CAAC,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACxC,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;IACtC,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;IAC1C,MAAM,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;AACrC,CAAC;AAED,SAAS,eAAe,CACvB,WAA8B,EAC9B,WAAwC,EACxC,YAA8B,EAC9B,0BAAsD,EACtD,0BAA+C,EAC/C,qBAAoC;IAEpC,QAAQ,WAAW,CAAC,IAAI,EAAE,CAAC;QAC1B,KAAK,MAAM,EAAE,CAAC;YACb,IAAI,qBAAqB,IAAI,IAAI,EAAE,CAAC;gBACnC,OAAO;YACR,CAAC;YAED,IAAI,cAAc,GAAG,0BAA0B,CAAC,GAAG,CAAC,qBAAqB,CAAC,IAAI,IAAI,CAAC;YACnF,IAAI,cAAc,IAAI,IAAI,EAAE,CAAC;gBAC5B,cAAc,GAAG,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC;gBAC7C,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC;gBACpE,0BAA0B,CAAC,GAAG,CAAC,qBAAqB,EAAE,cAAc,CAAC,CAAC;YACvE,CAAC;iBAAM,CAAC;gBACP,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;gBACnD,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC5B,OAAO;gBACR,CAAC;gBACD,KAAK,CAAC,IAAI,IAAI,WAAW,CAAC,IAAI,CAAC;YAChC,CAAC;YAED,WAAW,CAAC,IAAI,CAAC;gBAChB,IAAI,EAAE,YAAY;gBAClB,YAAY,EAAE,cAAc;gBAC5B,KAAK,EAAE,WAAW,CAAC,IAAI;gBACvB,OAAO,EAAE,YAAY;aACrB,CAAC,CAAC;YACH,OAAO;QACR,CAAC;QACD,KAAK,gBAAgB,EAAE,CAAC;YACvB,MAAM,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC;YACjD,MAAM,QAAQ,GAAoB;gBACjC,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,WAAW,CAAC,EAAE;gBAClB,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,SAAS,EAAE,EAAE;gBACb,WAAW,EAAE,EAAE;aACf,CAAC;YACF,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACpC,0BAA0B,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;YAChE,IAAI,qBAAqB,IAAI,IAAI,EAAE,CAAC;gBACnC,0BAA0B,CAAC,GAAG,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;YAC7D,CAAC;YACD,WAAW,CAAC,IAAI,CAAC;gBAChB,IAAI,EAAE,gBAAgB;gBACtB,YAAY;gBACZ,OAAO,EAAE,YAAY;aACrB,CAAC,CAAC;YACH,OAAO;QACR,CAAC;QACD,KAAK,gBAAgB,EAAE,CAAC;YACvB,MAAM,YAAY,GAAG,0BAA0B,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACvE,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;gBAC1B,OAAO;YACR,CAAC;YAED,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YACjD,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC/B,OAAO;YACR,CAAC;YAED,KAAK,CAAC,WAAW,IAAI,WAAW,CAAC,cAAc,CAAC;YAChD,KAAK,CAAC,SAAS,GAAG,kBAAkB,CAA0B,KAAK,CAAC,WAAW,CAAC,CAAC;YACjF,WAAW,CAAC,IAAI,CAAC;gBAChB,IAAI,EAAE,gBAAgB;gBACtB,YAAY;gBACZ,KAAK,EAAE,WAAW,CAAC,cAAc;gBACjC,OAAO,EAAE,YAAY;aACrB,CAAC,CAAC;YACH,OAAO;QACR,CAAC;QACD,KAAK,cAAc,EAAE,CAAC;YACrB,MAAM,YAAY,GAAG,0BAA0B,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACvE,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;gBAC1B,OAAO;YACR,CAAC;YAED,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YACjD,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;gBAChC,OAAO;YACR,CAAC;YAED,MAAM,aAAa,GAAa;gBAC/B,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,WAAW,CAAC,EAAE;gBAClB,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,SAAS,EAAE,WAAW,CAAC,SAAS;gBAChC,GAAG,CAAC,WAAW,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChE,GAAG,CAAC,WAAW,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,WAAW,CAAC,YAAY,EAAE,CAAC;aAC7F,CAAC;YACF,YAAY,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;YACnD,0BAA0B,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACrD,WAAW,CAAC,IAAI,CAAC;gBAChB,IAAI,EAAE,cAAc;gBACpB,YAAY;gBACZ,QAAQ,EAAE,aAAa;gBACvB,OAAO,EAAE,YAAY;aACrB,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,mBAAmB,CAC3B,YAAiC,EACjC,WAAwC,EACxC,YAA8B,EAC9B,0BAAsD,EACtD,0BAA+C,EAC/C,qBAAoC;IAEpC,IAAI,eAAe,GAAG,KAAK,CAAC;IAC5B,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;QACxC,IACC,WAAW,CAAC,IAAI,KAAK,gBAAgB;YACrC,WAAW,CAAC,IAAI,KAAK,gBAAgB;YACrC,WAAW,CAAC,IAAI,KAAK,cAAc,EAClC,CAAC;YACF,eAAe,GAAG,IAAI,CAAC;QACxB,CAAC;QAED,eAAe,CACd,WAAW,EACX,WAAW,EACX,YAAY,EACZ,0BAA0B,EAC1B,0BAA0B,EAC1B,qBAAqB,CACrB,CAAC;IACH,CAAC;IACD,OAAO,eAAe,CAAC;AACxB,CAAC;AAED,SAAS,eAAe,CACvB,YAA8B,EAC9B,WAA6B,EAC7B,WAAoB;IAEpB,MAAM,YAAY,GAAqB;QACtC,GAAG,kBAAkB,CAAC,WAAW,CAAC;QAClC,OAAO,EAAE,YAAY,CAAC,OAAO;KAC7B,CAAC;IAEF,IACC,CAAC,WAAW,IAAI,2BAA2B,CAAC,YAAY,CAAC,CAAC;QAC1D,CAAC,YAAY,CAAC,UAAU,KAAK,MAAM,IAAI,YAAY,CAAC,UAAU,KAAK,QAAQ,CAAC,EAC3E,CAAC;QACF,YAAY,CAAC,UAAU,GAAG,SAAS,CAAC;IACrC,CAAC;IAED,OAAO,YAAY,CAAC;AACrB,CAAC;AAED,SAAS,2BAA2B,CAAC,OAAyB;IAC7D,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE;QACrC,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,OAAQ,KAAyB,CAAC,WAAW,KAAK,QAAQ,CAAC;QAC7G,OAAO,YAAY,CAAC;IACrB,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,gCAAgC,CACxC,WAAwC,EACxC,YAA8B,EAC9B,0BAA+C;IAE/C,KAAK,MAAM,CAAC,YAAY,EAAE,KAAK,CAAC,IAAI,YAAY,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;QACpE,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,SAAS;QACV,CAAC;QAED,MAAM,aAAa,GAAa;YAC/B,IAAI,EAAE,UAAU;YAChB,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,UAAU,EAAE,IAAI;YAChB,YAAY,EAAE,0CAA0C;SACxD,CAAC;QACF,YAAY,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;QACnD,KAAK,MAAM,CAAC,WAAW,EAAE,kBAAkB,CAAC,IAAI,0BAA0B,EAAE,CAAC;YAC5E,IAAI,kBAAkB,KAAK,YAAY,EAAE,CAAC;gBACzC,0BAA0B,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAChD,CAAC;QACF,CAAC;QACD,WAAW,CAAC,IAAI,CAAC;YAChB,IAAI,EAAE,cAAc;YACpB,YAAY;YACZ,QAAQ,EAAE,aAAa;YACvB,OAAO,EAAE,YAAY;SACrB,CAAC,CAAC;IACJ,CAAC;AACF,CAAC;AAED,SAAS,wBAAwB,CAChC,WAAwC,EACxC,YAA8B,EAC9B,MAA0D,EAC1D,0BAAsD,EACtD,0BAA+C,EAC/C,qBAAoC,EACpC,WAAoB;IAEpB,MAAM,eAAe,GACpB,mBAAmB,CAClB,MAAM,CAAC,MAAM,EAAE,EACf,WAAW,EACX,YAAY,EACZ,0BAA0B,EAC1B,0BAA0B,EAC1B,qBAAqB,CACrB,IAAI,WAAW,CAAC;IAElB,MAAM,iBAAiB,GACtB,qBAAqB,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,0BAA0B,CAAC,GAAG,CAAC,qBAAqB,CAAC,IAAI,IAAI,CAAC,CAAC;IACxG,IAAI,iBAAiB,IAAI,IAAI,EAAE,CAAC;QAC/B,MAAM,UAAU,GAAG,YAAY,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAC3D,IAAI,UAAU,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YACjC,WAAW,CAAC,IAAI,CAAC;gBAChB,IAAI,EAAE,UAAU;gBAChB,YAAY,EAAE,iBAAiB;gBAC/B,OAAO,EAAE,UAAU,CAAC,IAAI;gBACxB,OAAO,EAAE,YAAY;aACrB,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAED,OAAO;QACN,qBAAqB,EAAE,IAAI;QAC3B,WAAW,EAAE,eAAe;KAC5B,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gCAAgC,CAC/C,WAAwC,EACxC,QAA0B,EAC1B,KAAa;IAEb,MAAM,WAAW,GAAG,IAAI,+BAA+B,EAAE,CAAC;IAC1D,MAAM,MAAM,GAAG,QAAQ,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAElD,KAAK,CAAC,KAAK,IAAmB,EAAE;QAC/B,IAAI,YAAY,GAA4B,IAAI,CAAC;QACjD,IAAI,qBAAqB,GAAkB,IAAI,CAAC;QAChD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAClC,MAAM,0BAA0B,GAAG,IAAI,GAAG,EAAyB,CAAC;QACpE,MAAM,8BAA8B,GAAG,IAAI,GAAG,EAAkB,CAAC;QACjE,MAAM,0BAA0B,GAAG,IAAI,GAAG,EAAkB,CAAC;QAE7D,MAAM,wBAAwB,GAAG,GAAS,EAAE;YAC3C,IAAI,CAAC,YAAY,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC7C,OAAO;YACR,CAAC;YAED,CAAC,EAAE,qBAAqB,EAAE,WAAW,EAAE,GAAG,wBAAwB,CACjE,WAAW,EACX,YAAY,EACZ,MAAM,EACN,0BAA0B,EAC1B,0BAA0B,EAC1B,qBAAqB,EACrB,WAAW,CACX,CAAC,CAAC;YACH,qBAAqB,GAAG,KAAK,CAAC;QAC/B,CAAC,CAAC;QAEF,MAAM,sCAAsC,GAAG,GAAS,EAAE;YACzD,IAAI,YAAY,EAAE,CAAC;gBAClB,gCAAgC,CAAC,WAAW,EAAE,YAAY,EAAE,0BAA0B,CAAC,CAAC;YACzF,CAAC;QACF,CAAC,CAAC;QAEF,IAAI,CAAC;YACJ,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;gBACvC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACpB,KAAK,OAAO,EAAE,CAAC;wBACd,YAAY,GAAG,kBAAkB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBACjD,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;wBAC3D,MAAM;oBACP,CAAC;oBACD,KAAK,YAAY,EAAE,CAAC;wBACnB,IAAI,YAAY,EAAE,CAAC;4BAClB,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAChD,CAAC;wBACD,qBAAqB,GAAG,KAAK,CAAC,YAAY,CAAC;wBAC3C,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;wBACzD,IAAI,YAAY,EAAE,CAAC;4BAClB,WAAW,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;wBACvD,CAAC;wBACD,MAAM;oBACP,CAAC;oBACD,KAAK,YAAY,EAAE,CAAC;wBACnB,IAAI,CAAC,YAAY,EAAE,CAAC;4BACnB,MAAM;wBACP,CAAC;wBACD,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/C,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BAC5B,qBAAqB,GAAG,IAAI,CAAC;wBAC9B,CAAC;wBACD,WAAW;4BACV,mBAAmB,CAClB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EACxB,WAAW,EACX,YAAY,EACZ,0BAA0B,EAC1B,0BAA0B,EAC1B,qBAAqB,CACrB,IAAI,WAAW,CAAC;wBAClB,MAAM;oBACP,CAAC;oBACD,KAAK,UAAU,EAAE,CAAC;wBACjB,IAAI,CAAC,YAAY,EAAE,CAAC;4BACnB,MAAM;wBACP,CAAC;wBACD,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/C,IAAI,qBAAqB,EAAE,CAAC;4BAC3B,CAAC,EAAE,qBAAqB,EAAE,WAAW,EAAE,GAAG,wBAAwB,CACjE,WAAW,EACX,YAAY,EACZ,MAAM,EACN,0BAA0B,EAC1B,0BAA0B,EAC1B,qBAAqB,EACrB,WAAW,CACX,CAAC,CAAC;4BACH,qBAAqB,GAAG,KAAK,CAAC;wBAC/B,CAAC;wBACD,MAAM;oBACP,CAAC;oBACD,KAAK,gBAAgB,EAAE,CAAC;wBACvB,IAAI,CAAC,YAAY,EAAE,CAAC;4BACnB,MAAM;wBACP,CAAC;wBACD,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/C,MAAM,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC;wBACjD,MAAM,aAAa,GAAoB,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;wBAC1E,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;wBACzC,8BAA8B,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;wBACrE,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;wBAClF,MAAM;oBACP,CAAC;oBACD,KAAK,gBAAgB,EAAE,CAAC;wBACvB,IAAI,CAAC,YAAY,EAAE,CAAC;4BACnB,MAAM;wBACP,CAAC;wBACD,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/C,MAAM,YAAY,GAAG,8BAA8B,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;wBAC5E,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;4BAC1B,MAAM;wBACP,CAAC;wBACD,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;wBACjD,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;4BAChC,MAAM;wBACP,CAAC;wBACD,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,KAAK,CAAC;wBAC9B,WAAW,CAAC,IAAI,CAAC;4BAChB,IAAI,EAAE,gBAAgB;4BACtB,YAAY;4BACZ,KAAK,EAAE,KAAK,CAAC,KAAK;4BAClB,OAAO,EAAE,YAAY;yBACrB,CAAC,CAAC;wBACH,MAAM;oBACP,CAAC;oBACD,KAAK,cAAc,EAAE,CAAC;wBACrB,IAAI,CAAC,YAAY,EAAE,CAAC;4BACnB,MAAM;wBACP,CAAC;wBACD,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/C,MAAM,YAAY,GAAG,8BAA8B,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;wBAC5E,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;4BAC1B,MAAM;wBACP,CAAC;wBACD,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;wBACjD,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;4BAChC,MAAM;wBACP,CAAC;wBACD,KAAK,CAAC,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC;wBAC/B,WAAW,CAAC,IAAI,CAAC;4BAChB,IAAI,EAAE,cAAc;4BACpB,YAAY;4BACZ,OAAO,EAAE,KAAK,CAAC,OAAO;4BACtB,OAAO,EAAE,YAAY;yBACrB,CAAC,CAAC;wBACH,MAAM;oBACP,CAAC;oBACD,KAAK,MAAM,EAAE,CAAC;wBACb,IAAI,CAAC,YAAY,EAAE,CAAC;4BACnB,YAAY,GAAG,kBAAkB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAClD,CAAC;wBACD,wBAAwB,EAAE,CAAC;wBAC3B,sCAAsC,EAAE,CAAC;wBACzC,MAAM,YAAY,GAAG,eAAe,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;wBAC/E,MAAM,iBAAiB,GAAG,WAAW,IAAI,2BAA2B,CAAC,YAAY,CAAC,CAAC;wBACnF,MAAM,WAAW,GAChB,iBAAiB,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC;4BAC1E,CAAC,CAAC,SAAS;4BACX,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;wBACjB,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;wBAC/E,WAAW,CAAC,GAAG,EAAE,CAAC;wBAClB,OAAO;oBACR,CAAC;oBACD,KAAK,OAAO,EAAE,CAAC;wBACd,IAAI,CAAC,YAAY,EAAE,CAAC;4BACnB,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACxB,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;4BAC7B,OAAO;wBACR,CAAC;wBAED,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;wBAC7C,wBAAwB,EAAE,CAAC;wBAC3B,sCAAsC,EAAE,CAAC;wBAEzC,MAAM,iBAAiB,GAAG,eAAe,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;wBAClF,IAAI,2BAA2B,CAAC,iBAAiB,CAAC,EAAE,CAAC;4BACpD,MAAM,gBAAgB,GAAqB;gCAC1C,GAAG,iBAAiB;gCACpB,UAAU,EAAE,SAAS;6BACrB,CAAC;4BACF,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC;4BACjF,WAAW,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;4BAClC,OAAO;wBACR,CAAC;wBACD,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC,CAAC;wBACpF,WAAW,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;wBACnC,OAAO;oBACR,CAAC;gBACF,CAAC;YACF,CAAC;YAED,MAAM,iBAAiB,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;YACrD,IAAI,CAAC,YAAY,EAAE,CAAC;gBACnB,YAAY,GAAG,kBAAkB,CAAC,iBAAiB,CAAC,CAAC;YACtD,CAAC;YACD,wBAAwB,EAAE,CAAC;YAC3B,sCAAsC,EAAE,CAAC;YACzC,WAAW,CAAC,GAAG,CAAC,eAAe,CAAC,YAAY,EAAE,iBAAiB,EAAE,WAAW,CAAC,CAAC,CAAC;QAChF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,wBAAwB,EAAE,CAAC;YAC3B,sCAAsC,EAAE,CAAC;YACzC,MAAM,eAAe,GAAG,YAAY,IAAI,kBAAkB,CAAC,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;YACvF,eAAe,CAAC,UAAU,GAAG,OAAO,CAAC;YACrC,eAAe,CAAC,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAC9F,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;YAC7E,WAAW,CAAC,GAAG,EAAE,CAAC;QACnB,CAAC;IACF,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,WAAW,CAAC;AACpB,CAAC","sourcesContent":["import type {\n\tAssistantMessage,\n\tAssistantMessageEventStream,\n\tThinkingContent,\n\tTool,\n\tToolCall,\n\tUsage,\n} from \"../types.ts\";\nimport { AssistantMessageEventStream as AssistantMessageEventStreamImpl } from \"../utils/event-stream.ts\";\nimport { parseStreamingJson } from \"../utils/json-parse.ts\";\nimport type { StreamParserEvent, ToolCallProtocol } from \"./types.ts\";\n\ntype PartialToolCall = ToolCall & {\n\tpartialJson: string;\n};\n\nfunction isPartialToolCall(block: AssistantMessage[\"content\"][number] | undefined): block is PartialToolCall {\n\treturn block?.type === \"toolCall\" && typeof (block as PartialToolCall).partialJson === \"string\";\n}\n\nfunction cloneUsage(usage: Usage): Usage {\n\treturn {\n\t\tinput: usage.input,\n\t\toutput: usage.output,\n\t\tcacheRead: usage.cacheRead,\n\t\tcacheWrite: usage.cacheWrite,\n\t\ttotalTokens: usage.totalTokens,\n\t\tcost: {\n\t\t\tinput: usage.cost.input,\n\t\t\toutput: usage.cost.output,\n\t\t\tcacheRead: usage.cost.cacheRead,\n\t\t\tcacheWrite: usage.cost.cacheWrite,\n\t\t\ttotal: usage.cost.total,\n\t\t},\n\t};\n}\n\nfunction createOuterMessage(message: AssistantMessage): AssistantMessage {\n\treturn {\n\t\trole: \"assistant\",\n\t\tapi: message.api,\n\t\tprovider: message.provider,\n\t\tmodel: message.model,\n\t\tresponseId: message.responseId,\n\t\tcontent: [],\n\t\tusage: cloneUsage(message.usage),\n\t\tstopReason: message.stopReason,\n\t\terrorMessage: message.errorMessage,\n\t\ttimestamp: message.timestamp,\n\t};\n}\n\nfunction syncOuterMetadata(target: AssistantMessage, source: AssistantMessage): void {\n\ttarget.api = source.api;\n\ttarget.provider = source.provider;\n\ttarget.model = source.model;\n\ttarget.responseId = source.responseId;\n\ttarget.usage = cloneUsage(source.usage);\n\ttarget.stopReason = source.stopReason;\n\ttarget.errorMessage = source.errorMessage;\n\ttarget.timestamp = source.timestamp;\n}\n\nfunction pushParserEvent(\n\tparserEvent: StreamParserEvent,\n\touterStream: AssistantMessageEventStream,\n\touterMessage: AssistantMessage,\n\ttextBlockIndexByInnerIndex: Map<number, number | null>,\n\ttoolCallIndexByParserIndex: Map<number, number>,\n\tcurrentInnerTextIndex: number | null,\n): void {\n\tswitch (parserEvent.type) {\n\t\tcase \"text\": {\n\t\t\tif (currentInnerTextIndex == null) {\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tlet textBlockIndex = textBlockIndexByInnerIndex.get(currentInnerTextIndex) ?? null;\n\t\t\tif (textBlockIndex == null) {\n\t\t\t\ttextBlockIndex = outerMessage.content.length;\n\t\t\t\touterMessage.content.push({ type: \"text\", text: parserEvent.text });\n\t\t\t\ttextBlockIndexByInnerIndex.set(currentInnerTextIndex, textBlockIndex);\n\t\t\t} else {\n\t\t\t\tconst block = outerMessage.content[textBlockIndex];\n\t\t\t\tif (block?.type !== \"text\") {\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\tblock.text += parserEvent.text;\n\t\t\t}\n\n\t\t\touterStream.push({\n\t\t\t\ttype: \"text_delta\",\n\t\t\t\tcontentIndex: textBlockIndex,\n\t\t\t\tdelta: parserEvent.text,\n\t\t\t\tpartial: outerMessage,\n\t\t\t});\n\t\t\treturn;\n\t\t}\n\t\tcase \"toolcall_start\": {\n\t\t\tconst contentIndex = outerMessage.content.length;\n\t\t\tconst toolCall: PartialToolCall = {\n\t\t\t\ttype: \"toolCall\",\n\t\t\t\tid: parserEvent.id,\n\t\t\t\tname: parserEvent.name,\n\t\t\t\targuments: {},\n\t\t\t\tpartialJson: \"\",\n\t\t\t};\n\t\t\touterMessage.content.push(toolCall);\n\t\t\ttoolCallIndexByParserIndex.set(parserEvent.index, contentIndex);\n\t\t\tif (currentInnerTextIndex != null) {\n\t\t\t\ttextBlockIndexByInnerIndex.set(currentInnerTextIndex, null);\n\t\t\t}\n\t\t\touterStream.push({\n\t\t\t\ttype: \"toolcall_start\",\n\t\t\t\tcontentIndex,\n\t\t\t\tpartial: outerMessage,\n\t\t\t});\n\t\t\treturn;\n\t\t}\n\t\tcase \"toolcall_delta\": {\n\t\t\tconst contentIndex = toolCallIndexByParserIndex.get(parserEvent.index);\n\t\t\tif (contentIndex == null) {\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst block = outerMessage.content[contentIndex];\n\t\t\tif (!isPartialToolCall(block)) {\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tblock.partialJson += parserEvent.argumentsDelta;\n\t\t\tblock.arguments = parseStreamingJson<Record<string, unknown>>(block.partialJson);\n\t\t\touterStream.push({\n\t\t\t\ttype: \"toolcall_delta\",\n\t\t\t\tcontentIndex,\n\t\t\t\tdelta: parserEvent.argumentsDelta,\n\t\t\t\tpartial: outerMessage,\n\t\t\t});\n\t\t\treturn;\n\t\t}\n\t\tcase \"toolcall_end\": {\n\t\t\tconst contentIndex = toolCallIndexByParserIndex.get(parserEvent.index);\n\t\t\tif (contentIndex == null) {\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst block = outerMessage.content[contentIndex];\n\t\t\tif (block?.type !== \"toolCall\") {\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst finalToolCall: ToolCall = {\n\t\t\t\ttype: \"toolCall\",\n\t\t\t\tid: parserEvent.id,\n\t\t\t\tname: parserEvent.name,\n\t\t\t\targuments: parserEvent.arguments,\n\t\t\t\t...(parserEvent.incomplete === true ? { incomplete: true } : {}),\n\t\t\t\t...(parserEvent.errorMessage === undefined ? {} : { errorMessage: parserEvent.errorMessage }),\n\t\t\t};\n\t\t\touterMessage.content[contentIndex] = finalToolCall;\n\t\t\ttoolCallIndexByParserIndex.delete(parserEvent.index);\n\t\t\touterStream.push({\n\t\t\t\ttype: \"toolcall_end\",\n\t\t\t\tcontentIndex,\n\t\t\t\ttoolCall: finalToolCall,\n\t\t\t\tpartial: outerMessage,\n\t\t\t});\n\t\t}\n\t}\n}\n\nfunction processParserEvents(\n\tparserEvents: StreamParserEvent[],\n\touterStream: AssistantMessageEventStream,\n\touterMessage: AssistantMessage,\n\ttextBlockIndexByInnerIndex: Map<number, number | null>,\n\ttoolCallIndexByParserIndex: Map<number, number>,\n\tcurrentInnerTextIndex: number | null,\n): boolean {\n\tlet emittedToolCall = false;\n\tfor (const parserEvent of parserEvents) {\n\t\tif (\n\t\t\tparserEvent.type === \"toolcall_start\" ||\n\t\t\tparserEvent.type === \"toolcall_delta\" ||\n\t\t\tparserEvent.type === \"toolcall_end\"\n\t\t) {\n\t\t\temittedToolCall = true;\n\t\t}\n\n\t\tpushParserEvent(\n\t\t\tparserEvent,\n\t\t\touterStream,\n\t\t\touterMessage,\n\t\t\ttextBlockIndexByInnerIndex,\n\t\t\ttoolCallIndexByParserIndex,\n\t\t\tcurrentInnerTextIndex,\n\t\t);\n\t}\n\treturn emittedToolCall;\n}\n\nfunction finalizeMessage(\n\touterMessage: AssistantMessage,\n\tdoneMessage: AssistantMessage,\n\tsawToolCall: boolean,\n): AssistantMessage {\n\tconst finalMessage: AssistantMessage = {\n\t\t...createOuterMessage(doneMessage),\n\t\tcontent: outerMessage.content,\n\t};\n\n\tif (\n\t\t(sawToolCall || hasFinalizedToolCallContent(finalMessage)) &&\n\t\t(finalMessage.stopReason === \"stop\" || finalMessage.stopReason === \"length\")\n\t) {\n\t\tfinalMessage.stopReason = \"toolUse\";\n\t}\n\n\treturn finalMessage;\n}\n\nfunction hasFinalizedToolCallContent(message: AssistantMessage): boolean {\n\treturn message.content.some((block) => {\n\t\tconst hasFinalized = block.type === \"toolCall\" && typeof (block as PartialToolCall).partialJson !== \"string\";\n\t\treturn hasFinalized;\n\t});\n}\n\nfunction finalizeDanglingPartialToolCalls(\n\touterStream: AssistantMessageEventStream,\n\touterMessage: AssistantMessage,\n\ttoolCallIndexByParserIndex: Map<number, number>,\n): void {\n\tfor (const [contentIndex, block] of outerMessage.content.entries()) {\n\t\tif (!isPartialToolCall(block)) {\n\t\t\tcontinue;\n\t\t}\n\n\t\tconst finalToolCall: ToolCall = {\n\t\t\ttype: \"toolCall\",\n\t\t\tid: block.id,\n\t\t\tname: block.name,\n\t\t\targuments: block.arguments,\n\t\t\tincomplete: true,\n\t\t\terrorMessage: \"Tool call stream ended before completion\",\n\t\t};\n\t\touterMessage.content[contentIndex] = finalToolCall;\n\t\tfor (const [parserIndex, mappedContentIndex] of toolCallIndexByParserIndex) {\n\t\t\tif (mappedContentIndex === contentIndex) {\n\t\t\t\ttoolCallIndexByParserIndex.delete(parserIndex);\n\t\t\t}\n\t\t}\n\t\touterStream.push({\n\t\t\ttype: \"toolcall_end\",\n\t\t\tcontentIndex,\n\t\t\ttoolCall: finalToolCall,\n\t\t\tpartial: outerMessage,\n\t\t});\n\t}\n}\n\nfunction finalizePendingTextBlock(\n\touterStream: AssistantMessageEventStream,\n\touterMessage: AssistantMessage,\n\tparser: ReturnType<ToolCallProtocol[\"createStreamParser\"]>,\n\ttextBlockIndexByInnerIndex: Map<number, number | null>,\n\ttoolCallIndexByParserIndex: Map<number, number>,\n\tcurrentInnerTextIndex: number | null,\n\tsawToolCall: boolean,\n): { currentInnerTextIndex: null; sawToolCall: boolean } {\n\tconst nextSawToolCall =\n\t\tprocessParserEvents(\n\t\t\tparser.finish(),\n\t\t\touterStream,\n\t\t\touterMessage,\n\t\t\ttextBlockIndexByInnerIndex,\n\t\t\ttoolCallIndexByParserIndex,\n\t\t\tcurrentInnerTextIndex,\n\t\t) || sawToolCall;\n\n\tconst outerContentIndex =\n\t\tcurrentInnerTextIndex == null ? null : (textBlockIndexByInnerIndex.get(currentInnerTextIndex) ?? null);\n\tif (outerContentIndex != null) {\n\t\tconst outerBlock = outerMessage.content[outerContentIndex];\n\t\tif (outerBlock?.type === \"text\") {\n\t\t\touterStream.push({\n\t\t\t\ttype: \"text_end\",\n\t\t\t\tcontentIndex: outerContentIndex,\n\t\t\t\tcontent: outerBlock.text,\n\t\t\t\tpartial: outerMessage,\n\t\t\t});\n\t\t}\n\t}\n\n\treturn {\n\t\tcurrentInnerTextIndex: null,\n\t\tsawToolCall: nextSawToolCall,\n\t};\n}\n\nexport function wrapStreamWithToolCallMiddleware(\n\tinnerStream: AssistantMessageEventStream,\n\tprotocol: ToolCallProtocol,\n\ttools: Tool[],\n): AssistantMessageEventStream {\n\tconst outerStream = new AssistantMessageEventStreamImpl();\n\tconst parser = protocol.createStreamParser(tools);\n\n\tvoid (async (): Promise<void> => {\n\t\tlet outerMessage: AssistantMessage | null = null;\n\t\tlet currentInnerTextIndex: number | null = null;\n\t\tlet sawToolCall = false;\n\t\tlet parserHasPendingInput = false;\n\t\tconst textBlockIndexByInnerIndex = new Map<number, number | null>();\n\t\tconst thinkingBlockIndexByInnerIndex = new Map<number, number>();\n\t\tconst toolCallIndexByParserIndex = new Map<number, number>();\n\n\t\tconst flushParserAtTermination = (): void => {\n\t\t\tif (!outerMessage || !parserHasPendingInput) {\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\t({ currentInnerTextIndex, sawToolCall } = finalizePendingTextBlock(\n\t\t\t\touterStream,\n\t\t\t\touterMessage,\n\t\t\t\tparser,\n\t\t\t\ttextBlockIndexByInnerIndex,\n\t\t\t\ttoolCallIndexByParserIndex,\n\t\t\t\tcurrentInnerTextIndex,\n\t\t\t\tsawToolCall,\n\t\t\t));\n\t\t\tparserHasPendingInput = false;\n\t\t};\n\n\t\tconst finalizeDanglingToolCallsAtTermination = (): void => {\n\t\t\tif (outerMessage) {\n\t\t\t\tfinalizeDanglingPartialToolCalls(outerStream, outerMessage, toolCallIndexByParserIndex);\n\t\t\t}\n\t\t};\n\n\t\ttry {\n\t\t\tfor await (const event of innerStream) {\n\t\t\t\tswitch (event.type) {\n\t\t\t\t\tcase \"start\": {\n\t\t\t\t\t\touterMessage = createOuterMessage(event.partial);\n\t\t\t\t\t\touterStream.push({ type: \"start\", partial: outerMessage });\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"text_start\": {\n\t\t\t\t\t\tif (outerMessage) {\n\t\t\t\t\t\t\tsyncOuterMetadata(outerMessage, event.partial);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcurrentInnerTextIndex = event.contentIndex;\n\t\t\t\t\t\ttextBlockIndexByInnerIndex.set(event.contentIndex, null);\n\t\t\t\t\t\tif (outerMessage) {\n\t\t\t\t\t\t\touterStream.push({ ...event, partial: outerMessage });\n\t\t\t\t\t\t}\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"text_delta\": {\n\t\t\t\t\t\tif (!outerMessage) {\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tsyncOuterMetadata(outerMessage, event.partial);\n\t\t\t\t\t\tif (event.delta.length > 0) {\n\t\t\t\t\t\t\tparserHasPendingInput = true;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tsawToolCall =\n\t\t\t\t\t\t\tprocessParserEvents(\n\t\t\t\t\t\t\t\tparser.feed(event.delta),\n\t\t\t\t\t\t\t\touterStream,\n\t\t\t\t\t\t\t\touterMessage,\n\t\t\t\t\t\t\t\ttextBlockIndexByInnerIndex,\n\t\t\t\t\t\t\t\ttoolCallIndexByParserIndex,\n\t\t\t\t\t\t\t\tcurrentInnerTextIndex,\n\t\t\t\t\t\t\t) || sawToolCall;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"text_end\": {\n\t\t\t\t\t\tif (!outerMessage) {\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tsyncOuterMetadata(outerMessage, event.partial);\n\t\t\t\t\t\tif (parserHasPendingInput) {\n\t\t\t\t\t\t\t({ currentInnerTextIndex, sawToolCall } = finalizePendingTextBlock(\n\t\t\t\t\t\t\t\touterStream,\n\t\t\t\t\t\t\t\touterMessage,\n\t\t\t\t\t\t\t\tparser,\n\t\t\t\t\t\t\t\ttextBlockIndexByInnerIndex,\n\t\t\t\t\t\t\t\ttoolCallIndexByParserIndex,\n\t\t\t\t\t\t\t\tcurrentInnerTextIndex,\n\t\t\t\t\t\t\t\tsawToolCall,\n\t\t\t\t\t\t\t));\n\t\t\t\t\t\t\tparserHasPendingInput = false;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"thinking_start\": {\n\t\t\t\t\t\tif (!outerMessage) {\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tsyncOuterMetadata(outerMessage, event.partial);\n\t\t\t\t\t\tconst contentIndex = outerMessage.content.length;\n\t\t\t\t\t\tconst thinkingBlock: ThinkingContent = { type: \"thinking\", thinking: \"\" };\n\t\t\t\t\t\touterMessage.content.push(thinkingBlock);\n\t\t\t\t\t\tthinkingBlockIndexByInnerIndex.set(event.contentIndex, contentIndex);\n\t\t\t\t\t\touterStream.push({ type: \"thinking_start\", contentIndex, partial: outerMessage });\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"thinking_delta\": {\n\t\t\t\t\t\tif (!outerMessage) {\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tsyncOuterMetadata(outerMessage, event.partial);\n\t\t\t\t\t\tconst contentIndex = thinkingBlockIndexByInnerIndex.get(event.contentIndex);\n\t\t\t\t\t\tif (contentIndex == null) {\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst block = outerMessage.content[contentIndex];\n\t\t\t\t\t\tif (block?.type !== \"thinking\") {\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tblock.thinking += event.delta;\n\t\t\t\t\t\touterStream.push({\n\t\t\t\t\t\t\ttype: \"thinking_delta\",\n\t\t\t\t\t\t\tcontentIndex,\n\t\t\t\t\t\t\tdelta: event.delta,\n\t\t\t\t\t\t\tpartial: outerMessage,\n\t\t\t\t\t\t});\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"thinking_end\": {\n\t\t\t\t\t\tif (!outerMessage) {\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tsyncOuterMetadata(outerMessage, event.partial);\n\t\t\t\t\t\tconst contentIndex = thinkingBlockIndexByInnerIndex.get(event.contentIndex);\n\t\t\t\t\t\tif (contentIndex == null) {\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst block = outerMessage.content[contentIndex];\n\t\t\t\t\t\tif (block?.type !== \"thinking\") {\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tblock.thinking = event.content;\n\t\t\t\t\t\touterStream.push({\n\t\t\t\t\t\t\ttype: \"thinking_end\",\n\t\t\t\t\t\t\tcontentIndex,\n\t\t\t\t\t\t\tcontent: event.content,\n\t\t\t\t\t\t\tpartial: outerMessage,\n\t\t\t\t\t\t});\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"done\": {\n\t\t\t\t\t\tif (!outerMessage) {\n\t\t\t\t\t\t\touterMessage = createOuterMessage(event.message);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tflushParserAtTermination();\n\t\t\t\t\t\tfinalizeDanglingToolCallsAtTermination();\n\t\t\t\t\t\tconst finalMessage = finalizeMessage(outerMessage, event.message, sawToolCall);\n\t\t\t\t\t\tconst finalizedToolCall = sawToolCall || hasFinalizedToolCallContent(finalMessage);\n\t\t\t\t\t\tconst finalReason =\n\t\t\t\t\t\t\tfinalizedToolCall && (event.reason === \"stop\" || event.reason === \"length\")\n\t\t\t\t\t\t\t\t? \"toolUse\"\n\t\t\t\t\t\t\t\t: event.reason;\n\t\t\t\t\t\touterStream.push({ type: \"done\", reason: finalReason, message: finalMessage });\n\t\t\t\t\t\touterStream.end();\n\t\t\t\t\t\treturn;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"error\": {\n\t\t\t\t\t\tif (!outerMessage) {\n\t\t\t\t\t\t\touterStream.push(event);\n\t\t\t\t\t\t\touterStream.end(event.error);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\tsyncOuterMetadata(outerMessage, event.error);\n\t\t\t\t\t\tflushParserAtTermination();\n\t\t\t\t\t\tfinalizeDanglingToolCallsAtTermination();\n\n\t\t\t\t\t\tconst finalErrorMessage = finalizeMessage(outerMessage, event.error, sawToolCall);\n\t\t\t\t\t\tif (hasFinalizedToolCallContent(finalErrorMessage)) {\n\t\t\t\t\t\t\tconst recoveredMessage: AssistantMessage = {\n\t\t\t\t\t\t\t\t...finalErrorMessage,\n\t\t\t\t\t\t\t\tstopReason: \"toolUse\",\n\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\touterStream.push({ type: \"done\", reason: \"toolUse\", message: recoveredMessage });\n\t\t\t\t\t\t\touterStream.end(recoveredMessage);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\touterStream.push({ type: \"error\", reason: event.reason, error: finalErrorMessage });\n\t\t\t\t\t\touterStream.end(finalErrorMessage);\n\t\t\t\t\t\treturn;\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tconst finalInnerMessage = await innerStream.result();\n\t\t\tif (!outerMessage) {\n\t\t\t\touterMessage = createOuterMessage(finalInnerMessage);\n\t\t\t}\n\t\t\tflushParserAtTermination();\n\t\t\tfinalizeDanglingToolCallsAtTermination();\n\t\t\touterStream.end(finalizeMessage(outerMessage, finalInnerMessage, sawToolCall));\n\t\t} catch (error) {\n\t\t\tflushParserAtTermination();\n\t\t\tfinalizeDanglingToolCallsAtTermination();\n\t\t\tconst fallbackMessage = outerMessage ?? createOuterMessage(await innerStream.result());\n\t\t\tfallbackMessage.stopReason = \"error\";\n\t\t\tfallbackMessage.errorMessage = error instanceof Error ? error.message : JSON.stringify(error);\n\t\t\touterStream.push({ type: \"error\", reason: \"error\", error: fallbackMessage });\n\t\t\touterStream.end();\n\t\t}\n\t})();\n\n\treturn outerStream;\n}\n"]}
+{"version":3,"file":"stream-wrapper.js","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,2BAA2B,IAAI,+BAA+B,EAAE,MAAM,0BAA0B,CAAC;AAC1G,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAGrE,MAAM,UAAU,gCAAgC,CAC/C,WAAwC,EACxC,QAA0B,EAC1B,KAAa;IAEb,MAAM,WAAW,GAAG,IAAI,+BAA+B,EAAE,CAAC;IAC1D,MAAM,MAAM,GAAG,QAAQ,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAElD,KAAK,CAAC,KAAK,IAAmB,EAAE;QAC/B,IAAI,UAAU,GAAmC,IAAI,CAAC;QACtD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAElC,MAAM,WAAW,GAAG,GAAS,EAAE;YAC9B,IAAI,CAAC,UAAU,IAAI,CAAC,qBAAqB;gBAAE,OAAO;YAClD,MAAM,MAAM,GAAG,UAAU,CAAC,mBAAmB,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/D,WAAW,GAAG,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;YAChD,UAAU,CAAC,UAAU,EAAE,CAAC;YACxB,qBAAqB,GAAG,KAAK,CAAC;QAC/B,CAAC,CAAC;QAEF,IAAI,CAAC;YACJ,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;gBACvC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACpB,KAAK,OAAO;wBACX,UAAU,GAAG,IAAI,uBAAuB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBACrE,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;wBACjE,MAAM;oBACP,KAAK,YAAY;wBAChB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;wBACpF,MAAM;oBACP,KAAK,YAAY,EAAE,CAAC;wBACnB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;4BAAE,qBAAqB,GAAG,IAAI,CAAC;wBACzD,MAAM,MAAM,GAAG,UAAU,CAAC,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;wBACxE,WAAW,GAAG,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;wBAChD,MAAM;oBACP,CAAC;oBACD,KAAK,UAAU;wBACd,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,WAAW,EAAE,CAAC;wBACd,MAAM;oBACP,KAAK,gBAAgB;wBACpB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC5D,MAAM;oBACP,KAAK,gBAAgB;wBACpB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,oBAAoB,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAChF,MAAM;oBACP,KAAK,cAAc;wBAClB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC5E,MAAM;oBACP,KAAK,MAAM,EAAE,CAAC;wBACb,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBACvE,WAAW,EAAE,CAAC;wBACd,UAAU,CAAC,yBAAyB,EAAE,CAAC;wBACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;wBAChE,MAAM,SAAS,GAAG,WAAW,IAAI,UAAU,CAAC,2BAA2B,EAAE,CAAC;wBAC1E,MAAM,MAAM,GACX,SAAS,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;wBAChG,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;wBACpD,WAAW,CAAC,GAAG,EAAE,CAAC;wBAClB,OAAO;oBACR,CAAC;oBACD,KAAK,OAAO,EAAE,CAAC;wBACd,IAAI,CAAC,UAAU,EAAE,CAAC;4BACjB,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACxB,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;4BAC7B,OAAO;wBACR,CAAC;wBACD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;wBAC7B,WAAW,EAAE,CAAC;wBACd,UAAU,CAAC,yBAAyB,EAAE,CAAC;wBACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;wBAC9D,IAAI,UAAU,CAAC,2BAA2B,EAAE,EAAE,CAAC;4BAC9C,MAAM,SAAS,GAAqB,EAAE,GAAG,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;4BAC1E,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;4BAC1E,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;4BAC3B,OAAO;wBACR,CAAC;wBACD,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;wBAC1E,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACzB,OAAO;oBACR,CAAC;oBACD,KAAK,gBAAgB,CAAC;oBACtB,KAAK,gBAAgB,CAAC;oBACtB,KAAK,cAAc;wBAClB,MAAM;gBACR,CAAC;YACF,CAAC;YAED,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;YAChD,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YACtE,WAAW,EAAE,CAAC;YACd,UAAU,CAAC,yBAAyB,EAAE,CAAC;YACvC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,WAAW,EAAE,CAAC;YACd,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;YACpF,UAAU,CAAC,yBAAyB,EAAE,CAAC;YACvC,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC;YACpC,QAAQ,CAAC,UAAU,GAAG,OAAO,CAAC;YAC9B,QAAQ,CAAC,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACvF,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YACtE,WAAW,CAAC,GAAG,EAAE,CAAC;QACnB,CAAC;IACF,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,WAAW,CAAC;AACpB,CAAC","sourcesContent":["import type { AssistantMessage, AssistantMessageEventStream, Tool } from \"../types.ts\";\nimport { AssistantMessageEventStream as AssistantMessageEventStreamImpl } from \"../utils/event-stream.ts\";\nimport { StreamMessageProjection } from \"./stream-wrapper-shared.ts\";\nimport type { ToolCallProtocol } from \"./types.ts\";\n\nexport function wrapStreamWithToolCallMiddleware(\n\tinnerStream: AssistantMessageEventStream,\n\tprotocol: ToolCallProtocol,\n\ttools: Tool[],\n): AssistantMessageEventStream {\n\tconst outerStream = new AssistantMessageEventStreamImpl();\n\tconst parser = protocol.createStreamParser(tools);\n\n\tvoid (async (): Promise<void> => {\n\t\tlet projection: StreamMessageProjection | null = null;\n\t\tlet sawToolCall = false;\n\t\tlet parserHasPendingInput = false;\n\n\t\tconst flushParser = (): void => {\n\t\t\tif (!projection || !parserHasPendingInput) return;\n\t\t\tconst result = projection.projectParserEvents(parser.finish());\n\t\t\tsawToolCall = sawToolCall || result.sawToolCall;\n\t\t\tprojection.finishText();\n\t\t\tparserHasPendingInput = false;\n\t\t};\n\n\t\ttry {\n\t\t\tfor await (const event of innerStream) {\n\t\t\t\tswitch (event.type) {\n\t\t\t\t\tcase \"start\":\n\t\t\t\t\t\tprojection = new StreamMessageProjection(outerStream, event.partial);\n\t\t\t\t\t\touterStream.push({ type: \"start\", partial: projection.message });\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_start\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.startText(event.contentIndex, event.partial.content[event.contentIndex]);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_delta\": {\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tif (event.delta.length > 0) parserHasPendingInput = true;\n\t\t\t\t\t\tconst result = projection.projectParserEvents(parser.feed(event.delta));\n\t\t\t\t\t\tsawToolCall = sawToolCall || result.sawToolCall;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"text_end\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tflushParser();\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_start\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.startThinking(event.contentIndex, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_delta\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.projectThinkingDelta(event.contentIndex, event.delta, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_end\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.finishThinking(event.contentIndex, event.content, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"done\": {\n\t\t\t\t\t\tprojection ??= new StreamMessageProjection(outerStream, event.message);\n\t\t\t\t\t\tflushParser();\n\t\t\t\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\t\t\t\tconst message = projection.finalize(event.message, sawToolCall);\n\t\t\t\t\t\tconst recovered = sawToolCall || projection.hasFinalizedToolCallContent();\n\t\t\t\t\t\tconst reason =\n\t\t\t\t\t\t\trecovered && (event.reason === \"stop\" || event.reason === \"length\") ? \"toolUse\" : event.reason;\n\t\t\t\t\t\touterStream.push({ type: \"done\", reason, message });\n\t\t\t\t\t\touterStream.end();\n\t\t\t\t\t\treturn;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"error\": {\n\t\t\t\t\t\tif (!projection) {\n\t\t\t\t\t\t\touterStream.push(event);\n\t\t\t\t\t\t\touterStream.end(event.error);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tprojection.sync(event.error);\n\t\t\t\t\t\tflushParser();\n\t\t\t\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\t\t\t\tconst message = projection.finalize(event.error, sawToolCall);\n\t\t\t\t\t\tif (projection.hasFinalizedToolCallContent()) {\n\t\t\t\t\t\t\tconst recovered: AssistantMessage = { ...message, stopReason: \"toolUse\" };\n\t\t\t\t\t\t\touterStream.push({ type: \"done\", reason: \"toolUse\", message: recovered });\n\t\t\t\t\t\t\touterStream.end(recovered);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\touterStream.push({ type: \"error\", reason: event.reason, error: message });\n\t\t\t\t\t\touterStream.end(message);\n\t\t\t\t\t\treturn;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"toolcall_start\":\n\t\t\t\t\tcase \"toolcall_delta\":\n\t\t\t\t\tcase \"toolcall_end\":\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tconst innerMessage = await innerStream.result();\n\t\t\tprojection ??= new StreamMessageProjection(outerStream, innerMessage);\n\t\t\tflushParser();\n\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\touterStream.end(projection.finalize(innerMessage, sawToolCall));\n\t\t} catch (error) {\n\t\t\tflushParser();\n\t\t\tprojection ??= new StreamMessageProjection(outerStream, await innerStream.result());\n\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\tconst fallback = projection.message;\n\t\t\tfallback.stopReason = \"error\";\n\t\t\tfallback.errorMessage = error instanceof Error ? error.message : JSON.stringify(error);\n\t\t\touterStream.push({ type: \"error\", reason: \"error\", error: fallback });\n\t\t\touterStream.end();\n\t\t}\n\t})();\n\n\treturn outerStream;\n}\n"]}