@javargasm/pi-kiro 0.4.3 → 0.4.4

package/dist/extension.js

@@ -551,2534 +551,2380 @@ import { existsSync as existsSync2, readFileSync as readFileSync2, writeFileSync
 import { homedir as homedir2 } from "node:os";
 import { join as join2 } from "node:path";
 
-// src/stream.ts
-init_debug();
-import { calculateCost, createAssistantMessageEventStream } from "@earendil-works/pi-ai";
-
-// src/event-parser.ts
-init_debug();
-function findJsonEnd(text, start) {
-  let braceCount = 0;
-  let inString = false;
-  let escapeNext = false;
-  for (let i = start;i < text.length; i++) {
-    const char = text[i];
-    if (escapeNext) {
-      escapeNext = false;
-      continue;
-    }
-    if (char === "\\") {
-      escapeNext = true;
-      continue;
-    }
-    if (char === '"') {
-      inString = !inString;
-      continue;
-    }
-    if (!inString) {
-      if (char === "{")
-        braceCount++;
-      else if (char === "}") {
-        braceCount--;
-        if (braceCount === 0)
-          return i;
-      }
-    }
-  }
-  return -1;
+// src/models.ts
+var KIRO_MODEL_IDS = new Set([
+  "claude-fable-5",
+  "claude-opus-4.8",
+  "claude-opus-4.7",
+  "claude-opus-4.6",
+  "claude-opus-4.6-1m",
+  "claude-sonnet-4.6",
+  "claude-sonnet-4.6-1m",
+  "claude-opus-4.5",
+  "claude-sonnet-4.5",
+  "claude-sonnet-4.5-1m",
+  "claude-sonnet-4",
+  "claude-haiku-4.5",
+  "deepseek-3.2",
+  "kimi-k2.5",
+  "minimax-m2.1",
+  "minimax-m2.5",
+  "glm-4.7",
+  "glm-4.7-flash",
+  "qwen3-coder-next",
+  "agi-nova-beta-1m",
+  "qwen3-coder-480b",
+  "auto"
+]);
+function dashToDot(modelId) {
+  return modelId.replace(/(\d)-(\d)/g, "$1.$2");
 }
-function parseKiroEvent(parsed) {
-  if (parsed.content !== undefined) {
-    return { type: "content", data: parsed.content };
-  }
-  if (parsed.reasoningText !== undefined || parsed.signature !== undefined || parsed.text !== undefined && !parsed.content && !parsed.name && !parsed.message) {
-    let text = "";
-    let signature;
-    if (parsed.reasoningText) {
-      const rt = parsed.reasoningText;
-      text = (rt.text ?? rt.Text) || "";
-      signature = rt.signature ?? rt.Signature;
-    } else {
-      text = parsed.text || "";
-      signature = parsed.signature;
-    }
-    return {
-      type: "reasoning",
-      data: { text, signature }
-    };
-  }
-  if (parsed.name && parsed.toolUseId) {
-    const rawInput = parsed.input;
-    const input = typeof rawInput === "string" ? rawInput : rawInput && typeof rawInput === "object" && Object.keys(rawInput).length > 0 ? JSON.stringify(rawInput) : "";
-    return {
-      type: "toolUse",
-      data: {
-        name: parsed.name,
-        toolUseId: parsed.toolUseId,
-        input,
-        stop: parsed.stop
-      }
-    };
-  }
-  if (parsed.input !== undefined && !parsed.name) {
-    return {
-      type: "toolUseInput",
-      data: {
-        input: typeof parsed.input === "string" ? parsed.input : JSON.stringify(parsed.input)
-      }
-    };
-  }
-  if (parsed.stop !== undefined && parsed.contextUsagePercentage === undefined) {
-    return { type: "toolUseStop", data: { stop: parsed.stop } };
-  }
-  if (parsed.contextUsagePercentage !== undefined) {
-    return {
-      type: "contextUsage",
-      data: { contextUsagePercentage: parsed.contextUsagePercentage }
-    };
-  }
-  if (parsed.followupPrompt !== undefined) {
-    return { type: "followupPrompt", data: parsed.followupPrompt };
-  }
-  if (parsed.error !== undefined || parsed.Error !== undefined) {
-    const err = parsed.error || parsed.Error || "unknown";
-    const message = parsed.message || parsed.Message || parsed.reason;
-    return {
-      type: "error",
-      data: {
-        error: typeof err === "string" ? err : JSON.stringify(err),
-        message
-      }
-    };
-  }
-  if (parsed.usage !== undefined) {
-    const u = parsed.usage;
-    return {
-      type: "usage",
-      data: {
-        inputTokens: u.inputTokens,
-        outputTokens: u.outputTokens
-      }
-    };
-  }
-  return null;
+function dotToDash(modelId) {
+  return modelId.replace(/(\d)\.(\d)/g, "$1-$2");
 }
-var EVENT_PATTERNS = [
-  '{"content":',
-  '{"reasoningText":',
-  '{"signature":',
-  '{"text":',
-  '{"name":',
-  '{"input":',
-  '{"stop":',
-  '{"contextUsagePercentage":',
-  '{"followupPrompt":',
-  '{"usage":',
-  '{"toolUseId":',
-  '{"unit":',
-  '{"error":',
-  '{"Error":',
-  '{"message":'
-];
-function findNextEventStart(buffer, from) {
-  let earliest = -1;
-  for (const pattern of EVENT_PATTERNS) {
-    const idx = buffer.indexOf(pattern, from);
-    if (idx >= 0 && (earliest < 0 || idx < earliest))
-      earliest = idx;
+function resolveKiroModel(modelId) {
+  const kiroId = dashToDot(modelId);
+  if (!KIRO_MODEL_IDS.has(kiroId)) {
+    throw new Error(`Unknown Kiro model ID: ${modelId}`);
   }
-  return earliest;
+  return kiroId;
 }
-function parseKiroEvents(buffer) {
-  const events = [];
-  let pos = 0;
-  while (pos < buffer.length) {
-    const jsonStart = findNextEventStart(buffer, pos);
-    if (jsonStart < 0) {
-      if (log.isDebug()) {
-        const gap = buffer.substring(pos);
-        const braceIdx = gap.indexOf('{"');
-        if (braceIdx >= 0) {
-          log.debug("event.unmatchedBrace", {
-            from: pos + braceIdx,
-            preview: gap.substring(braceIdx, Math.min(braceIdx + 200, gap.length))
-          });
-        }
-      }
-      break;
-    }
-    if (log.isDebug() && jsonStart > pos) {
-      const skipped = buffer.substring(pos, jsonStart);
-      const braceIdx = skipped.indexOf('{"');
-      if (braceIdx >= 0) {
-        log.debug("event.skippedBrace", {
-          from: pos + braceIdx,
-          preview: skipped.substring(braceIdx, Math.min(braceIdx + 200, skipped.length))
-        });
-      }
-    }
-    const jsonEnd = findJsonEnd(buffer, jsonStart);
-    if (jsonEnd < 0) {
-      return { events, remaining: buffer.substring(jsonStart) };
-    }
-    try {
-      const parsed = JSON.parse(buffer.substring(jsonStart, jsonEnd + 1));
-      const event = parseKiroEvent(parsed);
-      if (event) {
-        events.push(event);
-      } else if (log.isDebug()) {
-        log.debug("event.unknown", { keys: Object.keys(parsed), raw: parsed });
-      }
-    } catch (err) {
-      if (log.isDebug()) {
-        log.debug("event.parseFail", {
-          err: err instanceof Error ? err.message : String(err),
-          snippet: buffer.substring(jsonStart, Math.min(jsonEnd + 1, jsonStart + 200))
-        });
-      }
-    }
-    pos = jsonEnd + 1;
-  }
-  return { events, remaining: "" };
+var API_REGION_MAP = {
+  "us-west-1": "us-east-1",
+  "us-west-2": "us-east-1",
+  "us-east-2": "us-east-1",
+  "eu-west-1": "eu-central-1",
+  "eu-west-2": "eu-central-1",
+  "eu-west-3": "eu-central-1",
+  "eu-north-1": "eu-central-1",
+  "eu-south-1": "eu-central-1",
+  "eu-south-2": "eu-central-1",
+  "eu-central-2": "eu-central-1",
+  "ap-northeast-1": "us-east-1",
+  "ap-northeast-2": "us-east-1",
+  "ap-northeast-3": "us-east-1",
+  "ap-southeast-1": "us-east-1",
+  "ap-southeast-2": "us-east-1",
+  "ap-south-1": "us-east-1",
+  "ap-east-1": "us-east-1",
+  "ap-south-2": "us-east-1",
+  "ap-southeast-3": "us-east-1",
+  "ap-southeast-4": "us-east-1"
+};
+function resolveApiRegion(ssoRegion) {
+  if (!ssoRegion)
+    return "us-east-1";
+  return API_REGION_MAP[ssoRegion] ?? ssoRegion;
 }
-
-// src/health.ts
-var PERMANENT_PATTERNS = [
-  "Invalid refresh token",
-  "Invalid grant provided",
-  "InvalidGrantException",
-  "UnauthorizedClientException",
-  "AuthorizationPendingException",
-  "ExpiredTokenException",
-  "client is not registered",
-  "The security token is expired",
-  "Access denied"
-];
-function isPermanentError(reason) {
-  if (!reason)
-    return false;
-  return PERMANENT_PATTERNS.some((p) => reason.includes(p));
-}
-
-// src/thinking-parser.ts
-init_debug();
-var THINKING_END_TAG = "</thinking>";
-var THINKING_TAG_VARIANTS = [
-  { open: "<thinking>", close: "</thinking>" },
-  { open: "<think>", close: "</think>" },
-  { open: "<reasoning>", close: "</reasoning>" },
-  { open: "<thought>", close: "</thought>" }
-];
-function trailingPrefixLength(text, tag) {
-  const max = Math.min(text.length, tag.length - 1);
-  for (let len = max;len > 0; len--) {
-    if (text.endsWith(tag.slice(0, len)))
-      return len;
+var MODELS_BY_REGION = {
+  "us-east-1": new Set([
+    "claude-fable-5",
+    "claude-opus-4-8",
+    "claude-opus-4-7",
+    "claude-opus-4-6",
+    "claude-opus-4-6-1m",
+    "claude-sonnet-4-6",
+    "claude-sonnet-4-6-1m",
+    "claude-opus-4-5",
+    "claude-sonnet-4-5",
+    "claude-sonnet-4-5-1m",
+    "claude-sonnet-4",
+    "claude-haiku-4-5",
+    "deepseek-3-2",
+    "kimi-k2-5",
+    "minimax-m2-1",
+    "minimax-m2-5",
+    "glm-4-7",
+    "glm-4-7-flash",
+    "qwen3-coder-next",
+    "qwen3-coder-480b",
+    "agi-nova-beta-1m",
+    "auto"
+  ]),
+  "eu-central-1": new Set([
+    "claude-fable-5",
+    "claude-opus-4-8",
+    "claude-opus-4-7",
+    "claude-opus-4-6",
+    "claude-sonnet-4-6",
+    "claude-opus-4-5",
+    "claude-sonnet-4-5",
+    "claude-sonnet-4",
+    "claude-haiku-4-5",
+    "minimax-m2-1",
+    "minimax-m2-5",
+    "qwen3-coder-next",
+    "auto"
+  ])
+};
+function filterModelsByRegion(models, apiRegion) {
+  const allowed = MODELS_BY_REGION[apiRegion];
+  if (!allowed) {
+    console.warn(`[pi-kiro] Unknown API region "${apiRegion}" — no models available. Update MODELS_BY_REGION in models.ts.`);
+    return [];
   }
-  return 0;
+  return models.filter((m) => allowed.has(m.id));
 }
-function maxTrailingPrefixLength(text, tags) {
-  let max = 0;
-  for (const tag of tags) {
-    max = Math.max(max, trailingPrefixLength(text, tag));
-  }
-  return max;
+var RUNTIME_ENDPOINTS = {
+  "us-east-1": "https://runtime.us-east-1.kiro.dev",
+  "eu-central-1": "https://runtime.eu-central-1.kiro.dev"
+};
+function resolveRuntimeUrl(apiRegion) {
+  return RUNTIME_ENDPOINTS[apiRegion] ?? `https://runtime.${apiRegion}.kiro.dev`;
 }
-
-class ThinkingTagParser {
-  output;
-  stream;
-  textBuffer = "";
-  inThinking = false;
-  thinkingExtracted = false;
-  thinkingBlockIndex = null;
-  textBlockIndex = null;
-  lastTextBlockIndex = null;
-  activeEndTag = THINKING_END_TAG;
-  constructor(output, stream) {
-    this.output = output;
-    this.stream = stream;
-  }
-  processChunk(chunk) {
-    this.textBuffer += chunk;
-    if (log.isDebug()) {
-      log.debug("thinking.chunk", {
-        chunkLen: chunk.length,
-        bufferLen: this.textBuffer.length,
-        inThinking: this.inThinking,
-        thinkingExtracted: this.thinkingExtracted
-      });
-    }
-    while (this.textBuffer.length > 0) {
-      const prev = this.textBuffer.length;
-      if (!this.inThinking && !this.thinkingExtracted) {
-        this.processBeforeThinking();
-        if (this.textBuffer.length === 0)
-          break;
-      }
-      if (this.inThinking) {
-        this.processInsideThinking();
-        if (this.textBuffer.length === 0)
-          break;
-      }
-      if (this.thinkingExtracted) {
-        this.processAfterThinking();
-        break;
-      }
-      if (this.textBuffer.length >= prev)
-        break;
-    }
-  }
-  finalize() {
-    if (log.isDebug()) {
-      log.debug("thinking.finalize", {
-        bufferLen: this.textBuffer.length,
-        inThinking: this.inThinking,
-        thinkingExtracted: this.thinkingExtracted,
-        textBlockIndex: this.textBlockIndex,
-        thinkingBlockIndex: this.thinkingBlockIndex
-      });
-    }
-    if (this.textBuffer.length === 0)
-      return;
-    if (this.inThinking && this.thinkingBlockIndex !== null) {
-      const block = this.output.content[this.thinkingBlockIndex];
-      if (block) {
-        block.thinking += this.textBuffer;
-        this.stream.push({
-          type: "thinking_delta",
-          contentIndex: this.thinkingBlockIndex,
-          delta: this.textBuffer,
-          partial: this.output
-        });
-        this.stream.push({
-          type: "thinking_end",
-          contentIndex: this.thinkingBlockIndex,
-          content: block.thinking,
-          partial: this.output
-        });
-      }
-    } else {
-      this.emitText(this.textBuffer);
-    }
-    this.textBuffer = "";
-  }
-  getTextBlockIndex() {
-    return this.textBlockIndex ?? this.lastTextBlockIndex;
-  }
-  processBeforeThinking() {
-    let bestPos = -1;
-    let bestVariant = null;
-    for (const variant of THINKING_TAG_VARIANTS) {
-      const pos = this.textBuffer.indexOf(variant.open);
-      if (pos !== -1 && (bestPos === -1 || pos < bestPos)) {
-        bestPos = pos;
-        bestVariant = variant;
-      }
-    }
-    if (bestPos !== -1 && bestVariant) {
-      if (log.isDebug()) {
-        log.debug("thinking.open", { tag: bestVariant.open, at: bestPos });
-      }
-      if (bestPos > 0)
-        this.emitText(this.textBuffer.slice(0, bestPos));
-      this.textBuffer = this.textBuffer.slice(bestPos + bestVariant.open.length);
-      this.activeEndTag = bestVariant.close;
-      this.inThinking = true;
-      return;
-    }
-    const trailing = maxTrailingPrefixLength(this.textBuffer, THINKING_TAG_VARIANTS.map((v) => v.open));
-    const safeLen = this.textBuffer.length - trailing;
-    if (safeLen > 0) {
-      this.emitText(this.textBuffer.slice(0, safeLen));
-      this.textBuffer = this.textBuffer.slice(safeLen);
-    }
-  }
-  processInsideThinking() {
-    const endPos = this.textBuffer.indexOf(this.activeEndTag);
-    if (endPos !== -1) {
-      if (log.isDebug()) {
-        log.debug("thinking.close", { tag: this.activeEndTag, at: endPos });
-      }
-      if (endPos > 0)
-        this.emitThinking(this.textBuffer.slice(0, endPos));
-      if (this.thinkingBlockIndex !== null) {
-        const block = this.output.content[this.thinkingBlockIndex];
-        if (block) {
-          this.stream.push({
-            type: "thinking_end",
-            contentIndex: this.thinkingBlockIndex,
-            content: block.thinking,
-            partial: this.output
-          });
-        }
-      }
-      this.textBuffer = this.textBuffer.slice(endPos + this.activeEndTag.length);
-      this.inThinking = false;
-      this.thinkingExtracted = true;
-      this.lastTextBlockIndex = this.textBlockIndex;
-      this.textBlockIndex = null;
-      if (this.textBuffer.startsWith(`
-
-`))
-        this.textBuffer = this.textBuffer.slice(2);
-      return;
-    }
-    const trailing = trailingPrefixLength(this.textBuffer, this.activeEndTag);
-    const safeLen = this.textBuffer.length - trailing;
-    if (safeLen > 0) {
-      this.emitThinking(this.textBuffer.slice(0, safeLen));
-      this.textBuffer = this.textBuffer.slice(safeLen);
-    }
-  }
-  processAfterThinking() {
-    this.emitText(this.textBuffer);
-    this.textBuffer = "";
-  }
-  emitText(text) {
-    if (!text)
-      return;
-    if (this.textBlockIndex === null) {
-      this.textBlockIndex = this.output.content.length;
-      this.output.content.push({ type: "text", text: "" });
-      this.stream.push({ type: "text_start", contentIndex: this.textBlockIndex, partial: this.output });
-    }
-    const block = this.output.content[this.textBlockIndex];
-    if (!block)
-      return;
-    block.text += text;
-    this.stream.push({
-      type: "text_delta",
-      contentIndex: this.textBlockIndex,
-      delta: text,
-      partial: this.output
-    });
-  }
-  emitThinking(thinking) {
-    if (!thinking)
-      return;
-    if (this.thinkingBlockIndex === null) {
-      if (this.textBlockIndex !== null) {
-        this.thinkingBlockIndex = this.textBlockIndex;
-        this.output.content.splice(this.thinkingBlockIndex, 0, { type: "thinking", thinking: "" });
-        this.textBlockIndex = this.textBlockIndex + 1;
-      } else {
-        this.thinkingBlockIndex = this.output.content.length;
-        this.output.content.push({ type: "thinking", thinking: "" });
-      }
-      this.stream.push({
-        type: "thinking_start",
-        contentIndex: this.thinkingBlockIndex,
-        partial: this.output
-      });
+var BASE_URL = resolveRuntimeUrl("us-east-1");
+var ZERO_COST = Object.freeze({ input: 0, output: 0, cacheRead: 0, cacheWrite: 0 });
+var KIRO_DEFAULTS = {
+  api: "kiro-api",
+  provider: "kiro",
+  baseUrl: BASE_URL,
+  cost: ZERO_COST
+};
+var MULTIMODAL = ["text", "image"];
+var TEXT_ONLY = ["text"];
+var kiroModels = [
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-fable-5",
+    name: "Claude Fable 5",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 1e6,
+    maxTokens: 128000,
+    firstTokenTimeout: 180000,
+    supportedEfforts: ["low", "medium", "high", "xhigh", "max"],
+    supportsThinkingConfig: true
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-opus-4-8",
+    name: "Claude Opus 4.8",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 1e6,
+    maxTokens: 128000,
+    firstTokenTimeout: 180000,
+    supportedEfforts: ["low", "medium", "high", "xhigh", "max"],
+    supportsThinkingConfig: true
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-opus-4-7",
+    name: "Claude Opus 4.7",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 1e6,
+    maxTokens: 128000,
+    firstTokenTimeout: 180000,
+    supportedEfforts: ["low", "medium", "high", "xhigh", "max"],
+    supportsThinkingConfig: true
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-opus-4-6",
+    name: "Claude Opus 4.6",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 1e6,
+    maxTokens: 64000,
+    supportedEfforts: ["low", "medium", "high", "max"],
+    supportsThinkingConfig: true
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-opus-4-6-1m",
+    name: "Claude Opus 4.6 (1M)",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 1e6,
+    maxTokens: 64000,
+    supportedEfforts: ["low", "medium", "high", "max"],
+    supportsThinkingConfig: true
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-sonnet-4-6",
+    name: "Claude Sonnet 4.6",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 1e6,
+    maxTokens: 64000,
+    supportedEfforts: ["low", "medium", "high", "max"],
+    supportsThinkingConfig: true
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-sonnet-4-6-1m",
+    name: "Claude Sonnet 4.6 (1M)",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 1e6,
+    maxTokens: 64000,
+    supportedEfforts: ["low", "medium", "high", "max"],
+    supportsThinkingConfig: true
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-opus-4-5",
+    name: "Claude Opus 4.5",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 200000,
+    maxTokens: 64000
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-sonnet-4-5",
+    name: "Claude Sonnet 4.5",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 200000,
+    maxTokens: 65536
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-sonnet-4-5-1m",
+    name: "Claude Sonnet 4.5 (1M)",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 1e6,
+    maxTokens: 65536
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-sonnet-4",
+    name: "Claude Sonnet 4",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 200000,
+    maxTokens: 65536
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "claude-haiku-4-5",
+    name: "Claude Haiku 4.5",
+    reasoning: false,
+    input: MULTIMODAL,
+    contextWindow: 200000,
+    maxTokens: 65536
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "deepseek-3-2",
+    name: "DeepSeek 3.2",
+    reasoning: true,
+    input: TEXT_ONLY,
+    contextWindow: 128000,
+    maxTokens: 8192
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "kimi-k2-5",
+    name: "Kimi K2.5",
+    reasoning: true,
+    input: TEXT_ONLY,
+    contextWindow: 200000,
+    maxTokens: 8192
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "minimax-m2-5",
+    name: "MiniMax M2.5",
+    reasoning: false,
+    input: TEXT_ONLY,
+    contextWindow: 196000,
+    maxTokens: 64000
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "minimax-m2-1",
+    name: "MiniMax M2.1",
+    reasoning: false,
+    input: MULTIMODAL,
+    contextWindow: 196000,
+    maxTokens: 64000
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "glm-4-7",
+    name: "GLM 4.7",
+    reasoning: true,
+    input: TEXT_ONLY,
+    contextWindow: 128000,
+    maxTokens: 8192
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "glm-4-7-flash",
+    name: "GLM 4.7 Flash",
+    reasoning: false,
+    input: TEXT_ONLY,
+    contextWindow: 128000,
+    maxTokens: 8192
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "qwen3-coder-next",
+    name: "Qwen3 Coder Next",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 256000,
+    maxTokens: 64000
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "qwen3-coder-480b",
+    name: "Qwen3 Coder 480B",
+    reasoning: true,
+    input: TEXT_ONLY,
+    contextWindow: 128000,
+    maxTokens: 8192
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "agi-nova-beta-1m",
+    name: "AGI Nova Beta (1M)",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 1e6,
+    maxTokens: 65536
+  },
+  {
+    ...KIRO_DEFAULTS,
+    id: "auto",
+    name: "Auto",
+    reasoning: true,
+    input: MULTIMODAL,
+    contextWindow: 200000,
+    maxTokens: 65536
+  }
+];
+async function fetchAvailableModels(accessToken, apiRegion, profileArn) {
+  const url = `https://management.${apiRegion}.kiro.dev/?origin=KIRO_CLI&profileArn=${encodeURIComponent(profileArn)}`;
+  const resp = await fetch(url, {
+    method: "POST",
+    headers: {
+      Authorization: `Bearer ${accessToken}`,
+      "Content-Type": "application/x-amz-json-1.0",
+      "X-Amz-Target": "AmazonCodeWhispererService.ListAvailableModels",
+      "user-agent": "aws-sdk-rust/1.3.15 ua/2.1 api/codewhispererruntime/0.1.16551 os/macos lang/rust/1.92.0 md/appVersion-2.7.1 app/AmazonQ-For-CLI",
+      "x-amz-user-agent": "aws-sdk-rust/1.3.15 ua/2.1 api/codewhispererruntime/0.1.16551 os/macos lang/rust/1.92.0 m/F app/AmazonQ-For-CLI",
+      "x-amzn-codewhisperer-optout": "true",
+      "amz-sdk-request": "attempt=1; max=3",
+      "amz-sdk-invocation-id": crypto.randomUUID(),
+      accept: "*/*",
+      Pragma: "no-cache",
+      "Cache-Control": "no-cache"
+    },
+    body: JSON.stringify({ origin: "KIRO_CLI", profileArn })
+  });
+  if (!resp.ok) {
+    const body = await resp.text().catch(() => "");
+    if (resp.status === 401 || resp.status === 400 && body.includes("Invalid token")) {
+      throw new Error(`Authentication failed: 401 ListAvailableModels failed - ${body}`);
     }
-    const block = this.output.content[this.thinkingBlockIndex];
-    if (!block)
-      return;
-    block.thinking += thinking;
-    this.stream.push({
-      type: "thinking_delta",
-      contentIndex: this.thinkingBlockIndex,
-      delta: thinking,
-      partial: this.output
-    });
+    throw new Error(`ListAvailableModels failed: HTTP ${resp.status} - ${body}`);
   }
+  const data = await resp.json();
+  return (data.models ?? []).filter((m) => m.modelId !== "auto");
 }
-
-// src/tokenizer.ts
-function countTokens(text) {
-  if (!text)
-    return 0;
-  return Math.ceil(text.length / 4);
-}
-
-// src/transform.ts
-import { createHash } from "node:crypto";
-
-// src/kiro-defaults.ts
-var SYSTEM_SEED_INSTRUCTION = `Follow this instruction: # Kiro CLI Default Agent
-
-` + "You are the default Kiro CLI agent, bringing the power of AI-assisted development " + "directly to the user's terminal. You help with coding tasks, system operations, " + `AWS management, and development workflows.
-
-` + `The current model is {{modelId}}.
-`;
-var SYSTEM_SEED_ACK = "I will fully incorporate this information when generating my responses, " + "and explicitly acknowledge relevant parts of the summary when answering questions.";
-function resolveOS() {
-  switch (process.platform) {
-    case "darwin":
-      return "macos";
-    case "win32":
-      return "windows";
-    default:
-      return process.platform;
+var REASONING_FAMILIES = new Set([
+  "claude-fable",
+  "claude-sonnet",
+  "claude-opus",
+  "deepseek",
+  "kimi",
+  "glm",
+  "qwen",
+  "agi-nova",
+  "minimax"
+]);
+function isReasoningModel(dotId) {
+  for (const family of REASONING_FAMILIES) {
+    if (dotId.startsWith(family))
+      return true;
   }
+  return false;
 }
-var COMPACTION_THRESHOLD_PCT = 95;
-
-// src/transform.ts
-function normalizeMessages(messages) {
-  return messages.filter((msg) => msg.role !== "assistant" || msg.stopReason !== "error" && msg.stopReason !== "aborted");
+function firstTokenTimeout(dotId) {
+  if (dotId.startsWith("claude-fable") || dotId.startsWith("claude-opus"))
+    return 180000;
+  return 90000;
 }
-var TOOL_RESULT_LIMIT = 250000;
-var MAX_KIRO_IMAGES = 4;
-var MAX_KIRO_IMAGE_BYTES = 3750000;
-function truncate(text, limit) {
-  if (text.length <= limit)
-    return text;
-  const half = Math.floor(limit / 2);
-  return `${text.substring(0, half)}
-... [TRUNCATED] ...
-${text.substring(text.length - half)}`;
+function buildModelsFromApi(apiModels) {
+  return apiModels.map((m) => {
+    KIRO_MODEL_IDS.add(m.modelId);
+    const dashId = dotToDash(m.modelId);
+    const supportedTypes = m.supportedInputTypes ?? ["TEXT"];
+    const input = supportedTypes.includes("IMAGE") ? ["text", "image"] : ["text"];
+    const effortEnum = m.additionalModelRequestFieldsSchema?.properties?.output_config?.properties?.effort?.enum;
+    const supportedEfforts = Array.isArray(effortEnum) && effortEnum.length > 0 ? effortEnum : undefined;
+    const supportsThinkingConfig = !!m.additionalModelRequestFieldsSchema?.properties?.thinking;
+    return {
+      id: dashId,
+      name: m.modelName,
+      reasoning: isReasoningModel(m.modelId),
+      input,
+      contextWindow: m.tokenLimits?.maxInputTokens ?? 200000,
+      maxTokens: m.tokenLimits?.maxOutputTokens ?? 8192,
+      firstTokenTimeout: firstTokenTimeout(m.modelId),
+      ...supportedEfforts ? { supportedEfforts } : {},
+      ...supportsThinkingConfig ? { supportsThinkingConfig } : {}
+    };
+  });
 }
-function extractImages(msg) {
-  if (msg.role === "toolResult" || typeof msg.content === "string")
-    return [];
-  if (!Array.isArray(msg.content))
-    return [];
-  return msg.content.filter((c) => c.type === "image");
+var cachedDynamicModels = null;
+function getCachedDynamicModels() {
+  return cachedDynamicModels;
 }
-function getContentText(msg) {
-  if (msg.role === "toolResult") {
-    return msg.content.map((c) => c.type === "text" ? c.text : "").join("");
-  }
-  if (typeof msg.content === "string")
-    return msg.content;
-  if (!Array.isArray(msg.content))
-    return "";
-  return msg.content.map((c) => {
-    if (c.type === "text")
-      return c.text;
-    if (c.type === "thinking")
-      return c.thinking;
-    return "";
-  }).join("");
+function setCachedDynamicModels(models) {
+  cachedDynamicModels = models;
 }
-function parseToolArgs(input) {
-  if (input && typeof input === "object")
-    return input;
-  if (typeof input !== "string")
-    return {};
-  try {
-    return JSON.parse(input);
-  } catch {
-    return {};
-  }
+
+// src/oauth.ts
+init_debug();
+var BUILDER_ID_START_URL = "https://view.awsapps.com/start";
+var BUILDER_ID_REGION = "us-east-1";
+var SSO_SCOPES = [
+  "codewhisperer:completions",
+  "codewhisperer:analysis",
+  "codewhisperer:conversations",
+  "codewhisperer:transformations",
+  "codewhisperer:taskassist"
+];
+var IDC_PROBE_REGIONS = [
+  "us-east-1",
+  "eu-west-1",
+  "eu-central-1",
+  "us-east-2",
+  "eu-west-2",
+  "eu-west-3",
+  "eu-north-1",
+  "ap-southeast-1",
+  "ap-northeast-1",
+  "us-west-2"
+];
+var EXPIRES_BUFFER_MS = 5 * 60 * 1000;
+function abortableDelay(ms, signal) {
+  if (signal?.aborted)
+    return Promise.reject(signal.reason ?? new Error("Login cancelled"));
+  return new Promise((resolve2, reject) => {
+    const timer = setTimeout(resolve2, ms);
+    signal?.addEventListener("abort", () => {
+      clearTimeout(timer);
+      reject(signal.reason ?? new Error("Login cancelled"));
+    }, { once: true });
+  });
 }
-var KIRO_TOOL_USE_ID_RE = /^tooluse_[A-Za-z0-9]+$/;
-function toKiroToolUseId(id) {
-  if (KIRO_TOOL_USE_ID_RE.test(id))
-    return id;
-  const digest = createHash("sha256").update(id).digest("hex").slice(0, 22);
-  return `tooluse_${digest}`;
+async function tryRegisterAndAuthorize(startUrl, region) {
+  const oidcEndpoint = `https://oidc.${region}.amazonaws.com`;
+  const regResp = await fetch(`${oidcEndpoint}/client/register`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json", "User-Agent": "pi-kiro" },
+    body: JSON.stringify({
+      clientName: "pi-kiro",
+      clientType: "public",
+      scopes: SSO_SCOPES,
+      grantTypes: ["urn:ietf:params:oauth:grant-type:device_code", "refresh_token"]
+    })
+  });
+  if (!regResp.ok)
+    return null;
+  const { clientId, clientSecret } = await regResp.json();
+  const devResp = await fetch(`${oidcEndpoint}/device_authorization`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json", "User-Agent": "pi-kiro" },
+    body: JSON.stringify({ clientId, clientSecret, startUrl })
+  });
+  if (!devResp.ok)
+    return null;
+  return {
+    clientId,
+    clientSecret,
+    oidcEndpoint,
+    devAuth: await devResp.json()
+  };
 }
-function convertImagesToKiro(images) {
-  let omitted = 0;
-  const valid = [];
-  for (const img of images) {
-    const estimatedBytes = Math.ceil(img.data.length * 3 / 4);
-    if (estimatedBytes > MAX_KIRO_IMAGE_BYTES) {
-      omitted++;
+async function pollForToken(oidcEndpoint, clientId, clientSecret, devAuth, signal) {
+  const deadline = Date.now() + (devAuth.expiresIn || 600) * 1000;
+  const baseInterval = (devAuth.interval || 5) * 1000;
+  let interval = baseInterval;
+  while (Date.now() < deadline) {
+    if (signal?.aborted)
+      throw new Error("Login cancelled");
+    await abortableDelay(interval, signal);
+    let resp;
+    try {
+      resp = await fetch(`${oidcEndpoint}/token`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json", "User-Agent": "pi-kiro" },
+        body: JSON.stringify({
+          clientId,
+          clientSecret,
+          deviceCode: devAuth.deviceCode,
+          grantType: "urn:ietf:params:oauth:grant-type:device_code"
+        })
+      });
+    } catch {
       continue;
     }
-    if (valid.length >= MAX_KIRO_IMAGES) {
-      omitted++;
+    if (resp.status >= 500)
+      continue;
+    let data;
+    try {
+      data = await resp.json();
+    } catch {
+      if (!resp.ok) {
+        throw new Error(`Authorization failed: HTTP ${resp.status}`);
+      }
       continue;
     }
-    valid.push({
-      format: img.mimeType.split("/")[1] || "png",
-      source: { bytes: img.data }
-    });
+    if (!data.error && data.accessToken && data.refreshToken)
+      return data;
+    if (data.error === "authorization_pending")
+      continue;
+    if (data.error === "slow_down") {
+      interval += baseInterval;
+      continue;
+    }
+    if (data.error)
+      throw new Error(`Authorization failed: ${data.error}`);
   }
-  return { images: valid, omitted };
+  throw new Error("Authorization timed out");
 }
-function buildHistory(messages, _modelId, systemPrompt) {
-  const history = [];
-  let systemPrepended = false;
-  let currentMsgStartIdx = messages.length - 1;
-  while (currentMsgStartIdx > 0 && messages[currentMsgStartIdx]?.role === "toolResult") {
-    currentMsgStartIdx--;
-  }
-  const anchor = messages[currentMsgStartIdx];
-  if (anchor?.role === "assistant") {
-    const hasToolCall = Array.isArray(anchor.content) && anchor.content.some((b) => b.type === "toolCall");
-    if (!hasToolCall)
-      currentMsgStartIdx++;
-  }
-  const historyMessages = messages.slice(0, currentMsgStartIdx);
-  for (let i = 0;i < historyMessages.length; i++) {
-    const msg = historyMessages[i];
-    if (!msg)
-      continue;
-    if (msg.role === "user") {
-      let content = typeof msg.content === "string" ? msg.content : getContentText(msg);
-      if (systemPrompt && !systemPrepended) {
-        content = `${systemPrompt}
-
-${content}`;
-        systemPrepended = true;
-      }
-      const images = extractImages(msg);
-      const uim = {
-        content,
-        origin: "KIRO_CLI",
-        ...images.length > 0 ? { images: convertImagesToKiro(images).images } : {}
-      };
-      const prev2 = history[history.length - 1];
-      if (prev2?.userInputMessage) {
-        prev2.userInputMessage.content += `
-
-${uim.content}`;
-        if (uim.images) {
-          prev2.userInputMessage.images = [...prev2.userInputMessage.images ?? [], ...uim.images];
-        }
-      } else {
-        history.push({ userInputMessage: uim });
-      }
-      continue;
-    }
-    if (msg.role === "assistant") {
-      let armContent = "";
-      const armToolUses = [];
-      if (Array.isArray(msg.content)) {
-        for (const block of msg.content) {
-          if (block.type === "text") {
-            armContent += block.text;
-          } else if (block.type === "thinking") {
-            armContent = `<thinking>${block.thinking}</thinking>
-
-${armContent}`;
-          } else if (block.type === "toolCall") {
-            const tc = block;
-            armToolUses.push({
-              name: tc.name,
-              toolUseId: toKiroToolUseId(tc.id),
-              input: parseToolArgs(tc.arguments)
-            });
-          }
-        }
-      }
-      if (!armContent && armToolUses.length === 0)
-        continue;
-      history.push({
-        assistantResponseMessage: {
-          content: armContent,
-          ...armToolUses.length > 0 ? { toolUses: armToolUses } : {}
-        }
-      });
-      continue;
-    }
-    const trMsg = msg;
-    const toolResults = [
-      {
-        content: [{ text: truncate(getContentText(msg), TOOL_RESULT_LIMIT) }],
-        status: trMsg.isError ? "error" : "success",
-        toolUseId: toKiroToolUseId(trMsg.toolCallId)
-      }
-    ];
-    const trImages = [];
-    if (Array.isArray(trMsg.content)) {
-      for (const c of trMsg.content)
-        if (c.type === "image")
-          trImages.push(c);
-    }
-    let j = i + 1;
-    while (j < historyMessages.length && historyMessages[j]?.role === "toolResult") {
-      const next = historyMessages[j];
-      toolResults.push({
-        content: [{ text: truncate(getContentText(next), TOOL_RESULT_LIMIT) }],
-        status: next.isError ? "error" : "success",
-        toolUseId: toKiroToolUseId(next.toolCallId)
-      });
-      if (Array.isArray(next.content)) {
-        for (const c of next.content)
-          if (c.type === "image")
-            trImages.push(c);
-      }
-      j++;
-    }
-    i = j - 1;
-    const prev = history[history.length - 1];
-    if (prev?.userInputMessage) {
-      prev.userInputMessage.content += `
-
-Tool results provided.`;
-      if (trImages.length > 0) {
-        prev.userInputMessage.images = [
-          ...prev.userInputMessage.images ?? [],
-          ...convertImagesToKiro(trImages).images
-        ];
-      }
-      if (!prev.userInputMessage.userInputMessageContext) {
-        prev.userInputMessage.userInputMessageContext = {};
-      }
-      prev.userInputMessage.userInputMessageContext.toolResults = [
-        ...prev.userInputMessage.userInputMessageContext.toolResults ?? [],
-        ...toolResults
-      ];
-    } else {
-      history.push({
-        userInputMessage: {
-          content: "Tool results provided.",
-          origin: "KIRO_CLI",
-          ...trImages.length > 0 ? { images: convertImagesToKiro(trImages).images } : {},
-          userInputMessageContext: { toolResults }
-        }
-      });
-    }
+async function loginKiro(callbacks) {
+  const method = await callbacks.onSelect({
+    message: "Select login method:",
+    options: [
+      { id: "builder-id", label: "AWS Builder ID (personal account)" },
+      { id: "idc", label: "IAM Identity Center (enterprise SSO)" },
+      { id: "sync", label: "Import from Kiro CLI/IDE (auto-sync local DB)" },
+      { id: "desktop", label: "Desktop refresh token (manual)" }
+    ]
+  });
+  if (!method)
+    throw new Error("Login cancelled");
+  if (method === "sync") {
+    return loginCliSync(callbacks);
   }
-  return { history: collapseAgenticLoops(history), systemPrepended, currentMsgStartIdx };
+  if (method === "desktop") {
+    return loginDesktopManual(callbacks);
+  }
+  if (method === "builder-id") {
+    return runDeviceCodeFlow(callbacks, BUILDER_ID_START_URL, [BUILDER_ID_REGION], "builder-id");
+  }
+  const startUrl = (await callbacks.onPrompt({
+    message: "Paste your IAM Identity Center start URL:",
+    placeholder: "https://mycompany.awsapps.com/start",
+    allowEmpty: false
+  }))?.trim();
+  if (!startUrl || !startUrl.startsWith("http")) {
+    throw new Error(`Invalid start URL "${startUrl ?? ""}" — expected https://…`);
+  }
+  const regionRaw = await callbacks.onPrompt({
+    message: `Identity Center region, or blank to auto-detect (${IDC_PROBE_REGIONS.join(", ")})`,
+    placeholder: "us-east-1",
+    allowEmpty: true
+  });
+  const region = (regionRaw ?? "").trim();
+  const regions = region ? [region] : IDC_PROBE_REGIONS;
+  callbacks.onProgress?.(region ? `Connecting to ${region}…` : "Detecting your Identity Center region…");
+  return runDeviceCodeFlow(callbacks, startUrl, regions, "idc");
 }
-function collapseAgenticLoops(history) {
-  if (history.length < 4)
-    return history;
-  const result = [];
-  let i = 0;
-  while (i < history.length) {
-    const entry = history[i];
-    if (entry?.assistantResponseMessage?.toolUses && i + 1 < history.length && history[i + 1]?.userInputMessage?.userInputMessageContext?.toolResults) {
-      let j = i;
-      while (j < history.length) {
-        const asst = history[j];
-        if (!asst?.assistantResponseMessage?.toolUses)
-          break;
-        const nextUser = j + 1 < history.length ? history[j + 1] : null;
-        if (!nextUser?.userInputMessage?.userInputMessageContext?.toolResults)
-          break;
-        j += 2;
-      }
-      const pairCount = (j - i) / 2;
-      if (pairCount > 1) {
-        for (let k = i;k < j; k += 2) {
-          const asst = history[k];
-          const user = history[k + 1];
-          if (k === i) {
-            result.push(asst);
-          } else {
-            result.push({
-              assistantResponseMessage: {
-                content: "[tool calling continues]",
-                toolUses: asst.assistantResponseMessage.toolUses
-              }
-            });
-          }
-          result.push(user);
-        }
-      } else {
-        result.push(history[i], history[i + 1]);
-      }
-      i = j;
-    } else {
-      result.push(entry);
-      i++;
+async function loginCliSync(callbacks) {
+  callbacks.onProgress?.("Scanning for Kiro IDE credentials (~/.kiro/db)…");
+  const { importFromKiroCli: importFromKiroCli2 } = await Promise.resolve().then(() => (init_kiro_cli_sync(), exports_kiro_cli_sync));
+  const imported = await importFromKiroCli2();
+  if (!imported || !imported.accessToken && !imported.refreshToken) {
+    throw new Error(`No Kiro IDE credentials found.
+Make sure Kiro IDE is installed and you're logged in, then try again.
+Alternatively, use 'desktop' to paste a refresh token manually.`);
+  }
+  log.info("Successfully imported credentials from Kiro IDE");
+  callbacks.onProgress?.(`Imported from Kiro IDE (${imported.authMethod}, ${imported.region}${imported.email ? `, ${imported.email}` : ""})`);
+  if (imported.profileArn) {
+    try {
+      const apiRegion = resolveApiRegion(imported.region);
+      const apiModels = await fetchAvailableModels(imported.accessToken, apiRegion, imported.profileArn);
+      setCachedDynamicModels(buildModelsFromApi(apiModels));
+      log.info(`Fetched and cached ${apiModels.length} models after CLI sync`);
+    } catch (err) {
+      log.warn(`Failed to fetch models after CLI sync: ${err}`);
     }
   }
-  return result;
-}
-
-// src/stream.ts
-var FIRST_TOKEN_TIMEOUT_DEFAULT_MS = 90000;
-var IDLE_TIMEOUT_MS = 60000;
-var MAX_RETRIES = 3;
-var MAX_RETRY_DELAY_MS = 1e4;
-var CAPACITY_MAX_RETRIES = 3;
-var CAPACITY_BASE_DELAY_MS = 5000;
-var CAPACITY_MAX_DELAY_MS = 30000;
-var TRANSIENT_MAX_RETRIES = 3;
-var TRANSIENT_BASE_DELAY_MS = 2000;
-var TRANSIENT_MAX_DELAY_MS = 15000;
-var CONTEXT_TRUNCATION_MAX_RETRIES = 3;
-var CONTEXT_TRUNCATION_DROP_RATIO = 0.3;
-var TOO_BIG_PATTERNS = ["CONTENT_LENGTH_EXCEEDS_THRESHOLD", "Input is too long", "Improperly formed"];
-var NON_RETRYABLE_BODY_PATTERNS = ["MONTHLY_REQUEST_COUNT"];
-var CAPACITY_PATTERN = "INSUFFICIENT_MODEL_CAPACITY";
-function exponentialBackoff(attempt, baseMs, maxMs) {
-  return Math.min(baseMs * 2 ** attempt, maxMs);
-}
-function isTooBigError(status, body) {
-  return status === 413 || status === 400 && TOO_BIG_PATTERNS.some((p) => body.includes(p));
-}
-function isNonRetryableBodyError(body) {
-  return NON_RETRYABLE_BODY_PATTERNS.some((p) => body.includes(p));
-}
-function isCapacityError(body) {
-  return body.includes(CAPACITY_PATTERN);
-}
-function isTransientError(status) {
-  return status === 429 || status >= 500;
-}
-function firstTokenTimeoutForModel(modelId) {
-  const m = kiroModels.find((x) => x.id === modelId);
-  return m?.firstTokenTimeout ?? FIRST_TOKEN_TIMEOUT_DEFAULT_MS;
+  return kiroCredsFromCliImport(imported);
 }
-var HIDDEN_REASONING_PLACEHOLDER = "Reasoning hidden by provider";
-var HIDDEN_REASONING_COUNTDOWN_MS = 2000;
-function emitHiddenReasoningLate(output, stream) {
-  const contentIndex = output.content.length;
-  const block = {
-    type: "thinking",
-    thinking: HIDDEN_REASONING_PLACEHOLDER,
-    redacted: true
-  };
-  output.content.push(block);
-  stream.push({ type: "thinking_start", contentIndex, partial: output });
-  stream.push({
-    type: "thinking_delta",
-    contentIndex,
-    delta: HIDDEN_REASONING_PLACEHOLDER,
-    partial: output
-  });
-  stream.push({
-    type: "thinking_end",
-    contentIndex,
-    content: "",
-    partial: output
+async function loginDesktopManual(callbacks) {
+  const refreshRaw = await callbacks.onPrompt({
+    message: `Paste your Kiro desktop refresh token
+` + "(find it in ~/.kiro/db/kiro.db → auth_kv table):",
+    placeholder: "refresh-token",
+    allowEmpty: true
   });
-}
-function abortableDelay(ms, signal) {
-  if (signal?.aborted)
-    return Promise.reject(signal.reason);
-  return new Promise((resolve2, reject) => {
-    const timer = setTimeout(resolve2, ms);
-    signal?.addEventListener("abort", () => {
-      clearTimeout(timer);
-      reject(signal.reason);
-    }, { once: true });
+  const refreshToken = (refreshRaw ?? "").trim();
+  if (!refreshToken) {
+    throw new Error("Login cancelled — no refresh token provided");
+  }
+  const regionRaw = await callbacks.onPrompt({
+    message: "Kiro region:",
+    placeholder: "us-east-1",
+    allowEmpty: true
   });
+  const region = (regionRaw ?? "").trim() || "us-east-1";
+  const refreshCreds = {
+    refresh: `${refreshToken}|||desktop`,
+    access: "",
+    expires: 0,
+    clientId: "",
+    clientSecret: "",
+    region,
+    authMethod: "desktop"
+  };
+  callbacks.onProgress?.("Exchanging refresh token…");
+  return refreshKiroToken(refreshCreds);
 }
-var profileArnCache = new Map;
-var profileArnSkipResolution = false;
-function seedProfileArn(endpoint, arn) {
-  profileArnCache.set(endpoint, arn);
-}
-async function resolveProfileArn(accessToken, endpoint) {
-  if (profileArnSkipResolution)
-    return;
-  const cached = profileArnCache.get(endpoint);
-  if (cached !== undefined)
-    return cached;
-  try {
-    const ep = new URL(endpoint);
-    ep.hostname = ep.hostname.replace("runtime.", "management.");
-    ep.pathname = "/";
-    ep.search = "";
-    ep.hash = "";
-    const resp = await fetch(ep.toString(), {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/x-amz-json-1.0",
-        Authorization: `Bearer ${accessToken}`,
-        "X-Amz-Target": "AmazonCodeWhispererService.ListAvailableProfiles"
-      },
-      body: "{}"
-    });
-    if (!resp.ok) {
-      log.warn(`profileArn resolution failed: ${resp.status} ${resp.statusText}`);
-      return;
-    }
-    const j = await resp.json();
-    const arn = j.profiles?.find((p) => p.arn)?.arn;
-    if (!arn) {
-      log.warn("profileArn resolution returned no profile ARN");
-      return;
+async function runDeviceCodeFlow(callbacks, startUrl, regions, authMethod) {
+  let result = null;
+  let detectedRegion = "";
+  for (const region of regions) {
+    result = await tryRegisterAndAuthorize(startUrl, region);
+    if (result) {
+      detectedRegion = region;
+      if (regions.length > 1)
+        callbacks.onProgress?.(`Region: ${region}`);
+      break;
     }
-    profileArnCache.set(endpoint, arn);
-    return arn;
-  } catch (error) {
-    log.warn(`profileArn resolution threw: ${error instanceof Error ? error.message : String(error)}`);
-    return;
   }
-}
-function uniqueSorted(values) {
-  return [...new Set(values)].sort();
-}
-function duplicateValues(values) {
-  const seen = new Set;
-  const dupes = new Set;
-  for (const value of values) {
-    if (seen.has(value))
-      dupes.add(value);
-    else
-      seen.add(value);
+  if (!result || !detectedRegion) {
+    throw new Error(`Could not authorize ${startUrl} in ${regions.join(", ")}. Check your start URL${regions.length === 1 ? " and region" : ""} and try again.`);
+  }
+  callbacks.onAuth({
+    url: result.devAuth.verificationUriComplete,
+    instructions: `Code: ${result.devAuth.userCode}
+Complete authorization within 10 minutes.`
+  });
+  callbacks.onProgress?.("Waiting for browser authorization (up to 10 minutes)…");
+  const tok = await pollForToken(result.oidcEndpoint, result.clientId, result.clientSecret, result.devAuth, callbacks.signal);
+  if (!tok.accessToken || !tok.refreshToken) {
+    throw new Error("Authorization completed but no tokens returned");
   }
-  return [...dupes].sort();
-}
-function objectKeys(value) {
-  if (!value || typeof value !== "object" || Array.isArray(value))
-    return [];
-  return Object.keys(value).sort();
-}
-function summarizeToolUse(toolUse) {
-  const input = toolUse.input;
   return {
-    name: toolUse.name,
-    toolUseId: toolUse.toolUseId,
-    inputType: Array.isArray(input) ? "array" : typeof input,
-    inputKeys: objectKeys(input)
+    refresh: `${tok.refreshToken}|${result.clientId}|${result.clientSecret}|${authMethod}`,
+    access: tok.accessToken,
+    expires: Date.now() + (tok.expiresIn ?? 3600) * 1000 - EXPIRES_BUFFER_MS,
+    clientId: result.clientId,
+    clientSecret: result.clientSecret,
+    region: detectedRegion,
+    authMethod
   };
 }
-function summarizeToolResult(toolResult) {
-  return {
-    toolUseId: toolResult.toolUseId,
-    status: toolResult.status,
-    contentCount: toolResult.content.length,
-    textChars: toolResult.content.reduce((sum, part) => sum + part.text.length, 0)
-  };
+async function syncBackToKiroCli(result) {
+  if (result.kiroSyncSource !== "kiro-cli-db" || !result.kiroSyncTokenKey) {
+    log.debug("Credential sync-back skipped: credential did not originate from kiro-cli DB");
+    return;
+  }
+  try {
+    const { saveKiroCliCredentials: saveKiroCliCredentials2 } = await Promise.resolve().then(() => (init_kiro_cli_sync(), exports_kiro_cli_sync));
+    const synced = await saveKiroCliCredentials2({
+      accessToken: result.access,
+      refreshToken: result.refresh.split("|")[0] ?? "",
+      region: result.region,
+      authMethod: result.authMethod === "builder-id" ? "idc" : result.authMethod,
+      source: result.kiroSyncSource,
+      tokenKey: result.kiroSyncTokenKey
+    });
+    if (synced)
+      log.info("Synced refreshed credentials back to Kiro CLI DB");
+  } catch (err) {
+    log.debug(`Credential sync-back skipped: ${err}`);
+  }
 }
-function summarizeToolSpec(toolSpec) {
-  const spec = toolSpec.toolSpecification;
-  const schema = spec.inputSchema.json;
-  const properties = schema.properties;
-  const required = schema.required;
+function kiroCredsFromCliImport(imported) {
+  const hasOidcCreds = !!imported.clientId && !!imported.clientSecret;
+  const authMethod = hasOidcCreds && imported.authMethod === "idc" ? "idc" : "desktop";
+  const refreshPacked = hasOidcCreds ? `${imported.refreshToken}|${imported.clientId}|${imported.clientSecret ?? ""}|${authMethod}` : `${imported.refreshToken}|||desktop`;
   return {
-    name: spec.name,
-    descriptionChars: spec.description.length,
-    schemaKeys: objectKeys(schema),
-    propertyNames: properties && typeof properties === "object" && !Array.isArray(properties) ? Object.keys(properties).sort() : [],
-    requiredCount: Array.isArray(required) ? required.length : 0
+    refresh: refreshPacked,
+    access: imported.accessToken,
+    expires: Date.now() + 3600000 - EXPIRES_BUFFER_MS,
+    clientId: imported.clientId ?? "",
+    clientSecret: imported.clientSecret ?? "",
+    region: imported.region,
+    authMethod,
+    profileArn: imported.profileArn,
+    kiroSyncSource: imported.source,
+    kiroSyncTokenKey: imported.tokenKey
   };
 }
-function summarizeHistoryEntry(entry, index) {
-  if (entry.userInputMessage) {
-    const toolResults = entry.userInputMessage.userInputMessageContext?.toolResults ?? [];
-    return {
-      index,
-      role: "user",
-      contentChars: entry.userInputMessage.content.length,
-      imageCount: entry.userInputMessage.images?.length ?? 0,
-      ...toolResults.length > 0 ? { toolResults: toolResults.map(summarizeToolResult) } : {}
-    };
+async function refreshTokenInner(credentials) {
+  const parts = credentials.refresh.split("|");
+  const refreshToken = parts[0] ?? "";
+  const clientId = parts[1] ?? credentials.clientId ?? "";
+  const clientSecret = parts[2] ?? credentials.clientSecret ?? "";
+  const region = credentials.region;
+  const authMethod = credentials.authMethod;
+  if (!refreshToken || !region) {
+    throw new Error("Refresh token is missing region — re-login required");
+  }
+  if (authMethod !== "desktop" && (!clientId || !clientSecret)) {
+    throw new Error("Refresh token is missing clientId/clientSecret — re-login required");
   }
-  if (entry.assistantResponseMessage) {
-    const toolUses = entry.assistantResponseMessage.toolUses ?? [];
+  if (authMethod === "desktop") {
+    const desktopEndpoint = `https://prod.${region}.auth.desktop.kiro.dev/refreshToken`;
+    const resp2 = await fetch(desktopEndpoint, {
+      method: "POST",
+      headers: { "Content-Type": "application/json", "User-Agent": "pi-kiro" },
+      body: JSON.stringify({ refreshToken })
+    });
+    if (!resp2.ok) {
+      const body = await resp2.text().catch(() => "");
+      throw new Error(`Desktop token refresh failed: ${resp2.status} ${body}`);
+    }
+    const data2 = await resp2.json();
+    if (credentials.profileArn) {
+      try {
+        const apiRegion = resolveApiRegion(region);
+        const apiModels = await fetchAvailableModels(data2.accessToken, apiRegion, credentials.profileArn);
+        setCachedDynamicModels(buildModelsFromApi(apiModels));
+        log.info(`Fetched and cached ${apiModels.length} models after desktop token refresh`);
+      } catch (err) {
+        log.warn(`Failed to fetch models after desktop token refresh: ${err}`);
+      }
+    }
     return {
-      index,
-      role: "assistant",
-      contentChars: entry.assistantResponseMessage.content.length,
-      ...toolUses.length > 0 ? { toolUses: toolUses.map(summarizeToolUse) } : {}
+      refresh: `${data2.refreshToken}|||desktop`,
+      access: data2.accessToken,
+      expires: Date.now() + (data2.expiresIn ?? 3600) * 1000 - EXPIRES_BUFFER_MS,
+      clientId: "",
+      clientSecret: "",
+      region,
+      authMethod: "desktop",
+      profileArn: credentials.profileArn,
+      kiroSyncSource: credentials.kiroSyncSource,
+      kiroSyncTokenKey: credentials.kiroSyncTokenKey
     };
   }
-  return { index, role: "unknown", contentChars: 0 };
-}
-function collectToolIntegrity(history, currentToolResults) {
-  const toolUseIds = [];
-  const toolResultIds = currentToolResults.map((toolResult) => toolResult.toolUseId);
-  for (const entry of history) {
-    for (const toolUse of entry.assistantResponseMessage?.toolUses ?? []) {
-      toolUseIds.push(toolUse.toolUseId);
-    }
-    for (const toolResult of entry.userInputMessage?.userInputMessageContext?.toolResults ?? []) {
-      toolResultIds.push(toolResult.toolUseId);
+  const endpoint = `https://oidc.${region}.amazonaws.com/token`;
+  const resp = await fetch(endpoint, {
+    method: "POST",
+    headers: { "Content-Type": "application/json", "User-Agent": "pi-kiro" },
+    body: JSON.stringify({ clientId, clientSecret, refreshToken, grantType: "refresh_token" })
+  });
+  if (!resp.ok) {
+    const body = await resp.text().catch(() => "");
+    throw new Error(`Token refresh failed: ${resp.status} ${body}`);
+  }
+  const data = await resp.json();
+  if (credentials.profileArn) {
+    try {
+      const apiRegion = resolveApiRegion(region);
+      const apiModels = await fetchAvailableModels(data.accessToken, apiRegion, credentials.profileArn);
+      setCachedDynamicModels(buildModelsFromApi(apiModels));
+      log.info(`Fetched and cached ${apiModels.length} models after token refresh`);
+    } catch (err) {
+      log.warn(`Failed to fetch models after token refresh, falling back: ${err}`);
     }
   }
-  const useSet = new Set(toolUseIds);
-  const resultSet = new Set(toolResultIds);
   return {
-    toolUseIds: uniqueSorted(toolUseIds),
-    toolResultIds: uniqueSorted(toolResultIds),
-    duplicateToolUseIds: duplicateValues(toolUseIds),
-    duplicateToolResultIds: duplicateValues(toolResultIds),
-    unmatchedToolResults: uniqueSorted(toolResultIds.filter((id) => !useSet.has(id))),
-    toolUsesWithoutResults: uniqueSorted(toolUseIds.filter((id) => !resultSet.has(id)))
+    refresh: `${data.refreshToken}|${clientId}|${clientSecret}|${authMethod}`,
+    access: data.accessToken,
+    expires: Date.now() + (data.expiresIn ?? 3600) * 1000 - EXPIRES_BUFFER_MS,
+    clientId,
+    clientSecret,
+    region,
+    authMethod,
+    profileArn: credentials.profileArn,
+    kiroSyncSource: credentials.kiroSyncSource,
+    kiroSyncTokenKey: credentials.kiroSyncTokenKey
   };
 }
-function summarizeKiroRequest(request, requestBody) {
-  const current = request.conversationState.currentMessage.userInputMessage;
-  const history = request.conversationState.history ?? [];
-  const currentContext = current.userInputMessageContext;
-  const currentToolResults = currentContext?.toolResults ?? [];
-  const currentTools = currentContext?.tools ?? [];
-  const historySummary = history.map(summarizeHistoryEntry);
-  return {
-    conversationId: request.conversationState.conversationId,
-    modelId: current.modelId,
-    requestJsonChars: requestBody.length,
-    historyLen: history.length,
-    roleSequence: historySummary.map((entry) => entry.role),
-    history: historySummary,
-    current: {
-      contentChars: current.content.length,
-      imageCount: current.images?.length ?? 0,
-      toolResultCount: currentToolResults.length,
-      toolSpecCount: currentTools.length,
-      toolSpecNames: currentTools.map((toolSpec) => toolSpec.toolSpecification.name).sort(),
-      toolResults: currentToolResults.map(summarizeToolResult),
-      toolSpecs: currentTools.map(summarizeToolSpec)
-    },
-    toolIntegrity: collectToolIntegrity(history, currentToolResults),
-    additionalModelRequestFieldKeys: objectKeys(request.additionalModelRequestFields),
-    hasProfileArn: !!request.profileArn,
-    agentMode: request.agentMode
+async function refreshKiroToken(credentials) {
+  const inputMethod = credentials.authMethod;
+  const authMethod = inputMethod === "builder-id" || inputMethod === "idc" || inputMethod === "desktop" ? inputMethod : "idc";
+  if (inputMethod !== undefined && inputMethod !== "builder-id" && inputMethod !== "idc" && inputMethod !== "desktop") {
+    log.warn(`refreshKiroToken: unrecognized authMethod "${String(inputMethod)}" — defaulting to "idc"`);
+  }
+  const baseCreds = {
+    ...credentials,
+    clientId: credentials.clientId ?? credentials.refresh.split("|")[1] ?? "",
+    clientSecret: credentials.clientSecret ?? credentials.refresh.split("|")[2] ?? "",
+    region: credentials.region,
+    authMethod
   };
-}
-function emitToolCall(state, output, stream) {
-  if (!state.input.trim())
-    state.input = "{}";
-  let args;
+  const errors = [];
   try {
-    args = JSON.parse(state.input);
-    if (args && typeof args === "object" && "__tool_use_purpose" in args) {
-      delete args.__tool_use_purpose;
+    log.debug("refresh.cascade: layer 1 — normal refresh");
+    const result = await refreshTokenInner(baseCreds);
+    syncBackToKiroCli(result);
+    return result;
+  } catch (err) {
+    const msg = err instanceof Error ? err.message : String(err);
+    errors.push(`L1(normal): ${msg}`);
+    log.warn(`refresh.cascade: layer 1 failed — ${msg}`);
+  }
+  let freshImport = null;
+  try {
+    log.debug("refresh.cascade: layer 2 — fresh kiro-cli import");
+    const { importFromKiroCli: importFromKiroCli2 } = await Promise.resolve().then(() => (init_kiro_cli_sync(), exports_kiro_cli_sync));
+    freshImport = await importFromKiroCli2();
+    if (freshImport?.accessToken) {
+      const result = kiroCredsFromCliImport(freshImport);
+      log.info("refresh.cascade: layer 2 succeeded — using fresh kiro-cli credentials");
+      return result;
     }
-  } catch (e) {
-    log.warn(`failed to parse tool input for "${state.name}" (${state.toolUseId}): ${e instanceof Error ? e.message : String(e)}`);
-    return false;
+    errors.push("L2(fresh-import): no valid credentials found");
+    log.debug("refresh.cascade: layer 2 — no fresh credentials");
+  } catch (err) {
+    const msg = err instanceof Error ? err.message : String(err);
+    errors.push(`L2(fresh-import): ${msg}`);
+    log.warn(`refresh.cascade: layer 2 failed — ${msg}`);
   }
-  const contentIndex = output.content.length;
-  const toolCall = { type: "toolCall", id: state.toolUseId, name: state.name, arguments: args };
-  output.content.push(toolCall);
-  stream.push({ type: "toolcall_start", contentIndex, partial: output });
-  stream.push({ type: "toolcall_delta", contentIndex, delta: state.input, partial: output });
-  stream.push({ type: "toolcall_end", contentIndex, toolCall, partial: output });
-  return true;
+  if (freshImport?.refreshToken) {
+    try {
+      log.debug("refresh.cascade: layer 3 — refresh fresh kiro-cli creds");
+      const freshCreds = kiroCredsFromCliImport(freshImport);
+      const result = await refreshTokenInner(freshCreds);
+      syncBackToKiroCli(result);
+      log.info("refresh.cascade: layer 3 succeeded — refreshed fresh kiro-cli credentials");
+      return result;
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      errors.push(`L3(refresh-fresh): ${msg}`);
+      log.warn(`refresh.cascade: layer 3 failed — ${msg}`);
+    }
+  }
+  let expiredImport = null;
+  try {
+    log.debug("refresh.cascade: layer 4 — expired kiro-cli import");
+    const { getKiroCliCredentialsAllowExpired: getKiroCliCredentialsAllowExpired2 } = await Promise.resolve().then(() => (init_kiro_cli_sync(), exports_kiro_cli_sync));
+    expiredImport = await getKiroCliCredentialsAllowExpired2(freshImport);
+    if (expiredImport?.accessToken) {
+      const result = kiroCredsFromCliImport(expiredImport);
+      log.info("refresh.cascade: layer 4 succeeded — using expired kiro-cli credentials");
+      return result;
+    } else {
+      errors.push("L4(expired-import): no different expired credentials");
+      log.debug("refresh.cascade: layer 4 — no additional expired credentials");
+    }
+  } catch (err) {
+    const msg = err instanceof Error ? err.message : String(err);
+    errors.push(`L4(expired-import): ${msg}`);
+    log.warn(`refresh.cascade: layer 4 failed — ${msg}`);
+  }
+  if (expiredImport?.refreshToken) {
+    try {
+      log.debug("refresh.cascade: layer 5 — refresh expired kiro-cli creds");
+      const expiredCreds = kiroCredsFromCliImport(expiredImport);
+      const result = await refreshTokenInner(expiredCreds);
+      syncBackToKiroCli(result);
+      log.info("refresh.cascade: layer 5 succeeded — refreshed expired kiro-cli credentials");
+      return result;
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      errors.push(`L5(refresh-expired): ${msg}`);
+      log.warn(`refresh.cascade: layer 5 failed — ${msg}`);
+    }
+  }
+  throw new Error(`Kiro token refresh failed — all 5 cascade layers exhausted. ` + `Re-login required.
+${errors.join(`
+`)}`);
 }
-function streamKiro(model, context, options) {
-  const stream = createAssistantMessageEventStream();
-  (async () => {
-    const output = {
-      role: "assistant",
-      content: [],
-      api: model.api,
-      provider: model.provider,
-      model: model.id,
-      usage: {
-        input: 0,
-        output: 0,
-        cacheRead: 0,
-        cacheWrite: 0,
-        totalTokens: 0,
-        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
-      },
-      stopReason: "stop",
-      timestamp: Date.now()
+
+// src/stream.ts
+init_debug();
+import { calculateCost, createAssistantMessageEventStream } from "@earendil-works/pi-ai";
+
+// src/event-parser.ts
+init_debug();
+function findJsonEnd(text, start) {
+  let braceCount = 0;
+  let inString = false;
+  let escapeNext = false;
+  for (let i = start;i < text.length; i++) {
+    const char = text[i];
+    if (escapeNext) {
+      escapeNext = false;
+      continue;
+    }
+    if (char === "\\") {
+      escapeNext = true;
+      continue;
+    }
+    if (char === '"') {
+      inString = !inString;
+      continue;
+    }
+    if (!inString) {
+      if (char === "{")
+        braceCount++;
+      else if (char === "}") {
+        braceCount--;
+        if (braceCount === 0)
+          return i;
+      }
+    }
+  }
+  return -1;
+}
+function parseKiroEvent(parsed) {
+  if (parsed.content !== undefined) {
+    return { type: "content", data: parsed.content };
+  }
+  if (parsed.reasoningText !== undefined || parsed.signature !== undefined || parsed.text !== undefined && !parsed.content && !parsed.name && !parsed.message) {
+    let text = "";
+    let signature;
+    if (parsed.reasoningText) {
+      const rt = parsed.reasoningText;
+      text = (rt.text ?? rt.Text) || "";
+      signature = rt.signature ?? rt.Signature;
+    } else {
+      text = parsed.text || "";
+      signature = parsed.signature;
+    }
+    return {
+      type: "reasoning",
+      data: { text, signature }
     };
-    let hiddenShimTimer = null;
-    try {
-      const accessToken = options?.apiKey;
-      if (!accessToken) {
-        throw new Error("Kiro credentials not set. Run /login kiro.");
+  }
+  if (parsed.name && parsed.toolUseId) {
+    const rawInput = parsed.input;
+    const input = typeof rawInput === "string" ? rawInput : rawInput && typeof rawInput === "object" && Object.keys(rawInput).length > 0 ? JSON.stringify(rawInput) : "";
+    return {
+      type: "toolUse",
+      data: {
+        name: parsed.name,
+        toolUseId: parsed.toolUseId,
+        input,
+        stop: parsed.stop
       }
-      const endpoint = model.baseUrl || "https://runtime.us-east-1.kiro.dev";
-      const profileArn = await resolveProfileArn(accessToken, endpoint);
-      const kiroModelId = resolveKiroModel(model.id);
-      const thinkingEnabled = !!options?.reasoning || model.reasoning;
-      const reasoningHidden = !!model.reasoningHidden;
-      log.debug("request.init", {
-        endpoint,
-        model: model.id,
-        kiroModelId,
-        contextWindow: model.contextWindow,
-        thinkingEnabled,
-        reasoningHidden,
-        reasoning: options?.reasoning,
-        messageCount: context.messages.length,
-        toolCount: context.tools?.length ?? 0,
-        hasSystemPrompt: !!context.systemPrompt,
-        profileArn,
-        sessionId: options?.sessionId
-      });
-      let systemPrompt = context.systemPrompt ?? "";
-      if (thinkingEnabled && !reasoningHidden) {
-        const budget = options?.reasoning === "xhigh" ? 50000 : options?.reasoning === "high" ? 30000 : options?.reasoning === "medium" ? 20000 : 1e4;
-        systemPrompt = `<thinking_mode>enabled</thinking_mode><max_thinking_length>${budget}</max_thinking_length>${systemPrompt ? `
-${systemPrompt}` : ""}`;
+    };
+  }
+  if (parsed.input !== undefined && !parsed.name) {
+    return {
+      type: "toolUseInput",
+      data: {
+        input: typeof parsed.input === "string" ? parsed.input : JSON.stringify(parsed.input)
       }
-      const envState = {
-        operatingSystem: resolveOS(),
-        currentWorkingDirectory: process.cwd()
-      };
-      const conversationId = options?.sessionId ?? crypto.randomUUID();
-      let retryCount = 0;
-      while (retryCount <= MAX_RETRIES) {
-        if (options?.signal?.aborted)
-          throw options.signal.reason;
-        const normalized = normalizeMessages(context.messages);
-        const {
-          history,
-          systemPrepended,
-          currentMsgStartIdx
-        } = buildHistory(normalized, kiroModelId, systemPrompt);
-        const seedInstruction = SYSTEM_SEED_INSTRUCTION.replace("{{modelId}}", kiroModelId);
-        const seedPair = [
-          { userInputMessage: { content: seedInstruction, origin: "KIRO_CLI" } },
-          { assistantResponseMessage: { content: SYSTEM_SEED_ACK } }
-        ];
-        history.unshift(...seedPair);
-        const currentMessages = normalized.slice(currentMsgStartIdx);
-        const firstMsg = currentMessages[0];
-        let currentContent = "";
-        const currentToolResults = [];
-        let currentImages;
-        if (firstMsg?.role === "assistant") {
-          const am = firstMsg;
-          let armContent = "";
-          const armToolUses = [];
-          if (Array.isArray(am.content)) {
-            for (const b of am.content) {
-              if (b.type === "text") {
-                armContent += b.text;
-              } else if (b.type === "thinking") {
-                armContent = `<thinking>${b.thinking}</thinking>
-
-${armContent}`;
-              } else if (b.type === "toolCall") {
-                const tc = b;
-                armToolUses.push({
-                  name: tc.name,
-                  toolUseId: toKiroToolUseId(tc.id),
-                  input: parseToolArgs(tc.arguments)
-                });
-              }
-            }
-          }
-          if (armContent || armToolUses.length > 0) {
-            const last = history[history.length - 1];
-            if (last && !last.userInputMessage && last.assistantResponseMessage) {
-              last.assistantResponseMessage.content += `
-
-${armContent}`;
-              if (armToolUses.length > 0) {
-                last.assistantResponseMessage.toolUses = [
-                  ...last.assistantResponseMessage.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") {
-              const trm = m;
-              currentToolResults.push({
-                content: [{ text: truncate(getContentText(m), TOOL_RESULT_LIMIT) }],
-                status: trm.isError ? "error" : "success",
-                toolUseId: toKiroToolUseId(trm.toolCallId)
-              });
-              if (Array.isArray(trm.content)) {
-                for (const c of trm.content) {
-                  if (c.type === "image")
-                    toolResultImages.push(c);
-                }
-              }
-            }
-          }
-          if (toolResultImages.length > 0) {
-            const { images: converted, omitted } = convertImagesToKiro(toolResultImages);
-            if (omitted > 0)
-              log.warn(`${omitted} tool-result image(s) omitted (size/count limit)`);
-            currentImages = currentImages ? [...currentImages, ...converted] : converted;
-          }
-          currentContent = currentToolResults.length > 0 ? "Tool results provided." : "Please proceed with the task.";
-        } else if (firstMsg?.role === "toolResult") {
-          const toolResultImages = [];
-          for (const m of currentMessages) {
-            if (m?.role === "toolResult") {
-              const trm = m;
-              currentToolResults.push({
-                content: [{ text: truncate(getContentText(m), TOOL_RESULT_LIMIT) }],
-                status: trm.isError ? "error" : "success",
-                toolUseId: toKiroToolUseId(trm.toolCallId)
-              });
-              if (Array.isArray(trm.content)) {
-                for (const c of trm.content) {
-                  if (c.type === "image")
-                    toolResultImages.push(c);
-                }
-              }
-            }
-          }
-          if (toolResultImages.length > 0) {
-            const { images: converted, omitted } = convertImagesToKiro(toolResultImages);
-            if (omitted > 0)
-              log.warn(`${omitted} tool-result image(s) omitted (size/count limit)`);
-            currentImages = currentImages ? [...currentImages, ...converted] : converted;
-          }
-          currentContent = "Tool results provided.";
-        } else if (firstMsg?.role === "user") {
-          currentContent = typeof firstMsg.content === "string" ? firstMsg.content : getContentText(firstMsg);
-          if (systemPrompt && !systemPrepended) {
-            currentContent = `${systemPrompt}
-
-${currentContent}`;
-          }
-        }
-        const now = new Date;
-        const tzOffset = -now.getTimezoneOffset();
-        const tzSign = tzOffset >= 0 ? "+" : "-";
-        const tzH = String(Math.floor(Math.abs(tzOffset) / 60)).padStart(2, "0");
-        const tzM = String(Math.abs(tzOffset) % 60).padStart(2, "0");
-        const isoLocal = now.getFullYear() + "-" + String(now.getMonth() + 1).padStart(2, "0") + "-" + String(now.getDate()).padStart(2, "0") + "T" + String(now.getHours()).padStart(2, "0") + ":" + String(now.getMinutes()).padStart(2, "0") + ":" + String(now.getSeconds()).padStart(2, "0") + "." + String(now.getMilliseconds()).padStart(3, "0") + tzSign + tzH + ":" + tzM;
-        const weekday = now.toLocaleDateString("en-US", { weekday: "long" });
-        currentContent = `--- CONTEXT ENTRY BEGIN ---
-` + `Current time: ${weekday}, ${isoLocal}
-` + `--- CONTEXT ENTRY END ---
-
-` + `--- USER MESSAGE BEGIN ---
-` + `${currentContent}
-` + `--- USER MESSAGE END ---`;
-        let uimc = {
-          envState
-        };
-        if (currentToolResults.length > 0)
-          uimc.toolResults = currentToolResults;
-        if (context.tools?.length) {
-          const deepCleanSchema = (obj) => {
-            if (Array.isArray(obj)) {
-              return obj.map(deepCleanSchema);
-            } else if (obj !== null && typeof obj === "object") {
-              const cleaned = {};
-              for (const [k, v] of Object.entries(obj)) {
-                if (k === "$schema")
-                  continue;
-                if (k === "required" && Array.isArray(v) && v.length === 0)
-                  continue;
-                cleaned[k] = deepCleanSchema(v);
-              }
-              return cleaned;
-            }
-            return obj;
-          };
-          uimc.tools = context.tools.map((t) => {
-            const params = deepCleanSchema(t.parameters);
-            if (params.properties) {
-              params.properties.__tool_use_purpose = {
-                type: "string",
-                description: "A brief explanation why you are making this tool use."
-              };
-            }
-            return {
-              toolSpecification: {
-                name: t.name,
-                description: t.description || `Use ${t.name}`,
-                inputSchema: { json: params }
-              }
-            };
+    };
+  }
+  if (parsed.stop !== undefined && parsed.contextUsagePercentage === undefined) {
+    return { type: "toolUseStop", data: { stop: parsed.stop } };
+  }
+  if (parsed.contextUsagePercentage !== undefined) {
+    return {
+      type: "contextUsage",
+      data: { contextUsagePercentage: parsed.contextUsagePercentage }
+    };
+  }
+  if (parsed.followupPrompt !== undefined) {
+    return { type: "followupPrompt", data: parsed.followupPrompt };
+  }
+  if (parsed.error !== undefined || parsed.Error !== undefined) {
+    const err = parsed.error || parsed.Error || "unknown";
+    const message = parsed.message || parsed.Message || parsed.reason;
+    return {
+      type: "error",
+      data: {
+        error: typeof err === "string" ? err : JSON.stringify(err),
+        message
+      }
+    };
+  }
+  if (parsed.usage !== undefined) {
+    const u = parsed.usage;
+    return {
+      type: "usage",
+      data: {
+        inputTokens: u.inputTokens,
+        outputTokens: u.outputTokens
+      }
+    };
+  }
+  return null;
+}
+var EVENT_PATTERNS = [
+  '{"content":',
+  '{"reasoningText":',
+  '{"signature":',
+  '{"text":',
+  '{"name":',
+  '{"input":',
+  '{"stop":',
+  '{"contextUsagePercentage":',
+  '{"followupPrompt":',
+  '{"usage":',
+  '{"toolUseId":',
+  '{"unit":',
+  '{"error":',
+  '{"Error":',
+  '{"message":'
+];
+function findNextEventStart(buffer, from) {
+  let earliest = -1;
+  for (const pattern of EVENT_PATTERNS) {
+    const idx = buffer.indexOf(pattern, from);
+    if (idx >= 0 && (earliest < 0 || idx < earliest))
+      earliest = idx;
+  }
+  return earliest;
+}
+function parseKiroEvents(buffer) {
+  const events = [];
+  let pos = 0;
+  while (pos < buffer.length) {
+    const jsonStart = findNextEventStart(buffer, pos);
+    if (jsonStart < 0) {
+      if (log.isDebug()) {
+        const gap = buffer.substring(pos);
+        const braceIdx = gap.indexOf('{"');
+        if (braceIdx >= 0) {
+          log.debug("event.unmatchedBrace", {
+            from: pos + braceIdx,
+            preview: gap.substring(braceIdx, Math.min(braceIdx + 200, gap.length))
           });
         }
-        if (firstMsg?.role === "user") {
-          const imgs = extractImages(firstMsg);
-          if (imgs.length > 0) {
-            const { images: converted, omitted } = convertImagesToKiro(imgs);
-            if (omitted > 0)
-              log.warn(`${omitted} user image(s) omitted (size/count limit)`);
-            currentImages = converted;
-          }
-        }
-        const request = {
-          conversationState: {
-            chatTriggerType: "MANUAL",
-            agentTaskType: "vibe",
-            conversationId,
-            currentMessage: {
-              userInputMessage: {
-                content: currentContent,
-                modelId: kiroModelId,
-                origin: "KIRO_CLI",
-                ...currentImages ? { images: currentImages } : {},
-                userInputMessageContext: uimc
-              }
-            },
-            ...history.length > 0 ? { history } : {}
-          },
-          ...profileArn ? { profileArn } : {},
-          agentMode: "vibe"
-        };
-        const EFFORT_MAP = {
-          minimal: "low",
-          low: "medium",
-          medium: "high",
-          high: "xhigh",
-          xhigh: "max"
-        };
-        const staticModel = kiroModels.find((m) => m.id === model.id);
-        const dynamicModel = getCachedDynamicModels()?.find((m) => m.id === model.id);
-        const supportedEfforts = staticModel?.supportedEfforts ?? dynamicModel?.supportedEfforts;
-        const supportsThinkingConfig = staticModel?.supportsThinkingConfig ?? dynamicModel?.supportsThinkingConfig;
-        if (supportedEfforts && supportedEfforts.length > 0 && options?.reasoning && typeof options.reasoning === "string") {
-          const kiroEffort = EFFORT_MAP[options.reasoning];
-          if (kiroEffort && supportedEfforts.includes(kiroEffort)) {
-            request.additionalModelRequestFields = request.additionalModelRequestFields || {};
-            request.additionalModelRequestFields.output_config = { effort: kiroEffort };
-            log.debug("effort.set", { piReasoning: options.reasoning, kiroEffort, model: model.id });
-          }
-        }
-        if (supportsThinkingConfig && thinkingEnabled) {
-          request.additionalModelRequestFields = request.additionalModelRequestFields || {};
-          request.additionalModelRequestFields.thinking = {
-            type: "adaptive",
-            display: "summarized"
-          };
-          log.debug("thinking.set", { type: "adaptive", display: "summarized", model: model.id });
-        }
-        stream.push({ type: "start", partial: output });
-        if (reasoningHidden && thinkingEnabled && hiddenShimTimer === null) {
-          hiddenShimTimer = setTimeout(() => {
-            hiddenShimTimer = null;
-            emitHiddenReasoningLate(output, stream);
-          }, HIDDEN_REASONING_COUNTDOWN_MS);
-        }
-        let response;
-        let capacityRetryCount = 0;
-        let transientRetryCount = 0;
-        let contextTruncationAttempt = 0;
-        while (true) {
-          const osName = resolveOS();
-          const ua = `aws-sdk-rust/1.3.15 ua/2.1 api/codewhispererstreaming/0.1.16551 os/${osName} lang/rust/1.92.0 md/appVersion-2.7.1 app/AmazonQ-For-CLI`;
-          const xAmzUa = `aws-sdk-rust/1.3.15 ua/2.1 api/codewhispererstreaming/0.1.16551 os/${osName} lang/rust/1.92.0 m/F app/AmazonQ-For-CLI`;
-          const requestBody = JSON.stringify(request);
-          const requestShape = log.isDebug() ? summarizeKiroRequest(request, requestBody) : undefined;
-          if (requestShape)
-            log.debug("request.shape", requestShape);
-          log.debug("request.send", {
-            attempt: retryCount,
-            capacityAttempt: capacityRetryCount,
-            historyLen: history.length,
-            currentContentLen: currentContent.length,
-            hasImages: !!currentImages,
-            toolResultCount: currentToolResults.length,
-            requestJsonChars: requestBody.length
-          });
-          response = await fetch(endpoint, {
-            method: "POST",
-            headers: {
-              "Content-Type": "application/x-amz-json-1.0",
-              Accept: "*/*",
-              "Accept-Encoding": "gzip",
-              Authorization: `Bearer ${accessToken}`,
-              "X-Amz-Target": "AmazonCodeWhispererStreamingService.GenerateAssistantResponse",
-              "x-amzn-codewhisperer-optout": "true",
-              "amz-sdk-invocation-id": crypto.randomUUID(),
-              "amz-sdk-request": "attempt=1; max=3",
-              "user-agent": ua,
-              "x-amz-user-agent": xAmzUa,
-              Pragma: "no-cache",
-              "Cache-Control": "no-cache"
-            },
-            body: requestBody,
-            signal: options?.signal
-          });
-          if (response.ok)
-            break;
-          let errText = "";
-          try {
-            errText = await response.text();
-          } catch {
-            errText = "";
-          }
-          log.debug("response.error", {
-            status: response.status,
-            body: errText,
-            ...requestShape ? { requestShape } : {}
+      }
+      break;
+    }
+    if (log.isDebug() && jsonStart > pos) {
+      const skipped = buffer.substring(pos, jsonStart);
+      const braceIdx = skipped.indexOf('{"');
+      if (braceIdx >= 0) {
+        log.debug("event.skippedBrace", {
+          from: pos + braceIdx,
+          preview: skipped.substring(braceIdx, Math.min(braceIdx + 200, skipped.length))
+        });
+      }
+    }
+    const jsonEnd = findJsonEnd(buffer, jsonStart);
+    if (jsonEnd < 0) {
+      return { events, remaining: buffer.substring(jsonStart) };
+    }
+    try {
+      const parsed = JSON.parse(buffer.substring(jsonStart, jsonEnd + 1));
+      const event = parseKiroEvent(parsed);
+      if (event) {
+        events.push(event);
+      } else if (log.isDebug()) {
+        log.debug("event.unknown", { keys: Object.keys(parsed), raw: parsed });
+      }
+    } catch (err) {
+      if (log.isDebug()) {
+        log.debug("event.parseFail", {
+          err: err instanceof Error ? err.message : String(err),
+          snippet: buffer.substring(jsonStart, Math.min(jsonEnd + 1, jsonStart + 200))
+        });
+      }
+    }
+    pos = jsonEnd + 1;
+  }
+  return { events, remaining: "" };
+}
+
+// src/health.ts
+var PERMANENT_PATTERNS = [
+  "Invalid refresh token",
+  "Invalid grant provided",
+  "InvalidGrantException",
+  "UnauthorizedClientException",
+  "AuthorizationPendingException",
+  "ExpiredTokenException",
+  "client is not registered",
+  "The security token is expired",
+  "Access denied"
+];
+function isPermanentError(reason) {
+  if (!reason)
+    return false;
+  return PERMANENT_PATTERNS.some((p) => reason.includes(p));
+}
+
+// src/thinking-parser.ts
+init_debug();
+var THINKING_END_TAG = "</thinking>";
+var THINKING_TAG_VARIANTS = [
+  { open: "<thinking>", close: "</thinking>" },
+  { open: "<think>", close: "</think>" },
+  { open: "<reasoning>", close: "</reasoning>" },
+  { open: "<thought>", close: "</thought>" }
+];
+function trailingPrefixLength(text, tag) {
+  const max = Math.min(text.length, tag.length - 1);
+  for (let len = max;len > 0; len--) {
+    if (text.endsWith(tag.slice(0, len)))
+      return len;
+  }
+  return 0;
+}
+function maxTrailingPrefixLength(text, tags) {
+  let max = 0;
+  for (const tag of tags) {
+    max = Math.max(max, trailingPrefixLength(text, tag));
+  }
+  return max;
+}
+
+class ThinkingTagParser {
+  output;
+  stream;
+  textBuffer = "";
+  inThinking = false;
+  thinkingExtracted = false;
+  thinkingBlockIndex = null;
+  textBlockIndex = null;
+  lastTextBlockIndex = null;
+  activeEndTag = THINKING_END_TAG;
+  constructor(output, stream) {
+    this.output = output;
+    this.stream = stream;
+  }
+  processChunk(chunk) {
+    this.textBuffer += chunk;
+    if (log.isDebug()) {
+      log.debug("thinking.chunk", {
+        chunkLen: chunk.length,
+        bufferLen: this.textBuffer.length,
+        inThinking: this.inThinking,
+        thinkingExtracted: this.thinkingExtracted
+      });
+    }
+    while (this.textBuffer.length > 0) {
+      const prev = this.textBuffer.length;
+      if (!this.inThinking && !this.thinkingExtracted) {
+        this.processBeforeThinking();
+        if (this.textBuffer.length === 0)
+          break;
+      }
+      if (this.inThinking) {
+        this.processInsideThinking();
+        if (this.textBuffer.length === 0)
+          break;
+      }
+      if (this.thinkingExtracted) {
+        this.processAfterThinking();
+        break;
+      }
+      if (this.textBuffer.length >= prev)
+        break;
+    }
+  }
+  finalize() {
+    if (log.isDebug()) {
+      log.debug("thinking.finalize", {
+        bufferLen: this.textBuffer.length,
+        inThinking: this.inThinking,
+        thinkingExtracted: this.thinkingExtracted,
+        textBlockIndex: this.textBlockIndex,
+        thinkingBlockIndex: this.thinkingBlockIndex
+      });
+    }
+    if (this.textBuffer.length === 0)
+      return;
+    if (this.inThinking && this.thinkingBlockIndex !== null) {
+      const block = this.output.content[this.thinkingBlockIndex];
+      if (block) {
+        block.thinking += this.textBuffer;
+        this.stream.push({
+          type: "thinking_delta",
+          contentIndex: this.thinkingBlockIndex,
+          delta: this.textBuffer,
+          partial: this.output
+        });
+        this.stream.push({
+          type: "thinking_end",
+          contentIndex: this.thinkingBlockIndex,
+          content: block.thinking,
+          partial: this.output
+        });
+      }
+    } else {
+      this.emitText(this.textBuffer);
+    }
+    this.textBuffer = "";
+  }
+  getTextBlockIndex() {
+    return this.textBlockIndex ?? this.lastTextBlockIndex;
+  }
+  processBeforeThinking() {
+    let bestPos = -1;
+    let bestVariant = null;
+    for (const variant of THINKING_TAG_VARIANTS) {
+      const pos = this.textBuffer.indexOf(variant.open);
+      if (pos !== -1 && (bestPos === -1 || pos < bestPos)) {
+        bestPos = pos;
+        bestVariant = variant;
+      }
+    }
+    if (bestPos !== -1 && bestVariant) {
+      if (log.isDebug()) {
+        log.debug("thinking.open", { tag: bestVariant.open, at: bestPos });
+      }
+      if (bestPos > 0)
+        this.emitText(this.textBuffer.slice(0, bestPos));
+      this.textBuffer = this.textBuffer.slice(bestPos + bestVariant.open.length);
+      this.activeEndTag = bestVariant.close;
+      this.inThinking = true;
+      return;
+    }
+    const trailing = maxTrailingPrefixLength(this.textBuffer, THINKING_TAG_VARIANTS.map((v) => v.open));
+    const safeLen = this.textBuffer.length - trailing;
+    if (safeLen > 0) {
+      this.emitText(this.textBuffer.slice(0, safeLen));
+      this.textBuffer = this.textBuffer.slice(safeLen);
+    }
+  }
+  processInsideThinking() {
+    const endPos = this.textBuffer.indexOf(this.activeEndTag);
+    if (endPos !== -1) {
+      if (log.isDebug()) {
+        log.debug("thinking.close", { tag: this.activeEndTag, at: endPos });
+      }
+      if (endPos > 0)
+        this.emitThinking(this.textBuffer.slice(0, endPos));
+      if (this.thinkingBlockIndex !== null) {
+        const block = this.output.content[this.thinkingBlockIndex];
+        if (block) {
+          this.stream.push({
+            type: "thinking_end",
+            contentIndex: this.thinkingBlockIndex,
+            content: block.thinking,
+            partial: this.output
           });
-          if (isCapacityError(errText) && capacityRetryCount < CAPACITY_MAX_RETRIES) {
-            capacityRetryCount++;
-            const delayMs = exponentialBackoff(capacityRetryCount - 1, CAPACITY_BASE_DELAY_MS, CAPACITY_MAX_DELAY_MS);
-            log.warn(`INSUFFICIENT_MODEL_CAPACITY — retrying in ${delayMs}ms (${capacityRetryCount}/${CAPACITY_MAX_RETRIES})`);
-            await abortableDelay(delayMs, options?.signal);
-            continue;
-          }
-          if (isNonRetryableBodyError(errText) || isCapacityError(errText)) {
-            throw new Error(`Kiro API error: ${errText || response.statusText}`);
-          }
-          if (isTooBigError(response.status, errText)) {
-            if (contextTruncationAttempt < CONTEXT_TRUNCATION_MAX_RETRIES && history.length > 0) {
-              contextTruncationAttempt++;
-              const dropCount = Math.max(1, Math.floor(history.length * CONTEXT_TRUNCATION_DROP_RATIO));
-              const before = history.length;
-              history.splice(0, dropCount);
-              log.warn(`context too large — truncated history from ${before} to ${history.length} entries ` + `(attempt ${contextTruncationAttempt}/${CONTEXT_TRUNCATION_MAX_RETRIES})`);
-              request.conversationState.history = history.length > 0 ? history : undefined;
-              continue;
-            }
-            throw new Error(`Kiro API error: context_length_exceeded (${response.status} ${errText})`);
-          }
-          if (isTransientError(response.status) && transientRetryCount < TRANSIENT_MAX_RETRIES) {
-            transientRetryCount++;
-            const jitter = Math.floor(Math.random() * 1000);
-            const delayMs = exponentialBackoff(transientRetryCount - 1, TRANSIENT_BASE_DELAY_MS, TRANSIENT_MAX_DELAY_MS) + jitter;
-            log.warn(`transient error ${response.status} — retrying in ${delayMs}ms ` + `(${transientRetryCount}/${TRANSIENT_MAX_RETRIES})`);
-            await abortableDelay(delayMs, options?.signal);
-            continue;
-          }
-          if (response.status === 401) {
-            const permanent = isPermanentError(errText);
-            if (permanent) {
-              profileArnCache.delete(endpoint);
-              throw new Error(`Kiro API error: credentials permanently invalid — run /login kiro to re-authenticate. ${errText}`);
-            }
-          }
-          if (response.status === 403) {
-            profileArnCache.delete(endpoint);
-            throw new Error(`Kiro API error: access token rejected (403) — run /login kiro to re-authenticate. ${errText}`);
-          }
-          throw new Error(`Kiro API error: ${response.status} ${response.statusText} ${errText}`);
         }
-        if (capacityRetryCount > 0) {
-          log.info(`recovered from capacity pressure after ${capacityRetryCount} retries`);
+      }
+      this.textBuffer = this.textBuffer.slice(endPos + this.activeEndTag.length);
+      this.inThinking = false;
+      this.thinkingExtracted = true;
+      this.lastTextBlockIndex = this.textBlockIndex;
+      this.textBlockIndex = null;
+      if (this.textBuffer.startsWith(`
+
+`))
+        this.textBuffer = this.textBuffer.slice(2);
+      return;
+    }
+    const trailing = trailingPrefixLength(this.textBuffer, this.activeEndTag);
+    const safeLen = this.textBuffer.length - trailing;
+    if (safeLen > 0) {
+      this.emitThinking(this.textBuffer.slice(0, safeLen));
+      this.textBuffer = this.textBuffer.slice(safeLen);
+    }
+  }
+  processAfterThinking() {
+    this.emitText(this.textBuffer);
+    this.textBuffer = "";
+  }
+  emitText(text) {
+    if (!text)
+      return;
+    if (this.textBlockIndex === null) {
+      this.textBlockIndex = this.output.content.length;
+      this.output.content.push({ type: "text", text: "" });
+      this.stream.push({ type: "text_start", contentIndex: this.textBlockIndex, partial: this.output });
+    }
+    const block = this.output.content[this.textBlockIndex];
+    if (!block)
+      return;
+    block.text += text;
+    this.stream.push({
+      type: "text_delta",
+      contentIndex: this.textBlockIndex,
+      delta: text,
+      partial: this.output
+    });
+  }
+  emitThinking(thinking) {
+    if (!thinking)
+      return;
+    if (this.thinkingBlockIndex === null) {
+      if (this.textBlockIndex !== null) {
+        this.thinkingBlockIndex = this.textBlockIndex;
+        this.output.content.splice(this.thinkingBlockIndex, 0, { type: "thinking", thinking: "" });
+        this.textBlockIndex = this.textBlockIndex + 1;
+      } else {
+        this.thinkingBlockIndex = this.output.content.length;
+        this.output.content.push({ type: "thinking", thinking: "" });
+      }
+      this.stream.push({
+        type: "thinking_start",
+        contentIndex: this.thinkingBlockIndex,
+        partial: this.output
+      });
+    }
+    const block = this.output.content[this.thinkingBlockIndex];
+    if (!block)
+      return;
+    block.thinking += thinking;
+    this.stream.push({
+      type: "thinking_delta",
+      contentIndex: this.thinkingBlockIndex,
+      delta: thinking,
+      partial: this.output
+    });
+  }
+}
+
+// src/tokenizer.ts
+function countTokens(text) {
+  if (!text)
+    return 0;
+  return Math.ceil(text.length / 4);
+}
+
+// src/transform.ts
+import { createHash } from "node:crypto";
+function normalizeMessages(messages) {
+  return messages.filter((msg) => msg.role !== "assistant" || msg.stopReason !== "error" && msg.stopReason !== "aborted");
+}
+var TOOL_RESULT_LIMIT = 250000;
+var MAX_KIRO_IMAGES = 4;
+var MAX_KIRO_IMAGE_BYTES = 3750000;
+function truncate(text, limit) {
+  if (text.length <= limit)
+    return text;
+  const half = Math.floor(limit / 2);
+  return `${text.substring(0, half)}
+... [TRUNCATED] ...
+${text.substring(text.length - half)}`;
+}
+function extractImages(msg) {
+  if (msg.role === "toolResult" || typeof msg.content === "string")
+    return [];
+  if (!Array.isArray(msg.content))
+    return [];
+  return msg.content.filter((c) => c.type === "image");
+}
+function getContentText(msg) {
+  if (msg.role === "toolResult") {
+    return msg.content.map((c) => c.type === "text" ? c.text : "").join("");
+  }
+  if (typeof msg.content === "string")
+    return msg.content;
+  if (!Array.isArray(msg.content))
+    return "";
+  return msg.content.map((c) => {
+    if (c.type === "text")
+      return c.text;
+    if (c.type === "thinking")
+      return c.thinking;
+    return "";
+  }).join("");
+}
+function parseToolArgs(input) {
+  if (input && typeof input === "object")
+    return input;
+  if (typeof input !== "string")
+    return {};
+  try {
+    return JSON.parse(input);
+  } catch {
+    return {};
+  }
+}
+var KIRO_TOOL_USE_ID_RE = /^tooluse_[A-Za-z0-9]+$/;
+function toKiroToolUseId(id) {
+  if (KIRO_TOOL_USE_ID_RE.test(id))
+    return id;
+  const digest = createHash("sha256").update(id).digest("hex").slice(0, 22);
+  return `tooluse_${digest}`;
+}
+function convertImagesToKiro(images) {
+  let omitted = 0;
+  const valid = [];
+  for (const img of images) {
+    const estimatedBytes = Math.ceil(img.data.length * 3 / 4);
+    if (estimatedBytes > MAX_KIRO_IMAGE_BYTES) {
+      omitted++;
+      continue;
+    }
+    if (valid.length >= MAX_KIRO_IMAGES) {
+      omitted++;
+      continue;
+    }
+    valid.push({
+      format: img.mimeType.split("/")[1] || "png",
+      source: { bytes: img.data }
+    });
+  }
+  return { images: valid, omitted };
+}
+function buildHistory(messages, _modelId, systemPrompt) {
+  const history = [];
+  let systemPrepended = false;
+  let currentMsgStartIdx = messages.length - 1;
+  while (currentMsgStartIdx > 0 && messages[currentMsgStartIdx]?.role === "toolResult") {
+    currentMsgStartIdx--;
+  }
+  const anchor = messages[currentMsgStartIdx];
+  if (anchor?.role === "assistant") {
+    const hasToolCall = Array.isArray(anchor.content) && anchor.content.some((b) => b.type === "toolCall");
+    if (!hasToolCall)
+      currentMsgStartIdx++;
+  }
+  const historyMessages = messages.slice(0, currentMsgStartIdx);
+  for (let i = 0;i < historyMessages.length; i++) {
+    const msg = historyMessages[i];
+    if (!msg)
+      continue;
+    if (msg.role === "user") {
+      let content = typeof msg.content === "string" ? msg.content : getContentText(msg);
+      if (systemPrompt && !systemPrepended) {
+        content = `${systemPrompt}
+
+${content}`;
+        systemPrepended = true;
+      }
+      const images = extractImages(msg);
+      const uim = {
+        content,
+        origin: "KIRO_CLI",
+        ...images.length > 0 ? { images: convertImagesToKiro(images).images } : {}
+      };
+      const prev2 = history[history.length - 1];
+      if (prev2?.userInputMessage) {
+        prev2.userInputMessage.content += `
+
+${uim.content}`;
+        if (uim.images) {
+          prev2.userInputMessage.images = [...prev2.userInputMessage.images ?? [], ...uim.images];
         }
-        if (transientRetryCount > 0) {
-          log.info(`recovered from transient error after ${transientRetryCount} retries`);
+      } else {
+        history.push({ userInputMessage: uim });
+      }
+      continue;
+    }
+    if (msg.role === "assistant") {
+      let armContent = "";
+      const armToolUses = [];
+      if (Array.isArray(msg.content)) {
+        for (const block of msg.content) {
+          if (block.type === "text") {
+            armContent += block.text;
+          } else if (block.type === "thinking") {
+            armContent = `<thinking>${block.thinking}</thinking>
+
+${armContent}`;
+          } else if (block.type === "toolCall") {
+            const tc = block;
+            armToolUses.push({
+              name: tc.name,
+              toolUseId: toKiroToolUseId(tc.id),
+              input: parseToolArgs(tc.arguments)
+            });
+          }
         }
-        if (contextTruncationAttempt > 0) {
-          log.info(`recovered after ${contextTruncationAttempt} context truncation(s)`);
+      }
+      if (!armContent && armToolUses.length === 0)
+        continue;
+      history.push({
+        assistantResponseMessage: {
+          content: armContent,
+          ...armToolUses.length > 0 ? { toolUses: armToolUses } : {}
         }
-        const reader = response.body?.getReader();
-        if (!reader)
-          throw new Error("No response body");
-        const decoder = new TextDecoder;
-        let buffer = "";
-        let totalContent = "";
-        let lastContentData = "";
-        let usageEvent = null;
-        let receivedContextUsage = false;
-        let chunkSeq = 0;
-        let eventSeq = 0;
-        const thinkingParser = thinkingEnabled ? new ThinkingTagParser(output, stream) : null;
-        let textBlockIndex = null;
-        let emittedToolCalls = 0;
-        let sawAnyToolCalls = false;
-        let currentToolCall = null;
-        const flushToolCall = () => {
-          if (!currentToolCall)
-            return;
-          if (emitToolCall(currentToolCall, output, stream))
-            emittedToolCalls++;
-          currentToolCall = null;
-        };
-        const cancelHiddenShim = () => {
-          if (hiddenShimTimer) {
-            clearTimeout(hiddenShimTimer);
-            hiddenShimTimer = null;
-          }
-        };
-        let idleTimer = null;
-        let idleCancelled = false;
-        const resetIdle = () => {
-          if (idleTimer)
-            clearTimeout(idleTimer);
-          idleTimer = setTimeout(() => {
-            idleCancelled = true;
-            reader.cancel().catch(() => {});
-          }, IDLE_TIMEOUT_MS);
-        };
-        let gotFirstToken = false;
-        let firstTokenTimedOut = false;
-        let streamError = null;
-        const FIRST_TOKEN_SENTINEL = Symbol("firstTokenTimeout");
-        while (true) {
-          let readResult;
-          if (!gotFirstToken) {
-            const readPromise = reader.read();
-            let firstTokenTimer = null;
-            const result = await Promise.race([
-              readPromise,
-              new Promise((resolve2) => {
-                firstTokenTimer = setTimeout(() => resolve2(FIRST_TOKEN_SENTINEL), firstTokenTimeoutForModel(model.id));
-              })
-            ]);
-            if (firstTokenTimer)
-              clearTimeout(firstTokenTimer);
-            if (result === FIRST_TOKEN_SENTINEL) {
-              readPromise.catch(() => {});
-              reader.cancel().catch(() => {});
-              firstTokenTimedOut = true;
-              break;
-            }
-            readResult = result;
-            gotFirstToken = true;
-            resetIdle();
+      });
+      continue;
+    }
+    const trMsg = msg;
+    const toolResults = [
+      {
+        content: [{ text: truncate(getContentText(msg), TOOL_RESULT_LIMIT) }],
+        status: trMsg.isError ? "error" : "success",
+        toolUseId: toKiroToolUseId(trMsg.toolCallId)
+      }
+    ];
+    const trImages = [];
+    if (Array.isArray(trMsg.content)) {
+      for (const c of trMsg.content)
+        if (c.type === "image")
+          trImages.push(c);
+    }
+    let j = i + 1;
+    while (j < historyMessages.length && historyMessages[j]?.role === "toolResult") {
+      const next = historyMessages[j];
+      toolResults.push({
+        content: [{ text: truncate(getContentText(next), TOOL_RESULT_LIMIT) }],
+        status: next.isError ? "error" : "success",
+        toolUseId: toKiroToolUseId(next.toolCallId)
+      });
+      if (Array.isArray(next.content)) {
+        for (const c of next.content)
+          if (c.type === "image")
+            trImages.push(c);
+      }
+      j++;
+    }
+    i = j - 1;
+    const prev = history[history.length - 1];
+    if (prev?.userInputMessage) {
+      prev.userInputMessage.content += `
+
+Tool results provided.`;
+      if (trImages.length > 0) {
+        prev.userInputMessage.images = [
+          ...prev.userInputMessage.images ?? [],
+          ...convertImagesToKiro(trImages).images
+        ];
+      }
+      if (!prev.userInputMessage.userInputMessageContext) {
+        prev.userInputMessage.userInputMessageContext = {};
+      }
+      prev.userInputMessage.userInputMessageContext.toolResults = [
+        ...prev.userInputMessage.userInputMessageContext.toolResults ?? [],
+        ...toolResults
+      ];
+    } else {
+      history.push({
+        userInputMessage: {
+          content: "Tool results provided.",
+          origin: "KIRO_CLI",
+          ...trImages.length > 0 ? { images: convertImagesToKiro(trImages).images } : {},
+          userInputMessageContext: { toolResults }
+        }
+      });
+    }
+  }
+  return { history: collapseAgenticLoops(history), systemPrepended, currentMsgStartIdx };
+}
+function collapseAgenticLoops(history) {
+  if (history.length < 4)
+    return history;
+  const result = [];
+  let i = 0;
+  while (i < history.length) {
+    const entry = history[i];
+    if (entry?.assistantResponseMessage?.toolUses && i + 1 < history.length && history[i + 1]?.userInputMessage?.userInputMessageContext?.toolResults) {
+      let j = i;
+      while (j < history.length) {
+        const asst = history[j];
+        if (!asst?.assistantResponseMessage?.toolUses)
+          break;
+        const nextUser = j + 1 < history.length ? history[j + 1] : null;
+        if (!nextUser?.userInputMessage?.userInputMessageContext?.toolResults)
+          break;
+        j += 2;
+      }
+      const pairCount = (j - i) / 2;
+      if (pairCount > 1) {
+        for (let k = i;k < j; k += 2) {
+          const asst = history[k];
+          const user = history[k + 1];
+          if (k === i) {
+            result.push(asst);
           } else {
-            readResult = await reader.read();
-          }
-          const { done, value } = readResult;
-          if (done)
-            break;
-          const decoded = decoder.decode(value, { stream: true });
-          buffer += decoded;
-          if (log.isDebug()) {
-            log.debug("stream.chunk", {
-              seq: chunkSeq++,
-              bytes: value?.byteLength ?? 0,
-              decodedLen: decoded.length,
-              preview: previewChunk(decoded)
+            result.push({
+              assistantResponseMessage: {
+                content: "[tool calling continues]",
+                toolUses: asst.assistantResponseMessage.toolUses
+              }
             });
           }
-          const { events, remaining } = parseKiroEvents(buffer);
-          buffer = remaining;
-          if (events.length > 0)
-            resetIdle();
-          if (log.isDebug() && events.length > 0) {
-            for (const ev of events) {
-              log.debug("stream.event", { seq: eventSeq++, event: ev });
-            }
-          }
-          for (const event of events) {
-            switch (event.type) {
-              case "contextUsage": {
-                const pct = event.data.contextUsagePercentage;
-                output.usage.input = pct >= COMPACTION_THRESHOLD_PCT ? model.contextWindow + 1 : Math.round(pct / 100 * model.contextWindow);
-                receivedContextUsage = true;
-                log.debug("contextUsage", { pct, threshold: COMPACTION_THRESHOLD_PCT, willCompact: pct >= COMPACTION_THRESHOLD_PCT });
-                break;
-              }
-              case "reasoning": {
-                cancelHiddenShim();
-                if (output.content.length === 0 || output.content[output.content.length - 1]?.type !== "thinking") {
-                  output.content.push({ type: "thinking", thinking: "" });
-                  stream.push({ type: "thinking_start", contentIndex: output.content.length - 1, partial: output });
-                }
-                const contentIndex = output.content.length - 1;
-                const tc = output.content[contentIndex];
-                if (event.data.text) {
-                  tc.thinking += event.data.text;
-                  stream.push({ type: "thinking_delta", contentIndex, delta: event.data.text, partial: output });
-                }
-                if (event.data.signature) {
-                  tc.thinkingSignature = event.data.signature;
-                }
-                break;
-              }
-              case "content": {
-                if (event.data === lastContentData)
-                  continue;
-                lastContentData = event.data;
-                totalContent += event.data;
-                cancelHiddenShim();
-                if (thinkingParser) {
-                  thinkingParser.processChunk(event.data);
-                } else {
-                  if (textBlockIndex === null) {
-                    textBlockIndex = output.content.length;
-                    output.content.push({ type: "text", text: "" });
-                    stream.push({ type: "text_start", contentIndex: textBlockIndex, partial: output });
-                  }
-                  const block = output.content[textBlockIndex];
-                  if (block) {
-                    block.text += event.data;
-                    stream.push({
-                      type: "text_delta",
-                      contentIndex: textBlockIndex,
-                      delta: event.data,
-                      partial: output
-                    });
-                  }
-                }
-                break;
-              }
-              case "toolUse": {
-                const tc = event.data;
-                cancelHiddenShim();
-                sawAnyToolCalls = true;
-                if (!currentToolCall || currentToolCall.toolUseId !== tc.toolUseId) {
-                  flushToolCall();
-                  currentToolCall = { toolUseId: tc.toolUseId, name: tc.name, input: "" };
-                }
-                currentToolCall.input += tc.input || "";
-                if (tc.input)
-                  totalContent += tc.input;
-                if (tc.stop)
-                  flushToolCall();
-                break;
-              }
-              case "toolUseInput": {
-                if (currentToolCall)
-                  currentToolCall.input += event.data.input || "";
-                if (event.data.input)
-                  totalContent += event.data.input;
-                break;
-              }
-              case "toolUseStop": {
-                if (event.data.stop)
-                  flushToolCall();
-                break;
-              }
-              case "usage": {
-                usageEvent = event.data;
-                break;
-              }
-              case "error": {
-                streamError = event.data.message ? `${event.data.error}: ${event.data.message}` : event.data.error;
-                reader.cancel().catch(() => {});
-                break;
-              }
-            }
-            if (streamError)
-              break;
-          }
-        }
-        if (idleTimer)
-          clearTimeout(idleTimer);
-        if (firstTokenTimedOut || idleCancelled || streamError) {
-          if (retryCount < MAX_RETRIES) {
-            retryCount++;
-            const delayMs = exponentialBackoff(retryCount - 1, 1000, MAX_RETRY_DELAY_MS);
-            log.warn(`stream ${firstTokenTimedOut ? "first-token timed out" : idleCancelled ? "idle timed out" : `error: ${streamError}`} — retrying (${retryCount}/${MAX_RETRIES})`);
-            cancelHiddenShim();
-            await abortableDelay(delayMs, options?.signal);
-            output.content = [];
-            textBlockIndex = null;
-            continue;
-          }
-          if (streamError)
-            throw new Error(`Kiro API stream error after max retries: ${streamError}`);
-          throw new Error(`Kiro API error: ${firstTokenTimedOut ? "first token" : "idle"} timeout after max retries`);
-        }
-        cancelHiddenShim();
-        if (currentToolCall && emitToolCall(currentToolCall, output, stream))
-          emittedToolCalls++;
-        if (thinkingParser) {
-          thinkingParser.finalize();
-          textBlockIndex = thinkingParser.getTextBlockIndex();
-        }
-        if (textBlockIndex !== null) {
-          const block = output.content[textBlockIndex];
-          if (block) {
-            stream.push({
-              type: "text_end",
-              contentIndex: textBlockIndex,
-              content: block.text,
-              partial: output
-            });
-          }
-        }
-        if (usageEvent?.inputTokens !== undefined)
-          output.usage.input = usageEvent.inputTokens;
-        output.usage.output = usageEvent?.outputTokens ?? countTokens(totalContent);
-        output.usage.totalTokens = output.usage.input + output.usage.output;
-        try {
-          calculateCost(model, output.usage);
-        } catch {
-          output.usage.cost = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 };
-        }
-        const textBlock = textBlockIndex !== null ? output.content[textBlockIndex] : undefined;
-        const hasText = !!textBlock && textBlock.text.length > 0;
-        if (!hasText && !sawAnyToolCalls) {
-          if (retryCount < MAX_RETRIES) {
-            retryCount++;
-            const delayMs = exponentialBackoff(retryCount - 1, 1000, MAX_RETRY_DELAY_MS);
-            log.warn(`empty response — retrying (${retryCount}/${MAX_RETRIES})`);
-            cancelHiddenShim();
-            output.content = [];
-            textBlockIndex = null;
-            await abortableDelay(delayMs, options?.signal);
-            continue;
-          }
-          log.warn(`empty response persisted after ${MAX_RETRIES} retries`);
-          cancelHiddenShim();
-        }
-        if (!receivedContextUsage && emittedToolCalls === 0) {
-          output.stopReason = "length";
-        } else {
-          output.stopReason = emittedToolCalls > 0 ? "toolUse" : "stop";
+          result.push(user);
         }
-        stream.push({
-          type: "done",
-          reason: output.stopReason,
-          message: output
-        });
-        log.debug("response.done", {
-          stopReason: output.stopReason,
-          emittedToolCalls,
-          sawAnyToolCalls,
-          usage: output.usage
-        });
-        stream.end();
-        return;
-      }
-    } catch (error) {
-      output.stopReason = options?.signal?.aborted ? "aborted" : "error";
-      output.errorMessage = error instanceof Error ? error.message : String(error);
-      log.debug("response.caught", { stopReason: output.stopReason, error: output.errorMessage });
-      if (hiddenShimTimer) {
-        clearTimeout(hiddenShimTimer);
-        hiddenShimTimer = null;
+      } else {
+        result.push(history[i], history[i + 1]);
       }
-      stream.push({ type: "error", reason: output.stopReason, error: output });
-      stream.end();
+      i = j;
+    } else {
+      result.push(entry);
+      i++;
     }
-  })().catch(() => {
-    try {
-      stream.end();
-    } catch {}
-  });
-  return stream;
+  }
+  return result;
 }
 
-// src/models.ts
-var KIRO_MODEL_IDS = new Set([
-  "claude-fable-5",
-  "claude-opus-4.8",
-  "claude-opus-4.7",
-  "claude-opus-4.6",
-  "claude-opus-4.6-1m",
-  "claude-sonnet-4.6",
-  "claude-sonnet-4.6-1m",
-  "claude-opus-4.5",
-  "claude-sonnet-4.5",
-  "claude-sonnet-4.5-1m",
-  "claude-sonnet-4",
-  "claude-haiku-4.5",
-  "deepseek-3.2",
-  "kimi-k2.5",
-  "minimax-m2.1",
-  "minimax-m2.5",
-  "glm-4.7",
-  "glm-4.7-flash",
-  "qwen3-coder-next",
-  "agi-nova-beta-1m",
-  "qwen3-coder-480b",
-  "auto"
-]);
-function dotToDash(modelId) {
-  return modelId.replace(/(\d)\.(\d)/g, "$1-$2");
-}
-function resolveKiroModel(modelId) {
-  const kiroId = modelId.replace(/(\d)-(\d)/g, "$1.$2");
-  if (!KIRO_MODEL_IDS.has(kiroId)) {
-    throw new Error(`Unknown Kiro model ID: ${modelId}`);
+// src/kiro-defaults.ts
+var SYSTEM_SEED_INSTRUCTION = `Follow this instruction: # Kiro CLI Default Agent
+
+` + "You are the default Kiro CLI agent, bringing the power of AI-assisted development " + "directly to the user's terminal. You help with coding tasks, system operations, " + `AWS management, and development workflows.
+
+` + `The current model is {{modelId}}.
+`;
+var SYSTEM_SEED_ACK = "I will fully incorporate this information when generating my responses, " + "and explicitly acknowledge relevant parts of the summary when answering questions.";
+function resolveOS() {
+  switch (process.platform) {
+    case "darwin":
+      return "macos";
+    case "win32":
+      return "windows";
+    default:
+      return process.platform;
   }
-  return kiroId;
-}
-var API_REGION_MAP = {
-  "us-west-1": "us-east-1",
-  "us-west-2": "us-east-1",
-  "us-east-2": "us-east-1",
-  "eu-west-1": "eu-central-1",
-  "eu-west-2": "eu-central-1",
-  "eu-west-3": "eu-central-1",
-  "eu-north-1": "eu-central-1",
-  "eu-south-1": "eu-central-1",
-  "eu-south-2": "eu-central-1",
-  "eu-central-2": "eu-central-1",
-  "ap-northeast-1": "us-east-1",
-  "ap-northeast-2": "us-east-1",
-  "ap-northeast-3": "us-east-1",
-  "ap-southeast-1": "us-east-1",
-  "ap-southeast-2": "us-east-1",
-  "ap-south-1": "us-east-1",
-  "ap-east-1": "us-east-1",
-  "ap-south-2": "us-east-1",
-  "ap-southeast-3": "us-east-1",
-  "ap-southeast-4": "us-east-1"
-};
-function resolveApiRegion(ssoRegion) {
-  if (!ssoRegion)
-    return "us-east-1";
-  return API_REGION_MAP[ssoRegion] ?? ssoRegion;
 }
-var MODELS_BY_REGION = {
-  "us-east-1": new Set([
-    "claude-fable-5",
-    "claude-opus-4-8",
-    "claude-opus-4-7",
-    "claude-opus-4-6",
-    "claude-opus-4-6-1m",
-    "claude-sonnet-4-6",
-    "claude-sonnet-4-6-1m",
-    "claude-opus-4-5",
-    "claude-sonnet-4-5",
-    "claude-sonnet-4-5-1m",
-    "claude-sonnet-4",
-    "claude-haiku-4-5",
-    "deepseek-3-2",
-    "kimi-k2-5",
-    "minimax-m2-1",
-    "minimax-m2-5",
-    "glm-4-7",
-    "glm-4-7-flash",
-    "qwen3-coder-next",
-    "qwen3-coder-480b",
-    "agi-nova-beta-1m",
-    "auto"
-  ]),
-  "eu-central-1": new Set([
-    "claude-fable-5",
-    "claude-opus-4-8",
-    "claude-opus-4-7",
-    "claude-opus-4-6",
-    "claude-sonnet-4-6",
-    "claude-opus-4-5",
-    "claude-sonnet-4-5",
-    "claude-sonnet-4",
-    "claude-haiku-4-5",
-    "minimax-m2-1",
-    "minimax-m2-5",
-    "qwen3-coder-next",
-    "auto"
-  ])
-};
-function filterModelsByRegion(models, apiRegion) {
-  const allowed = MODELS_BY_REGION[apiRegion];
-  if (!allowed) {
-    console.warn(`[pi-kiro] Unknown API region "${apiRegion}" — no models available. Update MODELS_BY_REGION in models.ts.`);
-    return [];
-  }
-  return models.filter((m) => allowed.has(m.id));
+var COMPACTION_THRESHOLD_PCT = 95;
+
+// src/stream.ts
+var FIRST_TOKEN_TIMEOUT_DEFAULT_MS = 90000;
+var IDLE_TIMEOUT_MS = 60000;
+var MAX_RETRIES = 3;
+var MAX_RETRY_DELAY_MS = 1e4;
+var CAPACITY_MAX_RETRIES = 3;
+var CAPACITY_BASE_DELAY_MS = 5000;
+var CAPACITY_MAX_DELAY_MS = 30000;
+var TRANSIENT_MAX_RETRIES = 3;
+var TRANSIENT_BASE_DELAY_MS = 2000;
+var TRANSIENT_MAX_DELAY_MS = 15000;
+var CONTEXT_TRUNCATION_MAX_RETRIES = 3;
+var CONTEXT_TRUNCATION_DROP_RATIO = 0.3;
+var TOO_BIG_PATTERNS = ["CONTENT_LENGTH_EXCEEDS_THRESHOLD", "Input is too long", "Improperly formed"];
+var NON_RETRYABLE_BODY_PATTERNS = ["MONTHLY_REQUEST_COUNT"];
+var CAPACITY_PATTERN = "INSUFFICIENT_MODEL_CAPACITY";
+function exponentialBackoff(attempt, baseMs, maxMs) {
+  return Math.min(baseMs * 2 ** attempt, maxMs);
 }
-var RUNTIME_ENDPOINTS = {
-  "us-east-1": "https://runtime.us-east-1.kiro.dev",
-  "eu-central-1": "https://runtime.eu-central-1.kiro.dev"
-};
-function resolveRuntimeUrl(apiRegion) {
-  return RUNTIME_ENDPOINTS[apiRegion] ?? `https://runtime.${apiRegion}.kiro.dev`;
+function isTooBigError(status, body) {
+  return status === 413 || status === 400 && TOO_BIG_PATTERNS.some((p) => body.includes(p));
 }
-var BASE_URL = resolveRuntimeUrl("us-east-1");
-var ZERO_COST = Object.freeze({ input: 0, output: 0, cacheRead: 0, cacheWrite: 0 });
-var KIRO_DEFAULTS = {
-  api: "kiro-api",
-  provider: "kiro",
-  baseUrl: BASE_URL,
-  cost: ZERO_COST
-};
-var MULTIMODAL = ["text", "image"];
-var TEXT_ONLY = ["text"];
-var kiroModels = [
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-fable-5",
-    name: "Claude Fable 5",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 1e6,
-    maxTokens: 128000,
-    firstTokenTimeout: 180000,
-    supportedEfforts: ["low", "medium", "high", "xhigh", "max"],
-    supportsThinkingConfig: true
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-opus-4-8",
-    name: "Claude Opus 4.8",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 1e6,
-    maxTokens: 128000,
-    firstTokenTimeout: 180000,
-    supportedEfforts: ["low", "medium", "high", "xhigh", "max"],
-    supportsThinkingConfig: true
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-opus-4-7",
-    name: "Claude Opus 4.7",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 1e6,
-    maxTokens: 128000,
-    firstTokenTimeout: 180000,
-    supportedEfforts: ["low", "medium", "high", "xhigh", "max"],
-    supportsThinkingConfig: true
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-opus-4-6",
-    name: "Claude Opus 4.6",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 1e6,
-    maxTokens: 64000,
-    supportedEfforts: ["low", "medium", "high", "max"],
-    supportsThinkingConfig: true
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-opus-4-6-1m",
-    name: "Claude Opus 4.6 (1M)",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 1e6,
-    maxTokens: 64000,
-    supportedEfforts: ["low", "medium", "high", "max"],
-    supportsThinkingConfig: true
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-sonnet-4-6",
-    name: "Claude Sonnet 4.6",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 1e6,
-    maxTokens: 64000,
-    supportedEfforts: ["low", "medium", "high", "max"],
-    supportsThinkingConfig: true
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-sonnet-4-6-1m",
-    name: "Claude Sonnet 4.6 (1M)",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 1e6,
-    maxTokens: 64000,
-    supportedEfforts: ["low", "medium", "high", "max"],
-    supportsThinkingConfig: true
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-opus-4-5",
-    name: "Claude Opus 4.5",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 200000,
-    maxTokens: 64000
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-sonnet-4-5",
-    name: "Claude Sonnet 4.5",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 200000,
-    maxTokens: 65536
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-sonnet-4-5-1m",
-    name: "Claude Sonnet 4.5 (1M)",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 1e6,
-    maxTokens: 65536
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-sonnet-4",
-    name: "Claude Sonnet 4",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 200000,
-    maxTokens: 65536
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "claude-haiku-4-5",
-    name: "Claude Haiku 4.5",
-    reasoning: false,
-    input: MULTIMODAL,
-    contextWindow: 200000,
-    maxTokens: 65536
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "deepseek-3-2",
-    name: "DeepSeek 3.2",
-    reasoning: true,
-    input: TEXT_ONLY,
-    contextWindow: 128000,
-    maxTokens: 8192
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "kimi-k2-5",
-    name: "Kimi K2.5",
-    reasoning: true,
-    input: TEXT_ONLY,
-    contextWindow: 200000,
-    maxTokens: 8192
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "minimax-m2-5",
-    name: "MiniMax M2.5",
-    reasoning: false,
-    input: TEXT_ONLY,
-    contextWindow: 196000,
-    maxTokens: 64000
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "minimax-m2-1",
-    name: "MiniMax M2.1",
-    reasoning: false,
-    input: MULTIMODAL,
-    contextWindow: 196000,
-    maxTokens: 64000
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "glm-4-7",
-    name: "GLM 4.7",
-    reasoning: true,
-    input: TEXT_ONLY,
-    contextWindow: 128000,
-    maxTokens: 8192
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "glm-4-7-flash",
-    name: "GLM 4.7 Flash",
-    reasoning: false,
-    input: TEXT_ONLY,
-    contextWindow: 128000,
-    maxTokens: 8192
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "qwen3-coder-next",
-    name: "Qwen3 Coder Next",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 256000,
-    maxTokens: 64000
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "qwen3-coder-480b",
-    name: "Qwen3 Coder 480B",
-    reasoning: true,
-    input: TEXT_ONLY,
-    contextWindow: 128000,
-    maxTokens: 8192
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "agi-nova-beta-1m",
-    name: "AGI Nova Beta (1M)",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 1e6,
-    maxTokens: 65536
-  },
-  {
-    ...KIRO_DEFAULTS,
-    id: "auto",
-    name: "Auto",
-    reasoning: true,
-    input: MULTIMODAL,
-    contextWindow: 200000,
-    maxTokens: 65536
-  }
-];
-async function fetchAvailableModels(accessToken, apiRegion, fallbackProfileArn) {
-  const runtimeUrl = `https://runtime.${apiRegion}.kiro.dev/`;
-  let profileArn = await resolveProfileArn(accessToken, runtimeUrl);
-  if (!profileArn && fallbackProfileArn) {
-    profileArn = fallbackProfileArn;
-  }
-  if (!profileArn) {
-    throw new Error("Missing profileArn: cannot fetch available models.");
-  }
-  const url = `https://management.${apiRegion}.kiro.dev/ListAvailableModels?origin=KIRO_CLI&profileArn=${encodeURIComponent(profileArn)}`;
-  const resp = await fetch(url, {
-    method: "GET",
-    headers: {
-      Authorization: `Bearer ${accessToken}`,
-      Accept: "application/json",
-      "User-Agent": "pi-kiro"
-    }
-  });
-  if (!resp.ok) {
-    throw new Error(`ListAvailableModels failed: HTTP ${resp.status}`);
-  }
-  const data = await resp.json();
-  return (data.models ?? []).filter((m) => m.modelId !== "auto");
+function isNonRetryableBodyError(body) {
+  return NON_RETRYABLE_BODY_PATTERNS.some((p) => body.includes(p));
 }
-var REASONING_FAMILIES = new Set([
-  "claude-fable",
-  "claude-sonnet",
-  "claude-opus",
-  "deepseek",
-  "kimi",
-  "glm",
-  "qwen",
-  "agi-nova",
-  "minimax"
-]);
-function isReasoningModel(dotId) {
-  for (const family of REASONING_FAMILIES) {
-    if (dotId.startsWith(family))
-      return true;
-  }
-  return false;
-}
-function firstTokenTimeout(dotId) {
-  if (dotId.startsWith("claude-fable") || dotId.startsWith("claude-opus"))
-    return 180000;
-  return 90000;
+function isCapacityError(body) {
+  return body.includes(CAPACITY_PATTERN);
 }
-function buildModelsFromApi(apiModels) {
-  return apiModels.map((m) => {
-    KIRO_MODEL_IDS.add(m.modelId);
-    const dashId = dotToDash(m.modelId);
-    const supportedTypes = m.supportedInputTypes ?? ["TEXT"];
-    const input = supportedTypes.includes("IMAGE") ? ["text", "image"] : ["text"];
-    const effortEnum = m.additionalModelRequestFieldsSchema?.properties?.output_config?.properties?.effort?.enum;
-    const supportedEfforts = Array.isArray(effortEnum) && effortEnum.length > 0 ? effortEnum : undefined;
-    const supportsThinkingConfig = !!m.additionalModelRequestFieldsSchema?.properties?.thinking;
-    return {
-      id: dashId,
-      name: m.modelName,
-      reasoning: isReasoningModel(m.modelId),
-      input,
-      contextWindow: m.tokenLimits?.maxInputTokens ?? 200000,
-      maxTokens: m.tokenLimits?.maxOutputTokens ?? 8192,
-      firstTokenTimeout: firstTokenTimeout(m.modelId),
-      ...supportedEfforts ? { supportedEfforts } : {},
-      ...supportsThinkingConfig ? { supportsThinkingConfig } : {}
-    };
-  });
+function isTransientError(status) {
+  return status === 429 || status >= 500;
 }
-var cachedDynamicModels = null;
-function getCachedDynamicModels() {
-  return cachedDynamicModels;
+function firstTokenTimeoutForModel(modelId) {
+  const m = kiroModels.find((x) => x.id === modelId);
+  return m?.firstTokenTimeout ?? FIRST_TOKEN_TIMEOUT_DEFAULT_MS;
 }
-function setCachedDynamicModels(models) {
-  cachedDynamicModels = models;
+var HIDDEN_REASONING_PLACEHOLDER = "Reasoning hidden by provider";
+var HIDDEN_REASONING_COUNTDOWN_MS = 2000;
+function emitHiddenReasoningLate(output, stream) {
+  const contentIndex = output.content.length;
+  const block = {
+    type: "thinking",
+    thinking: HIDDEN_REASONING_PLACEHOLDER,
+    redacted: true
+  };
+  output.content.push(block);
+  stream.push({ type: "thinking_start", contentIndex, partial: output });
+  stream.push({
+    type: "thinking_delta",
+    contentIndex,
+    delta: HIDDEN_REASONING_PLACEHOLDER,
+    partial: output
+  });
+  stream.push({
+    type: "thinking_end",
+    contentIndex,
+    content: "",
+    partial: output
+  });
 }
-
-// src/oauth.ts
-init_debug();
-var BUILDER_ID_START_URL = "https://view.awsapps.com/start";
-var BUILDER_ID_REGION = "us-east-1";
-var SSO_SCOPES = [
-  "codewhisperer:completions",
-  "codewhisperer:analysis",
-  "codewhisperer:conversations",
-  "codewhisperer:transformations",
-  "codewhisperer:taskassist"
-];
-var IDC_PROBE_REGIONS = [
-  "us-east-1",
-  "eu-west-1",
-  "eu-central-1",
-  "us-east-2",
-  "eu-west-2",
-  "eu-west-3",
-  "eu-north-1",
-  "ap-southeast-1",
-  "ap-northeast-1",
-  "us-west-2"
-];
-var EXPIRES_BUFFER_MS = 5 * 60 * 1000;
 function abortableDelay2(ms, signal) {
   if (signal?.aborted)
-    return Promise.reject(signal.reason ?? new Error("Login cancelled"));
+    return Promise.reject(signal.reason);
   return new Promise((resolve2, reject) => {
     const timer = setTimeout(resolve2, ms);
     signal?.addEventListener("abort", () => {
       clearTimeout(timer);
-      reject(signal.reason ?? new Error("Login cancelled"));
+      reject(signal.reason);
     }, { once: true });
   });
 }
-async function tryRegisterAndAuthorize(startUrl, region) {
-  const oidcEndpoint = `https://oidc.${region}.amazonaws.com`;
-  const regResp = await fetch(`${oidcEndpoint}/client/register`, {
-    method: "POST",
-    headers: { "Content-Type": "application/json", "User-Agent": "pi-kiro" },
-    body: JSON.stringify({
-      clientName: "pi-kiro",
-      clientType: "public",
-      scopes: SSO_SCOPES,
-      grantTypes: ["urn:ietf:params:oauth:grant-type:device_code", "refresh_token"]
-    })
-  });
-  if (!regResp.ok)
-    return null;
-  const { clientId, clientSecret } = await regResp.json();
-  const devResp = await fetch(`${oidcEndpoint}/device_authorization`, {
-    method: "POST",
-    headers: { "Content-Type": "application/json", "User-Agent": "pi-kiro" },
-    body: JSON.stringify({ clientId, clientSecret, startUrl })
-  });
-  if (!devResp.ok)
-    return null;
-  return {
-    clientId,
-    clientSecret,
-    oidcEndpoint,
-    devAuth: await devResp.json()
-  };
-}
-async function pollForToken(oidcEndpoint, clientId, clientSecret, devAuth, signal) {
-  const deadline = Date.now() + (devAuth.expiresIn || 600) * 1000;
-  const baseInterval = (devAuth.interval || 5) * 1000;
-  let interval = baseInterval;
-  while (Date.now() < deadline) {
-    if (signal?.aborted)
-      throw new Error("Login cancelled");
-    await abortableDelay2(interval, signal);
-    let resp;
-    try {
-      resp = await fetch(`${oidcEndpoint}/token`, {
-        method: "POST",
-        headers: { "Content-Type": "application/json", "User-Agent": "pi-kiro" },
-        body: JSON.stringify({
-          clientId,
-          clientSecret,
-          deviceCode: devAuth.deviceCode,
-          grantType: "urn:ietf:params:oauth:grant-type:device_code"
-        })
-      });
-    } catch {
-      continue;
-    }
-    if (resp.status >= 500)
-      continue;
-    let data;
-    try {
-      data = await resp.json();
-    } catch {
-      if (!resp.ok) {
-        throw new Error(`Authorization failed: HTTP ${resp.status}`);
-      }
-      continue;
-    }
-    if (!data.error && data.accessToken && data.refreshToken)
-      return data;
-    if (data.error === "authorization_pending")
-      continue;
-    if (data.error === "slow_down") {
-      interval += baseInterval;
-      continue;
-    }
-    if (data.error)
-      throw new Error(`Authorization failed: ${data.error}`);
-  }
-  throw new Error("Authorization timed out");
+var profileArnStore = undefined;
+function seedProfileArn(arn) {
+  profileArnStore = arn;
 }
-async function loginKiro(callbacks) {
-  const method = await callbacks.onSelect({
-    message: "Select login method:",
-    options: [
-      { id: "builder-id", label: "AWS Builder ID (personal account)" },
-      { id: "idc", label: "IAM Identity Center (enterprise SSO)" },
-      { id: "sync", label: "Import from Kiro CLI/IDE (auto-sync local DB)" },
-      { id: "desktop", label: "Desktop refresh token (manual)" }
-    ]
-  });
-  if (!method)
-    throw new Error("Login cancelled");
-  if (method === "sync") {
-    return loginCliSync(callbacks);
-  }
-  if (method === "desktop") {
-    return loginDesktopManual(callbacks);
-  }
-  if (method === "builder-id") {
-    return runDeviceCodeFlow(callbacks, BUILDER_ID_START_URL, [BUILDER_ID_REGION], "builder-id");
-  }
-  const startUrl = (await callbacks.onPrompt({
-    message: "Paste your IAM Identity Center start URL:",
-    placeholder: "https://mycompany.awsapps.com/start",
-    allowEmpty: false
-  }))?.trim();
-  if (!startUrl || !startUrl.startsWith("http")) {
-    throw new Error(`Invalid start URL "${startUrl ?? ""}" — expected https://…`);
-  }
-  const regionRaw = await callbacks.onPrompt({
-    message: `Identity Center region, or blank to auto-detect (${IDC_PROBE_REGIONS.join(", ")})`,
-    placeholder: "us-east-1",
-    allowEmpty: true
-  });
-  const region = (regionRaw ?? "").trim();
-  const regions = region ? [region] : IDC_PROBE_REGIONS;
-  callbacks.onProgress?.(region ? `Connecting to ${region}…` : "Detecting your Identity Center region…");
-  return runDeviceCodeFlow(callbacks, startUrl, regions, "idc");
+function getProfileArn() {
+  return profileArnStore;
 }
-async function loginCliSync(callbacks) {
-  callbacks.onProgress?.("Scanning for Kiro IDE credentials (~/.kiro/db)…");
-  const { importFromKiroCli: importFromKiroCli2 } = await Promise.resolve().then(() => (init_kiro_cli_sync(), exports_kiro_cli_sync));
-  const imported = await importFromKiroCli2();
-  if (!imported || !imported.accessToken && !imported.refreshToken) {
-    throw new Error(`No Kiro IDE credentials found.
-Make sure Kiro IDE is installed and you're logged in, then try again.
-Alternatively, use 'desktop' to paste a refresh token manually.`);
-  }
-  log.info("Successfully imported credentials from Kiro IDE");
-  callbacks.onProgress?.(`Imported from Kiro IDE (${imported.authMethod}, ${imported.region}${imported.email ? `, ${imported.email}` : ""})`);
+function emitToolCall(state, output, stream) {
+  if (!state.input.trim())
+    state.input = "{}";
+  let args;
   try {
-    const apiRegion = resolveApiRegion(imported.region);
-    const apiModels = await fetchAvailableModels(imported.accessToken, apiRegion, imported.profileArn);
-    setCachedDynamicModels(buildModelsFromApi(apiModels));
-    log.info(`Fetched and cached ${apiModels.length} models after CLI sync`);
-  } catch (err) {
-    log.warn(`Failed to fetch models after CLI sync: ${err}`);
+    args = JSON.parse(state.input);
+    if (args && typeof args === "object" && "__tool_use_purpose" in args) {
+      delete args.__tool_use_purpose;
+    }
+  } catch (e) {
+    log.warn(`failed to parse tool input for "${state.name}" (${state.toolUseId}): ${e instanceof Error ? e.message : String(e)}`);
+    return false;
   }
-  return kiroCredsFromCliImport(imported);
+  const contentIndex = output.content.length;
+  const toolCall = { type: "toolCall", id: state.toolUseId, name: state.name, arguments: args };
+  output.content.push(toolCall);
+  stream.push({ type: "toolcall_start", contentIndex, partial: output });
+  stream.push({ type: "toolcall_delta", contentIndex, delta: state.input, partial: output });
+  stream.push({ type: "toolcall_end", contentIndex, toolCall, partial: output });
+  return true;
 }
-async function loginDesktopManual(callbacks) {
-  const refreshRaw = await callbacks.onPrompt({
-    message: `Paste your Kiro desktop refresh token
-` + "(find it in ~/.kiro/db/kiro.db → auth_kv table):",
-    placeholder: "refresh-token",
-    allowEmpty: true
-  });
-  const refreshToken = (refreshRaw ?? "").trim();
-  if (!refreshToken) {
-    throw new Error("Login cancelled — no refresh token provided");
-  }
-  const regionRaw = await callbacks.onPrompt({
-    message: "Kiro region:",
-    placeholder: "us-east-1",
-    allowEmpty: true
-  });
-  const region = (regionRaw ?? "").trim() || "us-east-1";
-  const refreshCreds = {
-    refresh: `${refreshToken}|||desktop`,
-    access: "",
-    expires: 0,
-    clientId: "",
-    clientSecret: "",
-    region,
-    authMethod: "desktop"
-  };
-  callbacks.onProgress?.("Exchanging refresh token…");
-  return refreshKiroToken(refreshCreds);
-}
-async function runDeviceCodeFlow(callbacks, startUrl, regions, authMethod) {
-  let result = null;
-  let detectedRegion = "";
-  for (const region of regions) {
-    result = await tryRegisterAndAuthorize(startUrl, region);
-    if (result) {
-      detectedRegion = region;
-      if (regions.length > 1)
-        callbacks.onProgress?.(`Region: ${region}`);
-      break;
-    }
-  }
-  if (!result || !detectedRegion) {
-    throw new Error(`Could not authorize ${startUrl} in ${regions.join(", ")}. Check your start URL${regions.length === 1 ? " and region" : ""} and try again.`);
-  }
-  callbacks.onAuth({
-    url: result.devAuth.verificationUriComplete,
-    instructions: `Code: ${result.devAuth.userCode}
-Complete authorization within 10 minutes.`
-  });
-  callbacks.onProgress?.("Waiting for browser authorization (up to 10 minutes)…");
-  const tok = await pollForToken(result.oidcEndpoint, result.clientId, result.clientSecret, result.devAuth, callbacks.signal);
-  if (!tok.accessToken || !tok.refreshToken) {
-    throw new Error("Authorization completed but no tokens returned");
-  }
-  try {
-    const apiRegion = resolveApiRegion(detectedRegion);
-    const apiModels = await fetchAvailableModels(tok.accessToken, apiRegion);
-    setCachedDynamicModels(buildModelsFromApi(apiModels));
-    log.info(`Fetched and cached ${apiModels.length} models after login`);
-  } catch (err) {
-    log.warn(`Failed to fetch models after login, falling back: ${err}`);
-  }
-  return {
-    refresh: `${tok.refreshToken}|${result.clientId}|${result.clientSecret}|${authMethod}`,
-    access: tok.accessToken,
-    expires: Date.now() + (tok.expiresIn ?? 3600) * 1000 - EXPIRES_BUFFER_MS,
-    clientId: result.clientId,
-    clientSecret: result.clientSecret,
-    region: detectedRegion,
-    authMethod
-  };
-}
-async function syncBackToKiroCli(result) {
-  if (result.kiroSyncSource !== "kiro-cli-db" || !result.kiroSyncTokenKey) {
-    log.debug("Credential sync-back skipped: credential did not originate from kiro-cli DB");
-    return;
-  }
-  try {
-    const { saveKiroCliCredentials: saveKiroCliCredentials2 } = await Promise.resolve().then(() => (init_kiro_cli_sync(), exports_kiro_cli_sync));
-    const synced = await saveKiroCliCredentials2({
-      accessToken: result.access,
-      refreshToken: result.refresh.split("|")[0] ?? "",
-      region: result.region,
-      authMethod: result.authMethod === "builder-id" ? "idc" : result.authMethod,
-      source: result.kiroSyncSource,
-      tokenKey: result.kiroSyncTokenKey
-    });
-    if (synced)
-      log.info("Synced refreshed credentials back to Kiro CLI DB");
-  } catch (err) {
-    log.debug(`Credential sync-back skipped: ${err}`);
-  }
-}
-function kiroCredsFromCliImport(imported) {
-  const hasOidcCreds = !!imported.clientId && !!imported.clientSecret;
-  const authMethod = hasOidcCreds && imported.authMethod === "idc" ? "idc" : "desktop";
-  const refreshPacked = hasOidcCreds ? `${imported.refreshToken}|${imported.clientId}|${imported.clientSecret ?? ""}|${authMethod}` : `${imported.refreshToken}|||desktop`;
-  return {
-    refresh: refreshPacked,
-    access: imported.accessToken,
-    expires: Date.now() + 3600000 - EXPIRES_BUFFER_MS,
-    clientId: imported.clientId ?? "",
-    clientSecret: imported.clientSecret ?? "",
-    region: imported.region,
-    authMethod,
-    kiroSyncSource: imported.source,
-    kiroSyncTokenKey: imported.tokenKey
-  };
-}
-async function refreshTokenInner(credentials) {
-  const parts = credentials.refresh.split("|");
-  const refreshToken = parts[0] ?? "";
-  const clientId = parts[1] ?? credentials.clientId ?? "";
-  const clientSecret = parts[2] ?? credentials.clientSecret ?? "";
-  const region = credentials.region;
-  const authMethod = credentials.authMethod;
-  if (!refreshToken || !region) {
-    throw new Error("Refresh token is missing region — re-login required");
-  }
-  if (authMethod !== "desktop" && (!clientId || !clientSecret)) {
-    throw new Error("Refresh token is missing clientId/clientSecret — re-login required");
-  }
-  if (authMethod === "desktop") {
-    const desktopEndpoint = `https://prod.${region}.auth.desktop.kiro.dev/refreshToken`;
-    const resp2 = await fetch(desktopEndpoint, {
-      method: "POST",
-      headers: { "Content-Type": "application/json", "User-Agent": "pi-kiro" },
-      body: JSON.stringify({ refreshToken })
-    });
-    if (!resp2.ok) {
-      const body = await resp2.text().catch(() => "");
-      throw new Error(`Desktop token refresh failed: ${resp2.status} ${body}`);
+function streamKiro(model, context, options) {
+  const stream = createAssistantMessageEventStream();
+  (async () => {
+    const output = {
+      role: "assistant",
+      content: [],
+      api: model.api,
+      provider: model.provider,
+      model: model.id,
+      usage: {
+        input: 0,
+        output: 0,
+        cacheRead: 0,
+        cacheWrite: 0,
+        totalTokens: 0,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
+      },
+      stopReason: "stop",
+      timestamp: Date.now()
+    };
+    let hiddenShimTimer = null;
+    try {
+      const accessToken = options?.apiKey;
+      if (!accessToken) {
+        throw new Error("Kiro credentials not set. Run /login kiro.");
+      }
+      const endpoint = model.baseUrl || "https://runtime.us-east-1.kiro.dev";
+      const profileArn = getProfileArn();
+      if (!profileArn) {
+        throw new Error("profileArn not resolved. Re-run /login kiro to refresh credentials.");
+      }
+      const kiroModelId = resolveKiroModel(model.id);
+      const thinkingEnabled = !!options?.reasoning || model.reasoning;
+      const reasoningHidden = !!model.reasoningHidden;
+      log.debug("request.init", {
+        endpoint,
+        model: model.id,
+        kiroModelId,
+        contextWindow: model.contextWindow,
+        thinkingEnabled,
+        reasoningHidden,
+        reasoning: options?.reasoning,
+        messageCount: context.messages.length,
+        toolCount: context.tools?.length ?? 0,
+        hasSystemPrompt: !!context.systemPrompt,
+        profileArn,
+        sessionId: options?.sessionId
+      });
+      let systemPrompt = context.systemPrompt ?? "";
+      if (thinkingEnabled && !reasoningHidden) {
+        const budget = options?.reasoning === "xhigh" ? 50000 : options?.reasoning === "high" ? 30000 : options?.reasoning === "medium" ? 20000 : 1e4;
+        systemPrompt = `<thinking_mode>enabled</thinking_mode><max_thinking_length>${budget}</max_thinking_length>${systemPrompt ? `
+${systemPrompt}` : ""}`;
+      }
+      const envState = {
+        operatingSystem: resolveOS(),
+        currentWorkingDirectory: process.cwd()
+      };
+      const conversationId = options?.sessionId ?? crypto.randomUUID();
+      let retryCount = 0;
+      while (retryCount <= MAX_RETRIES) {
+        if (options?.signal?.aborted)
+          throw options.signal.reason;
+        const normalized = normalizeMessages(context.messages);
+        const {
+          history,
+          systemPrepended,
+          currentMsgStartIdx
+        } = buildHistory(normalized, kiroModelId, systemPrompt);
+        const seedInstruction = SYSTEM_SEED_INSTRUCTION.replace("{{modelId}}", kiroModelId);
+        const seedPair = [
+          { userInputMessage: { content: seedInstruction, origin: "KIRO_CLI" } },
+          { assistantResponseMessage: { content: SYSTEM_SEED_ACK } }
+        ];
+        history.unshift(...seedPair);
+        const currentMessages = normalized.slice(currentMsgStartIdx);
+        const firstMsg = currentMessages[0];
+        let currentContent = "";
+        const currentToolResults = [];
+        let currentImages;
+        if (firstMsg?.role === "assistant") {
+          const am = firstMsg;
+          let armContent = "";
+          const armToolUses = [];
+          if (Array.isArray(am.content)) {
+            for (const b of am.content) {
+              if (b.type === "text") {
+                armContent += b.text;
+              } else if (b.type === "thinking") {
+                armContent = `<thinking>${b.thinking}</thinking>
+
+${armContent}`;
+              } else if (b.type === "toolCall") {
+                const tc = b;
+                armToolUses.push({
+                  name: tc.name,
+                  toolUseId: toKiroToolUseId(tc.id),
+                  input: parseToolArgs(tc.arguments)
+                });
+              }
+            }
+          }
+          if (armContent || armToolUses.length > 0) {
+            const last = history[history.length - 1];
+            if (last && !last.userInputMessage && last.assistantResponseMessage) {
+              last.assistantResponseMessage.content += `
+
+${armContent}`;
+              if (armToolUses.length > 0) {
+                last.assistantResponseMessage.toolUses = [
+                  ...last.assistantResponseMessage.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") {
+              const trm = m;
+              currentToolResults.push({
+                content: [{ text: truncate(getContentText(m), TOOL_RESULT_LIMIT) }],
+                status: trm.isError ? "error" : "success",
+                toolUseId: toKiroToolUseId(trm.toolCallId)
+              });
+              if (Array.isArray(trm.content)) {
+                for (const c of trm.content) {
+                  if (c.type === "image")
+                    toolResultImages.push(c);
+                }
+              }
+            }
+          }
+          if (toolResultImages.length > 0) {
+            const { images: converted, omitted } = convertImagesToKiro(toolResultImages);
+            if (omitted > 0)
+              log.warn(`${omitted} tool-result image(s) omitted (size/count limit)`);
+            currentImages = currentImages ? [...currentImages, ...converted] : converted;
+          }
+          currentContent = currentToolResults.length > 0 ? "Tool results provided." : "Please proceed with the task.";
+        } else if (firstMsg?.role === "toolResult") {
+          const toolResultImages = [];
+          for (const m of currentMessages) {
+            if (m?.role === "toolResult") {
+              const trm = m;
+              currentToolResults.push({
+                content: [{ text: truncate(getContentText(m), TOOL_RESULT_LIMIT) }],
+                status: trm.isError ? "error" : "success",
+                toolUseId: toKiroToolUseId(trm.toolCallId)
+              });
+              if (Array.isArray(trm.content)) {
+                for (const c of trm.content) {
+                  if (c.type === "image")
+                    toolResultImages.push(c);
+                }
+              }
+            }
+          }
+          if (toolResultImages.length > 0) {
+            const { images: converted, omitted } = convertImagesToKiro(toolResultImages);
+            if (omitted > 0)
+              log.warn(`${omitted} tool-result image(s) omitted (size/count limit)`);
+            currentImages = currentImages ? [...currentImages, ...converted] : converted;
+          }
+          currentContent = "Tool results provided.";
+        } else if (firstMsg?.role === "user") {
+          currentContent = typeof firstMsg.content === "string" ? firstMsg.content : getContentText(firstMsg);
+          if (systemPrompt && !systemPrepended) {
+            currentContent = `${systemPrompt}
+
+${currentContent}`;
+          }
+        }
+        const now = new Date;
+        const tzOffset = -now.getTimezoneOffset();
+        const tzSign = tzOffset >= 0 ? "+" : "-";
+        const tzH = String(Math.floor(Math.abs(tzOffset) / 60)).padStart(2, "0");
+        const tzM = String(Math.abs(tzOffset) % 60).padStart(2, "0");
+        const isoLocal = now.getFullYear() + "-" + String(now.getMonth() + 1).padStart(2, "0") + "-" + String(now.getDate()).padStart(2, "0") + "T" + String(now.getHours()).padStart(2, "0") + ":" + String(now.getMinutes()).padStart(2, "0") + ":" + String(now.getSeconds()).padStart(2, "0") + "." + String(now.getMilliseconds()).padStart(3, "0") + tzSign + tzH + ":" + tzM;
+        const weekday = now.toLocaleDateString("en-US", { weekday: "long" });
+        currentContent = `--- CONTEXT ENTRY BEGIN ---
+` + `Current time: ${weekday}, ${isoLocal}
+` + `--- CONTEXT ENTRY END ---
+
+` + `--- USER MESSAGE BEGIN ---
+` + `${currentContent}
+` + `--- USER MESSAGE END ---`;
+        let uimc = {
+          envState
+        };
+        if (currentToolResults.length > 0)
+          uimc.toolResults = currentToolResults;
+        if (context.tools?.length) {
+          const deepCleanSchema = (obj) => {
+            if (Array.isArray(obj)) {
+              return obj.map(deepCleanSchema);
+            } else if (obj !== null && typeof obj === "object") {
+              const cleaned = {};
+              for (const [k, v] of Object.entries(obj)) {
+                if (k === "$schema")
+                  continue;
+                if (k === "required" && Array.isArray(v) && v.length === 0)
+                  continue;
+                cleaned[k] = deepCleanSchema(v);
+              }
+              return cleaned;
+            }
+            return obj;
+          };
+          uimc.tools = context.tools.map((t) => {
+            const params = deepCleanSchema(t.parameters);
+            if (params.properties) {
+              params.properties.__tool_use_purpose = {
+                type: "string",
+                description: "A brief explanation why you are making this tool use."
+              };
+            }
+            return {
+              toolSpecification: {
+                name: t.name,
+                description: t.description || `Use ${t.name}`,
+                inputSchema: { json: params }
+              }
+            };
+          });
+        }
+        if (firstMsg?.role === "user") {
+          const imgs = extractImages(firstMsg);
+          if (imgs.length > 0) {
+            const { images: converted, omitted } = convertImagesToKiro(imgs);
+            if (omitted > 0)
+              log.warn(`${omitted} user image(s) omitted (size/count limit)`);
+            currentImages = converted;
+          }
+        }
+        const request = {
+          conversationState: {
+            chatTriggerType: "MANUAL",
+            agentTaskType: "vibe",
+            conversationId,
+            currentMessage: {
+              userInputMessage: {
+                content: currentContent,
+                modelId: kiroModelId,
+                origin: "KIRO_CLI",
+                ...currentImages ? { images: currentImages } : {},
+                userInputMessageContext: uimc
+              }
+            },
+            ...history.length > 0 ? { history } : {}
+          },
+          profileArn,
+          agentMode: "vibe"
+        };
+        const EFFORT_MAP = {
+          minimal: "low",
+          low: "medium",
+          medium: "high",
+          high: "xhigh",
+          xhigh: "max"
+        };
+        const staticModel = kiroModels.find((m) => m.id === model.id);
+        const dynamicModel = getCachedDynamicModels()?.find((m) => m.id === model.id);
+        const supportedEfforts = staticModel?.supportedEfforts ?? dynamicModel?.supportedEfforts;
+        const supportsThinkingConfig = staticModel?.supportsThinkingConfig ?? dynamicModel?.supportsThinkingConfig;
+        if (supportedEfforts && supportedEfforts.length > 0 && options?.reasoning && typeof options.reasoning === "string") {
+          const kiroEffort = EFFORT_MAP[options.reasoning];
+          if (kiroEffort && supportedEfforts.includes(kiroEffort)) {
+            request.additionalModelRequestFields = request.additionalModelRequestFields || {};
+            request.additionalModelRequestFields.output_config = { effort: kiroEffort };
+            log.debug("effort.set", { piReasoning: options.reasoning, kiroEffort, model: model.id });
+          }
+        }
+        if (supportsThinkingConfig && thinkingEnabled) {
+          request.additionalModelRequestFields = request.additionalModelRequestFields || {};
+          request.additionalModelRequestFields.thinking = {
+            type: "adaptive",
+            display: "summarized"
+          };
+          log.debug("thinking.set", { type: "adaptive", display: "summarized", model: model.id });
+        }
+        stream.push({ type: "start", partial: output });
+        if (reasoningHidden && thinkingEnabled && hiddenShimTimer === null) {
+          hiddenShimTimer = setTimeout(() => {
+            hiddenShimTimer = null;
+            emitHiddenReasoningLate(output, stream);
+          }, HIDDEN_REASONING_COUNTDOWN_MS);
+        }
+        let response;
+        let capacityRetryCount = 0;
+        let transientRetryCount = 0;
+        let contextTruncationAttempt = 0;
+        while (true) {
+          const osName = resolveOS();
+          const ua = `aws-sdk-rust/1.3.15 ua/2.1 api/codewhispererstreaming/0.1.16551 os/${osName} lang/rust/1.92.0 md/appVersion-2.7.1 app/AmazonQ-For-CLI`;
+          const xAmzUa = `aws-sdk-rust/1.3.15 ua/2.1 api/codewhispererstreaming/0.1.16551 os/${osName} lang/rust/1.92.0 m/F app/AmazonQ-For-CLI`;
+          const requestBody = JSON.stringify(request);
+          log.debug("request.send", {
+            attempt: retryCount,
+            capacityAttempt: capacityRetryCount,
+            historyLen: history.length,
+            currentContentLen: currentContent.length,
+            hasImages: !!currentImages,
+            toolResultCount: currentToolResults.length,
+            requestJsonChars: requestBody.length
+          });
+          response = await fetch(endpoint, {
+            method: "POST",
+            headers: {
+              "Content-Type": "application/x-amz-json-1.0",
+              Accept: "*/*",
+              "Accept-Encoding": "gzip",
+              Authorization: `Bearer ${accessToken}`,
+              "X-Amz-Target": "AmazonCodeWhispererStreamingService.GenerateAssistantResponse",
+              "x-amzn-codewhisperer-optout": "true",
+              "amz-sdk-invocation-id": crypto.randomUUID(),
+              "amz-sdk-request": "attempt=1; max=3",
+              "user-agent": ua,
+              "x-amz-user-agent": xAmzUa,
+              Pragma: "no-cache",
+              "Cache-Control": "no-cache"
+            },
+            body: requestBody,
+            signal: options?.signal
+          });
+          if (response.ok)
+            break;
+          let errText = "";
+          try {
+            errText = await response.text();
+          } catch {
+            errText = "";
+          }
+          log.debug("response.error", {
+            status: response.status,
+            body: errText
+          });
+          if (isCapacityError(errText) && capacityRetryCount < CAPACITY_MAX_RETRIES) {
+            capacityRetryCount++;
+            const delayMs = exponentialBackoff(capacityRetryCount - 1, CAPACITY_BASE_DELAY_MS, CAPACITY_MAX_DELAY_MS);
+            log.warn(`INSUFFICIENT_MODEL_CAPACITY — retrying in ${delayMs}ms (${capacityRetryCount}/${CAPACITY_MAX_RETRIES})`);
+            await abortableDelay2(delayMs, options?.signal);
+            continue;
+          }
+          if (isNonRetryableBodyError(errText) || isCapacityError(errText)) {
+            throw new Error(`Kiro API error: ${errText || response.statusText}`);
+          }
+          if (isTooBigError(response.status, errText)) {
+            if (contextTruncationAttempt < CONTEXT_TRUNCATION_MAX_RETRIES && history.length > 0) {
+              contextTruncationAttempt++;
+              const dropCount = Math.max(1, Math.floor(history.length * CONTEXT_TRUNCATION_DROP_RATIO));
+              const before = history.length;
+              history.splice(0, dropCount);
+              log.warn(`context too large — truncated history from ${before} to ${history.length} entries ` + `(attempt ${contextTruncationAttempt}/${CONTEXT_TRUNCATION_MAX_RETRIES})`);
+              request.conversationState.history = history.length > 0 ? history : undefined;
+              continue;
+            }
+            throw new Error(`Kiro API error: context_length_exceeded (${response.status} ${errText})`);
+          }
+          if (isTransientError(response.status) && transientRetryCount < TRANSIENT_MAX_RETRIES) {
+            transientRetryCount++;
+            const jitter = Math.floor(Math.random() * 1000);
+            const delayMs = exponentialBackoff(transientRetryCount - 1, TRANSIENT_BASE_DELAY_MS, TRANSIENT_MAX_DELAY_MS) + jitter;
+            log.warn(`transient error ${response.status} — retrying in ${delayMs}ms ` + `(${transientRetryCount}/${TRANSIENT_MAX_RETRIES})`);
+            await abortableDelay2(delayMs, options?.signal);
+            continue;
+          }
+          if (response.status === 401) {
+            const permanent = isPermanentError(errText);
+            if (permanent) {
+              throw new Error(`Kiro API error: credentials permanently invalid — run /login kiro to re-authenticate. ${errText}`);
+            }
+          }
+          if (response.status === 403) {
+            throw new Error(`Kiro API error: access token rejected (403) — run /login kiro to re-authenticate. ${errText}`);
+          }
+          throw new Error(`Kiro API error: ${response.status} ${response.statusText} ${errText}`);
+        }
+        if (capacityRetryCount > 0) {
+          log.info(`recovered from capacity pressure after ${capacityRetryCount} retries`);
+        }
+        if (transientRetryCount > 0) {
+          log.info(`recovered from transient error after ${transientRetryCount} retries`);
+        }
+        if (contextTruncationAttempt > 0) {
+          log.info(`recovered after ${contextTruncationAttempt} context truncation(s)`);
+        }
+        const reader = response.body?.getReader();
+        if (!reader)
+          throw new Error("No response body");
+        const decoder = new TextDecoder;
+        let buffer = "";
+        let totalContent = "";
+        let lastContentData = "";
+        let usageEvent = null;
+        let receivedContextUsage = false;
+        let chunkSeq = 0;
+        let eventSeq = 0;
+        const thinkingParser = thinkingEnabled ? new ThinkingTagParser(output, stream) : null;
+        let textBlockIndex = null;
+        let emittedToolCalls = 0;
+        let sawAnyToolCalls = false;
+        let currentToolCall = null;
+        const flushToolCall = () => {
+          if (!currentToolCall)
+            return;
+          if (emitToolCall(currentToolCall, output, stream))
+            emittedToolCalls++;
+          currentToolCall = null;
+        };
+        const cancelHiddenShim = () => {
+          if (hiddenShimTimer) {
+            clearTimeout(hiddenShimTimer);
+            hiddenShimTimer = null;
+          }
+        };
+        let idleTimer = null;
+        let idleCancelled = false;
+        const resetIdle = () => {
+          if (idleTimer)
+            clearTimeout(idleTimer);
+          idleTimer = setTimeout(() => {
+            idleCancelled = true;
+            reader.cancel().catch(() => {});
+          }, IDLE_TIMEOUT_MS);
+        };
+        let gotFirstToken = false;
+        let firstTokenTimedOut = false;
+        let streamError = null;
+        const FIRST_TOKEN_SENTINEL = Symbol("firstTokenTimeout");
+        while (true) {
+          let readResult;
+          if (!gotFirstToken) {
+            const readPromise = reader.read();
+            let firstTokenTimer = null;
+            const result = await Promise.race([
+              readPromise,
+              new Promise((resolve2) => {
+                firstTokenTimer = setTimeout(() => resolve2(FIRST_TOKEN_SENTINEL), firstTokenTimeoutForModel(model.id));
+              })
+            ]);
+            if (firstTokenTimer)
+              clearTimeout(firstTokenTimer);
+            if (result === FIRST_TOKEN_SENTINEL) {
+              readPromise.catch(() => {});
+              reader.cancel().catch(() => {});
+              firstTokenTimedOut = true;
+              break;
+            }
+            readResult = result;
+            gotFirstToken = true;
+            resetIdle();
+          } else {
+            readResult = await reader.read();
+          }
+          const { done, value } = readResult;
+          if (done)
+            break;
+          const decoded = decoder.decode(value, { stream: true });
+          buffer += decoded;
+          if (log.isDebug()) {
+            log.debug("stream.chunk", {
+              seq: chunkSeq++,
+              bytes: value?.byteLength ?? 0,
+              decodedLen: decoded.length,
+              preview: previewChunk(decoded)
+            });
+          }
+          const { events, remaining } = parseKiroEvents(buffer);
+          buffer = remaining;
+          if (events.length > 0)
+            resetIdle();
+          if (log.isDebug() && events.length > 0) {
+            for (const ev of events) {
+              log.debug("stream.event", { seq: eventSeq++, event: ev });
+            }
+          }
+          for (const event of events) {
+            switch (event.type) {
+              case "contextUsage": {
+                const pct = event.data.contextUsagePercentage;
+                output.usage.input = pct >= COMPACTION_THRESHOLD_PCT ? model.contextWindow + 1 : Math.round(pct / 100 * model.contextWindow);
+                receivedContextUsage = true;
+                log.debug("contextUsage", { pct, threshold: COMPACTION_THRESHOLD_PCT, willCompact: pct >= COMPACTION_THRESHOLD_PCT });
+                break;
+              }
+              case "reasoning": {
+                cancelHiddenShim();
+                if (output.content.length === 0 || output.content[output.content.length - 1]?.type !== "thinking") {
+                  output.content.push({ type: "thinking", thinking: "" });
+                  stream.push({ type: "thinking_start", contentIndex: output.content.length - 1, partial: output });
+                }
+                const contentIndex = output.content.length - 1;
+                const tc = output.content[contentIndex];
+                if (event.data.text) {
+                  tc.thinking += event.data.text;
+                  stream.push({ type: "thinking_delta", contentIndex, delta: event.data.text, partial: output });
+                }
+                if (event.data.signature) {
+                  tc.thinkingSignature = event.data.signature;
+                }
+                break;
+              }
+              case "content": {
+                if (event.data === lastContentData)
+                  continue;
+                lastContentData = event.data;
+                totalContent += event.data;
+                cancelHiddenShim();
+                if (thinkingParser) {
+                  thinkingParser.processChunk(event.data);
+                } else {
+                  if (textBlockIndex === null) {
+                    textBlockIndex = output.content.length;
+                    output.content.push({ type: "text", text: "" });
+                    stream.push({ type: "text_start", contentIndex: textBlockIndex, partial: output });
+                  }
+                  const block = output.content[textBlockIndex];
+                  if (block) {
+                    block.text += event.data;
+                    stream.push({
+                      type: "text_delta",
+                      contentIndex: textBlockIndex,
+                      delta: event.data,
+                      partial: output
+                    });
+                  }
+                }
+                break;
+              }
+              case "toolUse": {
+                const tc = event.data;
+                cancelHiddenShim();
+                sawAnyToolCalls = true;
+                if (!currentToolCall || currentToolCall.toolUseId !== tc.toolUseId) {
+                  flushToolCall();
+                  currentToolCall = { toolUseId: tc.toolUseId, name: tc.name, input: "" };
+                }
+                currentToolCall.input += tc.input || "";
+                if (tc.input)
+                  totalContent += tc.input;
+                if (tc.stop)
+                  flushToolCall();
+                break;
+              }
+              case "toolUseInput": {
+                if (currentToolCall)
+                  currentToolCall.input += event.data.input || "";
+                if (event.data.input)
+                  totalContent += event.data.input;
+                break;
+              }
+              case "toolUseStop": {
+                if (event.data.stop)
+                  flushToolCall();
+                break;
+              }
+              case "usage": {
+                usageEvent = event.data;
+                break;
+              }
+              case "error": {
+                streamError = event.data.message ? `${event.data.error}: ${event.data.message}` : event.data.error;
+                reader.cancel().catch(() => {});
+                break;
+              }
+            }
+            if (streamError)
+              break;
+          }
+        }
+        if (idleTimer)
+          clearTimeout(idleTimer);
+        if (firstTokenTimedOut || idleCancelled || streamError) {
+          if (retryCount < MAX_RETRIES) {
+            retryCount++;
+            const delayMs = exponentialBackoff(retryCount - 1, 1000, MAX_RETRY_DELAY_MS);
+            log.warn(`stream ${firstTokenTimedOut ? "first-token timed out" : idleCancelled ? "idle timed out" : `error: ${streamError}`} — retrying (${retryCount}/${MAX_RETRIES})`);
+            cancelHiddenShim();
+            await abortableDelay2(delayMs, options?.signal);
+            output.content = [];
+            textBlockIndex = null;
+            continue;
+          }
+          if (streamError)
+            throw new Error(`Kiro API stream error after max retries: ${streamError}`);
+          throw new Error(`Kiro API error: ${firstTokenTimedOut ? "first token" : "idle"} timeout after max retries`);
+        }
+        cancelHiddenShim();
+        if (currentToolCall && emitToolCall(currentToolCall, output, stream))
+          emittedToolCalls++;
+        if (thinkingParser) {
+          thinkingParser.finalize();
+          textBlockIndex = thinkingParser.getTextBlockIndex();
+        }
+        if (textBlockIndex !== null) {
+          const block = output.content[textBlockIndex];
+          if (block) {
+            stream.push({
+              type: "text_end",
+              contentIndex: textBlockIndex,
+              content: block.text,
+              partial: output
+            });
+          }
+        }
+        if (usageEvent?.inputTokens !== undefined)
+          output.usage.input = usageEvent.inputTokens;
+        output.usage.output = usageEvent?.outputTokens ?? countTokens(totalContent);
+        output.usage.totalTokens = output.usage.input + output.usage.output;
+        try {
+          calculateCost(model, output.usage);
+        } catch {
+          output.usage.cost = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 };
+        }
+        const textBlock = textBlockIndex !== null ? output.content[textBlockIndex] : undefined;
+        const hasText = !!textBlock && textBlock.text.length > 0;
+        if (!hasText && !sawAnyToolCalls) {
+          if (retryCount < MAX_RETRIES) {
+            retryCount++;
+            const delayMs = exponentialBackoff(retryCount - 1, 1000, MAX_RETRY_DELAY_MS);
+            log.warn(`empty response — retrying (${retryCount}/${MAX_RETRIES})`);
+            cancelHiddenShim();
+            output.content = [];
+            textBlockIndex = null;
+            await abortableDelay2(delayMs, options?.signal);
+            continue;
+          }
+          log.warn(`empty response persisted after ${MAX_RETRIES} retries`);
+          cancelHiddenShim();
+        }
+        if (!receivedContextUsage && emittedToolCalls === 0) {
+          output.stopReason = "length";
+        } else {
+          output.stopReason = emittedToolCalls > 0 ? "toolUse" : "stop";
+        }
+        stream.push({
+          type: "done",
+          reason: output.stopReason,
+          message: output
+        });
+        log.debug("response.done", {
+          stopReason: output.stopReason,
+          emittedToolCalls,
+          sawAnyToolCalls,
+          usage: output.usage
+        });
+        stream.end();
+        return;
+      }
+    } catch (error) {
+      output.stopReason = options?.signal?.aborted ? "aborted" : "error";
+      output.errorMessage = error instanceof Error ? error.message : String(error);
+      log.debug("response.caught", { stopReason: output.stopReason, error: output.errorMessage });
+      if (hiddenShimTimer) {
+        clearTimeout(hiddenShimTimer);
+        hiddenShimTimer = null;
+      }
+      stream.push({ type: "error", reason: output.stopReason, error: output });
+      stream.end();
     }
-    const data2 = await resp2.json();
+  })().catch(() => {
     try {
-      const apiRegion = resolveApiRegion(region);
-      const apiModels = await fetchAvailableModels(data2.accessToken, apiRegion);
-      setCachedDynamicModels(buildModelsFromApi(apiModels));
-      log.info(`Fetched and cached ${apiModels.length} models after desktop token refresh`);
-    } catch (err) {
-      log.warn(`Failed to fetch models after desktop token refresh: ${err}`);
-    }
-    return {
-      refresh: `${data2.refreshToken}|||desktop`,
-      access: data2.accessToken,
-      expires: Date.now() + (data2.expiresIn ?? 3600) * 1000 - EXPIRES_BUFFER_MS,
-      clientId: "",
-      clientSecret: "",
-      region,
-      authMethod: "desktop",
-      kiroSyncSource: credentials.kiroSyncSource,
-      kiroSyncTokenKey: credentials.kiroSyncTokenKey
-    };
-  }
-  const endpoint = `https://oidc.${region}.amazonaws.com/token`;
-  const resp = await fetch(endpoint, {
-    method: "POST",
-    headers: { "Content-Type": "application/json", "User-Agent": "pi-kiro" },
-    body: JSON.stringify({ clientId, clientSecret, refreshToken, grantType: "refresh_token" })
+      stream.end();
+    } catch {}
   });
-  if (!resp.ok) {
-    const body = await resp.text().catch(() => "");
-    throw new Error(`Token refresh failed: ${resp.status} ${body}`);
-  }
-  const data = await resp.json();
-  try {
-    const apiRegion = resolveApiRegion(region);
-    const apiModels = await fetchAvailableModels(data.accessToken, apiRegion);
-    setCachedDynamicModels(buildModelsFromApi(apiModels));
-    log.info(`Fetched and cached ${apiModels.length} models after token refresh`);
-  } catch (err) {
-    log.warn(`Failed to fetch models after token refresh, falling back: ${err}`);
-  }
-  return {
-    refresh: `${data.refreshToken}|${clientId}|${clientSecret}|${authMethod}`,
-    access: data.accessToken,
-    expires: Date.now() + (data.expiresIn ?? 3600) * 1000 - EXPIRES_BUFFER_MS,
-    clientId,
-    clientSecret,
-    region,
-    authMethod,
-    kiroSyncSource: credentials.kiroSyncSource,
-    kiroSyncTokenKey: credentials.kiroSyncTokenKey
-  };
-}
-async function refreshKiroToken(credentials) {
-  const inputMethod = credentials.authMethod;
-  const authMethod = inputMethod === "builder-id" || inputMethod === "idc" || inputMethod === "desktop" ? inputMethod : "idc";
-  if (inputMethod !== undefined && inputMethod !== "builder-id" && inputMethod !== "idc" && inputMethod !== "desktop") {
-    log.warn(`refreshKiroToken: unrecognized authMethod "${String(inputMethod)}" — defaulting to "idc"`);
-  }
-  const baseCreds = {
-    ...credentials,
-    clientId: credentials.clientId ?? credentials.refresh.split("|")[1] ?? "",
-    clientSecret: credentials.clientSecret ?? credentials.refresh.split("|")[2] ?? "",
-    region: credentials.region,
-    authMethod
-  };
-  const errors = [];
-  try {
-    log.debug("refresh.cascade: layer 1 — normal refresh");
-    const result = await refreshTokenInner(baseCreds);
-    syncBackToKiroCli(result);
-    return result;
-  } catch (err) {
-    const msg = err instanceof Error ? err.message : String(err);
-    errors.push(`L1(normal): ${msg}`);
-    log.warn(`refresh.cascade: layer 1 failed — ${msg}`);
-  }
-  let freshImport = null;
-  try {
-    log.debug("refresh.cascade: layer 2 — fresh kiro-cli import");
-    const { importFromKiroCli: importFromKiroCli2 } = await Promise.resolve().then(() => (init_kiro_cli_sync(), exports_kiro_cli_sync));
-    freshImport = await importFromKiroCli2();
-    if (freshImport?.accessToken) {
-      const result = kiroCredsFromCliImport(freshImport);
-      log.info("refresh.cascade: layer 2 succeeded — using fresh kiro-cli credentials");
-      return result;
-    }
-    errors.push("L2(fresh-import): no valid credentials found");
-    log.debug("refresh.cascade: layer 2 — no fresh credentials");
-  } catch (err) {
-    const msg = err instanceof Error ? err.message : String(err);
-    errors.push(`L2(fresh-import): ${msg}`);
-    log.warn(`refresh.cascade: layer 2 failed — ${msg}`);
-  }
-  if (freshImport?.refreshToken) {
-    try {
-      log.debug("refresh.cascade: layer 3 — refresh fresh kiro-cli creds");
-      const freshCreds = kiroCredsFromCliImport(freshImport);
-      const result = await refreshTokenInner(freshCreds);
-      syncBackToKiroCli(result);
-      log.info("refresh.cascade: layer 3 succeeded — refreshed fresh kiro-cli credentials");
-      return result;
-    } catch (err) {
-      const msg = err instanceof Error ? err.message : String(err);
-      errors.push(`L3(refresh-fresh): ${msg}`);
-      log.warn(`refresh.cascade: layer 3 failed — ${msg}`);
-    }
-  }
-  let expiredImport = null;
-  try {
-    log.debug("refresh.cascade: layer 4 — expired kiro-cli import");
-    const { getKiroCliCredentialsAllowExpired: getKiroCliCredentialsAllowExpired2 } = await Promise.resolve().then(() => (init_kiro_cli_sync(), exports_kiro_cli_sync));
-    expiredImport = await getKiroCliCredentialsAllowExpired2(freshImport);
-    if (expiredImport?.accessToken) {
-      const result = kiroCredsFromCliImport(expiredImport);
-      log.info("refresh.cascade: layer 4 succeeded — using expired kiro-cli credentials");
-      return result;
-    } else {
-      errors.push("L4(expired-import): no different expired credentials");
-      log.debug("refresh.cascade: layer 4 — no additional expired credentials");
-    }
-  } catch (err) {
-    const msg = err instanceof Error ? err.message : String(err);
-    errors.push(`L4(expired-import): ${msg}`);
-    log.warn(`refresh.cascade: layer 4 failed — ${msg}`);
-  }
-  if (expiredImport?.refreshToken) {
-    try {
-      log.debug("refresh.cascade: layer 5 — refresh expired kiro-cli creds");
-      const expiredCreds = kiroCredsFromCliImport(expiredImport);
-      const result = await refreshTokenInner(expiredCreds);
-      syncBackToKiroCli(result);
-      log.info("refresh.cascade: layer 5 succeeded — refreshed expired kiro-cli credentials");
-      return result;
-    } catch (err) {
-      const msg = err instanceof Error ? err.message : String(err);
-      errors.push(`L5(refresh-expired): ${msg}`);
-      log.warn(`refresh.cascade: layer 5 failed — ${msg}`);
-    }
-  }
-  throw new Error(`Kiro token refresh failed — all 5 cascade layers exhausted. ` + `Re-login required.
-${errors.join(`
-`)}`);
+  return stream;
 }
 
 // src/extension.ts
@@ -3128,27 +2974,63 @@ function readKiroCredentials() {
       }
     }
     const metadata = kiro.metadata;
-    const profileArn = typeof metadata?.profileArn === "string" ? metadata.profileArn : undefined;
+    const profileArn = [kiro.profileArn, metadata?.profileArn].find((v) => typeof v === "string");
     return {
       access: kiro.access,
+      refresh: typeof kiro.refresh === "string" ? kiro.refresh : "",
+      expires: typeof kiro.expires === "number" ? kiro.expires : 0,
       region: kiro.region || "us-east-1",
-      profileArn
+      profileArn,
+      clientId: typeof kiro.clientId === "string" ? kiro.clientId : undefined,
+      clientSecret: typeof kiro.clientSecret === "string" ? kiro.clientSecret : undefined,
+      authMethod: typeof kiro.authMethod === "string" ? kiro.authMethod : undefined
     };
   } catch {
     return null;
   }
 }
+function writeKiroCredentials(refreshed) {
+  try {
+    const authPath = join2(homedir2(), ".pi", "agent", "auth.json");
+    const raw = existsSync2(authPath) ? readFileSync2(authPath, "utf-8") : "{}";
+    const data = JSON.parse(raw);
+    const existing = data["kiro"] ?? {};
+    data["kiro"] = {
+      ...existing,
+      type: "oauth",
+      access: refreshed.access,
+      refresh: refreshed.refresh,
+      expires: refreshed.expires,
+      clientId: refreshed.clientId,
+      clientSecret: refreshed.clientSecret,
+      region: refreshed.region,
+      authMethod: refreshed.authMethod,
+      profileArn: refreshed.profileArn
+    };
+    writeFileSync(authPath, JSON.stringify(data, null, 2), { mode: 384 });
+  } catch (err) {
+    log.warn(`Failed to persist refreshed credentials: ${err}`);
+  }
+}
 async function extension_default(pi) {
   let modelDefs = toProviderModels(kiroModels);
   const creds = readKiroCredentials();
-  if (creds) {
+  if (creds?.profileArn) {
+    let accessToken = creds.access;
+    if (creds.refresh) {
+      try {
+        log.info("Refreshing token at startup…");
+        const refreshed = await refreshKiroToken(creds);
+        accessToken = refreshed.access;
+        writeKiroCredentials(refreshed);
+      } catch (err) {
+        log.warn(`Startup token refresh failed, trying with existing token: ${err}`);
+      }
+    }
     try {
       const apiRegion = resolveApiRegion(creds.region);
-      if (creds.profileArn) {
-        const runtimeUrl = resolveRuntimeUrl(apiRegion);
-        seedProfileArn(runtimeUrl + "/", creds.profileArn);
-      }
-      const apiModels = await fetchAvailableModels(creds.access, apiRegion, creds.profileArn);
+      seedProfileArn(creds.profileArn);
+      const apiModels = await fetchAvailableModels(accessToken, apiRegion, creds.profileArn);
       const dynamicDefs = buildModelsFromApi(apiModels);
       setCachedDynamicModels(dynamicDefs);
       modelDefs = toProviderModels(dynamicDefs);
@@ -3156,6 +3038,8 @@ async function extension_default(pi) {
     } catch (err) {
       log.warn(`Failed to fetch models at startup, using hardcoded fallback: ${err}`);
     }
+  } else {
+    log.warn("Run 'kiro login' to authenticate and fetch models dynamically. Note: This extension does not have the same authentication mechanism as other Kiro tools.");
   }
   pi.registerProvider("kiro", {
     baseUrl: "https://runtime.us-east-1.kiro.dev",
@@ -3168,8 +3052,12 @@ async function extension_default(pi) {
       refreshToken: refreshKiroToken,
       getApiKey: (cred) => cred.access,
       modifyModels: (allModels, cred) => {
-        const apiRegion = resolveApiRegion(cred.region);
+        const kc = cred;
+        const apiRegion = resolveApiRegion(kc.region);
         const nonKiro = allModels.filter((m) => m.provider !== "kiro");
+        if (kc.profileArn) {
+          seedProfileArn(kc.profileArn);
+        }
         const toKiroModel = (m) => ({
           ...m,
           api: "kiro-api",