@code-yeongyu/senpi-ai 2026.7.25-2

package/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/dist/tool-call-middleware/stream-wrapper.d.ts

@@ -0,0 +1,4 @@
+import type { AssistantMessageEventStream, Tool } from "../types.ts";
+import type { ToolCallProtocol } from "./types.ts";
+export declare function wrapStreamWithToolCallMiddleware(innerStream: AssistantMessageEventStream, protocol: ToolCallProtocol, tools: Tool[]): AssistantMessageEventStream;
+//# sourceMappingURL=stream-wrapper.d.ts.map

package/dist/tool-call-middleware/stream-wrapper.d.ts.map

@@ -0,0 +1 @@
+{"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/dist/tool-call-middleware/stream-wrapper.js

@@ -0,0 +1,121 @@
+import { AssistantMessageEventStream as AssistantMessageEventStreamImpl } from "../utils/event-stream.js";
+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 projection = null;
+        let sawToolCall = false;
+        let parserHasPendingInput = false;
+        const flushParser = () => {
+            if (!projection || !parserHasPendingInput)
+                return;
+            const result = projection.projectParserEvents(parser.finish());
+            sawToolCall = sawToolCall || result.sawToolCall;
+            projection.finishText();
+            parserHasPendingInput = false;
+        };
+        try {
+            for await (const event of innerStream) {
+                switch (event.type) {
+                    case "start":
+                        projection = new StreamMessageProjection(outerStream, event.partial);
+                        outerStream.push({ type: "start", partial: projection.message });
+                        break;
+                    case "text_start":
+                        if (!projection)
+                            break;
+                        projection.sync(event.partial);
+                        projection.startText(event.contentIndex, event.partial.content[event.contentIndex]);
+                        break;
+                    case "text_delta": {
+                        if (!projection)
+                            break;
+                        projection.sync(event.partial);
+                        if (event.delta.length > 0)
+                            parserHasPendingInput = true;
+                        const result = projection.projectParserEvents(parser.feed(event.delta));
+                        sawToolCall = sawToolCall || result.sawToolCall;
+                        break;
+                    }
+                    case "text_end":
+                        if (!projection)
+                            break;
+                        projection.sync(event.partial);
+                        flushParser();
+                        break;
+                    case "thinking_start":
+                        if (!projection)
+                            break;
+                        projection.sync(event.partial);
+                        projection.startThinking(event.contentIndex, event.partial);
+                        break;
+                    case "thinking_delta":
+                        if (!projection)
+                            break;
+                        projection.sync(event.partial);
+                        projection.projectThinkingDelta(event.contentIndex, event.delta, event.partial);
+                        break;
+                    case "thinking_end":
+                        if (!projection)
+                            break;
+                        projection.sync(event.partial);
+                        projection.finishThinking(event.contentIndex, event.content, event.partial);
+                        break;
+                    case "done": {
+                        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 (!projection) {
+                            outerStream.push(event);
+                            outerStream.end(event.error);
+                            return;
+                        }
+                        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: message });
+                        outerStream.end(message);
+                        return;
+                    }
+                    case "toolcall_start":
+                    case "toolcall_delta":
+                    case "toolcall_end":
+                        break;
+                }
+            }
+            const innerMessage = await innerStream.result();
+            projection ??= new StreamMessageProjection(outerStream, innerMessage);
+            flushParser();
+            projection.finalizeDanglingToolCalls();
+            outerStream.end(projection.finalize(innerMessage, sawToolCall));
+        }
+        catch (error) {
+            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();
+        }
+    })();
+    return outerStream;
+}
+//# sourceMappingURL=stream-wrapper.js.map

package/dist/tool-call-middleware/stream-wrapper.js.map

@@ -0,0 +1 @@
+{"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"]}

package/dist/tool-call-middleware/types.d.ts

