@arvoretech/pi-kiro-provider 0.8.1

package/README.md

@@ -0,0 +1,121 @@
+# @arvoretech/pi-kiro-provider
+
+A [pi](https://github.com/earendil-works/pi-coding-agent) provider extension that connects pi to the **Kiro API** (AWS CodeWhisperer/Q), exposing kiro-cli-verified models through one provider surface.
+
+## Why this exists
+
+Kiro gives you a strong free model menu, but pi needs a provider that speaks Kiro's auth, model catalog, and streaming protocol cleanly. `@arvoretech/pi-kiro-provider` handles that bridge, including:
+
+- AWS Builder ID, IAM Identity Center, Google, and GitHub login flows
+- shared credentials from an existing `kiro-cli` session when available
+- reasoning-aware streaming
+- region-aware model filtering so pi only shows models your Kiro region can actually use
+
+## Quick start
+
+Add to `.pi/settings.json`:
+
+```json
+{
+  "packages": ["npm:@arvoretech/pi-kiro-provider"]
+}
+```
+
+Or reference the local dist path for project-local development:
+
+```json
+{
+  "extensions": ["./arvore-pi-extensions/packages/kiro-provider/dist/index.js"]
+}
+```
+
+Then log in from pi:
+
+```text
+/login kiro
+```
+
+The login flow supports:
+- **AWS Builder ID** — native device-code flow, works well over SSH/remotes
+- **Your organization** — IAM Identity Center start URL
+- **Google** — social login via `kiro-cli`
+- **GitHub** — social login via `kiro-cli`
+
+If you already use [kiro-cli](https://kiro.dev), the provider can reuse those credentials instead of forcing a second login.
+
+## Models
+
+| Family | Models | Context | Reasoning |
+|--------|--------|---------|-----------|
+| Claude Opus | `claude-opus-4-7`, `claude-opus-4-6` | 1M | ✓ |
+| Claude Sonnet 5 | `claude-sonnet-5` | 1M | ✓ |
+| Claude Sonnet 4.6 | `claude-sonnet-4-6` | 1M | ✓ |
+| Claude Sonnet 4.5 | `claude-sonnet-4-5` | 200K | ✓ |
+| Claude Sonnet 4 | `claude-sonnet-4` | 200K | ✓ |
+| Claude Haiku 4.5 | `claude-haiku-4-5` | 200K | ✗ |
+| Claude Fable 5 | `claude-fable-5` | 1M | ✓ |
+| DeepSeek 3.2 | `deepseek-3-2` | 164K | ✓ |
+| MiniMax | `minimax-m2-1`, `minimax-m2-5` | 196K | ✗ |
+| GLM 5 | `glm-5` | 200K | ✓ |
+| Qwen3 Coder | `qwen3-coder-next` | 256K | ✓ |
+| Auto | `auto` | 1M | ✓ |
+
+All listed models are free to use through Kiro.
+
+## Usage
+
+Once logged in, select any Kiro model in pi:
+
+```text
+/model claude-sonnet-4-6
+```
+
+Or let Kiro pick automatically:
+
+```text
+/model auto
+```
+
+Reasoning is automatically enabled for supported models. Use `/reasoning` to adjust the thinking budget.
+
+## Retry Behavior
+
+Generic transient retries such as HTTP `429` and `5xx` are handled by `pi-coding-agent` at the session layer.
+
+This provider only keeps local recovery for Kiro-specific cases:
+- `403` auth races, where it can refresh credentials from `kiro-cli`
+- first-token / stalled-stream recovery
+- empty-stream retries
+- non-retryable Kiro body markers like `MONTHLY_REQUEST_COUNT` and `INSUFFICIENT_MODEL_CAPACITY`
+
+## Development
+
+```bash
+pnpm build       # Compile TypeScript
+pnpm check       # Type check (no emit)
+pnpm test        # Run the Vitest suite
+pnpm test:watch  # Watch mode
+```
+
+## Architecture
+
+The extension is organized as one feature per file:
+
+```
+src/
+├── index.ts            # Extension registration
+├── models.ts           # 12 model definitions + ID resolution
+├── oauth.ts            # Multi-provider auth (Builder ID / Google / GitHub)
+├── kiro-cli.ts         # kiro-cli credential sharing
+├── transform.ts        # Message format conversion
+├── history.ts          # Conversation history management
+├── thinking-parser.ts  # Streaming <thinking> tag parser
+├── event-parser.ts     # Kiro stream event parser
+└── stream.ts           # Main streaming orchestrator
+```
+
+See [src/](src/) — each file owns one feature (extension registration, model catalog, OAuth, kiro-cli credential sharing, message transform, history management, thinking-tag parsing, event parsing, and the streaming orchestrator).
+
+## License
+
+MIT

package/dist/bracket-tool-parser.d.ts

@@ -0,0 +1,12 @@
+export declare function findJsonEnd(text: string, start: number): number;
+export interface BracketToolCall {
+    toolUseId: string;
+    name: string;
+    arguments: Record<string, unknown>;
+}
+export interface BracketParseResult {
+    toolCalls: BracketToolCall[];
+    cleanedText: string;
+}
+export declare function parseBracketToolCalls(text: string): BracketParseResult;
+//# sourceMappingURL=bracket-tool-parser.d.ts.map

package/dist/bracket-tool-parser.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bracket-tool-parser.d.ts","sourceRoot":"","sources":["../src/bracket-tool-parser.ts"],"names":[],"mappings":"AAGA,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CA2B/D;AAED,MAAM,WAAW,eAAe;IAC/B,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,kBAAkB;IAClC,SAAS,EAAE,eAAe,EAAE,CAAC;IAC7B,WAAW,EAAE,MAAM,CAAC;CACpB;AAID,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,MAAM,GAAG,kBAAkB,CA8CtE"}

package/dist/bracket-tool-parser.js

@@ -0,0 +1,78 @@
+// ABOUTME: Extracts bracket-style tool calls from content text as a fallback.
+// ABOUTME: Parses [Called func_name with args: {...}] patterns from model output.
+export 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;
+}
+const BRACKET_PATTERN = /\[Called\s+([\w-]+)\s+with\s+args:\s*/g;
+export function parseBracketToolCalls(text) {
+    const toolCalls = [];
+    const removals = [];
+    // Reset the regex lastIndex to ensure consistent behavior
+    BRACKET_PATTERN.lastIndex = 0;
+    let match = BRACKET_PATTERN.exec(text);
+    while (match !== null) {
+        const name = match[1];
+        const jsonStart = match.index + match[0].length;
+        const braceIdx = text.indexOf("{", jsonStart);
+        if (braceIdx >= 0 && braceIdx === jsonStart) {
+            const jsonEndIdx = findJsonEnd(text, braceIdx);
+            if (jsonEndIdx >= 0) {
+                const afterJson = text.indexOf("]", jsonEndIdx + 1);
+                if (afterJson >= 0) {
+                    const between = text.substring(jsonEndIdx + 1, afterJson).trim();
+                    if (between.length === 0) {
+                        const jsonStr = text.substring(braceIdx, jsonEndIdx + 1);
+                        try {
+                            const args = JSON.parse(jsonStr);
+                            toolCalls.push({
+                                toolUseId: crypto.randomUUID(),
+                                name,
+                                arguments: args,
+                            });
+                            removals.push({ start: match.index, end: afterJson + 1 });
+                        }
+                        catch {
+                            // Malformed JSON — skip this match
+                        }
+                    }
+                }
+            }
+        }
+        match = BRACKET_PATTERN.exec(text);
+    }
+    // Build cleaned text by removing matched bracket patterns (reverse order to preserve indices)
+    let cleanedText = text;
+    for (let i = removals.length - 1; i >= 0; i--) {
+        const { start, end } = removals[i];
+        cleanedText = cleanedText.substring(0, start) + cleanedText.substring(end);
+    }
+    return { toolCalls, cleanedText };
+}
+//# sourceMappingURL=bracket-tool-parser.js.map

package/dist/bracket-tool-parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bracket-tool-parser.js","sourceRoot":"","sources":["../src/bracket-tool-parser.ts"],"names":[],"mappings":"AAAA,8EAA8E;AAC9E,kFAAkF;AAElF,MAAM,UAAU,WAAW,CAAC,IAAY,EAAE,KAAa;IACtD,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,IAAI,UAAU,GAAG,KAAK,CAAC;IACvB,KAAK,IAAI,CAAC,GAAG,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACrB,IAAI,UAAU,EAAE,CAAC;YAChB,UAAU,GAAG,KAAK,CAAC;YACnB,SAAS;QACV,CAAC;QACD,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YACnB,UAAU,GAAG,IAAI,CAAC;YAClB,SAAS;QACV,CAAC;QACD,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;YAClB,QAAQ,GAAG,CAAC,QAAQ,CAAC;YACrB,SAAS;QACV,CAAC;QACD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACf,IAAI,IAAI,KAAK,GAAG;gBAAE,UAAU,EAAE,CAAC;iBAC1B,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACvB,UAAU,EAAE,CAAC;gBACb,IAAI,UAAU,KAAK,CAAC;oBAAE,OAAO,CAAC,CAAC;YAChC,CAAC;QACF,CAAC;IACF,CAAC;IACD,OAAO,CAAC,CAAC,CAAC;AACX,CAAC;AAaD,MAAM,eAAe,GAAG,wCAAwC,CAAC;AAEjE,MAAM,UAAU,qBAAqB,CAAC,IAAY;IACjD,MAAM,SAAS,GAAsB,EAAE,CAAC;IACxC,MAAM,QAAQ,GAA0C,EAAE,CAAC;IAE3D,0DAA0D;IAC1D,eAAe,CAAC,SAAS,GAAG,CAAC,CAAC;IAC9B,IAAI,KAAK,GAA2B,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/D,OAAO,KAAK,KAAK,IAAI,EAAE,CAAC;QACvB,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;QAEhD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC9C,IAAI,QAAQ,IAAI,CAAC,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC7C,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YAC/C,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;gBACrB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC;gBACpD,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;oBACpB,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC;oBACjE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC;wBACzD,IAAI,CAAC;4BACJ,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BACjC,SAAS,CAAC,IAAI,CAAC;gCACd,SAAS,EAAE,MAAM,CAAC,UAAU,EAAE;gCAC9B,IAAI;gCACJ,SAAS,EAAE,IAAI;6BACf,CAAC,CAAC;4BACH,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,SAAS,GAAG,CAAC,EAAE,CAAC,CAAC;wBAC3D,CAAC;wBAAC,MAAM,CAAC;4BACR,mCAAmC;wBACpC,CAAC;oBACF,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;QACD,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,8FAA8F;IAC9F,IAAI,WAAW,GAAG,IAAI,CAAC;IACvB,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/C,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QACnC,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAC5E,CAAC;IAED,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC;AACnC,CAAC"}

package/dist/debug.d.ts

@@ -0,0 +1,3 @@
+export declare function debugEnabled(): boolean;
+export declare function debugLog(section: string, data?: unknown): void;
+//# sourceMappingURL=debug.d.ts.map

package/dist/debug.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"debug.d.ts","sourceRoot":"","sources":["../src/debug.ts"],"names":[],"mappings":"AAcA,wBAAgB,YAAY,IAAI,OAAO,CAEtC;AAED,wBAAgB,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,IAAI,CAkB9D"}

package/dist/debug.js

@@ -0,0 +1,49 @@
+// ABOUTME: Debug logging — dumps requests, responses, stream events, and errors
+// ABOUTME: to a log file when KIRO_DEBUG=1. Custom path via KIRO_DEBUG_LOG.
+import { appendFileSync, mkdirSync } from "node:fs";
+import { homedir } from "node:os";
+import { dirname, join } from "node:path";
+const ENABLED = !!process.env.KIRO_DEBUG && process.env.KIRO_DEBUG !== "0";
+const LOG_FILE = process.env.KIRO_DEBUG_LOG ||
+    join(homedir(), ".pi", "logs", "kiro-debug.log");
+let dirReady = false;
+export function debugEnabled() {
+    return ENABLED;
+}
+export function debugLog(section, data) {
+    if (!ENABLED)
+        return;
+    try {
+        if (!dirReady) {
+            mkdirSync(dirname(LOG_FILE), { recursive: true });
+            dirReady = true;
+        }
+        const ts = new Date().toISOString();
+        const body = data === undefined
+            ? ""
+            : typeof data === "string"
+                ? ` ${data}`
+                : ` ${JSON.stringify(data, redact, 2)}`;
+        appendFileSync(LOG_FILE, `${ts} [${section}]${body}\n`);
+    }
+    catch {
+        // best-effort; never break the provider
+    }
+}
+/** Redact auth tokens in JSON output. */
+function redact(key, value) {
+    if (typeof value !== "string")
+        return value;
+    const k = key.toLowerCase();
+    if (k === "authorization" ||
+        k === "access" ||
+        k === "refresh" ||
+        k.includes("token") ||
+        k.includes("secret")) {
+        return value.length > 8
+            ? `${value.slice(0, 4)}…${value.slice(-4)}(${value.length})`
+            : "<redacted>";
+    }
+    return value;
+}
+//# sourceMappingURL=debug.js.map

package/dist/debug.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"debug.js","sourceRoot":"","sources":["../src/debug.ts"],"names":[],"mappings":"AAAA,gFAAgF;AAChF,4EAA4E;AAE5E,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAE1C,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,KAAK,GAAG,CAAC;AAC3E,MAAM,QAAQ,GACb,OAAO,CAAC,GAAG,CAAC,cAAc;IAC1B,IAAI,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,gBAAgB,CAAC,CAAC;AAElD,IAAI,QAAQ,GAAG,KAAK,CAAC;AAErB,MAAM,UAAU,YAAY;IAC3B,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,OAAe,EAAE,IAAc;IACvD,IAAI,CAAC,OAAO;QAAE,OAAO;IACrB,IAAI,CAAC;QACJ,IAAI,CAAC,QAAQ,EAAE,CAAC;YACf,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,QAAQ,GAAG,IAAI,CAAC;QACjB,CAAC;QACD,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,IAAI,GACT,IAAI,KAAK,SAAS;YACjB,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,OAAO,IAAI,KAAK,QAAQ;gBACzB,CAAC,CAAC,IAAI,IAAI,EAAE;gBACZ,CAAC,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC;QAC3C,cAAc,CAAC,QAAQ,EAAE,GAAG,EAAE,KAAK,OAAO,IAAI,IAAI,IAAI,CAAC,CAAC;IACzD,CAAC;IAAC,MAAM,CAAC;QACR,wCAAwC;IACzC,CAAC;AACF,CAAC;AAED,yCAAyC;AACzC,SAAS,MAAM,CAAC,GAAW,EAAE,KAAc;IAC1C,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,MAAM,CAAC,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;IAC5B,IACC,CAAC,KAAK,eAAe;QACrB,CAAC,KAAK,QAAQ;QACd,CAAC,KAAK,SAAS;QACf,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;QACnB,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EACnB,CAAC;QACF,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC;YACtB,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG;YAC5D,CAAC,CAAC,YAAY,CAAC;IACjB,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC"}

