ns-kiro-core 0.1.2 → 0.2.0

package/dist/response-assembler.js

@@ -0,0 +1,330 @@
+// ABOUTME: Turns Kiro wire events into neutral content blocks and a stop reason.
+// ABOUTME: Owns what the response says; the request loop owns whether to ask again.
+import { KiroBlockBuffer } from "./blocks.js";
+import { parseBracketToolCalls } from "./bracket-tool-parser.js";
+import { calculateKiroCost } from "./cost.js";
+import { debugEnabled, debugLog, formatSafeError, redactSensitiveText } from "./debug.js";
+import { parseInvokeToolCalls } from "./invoke-tool-parser.js";
+import { ThinkingTagParser } from "./thinking-parser.js";
+import { countTokens } from "./tokenizer.js";
+/** Text that is an artifact of history padding rather than an answer. */
+const ECHO_NOISE_PATTERN = /^\s*(continue|\.+)\s*$/i;
+/**
+ * Accumulates one response.
+ *
+ * Block indexes are monotonic across the whole call, including across an
+ * internal retry, so the buffer and the usage totals outlive a single attempt
+ * while everything else is reset by {@link beginAttempt}.
+ */
+export class KiroResponseAssembler {
+    model;
+    thinkingEnabled;
+    pending = [];
+    blocks = new KiroBlockBuffer((event) => this.pending.push(event));
+    usage = {
+        input: 0,
+        output: 0,
+        totalTokens: 0,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+    };
+    totalContent = "";
+    lastContentData = "";
+    usageEvent = null;
+    receivedContextUsage = false;
+    thinkingParser = null;
+    nativeThinkingBlockIndex = null;
+    nativeThinkingEnded = false;
+    textBlockIndex = null;
+    emittedToolCalls = 0;
+    sawAnyToolCalls = false;
+    currentToolCall = null;
+    constructor(model, thinkingEnabled) {
+        this.model = model;
+        this.thinkingEnabled = thinkingEnabled;
+    }
+    /** Clear per-attempt state. Block indexes and accumulated usage are kept. */
+    beginAttempt() {
+        this.totalContent = "";
+        this.lastContentData = "";
+        this.usageEvent = null;
+        this.receivedContextUsage = false;
+        this.thinkingParser = this.thinkingEnabled ? new ThinkingTagParser(this.blocks) : null;
+        this.nativeThinkingBlockIndex = null;
+        this.nativeThinkingEnded = false;
+        this.textBlockIndex = null;
+        this.emittedToolCalls = 0;
+        this.sawAnyToolCalls = false;
+        this.currentToolCall = null;
+    }
+    /** Hand over the events buffered so far. */
+    takeEvents() {
+        return this.pending.splice(0, this.pending.length);
+    }
+    /** Announce that a retry discarded everything emitted so far. */
+    discard() {
+        this.blocks.reset();
+        this.textBlockIndex = null;
+    }
+    handle(frame) {
+        const { event, payload } = frame;
+        switch (event.type) {
+            case "contextUsage": {
+                const pct = event.data.contextUsagePercentage;
+                this.usage.input = Math.round((pct / 100) * this.model.contextWindow);
+                this.usage.contextPercent = pct;
+                this.receivedContextUsage = true;
+                break;
+            }
+            case "thinkingText": {
+                if (!this.thinkingEnabled)
+                    break;
+                this.blocks.appendThinking(this.ensureNativeThinkingBlock(), event.data);
+                this.totalContent += event.data;
+                break;
+            }
+            case "thinkingSignature": {
+                if (!this.thinkingEnabled)
+                    break;
+                this.ensureNativeThinkingBlock();
+                this.endNativeThinking(event.data);
+                break;
+            }
+            case "content": {
+                this.endNativeThinking();
+                // Kiro repeats the last content frame on some turns; a repeat carries no
+                // new text and must not be appended twice.
+                if (event.data === this.lastContentData)
+                    break;
+                this.lastContentData = event.data;
+                this.totalContent += event.data;
+                if (this.thinkingParser) {
+                    this.thinkingParser.processChunk(event.data);
+                }
+                else {
+                    if (this.textBlockIndex === null)
+                        this.textBlockIndex = this.blocks.openText();
+                    this.blocks.appendText(this.textBlockIndex, event.data);
+                }
+                break;
+            }
+            case "toolUse": {
+                const tc = event.data;
+                this.sawAnyToolCalls = true;
+                if (!this.currentToolCall || this.currentToolCall.toolUseId !== tc.toolUseId) {
+                    this.flushToolCall();
+                    this.currentToolCall = { toolUseId: tc.toolUseId, name: tc.name, input: "" };
+                }
+                this.currentToolCall.input += tc.input || "";
+                if (tc.input)
+                    this.totalContent += tc.input;
+                if (tc.stop)
+                    this.flushToolCall();
+                break;
+            }
+            case "toolUseInput": {
+                if (this.currentToolCall)
+                    this.currentToolCall.input += event.data.input || "";
+                if (event.data.input)
+                    this.totalContent += event.data.input;
+                break;
+            }
+            case "toolUseStop": {
+                if (event.data.stop)
+                    this.flushToolCall();
+                break;
+            }
+            case "usage": {
+                this.usageEvent = event.data;
+                // The parsed event keeps only the fields this package understands.
+                // Log the frame verbatim so a field Kiro adds — cache counters above
+                // all — is visible without having to guess its name first.
+                if (debugEnabled())
+                    debugLog("response.usageRaw", payload);
+                break;
+            }
+            // followupPrompt events are intentionally ignored
+        }
+    }
+    /**
+     * Close the turn: flush a trailing tool call, close thinking, and run the
+     * text-dialect recovery and echo-stripping passes.
+     */
+    endTurn() {
+        if (this.currentToolCall && this.emitToolCall(this.currentToolCall))
+            this.emittedToolCalls++;
+        this.currentToolCall = null;
+        this.endNativeThinking();
+        if (this.thinkingParser) {
+            this.thinkingParser.finalize();
+            this.textBlockIndex = this.thinkingParser.getTextBlockIndex();
+        }
+        this.recoverTextToolCalls();
+        this.stripEchoNoise();
+        const responseText = this.textBlockIndex === null ? "" : this.blocks.getText(this.textBlockIndex);
+        const hasText = responseText.length > 0;
+        return {
+            responseText,
+            hasText,
+            sawAnyToolCalls: this.sawAnyToolCalls,
+            emittedToolCalls: this.emittedToolCalls,
+            isEchoLoop: hasText && !this.sawAnyToolCalls && ECHO_NOISE_PATTERN.test(responseText),
+            isEmpty: !hasText && !this.sawAnyToolCalls,
+        };
+    }
+    /** Drop an echo the caller decided not to retry, so it is not read as a continuation signal. */
+    stripEcho() {
+        if (this.textBlockIndex !== null)
+            this.blocks.setText(this.textBlockIndex, "");
+    }
+    /** Close the text block and settle usage and the stop reason. */
+    complete() {
+        if (this.textBlockIndex !== null)
+            this.blocks.endText(this.textBlockIndex);
+        // Kiro does not reliably emit per-response output token counts. When the
+        // `usage` event is missing or reports only `inputTokens`, fall back to a
+        // tiktoken estimate over everything the assistant emitted — text plus
+        // tool-call input JSON. Otherwise tool-call-only turns report 0 output
+        // tokens and break consumers that watch it.
+        if (this.usageEvent?.inputTokens !== undefined)
+            this.usage.input = this.usageEvent.inputTokens;
+        this.usage.output = this.usageEvent?.outputTokens ?? countTokens(this.totalContent);
+        this.usage.totalTokens = this.usage.input + this.usage.output;
+        // Only set when reported: leaving these absent is what tells a host that
+        // Kiro said nothing about caching, rather than that nothing was cached.
+        if (this.usageEvent?.cacheReadTokens !== undefined)
+            this.usage.cacheRead = this.usageEvent.cacheReadTokens;
+        if (this.usageEvent?.cacheWriteTokens !== undefined)
+            this.usage.cacheWrite = this.usageEvent.cacheWriteTokens;
+        if (this.usageEvent?.credits !== undefined)
+            this.usage.credits = this.usageEvent.credits;
+        if (this.usageEvent?.creditUnit !== undefined)
+            this.usage.creditUnit = this.usageEvent.creditUnit;
+        this.usage.cost = calculateKiroCost(this.model.cost, this.usage);
+        // Use `emittedToolCalls`, not the count seen on the wire: a turn whose calls
+        // were all dropped for unparseable input must not report `toolUse`, because
+        // an empty turn with a tool-use stop stalls an agent loop waiting for
+        // results that will never arrive.
+        //
+        // `length` is inferred, not reported: Kiro sends no stop reason, so a turn
+        // that produced no tool call and never carried a contextUsage frame is
+        // treated as cut short.
+        //
+        // That rests on contextUsage closing every complete response. Checked
+        // 2026-09-06 across a short reply, a ~5000-character one, a tool-call turn,
+        // a model with no effort schema (claude-haiku-4.5) and a non-Claude model
+        // (glm-5): the frame arrived in all five, so its absence really does mark an
+        // abnormal turn. `response.done` logs `receivedContextUsage` in case a later
+        // Kiro stops sending it — a false `length` is read by hosts as truncation
+        // and prepends TRUNCATION_NOTICE, asking the model to continue work it
+        // already finished.
+        const stopReason = !this.receivedContextUsage && this.emittedToolCalls === 0
+            ? "length"
+            : this.emittedToolCalls > 0
+                ? "toolUse"
+                : "stop";
+        debugLog("response.done", {
+            stopReason,
+            receivedContextUsage: this.receivedContextUsage,
+            emittedToolCalls: this.emittedToolCalls,
+            sawAnyToolCalls: this.sawAnyToolCalls,
+            textLen: this.textBlockIndex === null ? 0 : this.blocks.getText(this.textBlockIndex).length,
+            usage: this.usage,
+        });
+        return { stopReason, usage: this.usage };
+    }
+    ensureNativeThinkingBlock() {
+        if (this.nativeThinkingBlockIndex === null)
+            this.nativeThinkingBlockIndex = this.blocks.openThinking();
+        return this.nativeThinkingBlockIndex;
+    }
+    endNativeThinking(signature) {
+        if (this.nativeThinkingBlockIndex === null || this.nativeThinkingEnded)
+            return;
+        this.nativeThinkingEnded = true;
+        this.blocks.endThinking(this.nativeThinkingBlockIndex, signature);
+    }
+    emitToolCall(state) {
+        if (!state.input.trim()) {
+            // Kiro omits the input payload when the model calls a tool with no
+            // arguments (e.g. `mcp({})`). Treat empty input as an empty object
+            // rather than skipping — these are valid zero-arg calls, not truncations.
+            state.input = "{}";
+        }
+        let args;
+        try {
+            args = JSON.parse(state.input);
+        }
+        catch (e) {
+            console.warn(`[kiro-core] Failed to parse tool input for "${state.name}" (toolUseId: ${state.toolUseId}): ${formatSafeError(e)}. Raw input (${state.input.length} chars): ${redactSensitiveText(state.input.substring(0, 200))}`);
+            return false;
+        }
+        const index = this.blocks.reserve();
+        this.pending.push({ type: "tool_call_start", index, id: state.toolUseId, name: state.name });
+        this.pending.push({ type: "tool_call_delta", index, id: state.toolUseId, argumentsDelta: state.input });
+        this.pending.push({ type: "tool_call_end", index, id: state.toolUseId, name: state.name, arguments: args });
+        return true;
+    }
+    flushToolCall() {
+        if (!this.currentToolCall)
+            return;
+        if (this.emitToolCall(this.currentToolCall))
+            this.emittedToolCalls++;
+        this.currentToolCall = null;
+    }
+    /**
+     * Extract text-dialect tool calls from content when no native tool calls
+     * arrived. Two dialects are recovered at this seam:
+     *   1. Kiro's own `[Called name with args: {...}]` bracket form.
+     *   2. Anthropic's `<invoke name="..."><parameter .../></invoke>` XML form,
+     *      which opus-class models emit as plain text at high context.
+     * Without this, the turn ends `stop` with zero tool calls — the agent loop
+     * sees a finished answer and an unattended session stalls with no error
+     * recorded anywhere.
+     *
+     * Models that emit native tool-use events opt out via
+     * `recoverTextToolCalls: false`. For them this pass has nothing to rescue
+     * and everything to break: prose that merely *quotes* the syntax — a model
+     * explaining how a tool is called — would be lifted into a real call the
+     * model never made. Absent means recover, so a model the catalog says
+     * nothing about keeps the fallback.
+     */
+    recoverTextToolCalls() {
+        if (this.model.recoverTextToolCalls === false || this.sawAnyToolCalls || this.textBlockIndex === null)
+            return;
+        let text = this.blocks.getText(this.textBlockIndex);
+        const recovered = [];
+        const bracketResult = parseBracketToolCalls(text);
+        if (bracketResult.toolCalls.length > 0) {
+            text = bracketResult.cleanedText;
+            recovered.push(...bracketResult.toolCalls);
+        }
+        const invokeResult = parseInvokeToolCalls(text);
+        if (invokeResult.toolCalls.length > 0) {
+            text = invokeResult.cleanedText;
+            recovered.push(...invokeResult.toolCalls);
+        }
+        if (recovered.length === 0)
+            return;
+        this.blocks.setText(this.textBlockIndex, text);
+        this.sawAnyToolCalls = true;
+        for (const call of recovered) {
+            if (this.emitToolCall({ toolUseId: call.toolUseId, name: call.name, input: JSON.stringify(call.arguments) })) {
+                this.emittedToolCalls++;
+            }
+        }
+    }
+    /**
+     * Strip echo noise: when tool calls are present and the text content is just
+     * "." or a similar short echo from history padding, remove it. This prevents
+     * the echo from accumulating in conversation history and reinforcing the
+     * pattern in future turns.
+     */
+    stripEchoNoise() {
+        if (this.emittedToolCalls === 0 || this.textBlockIndex === null)
+            return;
+        if (ECHO_NOISE_PATTERN.test(this.blocks.getText(this.textBlockIndex))) {
+            this.blocks.setText(this.textBlockIndex, "");
+        }
+    }
+}
+//# sourceMappingURL=response-assembler.js.map

