@code-yeongyu/senpi-ai 2026.7.25-2

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

@@ -0,0 +1,148 @@
+function cloneContentBlock(block) {
+    if (block.type === "toolCall")
+        return { ...block, arguments: { ...block.arguments } };
+    return { ...block };
+}
+function cloneToolCall(toolCall) {
+    return { ...toolCall, arguments: { ...toolCall.arguments } };
+}
+/** Maintains source-index order and native/recovered ID ownership for recovery streams. */
+export class RecoveryNativeProjection {
+    constructor(stream, message) {
+        this.rangesByInnerIndex = new Map();
+        this.nativeLifecycleByInnerIndex = new Map();
+        this.reservedIds = new Set();
+        this.recoveredIds = new Set();
+        this.recoveredIdByParserIndex = new Map();
+        this.highestProjectedInnerIndex = -1;
+        this.nextRecoveredId = 0;
+        this.stream = stream;
+        this.message = message;
+    }
+    reserveVisibleIds(source) {
+        for (const block of source.content) {
+            if (block.type === "toolCall")
+                this.reservedIds.add(block.id);
+        }
+    }
+    synchronizeLower(source, contentIndex) {
+        for (let innerIndex = 0; innerIndex < contentIndex && innerIndex < source.content.length; innerIndex += 1) {
+            if (this.rangesByInnerIndex.has(innerIndex))
+                continue;
+            if (!this.appendUnannounced(source, innerIndex))
+                return false;
+        }
+        return true;
+    }
+    synchronizeRemaining(source) {
+        for (let innerIndex = 0; innerIndex < source.content.length; innerIndex += 1) {
+            if (this.rangesByInnerIndex.has(innerIndex))
+                continue;
+            if (!this.appendUnannounced(source, innerIndex))
+                return false;
+        }
+        return true;
+    }
+    startText(innerIndex, outerIndex) {
+        if (!this.recordRange(innerIndex, { start: outerIndex, end: this.message.content.length }))
+            return false;
+        return true;
+    }
+    recordProjectedBlock(innerIndex, outerIndex) {
+        return this.recordRange(innerIndex, { start: outerIndex, end: outerIndex + 1 });
+    }
+    extendText(innerIndex) {
+        const range = this.rangesByInnerIndex.get(innerIndex);
+        if (range)
+            range.end = this.message.content.length;
+    }
+    projectNativeStart(source, innerIndex) {
+        if (this.nativeLifecycleByInnerIndex.has(innerIndex) || this.rangesByInnerIndex.has(innerIndex))
+            return "invalid";
+        const block = source.content[innerIndex];
+        if (block?.type !== "toolCall" || innerIndex <= this.highestProjectedInnerIndex)
+            return "invalid";
+        if (this.recoveredIds.has(block.id))
+            return "collision";
+        this.reservedIds.add(block.id);
+        const outerIndex = this.message.content.length;
+        this.message.content.push(cloneContentBlock(block));
+        if (!this.recordRange(innerIndex, { start: outerIndex, end: outerIndex + 1 }))
+            return "invalid";
+        this.nativeLifecycleByInnerIndex.set(innerIndex, "started");
+        this.stream.push({ type: "toolcall_start", contentIndex: outerIndex, partial: this.message });
+        return "projected";
+    }
+    projectNativeDelta(source, innerIndex, delta) {
+        if (this.nativeLifecycleByInnerIndex.get(innerIndex) !== "started")
+            return false;
+        const outerIndex = this.rangesByInnerIndex.get(innerIndex)?.start;
+        const block = source.content[innerIndex];
+        if (outerIndex == null || block?.type !== "toolCall")
+            return false;
+        this.message.content[outerIndex] = cloneContentBlock(block);
+        this.stream.push({ type: "toolcall_delta", contentIndex: outerIndex, delta, partial: this.message });
+        return true;
+    }
+    projectNativeEnd(innerIndex, toolCall) {
+        if (this.nativeLifecycleByInnerIndex.get(innerIndex) !== "started")
+            return false;
+        const outerIndex = this.rangesByInnerIndex.get(innerIndex)?.start;
+        if (outerIndex == null)
+            return false;
+        const finalToolCall = cloneToolCall(toolCall);
+        this.message.content[outerIndex] = finalToolCall;
+        this.nativeLifecycleByInnerIndex.set(innerIndex, "ended");
+        this.stream.push({
+            type: "toolcall_end",
+            contentIndex: outerIndex,
+            toolCall: finalToolCall,
+            partial: this.message,
+        });
+        return true;
+    }
+    assignRecoveredIds(events) {
+        return events.map((event) => {
+            if (event.type === "toolcall_start") {
+                const id = this.allocateRecoveredId();
+                this.recoveredIdByParserIndex.set(event.index, id);
+                return { ...event, id };
+            }
+            if (event.type === "toolcall_end") {
+                const id = this.recoveredIdByParserIndex.get(event.index) ?? event.id;
+                this.recoveredIdByParserIndex.delete(event.index);
+                return { ...event, id };
+            }
+            return event;
+        });
+    }
+    appendUnannounced(source, innerIndex) {
+        const block = source.content[innerIndex];
+        if (!block || innerIndex <= this.highestProjectedInnerIndex)
+            return false;
+        if (block.type === "toolCall" && this.recoveredIds.has(block.id))
+            return false;
+        const outerIndex = this.message.content.length;
+        this.message.content.push(cloneContentBlock(block));
+        return this.recordRange(innerIndex, { start: outerIndex, end: outerIndex + 1 });
+    }
+    recordRange(innerIndex, range) {
+        if (this.rangesByInnerIndex.has(innerIndex) || innerIndex <= this.highestProjectedInnerIndex)
+            return false;
+        this.rangesByInnerIndex.set(innerIndex, range);
+        this.highestProjectedInnerIndex = innerIndex;
+        return true;
+    }
+    allocateRecoveredId() {
+        for (;;) {
+            const id = `recovered-antml-${this.nextRecoveredId}`;
+            this.nextRecoveredId += 1;
+            if (this.reservedIds.has(id))
+                continue;
+            this.reservedIds.add(id);
+            this.recoveredIds.add(id);
+            return id;
+        }
+    }
+}
+//# sourceMappingURL=recovery-native-projection.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"recovery-native-projection.js","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-native-projection.ts"],"names":[],"mappings":"AAOA,SAAS,iBAAiB,CAAC,KAAmB;IAC7C,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;QAAE,OAAO,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,EAAE,GAAG,KAAK,CAAC,SAAS,EAAE,EAAE,CAAC;IACtF,OAAO,EAAE,GAAG,KAAK,EAAE,CAAC;AACrB,CAAC;AAED,SAAS,aAAa,CAAC,QAAkB;IACxC,OAAO,EAAE,GAAG,QAAQ,EAAE,SAAS,EAAE,EAAE,GAAG,QAAQ,CAAC,SAAS,EAAE,EAAE,CAAC;AAC9D,CAAC;AAED,2FAA2F;AAC3F,MAAM,OAAO,wBAAwB;IAWpC,YAAY,MAAmC,EAAE,OAAyB;QARzD,uBAAkB,GAAG,IAAI,GAAG,EAAwB,CAAC;QACrD,gCAA2B,GAAG,IAAI,GAAG,EAA2B,CAAC;QACjE,gBAAW,GAAG,IAAI,GAAG,EAAU,CAAC;QAChC,iBAAY,GAAG,IAAI,GAAG,EAAU,CAAC;QACjC,6BAAwB,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC9D,+BAA0B,GAAG,CAAC,CAAC,CAAC;QAChC,oBAAe,GAAG,CAAC,CAAC;QAG3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACxB,CAAC;IAED,iBAAiB,CAAC,MAAwB;QACzC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACpC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;gBAAE,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC/D,CAAC;IACF,CAAC;IAED,gBAAgB,CAAC,MAAwB,EAAE,YAAoB;QAC9D,KAAK,IAAI,UAAU,GAAG,CAAC,EAAE,UAAU,GAAG,YAAY,IAAI,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,IAAI,CAAC,EAAE,CAAC;YAC3G,IAAI,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC;gBAAE,SAAS;YACtD,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,UAAU,CAAC;gBAAE,OAAO,KAAK,CAAC;QAC/D,CAAC;QACD,OAAO,IAAI,CAAC;IACb,CAAC;IAED,oBAAoB,CAAC,MAAwB;QAC5C,KAAK,IAAI,UAAU,GAAG,CAAC,EAAE,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,IAAI,CAAC,EAAE,CAAC;YAC9E,IAAI,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC;gBAAE,SAAS;YACtD,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,UAAU,CAAC;gBAAE,OAAO,KAAK,CAAC;QAC/D,CAAC;QACD,OAAO,IAAI,CAAC;IACb,CAAC;IAED,SAAS,CAAC,UAAkB,EAAE,UAAkB;QAC/C,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAAE,OAAO,KAAK,CAAC;QACzG,OAAO,IAAI,CAAC;IACb,CAAC;IAED,oBAAoB,CAAC,UAAkB,EAAE,UAAkB;QAC1D,OAAO,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,GAAG,CAAC,EAAE,CAAC,CAAC;IACjF,CAAC;IAED,UAAU,CAAC,UAAkB;QAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACtD,IAAI,KAAK;YAAE,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;IACpD,CAAC;IAED,kBAAkB,CAAC,MAAwB,EAAE,UAAkB;QAC9D,IAAI,IAAI,CAAC,2BAA2B,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC;YAAE,OAAO,SAAS,CAAC;QAClH,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,IAAI,UAAU,IAAI,IAAI,CAAC,0BAA0B;YAAE,OAAO,SAAS,CAAC;QAClG,IAAI,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;YAAE,OAAO,WAAW,CAAC;QACxD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;QAC/C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC;QACpD,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,GAAG,CAAC,EAAE,CAAC;YAAE,OAAO,SAAS,CAAC;QAChG,IAAI,CAAC,2BAA2B,CAAC,GAAG,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAC5D,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,WAAW,CAAC;IACpB,CAAC;IAED,kBAAkB,CAAC,MAAwB,EAAE,UAAkB,EAAE,KAAa;QAC7E,IAAI,IAAI,CAAC,2BAA2B,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,SAAS;YAAE,OAAO,KAAK,CAAC;QACjF,MAAM,UAAU,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC;QAClE,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,UAAU,IAAI,IAAI,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU;YAAE,OAAO,KAAK,CAAC;QACnE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC5D,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;QACrG,OAAO,IAAI,CAAC;IACb,CAAC;IAED,gBAAgB,CAAC,UAAkB,EAAE,QAAkB;QACtD,IAAI,IAAI,CAAC,2BAA2B,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,SAAS;YAAE,OAAO,KAAK,CAAC;QACjF,MAAM,UAAU,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC;QAClE,IAAI,UAAU,IAAI,IAAI;YAAE,OAAO,KAAK,CAAC;QACrC,MAAM,aAAa,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC9C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,aAAa,CAAC;QACjD,IAAI,CAAC,2BAA2B,CAAC,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC1D,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YAChB,IAAI,EAAE,cAAc;YACpB,YAAY,EAAE,UAAU;YACxB,QAAQ,EAAE,aAAa;YACvB,OAAO,EAAE,IAAI,CAAC,OAAO;SACrB,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACb,CAAC;IAED,kBAAkB,CAAC,MAAoC;QACtD,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;gBACrC,MAAM,EAAE,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACtC,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBACnD,OAAO,EAAE,GAAG,KAAK,EAAE,EAAE,EAAE,CAAC;YACzB,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;gBACnC,MAAM,EAAE,GAAG,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACtE,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAClD,OAAO,EAAE,GAAG,KAAK,EAAE,EAAE,EAAE,CAAC;YACzB,CAAC;YACD,OAAO,KAAK,CAAC;QACd,CAAC,CAAC,CAAC;IACJ,CAAC;IAEO,iBAAiB,CAAC,MAAwB,EAAE,UAAkB;QACrE,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK,IAAI,UAAU,IAAI,IAAI,CAAC,0BAA0B;YAAE,OAAO,KAAK,CAAC;QAC1E,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;YAAE,OAAO,KAAK,CAAC;QAC/E,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;QAC/C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC;QACpD,OAAO,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,GAAG,CAAC,EAAE,CAAC,CAAC;IACjF,CAAC;IAEO,WAAW,CAAC,UAAkB,EAAE,KAAmB;QAC1D,IAAI,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,UAAU,IAAI,IAAI,CAAC,0BAA0B;YAAE,OAAO,KAAK,CAAC;QAC3G,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAC/C,IAAI,CAAC,0BAA0B,GAAG,UAAU,CAAC;QAC7C,OAAO,IAAI,CAAC;IACb,CAAC;IAEO,mBAAmB;QAC1B,SAAS,CAAC;YACT,MAAM,EAAE,GAAG,mBAAmB,IAAI,CAAC,eAAe,EAAE,CAAC;YACrD,IAAI,CAAC,eAAe,IAAI,CAAC,CAAC;YAC1B,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;gBAAE,SAAS;YACvC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACzB,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC1B,OAAO,EAAE,CAAC;QACX,CAAC;IACF,CAAC;CACD","sourcesContent":["import type { AssistantMessage, AssistantMessageEventStream, ToolCall } from \"../types.ts\";\nimport type { StreamParserEvent } from \"./types.ts\";\n\ntype ContentBlock = AssistantMessage[\"content\"][number];\ntype ContentRange = { start: number; end: number };\ntype NativeLifecycle = \"started\" | \"ended\";\n\nfunction cloneContentBlock(block: ContentBlock): ContentBlock {\n\tif (block.type === \"toolCall\") return { ...block, arguments: { ...block.arguments } };\n\treturn { ...block };\n}\n\nfunction cloneToolCall(toolCall: ToolCall): ToolCall {\n\treturn { ...toolCall, arguments: { ...toolCall.arguments } };\n}\n\n/** Maintains source-index order and native/recovered ID ownership for recovery streams. */\nexport class RecoveryNativeProjection {\n\tprivate readonly stream: AssistantMessageEventStream;\n\tprivate readonly message: AssistantMessage;\n\tprivate readonly rangesByInnerIndex = new Map<number, ContentRange>();\n\tprivate readonly nativeLifecycleByInnerIndex = new Map<number, NativeLifecycle>();\n\tprivate readonly reservedIds = new Set<string>();\n\tprivate readonly recoveredIds = new Set<string>();\n\tprivate readonly recoveredIdByParserIndex = new Map<number, string>();\n\tprivate highestProjectedInnerIndex = -1;\n\tprivate nextRecoveredId = 0;\n\n\tconstructor(stream: AssistantMessageEventStream, message: AssistantMessage) {\n\t\tthis.stream = stream;\n\t\tthis.message = message;\n\t}\n\n\treserveVisibleIds(source: AssistantMessage): void {\n\t\tfor (const block of source.content) {\n\t\t\tif (block.type === \"toolCall\") this.reservedIds.add(block.id);\n\t\t}\n\t}\n\n\tsynchronizeLower(source: AssistantMessage, contentIndex: number): boolean {\n\t\tfor (let innerIndex = 0; innerIndex < contentIndex && innerIndex < source.content.length; innerIndex += 1) {\n\t\t\tif (this.rangesByInnerIndex.has(innerIndex)) continue;\n\t\t\tif (!this.appendUnannounced(source, innerIndex)) return false;\n\t\t}\n\t\treturn true;\n\t}\n\n\tsynchronizeRemaining(source: AssistantMessage): boolean {\n\t\tfor (let innerIndex = 0; innerIndex < source.content.length; innerIndex += 1) {\n\t\t\tif (this.rangesByInnerIndex.has(innerIndex)) continue;\n\t\t\tif (!this.appendUnannounced(source, innerIndex)) return false;\n\t\t}\n\t\treturn true;\n\t}\n\n\tstartText(innerIndex: number, outerIndex: number): boolean {\n\t\tif (!this.recordRange(innerIndex, { start: outerIndex, end: this.message.content.length })) return false;\n\t\treturn true;\n\t}\n\n\trecordProjectedBlock(innerIndex: number, outerIndex: number): boolean {\n\t\treturn this.recordRange(innerIndex, { start: outerIndex, end: outerIndex + 1 });\n\t}\n\n\textendText(innerIndex: number): void {\n\t\tconst range = this.rangesByInnerIndex.get(innerIndex);\n\t\tif (range) range.end = this.message.content.length;\n\t}\n\n\tprojectNativeStart(source: AssistantMessage, innerIndex: number): \"projected\" | \"collision\" | \"invalid\" {\n\t\tif (this.nativeLifecycleByInnerIndex.has(innerIndex) || this.rangesByInnerIndex.has(innerIndex)) return \"invalid\";\n\t\tconst block = source.content[innerIndex];\n\t\tif (block?.type !== \"toolCall\" || innerIndex <= this.highestProjectedInnerIndex) return \"invalid\";\n\t\tif (this.recoveredIds.has(block.id)) return \"collision\";\n\t\tthis.reservedIds.add(block.id);\n\t\tconst outerIndex = this.message.content.length;\n\t\tthis.message.content.push(cloneContentBlock(block));\n\t\tif (!this.recordRange(innerIndex, { start: outerIndex, end: outerIndex + 1 })) return \"invalid\";\n\t\tthis.nativeLifecycleByInnerIndex.set(innerIndex, \"started\");\n\t\tthis.stream.push({ type: \"toolcall_start\", contentIndex: outerIndex, partial: this.message });\n\t\treturn \"projected\";\n\t}\n\n\tprojectNativeDelta(source: AssistantMessage, innerIndex: number, delta: string): boolean {\n\t\tif (this.nativeLifecycleByInnerIndex.get(innerIndex) !== \"started\") return false;\n\t\tconst outerIndex = this.rangesByInnerIndex.get(innerIndex)?.start;\n\t\tconst block = source.content[innerIndex];\n\t\tif (outerIndex == null || block?.type !== \"toolCall\") return false;\n\t\tthis.message.content[outerIndex] = cloneContentBlock(block);\n\t\tthis.stream.push({ type: \"toolcall_delta\", contentIndex: outerIndex, delta, partial: this.message });\n\t\treturn true;\n\t}\n\n\tprojectNativeEnd(innerIndex: number, toolCall: ToolCall): boolean {\n\t\tif (this.nativeLifecycleByInnerIndex.get(innerIndex) !== \"started\") return false;\n\t\tconst outerIndex = this.rangesByInnerIndex.get(innerIndex)?.start;\n\t\tif (outerIndex == null) return false;\n\t\tconst finalToolCall = cloneToolCall(toolCall);\n\t\tthis.message.content[outerIndex] = finalToolCall;\n\t\tthis.nativeLifecycleByInnerIndex.set(innerIndex, \"ended\");\n\t\tthis.stream.push({\n\t\t\ttype: \"toolcall_end\",\n\t\t\tcontentIndex: outerIndex,\n\t\t\ttoolCall: finalToolCall,\n\t\t\tpartial: this.message,\n\t\t});\n\t\treturn true;\n\t}\n\n\tassignRecoveredIds(events: readonly StreamParserEvent[]): StreamParserEvent[] {\n\t\treturn events.map((event) => {\n\t\t\tif (event.type === \"toolcall_start\") {\n\t\t\t\tconst id = this.allocateRecoveredId();\n\t\t\t\tthis.recoveredIdByParserIndex.set(event.index, id);\n\t\t\t\treturn { ...event, id };\n\t\t\t}\n\t\t\tif (event.type === \"toolcall_end\") {\n\t\t\t\tconst id = this.recoveredIdByParserIndex.get(event.index) ?? event.id;\n\t\t\t\tthis.recoveredIdByParserIndex.delete(event.index);\n\t\t\t\treturn { ...event, id };\n\t\t\t}\n\t\t\treturn event;\n\t\t});\n\t}\n\n\tprivate appendUnannounced(source: AssistantMessage, innerIndex: number): boolean {\n\t\tconst block = source.content[innerIndex];\n\t\tif (!block || innerIndex <= this.highestProjectedInnerIndex) return false;\n\t\tif (block.type === \"toolCall\" && this.recoveredIds.has(block.id)) return false;\n\t\tconst outerIndex = this.message.content.length;\n\t\tthis.message.content.push(cloneContentBlock(block));\n\t\treturn this.recordRange(innerIndex, { start: outerIndex, end: outerIndex + 1 });\n\t}\n\n\tprivate recordRange(innerIndex: number, range: ContentRange): boolean {\n\t\tif (this.rangesByInnerIndex.has(innerIndex) || innerIndex <= this.highestProjectedInnerIndex) return false;\n\t\tthis.rangesByInnerIndex.set(innerIndex, range);\n\t\tthis.highestProjectedInnerIndex = innerIndex;\n\t\treturn true;\n\t}\n\n\tprivate allocateRecoveredId(): string {\n\t\tfor (;;) {\n\t\t\tconst id = `recovered-antml-${this.nextRecoveredId}`;\n\t\t\tthis.nextRecoveredId += 1;\n\t\t\tif (this.reservedIds.has(id)) continue;\n\t\t\tthis.reservedIds.add(id);\n\t\t\tthis.recoveredIds.add(id);\n\t\t\treturn id;\n\t\t}\n\t}\n}\n"]}

