@mariozechner/pi 0.1.5 → 0.5.0

package/dist/commands/pods.d.ts

@@ -0,0 +1,21 @@
+/**
+ * List all pods
+ */
+export declare const listPods: () => void;
+/**
+ * Setup a new pod
+ */
+export declare const setupPod: (name: string, sshCmd: string, options: {
+    mount?: string;
+    modelsPath?: string;
+    vllm?: "release" | "nightly" | "gpt-oss";
+}) => Promise<void>;
+/**
+ * Switch active pod
+ */
+export declare const switchActivePod: (name: string) => void;
+/**
+ * Remove a pod from config
+ */
+export declare const removePodCommand: (name: string) => void;
+//# sourceMappingURL=pods.d.ts.map

package/dist/commands/pods.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pods.d.ts","sourceRoot":"","sources":["../../src/commands/pods.ts"],"names":[],"mappings":"AAUA;;GAEG;AACH,eAAO,MAAM,QAAQ,YAyBpB,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,QAAQ,GACpB,MAAM,MAAM,EACZ,QAAQ,MAAM,EACd,SAAS;IAAE,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS,CAAA;CAAE,kBA6H1F,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,eAAe,GAAI,MAAM,MAAM,SAa3C,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,gBAAgB,GAAI,MAAM,MAAM,SAU5C,CAAC"}

package/dist/commands/pods.js

