@code-yeongyu/senpi-ai 2026.7.25-2

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

@@ -0,0 +1 @@
+{"version":3,"file":"recovery-stream-wrapper.js","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-wrapper.ts"],"names":[],"mappings":"AACA,OAAO,EAA4B,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AACrG,OAAO,EAAE,mCAAmC,EAAE,MAAM,4BAA4B,CAAC;AACjF,OAAO,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAC3E,OAAO,EAA8B,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AAC7G,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAErE,MAAM,UAAU,4BAA4B,CAC3C,WAAwC,EACxC,KAAsB;IAEtB,MAAM,WAAW,GAAG,IAAI,mCAAmC,EAAE,CAAC;IAE9D,KAAK,CAAC,KAAK,IAAmB,EAAE;QAC/B,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;QAC1D,MAAM,WAAW,GAAyC,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC,aAAa,EAAE,CAAC;QAC1G,IAAI,UAAU,GAAmC,IAAI,CAAC;QACtD,IAAI,gBAAgB,GAAoC,IAAI,CAAC;QAC7D,IAAI,cAAc,GAAkC,IAAI,CAAC;QACzD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAClC,MAAM,gBAAgB,GAAG,IAAI,wBAAwB,EAAE,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,sBAAsB,CAAC,WAAW,CAAC,CAAC;QAEzD,MAAM,UAAU,GAAG,GAAS,EAAE;YAC7B,IAAI,CAAC,cAAc;gBAAE,OAAO;YAC5B,WAAW,GAAG,cAAc,CAAC,MAAM,EAAE,IAAI,WAAW,CAAC;YACrD,cAAc,GAAG,IAAI,CAAC;QACvB,CAAC,CAAC;QAEF,WAAW,CAAC,sBAAsB,CAAC,KAAK,IAAI,EAAE;YAC7C,qBAAqB,GAAG,IAAI,CAAC;YAC7B,IAAI,CAAC;gBACJ,MAAM,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC/B,UAAU,EAAE,CAAC;gBACb,QAAQ,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YAChC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,UAAU,EAAE,CAAC;gBACb,QAAQ,CAAC,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAC5C,MAAM,KAAK,CAAC;YACb,CAAC;QACF,CAAC,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAG,CAAC,MAAwB,EAAE,OAA8B,EAAQ,EAAE;YAC9F,IAAI,CAAC,UAAU,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gBAAE,OAAO;YAC3D,UAAU,EAAE,CAAC;YACb,iCAAiC,CAAC,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QAC7E,CAAC,CAAC;QAEF,MAAM,mBAAmB,GAAG,CAAC,MAAwB,EAAE,YAAoB,EAAW,EAAE;YACvF,IAAI,CAAC,UAAU,IAAI,CAAC,gBAAgB;gBAAE,OAAO,KAAK,CAAC;YACnD,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC,IAAI,YAAY,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBACtG,mBAAmB,CAAC,MAAM,EAAE,4BAA4B,CAAC,CAAC;gBAC1D,OAAO,KAAK,CAAC;YACd,CAAC;YACD,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxB,gBAAgB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC3C,IAAI,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,EAAE,YAAY,CAAC;gBAAE,OAAO,IAAI,CAAC;YACzE,mBAAmB,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACzC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,CAChB,MAAwB,EACxB,YAAoB,EACpB,IAAyB,EACzB,OAAO,GAA0B,6BAA6B,EACpD,EAAE;YACZ,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC,IAAI,YAAY,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBACtG,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBACrC,OAAO,KAAK,CAAC;YACd,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,YAAY,GACjB,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,CAAC;gBAC3C,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,CAAC;gBACnD,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,CAAC,CAAC;YACrD,IAAI,YAAY,IAAI,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAAE,OAAO,IAAI,CAAC;YACzE,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACrC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,CAChB,MAAwB,EACxB,YAAoB,EACpB,IAAyB,EACzB,OAAO,GAA0B,6BAA6B,EACpD,EAAE;YACZ,IAAI,gBAAgB,CAAC,QAAQ,CAAC,YAAY,EAAE,IAAI,CAAC;gBAAE,OAAO,IAAI,CAAC;YAC/D,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACrC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,mBAAmB,GAAG,CAAC,MAAwB,EAAW,EAAE;YACjE,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,MAAM,EAAE,EAAE,sBAAsB,EAAE,IAAI,EAAE,CAAC,CAAC;YAClG,gBAAgB,KAAK,IAAI,wBAAwB,CAAC,WAAW,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;YACnF,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxB,gBAAgB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC3C,UAAU,EAAE,CAAC;YACb,IAAI,gBAAgB,CAAC,oBAAoB,CAAC,MAAM,CAAC;gBAAE,OAAO,IAAI,CAAC;YAC/D,mBAAmB,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACzC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,IAAI,CAAC;YACJ,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;gBACvC,IAAI,qBAAqB;oBAAE,SAAS;gBACpC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACpB,KAAK,OAAO;wBACX,UAAU,GAAG,IAAI,uBAAuB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,EAAE;4BACpE,sBAAsB,EAAE,IAAI;yBAC5B,CAAC,CAAC;wBACH,gBAAgB,GAAG,IAAI,wBAAwB,CAAC,WAAW,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;wBACjF,gBAAgB,CAAC,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAClD,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;wBACjE,MAAM;oBACP,KAAK,YAAY,EAAE,CAAC;wBACnB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC;4BAAE,OAAO;wBACjE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,gBAAgB;4BAC9F,OAAO;wBACR,MAAM,QAAQ,GAAG,IAAI,sBAAsB,CAAC,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;wBACrG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;4BACpC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,cAAc,GAAG,QAAQ,CAAC;wBAC1B,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;wBACnD,MAAM;oBACP,CAAC;oBACD,KAAK,YAAY;wBAChB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC;4BAAE,OAAO;wBACjE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,cAAc;4BAAE,OAAO;wBACvF,WAAW,GAAG,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,WAAW,CAAC;wBAC9D,MAAM;oBACP,KAAK,UAAU;wBACd,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC;4BAAE,OAAO;wBACjE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC;4BAAE,OAAO;wBACpE,UAAU,EAAE,CAAC;wBACb,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;wBACjD,MAAM;oBACP,KAAK,MAAM;wBACV,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU;4BAAE,OAAO;wBAC/D,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;4BAAE,OAAO;wBAC5C,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;wBACpE,OAAO;oBACR,KAAK,OAAO;wBACX,IAAI,CAAC,UAAU,EAAE,CAAC;4BACjB,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gCAAE,OAAO;4BAC5C,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACxB,WAAW,CAAC,GAAG,EAAE,CAAC;4BAClB,OAAO;wBACR,CAAC;wBACD,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC;4BAAE,OAAO;wBAC9C,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;4BAAE,OAAO;wBAC5C,QAAQ,CAAC,WAAW,CAAC,UAAU,EAAE,KAAK,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;wBACzE,OAAO;oBACR,KAAK,gBAAgB,EAAE,CAAC;wBACvB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,EAAE,4BAA4B,CAAC;4BAAE,OAAO;wBACnG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB;4BAAE,OAAO;wBACzF,MAAM,MAAM,GAAG,gBAAgB,CAAC,kBAAkB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;wBACtF,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;4BAC5B,mBAAmB,CAClB,KAAK,CAAC,OAAO,EACb,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,4BAA4B,CACnE,CAAC;4BACF,OAAO;wBACR,CAAC;wBACD,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACvD,WAAW,GAAG,IAAI,CAAC;wBACnB,MAAM;oBACP,CAAC;oBACD,KAAK,gBAAgB;wBACpB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,EAAE,4BAA4B,CAAC;4BAAE,OAAO;wBACnG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB;4BAAE,OAAO;wBACzF,IAAI,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;4BAC1F,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,MAAM;oBACP,KAAK,cAAc;wBAClB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,EAAE,4BAA4B,CAAC;4BAAE,OAAO;wBACnG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB;4BAAE,OAAO;wBACzF,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;4BAC5E,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACrD,WAAW,GAAG,IAAI,CAAC;wBACnB,MAAM;oBACP,KAAK,gBAAgB,EAAE,CAAC;wBACvB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC;4BAAE,OAAO;wBACrE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,gBAAgB;4BAC9F,OAAO;wBACR,MAAM,UAAU,GAAG,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/E,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,EAAE,CAAC;4BAC5E,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACvD,MAAM;oBACP,CAAC;oBACD,KAAK,gBAAgB;wBACpB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC;4BAAE,OAAO;wBACrE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU;4BAAE,OAAO;wBACnF,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,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC;4BAAE,OAAO;wBACrE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU;4BAAE,OAAO;wBACnF,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC5E,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACrD,MAAM;gBACR,CAAC;YACF,CAAC;YAED,IAAI,qBAAqB,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gBAAE,OAAO;YACrE,UAAU,EAAE,CAAC;YACb,QAAQ,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,qBAAqB,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gBAAE,OAAO;YACrE,UAAU,EAAE,CAAC;YACb,QAAQ,CAAC,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;IACF,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,WAAW,CAAC;AACpB,CAAC","sourcesContent":["import type { AssistantMessage, AssistantMessageEvent, AssistantMessageEventStream, Tool } from \"../types.ts\";\nimport { type RecoveryContentKind, RecoveryContentLifecycle } from \"./recovery-content-lifecycle.ts\";\nimport { RecoveryAssistantMessageEventStream } from \"./recovery-event-stream.ts\";\nimport { RecoveryNativeProjection } from \"./recovery-native-projection.ts\";\nimport { type RecoveryStreamFailure, terminateRecoveryStreamForFailure } from \"./recovery-stream-failure.ts\";\nimport { RecoveryStreamTerminal } from \"./recovery-stream-terminal.ts\";\nimport { RecoveryTextProjection } from \"./recovery-text-projection.ts\";\nimport { StreamMessageProjection } from \"./stream-wrapper-shared.ts\";\n\nexport function wrapStreamWithInvokeRecovery(\n\tinnerStream: AssistantMessageEventStream,\n\ttools: readonly Tool[],\n): AssistantMessageEventStream {\n\tconst outerStream = new RecoveryAssistantMessageEventStream();\n\n\tvoid (async (): Promise<void> => {\n\t\tconst innerIterator = innerStream[Symbol.asyncIterator]();\n\t\tconst innerEvents: AsyncIterable<AssistantMessageEvent> = { [Symbol.asyncIterator]: () => innerIterator };\n\t\tlet projection: StreamMessageProjection | null = null;\n\t\tlet nativeProjection: RecoveryNativeProjection | null = null;\n\t\tlet textProjection: RecoveryTextProjection | null = null;\n\t\tlet sawToolCall = false;\n\t\tlet cancellationRequested = false;\n\t\tconst contentLifecycle = new RecoveryContentLifecycle();\n\t\tconst terminal = new RecoveryStreamTerminal(outerStream);\n\n\t\tconst finishText = (): void => {\n\t\t\tif (!textProjection) return;\n\t\t\tsawToolCall = textProjection.finish() || sawToolCall;\n\t\t\ttextProjection = null;\n\t\t};\n\n\t\touterStream.setCancellationHandler(async () => {\n\t\t\tcancellationRequested = true;\n\t\t\ttry {\n\t\t\t\tawait innerIterator.return?.();\n\t\t\t\tfinishText();\n\t\t\t\tterminal.cancelled(projection);\n\t\t\t} catch (error) {\n\t\t\t\tfinishText();\n\t\t\t\tterminal.iteratorFailure(projection, error);\n\t\t\t\tthrow error;\n\t\t\t}\n\t\t});\n\n\t\tconst terminateForFailure = (source: AssistantMessage, failure: RecoveryStreamFailure): void => {\n\t\t\tif (!projection || !outerStream.markSourceClosed()) return;\n\t\t\tfinishText();\n\t\t\tterminateRecoveryStreamForFailure(outerStream, projection, source, failure);\n\t\t};\n\n\t\tconst prepareContentEvent = (source: AssistantMessage, contentIndex: number): boolean => {\n\t\t\tif (!projection || !nativeProjection) return false;\n\t\t\tif (!Number.isSafeInteger(contentIndex) || contentIndex < 0 || contentIndex >= source.content.length) {\n\t\t\t\tterminateForFailure(source, \"invalid_native_event_order\");\n\t\t\t\treturn false;\n\t\t\t}\n\t\t\tprojection.sync(source);\n\t\t\tnativeProjection.reserveVisibleIds(source);\n\t\t\tif (nativeProjection.synchronizeLower(source, contentIndex)) return true;\n\t\t\tterminateForFailure(source, \"collision\");\n\t\t\treturn false;\n\t\t};\n\n\t\tconst canStart = (\n\t\t\tsource: AssistantMessage,\n\t\t\tcontentIndex: number,\n\t\t\tkind: RecoveryContentKind,\n\t\t\tfailure: RecoveryStreamFailure = \"invalid_content_event_order\",\n\t\t): boolean => {\n\t\t\tif (!Number.isSafeInteger(contentIndex) || contentIndex < 0 || contentIndex >= source.content.length) {\n\t\t\t\tterminateForFailure(source, failure);\n\t\t\t\treturn false;\n\t\t\t}\n\t\t\tconst block = source.content[contentIndex];\n\t\t\tconst blockMatches =\n\t\t\t\t(kind === \"text\" && block?.type === \"text\") ||\n\t\t\t\t(kind === \"thinking\" && block?.type === \"thinking\") ||\n\t\t\t\t(kind === \"toolCall\" && block?.type === \"toolCall\");\n\t\t\tif (blockMatches && contentLifecycle.canStart(contentIndex)) return true;\n\t\t\tterminateForFailure(source, failure);\n\t\t\treturn false;\n\t\t};\n\n\t\tconst isActive = (\n\t\t\tsource: AssistantMessage,\n\t\t\tcontentIndex: number,\n\t\t\tkind: RecoveryContentKind,\n\t\t\tfailure: RecoveryStreamFailure = \"invalid_content_event_order\",\n\t\t): boolean => {\n\t\t\tif (contentLifecycle.isActive(contentIndex, kind)) return true;\n\t\t\tterminateForFailure(source, failure);\n\t\t\treturn false;\n\t\t};\n\n\t\tconst synchronizeTerminal = (source: AssistantMessage): boolean => {\n\t\t\tprojection ??= new StreamMessageProjection(outerStream, source, { preserveSourceMetadata: true });\n\t\t\tnativeProjection ??= new RecoveryNativeProjection(outerStream, projection.message);\n\t\t\tprojection.sync(source);\n\t\t\tnativeProjection.reserveVisibleIds(source);\n\t\t\tfinishText();\n\t\t\tif (nativeProjection.synchronizeRemaining(source)) return true;\n\t\t\tterminateForFailure(source, \"collision\");\n\t\t\treturn false;\n\t\t};\n\n\t\ttry {\n\t\t\tfor await (const event of innerEvents) {\n\t\t\t\tif (cancellationRequested) continue;\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\t\tpreserveSourceMetadata: true,\n\t\t\t\t\t\t});\n\t\t\t\t\t\tnativeProjection = new RecoveryNativeProjection(outerStream, projection.message);\n\t\t\t\t\t\tnativeProjection.reserveVisibleIds(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 (!canStart(event.partial, event.contentIndex, \"text\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection || !nativeProjection)\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\tconst nextText = new RecoveryTextProjection(tools, projection, nativeProjection, event.contentIndex);\n\t\t\t\t\t\tif (!nextText.start(event.partial)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\ttextProjection = nextText;\n\t\t\t\t\t\tcontentLifecycle.start(event.contentIndex, \"text\");\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 (!isActive(event.partial, event.contentIndex, \"text\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !textProjection) return;\n\t\t\t\t\t\tsawToolCall = textProjection.feed(event.delta) || sawToolCall;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_end\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"text\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex)) return;\n\t\t\t\t\t\tfinishText();\n\t\t\t\t\t\tcontentLifecycle.end(event.contentIndex, \"text\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"done\":\n\t\t\t\t\t\tif (!synchronizeTerminal(event.message) || !projection) return;\n\t\t\t\t\t\tif (!outerStream.markSourceClosed()) return;\n\t\t\t\t\t\tterminal.done(projection, event.message, sawToolCall, event.reason);\n\t\t\t\t\t\treturn;\n\t\t\t\t\tcase \"error\":\n\t\t\t\t\t\tif (!projection) {\n\t\t\t\t\t\t\tif (!outerStream.markSourceClosed()) return;\n\t\t\t\t\t\t\touterStream.push(event);\n\t\t\t\t\t\t\touterStream.end();\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (!synchronizeTerminal(event.error)) return;\n\t\t\t\t\t\tif (!outerStream.markSourceClosed()) return;\n\t\t\t\t\t\tterminal.sourceError(projection, event.error, sawToolCall, event.reason);\n\t\t\t\t\t\treturn;\n\t\t\t\t\tcase \"toolcall_start\": {\n\t\t\t\t\t\tif (!canStart(event.partial, event.contentIndex, \"toolCall\", \"invalid_native_event_order\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection) return;\n\t\t\t\t\t\tconst status = nativeProjection.projectNativeStart(event.partial, event.contentIndex);\n\t\t\t\t\t\tif (status !== \"projected\") {\n\t\t\t\t\t\t\tterminateForFailure(\n\t\t\t\t\t\t\t\tevent.partial,\n\t\t\t\t\t\t\t\tstatus === \"collision\" ? \"collision\" : \"invalid_native_event_order\",\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcontentLifecycle.start(event.contentIndex, \"toolCall\");\n\t\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"toolcall_delta\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"toolCall\", \"invalid_native_event_order\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection) return;\n\t\t\t\t\t\tif (!nativeProjection.projectNativeDelta(event.partial, event.contentIndex, event.delta)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"toolcall_end\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"toolCall\", \"invalid_native_event_order\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection) return;\n\t\t\t\t\t\tif (!nativeProjection.projectNativeEnd(event.contentIndex, event.toolCall)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcontentLifecycle.end(event.contentIndex, \"toolCall\");\n\t\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_start\": {\n\t\t\t\t\t\tif (!canStart(event.partial, event.contentIndex, \"thinking\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection || !nativeProjection)\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\tconst outerIndex = projection.startThinking(event.contentIndex, event.partial);\n\t\t\t\t\t\tif (!nativeProjection.recordProjectedBlock(event.contentIndex, outerIndex)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcontentLifecycle.start(event.contentIndex, \"thinking\");\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 (!isActive(event.partial, event.contentIndex, \"thinking\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection) return;\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 (!isActive(event.partial, event.contentIndex, \"thinking\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection) return;\n\t\t\t\t\t\tprojection.finishThinking(event.contentIndex, event.content, event.partial);\n\t\t\t\t\t\tcontentLifecycle.end(event.contentIndex, \"thinking\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif (cancellationRequested || !outerStream.markSourceClosed()) return;\n\t\t\tfinishText();\n\t\t\tterminal.exhausted(projection);\n\t\t} catch (error) {\n\t\t\tif (cancellationRequested || !outerStream.markSourceClosed()) return;\n\t\t\tfinishText();\n\t\t\tterminal.iteratorFailure(projection, error);\n\t\t}\n\t})();\n\n\treturn outerStream;\n}\n"]}