package/dist/tool-call-middleware/recovery-stream-failure.d.ts

@@ -0,0 +1,5 @@
+import type { AssistantMessage, AssistantMessageEventStream } from "../types.ts";
+import type { StreamMessageProjection } from "./stream-wrapper-shared.ts";
+export type RecoveryStreamFailure = "collision" | "invalid_content_event_order" | "invalid_native_event_order";
+export declare function terminateRecoveryStreamForFailure(stream: AssistantMessageEventStream, projection: StreamMessageProjection, source: AssistantMessage, failure: RecoveryStreamFailure): void;
+//# sourceMappingURL=recovery-stream-failure.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"recovery-stream-failure.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-failure.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,2BAA2B,EAAE,MAAM,aAAa,CAAC;AACjF,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAE1E,MAAM,MAAM,qBAAqB,GAAG,WAAW,GAAG,6BAA6B,GAAG,4BAA4B,CAAC;AAqB/G,wBAAgB,iCAAiC,CAChD,MAAM,EAAE,2BAA2B,EACnC,UAAU,EAAE,uBAAuB,EACnC,MAAM,EAAE,gBAAgB,EACxB,OAAO,EAAE,qBAAqB,GAC5B,IAAI,CAiBN"}

package/dist/tool-call-middleware/recovery-stream-failure.js