@@ -0,0 +1,175 @@
+import chalk from "chalk";
+import { dirname, join } from "path";
+import { fileURLToPath } from "url";
+import { addPod, loadConfig, removePod, setActivePod } from "../config.js";
+import { scpFile, sshExec, sshExecStream } from "../ssh.js";
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+/**
+ * List all pods
+ */
+export const listPods = () => {
+    const config = loadConfig();
+    const podNames = Object.keys(config.pods);
+    if (podNames.length === 0) {
+        console.log("No pods configured. Use 'pi pods setup' to add a pod.");
+        return;
+    }
+    console.log("Configured pods:");
+    for (const name of podNames) {
+        const pod = config.pods[name];
+        const isActive = config.active === name;
+        const marker = isActive ? chalk.green("*") : " ";
+        const gpuCount = pod.gpus?.length || 0;
+        const gpuInfo = gpuCount > 0 ? `${gpuCount}x ${pod.gpus[0].name}` : "no GPUs detected";
+        const vllmInfo = pod.vllmVersion ? ` (vLLM: ${pod.vllmVersion})` : "";
+        console.log(`${marker} ${chalk.bold(name)} - ${gpuInfo}${vllmInfo} - ${pod.ssh}`);
+        if (pod.modelsPath) {
+            console.log(`    Models: ${pod.modelsPath}`);
+        }
+        if (pod.vllmVersion === "gpt-oss") {
+            console.log(chalk.yellow(`    ⚠️  GPT-OSS build - only for GPT-OSS models`));
+        }
+    }
+};
+/**
+ * Setup a new pod
+ */
+export const setupPod = async (name, sshCmd, options) => {
+    // Validate environment variables
+    const hfToken = process.env.HF_TOKEN;
+    const vllmApiKey = process.env.PI_API_KEY;
+    if (!hfToken) {
+        console.error(chalk.red("ERROR: HF_TOKEN environment variable is required"));
+        console.error("Get a token from: https://huggingface.co/settings/tokens");
+        console.error("Then run: export HF_TOKEN=your_token_here");
+        process.exit(1);
+    }
+    if (!vllmApiKey) {
+        console.error(chalk.red("ERROR: PI_API_KEY environment variable is required"));
+        console.error("Set an API key: export PI_API_KEY=your_api_key_here");
+        process.exit(1);
+    }
+    // Determine models path
+    let modelsPath = options.modelsPath;
+    if (!modelsPath && options.mount) {
+        // Extract path from mount command if not explicitly provided
+        // e.g., "mount -t nfs ... /mnt/sfs" -> "/mnt/sfs"
+        const parts = options.mount.split(" ");
+        modelsPath = parts[parts.length - 1];
+    }
+    if (!modelsPath) {
+        console.error(chalk.red("ERROR: --models-path is required (or must be extractable from --mount)"));
+        process.exit(1);
+    }
+    console.log(chalk.green(`Setting up pod '${name}'...`));
+    console.log(`SSH: ${sshCmd}`);
+    console.log(`Models path: ${modelsPath}`);
+    console.log(`vLLM version: ${options.vllm || "release"} ${options.vllm === "gpt-oss" ? chalk.yellow("(GPT-OSS special build)") : ""}`);
+    if (options.mount) {
+        console.log(`Mount command: ${options.mount}`);
+    }
+    console.log("");
+    // Test SSH connection
+    console.log("Testing SSH connection...");
+    const testResult = await sshExec(sshCmd, "echo 'SSH OK'");
+    if (testResult.exitCode !== 0) {
+        console.error(chalk.red("Failed to connect via SSH"));
+        console.error(testResult.stderr);
+        process.exit(1);
+    }
+    console.log(chalk.green("✓ SSH connection successful"));
+    // Copy setup script
+    console.log("Copying setup script...");
+    const scriptPath = join(__dirname, "../../scripts/pod_setup.sh");
+    const success = await scpFile(sshCmd, scriptPath, "/tmp/pod_setup.sh");
+    if (!success) {
+        console.error(chalk.red("Failed to copy setup script"));
+        process.exit(1);
+    }
+    console.log(chalk.green("✓ Setup script copied"));
+    // Build setup command
+    let setupCmd = `bash /tmp/pod_setup.sh --models-path '${modelsPath}' --hf-token '${hfToken}' --vllm-api-key '${vllmApiKey}'`;
+    if (options.mount) {
+        setupCmd += ` --mount '${options.mount}'`;
+    }
+    // Add vLLM version flag
+    const vllmVersion = options.vllm || "release";
+    setupCmd += ` --vllm '${vllmVersion}'`;
+    // Run setup script
+    console.log("");
+    console.log(chalk.yellow("Running setup (this will take 2-5 minutes)..."));
+    console.log("");
+    // Use forceTTY to preserve colors from apt, pip, etc.
+    const exitCode = await sshExecStream(sshCmd, setupCmd, { forceTTY: true });
+    if (exitCode !== 0) {
+        console.error(chalk.red("\nSetup failed. Check the output above for errors."));
+        process.exit(1);
+    }
+    // Parse GPU info from setup output
+    console.log("");
+    console.log("Detecting GPU configuration...");
+    const gpuResult = await sshExec(sshCmd, "nvidia-smi --query-gpu=index,name,memory.total --format=csv,noheader");
+    const gpus = [];
+    if (gpuResult.exitCode === 0 && gpuResult.stdout) {
+        const lines = gpuResult.stdout.trim().split("\n");
+        for (const line of lines) {
+            const [id, name, memory] = line.split(",").map((s) => s.trim());
+            if (id !== undefined) {
+                gpus.push({
+                    id: parseInt(id),
+                    name: name || "Unknown",
+                    memory: memory || "Unknown",
+                });
+            }
+        }
+    }
+    console.log(chalk.green(`✓ Detected ${gpus.length} GPU(s)`));
+    for (const gpu of gpus) {
+        console.log(`  GPU ${gpu.id}: ${gpu.name} (${gpu.memory})`);
+    }
+    // Save pod configuration
+    const pod = {
+        ssh: sshCmd,
+        gpus,
+        models: {},
+        modelsPath,
+        vllmVersion: options.vllm || "release",
+    };
+    addPod(name, pod);
+    console.log("");
+    console.log(chalk.green(`✓ Pod '${name}' setup complete and set as active pod`));
+    console.log("");
+    console.log("You can now deploy models with:");
+    console.log(chalk.cyan(`  pi start <model> --name <name>`));
+};
+/**
+ * Switch active pod
+ */
+export const switchActivePod = (name) => {
+    const config = loadConfig();
+    if (!config.pods[name]) {
+        console.error(chalk.red(`Pod '${name}' not found`));
+        console.log("\nAvailable pods:");
+        for (const podName of Object.keys(config.pods)) {
+            console.log(`  ${podName}`);
+        }
+        process.exit(1);
+    }
+    setActivePod(name);
+    console.log(chalk.green(`✓ Switched active pod to '${name}'`));
+};
+/**
+ * Remove a pod from config
+ */
+export const removePodCommand = (name) => {
+    const config = loadConfig();
+    if (!config.pods[name]) {
+        console.error(chalk.red(`Pod '${name}' not found`));
+        process.exit(1);
+    }
+    removePod(name);
+    console.log(chalk.green(`✓ Removed pod '${name}' from configuration`));
+    console.log(chalk.yellow("Note: This only removes the local configuration. The remote pod is not affected."));
+};
+//# sourceMappingURL=pods.js.map

