waifu-code 1.0.1

package/README.md

@@ -0,0 +1,51 @@
+# waifu CLI
+
+A streamlined proxy and wrapper for Claude Code that transparently routes API requests through NVIDIA NIM, eliminating the need for complex Python proxy setups. 
+
+By default, it uses `moonshotai/kimi-k2-thinking` for intelligent responses and natively supports Anthropic streaming APIs.
+
+## Installation
+
+Install globally via npm:
+
+```bash
+npm install -g .
+```
+
+## Usage
+
+Simply run:
+
+```bash
+waifu
+```
+
+On first launch, it will prompt you for your NVIDIA NIM API key. This key is saved automatically to `~/.waifu/config.json`.
+
+`waifu` will immediately start the integrated TypeScript proxy in the background on a random available port and securely launch your locally-installed `claude-code` CLI natively. No manual configuration or `$env` modifications are required!
+
+### Options
+
+```bash
+Usage: waifu [options]
+
+Run the Claude Code CLI through the NVIDIA NIM proxy.
+
+Options:
+  -v, --version           Output the current version
+  -k, --nim-key <key>     NVIDIA NIM API key (saved automatically)
+  -m, --model <model>     Specific NIM model to use (default: moonshotai/kimi-k2-thinking)
+  -p, --port <port>       Port to run the proxy on (default: auto)
+  --proxy-only            Start only the proxy server without launching claude
+  --verbose               Enable verbose logging for debugging
+  -h, --help              Display help for command
+```
+
+## How It Works
+
+This tool is a drop-in replacement for the original Python proxy server. It relies on a hyper-efficient native NodeJS integration:
+
+1. Intercepts Anthropic Server-Sent Events (SSE).
+2. Converts Claude's messages format directly to NIM compatible JSON.
+3. Fixes Anthropic API quirks (like `?beta=true` queries and streaming header strictness) seamlessly underneath the hood.
+4. Auto-detects and preserves `<think>` tags returned by supported models without breaking the CLI UI experience.

package/dist/cli.d.ts

@@ -0,0 +1,15 @@
+#!/usr/bin/env node
+/**
+ * waifu — Free coding assistant CLI.
+ *
+ * Starts an embedded proxy server that translates Anthropic API
+ * requests to NVIDIA NIM, then launches claude-code CLI with
+ * the proxy configured automatically.
+ *
+ * Usage:
+ *   waifu                      # Use saved NIM key
+ *   waifu --nim-key nvapi-xxx  # Provide NIM key (saved for next time)
+ *   waifu config               # Show current configuration
+ *   waifu config --nim-key xxx # Save NIM key without starting
+ */
+export {};

package/dist/cli.js