package/dist/tool-call-middleware/recovery-text-projection.d.ts

@@ -0,0 +1,23 @@
+import type { AssistantMessage, Tool } from "../types.ts";
+import type { RecoveryNativeProjection } from "./recovery-native-projection.ts";
+import type { StreamMessageProjection } from "./stream-wrapper-shared.ts";
+/** Owns one ordinary assistant text block's mask/parser/projection lifecycle. */
+export declare class RecoveryTextProjection {
+    private readonly projection;
+    private readonly nativeProjection;
+    private readonly innerIndex;
+    private readonly parser;
+    private readonly mask;
+    private activeInvoke;
+    private textBuffer;
+    private sawToolCall;
+    private finished;
+    constructor(tools: readonly Tool[], projection: StreamMessageProjection, nativeProjection: RecoveryNativeProjection, innerIndex: number);
+    start(source: AssistantMessage): boolean;
+    feed(text: string): boolean;
+    finish(): boolean;
+    private processSegment;
+    private projectParserEvents;
+    private flushText;
+}
+//# sourceMappingURL=recovery-text-projection.d.ts.map

package/dist/tool-call-middleware/recovery-text-projection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"recovery-text-projection.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-text-projection.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAI1D,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAChF,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAG1E,iFAAiF;AACjF,qBAAa,sBAAsB;IAClC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA0B;IACrD,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAA2B;IAC5D,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA2D;IAClF,OAAO,CAAC,QAAQ,CAAC,IAAI,CAA4B;IACjD,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,UAAU,CAAM;IACxB,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,QAAQ,CAAS;IAEzB,YACC,KAAK,EAAE,SAAS,IAAI,EAAE,EACtB,UAAU,EAAE,uBAAuB,EACnC,gBAAgB,EAAE,wBAAwB,EAC1C,UAAU,EAAE,MAAM,EAMlB;IAED,KAAK,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAGvC;IAED,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAQ1B;IAED,MAAM,IAAI,OAAO,CAShB;IAED,OAAO,CAAC,cAAc;IAMtB,OAAO,CAAC,mBAAmB;IAkB3B,OAAO,CAAC,SAAS;CAKjB"}