@@ -0,0 +1,36 @@
+const failureDetails = {
+    collision: {
+        diagnosticType: "text_tool_call_recovery_collision",
+        errorMessage: "Tool call ID collision in provider stream",
+        status: "collision",
+    },
+    invalid_content_event_order: {
+        diagnosticType: "text_tool_call_recovery_invalid_content_event",
+        errorMessage: "Invalid assistant content event order",
+        status: "invalid_content_event_order",
+    },
+    invalid_native_event_order: {
+        diagnosticType: "text_tool_call_recovery_invalid_native_event",
+        errorMessage: "Invalid native tool call event order",
+        status: "invalid_native_event_order",
+    },
+};
+export function terminateRecoveryStreamForFailure(stream, projection, source, failure) {
+    const details = failureDetails[failure];
+    projection.sync(source);
+    const message = projection.finalize(source, false);
+    message.content = message.content.filter((block) => block.type !== "toolCall");
+    message.stopReason = "error";
+    message.errorMessage = details.errorMessage;
+    message.diagnostics = [
+        ...(source.diagnostics ?? []),
+        {
+            type: details.diagnosticType,
+            timestamp: Date.now(),
+            details: { protocol: "antml", status: details.status },
+        },
+    ];
+    stream.push({ type: "error", reason: "error", error: message });
+    stream.end(message);
+}
+//# sourceMappingURL=recovery-stream-failure.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"recovery-stream-failure.js","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-failure.ts"],"names":[],"mappings":"AAKA,MAAM,cAAc,GACnB;IACC,SAAS,EAAE;QACV,cAAc,EAAE,mCAAmC;QACnD,YAAY,EAAE,2CAA2C;QACzD,MAAM,EAAE,WAAW;KACnB;IACD,2BAA2B,EAAE;QAC5B,cAAc,EAAE,+CAA+C;QAC/D,YAAY,EAAE,uCAAuC;QACrD,MAAM,EAAE,6BAA6B;KACrC;IACD,0BAA0B,EAAE;QAC3B,cAAc,EAAE,8CAA8C;QAC9D,YAAY,EAAE,sCAAsC;QACpD,MAAM,EAAE,4BAA4B;KACpC;CACD,CAAC;AAEH,MAAM,UAAU,iCAAiC,CAChD,MAAmC,EACnC,UAAmC,EACnC,MAAwB,EACxB,OAA8B;IAE9B,MAAM,OAAO,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IACxC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACxB,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACnD,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;IAC/E,OAAO,CAAC,UAAU,GAAG,OAAO,CAAC;IAC7B,OAAO,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IAC5C,OAAO,CAAC,WAAW,GAAG;QACrB,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;QAC7B;YACC,IAAI,EAAE,OAAO,CAAC,cAAc;YAC5B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,OAAO,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE;SACtD;KACD,CAAC;IACF,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;IAChE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;AACrB,CAAC","sourcesContent":["import type { AssistantMessage, AssistantMessageEventStream } from \"../types.ts\";\nimport type { StreamMessageProjection } from \"./stream-wrapper-shared.ts\";\n\nexport type RecoveryStreamFailure = \"collision\" | \"invalid_content_event_order\" | \"invalid_native_event_order\";\n\nconst failureDetails: Record<RecoveryStreamFailure, { diagnosticType: string; errorMessage: string; status: string }> =\n\t{\n\t\tcollision: {\n\t\t\tdiagnosticType: \"text_tool_call_recovery_collision\",\n\t\t\terrorMessage: \"Tool call ID collision in provider stream\",\n\t\t\tstatus: \"collision\",\n\t\t},\n\t\tinvalid_content_event_order: {\n\t\t\tdiagnosticType: \"text_tool_call_recovery_invalid_content_event\",\n\t\t\terrorMessage: \"Invalid assistant content event order\",\n\t\t\tstatus: \"invalid_content_event_order\",\n\t\t},\n\t\tinvalid_native_event_order: {\n\t\t\tdiagnosticType: \"text_tool_call_recovery_invalid_native_event\",\n\t\t\terrorMessage: \"Invalid native tool call event order\",\n\t\t\tstatus: \"invalid_native_event_order\",\n\t\t},\n\t};\n\nexport function terminateRecoveryStreamForFailure(\n\tstream: AssistantMessageEventStream,\n\tprojection: StreamMessageProjection,\n\tsource: AssistantMessage,\n\tfailure: RecoveryStreamFailure,\n): void {\n\tconst details = failureDetails[failure];\n\tprojection.sync(source);\n\tconst message = projection.finalize(source, false);\n\tmessage.content = message.content.filter((block) => block.type !== \"toolCall\");\n\tmessage.stopReason = \"error\";\n\tmessage.errorMessage = details.errorMessage;\n\tmessage.diagnostics = [\n\t\t...(source.diagnostics ?? []),\n\t\t{\n\t\t\ttype: details.diagnosticType,\n\t\t\ttimestamp: Date.now(),\n\t\t\tdetails: { protocol: \"antml\", status: details.status },\n\t\t},\n\t];\n\tstream.push({ type: \"error\", reason: \"error\", error: message });\n\tstream.end(message);\n}\n"]}