@@ -0,0 +1,188 @@
+#!/usr/bin/env node
+/**
+ * waifu — Free coding assistant CLI.
+ *
+ * Starts an embedded proxy server that translates Anthropic API
+ * requests to NVIDIA NIM, then launches claude-code CLI with
+ * the proxy configured automatically.
+ *
+ * Usage:
+ *   waifu                      # Use saved NIM key
+ *   waifu --nim-key nvapi-xxx  # Provide NIM key (saved for next time)
+ *   waifu config               # Show current configuration
+ *   waifu config --nim-key xxx # Save NIM key without starting
+ */
+import { spawn } from "node:child_process";
+import { randomUUID } from "node:crypto";
+import { Command } from "commander";
+import { resolveNimApiKey, resolveModel, loadConfig, saveConfig, getConfigFile, } from "./config.js";
+import { startProxyServer, waitForHealth } from "./proxy/server.js";
+import { DEFAULT_MODEL } from "./providers/nim.js";
+import { EventDetector } from "./observer/eventDetector.js";
+import { OverlayRenderer } from "./waifu/overlayRenderer.js";
+const VERSION = "1.0.0";
+const program = new Command();
+program
+    .name("waifu")
+    .description("Free coding assistant — Claude Code powered by NVIDIA NIM")
+    .version(VERSION)
+    .enablePositionalOptions()
+    .passThroughOptions();
+// ── Main command (default) ──
+program
+    .option("--nim-key <key>", "NVIDIA NIM API key")
+    .option("--model <model>", `Model to use (default: ${DEFAULT_MODEL})`)
+    .option("--port <port>", "Proxy server port (default: auto)")
+    .option("--proxy-only", "Start proxy server only (don't launch claude)")
+    .option("--no-waifu", "Disable waifu overlay and sounds")
+    .option("--verbose", "Enable verbose logging")
+    .action(async (opts) => {
+    const nimKey = resolveNimApiKey(opts.nimKey);
+    const model = resolveModel(opts.model);
+    const verbose = opts.verbose ?? false;
+    if (!nimKey) {
+        console.error("\x1b[31m✗ No NVIDIA NIM API key found.\x1b[0m\n\n" +
+            "Provide your key in one of these ways:\n" +
+            "  waifu --nim-key nvapi-xxx\n" +
+            "  waifu config --nim-key nvapi-xxx\n" +
+            "  export NIM_API_KEY=nvapi-xxx\n\n" +
+            "Get a free key at: \x1b[36mhttps://build.nvidia.com/settings/api-keys\x1b[0m");
+        process.exit(1);
+    }
+    // Save the key for next time if provided via CLI
+    if (opts.nimKey) {
+        const config = loadConfig();
+        config.nimApiKey = opts.nimKey;
+        if (opts.model)
+            config.model = opts.model;
+        saveConfig(config);
+        if (verbose)
+            console.log(`[waifu] Config saved to ${getConfigFile()}`);
+    }
+    // Generate internal auth token
+    const authToken = randomUUID();
+    // Initialize waifu overlay if enabled
+    const useWaifu = opts.waifu !== false;
+    let detector;
+    if (useWaifu) {
+        detector = new EventDetector();
+        const renderer = new OverlayRenderer(true);
+        detector.on("event", (ev) => renderer.render(ev));
+    }
+    // Start proxy server
+    if (verbose)
+        console.log("[waifu] Starting proxy server...");
+    const proxy = await startProxyServer({
+        nimApiKey: nimKey,
+        model,
+        authToken,
+        port: opts.port ? parseInt(opts.port, 10) : undefined,
+        detector,
+    });
+    console.log(`\x1b[32m✓ Proxy started\x1b[0m on http://${proxy.host}:${proxy.port} → NIM (${model})`);
+    // Wait for health check
+    const healthy = await waitForHealth(proxy.host, proxy.port);
+    if (!healthy) {
+        console.error("\x1b[31m✗ Proxy health check failed\x1b[0m");
+        await proxy.stop();
+        process.exit(1);
+    }
+    if (opts.proxyOnly) {
+        console.log("\n\x1b[33mProxy-only mode.\x1b[0m Use these env vars with claude:\n" +
+            `  ANTHROPIC_AUTH_TOKEN=${authToken}\n` +
+            `  ANTHROPIC_BASE_URL=http://${proxy.host}:${proxy.port}\n`);
+        // Keep running until interrupted
+        process.on("SIGINT", async () => {
+            console.log("\n[waifu] Shutting down...");
+            await proxy.stop();
+            process.exit(0);
+        });
+        process.on("SIGTERM", async () => {
+            await proxy.stop();
+            process.exit(0);
+        });
+        return;
+    }
+    // Launch claude-code CLI
+    if (verbose)
+        console.log("[waifu] Launching claude...");
+    const claudeCmd = "claude";
+    const claudeArgs = process.argv.slice(2).filter((arg) => !arg.startsWith("--nim-key") &&
+        !arg.startsWith("--model") &&
+        !arg.startsWith("--port") &&
+        !arg.startsWith("--proxy-only") &&
+        !arg.startsWith("--verbose") &&
+        // Also filter out the values for key/model/port
+        arg !== opts.nimKey &&
+        arg !== opts.model &&
+        arg !== opts.port &&
+        arg !== "--no-waifu");
+    const child = spawn(claudeCmd, claudeArgs, {
+        stdio: "inherit",
+        env: {
+            ...process.env,
+            ANTHROPIC_AUTH_TOKEN: authToken,
+            ANTHROPIC_BASE_URL: `http://${proxy.host}:${proxy.port}`,
+            // Disable the update check (we're not the official CLI)
+            CLAUDE_CODE_SKIP_UPDATE_CHECK: "1",
+        },
+        shell: process.platform === "win32",
+    });
+    child.on("error", (err) => {
+        if (err.code === "ENOENT") {
+            console.error("\x1b[31m✗ 'claude' CLI not found.\x1b[0m\n\n" +
+                "Install it first:\n" +
+                "  npm install -g @anthropic-ai/claude-code\n");
+        }
+        else {
+            console.error(`\x1b[31m✗ Failed to start claude: ${err.message}\x1b[0m`);
+        }
+        proxy.stop().then(() => process.exit(1));
+    });
+    child.on("exit", async (code) => {
+        await proxy.stop();
+        process.exit(code ?? 0);
+    });
+    // Cleanup on signals
+    const cleanup = async () => {
+        child.kill();
+        await proxy.stop();
+        process.exit(0);
+    };
+    process.on("SIGINT", cleanup);
+    process.on("SIGTERM", cleanup);
+});
+// ── Config subcommand ──
+program
+    .command("config")
+    .description("View or update waifu configuration")
+    .option("--nim-key <key>", "Set NVIDIA NIM API key")
+    .option("--model <model>", "Set default model")
+    .option("--reset", "Reset all configuration")
+    .action((opts) => {
+    if (opts.reset) {
+        saveConfig({});
+        console.log("✓ Configuration reset.");
+        return;
+    }
+    if (opts.nimKey || opts.model) {
+        const config = loadConfig();
+        if (opts.nimKey)
+            config.nimApiKey = opts.nimKey;
+        if (opts.model)
+            config.model = opts.model;
+        saveConfig(config);
+        console.log(`✓ Configuration saved to ${getConfigFile()}`);
+        return;
+    }
+    // Show current config
+    const config = loadConfig();
+    const key = config.nimApiKey;
+    console.log("\n\x1b[1mwaifu configuration\x1b[0m");
+    console.log(`  Config file: ${getConfigFile()}`);
+    console.log(`  NIM API Key: ${key ? key.slice(0, 8) + "..." + key.slice(-4) : "\x1b[33m(not set)\x1b[0m"}`);
+    console.log(`  Model:       ${config.model ?? `${DEFAULT_MODEL} (default)`}`);
+    console.log("");
+});
+program.parse();
+//# sourceMappingURL=cli.js.map