package/dist/tool-call-middleware/recovery-text-projection.js

@@ -0,0 +1,77 @@
+import { createAntmlInvokeRecoveryStreamParser } from "./protocols/antml/recovery-stream.js";
+import { createRecoveryCodeMask } from "./recovery-code-mask.js";
+import { appendRecoveryDiagnostic } from "./recovery-diagnostics.js";
+/** Owns one ordinary assistant text block's mask/parser/projection lifecycle. */
+export class RecoveryTextProjection {
+    constructor(tools, projection, nativeProjection, innerIndex) {
+        this.mask = createRecoveryCodeMask();
+        this.activeInvoke = false;
+        this.textBuffer = "";
+        this.sawToolCall = false;
+        this.finished = false;
+        this.projection = projection;
+        this.nativeProjection = nativeProjection;
+        this.innerIndex = innerIndex;
+        this.parser = createAntmlInvokeRecoveryStreamParser(tools);
+    }
+    start(source) {
+        const outerIndex = this.projection.startText(this.innerIndex, source.content[this.innerIndex]);
+        return this.nativeProjection.startText(this.innerIndex, outerIndex);
+    }
+    feed(text) {
+        for (let index = 0; index < text.length; index += 1) {
+            const character = text.charAt(index);
+            const options = this.activeInvoke ? { activeInvoke: true } : undefined;
+            for (const segment of this.mask.feed(character, options))
+                this.processSegment(segment);
+        }
+        this.flushText();
+        return this.sawToolCall;
+    }
+    finish() {
+        if (this.finished)
+            return this.sawToolCall;
+        for (const segment of this.mask.finish())
+            this.processSegment(segment);
+        this.projectParserEvents(this.parser.finish());
+        this.flushText();
+        this.projection.finishText();
+        this.nativeProjection.extendText(this.innerIndex);
+        this.finished = true;
+        return this.sawToolCall;
+    }
+    processSegment(segment) {
+        if (segment.recoveryBoundary)
+            this.projectParserEvents(this.parser.interrupt());
+        if (segment.scan)
+            this.projectParserEvents(this.parser.feed(segment.text));
+        else
+            this.textBuffer += segment.text;
+    }
+    projectParserEvents(events) {
+        for (const event of this.nativeProjection.assignRecoveredIds(events)) {
+            if (event.type === "text") {
+                this.textBuffer += event.text;
+                continue;
+            }
+            this.flushText();
+            const result = this.projection.projectParserEvents([event]);
+            this.sawToolCall = this.sawToolCall || result.sawToolCall;
+            if (event.type === "toolcall_start")
+                this.activeInvoke = true;
+            if (event.type === "toolcall_end") {
+                this.activeInvoke = false;
+                for (const toolCall of result.completedToolCalls)
+                    appendRecoveryDiagnostic(this.projection, toolCall);
+            }
+        }
+        this.nativeProjection.extendText(this.innerIndex);
+    }
+    flushText() {
+        if (this.textBuffer.length === 0)
+            return;
+        this.projection.projectParserEvents([{ type: "text", text: this.textBuffer }]);
+        this.textBuffer = "";
+    }
+}
+//# sourceMappingURL=recovery-text-projection.js.map