package/dist/commands/pods.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pods.js","sourceRoot":"","sources":["../../src/commands/pods.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC3E,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAG5D,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAEtC;;GAEG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC5B,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAE1C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;QACrE,OAAO;IACR,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;IAChC,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,KAAK,IAAI,CAAC;QACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;QACjD,MAAM,QAAQ,GAAG,GAAG,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC,CAAC;QACvC,MAAM,OAAO,GAAG,QAAQ,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,KAAK,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,kBAAkB,CAAC;QACvF,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,GAAG,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QACtE,OAAO,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,OAAO,GAAG,QAAQ,MAAM,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;QAClF,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,iDAAiD,CAAC,CAAC,CAAC;QAC9E,CAAC;IACF,CAAC;AACF,CAAC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,KAAK,EAC5B,IAAY,EACZ,MAAc,EACd,OAA0F,EACzF,EAAE;IACH,iCAAiC;IACjC,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;IACrC,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC;IAE1C,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC,CAAC;QAC7E,OAAO,CAAC,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAC1E,OAAO,CAAC,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC3D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IAED,IAAI,CAAC,UAAU,EAAE,CAAC;QACjB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC,CAAC;QAC/E,OAAO,CAAC,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACrE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IAED,wBAAwB;IACxB,IAAI,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IACpC,IAAI,CAAC,UAAU,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QAClC,6DAA6D;QAC7D,kDAAkD;QAClD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACvC,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACtC,CAAC;IAED,IAAI,CAAC,UAAU,EAAE,CAAC;QACjB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,wEAAwE,CAAC,CAAC,CAAC;QACnG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,mBAAmB,IAAI,MAAM,CAAC,CAAC,CAAC;IACxD,OAAO,CAAC,GAAG,CAAC,QAAQ,MAAM,EAAE,CAAC,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,gBAAgB,UAAU,EAAE,CAAC,CAAC;IAC1C,OAAO,CAAC,GAAG,CACV,iBAAiB,OAAO,CAAC,IAAI,IAAI,SAAS,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CACzH,CAAC;IACF,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,kBAAkB,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;IAChD,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAEhB,sBAAsB;IACtB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;IAC1D,IAAI,UAAU,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC,CAAC;QACtD,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC,CAAC;IAExD,oBAAoB;IACpB,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;IACvC,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,EAAE,4BAA4B,CAAC,CAAC;IACjE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,MAAM,EAAE,UAAU,EAAE,mBAAmB,CAAC,CAAC;IACvE,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC,CAAC;QACxD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;IAElD,sBAAsB;IACtB,IAAI,QAAQ,GAAG,yCAAyC,UAAU,iBAAiB,OAAO,qBAAqB,UAAU,GAAG,CAAC;IAC7H,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACnB,QAAQ,IAAI,aAAa,OAAO,CAAC,KAAK,GAAG,CAAC;IAC3C,CAAC;IACD,wBAAwB;IACxB,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,IAAI,SAAS,CAAC;IAC9C,QAAQ,IAAI,YAAY,WAAW,GAAG,CAAC;IAEvC,mBAAmB;IACnB,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAChB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,+CAA+C,CAAC,CAAC,CAAC;IAC3E,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAEhB,sDAAsD;IACtD,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3E,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;QACpB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC,CAAC;QAC/E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IAED,mCAAmC;IACnC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAChB,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;IAC9C,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,MAAM,EAAE,sEAAsE,CAAC,CAAC;IAEhH,MAAM,IAAI,GAAU,EAAE,CAAC;IACvB,IAAI,SAAS,CAAC,QAAQ,KAAK,CAAC,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;QAClD,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,MAAM,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YAChE,IAAI,EAAE,KAAK,SAAS,EAAE,CAAC;gBACtB,IAAI,CAAC,IAAI,CAAC;oBACT,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC;oBAChB,IAAI,EAAE,IAAI,IAAI,SAAS;oBACvB,MAAM,EAAE,MAAM,IAAI,SAAS;iBAC3B,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,IAAI,CAAC,MAAM,SAAS,CAAC,CAAC,CAAC;IAC7D,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,SAAS,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC;IAC7D,CAAC;IAED,yBAAyB;IACzB,MAAM,GAAG,GAAQ;QAChB,GAAG,EAAE,MAAM;QACX,IAAI;QACJ,MAAM,EAAE,EAAE;QACV,UAAU;QACV,WAAW,EAAE,OAAO,CAAC,IAAI,IAAI,SAAS;KACtC,CAAC;IAEF,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAClB,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAChB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,IAAI,wCAAwC,CAAC,CAAC,CAAC;IACjF,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAChB,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IAC/C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC,CAAC;AAC7D,CAAC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,IAAY,EAAE,EAAE;IAC/C,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,IAAI,aAAa,CAAC,CAAC,CAAC;QACpD,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACjC,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAChD,OAAO,CAAC,GAAG,CAAC,KAAK,OAAO,EAAE,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IAED,YAAY,CAAC,IAAI,CAAC,CAAC;IACnB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,6BAA6B,IAAI,GAAG,CAAC,CAAC,CAAC;AAChE,CAAC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,IAAY,EAAE,EAAE;IAChD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,IAAI,aAAa,CAAC,CAAC,CAAC;QACpD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IAED,SAAS,CAAC,IAAI,CAAC,CAAC;IAChB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,kBAAkB,IAAI,sBAAsB,CAAC,CAAC,CAAC;IACvE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,kFAAkF,CAAC,CAAC,CAAC;AAC/G,CAAC,CAAC"}