package/dist/response-assembler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"response-assembler.js","sourceRoot":"","sources":["../src/response-assembler.ts"],"names":[],"mappings":"AAAA,iFAAiF;AACjF,oFAAoF;AAEpF,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAC9C,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAE1F,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAG/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAG7C,yEAAyE;AACzE,MAAM,kBAAkB,GAAG,yBAAyB,CAAC;AAyBrD;;;;;;GAMG;AACH,MAAM,OAAO,qBAAqB;IAuBb;IACA;IAvBF,OAAO,GAAsB,EAAE,CAAC;IAChC,MAAM,GAAG,IAAI,eAAe,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IAClE,KAAK,GAAc;QAClC,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,WAAW,EAAE,CAAC;QACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;KACrE,CAAC;IAEM,YAAY,GAAG,EAAE,CAAC;IAClB,eAAe,GAAG,EAAE,CAAC;IACrB,UAAU,GAAyB,IAAI,CAAC;IACxC,oBAAoB,GAAG,KAAK,CAAC;IAC7B,cAAc,GAA6B,IAAI,CAAC;IAChD,wBAAwB,GAAkB,IAAI,CAAC;IAC/C,mBAAmB,GAAG,KAAK,CAAC;IAC5B,cAAc,GAAkB,IAAI,CAAC;IACrC,gBAAgB,GAAG,CAAC,CAAC;IACrB,eAAe,GAAG,KAAK,CAAC;IACxB,eAAe,GAA6B,IAAI,CAAC;IAEzD,YACmB,KAAgB,EAChB,eAAwB;QADxB,UAAK,GAAL,KAAK,CAAW;QAChB,oBAAe,GAAf,eAAe,CAAS;IACxC,CAAC;IAEJ,6EAA6E;IAC7E,YAAY;QACV,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC;QAC1B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACvF,IAAI,CAAC,wBAAwB,GAAG,IAAI,CAAC;QACrC,IAAI,CAAC,mBAAmB,GAAG,KAAK,CAAC;QACjC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,gBAAgB,GAAG,CAAC,CAAC;QAC1B,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC;QAC7B,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;IAC9B,CAAC;IAED,4CAA4C;IAC5C,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACrD,CAAC;IAED,iEAAiE;IACjE,OAAO;QACL,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;IAC7B,CAAC;IAED,MAAM,CAAC,KAAyB;QAC9B,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;QACjC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,sBAAsB,CAAC;gBAC9C,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;gBACtE,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,GAAG,CAAC;gBAChC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;gBACjC,MAAM;YACR,CAAC;YACD,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,IAAI,CAAC,IAAI,CAAC,eAAe;oBAAE,MAAM;gBACjC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gBACzE,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC,IAAI,CAAC;gBAChC,MAAM;YACR,CAAC;YACD,KAAK,mBAAmB,CAAC,CAAC,CAAC;gBACzB,IAAI,CAAC,IAAI,CAAC,eAAe;oBAAE,MAAM;gBACjC,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBACjC,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACnC,MAAM;YACR,CAAC;YACD,KAAK,SAAS,CAAC,CAAC,CAAC;gBACf,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBACzB,yEAAyE;gBACzE,2CAA2C;gBAC3C,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,eAAe;oBAAE,MAAM;gBAC/C,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC,IAAI,CAAC;gBAClC,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC,IAAI,CAAC;gBAChC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;oBACxB,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC/C,CAAC;qBAAM,CAAC;oBACN,IAAI,IAAI,CAAC,cAAc,KAAK,IAAI;wBAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;oBAC/E,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC1D,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,SAAS,CAAC,CAAC,CAAC;gBACf,MAAM,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC;gBACtB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;gBAC5B,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC,SAAS,KAAK,EAAE,CAAC,SAAS,EAAE,CAAC;oBAC7E,IAAI,CAAC,aAAa,EAAE,CAAC;oBACrB,IAAI,CAAC,eAAe,GAAG,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;gBAC/E,CAAC;gBACD,IAAI,CAAC,eAAe,CAAC,KAAK,IAAI,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC;gBAC7C,IAAI,EAAE,CAAC,KAAK;oBAAE,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC,KAAK,CAAC;gBAC5C,IAAI,EAAE,CAAC,IAAI;oBAAE,IAAI,CAAC,aAAa,EAAE,CAAC;gBAClC,MAAM;YACR,CAAC;YACD,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,IAAI,IAAI,CAAC,eAAe;oBAAE,IAAI,CAAC,eAAe,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;gBAC/E,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK;oBAAE,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;gBAC5D,MAAM;YACR,CAAC;YACD,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI;oBAAE,IAAI,CAAC,aAAa,EAAE,CAAC;gBAC1C,MAAM;YACR,CAAC;YACD,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC;gBAC7B,mEAAmE;gBACnE,qEAAqE;gBACrE,2DAA2D;gBAC3D,IAAI,YAAY,EAAE;oBAAE,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;gBAC3D,MAAM;YACR,CAAC;YACD,kDAAkD;QACpD,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,OAAO;QACL,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,eAAe,CAAC;YAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC7F,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC5B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAC;YAC/B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,EAAE,CAAC;QAChE,CAAC;QAED,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5B,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtB,MAAM,YAAY,GAAG,IAAI,CAAC,cAAc,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAClG,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC;QACxC,OAAO;YACL,YAAY;YACZ,OAAO;YACP,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,UAAU,EAAE,OAAO,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,kBAAkB,CAAC,IAAI,CAAC,YAAY,CAAC;YACrF,OAAO,EAAE,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,eAAe;SAC3C,CAAC;IACJ,CAAC;IAED,gGAAgG;IAChG,SAAS;QACP,IAAI,IAAI,CAAC,cAAc,KAAK,IAAI;YAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;IACjF,CAAC;IAED,iEAAiE;IACjE,QAAQ;QACN,IAAI,IAAI,CAAC,cAAc,KAAK,IAAI;YAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAE3E,yEAAyE;QACzE,yEAAyE;QACzE,sEAAsE;QACtE,uEAAuE;QACvE,4CAA4C;QAC5C,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,KAAK,SAAS;YAAE,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;QAC/F,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,YAAY,IAAI,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACpF,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAC9D,yEAAyE;QACzE,wEAAwE;QACxE,IAAI,IAAI,CAAC,UAAU,EAAE,eAAe,KAAK,SAAS;YAAE,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC;QAC3G,IAAI,IAAI,CAAC,UAAU,EAAE,gBAAgB,KAAK,SAAS;YAAE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,gBAAgB,CAAC;QAC9G,IAAI,IAAI,CAAC,UAAU,EAAE,OAAO,KAAK,SAAS;YAAE,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;QACzF,IAAI,IAAI,CAAC,UAAU,EAAE,UAAU,KAAK,SAAS;YAAE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;QAClG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAEjE,6EAA6E;QAC7E,4EAA4E;QAC5E,sEAAsE;QACtE,kCAAkC;QAClC,EAAE;QACF,2EAA2E;QAC3E,uEAAuE;QACvE,wBAAwB;QACxB,EAAE;QACF,sEAAsE;QACtE,4EAA4E;QAC5E,0EAA0E;QAC1E,6EAA6E;QAC7E,6EAA6E;QAC7E,0EAA0E;QAC1E,uEAAuE;QACvE,oBAAoB;QACpB,MAAM,UAAU,GACd,CAAC,IAAI,CAAC,oBAAoB,IAAI,IAAI,CAAC,gBAAgB,KAAK,CAAC;YACvD,CAAC,CAAC,QAAQ;YACV,CAAC,CAAC,IAAI,CAAC,gBAAgB,GAAG,CAAC;gBACzB,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,MAAM,CAAC;QAEf,QAAQ,CAAC,eAAe,EAAE;YACxB,UAAU;YACV,oBAAoB,EAAE,IAAI,CAAC,oBAAoB;YAC/C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,OAAO,EAAE,IAAI,CAAC,cAAc,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM;YAC3F,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC,CAAC;QAEH,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3C,CAAC;IAEO,yBAAyB;QAC/B,IAAI,IAAI,CAAC,wBAAwB,KAAK,IAAI;YAAE,IAAI,CAAC,wBAAwB,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;QACvG,OAAO,IAAI,CAAC,wBAAwB,CAAC;IACvC,CAAC;IAEO,iBAAiB,CAAC,SAAkB;QAC1C,IAAI,IAAI,CAAC,wBAAwB,KAAK,IAAI,IAAI,IAAI,CAAC,mBAAmB;YAAE,OAAO;QAC/E,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC;QAChC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,wBAAwB,EAAE,SAAS,CAAC,CAAC;IACpE,CAAC;IAEO,YAAY,CAAC,KAAwB;QAC3C,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;YACxB,mEAAmE;YACnE,mEAAmE;YACnE,0EAA0E;YAC1E,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC;QACrB,CAAC;QACD,IAAI,IAA6B,CAAC;QAClC,IAAI,CAAC;YACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAA4B,CAAC;QAC5D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CACV,+CAA+C,KAAK,CAAC,IAAI,iBAAiB,KAAK,CAAC,SAAS,MAAM,eAAe,CAAC,CAAC,CAAC,gBAAgB,KAAK,CAAC,KAAK,CAAC,MAAM,YAAY,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CACpN,CAAC;YACF,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,CAAC,SAAS,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7F,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,CAAC,SAAS,EAAE,cAAc,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QACxG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,CAAC,SAAS,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5G,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,aAAa;QACnB,IAAI,CAAC,IAAI,CAAC,eAAe;YAAE,OAAO;QAClC,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,eAAe,CAAC;YAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACrE,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;IAC9B,CAAC;IAED;;;;;;;;;;;;;;;;OAgBG;IACK,oBAAoB;QAC1B,IAAI,IAAI,CAAC,KAAK,CAAC,oBAAoB,KAAK,KAAK,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,cAAc,KAAK,IAAI;YAAE,OAAO;QAE9G,IAAI,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACpD,MAAM,SAAS,GAAmF,EAAE,CAAC;QACrG,MAAM,aAAa,GAAG,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAClD,IAAI,aAAa,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvC,IAAI,GAAG,aAAa,CAAC,WAAW,CAAC;YACjC,SAAS,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,YAAY,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,YAAY,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,IAAI,GAAG,YAAY,CAAC,WAAW,CAAC;YAChC,SAAS,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEnC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC5B,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;YAC7B,IAAI,IAAI,CAAC,YAAY,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC7G,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,cAAc;QACpB,IAAI,IAAI,CAAC,gBAAgB,KAAK,CAAC,IAAI,IAAI,CAAC,cAAc,KAAK,IAAI;YAAE,OAAO;QACxE,IAAI,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC;YACtE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;CACF"}