package/dist/tool-call-middleware/recovery-text-projection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"recovery-text-projection.js","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-text-projection.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,qCAAqC,EAAE,MAAM,sCAAsC,CAAC;AAC7F,OAAO,EAAE,sBAAsB,EAAgC,MAAM,yBAAyB,CAAC;AAC/F,OAAO,EAAE,wBAAwB,EAAE,MAAM,2BAA2B,CAAC;AAKrE,iFAAiF;AACjF,MAAM,OAAO,sBAAsB;IAWlC,YACC,KAAsB,EACtB,UAAmC,EACnC,gBAA0C,EAC1C,UAAkB;QAVF,SAAI,GAAG,sBAAsB,EAAE,CAAC;QACzC,iBAAY,GAAG,KAAK,CAAC;QACrB,eAAU,GAAG,EAAE,CAAC;QAChB,gBAAW,GAAG,KAAK,CAAC;QACpB,aAAQ,GAAG,KAAK,CAAC;QAQxB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,qCAAqC,CAAC,KAAK,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,MAAwB;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;QAC/F,OAAO,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IACrE,CAAC;IAED,IAAI,CAAC,IAAY;QAChB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YACrD,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACrC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;YACvE,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC;gBAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACxF,CAAC;QACD,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,OAAO,IAAI,CAAC,WAAW,CAAC;IACzB,CAAC;IAED,MAAM;QACL,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC,WAAW,CAAC;QAC3C,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACvE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/C,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,CAAC;QAC7B,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,OAAO,IAAI,CAAC,WAAW,CAAC;IACzB,CAAC;IAEO,cAAc,CAAC,OAAgC;QACtD,IAAI,OAAO,CAAC,gBAAgB;YAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;QAChF,IAAI,OAAO,CAAC,IAAI;YAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;;YACtE,IAAI,CAAC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IACtC,CAAC;IAEO,mBAAmB,CAAC,MAAoC;QAC/D,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,CAAC;YACtE,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC3B,IAAI,CAAC,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC;gBAC9B,SAAS;YACV,CAAC;YACD,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;YAC5D,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;YAC1D,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB;gBAAE,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YAC9D,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;gBACnC,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;gBAC1B,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,kBAAkB;oBAAE,wBAAwB,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YACvG,CAAC;QACF,CAAC;QACD,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACnD,CAAC;IAEO,SAAS;QAChB,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QACzC,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QAC/E,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;IACtB,CAAC;CACD","sourcesContent":["import type { AssistantMessage, Tool } from \"../types.ts\";\nimport { createAntmlInvokeRecoveryStreamParser } from \"./protocols/antml/recovery-stream.ts\";\nimport { createRecoveryCodeMask, type RecoveryCodeMaskSegment } from \"./recovery-code-mask.ts\";\nimport { appendRecoveryDiagnostic } from \"./recovery-diagnostics.ts\";\nimport type { RecoveryNativeProjection } from \"./recovery-native-projection.ts\";\nimport type { StreamMessageProjection } from \"./stream-wrapper-shared.ts\";\nimport type { StreamParserEvent } from \"./types.ts\";\n\n/** Owns one ordinary assistant text block's mask/parser/projection lifecycle. */\nexport class RecoveryTextProjection {\n\tprivate readonly projection: StreamMessageProjection;\n\tprivate readonly nativeProjection: RecoveryNativeProjection;\n\tprivate readonly innerIndex: number;\n\tprivate readonly parser: ReturnType<typeof createAntmlInvokeRecoveryStreamParser>;\n\tprivate readonly mask = createRecoveryCodeMask();\n\tprivate activeInvoke = false;\n\tprivate textBuffer = \"\";\n\tprivate sawToolCall = false;\n\tprivate finished = false;\n\n\tconstructor(\n\t\ttools: readonly Tool[],\n\t\tprojection: StreamMessageProjection,\n\t\tnativeProjection: RecoveryNativeProjection,\n\t\tinnerIndex: number,\n\t) {\n\t\tthis.projection = projection;\n\t\tthis.nativeProjection = nativeProjection;\n\t\tthis.innerIndex = innerIndex;\n\t\tthis.parser = createAntmlInvokeRecoveryStreamParser(tools);\n\t}\n\n\tstart(source: AssistantMessage): boolean {\n\t\tconst outerIndex = this.projection.startText(this.innerIndex, source.content[this.innerIndex]);\n\t\treturn this.nativeProjection.startText(this.innerIndex, outerIndex);\n\t}\n\n\tfeed(text: string): boolean {\n\t\tfor (let index = 0; index < text.length; index += 1) {\n\t\t\tconst character = text.charAt(index);\n\t\t\tconst options = this.activeInvoke ? { activeInvoke: true } : undefined;\n\t\t\tfor (const segment of this.mask.feed(character, options)) this.processSegment(segment);\n\t\t}\n\t\tthis.flushText();\n\t\treturn this.sawToolCall;\n\t}\n\n\tfinish(): boolean {\n\t\tif (this.finished) return this.sawToolCall;\n\t\tfor (const segment of this.mask.finish()) this.processSegment(segment);\n\t\tthis.projectParserEvents(this.parser.finish());\n\t\tthis.flushText();\n\t\tthis.projection.finishText();\n\t\tthis.nativeProjection.extendText(this.innerIndex);\n\t\tthis.finished = true;\n\t\treturn this.sawToolCall;\n\t}\n\n\tprivate processSegment(segment: RecoveryCodeMaskSegment): void {\n\t\tif (segment.recoveryBoundary) this.projectParserEvents(this.parser.interrupt());\n\t\tif (segment.scan) this.projectParserEvents(this.parser.feed(segment.text));\n\t\telse this.textBuffer += segment.text;\n\t}\n\n\tprivate projectParserEvents(events: readonly StreamParserEvent[]): void {\n\t\tfor (const event of this.nativeProjection.assignRecoveredIds(events)) {\n\t\t\tif (event.type === \"text\") {\n\t\t\t\tthis.textBuffer += event.text;\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tthis.flushText();\n\t\t\tconst result = this.projection.projectParserEvents([event]);\n\t\t\tthis.sawToolCall = this.sawToolCall || result.sawToolCall;\n\t\t\tif (event.type === \"toolcall_start\") this.activeInvoke = true;\n\t\t\tif (event.type === \"toolcall_end\") {\n\t\t\t\tthis.activeInvoke = false;\n\t\t\t\tfor (const toolCall of result.completedToolCalls) appendRecoveryDiagnostic(this.projection, toolCall);\n\t\t\t}\n\t\t}\n\t\tthis.nativeProjection.extendText(this.innerIndex);\n\t}\n\n\tprivate flushText(): void {\n\t\tif (this.textBuffer.length === 0) return;\n\t\tthis.projection.projectParserEvents([{ type: \"text\", text: this.textBuffer }]);\n\t\tthis.textBuffer = \"\";\n\t}\n}\n"]}

