ns-kiro-core 0.1.2 → 0.2.0

package/dist/request-builder.d.ts

@@ -0,0 +1,49 @@
+import type { KiroAdditionalModelRequestFields } from "./effort.js";
+import type { KiroModel } from "./models.js";
+import { type KiroHistoryEntry, type KiroUserInputMessage } from "./transform.js";
+import type { KiroMessage, KiroTool } from "./types.js";
+export interface KiroRequest {
+    conversationState: {
+        chatTriggerType: "MANUAL";
+        agentTaskType: "vibe";
+        conversationId: string;
+        currentMessage: {
+            userInputMessage: KiroUserInputMessage;
+        };
+        history?: KiroHistoryEntry[];
+    };
+    additionalModelRequestFields?: KiroAdditionalModelRequestFields;
+    profileArn: string;
+    agentMode?: string;
+}
+export interface BuildKiroRequestParams {
+    messages: KiroMessage[];
+    model: KiroModel;
+    kiroModelId: string;
+    /** Already carries any thinking markers the caller decided to prepend. */
+    systemPrompt: string;
+    tools?: KiroTool[];
+    conversationId: string;
+    profileArn: string;
+    additionalModelRequestFields?: KiroAdditionalModelRequestFields;
+}
+/**
+ * The request plus what was actually put on the wire.
+ *
+ * The counts describe the post-repair body, not the caller's input: they exist
+ * to be correlated against a rejection, so they must describe the bytes sent.
+ */
+export interface BuiltKiroRequest {
+    request: KiroRequest;
+    wireHistoryLength: number;
+    wireContentLength: number;
+    hasImages: boolean;
+    toolResultCount: number;
+}
+/**
+ * Assemble the body for one Kiro runtime call.
+ *
+ * Throws when the local history exceeds its budget, rather than silently
+ * dropping context the caller believes it sent.
+ */
+export declare function buildKiroRequest(params: BuildKiroRequestParams): BuiltKiroRequest;

package/dist/request-builder.js