package/dist/commands/prompt.d.ts

@@ -0,0 +1,7 @@
+interface PromptOptions {
+    pod?: string;
+    apiKey?: string;
+}
+export declare function promptModel(modelName: string, userArgs: string[], opts?: PromptOptions): Promise<void>;
+export {};
+//# sourceMappingURL=prompt.d.ts.map

package/dist/commands/prompt.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.d.ts","sourceRoot":"","sources":["../../src/commands/prompt.ts"],"names":[],"mappings":"AAQA,UAAU,aAAa;IACtB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;CAChB;AAMD,wBAAsB,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,IAAI,GAAE,aAAkB,iBAmEhG"}

package/dist/commands/prompt.js

@@ -0,0 +1,55 @@
+import { main as agentMain } from "@mariozechner/pi-agent";
+import chalk from "chalk";
+import { getActivePod, loadConfig } from "../config.js";
+// ────────────────────────────────────────────────────────────────────────────────
+// Main prompt function
+// ────────────────────────────────────────────────────────────────────────────────
+export async function promptModel(modelName, userArgs, opts = {}) {
+    // Get pod and model configuration
+    const activePod = opts.pod ? { name: opts.pod, pod: loadConfig().pods[opts.pod] } : getActivePod();
+    if (!activePod) {
+        console.error(chalk.red("No active pod. Use 'pi pods active <name>' to set one."));
+        process.exit(1);
+    }
+    const { name: podName, pod } = activePod;
+    const modelConfig = pod.models[modelName];
+    if (!modelConfig) {
+        console.error(chalk.red(`Model '${modelName}' not found on pod '${podName}'`));
+        process.exit(1);
+    }
+    // Extract host from SSH string
+    const host = pod.ssh
+        .split(" ")
+        .find((p) => p.includes("@"))
+        ?.split("@")[1] ?? "localhost";
+    // Build the system prompt for code navigation
+    const systemPrompt = `You help the user understand and navigate the codebase in the current working directory.
+
+You can read files, list directories, and execute shell commands via the respective tools.
+
+Do not output file contents you read via the read_file tool directly, unless asked to.
+
+Do not output markdown tables as part of your responses.
+
+Keep your responses concise and relevant to the user's request.
+
+File paths you output must include line numbers where possible, e.g. "src/index.ts:10-20" for lines 10 to 20 in src/index.ts.
+
+Current working directory: ${process.cwd()}`;
+    // Build arguments for agent main function
+    const args = [];
+    // Add base configuration that we control
+    args.push("--base-url", `http://${host}:${modelConfig.port}/v1`, "--model", modelConfig.model, "--api-key", opts.apiKey || process.env.PI_API_KEY || "dummy", "--api", modelConfig.model.toLowerCase().includes("gpt-oss") ? "responses" : "completions", "--system-prompt", systemPrompt);
+    // Pass through all user-provided arguments
+    // This includes messages, --continue, --json, etc.
+    args.push(...userArgs);
+    // Call agent main function directly
+    try {
+        await agentMain(args);
+    }
+    catch (err) {
+        console.error(chalk.red(`Agent error: ${err.message}`));
+        process.exit(1);
+    }
+}
+//# sourceMappingURL=prompt.js.map