package/dist/response-stream.d.ts

@@ -0,0 +1,39 @@
+import { type KiroWireEvent } from "./event-parser.js";
+/** Longest gap between two frames before the response is treated as stalled. */
+export declare const IDLE_TIMEOUT = 300000;
+/**
+ * Why the stream stopped. Every field is a retryable condition; all false means
+ * the response ended normally.
+ */
+export interface KiroEventStreamOutcome {
+    firstTokenTimedOut: boolean;
+    idleTimedOut: boolean;
+    /** Message from a Smithy protocol error or a Kiro `error` frame. */
+    error: string | null;
+}
+export interface KiroWireEventFrame {
+    event: KiroWireEvent;
+    /** The frame verbatim, for logging fields this package does not model. */
+    payload: Record<string, unknown>;
+}
+export interface KiroEventStreamOptions {
+    firstTokenTimeoutMs: number;
+    idleTimeoutMs?: number;
+}
+export interface KiroEventStreamReader {
+    frames: AsyncGenerator<KiroWireEventFrame>;
+    /** Populated by the time `frames` completes. */
+    outcome: KiroEventStreamOutcome;
+}
+/**
+ * Turn a Kiro response body into parsed wire events.
+ *
+ * Terminal conditions are reported through {@link KiroEventStreamOutcome}
+ * rather than thrown: the caller retries all three the same way, and a throw
+ * would force it to re-classify the cause it already knows.
+ *
+ * A Kiro `error` frame ends the stream and is not yielded — it carries no
+ * content, and every caller treats it as the same retryable failure as a
+ * protocol error.
+ */
+export declare function readKiroEventStream(body: ReadableStream<Uint8Array>, options: KiroEventStreamOptions): KiroEventStreamReader;