@@ -0,0 +1,229 @@
+// ABOUTME: Builds the Kiro runtime request body from a neutral conversation.
+// ABOUTME: Pure and side-effect free, so request shaping is testable without a transport.
+import { debugLog } from "./debug.js";
+import { addPlaceholderTools, assertHistoryWithinLimit, HISTORY_LIMIT, HISTORY_LIMIT_CONTEXT_WINDOW, prepareHistory, } from "./history.js";
+import { isKiroToolStructureRule, kiroConversationEntries, repairKiroConversation } from "./history-validator.js";
+import { buildHistory, convertImagesToKiro, convertToolsToKiro, EMPTY_CONTENT_PLACEHOLDER, extractImages, getContentText, normalizeMessages, relocateDisplacedToolResults, sanitizeSurrogates, TOOL_RESULT_LIMIT, toKiroToolUseId, truncate, } from "./transform.js";
+import { TRUNCATION_NOTICE, wasPreviousResponseTruncated } from "./truncation.js";
+function toolResultOf(message) {
+    return {
+        content: [{ text: truncate(getContentText(message), TOOL_RESULT_LIMIT) }],
+        status: message.isError ? "error" : "success",
+        toolUseId: toKiroToolUseId(message.toolCallId),
+    };
+}
+/**
+ * Assemble the body for one Kiro runtime call.
+ *
+ * Throws when the local history exceeds its budget, rather than silently
+ * dropping context the caller believes it sent.
+ */
+export function buildKiroRequest(params) {
+    const { messages, model, kiroModelId, systemPrompt, tools, conversationId, profileArn } = params;
+    // Relocate a tool result that arrived behind a later assistant turn than the
+    // one that called it, before anything positional runs. Interleaved concurrent
+    // tool executions produce that shape, and `sanitizeHistory` pairs
+    // POSITIONALLY, so without this pass the displaced result's issuing assistant
+    // is dropped and the real tool output is discarded.
+    const normalized = relocateDisplacedToolResults(normalizeMessages(messages));
+    const { history: rawHistory, systemPrepended, currentMsgStartIdx, } = buildHistory(normalized, kiroModelId, systemPrompt);
+    // Preserve semantic context locally; the host owns lossy compaction.
+    const history = prepareHistory(rawHistory, model.input.includes("image"));
+    const dynamicHistoryLimit = Math.floor((model.contextWindow / HISTORY_LIMIT_CONTEXT_WINDOW) * HISTORY_LIMIT);
+    const currentMessages = normalized.slice(currentMsgStartIdx);
+    const firstMsg = currentMessages[0];
+    let currentContent = "";
+    const currentToolResults = [];
+    let currentImages;
+    if (firstMsg?.role === "assistant") {
+        let armContent = "";
+        const armToolUses = [];
+        for (const block of firstMsg.content) {
+            if (block.type === "text")
+                armContent += block.text;
+            // Reasoning is deliberately NOT serialized into the assistant text
+            // channel, matching `buildHistory`. Flattening it to
+            // `<thinking>...</thinking>` writes literal markup into the string the
+            // model reads back as its own prior speech.
+            else if (block.type === "toolCall") {
+                armToolUses.push({ name: block.name, toolUseId: toKiroToolUseId(block.id), input: block.arguments });
+            }
+        }
+        if (armContent || armToolUses.length > 0) {
+            const prevArm = history[history.length - 1]?.assistantResponseMessage;
+            if (history.length > 0 && !history[history.length - 1]?.userInputMessage && prevArm) {
+                // Merge into previous assistant message to maintain alternation
+                // without synthetic padding. Join only non-empty sides: a turn that
+                // carried only reasoning or only a tool call leaves `armContent`
+                // empty, and an unconditional separator would append a bare `\n\n`
+                // onto text the model actually produced.
+                prevArm.content =
+                    prevArm.content && armContent ? `${prevArm.content}\n\n${armContent}` : prevArm.content || armContent;
+                if (armToolUses.length > 0)
+                    prevArm.toolUses = [...(prevArm.toolUses || []), ...armToolUses];
+            }
+            else {
+                history.push({
+                    assistantResponseMessage: {
+                        content: armContent,
+                        ...(armToolUses.length > 0 ? { toolUses: armToolUses } : {}),
+                    },
+                });
+            }
+        }
+        const toolResultImages = [];
+        for (let i = 1; i < currentMessages.length; i++) {
+            const m = currentMessages[i];
+            if (m?.role !== "toolResult")
+                continue;
+            currentToolResults.push(toolResultOf(m));
+            toolResultImages.push(...extractImages(m));
+        }
+        if (toolResultImages.length > 0)
+            currentImages = convertImagesToKiro(toolResultImages);
+        // A tool turn carries its payload in `userInputMessageContext.toolResults`,
+        // so it needs no text. Leaving this empty also leaves the fallback below
+        // free to fill in only genuinely payload-less turns.
+        currentContent = "";
+    }
+    else if (firstMsg?.role === "toolResult") {
+        const toolResultImages = [];
+        for (const m of currentMessages) {
+            if (m.role !== "toolResult")
+                continue;
+            currentToolResults.push(toolResultOf(m));
+            toolResultImages.push(...extractImages(m));
+        }
+        if (toolResultImages.length > 0)
+            currentImages = convertImagesToKiro(toolResultImages);
+        // Empty by design — `toolResults` is this turn's payload.
+        currentContent = "";
+    }
+    else if (firstMsg?.role === "user") {
+        currentContent = getContentText(firstMsg);
+        if (systemPrompt && !systemPrepended)
+            currentContent = `${systemPrompt}\n\n${currentContent}`;
+    }
+    // Current assistant tool calls are outbound history too, so enforce the
+    // budget only after they have been appended.
+    assertHistoryWithinLimit(history, dynamicHistoryLimit);
+    if (wasPreviousResponseTruncated(messages)) {
+        currentContent = currentContent === "" ? TRUNCATION_NOTICE : `${TRUNCATION_NOTICE}\n\n${currentContent}`;
+    }
+    // Always synthesize placeholder specs for tool names referenced in history,
+    // even when the caller declares none. Without this, a call that inherits a
+    // tool-rich conversation but declares no current tools is rejected by Kiro
+    // as "Improperly formed request", because history references toolUses with
+    // no tool catalog.
+    let uimc;
+    const baseTools = tools?.length ? convertToolsToKiro(tools) : [];
+    const finalTools = history.length > 0 ? addPlaceholderTools(baseTools, history) : baseTools;
+    if (currentToolResults.length > 0 || finalTools.length > 0) {
+        uimc = {};
+        if (currentToolResults.length > 0)
+            uimc.toolResults = currentToolResults;
+        if (finalTools.length > 0)
+            uimc.tools = finalTools;
+    }
+    if (firstMsg?.role === "user") {
+        const imgs = extractImages(firstMsg);
+        if (imgs.length > 0)
+            currentImages = convertImagesToKiro(imgs);
+    }
+    // A turn with neither text nor tool results has no payload at all: an
+    // image-only user message, or an empty-text one. Send a neutral prompt so
+    // its attachments still reach the model.
+    //
+    // The `currentToolResults` guard is load-bearing. Without it this line
+    // refills every tool turn that deliberately left `currentContent` empty.
+    // Kiro's rule is content **or** tool results — see EMPTY_CONTENT_PLACEHOLDER.
+    if (currentContent === "" && currentToolResults.length === 0)
+        currentContent = EMPTY_CONTENT_PLACEHOLDER;
+    // Pre-send REPAIR against the rules first-party Kiro Agent enforces.
+    // `prepareHistory` covers the shapes this provider itself produces, but not
+    // every shape a caller can hand us: `sanitizeHistory` tests tool pairing by
+    // POSITION, so an assistant entry with `toolUses` survives whenever the next
+    // entry carries any `toolResults` at all, matching ids or not, and
+    // `injectSyntheticToolCalls` only rescues orphaned RESULTS. A mismatched
+    // pair — both partners present, paired with each other's counterpart —
+    // passes both passes untouched and is rejected on the wire with
+    // `400 TOOL_USE_RESULT_MISMATCH`.
+    //
+    // Repair runs on the WHOLE conversation and is split back afterwards.
+    // Repairing `history` alone would be wrong in the ordinary case: its last
+    // entry is normally the assistant whose `toolUses` this very request
+    // answers, so rule 4 would synthesize a FAILED result for a call whose real
+    // output is sitting in the current message.
+    const conversationEntries = kiroConversationEntries(history, {
+        content: currentContent,
+        modelId: kiroModelId,
+        origin: "KIRO_CLI",
+        ...(uimc ? { userInputMessageContext: uimc } : {}),
+    });
+    const repair = repairKiroConversation(conversationEntries);
+    if (repair.diagnostics.length > 0) {
+        debugLog("request.invariants", { errors: repair.diagnostics, remaining: repair.remaining });
+    }
+    // Split back. Repair keeps the current message last in every case but total
+    // collapse, where a conversation that is *only* a bare tool-result carrier
+    // has no valid opening entry and step 1 consumes it.
+    const repairedCurrent = repair.entries[repair.entries.length - 1]?.userInputMessage;
+    // Read the repaired context EXACTLY, including when repair removed it. A
+    // `?? uimc` fallback would undo the repair in the one case that matters
+    // most: stripping every orphaned tool result leaves a turn with no context
+    // at all, and falling back would put the orphans — the shape the backend
+    // rejects — straight back onto the wire.
+    let wireHistory;
+    let wireContent;
+    let wireUimc;
+    if (repairedCurrent) {
+        wireHistory = repair.entries.slice(0, -1);
+        wireContent = repairedCurrent.content;
+        wireUimc = repairedCurrent.userInputMessageContext;
+    }
+    else {
+        // Collapsed. Apply what repair would have applied to a lone carrier: drop
+        // the results that answer nothing, keep any tool catalog, and give the
+        // empty turn the neutral prompt.
+        wireHistory = [];
+        wireContent = currentContent || EMPTY_CONTENT_PLACEHOLDER;
+        wireUimc = uimc?.tools?.length ? { tools: uimc.tools } : undefined;
+    }
+    if (repair.remaining.length > 0) {
+        const structural = repair.remaining.filter((e) => isKiroToolStructureRule(e.rule));
+        if (structural.length > 0) {
+            console.warn(`[kiro-core] outbound history still violates ${structural
+                .map((e) => `${e.rule}@${e.index}`)
+                .join(", ")} after repair — Kiro may reject this request`);
+        }
+    }
+    return {
+        request: {
+            conversationState: {
+                chatTriggerType: "MANUAL",
+                agentTaskType: "vibe",
+                conversationId,
+                currentMessage: {
+                    userInputMessage: {
+                        content: sanitizeSurrogates(wireContent),
+                        modelId: kiroModelId,
+                        origin: "KIRO_CLI",
+                        ...(currentImages ? { images: currentImages } : {}),
+                        ...(wireUimc ? { userInputMessageContext: wireUimc } : {}),
+                    },
+                },
+                ...(wireHistory.length > 0 ? { history: wireHistory } : {}),
+            },
+            ...(params.additionalModelRequestFields
+                ? { additionalModelRequestFields: params.additionalModelRequestFields }
+                : {}),
+            profileArn,
+            agentMode: "vibe",
+        },
+        wireHistoryLength: wireHistory.length,
+        wireContentLength: wireContent.length,
+        hasImages: !!currentImages,
+        toolResultCount: wireUimc?.toolResults?.length ?? 0,
+    };
+}
+//# sourceMappingURL=request-builder.js.map