package/dist/commands/prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../src/commands/prompt.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,IAAI,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAWxD,mFAAmF;AACnF,uBAAuB;AACvB,mFAAmF;AAEnF,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,SAAiB,EAAE,QAAkB,EAAE,OAAsB,EAAE;IAChG,kCAAkC;IAClC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC;IAEnG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,wDAAwD,CAAC,CAAC,CAAC;QACnF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IAED,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,SAAS,CAAC;IACzC,MAAM,WAAW,GAAG,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAE1C,IAAI,CAAC,WAAW,EAAE,CAAC;QAClB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,SAAS,uBAAuB,OAAO,GAAG,CAAC,CAAC,CAAC;QAC/E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IAED,+BAA+B;IAC/B,MAAM,IAAI,GACT,GAAG,CAAC,GAAG;SACL,KAAK,CAAC,GAAG,CAAC;SACV,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QAC7B,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC;IAEjC,8CAA8C;IAC9C,MAAM,YAAY,GAAG;;;;;;;;;;;;6BAYO,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC;IAE5C,0CAA0C;IAC1C,MAAM,IAAI,GAAa,EAAE,CAAC;IAE1B,yCAAyC;IACzC,IAAI,CAAC,IAAI,CACR,YAAY,EACZ,UAAU,IAAI,IAAI,WAAW,CAAC,IAAI,KAAK,EACvC,SAAS,EACT,WAAW,CAAC,KAAK,EACjB,WAAW,EACX,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,OAAO,EAChD,OAAO,EACP,WAAW,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,EACjF,iBAAiB,EACjB,YAAY,CACZ,CAAC;IAEF,2CAA2C;IAC3C,mDAAmD;IACnD,IAAI,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;IAEvB,oCAAoC;IACpC,IAAI,CAAC;QACJ,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QACnB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,gBAAgB,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QACxD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;AACF,CAAC"}

package/dist/config.d.ts