package/dist/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";AAEA;;;;;;;;;;;;GAYG;AAEH,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EACL,gBAAgB,EAChB,YAAY,EACZ,UAAU,EACV,UAAU,EACV,aAAa,GACd,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAC5D,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAI7D,MAAM,OAAO,GAAG,OAAO,CAAC;AAExB,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAE9B,OAAO;KACJ,IAAI,CAAC,OAAO,CAAC;KACb,WAAW,CAAC,2DAA2D,CAAC;KACxE,OAAO,CAAC,OAAO,CAAC;KAChB,uBAAuB,EAAE;KACzB,kBAAkB,EAAE,CAAC;AAExB,+BAA+B;AAC/B,OAAO;KACJ,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC;KAC/C,MAAM,CAAC,iBAAiB,EAAE,0BAA0B,aAAa,GAAG,CAAC;KACrE,MAAM,CAAC,eAAe,EAAE,mCAAmC,CAAC;KAC5D,MAAM,CAAC,cAAc,EAAE,+CAA+C,CAAC;KACvE,MAAM,CAAC,YAAY,EAAE,kCAAkC,CAAC;KACxD,MAAM,CAAC,WAAW,EAAE,wBAAwB,CAAC;KAE7C,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;IACrB,MAAM,MAAM,GAAG,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACvC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,KAAK,CAAC;IAEtC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,KAAK,CACX,mDAAmD;YACnD,0CAA0C;YAC1C,+BAA+B;YAC/B,sCAAsC;YACtC,oCAAoC;YACpC,8EAA8E,CAC/E,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,iDAAiD;IACjD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAChB,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAC5B,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC;QAC/B,IAAI,IAAI,CAAC,KAAK;YAAE,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QAC1C,UAAU,CAAC,MAAM,CAAC,CAAC;QACnB,IAAI,OAAO;YAAE,OAAO,CAAC,GAAG,CAAC,2BAA2B,aAAa,EAAE,EAAE,CAAC,CAAC;IACzE,CAAC;IAED,+BAA+B;IAC/B,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;IAE/B,sCAAsC;IACtC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC;IACtC,IAAI,QAAmC,CAAC;IACxC,IAAI,QAAQ,EAAE,CAAC;QACb,QAAQ,GAAG,IAAI,aAAa,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC;QAC3C,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAS,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,qBAAqB;IACrB,IAAI,OAAO;QAAE,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAE7D,MAAM,KAAK,GAAG,MAAM,gBAAgB,CAAC;QACnC,SAAS,EAAE,MAAM;QACjB,KAAK;QACL,SAAS;QACT,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;QACrD,QAAQ;KACT,CAAC,CAAC;IAGH,OAAO,CAAC,GAAG,CACT,4CAA4C,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,WAAW,KAAK,GAAG,CACxF,CAAC;IAEF,wBAAwB;IACxB,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IAC5D,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAC5D,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,CACT,qEAAqE;YACrE,0BAA0B,SAAS,IAAI;YACvC,+BAA+B,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,IAAI,CAC5D,CAAC;QACF,iCAAiC;QACjC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;YAC9B,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;YAC1C,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;YAC/B,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,yBAAyB;IACzB,IAAI,OAAO;QAAE,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAExD,MAAM,SAAS,GAAG,QAAQ,CAAC;IAC3B,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAC7C,CAAC,GAAG,EAAE,EAAE,CACN,CAAC,GAAG,CAAC,UAAU,CAAC,WAAW,CAAC;QAC5B,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC;QAC1B,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC;QACzB,CAAC,GAAG,CAAC,UAAU,CAAC,cAAc,CAAC;QAC/B,CAAC,GAAG,CAAC,UAAU,CAAC,WAAW,CAAC;QAC5B,gDAAgD;QAChD,GAAG,KAAK,IAAI,CAAC,MAAM;QACnB,GAAG,KAAK,IAAI,CAAC,KAAK;QAClB,GAAG,KAAK,IAAI,CAAC,IAAI;QACjB,GAAG,KAAK,YAAY,CACvB,CAAC;IAEF,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,EAAE,UAAU,EAAE;QACzC,KAAK,EAAE,SAAS;QAChB,GAAG,EAAE;YACH,GAAG,OAAO,CAAC,GAAG;YACd,oBAAoB,EAAE,SAAS;YAC/B,kBAAkB,EAAE,UAAU,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,EAAE;YACxD,wDAAwD;YACxD,6BAA6B,EAAE,GAAG;SACnC;QACD,KAAK,EAAE,OAAO,CAAC,QAAQ,KAAK,OAAO;KACpC,CAAC,CAAC;IAIH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;QACxB,IAAK,GAA6B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACrD,OAAO,CAAC,KAAK,CACX,8CAA8C;gBAC9C,qBAAqB;gBACrB,8CAA8C,CAC/C,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,qCAAqC,GAAG,CAAC,OAAO,SAAS,CAAC,CAAC;QAC3E,CAAC;QACD,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;QAC9B,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,qBAAqB;IACrB,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;QACzB,KAAK,CAAC,IAAI,EAAE,CAAC;QACb,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC;IAEF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC9B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;AACjC,CAAC,CAAC,CAAC;AAEL,0BAA0B;AAC1B,OAAO;KACJ,OAAO,CAAC,QAAQ,CAAC;KACjB,WAAW,CAAC,oCAAoC,CAAC;KACjD,MAAM,CAAC,iBAAiB,EAAE,wBAAwB,CAAC;KACnD,MAAM,CAAC,iBAAiB,EAAE,mBAAmB,CAAC;KAC9C,MAAM,CAAC,SAAS,EAAE,yBAAyB,CAAC;KAC5C,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;IACf,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,UAAU,CAAC,EAAE,CAAC,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QACtC,OAAO;IACT,CAAC;IAED,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAC5B,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC;QAChD,IAAI,IAAI,CAAC,KAAK;YAAE,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QAC1C,UAAU,CAAC,MAAM,CAAC,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,4BAA4B,aAAa,EAAE,EAAE,CAAC,CAAC;QAC3D,OAAO;IACT,CAAC;IAED,sBAAsB;IACtB,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC;IAC7B,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,kBAAkB,aAAa,EAAE,EAAE,CAAC,CAAC;IACjD,OAAO,CAAC,GAAG,CACT,kBAAkB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,0BAA0B,EAAE,CAC/F,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,kBAAkB,MAAM,CAAC,KAAK,IAAI,GAAG,aAAa,YAAY,EAAE,CAAC,CAAC;IAC9E,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEL,OAAO,CAAC,KAAK,EAAE,CAAC"}