package/dist/response-stream.js

@@ -0,0 +1,123 @@
+// ABOUTME: Reads Kiro's AWS event-stream response into parsed wire events.
+// ABOUTME: Owns framing and the stall timeouts; block assembly lives elsewhere.
+import { UniversalEventStreamMarshaller } from "@smithy/core/event-streams";
+import { debugEnabled, debugLog } from "./debug.js";
+import { parseKiroEvent } from "./event-parser.js";
+/** Longest gap between two frames before the response is treated as stalled. */
+export const IDLE_TIMEOUT = 300_000;
+const eventStreamMarshaller = new UniversalEventStreamMarshaller({
+    utf8Encoder: (input) => new TextDecoder().decode(input),
+    utf8Decoder: (input) => new TextEncoder().encode(input),
+});
+/**
+ * Turn a Kiro response body into parsed wire events.
+ *
+ * Terminal conditions are reported through {@link KiroEventStreamOutcome}
+ * rather than thrown: the caller retries all three the same way, and a throw
+ * would force it to re-classify the cause it already knows.
+ *
+ * A Kiro `error` frame ends the stream and is not yielded — it carries no
+ * content, and every caller treats it as the same retryable failure as a
+ * protocol error.
+ */
+export function readKiroEventStream(body, options) {
+    const outcome = { firstTokenTimedOut: false, idleTimedOut: false, error: null };
+    const idleTimeoutMs = options.idleTimeoutMs ?? IDLE_TIMEOUT;
+    const bodyReader = body.getReader();
+    async function* frames() {
+        let idleTimer = null;
+        const resetIdle = () => {
+            if (idleTimer)
+                clearTimeout(idleTimer);
+            idleTimer = setTimeout(() => {
+                outcome.idleTimedOut = true;
+                void bodyReader.cancel().catch(() => { });
+            }, idleTimeoutMs);
+        };
+        // Smithy's marshaller handles chunk reassembly, CRC validation, protocol
+        // error/exception detection, and payload deserialization.
+        const bodyIterable = {
+            async *[Symbol.asyncIterator]() {
+                try {
+                    while (true) {
+                        const { done, value } = await bodyReader.read();
+                        if (done)
+                            return;
+                        yield value;
+                    }
+                }
+                finally {
+                    bodyReader.releaseLock();
+                }
+            },
+        };
+        const utf8Decoder = new TextDecoder();
+        const eventStream = eventStreamMarshaller.deserialize(bodyIterable, async (event) => {
+            const entry = Object.entries(event)[0];
+            if (!entry)
+                throw new Error("Received an empty event stream message");
+            const [key, msg] = entry;
+            const parsed = JSON.parse(utf8Decoder.decode(msg.body));
+            return { [key]: parsed };
+        });
+        const iterator = eventStream[Symbol.asyncIterator]();
+        let gotFirstToken = false;
+        const FIRST_TOKEN_SENTINEL = Symbol("firstTokenTimeout");
+        try {
+            while (true) {
+                let iterResult;
+                try {
+                    if (!gotFirstToken) {
+                        const readPromise = iterator.next();
+                        const result = await Promise.race([
+                            readPromise,
+                            new Promise((resolve) => setTimeout(() => resolve(FIRST_TOKEN_SENTINEL), options.firstTokenTimeoutMs)),
+                        ]);
+                        if (result === FIRST_TOKEN_SENTINEL) {
+                            readPromise.catch(() => { }); // suppress dangling rejection
+                            void bodyReader.cancel().catch(() => { });
+                            outcome.firstTokenTimedOut = true;
+                            return;
+                        }
+                        iterResult = result;
+                        gotFirstToken = true;
+                        resetIdle();
+                    }
+                    else {
+                        iterResult = await iterator.next();
+                    }
+                }
+                catch (e) {
+                    // Smithy throws on `:message-type` error/exception headers.
+                    outcome.error =
+                        e instanceof Error
+                            ? e.message
+                            : (typeof e === "object" && e !== null ? JSON.stringify(e) : String(e)) || "Unknown stream error";
+                    return;
+                }
+                const { done, value } = iterResult;
+                if (done)
+                    return;
+                resetIdle();
+                const payload = Object.values(value)[0];
+                const event = parseKiroEvent(payload);
+                if (!event)
+                    continue;
+                if (debugEnabled())
+                    debugLog("stream.events", [event]);
+                if (event.type === "error") {
+                    outcome.error = event.data.message ? `${event.data.error}: ${event.data.message}` : event.data.error;
+                    void bodyReader.cancel().catch(() => { });
+                    return;
+                }
+                yield { event, payload };
+            }
+        }
+        finally {
+            if (idleTimer)
+                clearTimeout(idleTimer);
+        }
+    }
+    return { frames: frames(), outcome };
+}
+//# sourceMappingURL=response-stream.js.map