package/dist/tool-call-middleware/stream-message-metadata.d.ts

@@ -0,0 +1,5 @@
+import type { AssistantMessage } from "../types.ts";
+import type { AssistantMessageDiagnostic } from "../utils/diagnostics.ts";
+export declare function cloneAssistantMessageMetadata(source: AssistantMessage, content: AssistantMessage["content"], projectedDiagnostics: readonly AssistantMessageDiagnostic[], preserveAll: boolean): AssistantMessage;
+export declare function syncAssistantMessageMetadata(target: AssistantMessage, source: AssistantMessage, projectedDiagnostics: readonly AssistantMessageDiagnostic[], preserveAll: boolean): void;
+//# sourceMappingURL=stream-message-metadata.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"stream-message-metadata.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-message-metadata.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAS,MAAM,aAAa,CAAC;AAC3D,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC;AAsC1E,wBAAgB,6BAA6B,CAC5C,MAAM,EAAE,gBAAgB,EACxB,OAAO,EAAE,gBAAgB,CAAC,SAAS,CAAC,EACpC,oBAAoB,EAAE,SAAS,0BAA0B,EAAE,EAC3D,WAAW,EAAE,OAAO,GAClB,gBAAgB,CAOlB;AAED,wBAAgB,4BAA4B,CAC3C,MAAM,EAAE,gBAAgB,EACxB,MAAM,EAAE,gBAAgB,EACxB,oBAAoB,EAAE,SAAS,0BAA0B,EAAE,EAC3D,WAAW,EAAE,OAAO,GAClB,IAAI,CAgBN"}

package/dist/tool-call-middleware/stream-message-metadata.js

@@ -0,0 +1,63 @@
+function cloneUsage(usage) {
+    return { ...usage, cost: { ...usage.cost } };
+}
+function cloneLegacyUsage(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 cloneLegacyMessage(source, content) {
+    return {
+        role: "assistant",
+        api: source.api,
+        provider: source.provider,
+        model: source.model,
+        responseId: source.responseId,
+        content,
+        usage: cloneLegacyUsage(source.usage),
+        stopReason: source.stopReason,
+        errorMessage: source.errorMessage,
+        timestamp: source.timestamp,
+    };
+}
+export function cloneAssistantMessageMetadata(source, content, projectedDiagnostics, preserveAll) {
+    if (!preserveAll)
+        return cloneLegacyMessage(source, content);
+    const message = { ...source, content, usage: cloneUsage(source.usage) };
+    const diagnostics = [...(source.diagnostics ?? []), ...projectedDiagnostics];
+    if (diagnostics.length > 0)
+        message.diagnostics = diagnostics;
+    else
+        delete message.diagnostics;
+    return message;
+}
+export function syncAssistantMessageMetadata(target, source, projectedDiagnostics, preserveAll) {
+    if (!preserveAll) {
+        target.api = source.api;
+        target.provider = source.provider;
+        target.model = source.model;
+        target.responseId = source.responseId;
+        target.usage = cloneLegacyUsage(source.usage);
+        target.stopReason = source.stopReason;
+        target.errorMessage = source.errorMessage;
+        target.timestamp = source.timestamp;
+        return;
+    }
+    const content = target.content;
+    const targetRecord = target;
+    for (const key of Object.keys(targetRecord))
+        delete targetRecord[key];
+    Object.assign(targetRecord, cloneAssistantMessageMetadata(source, content, projectedDiagnostics, true));
+}
+//# sourceMappingURL=stream-message-metadata.js.map

package/dist/tool-call-middleware/stream-message-metadata.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-message-metadata.js","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-message-metadata.ts"],"names":[],"mappings":"AAGA,SAAS,UAAU,CAAC,KAAY;IAC/B,OAAO,EAAE,GAAG,KAAK,EAAE,IAAI,EAAE,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;AAC9C,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAY;IACrC,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,MAAwB,EAAE,OAAoC;IACzF,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,GAAG,EAAE,MAAM,CAAC,GAAG;QACf,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,OAAO;QACP,KAAK,EAAE,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC;QACrC,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,SAAS,EAAE,MAAM,CAAC,SAAS;KAC3B,CAAC;AACH,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC5C,MAAwB,EACxB,OAAoC,EACpC,oBAA2D,EAC3D,WAAoB;IAEpB,IAAI,CAAC,WAAW;QAAE,OAAO,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC7D,MAAM,OAAO,GAAqB,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;IAC1F,MAAM,WAAW,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC,EAAE,GAAG,oBAAoB,CAAC,CAAC;IAC7E,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC;;QACzD,OAAO,OAAO,CAAC,WAAW,CAAC;IAChC,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,4BAA4B,CAC3C,MAAwB,EACxB,MAAwB,EACxB,oBAA2D,EAC3D,WAAoB;IAEpB,IAAI,CAAC,WAAW,EAAE,CAAC;QAClB,MAAM,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACxB,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAClC,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC5B,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QACtC,MAAM,CAAC,KAAK,GAAG,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QACtC,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QAC1C,MAAM,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACpC,OAAO;IACR,CAAC;IACD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;IAC/B,MAAM,YAAY,GAAG,MAA4C,CAAC;IAClE,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC;QAAE,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC;IACtE,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE,6BAA6B,CAAC,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,IAAI,CAAC,CAAC,CAAC;AACzG,CAAC","sourcesContent":["import type { AssistantMessage, Usage } from \"../types.ts\";\nimport type { AssistantMessageDiagnostic } from \"../utils/diagnostics.ts\";\n\nfunction cloneUsage(usage: Usage): Usage {\n\treturn { ...usage, cost: { ...usage.cost } };\n}\n\nfunction cloneLegacyUsage(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 cloneLegacyMessage(source: AssistantMessage, content: AssistantMessage[\"content\"]): AssistantMessage {\n\treturn {\n\t\trole: \"assistant\",\n\t\tapi: source.api,\n\t\tprovider: source.provider,\n\t\tmodel: source.model,\n\t\tresponseId: source.responseId,\n\t\tcontent,\n\t\tusage: cloneLegacyUsage(source.usage),\n\t\tstopReason: source.stopReason,\n\t\terrorMessage: source.errorMessage,\n\t\ttimestamp: source.timestamp,\n\t};\n}\n\nexport function cloneAssistantMessageMetadata(\n\tsource: AssistantMessage,\n\tcontent: AssistantMessage[\"content\"],\n\tprojectedDiagnostics: readonly AssistantMessageDiagnostic[],\n\tpreserveAll: boolean,\n): AssistantMessage {\n\tif (!preserveAll) return cloneLegacyMessage(source, content);\n\tconst message: AssistantMessage = { ...source, content, usage: cloneUsage(source.usage) };\n\tconst diagnostics = [...(source.diagnostics ?? []), ...projectedDiagnostics];\n\tif (diagnostics.length > 0) message.diagnostics = diagnostics;\n\telse delete message.diagnostics;\n\treturn message;\n}\n\nexport function syncAssistantMessageMetadata(\n\ttarget: AssistantMessage,\n\tsource: AssistantMessage,\n\tprojectedDiagnostics: readonly AssistantMessageDiagnostic[],\n\tpreserveAll: boolean,\n): void {\n\tif (!preserveAll) {\n\t\ttarget.api = source.api;\n\t\ttarget.provider = source.provider;\n\t\ttarget.model = source.model;\n\t\ttarget.responseId = source.responseId;\n\t\ttarget.usage = cloneLegacyUsage(source.usage);\n\t\ttarget.stopReason = source.stopReason;\n\t\ttarget.errorMessage = source.errorMessage;\n\t\ttarget.timestamp = source.timestamp;\n\t\treturn;\n\t}\n\tconst content = target.content;\n\tconst targetRecord = target as unknown as Record<string, unknown>;\n\tfor (const key of Object.keys(targetRecord)) delete targetRecord[key];\n\tObject.assign(targetRecord, cloneAssistantMessageMetadata(source, content, projectedDiagnostics, true));\n}\n"]}