package/dist/event-parser.d.ts

@@ -0,0 +1,44 @@
+export type KiroStreamEvent = {
+    type: "content";
+    data: string;
+} | {
+    type: "toolUse";
+    data: {
+        name: string;
+        toolUseId: string;
+        input: string;
+        stop?: boolean;
+    };
+} | {
+    type: "toolUseInput";
+    data: {
+        input: string;
+    };
+} | {
+    type: "toolUseStop";
+    data: {
+        stop: boolean;
+    };
+} | {
+    type: "contextUsage";
+    data: {
+        contextUsagePercentage: number;
+    };
+} | {
+    type: "followupPrompt";
+    data: string;
+} | {
+    type: "usage";
+    data: {
+        inputTokens?: number;
+        outputTokens?: number;
+    };
+} | {
+    type: "error";
+    data: {
+        error: string;
+        message?: string;
+    };
+};
+export declare function parseKiroEvent(parsed: Record<string, unknown>): KiroStreamEvent | null;
+//# sourceMappingURL=event-parser.d.ts.map

package/dist/event-parser.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"event-parser.d.ts","sourceRoot":"","sources":["../src/event-parser.ts"],"names":[],"mappings":"AAGA,MAAM,MAAM,eAAe,GACxB;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACjC;IACA,IAAI,EAAE,SAAS,CAAC;IAChB,IAAI,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC;CACxE,GACD;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,IAAI,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GACjD;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,IAAI,EAAE;QAAE,IAAI,EAAE,OAAO,CAAA;KAAE,CAAA;CAAE,GAChD;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,IAAI,EAAE;QAAE,sBAAsB,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GAClE;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACxC;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,IAAI,EAAE;QAAE,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GACxE;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,IAAI,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,CAAC;AAEhE,wBAAgB,cAAc,CAC7B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC7B,eAAe,GAAG,IAAI,CAkExB"}