package/dist/request-builder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-builder.js","sourceRoot":"","sources":["../src/request-builder.ts"],"names":[],"mappings":"AAAA,6EAA6E;AAC7E,0FAA0F;AAE1F,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAEtC,OAAO,EACL,mBAAmB,EACnB,wBAAwB,EACxB,aAAa,EACb,4BAA4B,EAC5B,cAAc,GACf,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AAElH,OAAO,EACL,YAAY,EACZ,mBAAmB,EACnB,kBAAkB,EAClB,yBAAyB,EACzB,aAAa,EACb,cAAc,EAMd,iBAAiB,EACjB,4BAA4B,EAC5B,kBAAkB,EAClB,iBAAiB,EACjB,eAAe,EACf,QAAQ,GACT,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AA4ClF,SAAS,YAAY,CAAC,OAAqD;IACzE,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,iBAAiB,CAAC,EAAE,CAAC;QACzE,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;QAC7C,SAAS,EAAE,eAAe,CAAC,OAAO,CAAC,UAAU,CAAC;KAC/C,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAAC,MAA8B;IAC7D,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,KAAK,EAAE,cAAc,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC;IAEjG,6EAA6E;IAC7E,8EAA8E;IAC9E,kEAAkE;IAClE,8EAA8E;IAC9E,oDAAoD;IACpD,MAAM,UAAU,GAAG,4BAA4B,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC7E,MAAM,EACJ,OAAO,EAAE,UAAU,EACnB,eAAe,EACf,kBAAkB,GACnB,GAAG,YAAY,CAAC,UAAU,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;IACxD,qEAAqE;IACrE,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,EAAE,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,MAAM,mBAAmB,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,aAAa,GAAG,4BAA4B,CAAC,GAAG,aAAa,CAAC,CAAC;IAC7G,MAAM,eAAe,GAAG,UAAU,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IAC7D,MAAM,QAAQ,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC;IACpC,IAAI,cAAc,GAAG,EAAE,CAAC;IACxB,MAAM,kBAAkB,GAAqB,EAAE,CAAC;IAChD,IAAI,aAAsC,CAAC;IAE3C,IAAI,QAAQ,EAAE,IAAI,KAAK,WAAW,EAAE,CAAC;QACnC,IAAI,UAAU,GAAG,EAAE,CAAC;QACpB,MAAM,WAAW,GAA+E,EAAE,CAAC;QACnG,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;gBAAE,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC;YACpD,mEAAmE;YACnE,qDAAqD;YACrD,uEAAuE;YACvE,4CAA4C;iBACvC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACnC,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;YACvG,CAAC;QACH,CAAC;QACD,IAAI,UAAU,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,wBAAwB,CAAC;YACtE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,gBAAgB,IAAI,OAAO,EAAE,CAAC;gBACpF,gEAAgE;gBAChE,oEAAoE;gBACpE,iEAAiE;gBACjE,mEAAmE;gBACnE,yCAAyC;gBACzC,OAAO,CAAC,OAAO;oBACb,OAAO,CAAC,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,OAAO,OAAO,UAAU,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,IAAI,UAAU,CAAC;gBACxG,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC;oBAAE,OAAO,CAAC,QAAQ,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,GAAG,WAAW,CAAC,CAAC;YAC/F,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,IAAI,CAAC;oBACX,wBAAwB,EAAE;wBACxB,OAAO,EAAE,UAAU;wBACnB,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC7D;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,MAAM,gBAAgB,GAAG,EAAE,CAAC;QAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAChD,MAAM,CAAC,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,CAAC,EAAE,IAAI,KAAK,YAAY;gBAAE,SAAS;YACvC,kBAAkB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;YACzC,gBAAgB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC;YAAE,aAAa,GAAG,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;QACvF,4EAA4E;QAC5E,yEAAyE;QACzE,qDAAqD;QACrD,cAAc,GAAG,EAAE,CAAC;IACtB,CAAC;SAAM,IAAI,QAAQ,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;QAC3C,MAAM,gBAAgB,GAAG,EAAE,CAAC;QAC5B,KAAK,MAAM,CAAC,IAAI,eAAe,EAAE,CAAC;YAChC,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY;gBAAE,SAAS;YACtC,kBAAkB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;YACzC,gBAAgB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC;YAAE,aAAa,GAAG,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;QACvF,0DAA0D;QAC1D,cAAc,GAAG,EAAE,CAAC;IACtB,CAAC;SAAM,IAAI,QAAQ,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;QACrC,cAAc,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,YAAY,IAAI,CAAC,eAAe;YAAE,cAAc,GAAG,GAAG,YAAY,OAAO,cAAc,EAAE,CAAC;IAChG,CAAC;IAED,wEAAwE;IACxE,6CAA6C;IAC7C,wBAAwB,CAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;IACvD,IAAI,4BAA4B,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC3C,cAAc,GAAG,cAAc,KAAK,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,GAAG,iBAAiB,OAAO,cAAc,EAAE,CAAC;IAC3G,CAAC;IACD,4EAA4E;IAC5E,2EAA2E;IAC3E,2EAA2E;IAC3E,2EAA2E;IAC3E,mBAAmB;IACnB,IAAI,IAA6C,CAAC;IAClD,MAAM,SAAS,GAAG,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACjE,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5F,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3D,IAAI,GAAG,EAAE,CAAC;QACV,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC;YAAE,IAAI,CAAC,WAAW,GAAG,kBAAkB,CAAC;QACzE,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC;YAAE,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC;IACrD,CAAC;IACD,IAAI,QAAQ,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;QACrC,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC;YAAE,aAAa,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IACjE,CAAC;IACD,sEAAsE;IACtE,0EAA0E;IAC1E,yCAAyC;IACzC,EAAE;IACF,uEAAuE;IACvE,yEAAyE;IACzE,8EAA8E;IAC9E,IAAI,cAAc,KAAK,EAAE,IAAI,kBAAkB,CAAC,MAAM,KAAK,CAAC;QAAE,cAAc,GAAG,yBAAyB,CAAC;IAEzG,qEAAqE;IACrE,4EAA4E;IAC5E,4EAA4E;IAC5E,6EAA6E;IAC7E,mEAAmE;IACnE,yEAAyE;IACzE,uEAAuE;IACvE,gEAAgE;IAChE,kCAAkC;IAClC,EAAE;IACF,sEAAsE;IACtE,0EAA0E;IAC1E,qEAAqE;IACrE,4EAA4E;IAC5E,4CAA4C;IAC5C,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,OAAO,EAAE;QAC3D,OAAO,EAAE,cAAc;QACvB,OAAO,EAAE,WAAW;QACpB,MAAM,EAAE,UAAU;QAClB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,uBAAuB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACnD,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,sBAAsB,CAAC,mBAAmB,CAAC,CAAC;IAC3D,IAAI,MAAM,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAClC,QAAQ,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,WAAW,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;IAC9F,CAAC;IACD,4EAA4E;IAC5E,2EAA2E;IAC3E,qDAAqD;IACrD,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,gBAAgB,CAAC;IACpF,yEAAyE;IACzE,wEAAwE;IACxE,2EAA2E;IAC3E,yEAAyE;IACzE,yCAAyC;IACzC,IAAI,WAA+B,CAAC;IACpC,IAAI,WAAmB,CAAC;IACxB,IAAI,QAAiD,CAAC;IACtD,IAAI,eAAe,EAAE,CAAC;QACpB,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1C,WAAW,GAAG,eAAe,CAAC,OAAO,CAAC;QACtC,QAAQ,GAAG,eAAe,CAAC,uBAAuB,CAAC;IACrD,CAAC;SAAM,CAAC;QACN,0EAA0E;QAC1E,uEAAuE;QACvE,iCAAiC;QACjC,WAAW,GAAG,EAAE,CAAC;QACjB,WAAW,GAAG,cAAc,IAAI,yBAAyB,CAAC;QAC1D,QAAQ,GAAG,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACrE,CAAC;IACD,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,MAAM,UAAU,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACnF,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,IAAI,CACV,+CAA+C,UAAU;iBACtD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC;iBAClC,IAAI,CAAC,IAAI,CAAC,8CAA8C,CAC5D,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO;QACL,OAAO,EAAE;YACP,iBAAiB,EAAE;gBACjB,eAAe,EAAE,QAAQ;gBACzB,aAAa,EAAE,MAAM;gBACrB,cAAc;gBACd,cAAc,EAAE;oBACd,gBAAgB,EAAE;wBAChB,OAAO,EAAE,kBAAkB,CAAC,WAAW,CAAC;wBACxC,OAAO,EAAE,WAAW;wBACpB,MAAM,EAAE,UAAU;wBAClB,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBACnD,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,uBAAuB,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC3D;iBACF;gBACD,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5D;YACD,GAAG,CAAC,MAAM,CAAC,4BAA4B;gBACrC,CAAC,CAAC,EAAE,4BAA4B,EAAE,MAAM,CAAC,4BAA4B,EAAE;gBACvE,CAAC,CAAC,EAAE,CAAC;YACP,UAAU;YACV,SAAS,EAAE,MAAM;SAClB;QACD,iBAAiB,EAAE,WAAW,CAAC,MAAM;QACrC,iBAAiB,EAAE,WAAW,CAAC,MAAM;QACrC,SAAS,EAAE,CAAC,CAAC,aAAa;QAC1B,eAAe,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,IAAI,CAAC;KACpD,CAAC;AACJ,CAAC"}

