@juspay/neurolink 1.6.0 → 1.9.0

package/dist/cli/commands/ollama.d.ts

@@ -1,4 +1,4 @@
-import type { Argv } from 'yargs';
+import type { Argv } from "yargs";
 export declare const ollamaCommand: {
     command: string;
     describe: string;

package/dist/cli/commands/ollama.js

@@ -1,40 +1,40 @@
-import { execSync } from 'child_process';
-import chalk from 'chalk';
-import ora from 'ora';
-import inquirer from 'inquirer';
+import { execSync } from "child_process";
+import chalk from "chalk";
+import ora from "ora";
+import inquirer from "inquirer";
 export const ollamaCommand = {
-    command: 'ollama <command>',
-    describe: 'Manage Ollama local AI models',
+    command: "ollama <command>",
+    describe: "Manage Ollama local AI models",
     builder: (yargs) => {
         return yargs
-            .command('list-models', 'List installed Ollama models', {}, listModelsHandler)
-            .command('pull <model>', 'Download an Ollama model', {
+            .command("list-models", "List installed Ollama models", {}, listModelsHandler)
+            .command("pull <model>", "Download an Ollama model", {
             model: {
-                describe: 'Model name to download',
-                type: 'string',
-                demandOption: true
-            }
+                describe: "Model name to download",
+                type: "string",
+                demandOption: true,
+            },
         }, pullModelHandler)
-            .command('remove <model>', 'Remove an Ollama model', {
+            .command("remove <model>", "Remove an Ollama model", {
             model: {
-                describe: 'Model name to remove',
-                type: 'string',
-                demandOption: true
-            }
+                describe: "Model name to remove",
+                type: "string",
+                demandOption: true,
+            },
         }, removeModelHandler)
-            .command('status', 'Check Ollama service status', {}, statusHandler)
-            .command('start', 'Start Ollama service', {}, startHandler)
-            .command('stop', 'Stop Ollama service', {}, stopHandler)
-            .command('setup', 'Interactive Ollama setup', {}, setupHandler)
-            .demandCommand(1, 'Please specify a command');
+            .command("status", "Check Ollama service status", {}, statusHandler)
+            .command("start", "Start Ollama service", {}, startHandler)
+            .command("stop", "Stop Ollama service", {}, stopHandler)
+            .command("setup", "Interactive Ollama setup", {}, setupHandler)
+            .demandCommand(1, "Please specify a command");
     },
-    handler: () => { } // No-op handler as subcommands handle everything
+    handler: () => { }, // No-op handler as subcommands handle everything
 };
 async function listModelsHandler() {
-    const spinner = ora('Fetching installed models...').start();
+    const spinner = ora("Fetching installed models...").start();
     try {
-        const output = execSync('ollama list', { encoding: 'utf8' });
-        spinner.succeed('Installed models:');
+        const output = execSync("ollama list", { encoding: "utf8" });
+        spinner.succeed("Installed models:");
         if (output.trim()) {
             console.log(output);
         }
@@ -43,24 +43,24 @@ async function listModelsHandler() {
         }
     }
     catch (error) {
-        spinner.fail('Failed to list models. Is Ollama installed?');
-        console.error(chalk.red('Error:', error.message));
-        console.log(chalk.blue('\nTip: Install Ollama from https://ollama.ai'));
+        spinner.fail("Failed to list models. Is Ollama installed?");
+        console.error(chalk.red("Error:", error.message));
+        console.log(chalk.blue("\nTip: Install Ollama from https://ollama.ai"));
         process.exit(1);
     }
 }
 async function pullModelHandler(argv) {
     const { model } = argv;
     console.log(chalk.blue(`Downloading model: ${model}`));
-    console.log(chalk.gray('This may take several minutes...'));
+    console.log(chalk.gray("This may take several minutes..."));
     try {
-        execSync(`ollama pull ${model}`, { stdio: 'inherit' });
+        execSync(`ollama pull ${model}`, { stdio: "inherit" });
         console.log(chalk.green(`\n✅ Successfully downloaded ${model}`));
         console.log(chalk.blue(`\nTest it with: npx @juspay/neurolink generate-text "Hello!" --provider ollama --model ${model}`));
     }
     catch (error) {
         console.error(chalk.red(`\n❌ Failed to download ${model}`));
-        console.error(chalk.red('Error:', error.message));
+        console.error(chalk.red("Error:", error.message));
         process.exit(1);
     }
 }
@@ -69,36 +69,38 @@ async function removeModelHandler(argv) {
     // Confirm removal
     const { confirm } = await inquirer.prompt([
         {
-            type: 'confirm',
-            name: 'confirm',
+            type: "confirm",
+            name: "confirm",
             message: `Are you sure you want to remove model "${model}"?`,
-            default: false
-        }
+            default: false,
+        },
     ]);
     if (!confirm) {
-        console.log(chalk.yellow('Removal cancelled.'));
+        console.log(chalk.yellow("Removal cancelled."));
         return;
     }
     const spinner = ora(`Removing model ${model}...`).start();
     try {
-        execSync(`ollama rm ${model}`, { encoding: 'utf8' });
+        execSync(`ollama rm ${model}`, { encoding: "utf8" });
         spinner.succeed(`Successfully removed ${model}`);
     }
     catch (error) {
         spinner.fail(`Failed to remove ${model}`);
-        console.error(chalk.red('Error:', error.message));
+        console.error(chalk.red("Error:", error.message));
         process.exit(1);
     }
 }
 async function statusHandler() {
-    const spinner = ora('Checking Ollama service status...').start();
+    const spinner = ora("Checking Ollama service status...").start();
     try {
         // Try to run a simple command
-        execSync('ollama list', { encoding: 'utf8' });
-        spinner.succeed('Ollama service is running');
+        execSync("ollama list", { encoding: "utf8" });
+        spinner.succeed("Ollama service is running");
         // Get additional info
         try {
-            const response = execSync('curl -s http://localhost:11434/api/tags', { encoding: 'utf8' });
+            const response = execSync("curl -s http://localhost:11434/api/tags", {
+                encoding: "utf8",
+            });
             const data = JSON.parse(response);
             if (data.models && data.models.length > 0) {
                 console.log(chalk.green(`\n${data.models.length} models available`));
@@ -109,156 +111,156 @@ async function statusHandler() {
         }
     }
     catch (error) {
-        spinner.fail('Ollama service is not running');
-        console.log(chalk.yellow('\nStart Ollama with: ollama serve'));
-        console.log(chalk.blue('Or restart the Ollama app if using the desktop version'));
+        spinner.fail("Ollama service is not running");
+        console.log(chalk.yellow("\nStart Ollama with: ollama serve"));
+        console.log(chalk.blue("Or restart the Ollama app if using the desktop version"));
         process.exit(1);
     }
 }
 async function startHandler() {
-    console.log(chalk.blue('Starting Ollama service...'));
+    console.log(chalk.blue("Starting Ollama service..."));
     try {
         // Check if already running
         try {
-            execSync('ollama list', { encoding: 'utf8' });
-            console.log(chalk.yellow('Ollama service is already running!'));
+            execSync("ollama list", { encoding: "utf8" });
+            console.log(chalk.yellow("Ollama service is already running!"));
             return;
         }
         catch {
             // Not running, continue to start
         }
         // Different approaches for different platforms
-        if (process.platform === 'darwin') {
+        if (process.platform === "darwin") {
             // macOS
-            console.log(chalk.gray('Starting Ollama on macOS...'));
+            console.log(chalk.gray("Starting Ollama on macOS..."));
             try {
-                execSync('open -a Ollama');
-                console.log(chalk.green('✅ Ollama app started'));
+                execSync("open -a Ollama");
+                console.log(chalk.green("✅ Ollama app started"));
             }
             catch {
                 // Try service command
-                execSync('ollama serve > /dev/null 2>&1 &', { stdio: 'ignore' });
-                console.log(chalk.green('✅ Ollama service started'));
+                execSync("ollama serve > /dev/null 2>&1 &", { stdio: "ignore" });
+                console.log(chalk.green("✅ Ollama service started"));
             }
         }
-        else if (process.platform === 'linux') {
+        else if (process.platform === "linux") {
             // Linux
-            console.log(chalk.gray('Starting Ollama service on Linux...'));
+            console.log(chalk.gray("Starting Ollama service on Linux..."));
             try {
-                execSync('systemctl start ollama', { encoding: 'utf8' });
-                console.log(chalk.green('✅ Ollama service started'));
+                execSync("systemctl start ollama", { encoding: "utf8" });
+                console.log(chalk.green("✅ Ollama service started"));
             }
             catch {
                 // Try direct command
-                execSync('ollama serve > /dev/null 2>&1 &', { stdio: 'ignore' });
-                console.log(chalk.green('✅ Ollama service started'));
+                execSync("ollama serve > /dev/null 2>&1 &", { stdio: "ignore" });
+                console.log(chalk.green("✅ Ollama service started"));
             }
         }
         else {
             // Windows
-            console.log(chalk.gray('Starting Ollama on Windows...'));
-            execSync('start ollama serve', { stdio: 'ignore' });
-            console.log(chalk.green('✅ Ollama service started'));
+            console.log(chalk.gray("Starting Ollama on Windows..."));
+            execSync("start ollama serve", { stdio: "ignore" });
+            console.log(chalk.green("✅ Ollama service started"));
         }
-        console.log(chalk.blue('\nWait a few seconds for the service to initialize...'));
+        console.log(chalk.blue("\nWait a few seconds for the service to initialize..."));
     }
     catch (error) {
-        console.error(chalk.red('Failed to start Ollama service'));
-        console.error(chalk.red('Error:', error.message));
-        console.log(chalk.blue('\nTry starting Ollama manually or check installation'));
+        console.error(chalk.red("Failed to start Ollama service"));
+        console.error(chalk.red("Error:", error.message));
+        console.log(chalk.blue("\nTry starting Ollama manually or check installation"));
         process.exit(1);
     }
 }
 async function stopHandler() {
-    const spinner = ora('Stopping Ollama service...').start();
+    const spinner = ora("Stopping Ollama service...").start();
     try {
-        if (process.platform === 'darwin') {
+        if (process.platform === "darwin") {
             // macOS
             try {
-                execSync('pkill ollama', { encoding: 'utf8' });
+                execSync("pkill ollama", { encoding: "utf8" });
             }
             catch {
-                execSync('killall Ollama', { encoding: 'utf8' });
+                execSync("killall Ollama", { encoding: "utf8" });
             }
         }
-        else if (process.platform === 'linux') {
+        else if (process.platform === "linux") {
             // Linux
             try {
-                execSync('systemctl stop ollama', { encoding: 'utf8' });
+                execSync("systemctl stop ollama", { encoding: "utf8" });
             }
             catch {
-                execSync('pkill ollama', { encoding: 'utf8' });
+                execSync("pkill ollama", { encoding: "utf8" });
             }
         }
         else {
             // Windows
-            execSync('taskkill /F /IM ollama.exe', { encoding: 'utf8' });
+            execSync("taskkill /F /IM ollama.exe", { encoding: "utf8" });
         }
-        spinner.succeed('Ollama service stopped');
+        spinner.succeed("Ollama service stopped");
     }
     catch (error) {
-        spinner.fail('Failed to stop Ollama service');
-        console.error(chalk.red('It may not be running or requires manual stop'));
+        spinner.fail("Failed to stop Ollama service");
+        console.error(chalk.red("It may not be running or requires manual stop"));
     }
 }
 async function setupHandler() {
-    console.log(chalk.blue('🦙 Welcome to Ollama Setup!\n'));
+    console.log(chalk.blue("🦙 Welcome to Ollama Setup!\n"));
     // Check if Ollama is installed
-    const checkSpinner = ora('Checking Ollama installation...').start();
+    const checkSpinner = ora("Checking Ollama installation...").start();
     let isInstalled = false;
     try {
-        execSync('ollama --version', { encoding: 'utf8' });
+        execSync("ollama --version", { encoding: "utf8" });
         isInstalled = true;
-        checkSpinner.succeed('Ollama is installed');
+        checkSpinner.succeed("Ollama is installed");
     }
     catch {
-        checkSpinner.fail('Ollama is not installed');
+        checkSpinner.fail("Ollama is not installed");
     }
     if (!isInstalled) {
-        console.log(chalk.yellow('\nOllama needs to be installed first.'));
-        console.log(chalk.blue('\nInstallation instructions:'));
-        if (process.platform === 'darwin') {
-            console.log('\nFor macOS:');
-            console.log(chalk.gray('  brew install ollama'));
-            console.log(chalk.gray('  # or download from https://ollama.ai'));
+        console.log(chalk.yellow("\nOllama needs to be installed first."));
+        console.log(chalk.blue("\nInstallation instructions:"));
+        if (process.platform === "darwin") {
+            console.log("\nFor macOS:");
+            console.log(chalk.gray("  brew install ollama"));
+            console.log(chalk.gray("  # or download from https://ollama.ai"));
         }
-        else if (process.platform === 'linux') {
-            console.log('\nFor Linux:');
-            console.log(chalk.gray('  curl -fsSL https://ollama.ai/install.sh | sh'));
+        else if (process.platform === "linux") {
+            console.log("\nFor Linux:");
+            console.log(chalk.gray("  curl -fsSL https://ollama.ai/install.sh | sh"));
         }
         else {
-            console.log('\nFor Windows:');
-            console.log(chalk.gray('  Download from https://ollama.ai'));
+            console.log("\nFor Windows:");
+            console.log(chalk.gray("  Download from https://ollama.ai"));
         }
         const { proceedAnyway } = await inquirer.prompt([
             {
-                type: 'confirm',
-                name: 'proceedAnyway',
-                message: 'Would you like to continue with setup anyway?',
-                default: false
-            }
+                type: "confirm",
+                name: "proceedAnyway",
+                message: "Would you like to continue with setup anyway?",
+                default: false,
+            },
         ]);
         if (!proceedAnyway) {
-            console.log(chalk.blue('\nInstall Ollama and run setup again!'));
+            console.log(chalk.blue("\nInstall Ollama and run setup again!"));
             return;
         }
     }
     // Check if service is running
     let serviceRunning = false;
     try {
-        execSync('ollama list', { encoding: 'utf8' });
+        execSync("ollama list", { encoding: "utf8" });
         serviceRunning = true;
-        console.log(chalk.green('\n✅ Ollama service is running'));
+        console.log(chalk.green("\n✅ Ollama service is running"));
     }
     catch {
-        console.log(chalk.yellow('\n⚠️  Ollama service is not running'));
+        console.log(chalk.yellow("\n⚠️  Ollama service is not running"));
         const { startService } = await inquirer.prompt([
             {
-                type: 'confirm',
-                name: 'startService',
-                message: 'Would you like to start the Ollama service?',
-                default: true
-            }
+                type: "confirm",
+                name: "startService",
+                message: "Would you like to start the Ollama service?",
+                default: true,
+            },
         ]);
         if (startService) {
             await startHandler();
@@ -267,45 +269,51 @@ async function setupHandler() {
     }
     if (serviceRunning) {
         // List available models
-        console.log(chalk.blue('\n📦 Popular Ollama models:'));
-        console.log('  • llama2 (7B) - General purpose');
-        console.log('  • codellama (7B) - Code generation');
-        console.log('  • mistral (7B) - Fast and efficient');
-        console.log('  • tinyllama (1B) - Lightweight');
-        console.log('  • phi (2.7B) - Microsoft\'s compact model');
+        console.log(chalk.blue("\n📦 Popular Ollama models:"));
+        console.log("  • llama2 (7B) - General purpose");
+        console.log("  • codellama (7B) - Code generation");
+        console.log("  • mistral (7B) - Fast and efficient");
+        console.log("  • tinyllama (1B) - Lightweight");
+        console.log("  • phi (2.7B) - Microsoft's compact model");
         const { downloadModel } = await inquirer.prompt([
             {
-                type: 'confirm',
-                name: 'downloadModel',
-                message: 'Would you like to download a model?',
-                default: true
-            }
+                type: "confirm",
+                name: "downloadModel",
+                message: "Would you like to download a model?",
+                default: true,
+            },
         ]);
         if (downloadModel) {
             const { selectedModel } = await inquirer.prompt([
                 {
-                    type: 'list',
-                    name: 'selectedModel',
-                    message: 'Select a model to download:',
+                    type: "list",
+                    name: "selectedModel",
+                    message: "Select a model to download:",
                     choices: [
-                        { name: 'llama2 (7B) - Recommended for general use', value: 'llama2' },
-                        { name: 'codellama (7B) - Best for code generation', value: 'codellama' },
-                        { name: 'mistral (7B) - Fast and efficient', value: 'mistral' },
-                        { name: 'tinyllama (1B) - Lightweight, fast', value: 'tinyllama' },
-                        { name: 'phi (2.7B) - Microsoft\'s compact model', value: 'phi' },
-                        { name: 'Other (enter manually)', value: 'other' }
-                    ]
-                }
+                        {
+                            name: "llama2 (7B) - Recommended for general use",
+                            value: "llama2",
+                        },
+                        {
+                            name: "codellama (7B) - Best for code generation",
+                            value: "codellama",
+                        },
+                        { name: "mistral (7B) - Fast and efficient", value: "mistral" },
+                        { name: "tinyllama (1B) - Lightweight, fast", value: "tinyllama" },
+                        { name: "phi (2.7B) - Microsoft's compact model", value: "phi" },
+                        { name: "Other (enter manually)", value: "other" },
+                    ],
+                },
             ]);
             let modelToDownload = selectedModel;
-            if (selectedModel === 'other') {
+            if (selectedModel === "other") {
                 const { customModel } = await inquirer.prompt([
                     {
-                        type: 'input',
-                        name: 'customModel',
-                        message: 'Enter the model name:',
-                        validate: (input) => input.trim().length > 0 || 'Model name is required'
-                    }
+                        type: "input",
+                        name: "customModel",
+                        message: "Enter the model name:",
+                        validate: (input) => input.trim().length > 0 || "Model name is required",
+                    },
                 ]);
                 modelToDownload = customModel;
             }
@@ -313,11 +321,13 @@ async function setupHandler() {
         }
     }
     // Final instructions
-    console.log(chalk.green('\n✅ Setup complete!\n'));
-    console.log(chalk.blue('Next steps:'));
-    console.log('1. List models: ' + chalk.gray('neurolink ollama list-models'));
-    console.log('2. Generate text: ' + chalk.gray('neurolink generate-text "Hello!" --provider ollama'));
-    console.log('3. Use specific model: ' + chalk.gray('neurolink generate-text "Hello!" --provider ollama --model codellama'));
-    console.log(chalk.gray('\nFor more information, see: https://docs.neurolink.ai/providers/ollama'));
+    console.log(chalk.green("\n✅ Setup complete!\n"));
+    console.log(chalk.blue("Next steps:"));
+    console.log("1. List models: " + chalk.gray("neurolink ollama list-models"));
+    console.log("2. Generate text: " +
+        chalk.gray('neurolink generate-text "Hello!" --provider ollama'));
+    console.log("3. Use specific model: " +
+        chalk.gray('neurolink generate-text "Hello!" --provider ollama --model codellama'));
+    console.log(chalk.gray("\nFor more information, see: https://docs.neurolink.ai/providers/ollama"));
 }
 export default ollamaCommand;