package/dist/event-parser.js

@@ -0,0 +1,66 @@
+// ABOUTME: Kiro stream event type definitions and JSON-to-typed-event mapping.
+// ABOUTME: Binary framing is handled by @smithy/core EventStreamMarshaller in stream.ts.
+export function parseKiroEvent(parsed) {
+    if (parsed.content !== undefined)
+        return { type: "content", data: parsed.content };
+    if (parsed.name && parsed.toolUseId) {
+        const input = typeof parsed.input === "string"
+            ? parsed.input
+            : parsed.input &&
+                typeof parsed.input === "object" &&
+                Object.keys(parsed.input).length > 0
+                ? JSON.stringify(parsed.input)
+                : "";
+        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 error = (parsed.error || parsed.Error || "unknown");
+        const message = (parsed.message || parsed.Message || parsed.reason);
+        return {
+            type: "error",
+            data: {
+                error: typeof error === "string" ? error : JSON.stringify(error),
+                message,
+            },
+        };
+    }
+    if (parsed.usage !== undefined) {
+        const u = parsed.usage;
+        return {
+            type: "usage",
+            data: {
+                inputTokens: u.inputTokens,
+                outputTokens: u.outputTokens,
+            },
+        };
+    }
+    return null;
+}
+//# sourceMappingURL=event-parser.js.map