package/dist/tool-call-middleware/stream-thinking-projection.d.ts

@@ -0,0 +1,13 @@
+import type { AssistantMessage, AssistantMessageEventStream } from "../types.ts";
+/** Replays canonical thinking events without exposing their content to text parsers. */
+export declare class StreamThinkingProjection {
+    private readonly stream;
+    private readonly message;
+    private readonly outerIndexByInnerIndex;
+    private readonly preserveSourceMetadata;
+    constructor(stream: AssistantMessageEventStream, message: AssistantMessage, preserveSourceMetadata: boolean);
+    start(innerIndex: number, source: AssistantMessage): number;
+    delta(innerIndex: number, delta: string, source: AssistantMessage): void;
+    end(innerIndex: number, content: string, source: AssistantMessage): void;
+}
+//# sourceMappingURL=stream-thinking-projection.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"stream-thinking-projection.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-thinking-projection.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,2BAA2B,EAAmB,MAAM,aAAa,CAAC;AAMlG,wFAAwF;AACxF,qBAAa,wBAAwB;IACpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8B;IACrD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAA6B;IACpE,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAU;IAEjD,YAAY,MAAM,EAAE,2BAA2B,EAAE,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,OAAO,EAI1G;IAED,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,MAAM,CAW1D;IAED,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI,CAWvE;IAED,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI,CAWvE;CACD"}

package/dist/tool-call-middleware/stream-thinking-projection.js

@@ -0,0 +1,52 @@
+function cloneThinking(block, thinking = block.thinking) {
+    return { ...block, thinking };
+}
+/** Replays canonical thinking events without exposing their content to text parsers. */
+export class StreamThinkingProjection {
+    constructor(stream, message, preserveSourceMetadata) {
+        this.outerIndexByInnerIndex = new Map();
+        this.stream = stream;
+        this.message = message;
+        this.preserveSourceMetadata = preserveSourceMetadata;
+    }
+    start(innerIndex, source) {
+        const sourceBlock = source.content[innerIndex];
+        const block = this.preserveSourceMetadata && sourceBlock?.type === "thinking"
+            ? cloneThinking(sourceBlock)
+            : { type: "thinking", thinking: "" };
+        const outerIndex = this.message.content.length;
+        this.message.content.push(block);
+        this.outerIndexByInnerIndex.set(innerIndex, outerIndex);
+        this.stream.push({ type: "thinking_start", contentIndex: outerIndex, partial: this.message });
+        return outerIndex;
+    }
+    delta(innerIndex, delta, source) {
+        const outerIndex = this.outerIndexByInnerIndex.get(innerIndex);
+        if (outerIndex == null)
+            return;
+        const current = this.message.content[outerIndex];
+        if (current?.type !== "thinking")
+            return;
+        const sourceBlock = source.content[innerIndex];
+        this.message.content[outerIndex] =
+            this.preserveSourceMetadata && sourceBlock?.type === "thinking"
+                ? cloneThinking(sourceBlock)
+                : cloneThinking(current, current.thinking + delta);
+        this.stream.push({ type: "thinking_delta", contentIndex: outerIndex, delta, partial: this.message });
+    }
+    end(innerIndex, content, source) {
+        const outerIndex = this.outerIndexByInnerIndex.get(innerIndex);
+        if (outerIndex == null)
+            return;
+        const current = this.message.content[outerIndex];
+        if (current?.type !== "thinking")
+            return;
+        const sourceBlock = source.content[innerIndex];
+        this.message.content[outerIndex] =
+            this.preserveSourceMetadata && sourceBlock?.type === "thinking"
+                ? cloneThinking(sourceBlock, content)
+                : cloneThinking(current, content);
+        this.stream.push({ type: "thinking_end", contentIndex: outerIndex, content, partial: this.message });
+    }
+}
+//# sourceMappingURL=stream-thinking-projection.js.map

package/dist/tool-call-middleware/stream-thinking-projection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-thinking-projection.js","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-thinking-projection.ts"],"names":[],"mappings":"AAEA,SAAS,aAAa,CAAC,KAAsB,EAAE,QAAQ,GAAG,KAAK,CAAC,QAAQ;IACvE,OAAO,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,CAAC;AAC/B,CAAC;AAED,wFAAwF;AACxF,MAAM,OAAO,wBAAwB;IAMpC,YAAY,MAAmC,EAAE,OAAyB,EAAE,sBAA+B;QAH1F,2BAAsB,GAAG,IAAI,GAAG,EAAkB,CAAC;QAInE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,UAAkB,EAAE,MAAwB;QACjD,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC/C,MAAM,KAAK,GACV,IAAI,CAAC,sBAAsB,IAAI,WAAW,EAAE,IAAI,KAAK,UAAU;YAC9D,CAAC,CAAC,aAAa,CAAC,WAAW,CAAC;YAC5B,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;QACvC,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;QAC/C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QACxD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC9F,OAAO,UAAU,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,UAAkB,EAAE,KAAa,EAAE,MAAwB;QAChE,MAAM,UAAU,GAAG,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC/D,IAAI,UAAU,IAAI,IAAI;YAAE,OAAO;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACjD,IAAI,OAAO,EAAE,IAAI,KAAK,UAAU;YAAE,OAAO;QACzC,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC;YAC/B,IAAI,CAAC,sBAAsB,IAAI,WAAW,EAAE,IAAI,KAAK,UAAU;gBAC9D,CAAC,CAAC,aAAa,CAAC,WAAW,CAAC;gBAC5B,CAAC,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,GAAG,KAAK,CAAC,CAAC;QACrD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IACtG,CAAC;IAED,GAAG,CAAC,UAAkB,EAAE,OAAe,EAAE,MAAwB;QAChE,MAAM,UAAU,GAAG,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC/D,IAAI,UAAU,IAAI,IAAI;YAAE,OAAO;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACjD,IAAI,OAAO,EAAE,IAAI,KAAK,UAAU;YAAE,OAAO;QACzC,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC;YAC/B,IAAI,CAAC,sBAAsB,IAAI,WAAW,EAAE,IAAI,KAAK,UAAU;gBAC9D,CAAC,CAAC,aAAa,CAAC,WAAW,EAAE,OAAO,CAAC;gBACrC,CAAC,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACpC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IACtG,CAAC;CACD","sourcesContent":["import type { AssistantMessage, AssistantMessageEventStream, ThinkingContent } from \"../types.ts\";\n\nfunction cloneThinking(block: ThinkingContent, thinking = block.thinking): ThinkingContent {\n\treturn { ...block, thinking };\n}\n\n/** Replays canonical thinking events without exposing their content to text parsers. */\nexport class StreamThinkingProjection {\n\tprivate readonly stream: AssistantMessageEventStream;\n\tprivate readonly message: AssistantMessage;\n\tprivate readonly outerIndexByInnerIndex = new Map<number, number>();\n\tprivate readonly preserveSourceMetadata: boolean;\n\n\tconstructor(stream: AssistantMessageEventStream, message: AssistantMessage, preserveSourceMetadata: boolean) {\n\t\tthis.stream = stream;\n\t\tthis.message = message;\n\t\tthis.preserveSourceMetadata = preserveSourceMetadata;\n\t}\n\n\tstart(innerIndex: number, source: AssistantMessage): number {\n\t\tconst sourceBlock = source.content[innerIndex];\n\t\tconst block: ThinkingContent =\n\t\t\tthis.preserveSourceMetadata && sourceBlock?.type === \"thinking\"\n\t\t\t\t? cloneThinking(sourceBlock)\n\t\t\t\t: { type: \"thinking\", thinking: \"\" };\n\t\tconst outerIndex = this.message.content.length;\n\t\tthis.message.content.push(block);\n\t\tthis.outerIndexByInnerIndex.set(innerIndex, outerIndex);\n\t\tthis.stream.push({ type: \"thinking_start\", contentIndex: outerIndex, partial: this.message });\n\t\treturn outerIndex;\n\t}\n\n\tdelta(innerIndex: number, delta: string, source: AssistantMessage): void {\n\t\tconst outerIndex = this.outerIndexByInnerIndex.get(innerIndex);\n\t\tif (outerIndex == null) return;\n\t\tconst current = this.message.content[outerIndex];\n\t\tif (current?.type !== \"thinking\") return;\n\t\tconst sourceBlock = source.content[innerIndex];\n\t\tthis.message.content[outerIndex] =\n\t\t\tthis.preserveSourceMetadata && sourceBlock?.type === \"thinking\"\n\t\t\t\t? cloneThinking(sourceBlock)\n\t\t\t\t: cloneThinking(current, current.thinking + delta);\n\t\tthis.stream.push({ type: \"thinking_delta\", contentIndex: outerIndex, delta, partial: this.message });\n\t}\n\n\tend(innerIndex: number, content: string, source: AssistantMessage): void {\n\t\tconst outerIndex = this.outerIndexByInnerIndex.get(innerIndex);\n\t\tif (outerIndex == null) return;\n\t\tconst current = this.message.content[outerIndex];\n\t\tif (current?.type !== \"thinking\") return;\n\t\tconst sourceBlock = source.content[innerIndex];\n\t\tthis.message.content[outerIndex] =\n\t\t\tthis.preserveSourceMetadata && sourceBlock?.type === \"thinking\"\n\t\t\t\t? cloneThinking(sourceBlock, content)\n\t\t\t\t: cloneThinking(current, content);\n\t\tthis.stream.push({ type: \"thinking_end\", contentIndex: outerIndex, content, partial: this.message });\n\t}\n}\n"]}

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

