ns-kiro-core 0.2.3 → 0.3.0

package/dist/blocks.d.ts

@@ -1,4 +1,5 @@
 import type { KiroStreamEvent } from "./types.js";
+type BlockKind = "text" | "thinking" | "toolCall";
 /**
  * Allocates block indexes and accumulates their content while emitting the
  * neutral stream events for each transition.
@@ -21,6 +22,8 @@ export declare class KiroBlockBuffer {
     /** Reserve an index for a block this buffer does not accumulate (a tool call). */
     reserve(): number;
     getText(index: number): string;
+    /** Kinds of the live blocks, in allocation order. */
+    kinds(): BlockKind[];
     /**
      * Rewrite one accumulated block before it is closed. Used by the recovery
      * passes that lift tool calls out of text and by echo-noise stripping, both
@@ -30,3 +33,4 @@ export declare class KiroBlockBuffer {
     /** Announce that an internal retry discarded everything emitted so far. */
     reset(): void;
 }
+export {};

package/dist/blocks.js

@@ -64,6 +64,10 @@ export class KiroBlockBuffer {
     getText(index) {
         return this.blocks.get(index)?.text ?? "";
     }
+    /** Kinds of the live blocks, in allocation order. */
+    kinds() {
+        return [...this.blocks.values()].map((block) => block.kind);
+    }
     /**
      * Rewrite one accumulated block before it is closed. Used by the recovery
      * passes that lift tool calls out of text and by echo-noise stripping, both

package/dist/blocks.js.map

@@ -1 +1 @@
-{"version":3,"file":"blocks.js","sourceRoot":"","sources":["../src/blocks.ts"],"names":[],"mappings":"AAAA,mFAAmF;AACnF,uFAAuF;AAYvF;;;;;;;GAOG;AACH,MAAM,OAAO,eAAe;IAIG;IAHrB,MAAM,GAAG,IAAI,GAAG,EAAiB,CAAC;IAClC,SAAS,GAAG,CAAC,CAAC;IAEtB,YAA6B,IAAsC;QAAtC,SAAI,GAAJ,IAAI,CAAkC;IAAG,CAAC;IAEvE,QAAQ;QACN,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;QACzC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,KAAa;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK;YAAE,OAAO;QAC7B,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC;QACpB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,OAAO,CAAC,KAAa;QACnB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK;YAAE,OAAO;QAClC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,YAAY;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACrE,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,cAAc,CAAC,KAAa,EAAE,KAAa;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK;YAAE,OAAO;QAC7B,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC;QACpB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,WAAW,CAAC,KAAa,EAAE,SAAkB;QAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK;YAAE,OAAO;QAClC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACxG,CAAC;IAED,kFAAkF;IAClF,OAAO;QACL,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACpE,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,CAAC,KAAa;QACnB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;IAC5C,CAAC;IAED;;;;OAIG;IACH,OAAO,CAAC,KAAa,EAAE,IAAY;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,KAAK;YAAE,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;IAC/B,CAAC;IAED,2EAA2E;IAC3E,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC/B,CAAC;CACF"}
+{"version":3,"file":"blocks.js","sourceRoot":"","sources":["../src/blocks.ts"],"names":[],"mappings":"AAAA,mFAAmF;AACnF,uFAAuF;AAYvF;;;;;;;GAOG;AACH,MAAM,OAAO,eAAe;IAIG;IAHrB,MAAM,GAAG,IAAI,GAAG,EAAiB,CAAC;IAClC,SAAS,GAAG,CAAC,CAAC;IAEtB,YAA6B,IAAsC;QAAtC,SAAI,GAAJ,IAAI,CAAkC;IAAG,CAAC;IAEvE,QAAQ;QACN,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;QACzC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,KAAa;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK;YAAE,OAAO;QAC7B,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC;QACpB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,OAAO,CAAC,KAAa;QACnB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK;YAAE,OAAO;QAClC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,YAAY;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACrE,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,cAAc,CAAC,KAAa,EAAE,KAAa;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK;YAAE,OAAO;QAC7B,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC;QACpB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,WAAW,CAAC,KAAa,EAAE,SAAkB;QAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK;YAAE,OAAO;QAClC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACxG,CAAC;IAED,kFAAkF;IAClF,OAAO;QACL,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACpE,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,CAAC,KAAa;QACnB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;IAC5C,CAAC;IAED,qDAAqD;IACrD,KAAK;QACH,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC9D,CAAC;IAED;;;;OAIG;IACH,OAAO,CAAC,KAAa,EAAE,IAAY;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,KAAK;YAAE,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;IAC/B,CAAC;IAED,2EAA2E;IAC3E,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC/B,CAAC;CACF"}

package/dist/cache-estimator.d.ts

@@ -0,0 +1,11 @@
+import type { KiroWireUsage } from "./event-parser.js";
+import type { KiroUsage } from "./types.js";
+import type { KiroUsageTracking } from "./usage-tracking.js";
+/**
+ * Estimate repeated prompt tokens as cache reads, then record the successful
+ * turn as the next baseline. Mutates `usage` only when every conservative guard
+ * passes. Real wire cache counters always win.
+ */
+export declare function applyCacheEstimate(conversationId: string, usage: KiroUsage, wireUsage: KiroWireUsage | null, config: KiroUsageTracking, now?: number): number;
+/** Clear estimator state between tests. */
+export declare function resetCacheEstimatorForTests(): void;

package/dist/cache-estimator.js

@@ -0,0 +1,61 @@
+// ABOUTME: Conservative in-memory cache-read estimation for Kiro usage records.
+// ABOUTME: Reclassifies repeated prompt input without changing total tokens or dollar cost.
+const stateByConversation = new Map();
+const MAX_TRACKED_CONVERSATIONS = 128;
+let warnedWireCacheCounters = false;
+/**
+ * Estimate repeated prompt tokens as cache reads, then record the successful
+ * turn as the next baseline. Mutates `usage` only when every conservative guard
+ * passes. Real wire cache counters always win.
+ */
+export function applyCacheEstimate(conversationId, usage, wireUsage, config, now = Date.now()) {
+    const previous = stateByConversation.get(conversationId);
+    const promptTokens = usage.input + (usage.cacheRead ?? 0) + (usage.cacheWrite ?? 0) + usage.output;
+    const wireReportedCache = wireUsage?.cacheReadInputTokens !== undefined || wireUsage?.cacheWriteInputTokens !== undefined;
+    if (wireReportedCache) {
+        if (config.estimateCacheUsage && !warnedWireCacheCounters) {
+            warnedWireCacheCounters = true;
+            console.warn("[kiro-core] Kiro now reports cache token counters; cache estimation was skipped and may be disabled.");
+        }
+        noteSuccessfulTurn(conversationId, promptTokens, usage.input, now);
+        return 0;
+    }
+    if (!config.estimateCacheUsage || usage.input <= 0)
+        return 0;
+    if (!previous) {
+        noteSuccessfulTurn(conversationId, promptTokens, usage.input, now);
+        return 0;
+    }
+    const expired = config.estimatedCacheTimeout > 0 && now - previous.turnTimestamp > config.estimatedCacheTimeout;
+    const contextShrank = usage.input < previous.promptTokens * 0.6;
+    if (expired || contextShrank) {
+        noteSuccessfulTurn(conversationId, promptTokens, usage.input, now);
+        return 0;
+    }
+    const estimatedRead = Math.min(previous.promptTokens, usage.input);
+    if (estimatedRead > 0) {
+        usage.input -= estimatedRead;
+        usage.cacheRead = (usage.cacheRead ?? 0) + estimatedRead;
+        usage.cacheEstimated = true;
+    }
+    noteSuccessfulTurn(conversationId, promptTokens, promptTokens, now);
+    return estimatedRead;
+}
+function noteSuccessfulTurn(conversationId, promptTokens, input, now) {
+    if (input <= 0 || promptTokens <= 0)
+        return;
+    // Refresh insertion order so the map is a bounded least-recently-used set.
+    stateByConversation.delete(conversationId);
+    stateByConversation.set(conversationId, { promptTokens, turnTimestamp: now });
+    if (stateByConversation.size > MAX_TRACKED_CONVERSATIONS) {
+        const oldest = stateByConversation.keys().next().value;
+        if (oldest !== undefined)
+            stateByConversation.delete(oldest);
+    }
+}
+/** Clear estimator state between tests. */
+export function resetCacheEstimatorForTests() {
+    stateByConversation.clear();
+    warnedWireCacheCounters = false;
+}
+//# sourceMappingURL=cache-estimator.js.map

package/dist/cache-estimator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache-estimator.js","sourceRoot":"","sources":["../src/cache-estimator.ts"],"names":[],"mappings":"AAAA,gFAAgF;AAChF,4FAA4F;AAW5F,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA8B,CAAC;AAClE,MAAM,yBAAyB,GAAG,GAAG,CAAC;AACtC,IAAI,uBAAuB,GAAG,KAAK,CAAC;AAEpC;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAChC,cAAsB,EACtB,KAAgB,EAChB,SAA+B,EAC/B,MAAyB,EACzB,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IAEhB,MAAM,QAAQ,GAAG,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IACzD,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC;IACnG,MAAM,iBAAiB,GACrB,SAAS,EAAE,oBAAoB,KAAK,SAAS,IAAI,SAAS,EAAE,qBAAqB,KAAK,SAAS,CAAC;IAElG,IAAI,iBAAiB,EAAE,CAAC;QACtB,IAAI,MAAM,CAAC,kBAAkB,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAC1D,uBAAuB,GAAG,IAAI,CAAC;YAC/B,OAAO,CAAC,IAAI,CACV,sGAAsG,CACvG,CAAC;QACJ,CAAC;QACD,kBAAkB,CAAC,cAAc,EAAE,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QACnE,OAAO,CAAC,CAAC;IACX,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,KAAK,CAAC,KAAK,IAAI,CAAC;QAAE,OAAO,CAAC,CAAC;IAE7D,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,kBAAkB,CAAC,cAAc,EAAE,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QACnE,OAAO,CAAC,CAAC;IACX,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,qBAAqB,GAAG,CAAC,IAAI,GAAG,GAAG,QAAQ,CAAC,aAAa,GAAG,MAAM,CAAC,qBAAqB,CAAC;IAChH,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,GAAG,QAAQ,CAAC,YAAY,GAAG,GAAG,CAAC;IAChE,IAAI,OAAO,IAAI,aAAa,EAAE,CAAC;QAC7B,kBAAkB,CAAC,cAAc,EAAE,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QACnE,OAAO,CAAC,CAAC;IACX,CAAC;IAED,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;IACnE,IAAI,aAAa,GAAG,CAAC,EAAE,CAAC;QACtB,KAAK,CAAC,KAAK,IAAI,aAAa,CAAC;QAC7B,KAAK,CAAC,SAAS,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC,GAAG,aAAa,CAAC;QACzD,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;IAC9B,CAAC;IAED,kBAAkB,CAAC,cAAc,EAAE,YAAY,EAAE,YAAY,EAAE,GAAG,CAAC,CAAC;IACpE,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,SAAS,kBAAkB,CAAC,cAAsB,EAAE,YAAoB,EAAE,KAAa,EAAE,GAAW;IAClG,IAAI,KAAK,IAAI,CAAC,IAAI,YAAY,IAAI,CAAC;QAAE,OAAO;IAC5C,2EAA2E;IAC3E,mBAAmB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;IAC3C,mBAAmB,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,YAAY,EAAE,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9E,IAAI,mBAAmB,CAAC,IAAI,GAAG,yBAAyB,EAAE,CAAC;QACzD,MAAM,MAAM,GAAG,mBAAmB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;QACvD,IAAI,MAAM,KAAK,SAAS;YAAE,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC/D,CAAC;AACH,CAAC;AAED,2CAA2C;AAC3C,MAAM,UAAU,2BAA2B;IACzC,mBAAmB,CAAC,KAAK,EAAE,CAAC;IAC5B,uBAAuB,GAAG,KAAK,CAAC;AAClC,CAAC"}

package/dist/endpoints.d.ts

@@ -5,4 +5,12 @@ export interface KiroEndpoints {
 }
 export declare function resolveApiRegion(ssoRegion: string | undefined): string;
 export declare function getKiroEndpoints(region: string): KiroEndpoints;
+/**
+ * A Kiro profile is owned by one region and its ARN carries that region. The
+ * runtime API rejects a profile ARN issued in another region with a generic
+ * `Improperly formed request.`, so runtime host selection has to follow the
+ * profile rather than the SSO-derived region: an Identity Center instance in
+ * us-east-1 can own a profile in eu-central-1.
+ */
+export declare function getKiroRegionFromProfileArn(profileArn: string | undefined): string | undefined;
 export declare function getKiroRegionFromEndpoint(endpoint: string): string | undefined;

package/dist/endpoints.js

@@ -7,6 +7,7 @@ const API_REGION_MAP = {
     "ap-southeast-1": "us-east-1",
     "ap-southeast-2": "us-east-1",
     "ap-northeast-1": "us-east-1",
+    "ap-northeast-2": "us-east-1",
     "ap-south-1": "us-east-1",
     "eu-west-1": "eu-central-1",
     "eu-west-2": "eu-central-1",
@@ -28,6 +29,19 @@ export function getKiroEndpoints(region) {
         runtime: `https://runtime.${region}.kiro.dev/`,
     };
 }
+/**
+ * A Kiro profile is owned by one region and its ARN carries that region. The
+ * runtime API rejects a profile ARN issued in another region with a generic
+ * `Improperly formed request.`, so runtime host selection has to follow the
+ * profile rather than the SSO-derived region: an Identity Center instance in
+ * us-east-1 can own a profile in eu-central-1.
+ */
+export function getKiroRegionFromProfileArn(profileArn) {
+    if (!profileArn)
+        return undefined;
+    const region = profileArn.split(":")[3];
+    return region && /^[a-z]{2}(?:-[a-z]+)+-\d$/.test(region) ? region : undefined;
+}
 export function getKiroRegionFromEndpoint(endpoint) {
     try {
         const [service, region, ...suffix] = new URL(endpoint).hostname.split(".");

package/dist/endpoints.js.map

@@ -1 +1 @@
-{"version":3,"file":"endpoints.js","sourceRoot":"","sources":["../src/endpoints.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAC/E,sFAAsF;AAEtF,MAAM,cAAc,GAA2B;IAC7C,WAAW,EAAE,WAAW;IACxB,WAAW,EAAE,WAAW;IACxB,WAAW,EAAE,WAAW;IACxB,gBAAgB,EAAE,WAAW;IAC7B,gBAAgB,EAAE,WAAW;IAC7B,gBAAgB,EAAE,WAAW;IAC7B,YAAY,EAAE,WAAW;IACzB,WAAW,EAAE,cAAc;IAC3B,WAAW,EAAE,cAAc;IAC3B,WAAW,EAAE,cAAc;IAC3B,YAAY,EAAE,cAAc;IAC5B,YAAY,EAAE,cAAc;IAC5B,YAAY,EAAE,cAAc;IAC5B,cAAc,EAAE,cAAc;CAC/B,CAAC;AAQF,MAAM,UAAU,gBAAgB,CAAC,SAA6B;IAC5D,IAAI,CAAC,SAAS;QAAE,OAAO,WAAW,CAAC;IACnC,OAAO,cAAc,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAc;IAC7C,OAAO;QACL,MAAM;QACN,UAAU,EAAE,sBAAsB,MAAM,YAAY;QACpD,OAAO,EAAE,mBAAmB,MAAM,YAAY;KAC/C,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,QAAgB;IACxD,IAAI,CAAC;QACH,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC3E,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChC,IAAI,CAAC,OAAO,KAAK,YAAY,IAAI,OAAO,KAAK,SAAS,CAAC,IAAI,MAAM,KAAK,UAAU;YAAE,OAAO,MAAM,CAAC;QAChG,OAAO,SAAS,CAAC;IACnB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC"}
+{"version":3,"file":"endpoints.js","sourceRoot":"","sources":["../src/endpoints.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAC/E,sFAAsF;AAEtF,MAAM,cAAc,GAA2B;IAC7C,WAAW,EAAE,WAAW;IACxB,WAAW,EAAE,WAAW;IACxB,WAAW,EAAE,WAAW;IACxB,gBAAgB,EAAE,WAAW;IAC7B,gBAAgB,EAAE,WAAW;IAC7B,gBAAgB,EAAE,WAAW;IAC7B,gBAAgB,EAAE,WAAW;IAC7B,YAAY,EAAE,WAAW;IACzB,WAAW,EAAE,cAAc;IAC3B,WAAW,EAAE,cAAc;IAC3B,WAAW,EAAE,cAAc;IAC3B,YAAY,EAAE,cAAc;IAC5B,YAAY,EAAE,cAAc;IAC5B,YAAY,EAAE,cAAc;IAC5B,cAAc,EAAE,cAAc;CAC/B,CAAC;AAQF,MAAM,UAAU,gBAAgB,CAAC,SAA6B;IAC5D,IAAI,CAAC,SAAS;QAAE,OAAO,WAAW,CAAC;IACnC,OAAO,cAAc,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAc;IAC7C,OAAO;QACL,MAAM;QACN,UAAU,EAAE,sBAAsB,MAAM,YAAY;QACpD,OAAO,EAAE,mBAAmB,MAAM,YAAY;KAC/C,CAAC;AACJ,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,2BAA2B,CAAC,UAA8B;IACxE,IAAI,CAAC,UAAU;QAAE,OAAO,SAAS,CAAC;IAClC,MAAM,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACxC,OAAO,MAAM,IAAI,2BAA2B,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AACjF,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,QAAgB;IACxD,IAAI,CAAC;QACH,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC3E,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChC,IAAI,CAAC,OAAO,KAAK,YAAY,IAAI,OAAO,KAAK,SAAS,CAAC,IAAI,MAAM,KAAK,UAAU;YAAE,OAAO,MAAM,CAAC;QAChG,OAAO,SAAS,CAAC;IACnB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC"}

package/dist/errors.d.ts

@@ -0,0 +1,39 @@
+export { parseRetryAfterMs } from "./retry.js";
+/** Retries this provider already performed internally before giving up. */
+export interface KiroProviderAttempts {
+    /** 403 credential-refresh retries (`exponentialBackoff(n, 500, MAX_RETRY_DELAY)`). */
+    credentialRefresh: number;
+    /** INSUFFICIENT_MODEL_CAPACITY retries (`capacityRetryConfig`, 5s base / 30s ceiling). */
+    capacity: number;
+}
+/**
+ * A Kiro runtime HTTP failure with its classification preserved.
+ *
+ * `message` is deliberately identical to the string this provider has always
+ * thrown — downstream consumers match on that text, and the typed fields are
+ * strictly additive. Read them instead of re-parsing `message`.
+ *
+ * Unlike the upstream pi extension, `streamKiro` here is a generator that
+ * rejects with this error, so the typed fields reach the host adapter directly
+ * on the thrown value rather than through a diagnostics channel.
+ */
+export declare class KiroApiError extends Error {
+    readonly status: number;
+    readonly reasonCode?: string | undefined;
+    readonly retryAfterMs?: number | undefined;
+    readonly providerAttempts?: KiroProviderAttempts | undefined;
+    constructor(message: string, status: number, reasonCode?: string | undefined, retryAfterMs?: number | undefined, providerAttempts?: KiroProviderAttempts | undefined);
+}
+/**
+ * Pull Kiro's reason code out of an error body.
+ *
+ * Prefers the parsed JSON `reason` field, which is what Kiro actually sends
+ * (`{"message":"Improperly formed request.","reason":"REQUEST_BODY_INVALID"}`).
+ * Falls back to scanning for a known marker so a plain-text or
+ * event-stream-wrapped body still classifies.
+ *
+ * Returns undefined rather than guessing when the body carries no code: a
+ * bare 413 or `Input is too long` has no reason code, and inventing one would
+ * make an absent classification indistinguishable from a real one.
+ */
+export declare function extractKiroReasonCode(errorText: string): string | undefined;

package/dist/errors.js

@@ -0,0 +1,82 @@
+// ABOUTME: Typed error for Kiro runtime HTTP failures, carrying the class the
+// ABOUTME: throw site already computed so consumers need not re-parse the message.
+import { KIRO_REASON_CODES } from "./retry.js";
+/**
+ * Known reason-code markers to scan for when the body is not parseable JSON.
+ *
+ * Deliberately module-private and deliberately NOT named `KIRO_REASON_CODES`:
+ * this is only a fallback marker list for `extractKiroReasonCode`, not a public
+ * vocabulary. The literals it shares with `src/retry.ts` (`TOO_BIG_PATTERNS`,
+ * `NON_RETRYABLE_BODY_PATTERNS`, `CAPACITY_PATTERN`) are matched there for
+ * *classification*, which is a separate concern — `Input is too long` is a
+ * classification marker with no reason code, and `REQUEST_BODY_INVALID` is a
+ * reason code that must never classify as too-big.
+ *
+ * Not exhaustive: `reasonCode` is a passthrough of whatever the body carried, so
+ * an unrecognized code still reaches the consumer verbatim.
+ */
+const REASON_CODE_MARKERS = [
+    KIRO_REASON_CODES.CONTENT_LENGTH_EXCEEDS_THRESHOLD,
+    KIRO_REASON_CODES.MONTHLY_REQUEST_COUNT,
+    KIRO_REASON_CODES.INSUFFICIENT_MODEL_CAPACITY,
+    KIRO_REASON_CODES.REQUEST_BODY_INVALID,
+];
+// Retry-header units and malformed-value handling are owned by retry.ts.
+// Re-export the canonical helper to preserve this module's public surface
+// without maintaining a second parser that can drift from retry behavior.
+export { parseRetryAfterMs } from "./retry.js";
+/**
+ * A Kiro runtime HTTP failure with its classification preserved.
+ *
+ * `message` is deliberately identical to the string this provider has always
+ * thrown — downstream consumers match on that text, and the typed fields are
+ * strictly additive. Read them instead of re-parsing `message`.
+ *
+ * Unlike the upstream pi extension, `streamKiro` here is a generator that
+ * rejects with this error, so the typed fields reach the host adapter directly
+ * on the thrown value rather than through a diagnostics channel.
+ */
+export class KiroApiError extends Error {
+    status;
+    reasonCode;
+    retryAfterMs;
+    providerAttempts;
+    constructor(message, status, reasonCode, retryAfterMs, providerAttempts) {
+        super(message);
+        this.status = status;
+        this.reasonCode = reasonCode;
+        this.retryAfterMs = retryAfterMs;
+        this.providerAttempts = providerAttempts;
+        this.name = "KiroApiError";
+    }
+}
+/**
+ * Pull Kiro's reason code out of an error body.
+ *
+ * Prefers the parsed JSON `reason` field, which is what Kiro actually sends
+ * (`{"message":"Improperly formed request.","reason":"REQUEST_BODY_INVALID"}`).
+ * Falls back to scanning for a known marker so a plain-text or
+ * event-stream-wrapped body still classifies.
+ *
+ * Returns undefined rather than guessing when the body carries no code: a
+ * bare 413 or `Input is too long` has no reason code, and inventing one would
+ * make an absent classification indistinguishable from a real one.
+ */
+export function extractKiroReasonCode(errorText) {
+    if (!errorText)
+        return undefined;
+    const trimmed = errorText.trim();
+    if (trimmed.startsWith("{")) {
+        try {
+            const parsed = JSON.parse(trimmed);
+            const reason = parsed.reason ?? parsed.reasonCode;
+            if (typeof reason === "string" && reason.length > 0)
+                return reason;
+        }
+        catch {
+            // Fall through to marker scan — a truncated or wrapped body is still useful.
+        }
+    }
+    return REASON_CODE_MARKERS.find((code) => errorText.includes(code));
+}
+//# sourceMappingURL=errors.js.map

package/dist/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../src/errors.ts"],"names":[],"mappings":"AAAA,8EAA8E;AAC9E,mFAAmF;AAEnF,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE/C;;;;;;;;;;;;;GAaG;AACH,MAAM,mBAAmB,GAAG;IAC1B,iBAAiB,CAAC,gCAAgC;IAClD,iBAAiB,CAAC,qBAAqB;IACvC,iBAAiB,CAAC,2BAA2B;IAC7C,iBAAiB,CAAC,oBAAoB;CAC9B,CAAC;AAEX,yEAAyE;AACzE,0EAA0E;AAC1E,0EAA0E;AAC1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAU/C;;;;;;;;;;GAUG;AACH,MAAM,OAAO,YAAa,SAAQ,KAAK;IAG1B;IACA;IACA;IACA;IALX,YACE,OAAe,EACN,MAAc,EACd,UAAmB,EACnB,YAAqB,EACrB,gBAAuC;QAEhD,KAAK,CAAC,OAAO,CAAC,CAAC;QALN,WAAM,GAAN,MAAM,CAAQ;QACd,eAAU,GAAV,UAAU,CAAS;QACnB,iBAAY,GAAZ,YAAY,CAAS;QACrB,qBAAgB,GAAhB,gBAAgB,CAAuB;QAGhD,IAAI,CAAC,IAAI,GAAG,cAAc,CAAC;IAC7B,CAAC;CACF;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,qBAAqB,CAAC,SAAiB;IACrD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IACjC,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,EAAE,CAAC;IACjC,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAA+C,CAAC;YACjF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,UAAU,CAAC;YAClD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;gBAAE,OAAO,MAAM,CAAC;QACrE,CAAC;QAAC,MAAM,CAAC;YACP,6EAA6E;QAC/E,CAAC;IACH,CAAC;IACD,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;AACtE,CAAC"}

package/dist/event-parser.d.ts

@@ -1,3 +1,85 @@
+/**
+ * Members of the `ChatResponseStream` tagged union emitted by
+ * `generateAssistantResponse` on `runtime.{region}.kiro.dev`.
+ *
+ * Source of truth: the generated Smithy client for the same service
+ * (`@amzn/kiro-runtime-service-typescript-client`, `ChatResponseStream`).
+ * The frame's `:event-type` header carries one of these keys, so routing is a
+ * switch on the key rather than a guess based on which fields happen to be set.
+ */
+export declare const KIRO_EVENT_KEYS: readonly ["assistantResponseEvent", "codeReferenceEvent", "contextUsageEvent", "documentCitationEvent", "error", "metadataEvent", "meteringEvent", "reasoningContentEvent", "serviceUnavailableError", "throttlingError", "toolResultEvent", "toolUseEvent", "validationError"];
+export type KiroEventKey = (typeof KIRO_EVENT_KEYS)[number];
+export declare function isKiroEventKey(key: string): key is KiroEventKey;
+/**
+ * What a `metadataEvent` frame's `tokenUsage` reports.
+ *
+ * Observed 2026-09-06 against `claude-sonnet-5` in us-east-1: Kiro does not
+ * send token counts at all — the billing record arrives on `meteringEvent`
+ * instead. Token counts are still parsed because the object form is what the
+ * underlying services emit and Kiro may yet expose it.
+ *
+ * Kiro does cache prompts server-side (a repeated prefix billed 0.0350 against
+ * 0.0659 for a fresh one, and a changed prefix went straight back to 0.0659),
+ * but it reports no cache token counts and its request schema
+ * (`additionalProperties: false`) offers no way to ask for caching. The cache
+ * fields below are therefore read defensively, in the spellings the upstream
+ * services use, and stay `undefined` when absent rather than becoming `0`.
+ */
+export interface KiroWireUsage {
+    /** `TokenUsage.uncachedInputTokens` — input tokens billed at full rate. */
+    inputTokens?: number;
+    outputTokens?: number;
+    totalTokens?: number;
+    cacheReadInputTokens?: number;
+    cacheWriteInputTokens?: number;
+    contextUsagePercentage?: number;
+    /**
+     * `TokenUsage.normalizedTokenUsage` — usage normalized by MPS from credit
+     * information. Distinct from `MeteringEvent.usage`, which is a raw credit
+     * count; surfaced so it is not dropped at the boundary like the rest of
+     * `metadataEvent` was.
+     */
+    normalizedTokenUsage?: number;
+    /** `MetadataEvent.stopReason`, passed through verbatim. */
+    rawStopReason?: string;
+    /** `MetadataEvent.stopDetails`, passed through verbatim. */
+    stopDetails?: Record<string, unknown>;
+}
+/** Which modeled union member produced an error event. */
+export type KiroErrorKind = "internalServer" | "throttling" | "validation" | "serviceUnavailable" | "unknown";
+/**
+ * The four error members of `ChatResponseStream` target `@error` shapes, so the
+ * service frames them as `:message-type: exception` with the union member name
+ * in `:exception-type` — not as ordinary `event` frames. Mapping the member to
+ * its exception class here keeps both framings on one table.
+ */
+export declare const KIRO_ERROR_MEMBERS: Readonly<Record<string, {
+    kind: KiroErrorKind;
+    exception: string;
+}>>;
+/**
+ * Look up an error member by a token the SERVICE chose.
+ *
+ * `KIRO_ERROR_TOKENS` is an ordinary object literal, so a bare index would
+ * also resolve inherited `Object.prototype` members: an `:exception-type` of
+ * `toString` or `constructor` returns a truthy value whose `kind` and
+ * `exception` are both undefined, silently discarding the member name this
+ * routing exists to preserve. Only own properties count as modeled members.
+ */
+export declare function lookupKiroErrorMember(key: string): {
+    kind: KiroErrorKind;
+    exception: string;
+} | undefined;
+export interface KiroErrorData {
+    /** Exception class name, or the legacy free-form `error` string. */
+    error: string;
+    message?: string;
+    kind: KiroErrorKind;
+    /** `ThrottlingException.reason` / `ValidationException.reason`, passed through. */
+    reason?: string;
+    /** `ThrottlingException.retryAfterMilliseconds`. */
+    retryAfterMilliseconds?: number;
+}
 export type KiroWireEvent = {
     type: "content";
     data: string;
@@ -36,38 +118,57 @@ export type KiroWireEvent = {
 } | {
     type: "usage";
     data: KiroWireUsage;
+}
+/** `MeteringEvent` — `usage` is a COUNT OF CREDITS, not tokens. */
+ | {
+    type: "metering";
+    data: {
+        credits?: number;
+        unit?: string;
+        unitPlural?: string;
+    };
 } | {
     type: "error";
+    data: KiroErrorData;
+}
+/** A known union member with no consumer yet. Kept distinct from unparseable. */
+ | {
+    type: "ignored";
     data: {
-        error: string;
-        message?: string;
+        key: string;
     };
 };
 /**
- * What a `usage` frame reports.
+ * Route a decoded stream frame by its modeled `:event-type` key.
  *
- * Observed 2026-09-06 against `claude-sonnet-5` in us-east-1: Kiro does not
- * send token counts at all. The frame is a billing record —
- * `{unit: "credit", unitPlural: "credits", usage: 0.0659}` — so `usage` is a
- * NUMBER, not an object, and reading `inputTokens` off it silently yields
- * nothing. Token counts are still parsed because the object form is what the
- * underlying services emit and Kiro may yet expose it.
+ * `key` is the `ChatResponseStream` union member name from the frame header.
+ * An unrecognized key falls back to {@link parseKiroEventByShape} so a member
+ * added server-side degrades instead of breaking the stream.
  *
- * Kiro does cache prompts server-side (a repeated prefix billed 0.0350 against
- * 0.0659 for a fresh one, and a changed prefix went straight back to 0.0659),
- * but it reports no cache token counts and its request schema
- * (`additionalProperties: false`) offers no way to ask for caching. The cache
- * fields below are therefore read defensively, in the spellings the upstream
- * services use, and stay `undefined` when absent rather than becoming `0`.
+ * Note: a frame whose `:event-type` is the literal `$unknown` never reaches
+ * here. The Smithy marshaller drops any event frame for which the deserializer
+ * returns a `$unknown` property, and the deserializer keys its result by the
+ * header value, so `$unknown` is discarded one layer up. The fallback below is
+ * therefore reached only by a real, unrecognized member name.
  */
-export interface KiroWireUsage {
-    inputTokens?: number;
-    outputTokens?: number;
-    cacheReadTokens?: number;
-    cacheWriteTokens?: number;
-    /** Billed amount for the turn — the only usage figure Kiro actually reports. */
-    credits?: number;
-    /** Unit the amount is denominated in, e.g. `"credit"`. */
-    creditUnit?: string;
-}
-export declare function parseKiroEvent(parsed: Record<string, unknown>): KiroWireEvent | null;
+export declare function parseKiroEvent(key: string, parsed: Record<string, unknown>): KiroWireEvent | null;
+/**
+ * Route an `:message-type: exception` frame by its `:exception-type` token.
+ *
+ * The Smithy marshaller throws whatever the deserializer returns for that key,
+ * so this is the only place the modeled exception class, `reason`, and
+ * `retryAfterMilliseconds` are still structured. Accepts either token form the
+ * service uses (union member name or exception class name). Returns `null` for
+ * a token that is not one of the four modeled errors; the caller is responsible
+ * for still preserving that name (see `response-stream.ts`), because Smithy's
+ * own raw-body fallback only fires for a `$unknown` result this deserializer
+ * never produces.
+ */
+export declare function parseKiroExceptionFrame(key: string, parsed: Record<string, unknown>): KiroErrorData | null;
+/**
+ * Fail-open fallback for frames carrying an unrecognized `:event-type`.
+ *
+ * Order-dependent field sniffing. Only reachable when the frame's key is not a
+ * known `ChatResponseStream` member; modeled frames never reach here.
+ */
+export declare function parseKiroEventByShape(parsed: Record<string, unknown>): KiroWireEvent | null;