package/dist/event-parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"event-parser.js","sourceRoot":"","sources":["../src/event-parser.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAC/E,yFAAyF;AAezF,MAAM,UAAU,cAAc,CAC7B,MAA+B;IAE/B,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS;QAC/B,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,CAAC,OAAiB,EAAE,CAAC;IAC5D,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QACrC,MAAM,KAAK,GACV,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;YAC/B,CAAC,CAAC,MAAM,CAAC,KAAK;YACd,CAAC,CAAC,MAAM,CAAC,KAAK;gBACZ,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;gBAChC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAgC,CAAC,CAAC,MAAM,GAAG,CAAC;gBAChE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC;gBAC9B,CAAC,CAAC,EAAE,CAAC;QACR,OAAO;YACN,IAAI,EAAE,SAAS;YACf,IAAI,EAAE;gBACL,IAAI,EAAE,MAAM,CAAC,IAAc;gBAC3B,SAAS,EAAE,MAAM,CAAC,SAAmB;gBACrC,KAAK;gBACL,IAAI,EAAE,MAAM,CAAC,IAA2B;aACxC;SACD,CAAC;IACH,CAAC;IACD,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAChD,OAAO;YACN,IAAI,EAAE,cAAc;YACpB,IAAI,EAAE;gBACL,KAAK,EACJ,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;oBAC/B,CAAC,CAAC,MAAM,CAAC,KAAK;oBACd,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC;aAChC;SACD,CAAC;IACH,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,IAAI,MAAM,CAAC,sBAAsB,KAAK,SAAS;QAC3E,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAe,EAAE,EAAE,CAAC;IACxE,IAAI,MAAM,CAAC,sBAAsB,KAAK,SAAS;QAC9C,OAAO;YACN,IAAI,EAAE,cAAc;YACpB,IAAI,EAAE,EAAE,sBAAsB,EAAE,MAAM,CAAC,sBAAgC,EAAE;SACzE,CAAC;IACH,IAAI,MAAM,CAAC,cAAc,KAAK,SAAS;QACtC,OAAO,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,CAAC,cAAwB,EAAE,CAAC;IAC1E,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAC9D,MAAM,KAAK,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,IAAI,SAAS,CAAW,CAAC;QACpE,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,CAEtD,CAAC;QACb,OAAO;YACN,IAAI,EAAE,OAAO;YACb,IAAI,EAAE;gBACL,KAAK,EAAE,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;gBAChE,OAAO;aACP;SACD,CAAC;IACH,CAAC;IACD,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAChC,MAAM,CAAC,GAAG,MAAM,CAAC,KAAgC,CAAC;QAClD,OAAO;YACN,IAAI,EAAE,OAAO;YACb,IAAI,EAAE;gBACL,WAAW,EAAE,CAAC,CAAC,WAAiC;gBAChD,YAAY,EAAE,CAAC,CAAC,YAAkC;aAClD;SACD,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC"}