package/dist/response-assembler.d.ts

@@ -0,0 +1,89 @@
+import type { KiroModel } from "./models.js";
+import type { KiroWireEventFrame } from "./response-stream.js";
+import type { KiroStreamEvent, KiroUsage } from "./types.js";
+/** What the finished attempt produced, before the caller decides to keep it. */
+export interface KiroAttemptSummary {
+    responseText: string;
+    hasText: boolean;
+    sawAnyToolCalls: boolean;
+    emittedToolCalls: number;
+    /** The whole turn is a "Continue" echo taught by synthetic padding. */
+    isEchoLoop: boolean;
+    /** No text and no tool calls: a 200 that said nothing. */
+    isEmpty: boolean;
+}
+export interface KiroCompletedResponse {
+    stopReason: "stop" | "toolUse" | "length";
+    usage: KiroUsage;
+}
+/**
+ * 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 declare class KiroResponseAssembler {
+    private readonly model;
+    private readonly thinkingEnabled;
+    private readonly pending;
+    private readonly blocks;
+    private readonly usage;
+    private totalContent;
+    private lastContentData;
+    private usageEvent;
+    private receivedContextUsage;
+    private thinkingParser;
+    private nativeThinkingBlockIndex;
+    private nativeThinkingEnded;
+    private textBlockIndex;
+    private emittedToolCalls;
+    private sawAnyToolCalls;
+    private currentToolCall;
+    constructor(model: KiroModel, thinkingEnabled: boolean);
+    /** Clear per-attempt state. Block indexes and accumulated usage are kept. */
+    beginAttempt(): void;
+    /** Hand over the events buffered so far. */
+    takeEvents(): KiroStreamEvent[];
+    /** Announce that a retry discarded everything emitted so far. */
+    discard(): void;
+    handle(frame: KiroWireEventFrame): void;
+    /**
+     * Close the turn: flush a trailing tool call, close thinking, and run the
+     * text-dialect recovery and echo-stripping passes.
+     */
+    endTurn(): KiroAttemptSummary;
+    /** Drop an echo the caller decided not to retry, so it is not read as a continuation signal. */
+    stripEcho(): void;
+    /** Close the text block and settle usage and the stop reason. */
+    complete(): KiroCompletedResponse;
+    private ensureNativeThinkingBlock;
+    private endNativeThinking;
+    private emitToolCall;
+    private flushToolCall;
+    /**
+     * 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.
+     */
+    private recoverTextToolCalls;
+    /**
+     * 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.
+     */
+    private stripEchoNoise;
+}