@@ -0,0 +1,11 @@
+import type { Config, Pod } from "./types.js";
+export declare const loadConfig: () => Config;
+export declare const saveConfig: (config: Config) => void;
+export declare const getActivePod: () => {
+    name: string;
+    pod: Pod;
+} | null;
+export declare const addPod: (name: string, pod: Pod) => void;
+export declare const removePod: (name: string) => void;
+export declare const setActivePod: (name: string) => void;
+//# sourceMappingURL=config.d.ts.map

package/dist/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,YAAY,CAAC;AAe9C,eAAO,MAAM,UAAU,QAAO,MAa7B,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,QAAQ,MAAM,KAAG,IAQ3C,CAAC;AAEF,eAAO,MAAM,YAAY,QAAO;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,GAAG,CAAA;CAAE,GAAG,IAM5D,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,MAAM,MAAM,EAAE,KAAK,GAAG,KAAG,IAQ/C,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,MAAM,MAAM,KAAG,IAQxC,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,MAAM,MAAM,KAAG,IAQ3C,CAAC"}

package/dist/config.js

@@ -0,0 +1,74 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "fs";
+import { homedir } from "os";
+import { join } from "path";
+// Get config directory from env or use default
+const getConfigDir = () => {
+    const configDir = process.env.PI_CONFIG_DIR || join(homedir(), ".pi");
+    if (!existsSync(configDir)) {
+        mkdirSync(configDir, { recursive: true });
+    }
+    return configDir;
+};
+const getConfigPath = () => {
+    return join(getConfigDir(), "pods.json");
+};
+export const loadConfig = () => {
+    const configPath = getConfigPath();
+    if (!existsSync(configPath)) {
+        // Return empty config if file doesn't exist
+        return { pods: {} };
+    }
+    try {
+        const data = readFileSync(configPath, "utf-8");
+        return JSON.parse(data);
+    }
+    catch (e) {
+        console.error(`Error reading config: ${e}`);
+        return { pods: {} };
+    }
+};
+export const saveConfig = (config) => {
+    const configPath = getConfigPath();
+    try {
+        writeFileSync(configPath, JSON.stringify(config, null, 2));
+    }
+    catch (e) {
+        console.error(`Error saving config: ${e}`);
+        process.exit(1);
+    }
+};
+export const getActivePod = () => {
+    const config = loadConfig();
+    if (!config.active || !config.pods[config.active]) {
+        return null;
+    }
+    return { name: config.active, pod: config.pods[config.active] };
+};
+export const addPod = (name, pod) => {
+    const config = loadConfig();
+    config.pods[name] = pod;
+    // If no active pod, make this one active
+    if (!config.active) {
+        config.active = name;
+    }
+    saveConfig(config);
+};
+export const removePod = (name) => {
+    const config = loadConfig();
+    delete config.pods[name];
+    // If this was the active pod, clear active
+    if (config.active === name) {
+        config.active = undefined;
+    }
+    saveConfig(config);
+};
+export const setActivePod = (name) => {
+    const config = loadConfig();
+    if (!config.pods[name]) {
+        console.error(`Pod '${name}' not found`);
+        process.exit(1);
+    }
+    config.active = name;
+    saveConfig(config);
+};
+//# 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,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxE,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAC7B,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAG5B,+CAA+C;AAC/C,MAAM,YAAY,GAAG,GAAW,EAAE;IACjC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAC;IACtE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC5B,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3C,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,GAAW,EAAE;IAClC,OAAO,IAAI,CAAC,YAAY,EAAE,EAAE,WAAW,CAAC,CAAC;AAC1C,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,UAAU,GAAG,GAAW,EAAE;IACtC,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAC7B,4CAA4C;QAC5C,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;IACrB,CAAC;IACD,IAAI,CAAC;QACJ,MAAM,IAAI,GAAG,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACZ,OAAO,CAAC,KAAK,CAAC,yBAAyB,CAAC,EAAE,CAAC,CAAC;QAC5C,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;IACrB,CAAC;AACF,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,MAAc,EAAQ,EAAE;IAClD,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,IAAI,CAAC;QACJ,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IAC5D,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACZ,OAAO,CAAC,KAAK,CAAC,wBAAwB,CAAC,EAAE,CAAC,CAAC;QAC3C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;AACF,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,YAAY,GAAG,GAAsC,EAAE;IACnE,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QACnD,OAAO,IAAI,CAAC;IACb,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;AACjE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,MAAM,GAAG,CAAC,IAAY,EAAE,GAAQ,EAAQ,EAAE;IACtD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC;IACxB,yCAAyC;IACzC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACpB,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC;IACtB,CAAC;IACD,UAAU,CAAC,MAAM,CAAC,CAAC;AACpB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,SAAS,GAAG,CAAC,IAAY,EAAQ,EAAE;IAC/C,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzB,2CAA2C;IAC3C,IAAI,MAAM,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;QAC5B,MAAM,CAAC,MAAM,GAAG,SAAS,CAAC;IAC3B,CAAC;IACD,UAAU,CAAC,MAAM,CAAC,CAAC;AACpB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,IAAY,EAAQ,EAAE;IAClD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,aAAa,CAAC,CAAC;QACzC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;IACD,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,UAAU,CAAC,MAAM,CAAC,CAAC;AACpB,CAAC,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./types.js";
+//# 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":"AACA,cAAc,YAAY,CAAC"}