package/dist/tool-call-middleware/recovery-stream-terminal.d.ts

@@ -0,0 +1,15 @@
+import type { AssistantMessage, AssistantMessageEventStream, StopReason } from "../types.ts";
+import type { StreamMessageProjection } from "./stream-wrapper-shared.ts";
+/** Emits one terminal event after recovery projection state has been flushed. */
+export declare class RecoveryStreamTerminal {
+    private emitted;
+    private readonly stream;
+    constructor(stream: AssistantMessageEventStream);
+    done(projection: StreamMessageProjection, source: AssistantMessage, sawToolCall: boolean, reason: Extract<StopReason, "stop" | "length" | "toolUse">): void;
+    sourceError(projection: StreamMessageProjection, source: AssistantMessage, sawToolCall: boolean, reason: Extract<StopReason, "aborted" | "error">): void;
+    iteratorFailure(projection: StreamMessageProjection | null, error: unknown): void;
+    exhausted(projection: StreamMessageProjection | null): void;
+    cancelled(projection: StreamMessageProjection | null): void;
+    private emit;
+}
+//# sourceMappingURL=recovery-stream-terminal.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"recovery-stream-terminal.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-terminal.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,2BAA2B,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC7F,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAM1E,iFAAiF;AACjF,qBAAa,sBAAsB;IAClC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8B;IAErD,YAAY,MAAM,EAAE,2BAA2B,EAE9C;IAED,IAAI,CACH,UAAU,EAAE,uBAAuB,EACnC,MAAM,EAAE,gBAAgB,EACxB,WAAW,EAAE,OAAO,EACpB,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS,CAAC,GACxD,IAAI,CAON;IAED,WAAW,CACV,UAAU,EAAE,uBAAuB,EACnC,MAAM,EAAE,gBAAgB,EACxB,WAAW,EAAE,OAAO,EACpB,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,SAAS,GAAG,OAAO,CAAC,GAC9C,IAAI,CAkBN;IAED,eAAe,CAAC,UAAU,EAAE,uBAAuB,GAAG,IAAI,EAAE,KAAK,EAAE,OAAO,GAAG,IAAI,CAYhF;IAED,SAAS,CAAC,UAAU,EAAE,uBAAuB,GAAG,IAAI,GAAG,IAAI,CAW1D;IAED,SAAS,CAAC,UAAU,EAAE,uBAAuB,GAAG,IAAI,GAAG,IAAI,CAY1D;IAED,OAAO,CAAC,IAAI;CAMZ"}