package/dist/response-stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"response-stream.js","sourceRoot":"","sources":["../src/response-stream.ts"],"names":[],"mappings":"AAAA,2EAA2E;AAC3E,gFAAgF;AAEhF,OAAO,EAAE,8BAA8B,EAAE,MAAM,4BAA4B,CAAC;AAE5E,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,EAAsB,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEvE,gFAAgF;AAChF,MAAM,CAAC,MAAM,YAAY,GAAG,OAAO,CAAC;AAEpC,MAAM,qBAAqB,GAAG,IAAI,8BAA8B,CAAC;IAC/D,WAAW,EAAE,CAAC,KAAiB,EAAE,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;IACnE,WAAW,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;CAChE,CAAC,CAAC;AA8BH;;;;;;;;;;GAUG;AACH,MAAM,UAAU,mBAAmB,CACjC,IAAgC,EAChC,OAA+B;IAE/B,MAAM,OAAO,GAA2B,EAAE,kBAAkB,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IACxG,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,YAAY,CAAC;IAC5D,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAEpC,KAAK,SAAS,CAAC,CAAC,MAAM;QACpB,IAAI,SAAS,GAAyC,IAAI,CAAC;QAC3D,MAAM,SAAS,GAAG,GAAG,EAAE;YACrB,IAAI,SAAS;gBAAE,YAAY,CAAC,SAAS,CAAC,CAAC;YACvC,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC1B,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;gBAC5B,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC3C,CAAC,EAAE,aAAa,CAAC,CAAC;QACpB,CAAC,CAAC;QAEF,yEAAyE;QACzE,0DAA0D;QAC1D,MAAM,YAAY,GAA8B;YAC9C,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;gBAC3B,IAAI,CAAC;oBACH,OAAO,IAAI,EAAE,CAAC;wBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,UAAU,CAAC,IAAI,EAAE,CAAC;wBAChD,IAAI,IAAI;4BAAE,OAAO;wBACjB,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC;wBAAS,CAAC;oBACT,UAAU,CAAC,WAAW,EAAE,CAAC;gBAC3B,CAAC;YACH,CAAC;SACF,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;QACtC,MAAM,WAAW,GAAG,qBAAqB,CAAC,WAAW,CAAC,YAAY,EAAE,KAAK,EAAE,KAA8B,EAAE,EAAE;YAC3G,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACvC,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;YACtE,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC;YACzB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAA4B,CAAC;YACnF,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,MAAM,EAA6B,CAAC;QACtD,CAAC,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC,EAA4C,CAAC;QAE/F,IAAI,aAAa,GAAG,KAAK,CAAC;QAC1B,MAAM,oBAAoB,GAAG,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAEzD,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;gBACZ,IAAI,UAAmD,CAAC;gBACxD,IAAI,CAAC;oBACH,IAAI,CAAC,aAAa,EAAE,CAAC;wBACnB,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;wBACpC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;4BAChC,WAAW;4BACX,IAAI,OAAO,CAA8B,CAAC,OAAO,EAAE,EAAE,CACnD,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,oBAAoB,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAC7E;yBACF,CAAC,CAAC;wBACH,IAAI,MAAM,KAAK,oBAAoB,EAAE,CAAC;4BACpC,WAAW,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC,8BAA8B;4BAC3D,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;4BACzC,OAAO,CAAC,kBAAkB,GAAG,IAAI,CAAC;4BAClC,OAAO;wBACT,CAAC;wBACD,UAAU,GAAG,MAAiD,CAAC;wBAC/D,aAAa,GAAG,IAAI,CAAC;wBACrB,SAAS,EAAE,CAAC;oBACd,CAAC;yBAAM,CAAC;wBACN,UAAU,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACrC,CAAC;gBACH,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,4DAA4D;oBAC5D,OAAO,CAAC,KAAK;wBACX,CAAC,YAAY,KAAK;4BAChB,CAAC,CAAC,CAAC,CAAC,OAAO;4BACX,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,sBAAsB,CAAC;oBACtG,OAAO;gBACT,CAAC;gBAED,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,UAAU,CAAC;gBACnC,IAAI,IAAI;oBAAE,OAAO;gBACjB,SAAS,EAAE,CAAC;gBACZ,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,KAAgC,CAAC,CAAC,CAAC,CAA4B,CAAC;gBAC9F,MAAM,KAAK,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;gBACtC,IAAI,CAAC,KAAK;oBAAE,SAAS;gBACrB,IAAI,YAAY,EAAE;oBAAE,QAAQ,CAAC,eAAe,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;gBACvD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBAC3B,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;oBACrG,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;oBACzC,OAAO;gBACT,CAAC;gBACD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;YAC3B,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,SAAS;gBAAE,YAAY,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,CAAC;AACvC,CAAC"}