@@ -0,0 +1,109 @@
+import type { ImageContent, TextContent, Tool } from "../types.ts";
+/**
+ * Supported tool call formats for models that don't natively support tool calling.
+ * - "hermes": Hermes 2/3 format with special delimiters
+ * - "morph-xml": XML-based tool call format
+ * - "xml": Deprecated alias for "morph-xml"
+ * - "yaml-xml": YAML arguments inside XML tool tags
+ * - "gemma4-delimiter": Gemma 4 specific delimiter format
+ * - "anthropic-xml": Legacy Anthropic invoke/parameter XML format
+ * - "antml": ANTML function_calls/invoke format with Claude-Code-style failure tolerance
+ */
+export type ToolCallFormat = "hermes" | "xml" | "morph-xml" | "yaml-xml" | "gemma4-delimiter" | "anthropic-xml" | "antml";
+/**
+ * Content type for tool results (text or image)
+ */
+export type ToolResultContent = TextContent | ImageContent;
+export interface ParserOptions {
+    emitRawToolCallTextOnError?: boolean;
+    onError?: (message: string, metadata?: Record<string, unknown>) => void;
+}
+/**
+ * A parsed tool call extracted from generated text
+ */
+export interface ParsedToolCall {
+    name: string;
+    arguments: Record<string, unknown>;
+}
+/**
+ * Events emitted by the stream parser during text processing.
+ * Maps 1:1 to AssistantMessageEventStream events for seamless integration.
+ */
+export type StreamParserEvent = {
+    type: "text";
+    text: string;
+} | {
+    type: "toolcall_start";
+    index: number;
+    name: string;
+    id: string;
+} | {
+    type: "toolcall_delta";
+    index: number;
+    argumentsDelta: string;
+} | {
+    type: "toolcall_end";
+    index: number;
+    name: string;
+    id: string;
+    arguments: Record<string, unknown>;
+    incomplete?: true;
+    errorMessage?: string;
+};
+/**
+ * Stream parser for incremental tool call parsing during streaming.
+ */
+export interface StreamParser {
+    /**
+     * Feed a text delta into the parser and receive any events generated.
+     * @param textDelta - The next chunk of text to parse
+     * @returns Array of parser events (text or tool call events)
+     */
+    feed(textDelta: string): StreamParserEvent[];
+    /**
+     * Signal that the stream has ended and receive any final events.
+     * @returns Array of final parser events
+     */
+    finish(): StreamParserEvent[];
+}
+/**
+ * Protocol interface for formatting and parsing tool calls in different formats.
+ * Implementations handle the specifics of each tool call format (Hermes, XML, etc.)
+ */
+export interface ToolCallProtocol {
+    /**
+     * Format tools into a system prompt that instructs the model how to use tools.
+     * @param tools - Array of available tools
+     * @returns Formatted system prompt string
+     */
+    formatToolsSystemPrompt(tools: Tool[]): string;
+    /**
+     * Format a tool response for inclusion in the conversation context.
+     * @param toolName - Name of the tool that was called
+     * @param toolCallId - Unique identifier for this tool call
+     * @param content - Array of content blocks (text or images) representing the result
+     * @returns Formatted tool response string
+     */
+    formatToolResponse(toolName: string, toolCallId: string, content: ToolResultContent[]): string;
+    /**
+     * Format a tool call for sending to the model.
+     * @param name - Name of the tool to call
+     * @param args - Arguments for the tool call
+     * @returns Formatted tool call string
+     */
+    formatToolCall(name: string, args: Record<string, unknown>): string;
+    /**
+     * Parse generated text to extract tool calls.
+     * @param text - The generated text to parse
+     * @param tools - Available tools for validation
+     * @returns Array of parsed tool calls
+     */
+    parseGeneratedText(text: string, tools: Tool[], options?: ParserOptions): ParsedToolCall[];
+    /**
+     * Create a stream parser for incremental parsing during streaming.
+     * @param tools - Available tools for the parser to recognize
+     * @returns A new StreamParser instance
+     */
+    createStreamParser(tools: Tool[], options?: ParserOptions): StreamParser;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/tool-call-middleware/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAEnE;;;;;;;;;GASG;AACH,MAAM,MAAM,cAAc,GACvB,QAAQ,GACR,KAAK,GACL,WAAW,GACX,UAAU,GACV,kBAAkB,GAClB,eAAe,GACf,OAAO,CAAC;AAEX;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG,WAAW,GAAG,YAAY,CAAC;AAE3D,MAAM,WAAW,aAAa;IAC7B,0BAA0B,CAAC,EAAE,OAAO,CAAC;IACrC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;CACxE;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED;;;GAGG;AACH,MAAM,MAAM,iBAAiB,GAC1B;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAC9B;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,MAAM,CAAA;CAAE,GACnE;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,cAAc,EAAE,MAAM,CAAA;CAAE,GACjE;IACA,IAAI,EAAE,cAAc,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,UAAU,CAAC,EAAE,IAAI,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;CACrB,CAAC;AAEL;;GAEG;AACH,MAAM,WAAW,YAAY;IAC5B;;;;OAIG;IACH,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,iBAAiB,EAAE,CAAC;IAE7C;;;OAGG;IACH,MAAM,IAAI,iBAAiB,EAAE,CAAC;CAC9B;AAED;;;GAGG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,uBAAuB,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC;IAE/C;;;;;;OAMG;IACH,kBAAkB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,iBAAiB,EAAE,GAAG,MAAM,CAAC;IAE/F;;;;;OAKG;IACH,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC;IAEpE;;;;;OAKG;IACH,kBAAkB,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,cAAc,EAAE,CAAC;IAE3F;;;;OAIG;IACH,kBAAkB,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,YAAY,CAAC;CACzE"}

package/dist/tool-call-middleware/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/tool-call-middleware/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/tool-call-middleware/types.ts"],"names":[],"mappings":"","sourcesContent":["import type { ImageContent, TextContent, Tool } from \"../types.ts\";\n\n/**\n * Supported tool call formats for models that don't natively support tool calling.\n * - \"hermes\": Hermes 2/3 format with special delimiters\n * - \"morph-xml\": XML-based tool call format\n * - \"xml\": Deprecated alias for \"morph-xml\"\n * - \"yaml-xml\": YAML arguments inside XML tool tags\n * - \"gemma4-delimiter\": Gemma 4 specific delimiter format\n * - \"anthropic-xml\": Legacy Anthropic invoke/parameter XML format\n * - \"antml\": ANTML function_calls/invoke format with Claude-Code-style failure tolerance\n */\nexport type ToolCallFormat =\n\t| \"hermes\"\n\t| \"xml\"\n\t| \"morph-xml\"\n\t| \"yaml-xml\"\n\t| \"gemma4-delimiter\"\n\t| \"anthropic-xml\"\n\t| \"antml\";\n\n/**\n * Content type for tool results (text or image)\n */\nexport type ToolResultContent = TextContent | ImageContent;\n\nexport interface ParserOptions {\n\temitRawToolCallTextOnError?: boolean;\n\tonError?: (message: string, metadata?: Record<string, unknown>) => void;\n}\n\n/**\n * A parsed tool call extracted from generated text\n */\nexport interface ParsedToolCall {\n\tname: string;\n\targuments: Record<string, unknown>;\n}\n\n/**\n * Events emitted by the stream parser during text processing.\n * Maps 1:1 to AssistantMessageEventStream events for seamless integration.\n */\nexport type StreamParserEvent =\n\t| { type: \"text\"; text: string }\n\t| { type: \"toolcall_start\"; index: number; name: string; id: string }\n\t| { type: \"toolcall_delta\"; index: number; argumentsDelta: string }\n\t| {\n\t\t\ttype: \"toolcall_end\";\n\t\t\tindex: number;\n\t\t\tname: string;\n\t\t\tid: string;\n\t\t\targuments: Record<string, unknown>;\n\t\t\tincomplete?: true;\n\t\t\terrorMessage?: string;\n\t  };\n\n/**\n * Stream parser for incremental tool call parsing during streaming.\n */\nexport interface StreamParser {\n\t/**\n\t * Feed a text delta into the parser and receive any events generated.\n\t * @param textDelta - The next chunk of text to parse\n\t * @returns Array of parser events (text or tool call events)\n\t */\n\tfeed(textDelta: string): StreamParserEvent[];\n\n\t/**\n\t * Signal that the stream has ended and receive any final events.\n\t * @returns Array of final parser events\n\t */\n\tfinish(): StreamParserEvent[];\n}\n\n/**\n * Protocol interface for formatting and parsing tool calls in different formats.\n * Implementations handle the specifics of each tool call format (Hermes, XML, etc.)\n */\nexport interface ToolCallProtocol {\n\t/**\n\t * Format tools into a system prompt that instructs the model how to use tools.\n\t * @param tools - Array of available tools\n\t * @returns Formatted system prompt string\n\t */\n\tformatToolsSystemPrompt(tools: Tool[]): string;\n\n\t/**\n\t * Format a tool response for inclusion in the conversation context.\n\t * @param toolName - Name of the tool that was called\n\t * @param toolCallId - Unique identifier for this tool call\n\t * @param content - Array of content blocks (text or images) representing the result\n\t * @returns Formatted tool response string\n\t */\n\tformatToolResponse(toolName: string, toolCallId: string, content: ToolResultContent[]): string;\n\n\t/**\n\t * Format a tool call for sending to the model.\n\t * @param name - Name of the tool to call\n\t * @param args - Arguments for the tool call\n\t * @returns Formatted tool call string\n\t */\n\tformatToolCall(name: string, args: Record<string, unknown>): string;\n\n\t/**\n\t * Parse generated text to extract tool calls.\n\t * @param text - The generated text to parse\n\t * @param tools - Available tools for validation\n\t * @returns Array of parsed tool calls\n\t */\n\tparseGeneratedText(text: string, tools: Tool[], options?: ParserOptions): ParsedToolCall[];\n\n\t/**\n\t * Create a stream parser for incremental parsing during streaming.\n\t * @param tools - Available tools for the parser to recognize\n\t * @returns A new StreamParser instance\n\t */\n\tcreateStreamParser(tools: Tool[], options?: ParserOptions): StreamParser;\n}\n"]}