package/dist/tool-call-middleware/recovery-stream-terminal.js

@@ -0,0 +1,88 @@
+function errorText(error) {
+    return error instanceof Error ? error.message : String(error);
+}
+/** Emits one terminal event after recovery projection state has been flushed. */
+export class RecoveryStreamTerminal {
+    constructor(stream) {
+        this.emitted = false;
+        this.stream = stream;
+    }
+    done(projection, source, sawToolCall, reason) {
+        if (this.emitted)
+            return;
+        projection.finalizeDanglingToolCalls();
+        const message = projection.finalize(source, sawToolCall);
+        const recovered = sawToolCall || message.content.some((block) => block.type === "toolCall");
+        const finalReason = recovered && (reason === "stop" || reason === "length") ? "toolUse" : reason;
+        this.emit({ type: "done", reason: finalReason, message });
+    }
+    sourceError(projection, source, sawToolCall, reason) {
+        if (this.emitted)
+            return;
+        projection.finalizeDanglingToolCalls();
+        const message = projection.finalize(source, sawToolCall);
+        if (reason === "aborted") {
+            message.content = message.content.filter((block) => block.type !== "toolCall");
+            message.stopReason = "aborted";
+            message.errorMessage ??= "Request was aborted";
+            this.emit({ type: "error", reason: "aborted", error: message });
+            return;
+        }
+        if (projection.hasFinalizedToolCallContent()) {
+            message.stopReason = "toolUse";
+            this.emit({ type: "done", reason: "toolUse", message });
+            return;
+        }
+        message.stopReason = "error";
+        this.emit({ type: "error", reason: "error", error: message });
+    }
+    iteratorFailure(projection, error) {
+        if (this.emitted)
+            return;
+        if (!projection) {
+            this.emitted = true;
+            this.stream.fail(error);
+            return;
+        }
+        projection.finalizeDanglingToolCalls();
+        const message = projection.finalize(projection.message, false);
+        message.stopReason = "error";
+        message.errorMessage = errorText(error) || "Assistant message stream failed";
+        this.emit({ type: "error", reason: "error", error: message });
+    }
+    exhausted(projection) {
+        if (this.emitted)
+            return;
+        if (!projection) {
+            this.iteratorFailure(null, new Error("Assistant message stream ended without a terminal event"));
+            return;
+        }
+        projection.finalizeDanglingToolCalls();
+        const message = projection.finalize(projection.message, false);
+        message.stopReason = "error";
+        message.errorMessage = "Assistant message stream ended without a terminal event";
+        this.emit({ type: "error", reason: "error", error: message });
+    }
+    cancelled(projection) {
+        if (this.emitted)
+            return;
+        if (!projection) {
+            this.iteratorFailure(null, new Error("Assistant message stream consumption was cancelled"));
+            return;
+        }
+        projection.finalizeDanglingToolCalls();
+        const message = projection.finalize(projection.message, false);
+        message.content = message.content.filter((block) => block.type !== "toolCall");
+        message.stopReason = "error";
+        message.errorMessage = "Assistant message stream consumption was cancelled";
+        this.emit({ type: "error", reason: "error", error: message });
+    }
+    emit(event) {
+        if (this.emitted)
+            return;
+        this.emitted = true;
+        this.stream.push(event);
+        this.stream.end();
+    }
+}
+//# sourceMappingURL=recovery-stream-terminal.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"recovery-stream-terminal.js","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-terminal.ts"],"names":[],"mappings":"AAGA,SAAS,SAAS,CAAC,KAAc;IAChC,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC/D,CAAC;AAED,iFAAiF;AACjF,MAAM,OAAO,sBAAsB;IAIlC,YAAY,MAAmC;QAHvC,YAAO,GAAG,KAAK,CAAC;QAIvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IAED,IAAI,CACH,UAAmC,EACnC,MAAwB,EACxB,WAAoB,EACpB,MAA0D;QAE1D,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,UAAU,CAAC,yBAAyB,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QAC5F,MAAM,WAAW,GAAG,SAAS,IAAI,CAAC,MAAM,KAAK,MAAM,IAAI,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC;QACjG,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,WAAW,CACV,UAAmC,EACnC,MAAwB,EACxB,WAAoB,EACpB,MAAgD;QAEhD,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,UAAU,CAAC,yBAAyB,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACzD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YAC1B,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YAC/E,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;YAC/B,OAAO,CAAC,YAAY,KAAK,qBAAqB,CAAC;YAC/C,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YAChE,OAAO;QACR,CAAC;QACD,IAAI,UAAU,CAAC,2BAA2B,EAAE,EAAE,CAAC;YAC9C,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;YAC/B,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;YACxD,OAAO;QACR,CAAC;QACD,OAAO,CAAC,UAAU,GAAG,OAAO,CAAC;QAC7B,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,eAAe,CAAC,UAA0C,EAAE,KAAc;QACzE,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,UAAU,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxB,OAAO;QACR,CAAC;QACD,UAAU,CAAC,yBAAyB,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC/D,OAAO,CAAC,UAAU,GAAG,OAAO,CAAC;QAC7B,OAAO,CAAC,YAAY,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,iCAAiC,CAAC;QAC7E,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,SAAS,CAAC,UAA0C;QACnD,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,UAAU,EAAE,CAAC;YACjB,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC,CAAC;YACjG,OAAO;QACR,CAAC;QACD,UAAU,CAAC,yBAAyB,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC/D,OAAO,CAAC,UAAU,GAAG,OAAO,CAAC;QAC7B,OAAO,CAAC,YAAY,GAAG,yDAAyD,CAAC;QACjF,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,SAAS,CAAC,UAA0C;QACnD,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,UAAU,EAAE,CAAC;YACjB,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC,CAAC;YAC5F,OAAO;QACR,CAAC;QACD,UAAU,CAAC,yBAAyB,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC/D,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QAC/E,OAAO,CAAC,UAAU,GAAG,OAAO,CAAC;QAC7B,OAAO,CAAC,YAAY,GAAG,oDAAoD,CAAC;QAC5E,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;IAC/D,CAAC;IAEO,IAAI,CAAC,KAA8F;QAC1G,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC;CACD","sourcesContent":["import type { AssistantMessage, AssistantMessageEventStream, StopReason } from \"../types.ts\";\nimport type { StreamMessageProjection } from \"./stream-wrapper-shared.ts\";\n\nfunction errorText(error: unknown): string {\n\treturn error instanceof Error ? error.message : String(error);\n}\n\n/** Emits one terminal event after recovery projection state has been flushed. */\nexport class RecoveryStreamTerminal {\n\tprivate emitted = false;\n\tprivate readonly stream: AssistantMessageEventStream;\n\n\tconstructor(stream: AssistantMessageEventStream) {\n\t\tthis.stream = stream;\n\t}\n\n\tdone(\n\t\tprojection: StreamMessageProjection,\n\t\tsource: AssistantMessage,\n\t\tsawToolCall: boolean,\n\t\treason: Extract<StopReason, \"stop\" | \"length\" | \"toolUse\">,\n\t): void {\n\t\tif (this.emitted) return;\n\t\tprojection.finalizeDanglingToolCalls();\n\t\tconst message = projection.finalize(source, sawToolCall);\n\t\tconst recovered = sawToolCall || message.content.some((block) => block.type === \"toolCall\");\n\t\tconst finalReason = recovered && (reason === \"stop\" || reason === \"length\") ? \"toolUse\" : reason;\n\t\tthis.emit({ type: \"done\", reason: finalReason, message });\n\t}\n\n\tsourceError(\n\t\tprojection: StreamMessageProjection,\n\t\tsource: AssistantMessage,\n\t\tsawToolCall: boolean,\n\t\treason: Extract<StopReason, \"aborted\" | \"error\">,\n\t): void {\n\t\tif (this.emitted) return;\n\t\tprojection.finalizeDanglingToolCalls();\n\t\tconst message = projection.finalize(source, sawToolCall);\n\t\tif (reason === \"aborted\") {\n\t\t\tmessage.content = message.content.filter((block) => block.type !== \"toolCall\");\n\t\t\tmessage.stopReason = \"aborted\";\n\t\t\tmessage.errorMessage ??= \"Request was aborted\";\n\t\t\tthis.emit({ type: \"error\", reason: \"aborted\", error: message });\n\t\t\treturn;\n\t\t}\n\t\tif (projection.hasFinalizedToolCallContent()) {\n\t\t\tmessage.stopReason = \"toolUse\";\n\t\t\tthis.emit({ type: \"done\", reason: \"toolUse\", message });\n\t\t\treturn;\n\t\t}\n\t\tmessage.stopReason = \"error\";\n\t\tthis.emit({ type: \"error\", reason: \"error\", error: message });\n\t}\n\n\titeratorFailure(projection: StreamMessageProjection | null, error: unknown): void {\n\t\tif (this.emitted) return;\n\t\tif (!projection) {\n\t\t\tthis.emitted = true;\n\t\t\tthis.stream.fail(error);\n\t\t\treturn;\n\t\t}\n\t\tprojection.finalizeDanglingToolCalls();\n\t\tconst message = projection.finalize(projection.message, false);\n\t\tmessage.stopReason = \"error\";\n\t\tmessage.errorMessage = errorText(error) || \"Assistant message stream failed\";\n\t\tthis.emit({ type: \"error\", reason: \"error\", error: message });\n\t}\n\n\texhausted(projection: StreamMessageProjection | null): void {\n\t\tif (this.emitted) return;\n\t\tif (!projection) {\n\t\t\tthis.iteratorFailure(null, new Error(\"Assistant message stream ended without a terminal event\"));\n\t\t\treturn;\n\t\t}\n\t\tprojection.finalizeDanglingToolCalls();\n\t\tconst message = projection.finalize(projection.message, false);\n\t\tmessage.stopReason = \"error\";\n\t\tmessage.errorMessage = \"Assistant message stream ended without a terminal event\";\n\t\tthis.emit({ type: \"error\", reason: \"error\", error: message });\n\t}\n\n\tcancelled(projection: StreamMessageProjection | null): void {\n\t\tif (this.emitted) return;\n\t\tif (!projection) {\n\t\t\tthis.iteratorFailure(null, new Error(\"Assistant message stream consumption was cancelled\"));\n\t\t\treturn;\n\t\t}\n\t\tprojection.finalizeDanglingToolCalls();\n\t\tconst message = projection.finalize(projection.message, false);\n\t\tmessage.content = message.content.filter((block) => block.type !== \"toolCall\");\n\t\tmessage.stopReason = \"error\";\n\t\tmessage.errorMessage = \"Assistant message stream consumption was cancelled\";\n\t\tthis.emit({ type: \"error\", reason: \"error\", error: message });\n\t}\n\n\tprivate emit(event: Extract<Parameters<AssistantMessageEventStream[\"push\"]>[0], { type: \"done\" | \"error\" }>): void {\n\t\tif (this.emitted) return;\n\t\tthis.emitted = true;\n\t\tthis.stream.push(event);\n\t\tthis.stream.end();\n\t}\n}\n"]}

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

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"recovery-stream-wrapper.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAA2C,2BAA2B,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAS9G,wBAAgB,4BAA4B,CAC3C,WAAW,EAAE,2BAA2B,EACxC,KAAK,EAAE,SAAS,IAAI,EAAE,GACpB,2BAA2B,CAwN7B"}

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