package/dist/config.d.ts

@@ -0,0 +1,35 @@
+/**
+ * Configuration management for waifu CLI.
+ *
+ * Stores NIM API key and settings in ~/.waifu/config.json.
+ * Cross-platform: uses os.homedir() for the config directory.
+ */
+export interface WaifuConfig {
+    nimApiKey?: string;
+    model?: string;
+}
+/** Load config from disk. Returns empty config if file doesn't exist. */
+export declare function loadConfig(): WaifuConfig;
+/** Save config to disk. Creates the config directory if needed. */
+export declare function saveConfig(config: WaifuConfig): void;
+/**
+ * Resolve the NIM API key from (in priority order):
+ *   1. CLI flag --nim-key
+ *   2. Environment variable NIM_API_KEY or NVIDIA_NIM_API_KEY
+ *   3. Persisted config file ~/.waifu/config.json
+ *
+ * Returns the key or null if not found.
+ */
+export declare function resolveNimApiKey(cliKey?: string): string | null;
+/**
+ * Resolve the model from (in priority order):
+ *   1. CLI flag --model
+ *   2. Environment variable WAIFU_MODEL
+ *   3. Persisted config file
+ *   4. Default: moonshotai/kimi-k2.5
+ */
+export declare function resolveModel(cliModel?: string): string;
+/** Get the config directory path. */
+export declare function getConfigDir(): string;
+/** Get the config file path. */
+export declare function getConfigFile(): string;