package/dist/index.js

@@ -0,0 +1,3 @@
+// Main library exports
+export * from "./types.js";
+//# 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,uBAAuB;AACvB,cAAc,YAAY,CAAC"}

package/dist/model-configs.d.ts

@@ -0,0 +1,22 @@
+import type { GPU } from "./types.js";
+/**
+ * Get the best configuration for a model based on available GPUs
+ */
+export declare const getModelConfig: (modelId: string, gpus: GPU[], requestedGpuCount: number) => {
+    args: string[];
+    env?: Record<string, string>;
+    notes?: string;
+} | null;
+/**
+ * Check if a model is known
+ */
+export declare const isKnownModel: (modelId: string) => boolean;
+/**
+ * Get all known models
+ */
+export declare const getKnownModels: () => string[];
+/**
+ * Get model display name
+ */
+export declare const getModelName: (modelId: string) => string;
+//# sourceMappingURL=model-configs.d.ts.map

package/dist/model-configs.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-configs.d.ts","sourceRoot":"","sources":["../src/model-configs.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,YAAY,CAAC;AA2BtC;;GAEG;AACH,eAAO,MAAM,cAAc,GAC1B,SAAS,MAAM,EACf,MAAM,GAAG,EAAE,EACX,mBAAmB,MAAM,KACvB;IAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IAAC,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,IAoDrE,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,YAAY,GAAI,SAAS,MAAM,KAAG,OAE9C,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,cAAc,QAAO,MAAM,EAEvC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,YAAY,GAAI,SAAS,MAAM,KAAG,MAE9C,CAAC"}

package/dist/model-configs.js