@@ -0,0 +1,233 @@
+import { RecoveryContentLifecycle } from "./recovery-content-lifecycle.js";
+import { RecoveryAssistantMessageEventStream } from "./recovery-event-stream.js";
+import { RecoveryNativeProjection } from "./recovery-native-projection.js";
+import { terminateRecoveryStreamForFailure } from "./recovery-stream-failure.js";
+import { RecoveryStreamTerminal } from "./recovery-stream-terminal.js";
+import { RecoveryTextProjection } from "./recovery-text-projection.js";
+import { StreamMessageProjection } from "./stream-wrapper-shared.js";
+export function wrapStreamWithInvokeRecovery(innerStream, tools) {
+    const outerStream = new RecoveryAssistantMessageEventStream();
+    void (async () => {
+        const innerIterator = innerStream[Symbol.asyncIterator]();
+        const innerEvents = { [Symbol.asyncIterator]: () => innerIterator };
+        let projection = null;
+        let nativeProjection = null;
+        let textProjection = null;
+        let sawToolCall = false;
+        let cancellationRequested = false;
+        const contentLifecycle = new RecoveryContentLifecycle();
+        const terminal = new RecoveryStreamTerminal(outerStream);
+        const finishText = () => {
+            if (!textProjection)
+                return;
+            sawToolCall = textProjection.finish() || sawToolCall;
+            textProjection = null;
+        };
+        outerStream.setCancellationHandler(async () => {
+            cancellationRequested = true;
+            try {
+                await innerIterator.return?.();
+                finishText();
+                terminal.cancelled(projection);
+            }
+            catch (error) {
+                finishText();
+                terminal.iteratorFailure(projection, error);
+                throw error;
+            }
+        });
+        const terminateForFailure = (source, failure) => {
+            if (!projection || !outerStream.markSourceClosed())
+                return;
+            finishText();
+            terminateRecoveryStreamForFailure(outerStream, projection, source, failure);
+        };
+        const prepareContentEvent = (source, contentIndex) => {
+            if (!projection || !nativeProjection)
+                return false;
+            if (!Number.isSafeInteger(contentIndex) || contentIndex < 0 || contentIndex >= source.content.length) {
+                terminateForFailure(source, "invalid_native_event_order");
+                return false;
+            }
+            projection.sync(source);
+            nativeProjection.reserveVisibleIds(source);
+            if (nativeProjection.synchronizeLower(source, contentIndex))
+                return true;
+            terminateForFailure(source, "collision");
+            return false;
+        };
+        const canStart = (source, contentIndex, kind, failure = "invalid_content_event_order") => {
+            if (!Number.isSafeInteger(contentIndex) || contentIndex < 0 || contentIndex >= source.content.length) {
+                terminateForFailure(source, failure);
+                return false;
+            }
+            const block = source.content[contentIndex];
+            const blockMatches = (kind === "text" && block?.type === "text") ||
+                (kind === "thinking" && block?.type === "thinking") ||
+                (kind === "toolCall" && block?.type === "toolCall");
+            if (blockMatches && contentLifecycle.canStart(contentIndex))
+                return true;
+            terminateForFailure(source, failure);
+            return false;
+        };
+        const isActive = (source, contentIndex, kind, failure = "invalid_content_event_order") => {
+            if (contentLifecycle.isActive(contentIndex, kind))
+                return true;
+            terminateForFailure(source, failure);
+            return false;
+        };
+        const synchronizeTerminal = (source) => {
+            projection ??= new StreamMessageProjection(outerStream, source, { preserveSourceMetadata: true });
+            nativeProjection ??= new RecoveryNativeProjection(outerStream, projection.message);
+            projection.sync(source);
+            nativeProjection.reserveVisibleIds(source);
+            finishText();
+            if (nativeProjection.synchronizeRemaining(source))
+                return true;
+            terminateForFailure(source, "collision");
+            return false;
+        };
+        try {
+            for await (const event of innerEvents) {
+                if (cancellationRequested)
+                    continue;
+                switch (event.type) {
+                    case "start":
+                        projection = new StreamMessageProjection(outerStream, event.partial, {
+                            preserveSourceMetadata: true,
+                        });
+                        nativeProjection = new RecoveryNativeProjection(outerStream, projection.message);
+                        nativeProjection.reserveVisibleIds(event.partial);
+                        outerStream.push({ type: "start", partial: projection.message });
+                        break;
+                    case "text_start": {
+                        if (!canStart(event.partial, event.contentIndex, "text"))
+                            return;
+                        if (!prepareContentEvent(event.partial, event.contentIndex) || !projection || !nativeProjection)
+                            return;
+                        const nextText = new RecoveryTextProjection(tools, projection, nativeProjection, event.contentIndex);
+                        if (!nextText.start(event.partial)) {
+                            terminateForFailure(event.partial, "invalid_native_event_order");
+                            return;
+                        }
+                        textProjection = nextText;
+                        contentLifecycle.start(event.contentIndex, "text");
+                        break;
+                    }
+                    case "text_delta":
+                        if (!isActive(event.partial, event.contentIndex, "text"))
+                            return;
+                        if (!prepareContentEvent(event.partial, event.contentIndex) || !textProjection)
+                            return;
+                        sawToolCall = textProjection.feed(event.delta) || sawToolCall;
+                        break;
+                    case "text_end":
+                        if (!isActive(event.partial, event.contentIndex, "text"))
+                            return;
+                        if (!prepareContentEvent(event.partial, event.contentIndex))
+                            return;
+                        finishText();
+                        contentLifecycle.end(event.contentIndex, "text");
+                        break;
+                    case "done":
+                        if (!synchronizeTerminal(event.message) || !projection)
+                            return;
+                        if (!outerStream.markSourceClosed())
+                            return;
+                        terminal.done(projection, event.message, sawToolCall, event.reason);
+                        return;
+                    case "error":
+                        if (!projection) {
+                            if (!outerStream.markSourceClosed())
+                                return;
+                            outerStream.push(event);
+                            outerStream.end();
+                            return;
+                        }
+                        if (!synchronizeTerminal(event.error))
+                            return;
+                        if (!outerStream.markSourceClosed())
+                            return;
+                        terminal.sourceError(projection, event.error, sawToolCall, event.reason);
+                        return;
+                    case "toolcall_start": {
+                        if (!canStart(event.partial, event.contentIndex, "toolCall", "invalid_native_event_order"))
+                            return;
+                        if (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection)
+                            return;
+                        const status = nativeProjection.projectNativeStart(event.partial, event.contentIndex);
+                        if (status !== "projected") {
+                            terminateForFailure(event.partial, status === "collision" ? "collision" : "invalid_native_event_order");
+                            return;
+                        }
+                        contentLifecycle.start(event.contentIndex, "toolCall");
+                        sawToolCall = true;
+                        break;
+                    }
+                    case "toolcall_delta":
+                        if (!isActive(event.partial, event.contentIndex, "toolCall", "invalid_native_event_order"))
+                            return;
+                        if (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection)
+                            return;
+                        if (!nativeProjection.projectNativeDelta(event.partial, event.contentIndex, event.delta)) {
+                            terminateForFailure(event.partial, "invalid_native_event_order");
+                            return;
+                        }
+                        break;
+                    case "toolcall_end":
+                        if (!isActive(event.partial, event.contentIndex, "toolCall", "invalid_native_event_order"))
+                            return;
+                        if (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection)
+                            return;
+                        if (!nativeProjection.projectNativeEnd(event.contentIndex, event.toolCall)) {
+                            terminateForFailure(event.partial, "invalid_native_event_order");
+                            return;
+                        }
+                        contentLifecycle.end(event.contentIndex, "toolCall");
+                        sawToolCall = true;
+                        break;
+                    case "thinking_start": {
+                        if (!canStart(event.partial, event.contentIndex, "thinking"))
+                            return;
+                        if (!prepareContentEvent(event.partial, event.contentIndex) || !projection || !nativeProjection)
+                            return;
+                        const outerIndex = projection.startThinking(event.contentIndex, event.partial);
+                        if (!nativeProjection.recordProjectedBlock(event.contentIndex, outerIndex)) {
+                            terminateForFailure(event.partial, "invalid_native_event_order");
+                            return;
+                        }
+                        contentLifecycle.start(event.contentIndex, "thinking");
+                        break;
+                    }
+                    case "thinking_delta":
+                        if (!isActive(event.partial, event.contentIndex, "thinking"))
+                            return;
+                        if (!prepareContentEvent(event.partial, event.contentIndex) || !projection)
+                            return;
+                        projection.projectThinkingDelta(event.contentIndex, event.delta, event.partial);
+                        break;
+                    case "thinking_end":
+                        if (!isActive(event.partial, event.contentIndex, "thinking"))
+                            return;
+                        if (!prepareContentEvent(event.partial, event.contentIndex) || !projection)
+                            return;
+                        projection.finishThinking(event.contentIndex, event.content, event.partial);
+                        contentLifecycle.end(event.contentIndex, "thinking");
+                        break;
+                }
+            }
+            if (cancellationRequested || !outerStream.markSourceClosed())
+                return;
+            finishText();
+            terminal.exhausted(projection);
+        }
+        catch (error) {
+            if (cancellationRequested || !outerStream.markSourceClosed())
+                return;
+            finishText();
+            terminal.iteratorFailure(projection, error);
+        }
+    })();
+    return outerStream;
+}
+//# sourceMappingURL=recovery-stream-wrapper.js.map