package/dist/config.js

@@ -0,0 +1,73 @@
+/**
+ * Configuration management for waifu CLI.
+ *
+ * Stores NIM API key and settings in ~/.waifu/config.json.
+ * Cross-platform: uses os.homedir() for the config directory.
+ */
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+const CONFIG_DIR = join(homedir(), ".waifu");
+const CONFIG_FILE = join(CONFIG_DIR, "config.json");
+/** Load config from disk. Returns empty config if file doesn't exist. */
+export function loadConfig() {
+    try {
+        if (existsSync(CONFIG_FILE)) {
+            const raw = readFileSync(CONFIG_FILE, "utf-8");
+            return JSON.parse(raw);
+        }
+    }
+    catch {
+        // Corrupted config — treat as empty
+    }
+    return {};
+}
+/** Save config to disk. Creates the config directory if needed. */
+export function saveConfig(config) {
+    if (!existsSync(CONFIG_DIR)) {
+        mkdirSync(CONFIG_DIR, { recursive: true });
+    }
+    writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2) + "\n", "utf-8");
+}
+/**
+ * Resolve the NIM API key from (in priority order):
+ *   1. CLI flag --nim-key
+ *   2. Environment variable NIM_API_KEY or NVIDIA_NIM_API_KEY
+ *   3. Persisted config file ~/.waifu/config.json
+ *
+ * Returns the key or null if not found.
+ */
+export function resolveNimApiKey(cliKey) {
+    if (cliKey)
+        return cliKey;
+    if (process.env.NIM_API_KEY)
+        return process.env.NIM_API_KEY;
+    if (process.env.NVIDIA_NIM_API_KEY)
+        return process.env.NVIDIA_NIM_API_KEY;
+    const config = loadConfig();
+    return config.nimApiKey ?? null;
+}
+/**
+ * Resolve the model from (in priority order):
+ *   1. CLI flag --model
+ *   2. Environment variable WAIFU_MODEL
+ *   3. Persisted config file
+ *   4. Default: moonshotai/kimi-k2.5
+ */
+export function resolveModel(cliModel) {
+    if (cliModel)
+        return cliModel;
+    if (process.env.WAIFU_MODEL)
+        return process.env.WAIFU_MODEL;
+    const config = loadConfig();
+    return config.model ?? "moonshotai/kimi-k2-thinking";
+}
+/** Get the config directory path. */
+export function getConfigDir() {
+    return CONFIG_DIR;
+}
+/** Get the config file path. */
+export function getConfigFile() {
+    return CONFIG_FILE;
+}
+//# sourceMappingURL=config.js.map