package/dist/history.d.ts

@@ -0,0 +1,13 @@
+import type { KiroHistoryEntry, KiroToolSpec } from "./transform.js";
+export declare const HISTORY_LIMIT = 850000;
+/** The context window size (in tokens) that HISTORY_LIMIT was calibrated for. */
+export declare const HISTORY_LIMIT_CONTEXT_WINDOW = 200000;
+/** Remove images from history entries — they've already been processed by the
+ *  model in previous turns and re-sending them wastes context / causes 413s. */
+export declare function stripHistoryImages(history: KiroHistoryEntry[]): KiroHistoryEntry[];
+export declare function sanitizeHistory(history: KiroHistoryEntry[]): KiroHistoryEntry[];
+export declare function injectSyntheticToolCalls(history: KiroHistoryEntry[]): KiroHistoryEntry[];
+export declare function truncateHistory(history: KiroHistoryEntry[], limit: number): KiroHistoryEntry[];
+export declare function extractToolNamesFromHistory(history: KiroHistoryEntry[]): Set<string>;
+export declare function addPlaceholderTools(tools: KiroToolSpec[], history: KiroHistoryEntry[]): KiroToolSpec[];
+//# sourceMappingURL=history.d.ts.map

package/dist/history.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"history.d.ts","sourceRoot":"","sources":["../src/history.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAErE,eAAO,MAAM,aAAa,SAAS,CAAC;AACpC,iFAAiF;AACjF,eAAO,MAAM,4BAA4B,SAAS,CAAC;AAEnD;gFACgF;AAChF,wBAAgB,kBAAkB,CACjC,OAAO,EAAE,gBAAgB,EAAE,GACzB,gBAAgB,EAAE,CAMpB;AAED,wBAAgB,eAAe,CAC9B,OAAO,EAAE,gBAAgB,EAAE,GACzB,gBAAgB,EAAE,CAgCpB;AAED,wBAAgB,wBAAwB,CACvC,OAAO,EAAE,gBAAgB,EAAE,GACzB,gBAAgB,EAAE,CA8BpB;AAED,wBAAgB,eAAe,CAC9B,OAAO,EAAE,gBAAgB,EAAE,EAC3B,KAAK,EAAE,MAAM,GACX,gBAAgB,EAAE,CAWpB;AAED,wBAAgB,2BAA2B,CAC1C,OAAO,EAAE,gBAAgB,EAAE,GACzB,GAAG,CAAC,MAAM,CAAC,CAQb;AAED,wBAAgB,mBAAmB,CAClC,KAAK,EAAE,YAAY,EAAE,EACrB,OAAO,EAAE,gBAAgB,EAAE,GACzB,YAAY,EAAE,CAkBhB"}

package/dist/history.js