@@ -0,0 +1,41 @@
+import type { AssistantMessage, AssistantMessageEventStream, ToolCall } from "../types.ts";
+import type { AssistantMessageDiagnostic } from "../utils/diagnostics.ts";
+import type { StreamParserEvent } from "./types.ts";
+export type ParserProjectionResult = {
+    readonly sawToolCall: boolean;
+    readonly completedToolCalls: readonly ToolCall[];
+};
+type StreamMessageProjectionOptions = {
+    readonly preserveSourceMetadata?: boolean;
+};
+/** Projects parser events into the canonical outer assistant stream. */
+export declare class StreamMessageProjection {
+    readonly message: AssistantMessage;
+    private readonly stream;
+    private currentInnerTextIndex;
+    private readonly textBlockIndexByInnerIndex;
+    private readonly lastTextBlockIndexByInnerIndex;
+    private readonly toolCallIndexByParserIndex;
+    private readonly projectedDiagnostics;
+    private readonly thinking;
+    private readonly preserveSourceMetadata;
+    private source;
+    constructor(stream: AssistantMessageEventStream, source: AssistantMessage, options?: StreamMessageProjectionOptions);
+    sync(source: AssistantMessage): void;
+    appendDiagnostic(diagnostic: AssistantMessageDiagnostic): void;
+    startText(contentIndex: number, sourceBlock?: AssistantMessage["content"][number]): number;
+    startThinking(contentIndex: number, source: AssistantMessage): number;
+    projectThinkingDelta(contentIndex: number, delta: string, source: AssistantMessage): void;
+    finishThinking(contentIndex: number, content: string, source: AssistantMessage): void;
+    projectParserEvents(events: readonly StreamParserEvent[]): ParserProjectionResult;
+    finishText(): void;
+    finalizeDanglingToolCalls(): void;
+    finalize(doneMessage: AssistantMessage, sawToolCall: boolean): AssistantMessage;
+    hasFinalizedToolCallContent(): boolean;
+    private pushText;
+    private startToolCall;
+    private updateToolCall;
+    private endToolCall;
+}
+export {};
+//# sourceMappingURL=stream-wrapper-shared.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"stream-wrapper-shared.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper-shared.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,2BAA2B,EAAe,QAAQ,EAAE,MAAM,aAAa,CAAC;AACxG,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC;AAI1E,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAIpD,MAAM,MAAM,sBAAsB,GAAG;IACpC,QAAQ,CAAC,WAAW,EAAE,OAAO,CAAC;IAC9B,QAAQ,CAAC,kBAAkB,EAAE,SAAS,QAAQ,EAAE,CAAC;CACjD,CAAC;AAEF,KAAK,8BAA8B,GAAG;IAAE,QAAQ,CAAC,sBAAsB,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC;AAUpF,wEAAwE;AACxE,qBAAa,uBAAuB;IACnC,QAAQ,CAAC,OAAO,EAAE,gBAAgB,CAAC;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8B;IACrD,OAAO,CAAC,qBAAqB,CAAuB;IACpD,OAAO,CAAC,QAAQ,CAAC,0BAA0B,CAAoC;IAC/E,OAAO,CAAC,QAAQ,CAAC,8BAA8B,CAA6B;IAC5E,OAAO,CAAC,QAAQ,CAAC,0BAA0B,CAA6B;IACxE,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAoC;IACzE,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA2B;IACpD,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAU;IACjD,OAAO,CAAC,MAAM,CAAmB;IAEjC,YACC,MAAM,EAAE,2BAA2B,EACnC,MAAM,EAAE,gBAAgB,EACxB,OAAO,GAAE,8BAAmC,EAO5C;IAED,IAAI,CAAC,MAAM,EAAE,gBAAgB,GAAG,IAAI,CAGnC;IAED,gBAAgB,CAAC,UAAU,EAAE,0BAA0B,GAAG,IAAI,CAG7D;IAED,SAAS,CAAC,YAAY,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,gBAAgB,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,GAAG,MAAM,CAczF;IAED,aAAa,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,MAAM,CAEpE;IAED,oBAAoB,CAAC,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI,CAExF;IAED,cAAc,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI,CAEpF;IAED,mBAAmB,CAAC,MAAM,EAAE,SAAS,iBAAiB,EAAE,GAAG,sBAAsB,CA2BhF;IAED,UAAU,IAAI,IAAI,CAqBjB;IAED,yBAAyB,IAAI,IAAI,CAiBhC;IAED,QAAQ,CAAC,WAAW,EAAE,gBAAgB,EAAE,WAAW,EAAE,OAAO,GAAG,gBAAgB,CAc9E;IAED,2BAA2B,IAAI,OAAO,CAErC;IAED,OAAO,CAAC,QAAQ;IAgBhB,OAAO,CAAC,aAAa;IA2BrB,OAAO,CAAC,cAAc;IAUtB,OAAO,CAAC,WAAW;CAgBnB"}

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