package/dist/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC7E,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAOjC,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,QAAQ,CAAC,CAAC;AAC7C,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;AAEpD,yEAAyE;AACzE,MAAM,UAAU,UAAU;IACxB,IAAI,CAAC;QACH,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YAC5B,MAAM,GAAG,GAAG,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YAC/C,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAgB,CAAC;QACxC,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,oCAAoC;IACtC,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,mEAAmE;AACnE,MAAM,UAAU,UAAU,CAAC,MAAmB;IAC5C,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAC5B,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7C,CAAC;IACD,aAAa,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;AAC9E,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,gBAAgB,CAAC,MAAe;IAC9C,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAC1B,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW;QAAE,OAAO,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;IAC5D,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB;QAAE,OAAO,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IAE1E,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,OAAO,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC;AAClC,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,YAAY,CAAC,QAAiB;IAC5C,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC9B,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW;QAAE,OAAO,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;IAE5D,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,OAAO,MAAM,CAAC,KAAK,IAAI,6BAA6B,CAAC;AACvD,CAAC;AAED,qCAAqC;AACrC,MAAM,UAAU,YAAY;IAC1B,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,gCAAgC;AAChC,MAAM,UAAU,aAAa;IAC3B,OAAO,WAAW,CAAC;AACrB,CAAC"}

package/dist/observer/eventDetector.d.ts

@@ -0,0 +1,16 @@
+import { EventEmitter } from "node:events";
+export type WaifuEvent = "thinking" | "permission" | "error" | "completion" | "idle";
+/**
+ * EventDetector now receives high-level events from the Proxy Server
+ * instead of raw terminal bytes.
+ */
+export declare class EventDetector extends EventEmitter {
+    private currentState;
+    constructor();
+    /** Called when the model starts sending reasoning (thinking) content */
+    onThinking(): void;
+    /** Called when the message stream is complete */
+    setCompletion(): void;
+    private updateState;
+    reset(): void;
+}

package/dist/observer/eventDetector.js

@@ -0,0 +1,30 @@
+import { EventEmitter } from "node:events";
+/**
+ * EventDetector now receives high-level events from the Proxy Server
+ * instead of raw terminal bytes.
+ */
+export class EventDetector extends EventEmitter {
+    currentState = "idle";
+    constructor() {
+        super();
+    }
+    /** Called when the model starts sending reasoning (thinking) content */
+    onThinking() {
+        this.updateState("thinking");
+    }
+    /** Called when the message stream is complete */
+    setCompletion() {
+        // Reset to idle after a short delay
+        setTimeout(() => this.updateState("idle"), 1000);
+    }
+    updateState(newState) {
+        if (this.currentState === newState)
+            return;
+        this.currentState = newState;
+        this.emit("event", newState);
+    }
+    reset() {
+        this.updateState("idle");
+    }
+}
+//# sourceMappingURL=eventDetector.js.map

package/dist/observer/eventDetector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"eventDetector.js","sourceRoot":"","sources":["../../src/observer/eventDetector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAI3C;;;GAGG;AACH,MAAM,OAAO,aAAc,SAAQ,YAAY;IACrC,YAAY,GAAe,MAAM,CAAC;IAE1C;QACE,KAAK,EAAE,CAAC;IACV,CAAC;IAED,wEAAwE;IACjE,UAAU;QACf,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAC/B,CAAC;IAED,iDAAiD;IAC1C,aAAa;QAClB,oCAAoC;QACpC,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;IACnD,CAAC;IAGO,WAAW,CAAC,QAAoB;QACtC,IAAI,IAAI,CAAC,YAAY,KAAK,QAAQ;YAAE,OAAO;QAC3C,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC;QAC7B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAEM,KAAK;QACV,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAC3B,CAAC;CACF"}

package/dist/observer/streamObserver.d.ts