@@ -0,0 +1,121 @@
+// Feature 6: History Management
+export const HISTORY_LIMIT = 850000;
+/** The context window size (in tokens) that HISTORY_LIMIT was calibrated for. */
+export const HISTORY_LIMIT_CONTEXT_WINDOW = 200000;
+/** Remove images from history entries — they've already been processed by the
+ *  model in previous turns and re-sending them wastes context / causes 413s. */
+export function stripHistoryImages(history) {
+    return history.map((entry) => {
+        if (!entry.userInputMessage?.images)
+            return entry;
+        const { images, ...rest } = entry.userInputMessage;
+        return { ...entry, userInputMessage: { ...rest } };
+    });
+}
+export function sanitizeHistory(history) {
+    // Strip leading entries that would make the history invalid
+    while (history.length > 0 &&
+        (!history[0]?.userInputMessage ||
+            history[0].userInputMessage.userInputMessageContext?.toolResults))
+        history = history.slice(1);
+    const result = [];
+    for (let i = 0; i < history.length; i++) {
+        const m = history[i];
+        if (!m)
+            continue;
+        // Skip assistant messages with no content and no tool uses (e.g. from API errors)
+        if (m.assistantResponseMessage &&
+            !m.assistantResponseMessage.toolUses &&
+            !m.assistantResponseMessage.content)
+            continue;
+        if (m.assistantResponseMessage?.toolUses) {
+            const next = history[i + 1];
+            if (next?.userInputMessage?.userInputMessageContext?.toolResults)
+                result.push(m);
+        }
+        else if (m.userInputMessage?.userInputMessageContext?.toolResults) {
+            const prev = result[result.length - 1];
+            if (prev?.assistantResponseMessage?.toolUses)
+                result.push(m);
+        }
+        else {
+            result.push(m);
+        }
+    }
+    // Leading invalid entries already stripped above
+    return result;
+}
+export function injectSyntheticToolCalls(history) {
+    const validIds = new Set();
+    for (const entry of history) {
+        for (const tu of entry.assistantResponseMessage?.toolUses ?? []) {
+            if (tu.toolUseId)
+                validIds.add(tu.toolUseId);
+        }
+    }
+    const result = [];
+    for (const entry of history) {
+        const toolResults = entry.userInputMessage?.userInputMessageContext?.toolResults;
+        if (toolResults) {
+            const orphaned = toolResults.filter((tr) => !validIds.has(tr.toolUseId));
+            if (orphaned.length > 0) {
+                result.push({
+                    assistantResponseMessage: {
+                        content: "Tool calls were made.",
+                        toolUses: orphaned.map((tr) => ({
+                            name: "unknown_tool",
+                            toolUseId: tr.toolUseId,
+                            input: {},
+                        })),
+                    },
+                });
+                for (const tr of orphaned)
+                    validIds.add(tr.toolUseId);
+            }
+        }
+        result.push(entry);
+    }
+    return result;
+}
+export function truncateHistory(history, limit) {
+    let sanitized = sanitizeHistory(stripHistoryImages(history));
+    let historySize = JSON.stringify(sanitized).length;
+    while (historySize > limit && sanitized.length > 2) {
+        sanitized.shift();
+        while (sanitized.length > 0 && !sanitized[0]?.userInputMessage)
+            sanitized.shift();
+        sanitized = sanitizeHistory(sanitized);
+        historySize = JSON.stringify(sanitized).length;
+    }
+    return injectSyntheticToolCalls(sanitized);
+}
+export function extractToolNamesFromHistory(history) {
+    const names = new Set();
+    for (const entry of history) {
+        for (const tu of entry.assistantResponseMessage?.toolUses ?? []) {
+            if (tu.name)
+                names.add(tu.name);
+        }
+    }
+    return names;
+}
+export function addPlaceholderTools(tools, history) {
+    const historyNames = extractToolNamesFromHistory(history);
+    if (historyNames.size === 0)
+        return tools;
+    const existing = new Set(tools.map((t) => t.toolSpecification?.name).filter(Boolean));
+    const missing = Array.from(historyNames).filter((n) => !existing.has(n));
+    if (missing.length === 0)
+        return tools;
+    return [
+        ...tools,
+        ...missing.map((name) => ({
+            toolSpecification: {
+                name,
+                description: "Tool",
+                inputSchema: { json: { type: "object", properties: {} } },
+            },
+        })),
+    ];
+}
+//# sourceMappingURL=history.js.map