@@ -0,0 +1,209 @@
+import { parseStreamingJson } from "../utils/json-parse.js";
+import { cloneAssistantMessageMetadata, syncAssistantMessageMetadata } from "./stream-message-metadata.js";
+import { StreamThinkingProjection } from "./stream-thinking-projection.js";
+function isPartialToolCall(block) {
+    return block?.type === "toolCall" && "partialJson" in block && typeof block.partialJson === "string";
+}
+function assertNever(value) {
+    throw new Error(`Unexpected parser event: ${JSON.stringify(value)}`);
+}
+/** Projects parser events into the canonical outer assistant stream. */
+export class StreamMessageProjection {
+    constructor(stream, source, options = {}) {
+        this.currentInnerTextIndex = null;
+        this.textBlockIndexByInnerIndex = new Map();
+        this.lastTextBlockIndexByInnerIndex = new Map();
+        this.toolCallIndexByParserIndex = new Map();
+        this.projectedDiagnostics = [];
+        this.stream = stream;
+        this.source = source;
+        this.preserveSourceMetadata = options.preserveSourceMetadata === true;
+        this.message = cloneAssistantMessageMetadata(source, [], this.projectedDiagnostics, this.preserveSourceMetadata);
+        this.thinking = new StreamThinkingProjection(stream, this.message, this.preserveSourceMetadata);
+    }
+    sync(source) {
+        this.source = source;
+        syncAssistantMessageMetadata(this.message, source, this.projectedDiagnostics, this.preserveSourceMetadata);
+    }
+    appendDiagnostic(diagnostic) {
+        this.projectedDiagnostics.push(diagnostic);
+        syncAssistantMessageMetadata(this.message, this.source, this.projectedDiagnostics, this.preserveSourceMetadata);
+    }
+    startText(contentIndex, sourceBlock) {
+        this.currentInnerTextIndex = contentIndex;
+        const outerIndex = this.message.content.length;
+        if (this.preserveSourceMetadata) {
+            const block = sourceBlock?.type === "text" ? { ...sourceBlock, text: "" } : { type: "text", text: "" };
+            this.message.content.push(block);
+            this.textBlockIndexByInnerIndex.set(contentIndex, outerIndex);
+            this.lastTextBlockIndexByInnerIndex.set(contentIndex, outerIndex);
+        }
+        else {
+            this.textBlockIndexByInnerIndex.set(contentIndex, null);
+        }
+        this.stream.push({ type: "text_start", contentIndex: outerIndex, partial: this.message });
+        return outerIndex;
+    }
+    startThinking(contentIndex, source) {
+        return this.thinking.start(contentIndex, source);
+    }
+    projectThinkingDelta(contentIndex, delta, source) {
+        this.thinking.delta(contentIndex, delta, source);
+    }
+    finishThinking(contentIndex, content, source) {
+        this.thinking.end(contentIndex, content, source);
+    }
+    projectParserEvents(events) {
+        let sawToolCall = false;
+        const completedToolCalls = [];
+        for (const event of events) {
+            switch (event.type) {
+                case "text":
+                    this.pushText(event.text);
+                    break;
+                case "toolcall_start":
+                    sawToolCall = true;
+                    this.startToolCall(event);
+                    break;
+                case "toolcall_delta":
+                    sawToolCall = true;
+                    this.updateToolCall(event.index, event.argumentsDelta);
+                    break;
+                case "toolcall_end": {
+                    sawToolCall = true;
+                    const completed = this.endToolCall(event);
+                    if (completed)
+                        completedToolCalls.push(completed);
+                    break;
+                }
+                default:
+                    assertNever(event);
+            }
+        }
+        return { sawToolCall, completedToolCalls };
+    }
+    finishText() {
+        const outerContentIndex = this.currentInnerTextIndex == null
+            ? null
+            : (this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex) ??
+                (this.preserveSourceMetadata
+                    ? this.lastTextBlockIndexByInnerIndex.get(this.currentInnerTextIndex)
+                    : null) ??
+                null);
+        if (outerContentIndex != null) {
+            const block = this.message.content[outerContentIndex];
+            if (block?.type === "text") {
+                this.stream.push({
+                    type: "text_end",
+                    contentIndex: outerContentIndex,
+                    content: block.text,
+                    partial: this.message,
+                });
+            }
+        }
+        this.currentInnerTextIndex = null;
+    }
+    finalizeDanglingToolCalls() {
+        for (const [contentIndex, block] of this.message.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",
+            };
+            this.message.content[contentIndex] = finalToolCall;
+            for (const [parserIndex, mappedContentIndex] of this.toolCallIndexByParserIndex) {
+                if (mappedContentIndex === contentIndex)
+                    this.toolCallIndexByParserIndex.delete(parserIndex);
+            }
+            this.stream.push({ type: "toolcall_end", contentIndex, toolCall: finalToolCall, partial: this.message });
+        }
+    }
+    finalize(doneMessage, sawToolCall) {
+        const finalMessage = cloneAssistantMessageMetadata(doneMessage, this.message.content, this.projectedDiagnostics, this.preserveSourceMetadata);
+        if ((sawToolCall || this.hasFinalizedToolCallContent()) &&
+            (finalMessage.stopReason === "stop" || finalMessage.stopReason === "length")) {
+            finalMessage.stopReason = "toolUse";
+        }
+        return finalMessage;
+    }
+    hasFinalizedToolCallContent() {
+        return this.message.content.some((block) => block.type === "toolCall" && !isPartialToolCall(block));
+    }
+    pushText(text) {
+        if (this.currentInnerTextIndex == null)
+            return;
+        let contentIndex = this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex) ?? null;
+        if (contentIndex == null) {
+            contentIndex = this.message.content.length;
+            this.message.content.push({ type: "text", text });
+            this.textBlockIndexByInnerIndex.set(this.currentInnerTextIndex, contentIndex);
+            this.lastTextBlockIndexByInnerIndex.set(this.currentInnerTextIndex, contentIndex);
+        }
+        else {
+            const block = this.message.content[contentIndex];
+            if (block?.type !== "text")
+                return;
+            block.text += text;
+        }
+        this.stream.push({ type: "text_delta", contentIndex, delta: text, partial: this.message });
+    }
+    startToolCall(event) {
+        if (this.currentInnerTextIndex != null) {
+            const textIndex = this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex);
+            const block = textIndex == null ? undefined : this.message.content[textIndex];
+            if (textIndex === this.message.content.length - 1 &&
+                block?.type === "text" &&
+                block.text.length === 0 &&
+                block.textSignature === undefined) {
+                this.message.content.pop();
+            }
+        }
+        const contentIndex = this.message.content.length;
+        const toolCall = {
+            type: "toolCall",
+            id: event.id,
+            name: event.name,
+            arguments: {},
+            partialJson: "",
+        };
+        this.message.content.push(toolCall);
+        this.toolCallIndexByParserIndex.set(event.index, contentIndex);
+        if (this.currentInnerTextIndex != null)
+            this.textBlockIndexByInnerIndex.set(this.currentInnerTextIndex, null);
+        this.stream.push({ type: "toolcall_start", contentIndex, partial: this.message });
+    }
+    updateToolCall(parserIndex, delta) {
+        const contentIndex = this.toolCallIndexByParserIndex.get(parserIndex);
+        if (contentIndex == null)
+            return;
+        const block = this.message.content[contentIndex];
+        if (!isPartialToolCall(block))
+            return;
+        block.partialJson += delta;
+        block.arguments = parseStreamingJson(block.partialJson);
+        this.stream.push({ type: "toolcall_delta", contentIndex, delta, partial: this.message });
+    }
+    endToolCall(event) {
+        const contentIndex = this.toolCallIndexByParserIndex.get(event.index);
+        if (contentIndex == null)
+            return undefined;
+        const finalToolCall = {
+            type: "toolCall",
+            id: event.id,
+            name: event.name,
+            arguments: event.arguments,
+            ...(event.incomplete === true ? { incomplete: true } : {}),
+            ...(event.errorMessage === undefined ? {} : { errorMessage: event.errorMessage }),
+        };
+        this.message.content[contentIndex] = finalToolCall;
+        this.toolCallIndexByParserIndex.delete(event.index);
+        this.stream.push({ type: "toolcall_end", contentIndex, toolCall: finalToolCall, partial: this.message });
+        return finalToolCall;
+    }
+}
+//# sourceMappingURL=stream-wrapper-shared.js.map