@@ -0,0 +1,8 @@
+import { ChildProcess } from "node:child_process";
+import { EventDetector } from "./eventDetector.js";
+import { OverlayRenderer } from "../waifu/overlayRenderer.js";
+/**
+ * StreamObserver listens to a child process's stdout and routes chunks
+ * to terminal and our event detector / renderer for the waifu overlay.
+ */
+export declare function createStreamObserver(child: ChildProcess, detector: EventDetector, renderer: OverlayRenderer): void;

package/dist/observer/streamObserver.js

@@ -0,0 +1,24 @@
+/**
+ * StreamObserver listens to a child process's stdout and routes chunks
+ * to terminal and our event detector / renderer for the waifu overlay.
+ */
+export function createStreamObserver(child, detector, renderer) {
+    if (!child.stdout) {
+        throw new Error("Child process has no stdout to observe.");
+    }
+    child.stdout.on("data", (chunk) => {
+        const text = chunk.toString("utf8");
+        // 1. Immediately write to terminal (CRITICAL: DO NOT DELAY)
+        process.stdout.write(chunk);
+        // 2. Feed to detector for analysis
+        detector.onChunk(text);
+    });
+    child.on("exit", () => {
+        detector.setCompletion();
+    });
+    // Listen for detector events
+    detector.on("event", (event) => {
+        renderer.render(event);
+    });
+}
+//# sourceMappingURL=streamObserver.js.map

package/dist/observer/streamObserver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamObserver.js","sourceRoot":"","sources":["../../src/observer/streamObserver.ts"],"names":[],"mappings":"AAIA;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAClC,KAAmB,EACnB,QAAuB,EACvB,QAAyB;IAEzB,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;QACxC,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAEpC,4DAA4D;QAC5D,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAE5B,mCAAmC;QACnC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,QAAQ,CAAC,aAAa,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,6BAA6B;IAC7B,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAa,EAAE,EAAE;QACrC,QAAQ,CAAC,MAAM,CAAC,KAAY,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/providers/converter.d.ts

@@ -0,0 +1,18 @@
+/**
+ * Anthropic ↔ OpenAI message format converter.
+ *
+ * Converts Anthropic-style messages (with content blocks like text, thinking,
+ * tool_use, tool_result) to OpenAI chat completions format.
+ * Port of Python AnthropicToOpenAIConverter from the proxy server.
+ */
+/** Convert a list of Anthropic messages to OpenAI format. */
+export declare function convertMessages(messages: any[]): any[];
+/** Convert Anthropic tools to OpenAI format. */
+export declare function convertTools(tools: any[]): any[];
+/** Convert Anthropic system prompt to OpenAI format. */
+export declare function convertSystemPrompt(system: string | any[] | undefined | null): {
+    role: string;
+    content: string;
+} | null;
+/** Build the common parts of an OpenAI-format request body. */
+export declare function buildBaseRequestBody(requestData: any): any;

package/dist/providers/converter.js