package/dist/history.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"history.js","sourceRoot":"","sources":["../src/history.ts"],"names":[],"mappings":"AAAA,gCAAgC;AAIhC,MAAM,CAAC,MAAM,aAAa,GAAG,MAAM,CAAC;AACpC,iFAAiF;AACjF,MAAM,CAAC,MAAM,4BAA4B,GAAG,MAAM,CAAC;AAEnD;gFACgF;AAChF,MAAM,UAAU,kBAAkB,CACjC,OAA2B;IAE3B,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QAC5B,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE,MAAM;YAAE,OAAO,KAAK,CAAC;QAClD,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,KAAK,CAAC,gBAAgB,CAAC;QACnD,OAAO,EAAE,GAAG,KAAK,EAAE,gBAAgB,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,eAAe,CAC9B,OAA2B;IAE3B,4DAA4D;IAC5D,OACC,OAAO,CAAC,MAAM,GAAG,CAAC;QAClB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,gBAAgB;YAC7B,OAAO,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,uBAAuB,EAAE,WAAW,CAAC;QAElE,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC5B,MAAM,MAAM,GAAuB,EAAE,CAAC;IACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACrB,IAAI,CAAC,CAAC;YAAE,SAAS;QACjB,kFAAkF;QAClF,IACC,CAAC,CAAC,wBAAwB;YAC1B,CAAC,CAAC,CAAC,wBAAwB,CAAC,QAAQ;YACpC,CAAC,CAAC,CAAC,wBAAwB,CAAC,OAAO;YAEnC,SAAS;QACV,IAAI,CAAC,CAAC,wBAAwB,EAAE,QAAQ,EAAE,CAAC;YAC1C,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC5B,IAAI,IAAI,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,WAAW;gBAC/D,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;aAAM,IAAI,CAAC,CAAC,gBAAgB,EAAE,uBAAuB,EAAE,WAAW,EAAE,CAAC;YACrE,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACvC,IAAI,IAAI,EAAE,wBAAwB,EAAE,QAAQ;gBAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9D,CAAC;aAAM,CAAC;YACP,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChB,CAAC;IACF,CAAC;IACD,iDAAiD;IACjD,OAAO,MAAM,CAAC;AACf,CAAC;AAED,MAAM,UAAU,wBAAwB,CACvC,OAA2B;IAE3B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IACnC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC7B,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,wBAAwB,EAAE,QAAQ,IAAI,EAAE,EAAE,CAAC;YACjE,IAAI,EAAE,CAAC,SAAS;gBAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC;IACF,CAAC;IACD,MAAM,MAAM,GAAuB,EAAE,CAAC;IACtC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,WAAW,GAChB,KAAK,CAAC,gBAAgB,EAAE,uBAAuB,EAAE,WAAW,CAAC;QAC9D,IAAI,WAAW,EAAE,CAAC;YACjB,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;YACzE,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,MAAM,CAAC,IAAI,CAAC;oBACX,wBAAwB,EAAE;wBACzB,OAAO,EAAE,uBAAuB;wBAChC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;4BAC/B,IAAI,EAAE,cAAc;4BACpB,SAAS,EAAE,EAAE,CAAC,SAAS;4BACvB,KAAK,EAAE,EAAE;yBACT,CAAC,CAAC;qBACH;iBACD,CAAC,CAAC;gBACH,KAAK,MAAM,EAAE,IAAI,QAAQ;oBAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;YACvD,CAAC;QACF,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpB,CAAC;IACD,OAAO,MAAM,CAAC;AACf,CAAC;AAED,MAAM,UAAU,eAAe,CAC9B,OAA2B,EAC3B,KAAa;IAEb,IAAI,SAAS,GAAG,eAAe,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC7D,IAAI,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;IACnD,OAAO,WAAW,GAAG,KAAK,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpD,SAAS,CAAC,KAAK,EAAE,CAAC;QAClB,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,gBAAgB;YAC7D,SAAS,CAAC,KAAK,EAAE,CAAC;QACnB,SAAS,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;QACvC,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;IAChD,CAAC;IACD,OAAO,wBAAwB,CAAC,SAAS,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,UAAU,2BAA2B,CAC1C,OAA2B;IAE3B,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;IAChC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC7B,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,wBAAwB,EAAE,QAAQ,IAAI,EAAE,EAAE,CAAC;YACjE,IAAI,EAAE,CAAC,IAAI;gBAAE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;IACF,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAED,MAAM,UAAU,mBAAmB,CAClC,KAAqB,EACrB,OAA2B;IAE3B,MAAM,YAAY,GAAG,2BAA2B,CAAC,OAAO,CAAC,CAAC;IAC1D,IAAI,YAAY,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAC1C,MAAM,QAAQ,GAAG,IAAI,GAAG,CACvB,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAC3D,CAAC;IACF,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACzE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACvC,OAAO;QACN,GAAG,KAAK;QACR,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACzB,iBAAiB,EAAE;gBAClB,IAAI;gBACJ,WAAW,EAAE,MAAM;gBACnB,WAAW,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAiB,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE;aAClE;SACD,CAAC,CAAC;KACH,CAAC;AACH,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,6 @@
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+export type { KiroStreamEvent } from "./event-parser.js";
+export { KIRO_MODEL_IDS, kiroModels, resolveApiRegion, resolveKiroModel, } from "./models.js";
+export { streamKiro } from "./stream.js";
+export default function (pi: ExtensionAPI): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AASpE,YAAY,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,EACN,cAAc,EACd,UAAU,EACV,gBAAgB,EAChB,gBAAgB,GAChB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,MAAM,CAAC,OAAO,WAAW,EAAE,EAAE,YAAY,QAgCxC"}

package/dist/index.js