@@ -0,0 +1,75 @@
+import { readFileSync } from "fs";
+import { dirname, join } from "path";
+import { fileURLToPath } from "url";
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+// Load models configuration - resolve relative to this file
+const modelsJsonPath = join(__dirname, "models.json");
+const modelsData = JSON.parse(readFileSync(modelsJsonPath, "utf-8"));
+/**
+ * Get the best configuration for a model based on available GPUs
+ */
+export const getModelConfig = (modelId, gpus, requestedGpuCount) => {
+    const modelInfo = modelsData.models[modelId];
+    if (!modelInfo) {
+        // Unknown model, no default config
+        return null;
+    }
+    // Extract GPU type from the first GPU name (e.g., "NVIDIA H200" -> "H200")
+    const gpuType = gpus[0]?.name?.replace("NVIDIA", "")?.trim()?.split(" ")[0] || "";
+    // Find best matching config
+    let bestConfig = null;
+    for (const config of modelInfo.configs) {
+        // Check GPU count
+        if (config.gpuCount !== requestedGpuCount) {
+            continue;
+        }
+        // Check GPU type if specified
+        if (config.gpuTypes && config.gpuTypes.length > 0) {
+            const typeMatches = config.gpuTypes.some((type) => gpuType.includes(type) || type.includes(gpuType));
+            if (!typeMatches) {
+                continue;
+            }
+        }
+        // This config matches
+        bestConfig = config;
+        break;
+    }
+    // If no exact match, try to find a config with just the right GPU count
+    if (!bestConfig) {
+        for (const config of modelInfo.configs) {
+            if (config.gpuCount === requestedGpuCount) {
+                bestConfig = config;
+                break;
+            }
+        }
+    }
+    if (!bestConfig) {
+        // No suitable config found
+        return null;
+    }
+    return {
+        args: [...bestConfig.args],
+        env: bestConfig.env ? { ...bestConfig.env } : undefined,
+        notes: bestConfig.notes || modelInfo.notes,
+    };
+};
+/**
+ * Check if a model is known
+ */
+export const isKnownModel = (modelId) => {
+    return modelId in modelsData.models;
+};
+/**
+ * Get all known models
+ */
+export const getKnownModels = () => {
+    return Object.keys(modelsData.models);
+};
+/**
+ * Get model display name
+ */
+export const getModelName = (modelId) => {
+    return modelsData.models[modelId]?.name || modelId;
+};
+//# sourceMappingURL=model-configs.js.map

package/dist/model-configs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-configs.js","sourceRoot":"","sources":["../src/model-configs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AAGpC,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAoBtC,4DAA4D;AAC5D,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;AACtD,MAAM,UAAU,GAAe,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC;AAEjF;;GAEG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,CAC7B,OAAe,EACf,IAAW,EACX,iBAAyB,EACiD,EAAE;IAC5E,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC7C,IAAI,CAAC,SAAS,EAAE,CAAC;QAChB,mCAAmC;QACnC,OAAO,IAAI,CAAC;IACb,CAAC;IAED,2EAA2E;IAC3E,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAElF,4BAA4B;IAC5B,IAAI,UAAU,GAAuB,IAAI,CAAC;IAE1C,KAAK,MAAM,MAAM,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;QACxC,kBAAkB;QAClB,IAAI,MAAM,CAAC,QAAQ,KAAK,iBAAiB,EAAE,CAAC;YAC3C,SAAS;QACV,CAAC;QAED,8BAA8B;QAC9B,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnD,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YACrG,IAAI,CAAC,WAAW,EAAE,CAAC;gBAClB,SAAS;YACV,CAAC;QACF,CAAC;QAED,sBAAsB;QACtB,UAAU,GAAG,MAAM,CAAC;QACpB,MAAM;IACP,CAAC;IAED,wEAAwE;IACxE,IAAI,CAAC,UAAU,EAAE,CAAC;QACjB,KAAK,MAAM,MAAM,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;YACxC,IAAI,MAAM,CAAC,QAAQ,KAAK,iBAAiB,EAAE,CAAC;gBAC3C,UAAU,GAAG,MAAM,CAAC;gBACpB,MAAM;YACP,CAAC;QACF,CAAC;IACF,CAAC;IAED,IAAI,CAAC,UAAU,EAAE,CAAC;QACjB,2BAA2B;QAC3B,OAAO,IAAI,CAAC;IACb,CAAC;IAED,OAAO;QACN,IAAI,EAAE,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC;QAC1B,GAAG,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,SAAS;QACvD,KAAK,EAAE,UAAU,CAAC,KAAK,IAAI,SAAS,CAAC,KAAK;KAC1C,CAAC;AACH,CAAC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,OAAe,EAAW,EAAE;IACxD,OAAO,OAAO,IAAI,UAAU,CAAC,MAAM,CAAC;AACrC,CAAC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,GAAa,EAAE;IAC5C,OAAO,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;AACvC,CAAC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,OAAe,EAAU,EAAE;IACvD,OAAO,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,IAAI,IAAI,OAAO,CAAC;AACpD,CAAC,CAAC"}