@@ -0,0 +1,168 @@
+/**
+ * Anthropic ↔ OpenAI message format converter.
+ *
+ * Converts Anthropic-style messages (with content blocks like text, thinking,
+ * tool_use, tool_result) to OpenAI chat completions format.
+ * Port of Python AnthropicToOpenAIConverter from the proxy server.
+ */
+/* eslint-disable @typescript-eslint/no-explicit-any */
+function getBlockAttr(block, attr, defaultVal = undefined) {
+    if (block && typeof block === "object") {
+        return block[attr] ?? defaultVal;
+    }
+    return defaultVal;
+}
+function getBlockType(block) {
+    return getBlockAttr(block, "type");
+}
+/** Convert a list of Anthropic messages to OpenAI format. */
+export function convertMessages(messages) {
+    const result = [];
+    for (const msg of messages) {
+        const role = msg.role;
+        const content = msg.content;
+        if (typeof content === "string") {
+            result.push({ role, content });
+        }
+        else if (Array.isArray(content)) {
+            if (role === "assistant") {
+                result.push(...convertAssistantMessage(content));
+            }
+            else if (role === "user") {
+                result.push(...convertUserMessage(content));
+            }
+        }
+        else {
+            result.push({ role, content: String(content) });
+        }
+    }
+    return result;
+}
+function convertAssistantMessage(content) {
+    const contentParts = [];
+    const toolCalls = [];
+    for (const block of content) {
+        const blockType = getBlockType(block);
+        if (blockType === "text") {
+            contentParts.push(getBlockAttr(block, "text", ""));
+        }
+        else if (blockType === "thinking") {
+            const thinking = getBlockAttr(block, "thinking", "");
+            contentParts.push(`<think>\n${thinking}\n</think>`);
+        }
+        else if (blockType === "tool_use") {
+            const toolInput = getBlockAttr(block, "input", {});
+            toolCalls.push({
+                id: getBlockAttr(block, "id"),
+                type: "function",
+                function: {
+                    name: getBlockAttr(block, "name"),
+                    arguments: typeof toolInput === "object"
+                        ? JSON.stringify(toolInput)
+                        : String(toolInput),
+                },
+            });
+        }
+    }
+    let contentStr = contentParts.join("\n\n");
+    // Ensure content is never empty for assistant messages
+    // NIM requires non-empty content if there are no tool calls
+    if (!contentStr && toolCalls.length === 0) {
+        contentStr = " ";
+    }
+    const msg = { role: "assistant", content: contentStr };
+    if (toolCalls.length > 0) {
+        msg.tool_calls = toolCalls;
+    }
+    return [msg];
+}
+function convertUserMessage(content) {
+    const result = [];
+    const textParts = [];
+    const flushText = () => {
+        if (textParts.length > 0) {
+            result.push({ role: "user", content: textParts.join("\n") });
+            textParts.length = 0;
+        }
+    };
+    for (const block of content) {
+        const blockType = getBlockType(block);
+        if (blockType === "text") {
+            textParts.push(getBlockAttr(block, "text", ""));
+        }
+        else if (blockType === "tool_result") {
+            flushText();
+            let toolContent = getBlockAttr(block, "content", "");
+            if (Array.isArray(toolContent)) {
+                toolContent = toolContent
+                    .map((item) => typeof item === "object" ? item.text ?? String(item) : String(item))
+                    .join("\n");
+            }
+            result.push({
+                role: "tool",
+                tool_call_id: getBlockAttr(block, "tool_use_id"),
+                content: toolContent ? String(toolContent) : "",
+            });
+        }
+    }
+    flushText();
+    return result;
+}
+/** Convert Anthropic tools to OpenAI format. */
+export function convertTools(tools) {
+    return tools.map((tool) => ({
+        type: "function",
+        function: {
+            name: tool.name,
+            description: tool.description || "",
+            parameters: tool.input_schema,
+        },
+    }));
+}
+/** Convert Anthropic system prompt to OpenAI format. */
+export function convertSystemPrompt(system) {
+    if (typeof system === "string") {
+        return { role: "system", content: system };
+    }
+    if (Array.isArray(system)) {
+        const textParts = system
+            .filter((block) => getBlockType(block) === "text")
+            .map((block) => getBlockAttr(block, "text", ""));
+        if (textParts.length > 0) {
+            return { role: "system", content: textParts.join("\n\n").trim() };
+        }
+    }
+    return null;
+}
+/** Build the common parts of an OpenAI-format request body. */
+export function buildBaseRequestBody(requestData) {
+    const messages = convertMessages(requestData.messages);
+    const system = requestData.system;
+    if (system) {
+        const systemMsg = convertSystemPrompt(system);
+        if (systemMsg) {
+            messages.unshift(systemMsg);
+        }
+    }
+    const body = { model: requestData.model, messages };
+    if (requestData.max_tokens != null) {
+        body.max_tokens = requestData.max_tokens;
+    }
+    if (requestData.temperature != null) {
+        body.temperature = requestData.temperature;
+    }
+    if (requestData.top_p != null) {
+        body.top_p = requestData.top_p;
+    }
+    if (requestData.stop_sequences) {
+        body.stop = requestData.stop_sequences;
+    }
+    if (requestData.tools && requestData.tools.length > 0) {
+        body.tools = convertTools(requestData.tools);
+    }
+    if (requestData.tool_choice) {
+        body.tool_choice = requestData.tool_choice;
+    }
+    return body;
+}
+//# sourceMappingURL=converter.js.map