@@ -0,0 +1,44 @@
+// Feature 1: Extension Registration
+//
+// Entry point that wires all features together via pi.registerProvider().
+import { getKiroCliCredentials } from "./kiro-cli.js";
+import { setExtensionContext } from "./login-ui.js";
+import { getCachedModels, kiroModels, resolveApiRegion } from "./models.js";
+import { loginKiro, refreshKiroToken } from "./oauth.js";
+import { streamKiro } from "./stream.js";
+import { fetchKiroUsage } from "./usage.js";
+export { KIRO_MODEL_IDS, kiroModels, resolveApiRegion, resolveKiroModel, } from "./models.js";
+export { streamKiro } from "./stream.js";
+export default function (pi) {
+    // Capture ctx for the custom TUI login component
+    pi.on("session_start", async (_event, ctx) => {
+        setExtensionContext(ctx);
+    });
+    pi.registerProvider("kiro", {
+        baseUrl: "https://q.us-east-1.amazonaws.com/generateAssistantResponse",
+        api: "kiro-api",
+        models: kiroModels,
+        oauth: {
+            // Name reflects all supported auth methods: AWS Builder ID, Google, GitHub
+            name: "Kiro (Builder ID / Google / GitHub)",
+            login: loginKiro,
+            refreshToken: refreshKiroToken,
+            getApiKey: (cred) => cred.access,
+            getCliCredentials: getKiroCliCredentials,
+            modifyModels: (models, cred) => {
+                const apiRegion = resolveApiRegion(cred.region);
+                const cachedKiro = getCachedModels(apiRegion);
+                const nonKiro = models.filter((m) => m.provider !== "kiro");
+                const modifiedKiro = cachedKiro.map((m) => ({
+                    ...m,
+                    baseUrl: `https://q.${apiRegion}.amazonaws.com/generateAssistantResponse`,
+                }));
+                return [...nonKiro, ...modifiedKiro];
+            },
+            fetchUsage: fetchKiroUsage,
+            // biome-ignore lint/suspicious/noExplicitAny: ProviderConfig.oauth doesn't include getCliCredentials but OAuthProviderInterface does
+        },
+        streamSimple: streamKiro,
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,oCAAoC;AACpC,EAAE;AACF,0EAA0E;AAI1E,OAAO,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAE5E,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAG5C,OAAO,EACN,cAAc,EACd,UAAU,EACV,gBAAgB,EAChB,gBAAgB,GAChB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,MAAM,CAAC,OAAO,WAAW,EAAgB;IACxC,iDAAiD;IACjD,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;QAC5C,mBAAmB,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,gBAAgB,CAAC,MAAM,EAAE;QAC3B,OAAO,EAAE,6DAA6D;QACtE,GAAG,EAAE,UAAU;QACf,MAAM,EAAE,UAAU;QAClB,KAAK,EAAE;YACN,2EAA2E;YAC3E,IAAI,EAAE,qCAAqC;YAC3C,KAAK,EAAE,SAAS;YAChB,YAAY,EAAE,gBAAgB;YAC9B,SAAS,EAAE,CAAC,IAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM;YAClD,iBAAiB,EAAE,qBAAqB;YACxC,YAAY,EAAE,CAAC,MAAoB,EAAE,IAAsB,EAAE,EAAE;gBAC9D,MAAM,SAAS,GAAG,gBAAgB,CAAE,IAAwB,CAAC,MAAM,CAAC,CAAC;gBACrE,MAAM,UAAU,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;gBAC9C,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAa,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC;gBACxE,MAAM,YAAY,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAa,EAAE,EAAE,CAAC,CAAC;oBACvD,GAAG,CAAC;oBACJ,OAAO,EAAE,aAAa,SAAS,0CAA0C;iBACzE,CAAC,CAAC,CAAC;gBAEJ,OAAO,CAAC,GAAG,OAAO,EAAE,GAAG,YAAY,CAAC,CAAC;YACtC,CAAC;YACD,UAAU,EAAE,cAAc;YAC1B,qIAAqI;SAC9H;QACR,YAAY,EAAE,UAAU;KACxB,CAAC,CAAC;AACJ,CAAC"}

package/dist/kiro-cli.d.ts

@@ -0,0 +1,32 @@
+import type { KiroAuthMethod, KiroCredentials } from "./oauth.js";
+export declare function getKiroCliDbPath(): string | undefined;
+export declare function getKiroCliCredentials(): KiroCredentials | undefined;
+/**
+ * Like getKiroCliCredentials but returns credentials even when the access token
+ * is expired, as long as a refresh token exists. Used to attempt a token refresh
+ * before falling back to the full device code login flow.
+ */
+export declare function getKiroCliCredentialsAllowExpired(): KiroCredentials | undefined;
+declare function tryKiroCliToken(dbPath: string, tokenKey: string, authMethod: KiroAuthMethod, allowExpired?: boolean): KiroCredentials | undefined;
+export { tryKiroCliToken };
+/**
+ * Get the social token (Google/GitHub) from kiro-cli if available.
+ * Returns undefined if no valid social token exists.
+ * This is used to prefer social login when the user has logged in that way.
+ */
+export declare function getKiroCliSocialToken(): KiroCredentials | undefined;
+/**
+ * Like getKiroCliSocialToken but returns credentials even when the access token
+ * is expired, as long as a refresh token exists.
+ */
+export declare function getKiroCliSocialTokenAllowExpired(): KiroCredentials | undefined;
+export declare function saveKiroCliCredentials(creds: KiroCredentials): void;
+/**
+ * Ask kiro-cli to refresh its own tokens via `kiro-cli debug refresh-auth-token`,
+ * then re-read the SQLite DB for fresh credentials.
+ *
+ * Returns refreshed credentials on success, or undefined if kiro-cli is not
+ * installed, the command fails, or the DB still has no valid tokens afterward.
+ */
+export declare function refreshViaKiroCli(): KiroCredentials | undefined;
+//# sourceMappingURL=kiro-cli.d.ts.map