@drakulavich/parakeet-cli 0.3.0 → 0.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@drakulavich/parakeet-cli",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "Fast multilingual speech-to-text CLI powered by NVIDIA Parakeet ONNX models",
   "type": "module",
   "bin": {

package/src/cli.ts

@@ -1,6 +1,7 @@
 #!/usr/bin/env bun
 
 import { transcribe } from "./lib";
+import { downloadModel } from "./models";
 
 async function main(): Promise<void> {
   const args = process.argv.slice(2);
@@ -11,20 +12,34 @@ async function main(): Promise<void> {
     process.exit(0);
   }
 
-  const noCache = args.includes("--no-cache");
-  const file = args.filter((a) => !a.startsWith("--"))[0];
+  const positional = args.filter((a) => !a.startsWith("--"));
+
+  if (positional[0] === "install") {
+    const noCache = args.includes("--no-cache");
+    try {
+      await downloadModel(noCache);
+    } catch (err: unknown) {
+      const message = err instanceof Error ? err.message : String(err);
+      console.error(`Error: ${message}`);
+      process.exit(1);
+    }
+    process.exit(0);
+  }
+
+  const file = positional[0];
 
   if (!file) {
-    console.error("Usage: parakeet [--no-cache] <audio_file>");
+    console.error("Usage: parakeet [--version] <audio_file>");
+    console.error("       parakeet install [--no-cache]");
     process.exit(1);
   }
 
   try {
-    const text = await transcribe(file, { noCache });
+    const text = await transcribe(file);
     if (text) process.stdout.write(text + "\n");
   } catch (err: unknown) {
     const message = err instanceof Error ? err.message : String(err);
-    console.error(`Error: ${message}`);
+    console.error(message);
     process.exit(1);
   }
 }

package/src/lib.ts

@@ -1,11 +1,9 @@
 import { existsSync } from "fs";
-import { transcribe as internalTranscribe } from "./transcribe";
+import { transcribe as internalTranscribe, type TranscribeOptions } from "./transcribe";
+import { downloadModel } from "./models";
 
-export interface TranscribeOptions {
-  beamWidth?: number;
-  noCache?: boolean;
-  modelDir?: string;
-}
+export type { TranscribeOptions };
+export { downloadModel };
 
 export async function transcribe(
   audioPath: string,
@@ -15,9 +13,5 @@ export async function transcribe(
     throw new Error(`File not found: ${audioPath}`);
   }
 
-  return internalTranscribe(audioPath, {
-    noCache: options.noCache ?? false,
-    beamWidth: options.beamWidth,
-    modelDir: options.modelDir,
-  });
+  return internalTranscribe(audioPath, options);
 }

package/src/models.ts

@@ -21,10 +21,31 @@ export function isModelCached(dir?: string): boolean {
   return MODEL_FILES.every((f) => existsSync(join(d, f)));
 }
 
-export async function ensureModel(noCache = false, modelDir?: string): Promise<string> {
+export function requireModel(modelDir?: string): string {
   const dir = modelDir ?? getModelDir();
 
-  if (!noCache && isModelCached()) {
+  if (!isModelCached(dir)) {
+    const lines = [
+      `Error: Model not found at ${dir}`,
+      "",
+      "╔══════════════════════════════════════════════════════════╗",
+      "║ Looks like Parakeet model is not downloaded yet.         ║",
+      "║ Please run the following command to download the model:  ║",
+      "║                                                          ║",
+      "║     npx @drakulavich/parakeet-cli install                ║",
+      "╚══════════════════════════════════════════════════════════╝",
+    ];
+    throw new Error(lines.join("\n"));
+  }
+
+  return dir;
+}
+
+export async function downloadModel(noCache = false, modelDir?: string): Promise<string> {
+  const dir = modelDir ?? getModelDir();
+
+  if (!noCache && isModelCached(dir)) {
+    console.error("Model already downloaded.");
     return dir;
   }
 
@@ -47,5 +68,6 @@ export async function ensureModel(noCache = false, modelDir?: string): Promise<s
     await Bun.write(dest, res);
   }
 
+  console.error("Model downloaded successfully.");
   return dir;
 }

package/src/transcribe.ts

@@ -1,4 +1,4 @@
-import { ensureModel } from "./models";
+import { requireModel } from "./models";
 import { convertToFloat32PCM } from "./audio";
 import { initPreprocessor, preprocess } from "./preprocess";
 import { initEncoder, encode } from "./encoder";
@@ -25,7 +25,6 @@ const DECODER_LAYERS = 2;
 const DECODER_HIDDEN = 640;
 
 export interface TranscribeOptions {
-  noCache?: boolean;
   beamWidth?: number;
   modelDir?: string;
 }
@@ -40,9 +39,8 @@ export async function transcribe(audioPath: string, opts: TranscribeOptions = {}
     return "";
   }
 
-  const noCache = opts.noCache ?? false;
   const beamWidth = opts.beamWidth ?? 4;
-  const modelDir = await ensureModel(noCache, opts.modelDir);
+  const modelDir = requireModel(opts.modelDir);
   const tokenizer = await Tokenizer.fromFile(join(modelDir, "vocab.txt"));
 
   await initPreprocessor(modelDir);