@juspay/neurolink 7.1.0 → 7.2.0

package/dist/cli/index.js

@@ -10,6 +10,7 @@ import { hideBin } from "yargs/helpers";
 import chalk from "chalk";
 import { addOllamaCommands } from "./commands/ollama.js";
 import { CLICommandFactory } from "./factories/commandFactory.js";
+import { logger } from "../lib/utils/logger.js";
 // Load environment variables from .env file
 try {
     // Try to import and configure dotenv
@@ -81,24 +82,24 @@ function handleError(error, context) {
             "Authorization error: You are not authorized to perform this action or access this resource.";
     }
     // If no specific condition matched, genericMessage remains error.message
-    console.error(chalk.red(`❌ ${context} failed: ${genericMessage}`));
+    logger.error(chalk.red(`❌ ${context} failed: ${genericMessage}`));
     // Smart hints for common errors (just string matching!)
     if (genericMessage.toLowerCase().includes("api key") ||
         genericMessage.toLowerCase().includes("credential")) {
-        console.error(chalk.yellow("💡 Set Google AI Studio API key (RECOMMENDED): export GOOGLE_AI_API_KEY=AIza-..."));
-        console.error(chalk.yellow("💡 Or set OpenAI API key: export OPENAI_API_KEY=sk-..."));
-        console.error(chalk.yellow("💡 Or set AWS Bedrock credentials: export AWS_ACCESS_KEY_ID=... AWS_SECRET_ACCESS_KEY=... AWS_REGION=us-east-1"));
-        console.error(chalk.yellow("💡 Or set Google Vertex AI credentials: export GOOGLE_APPLICATION_CREDENTIALS=/path/to/key.json"));
-        console.error(chalk.yellow("💡 Or set Anthropic API key: export ANTHROPIC_API_KEY=sk-ant-..."));
-        console.error(chalk.yellow("💡 Or set Azure OpenAI credentials: export AZURE_OPENAI_API_KEY=... AZURE_OPENAI_ENDPOINT=..."));
+        logger.error(chalk.yellow("💡 Set Google AI Studio API key (RECOMMENDED): export GOOGLE_AI_API_KEY=AIza-..."));
+        logger.error(chalk.yellow("💡 Or set OpenAI API key: export OPENAI_API_KEY=sk-..."));
+        logger.error(chalk.yellow("💡 Or set AWS Bedrock credentials: export AWS_ACCESS_KEY_ID=... AWS_SECRET_ACCESS_KEY=... AWS_REGION=us-east-1"));
+        logger.error(chalk.yellow("💡 Or set Google Vertex AI credentials: export GOOGLE_APPLICATION_CREDENTIALS=/path/to/key.json"));
+        logger.error(chalk.yellow("💡 Or set Anthropic API key: export ANTHROPIC_API_KEY=sk-ant-..."));
+        logger.error(chalk.yellow("💡 Or set Azure OpenAI credentials: export AZURE_OPENAI_API_KEY=... AZURE_OPENAI_ENDPOINT=..."));
     }
     if (error.message.toLowerCase().includes("rate limit")) {
-        console.error(chalk.yellow("💡 Try again in a few moments or use --provider vertex"));
+        logger.error(chalk.yellow("💡 Try again in a few moments or use --provider vertex"));
     }
     if (error.message.toLowerCase().includes("not authorized") ||
         error.message.toLowerCase().includes("permission denied")) {
-        console.error(chalk.yellow("💡 Check your account permissions for the selected model/service."));
-        console.error(chalk.yellow("💡 For AWS Bedrock, ensure you have permissions for the specific model and consider using inference profile ARNs."));
+        logger.error(chalk.yellow("💡 Check your account permissions for the selected model/service."));
+        logger.error(chalk.yellow("💡 For AWS Bedrock, ensure you have permissions for the specific model and consider using inference profile ARNs."));
     }
     process.exit(1);
 }
@@ -118,6 +119,14 @@ const cli = yargs(args)
     .epilogue("For more info: https://github.com/juspay/neurolink")
     .showHelpOnFail(true, "Specify --help for available options")
     .middleware((argv) => {
+    // Handle no-color option globally
+    if (argv.noColor || process.env.NO_COLOR || !process.stdout.isTTY) {
+        process.env.FORCE_COLOR = "0";
+    }
+    // Handle custom config file
+    if (argv.configFile) {
+        process.env.NEUROLINK_CONFIG_FILE = argv.configFile;
+    }
     // Control SDK logging based on debug flag
     if (argv.debug) {
         process.env.NEUROLINK_DEBUG = "true";
@@ -205,6 +214,12 @@ const cli = yargs(args)
     .command(CLICommandFactory.createProviderCommands())
     // Status command alias - Using CLICommandFactory
     .command(CLICommandFactory.createStatusCommand())
+    // Models Command Group - Using CLICommandFactory
+    .command(CLICommandFactory.createModelsCommands())
+    // MCP Command Group - Using CLICommandFactory
+    .command(CLICommandFactory.createMCPCommands())
+    // Discover Command - Using CLICommandFactory
+    .command(CLICommandFactory.createDiscoverCommand())
     // Configuration Command Group - Using CLICommandFactory
     .command(CLICommandFactory.createConfigCommands())
     // Get Best Provider Command - Using CLICommandFactory

package/dist/cli/utils/completeSetup.js

@@ -7,6 +7,7 @@
 import { runInteractiveSetup, testProviderConnectivity, displaySetupSummary, } from "./interactiveSetup.js";
 import { updateEnvFile, displayEnvUpdateSummary } from "./envManager.js";
 import chalk from "chalk";
+import { logger } from "../../lib/utils/logger.js";
 /**
  * Run the complete interactive setup process
  */
@@ -14,20 +15,20 @@ export async function runCompleteSetup(quiet = false) {
     try {
         // Step 1: Run interactive setup wizard
         if (!quiet) {
-            console.log(chalk.blue("🚀 Starting NeuroLink Configuration Setup...\n"));
+            logger.always(chalk.blue("🚀 Starting NeuroLink Configuration Setup...\n"));
         }
         const setupResult = await runInteractiveSetup(quiet);
         // If no providers selected, exit early
         if (setupResult.selectedProviders.length === 0) {
             if (!quiet) {
-                console.log(chalk.yellow("⚠️  No providers selected. Setup cancelled."));
+                logger.always(chalk.yellow("⚠️  No providers selected. Setup cancelled."));
             }
             return setupResult;
         }
         // Step 2: Update environment file with credentials
         if (Object.keys(setupResult.credentials).length > 0) {
             if (!quiet) {
-                console.log(chalk.blue("\n💾 Updating environment configuration...\n"));
+                logger.always(chalk.blue("\n💾 Updating environment configuration...\n"));
             }
             try {
                 const envResult = updateEnvFile(setupResult.credentials, ".env", true);
@@ -42,14 +43,14 @@ export async function runCompleteSetup(quiet = false) {
             }
             catch (error) {
                 if (!quiet) {
-                    console.error(chalk.red(`❌ Failed to update environment file: ${error instanceof Error ? error.message : String(error)}`));
+                    logger.error(chalk.red(`❌ Failed to update environment file: ${error instanceof Error ? error.message : String(error)}`));
                 }
                 throw error;
             }
         }
         // Step 3: Test provider connectivity
         if (!quiet) {
-            console.log(chalk.blue("\n🧪 Testing configured providers...\n"));
+            logger.always(chalk.blue("\n🧪 Testing configured providers...\n"));
         }
         setupResult.testResults = await testProviderConnectivity(setupResult.selectedProviders, quiet);
         // Step 4: Display summary
@@ -58,8 +59,8 @@ export async function runCompleteSetup(quiet = false) {
     }
     catch (error) {
         if (!quiet) {
-            console.error(chalk.red(`❌ Setup failed: ${error instanceof Error ? error.message : String(error)}`));
-            console.log(chalk.yellow("💡 You can retry setup with: neurolink config setup"));
+            logger.error(chalk.red(`❌ Setup failed: ${error instanceof Error ? error.message : String(error)}`));
+            logger.always(chalk.yellow("💡 You can retry setup with: neurolink config setup"));
         }
         throw error;
     }
@@ -75,7 +76,7 @@ export async function configSetup(quiet = false) {
  */
 export async function configInit(quiet = false) {
     if (!quiet) {
-        console.log(chalk.gray("📝 config init is an alias for config setup\n"));
+        logger.always(chalk.gray("📝 config init is an alias for config setup\n"));
     }
     await runCompleteSetup(quiet);
 }

package/dist/cli/utils/envManager.js

@@ -5,6 +5,7 @@
  */
 import fs from "fs";
 import chalk from "chalk";
+import { logger } from "../../lib/utils/logger.js";
 /**
  * Create a timestamped backup of the existing .env file
  */
@@ -158,23 +159,23 @@ export function displayEnvUpdateSummary(result, quiet = false) {
         return;
     }
     if (result.backup.existed && result.backup.backupPath) {
-        console.log(chalk.gray(`💾 Created backup: ${result.backup.backupPath}`));
+        logger.always(chalk.gray(`💾 Created backup: ${result.backup.backupPath}`));
     }
     if (result.added.length > 0) {
-        console.log(chalk.green(`➕ Added ${result.added.length} new variables: ${result.added.join(", ")}`));
+        logger.always(chalk.green(`➕ Added ${result.added.length} new variables: ${result.added.join(", ")}`));
     }
     if (result.updated.length > 0) {
-        console.log(chalk.yellow(`🔄 Updated ${result.updated.length} existing variables: ${result.updated.join(", ")}`));
+        logger.always(chalk.yellow(`🔄 Updated ${result.updated.length} existing variables: ${result.updated.join(", ")}`));
     }
     if (result.unchanged.length > 0) {
-        console.log(chalk.gray(`✓ ${result.unchanged.length} variables unchanged: ${result.unchanged.join(", ")}`));
+        logger.always(chalk.gray(`✓ ${result.unchanged.length} variables unchanged: ${result.unchanged.join(", ")}`));
     }
     const totalChanges = result.added.length + result.updated.length;
     if (totalChanges > 0) {
-        console.log(chalk.blue(`📝 Environment file updated with ${totalChanges} changes`));
+        logger.always(chalk.blue(`📝 Environment file updated with ${totalChanges} changes`));
     }
     else {
-        console.log(chalk.gray("📝 No changes needed to environment file"));
+        logger.always(chalk.gray("📝 No changes needed to environment file"));
     }
 }
 /**

package/dist/cli/utils/interactiveSetup.js

@@ -8,6 +8,7 @@ import { AIProviderName } from "../../lib/core/types.js";
 import { NeuroLink } from "../../lib/neurolink.js";
 import chalk from "chalk";
 import ora from "ora";
+import { logger } from "../../lib/utils/logger.js";
 export const PROVIDER_CONFIGS = [
     {
         id: AIProviderName.OPENAI,
@@ -145,8 +146,8 @@ export async function runInteractiveSetup(quiet = false) {
         testResults: [],
     };
     if (!quiet) {
-        console.log(chalk.blue("\n🎉 Welcome to NeuroLink Interactive Setup!"));
-        console.log(chalk.gray("This wizard will help you configure AI providers for NeuroLink.\n"));
+        logger.always(chalk.blue("\n🎉 Welcome to NeuroLink Interactive Setup!"));
+        logger.always(chalk.gray("This wizard will help you configure AI providers for NeuroLink.\n"));
     }
     // Step 1: Provider Selection
     const providerChoices = PROVIDER_CONFIGS.map((config) => ({
@@ -171,7 +172,7 @@ export async function runInteractiveSetup(quiet = false) {
     result.selectedProviders = selectedProviders;
     // Step 2: Credential Collection
     if (!quiet) {
-        console.log(chalk.blue("\n🔑 Collecting credentials for selected providers...\n"));
+        logger.always(chalk.blue("\n🔑 Collecting credentials for selected providers...\n"));
     }
     for (const providerId of selectedProviders) {
         const config = PROVIDER_CONFIGS.find((c) => c.id === providerId);
@@ -179,7 +180,7 @@ export async function runInteractiveSetup(quiet = false) {
             continue;
         }
         if (!quiet) {
-            console.log(chalk.yellow(`\n📋 Configuring ${config.name}:`));
+            logger.always(chalk.yellow(`\n📋 Configuring ${config.name}:`));
         }
         for (const envVar of config.envVars) {
             const currentValue = process.env[envVar.key];
@@ -235,7 +236,7 @@ export async function testProviderConnectivity(providers, quiet = false) {
     const sdk = new NeuroLink();
     const results = [];
     if (!quiet) {
-        console.log(chalk.blue("\n🧪 Testing provider connectivity...\n"));
+        logger.always(chalk.blue("\n🧪 Testing provider connectivity...\n"));
     }
     const spinner = quiet ? null : ora().start();
     for (const provider of providers) {
@@ -252,7 +253,7 @@ export async function testProviderConnectivity(providers, quiet = false) {
                 spinner.start(); // Restart for next provider
             }
             else if (!quiet) {
-                console.log(`${provider}: ${chalk.green("✅ Working")} (${duration}ms)`);
+                logger.always(`${provider}: ${chalk.green("✅ Working")} (${duration}ms)`);
             }
         }
         catch (error) {
@@ -263,7 +264,7 @@ export async function testProviderConnectivity(providers, quiet = false) {
                 spinner.start(); // Restart for next provider
             }
             else if (!quiet) {
-                console.error(`${provider}: ${chalk.red("❌ Failed")} - ${errorMessage.split("\n")[0]}`);
+                logger.error(`${provider}: ${chalk.red("❌ Failed")} - ${errorMessage.split("\n")[0]}`);
             }
         }
     }
@@ -281,22 +282,22 @@ export function displaySetupSummary(result, quiet = false) {
     }
     const working = result.testResults.filter((r) => r.status === "working").length;
     const total = result.testResults.length;
-    console.log(chalk.blue("\n📊 Setup Summary:"));
-    console.log(chalk.blue("================"));
-    console.log(`Selected providers: ${result.selectedProviders.length}`);
-    console.log(`Working providers: ${working}/${total}`);
+    logger.always(chalk.blue("\n📊 Setup Summary:"));
+    logger.always(chalk.blue("================"));
+    logger.always(`Selected providers: ${result.selectedProviders.length}`);
+    logger.always(`Working providers: ${working}/${total}`);
     if (result.envFileBackup) {
-        console.log(chalk.gray(`Environment backup: ${result.envFileBackup}`));
+        logger.always(chalk.gray(`Environment backup: ${result.envFileBackup}`));
     }
     if (working > 0) {
-        console.log(chalk.green("\n✅ Setup completed successfully!"));
-        console.log(chalk.yellow("💡 You can now use NeuroLink with your configured providers."));
-        console.log(chalk.gray('   Try: neurolink generate "Hello, AI!"'));
+        logger.always(chalk.green("\n✅ Setup completed successfully!"));
+        logger.always(chalk.yellow("💡 You can now use NeuroLink with your configured providers."));
+        logger.always(chalk.gray('   Try: neurolink generate "Hello, AI!"'));
     }
     else {
-        console.log(chalk.red("\n❌ No providers are working."));
-        console.log(chalk.yellow("💡 Please check your credentials and try again."));
-        console.log(chalk.gray("   Run: neurolink config setup"));
+        logger.always(chalk.red("\n❌ No providers are working."));
+        logger.always(chalk.yellow("💡 Please check your credentials and try again."));
+        logger.always(chalk.gray("   Run: neurolink config setup"));
     }
-    console.log(chalk.blue("\n📚 Documentation: https://github.com/juspay/neurolink#setup"));
+    logger.always(chalk.blue("\n📚 Documentation: https://github.com/juspay/neurolink#setup"));
 }

package/dist/core/analytics.js

@@ -50,52 +50,39 @@ export function createAnalytics(provider, model, result, responseTime, context)
  * Extract token usage from various AI result formats
  */
 function extractTokenUsage(result) {
-    // Handle different response formats
+    // Use properly typed usage object from BaseProvider or direct AI SDK
     if (result.usage &&
         typeof result.usage === "object" &&
         result.usage !== null) {
         const usage = result.usage;
-        // Standard format
-        if (typeof usage.promptTokens === "number" &&
-            typeof usage.completionTokens === "number") {
-            return {
-                input: usage.promptTokens || 0,
-                output: usage.completionTokens || 0,
-                total: typeof usage.totalTokens === "number"
-                    ? usage.totalTokens
-                    : usage.promptTokens + usage.completionTokens,
-            };
+        // Try BaseProvider normalized format first (inputTokens/outputTokens)
+        if (typeof usage.inputTokens === "number" ||
+            typeof usage.outputTokens === "number") {
+            const input = typeof usage.inputTokens === "number" ? usage.inputTokens : 0;
+            const output = typeof usage.outputTokens === "number" ? usage.outputTokens : 0;
+            const total = typeof usage.totalTokens === "number"
+                ? usage.totalTokens
+                : input + output;
+            return { input, output, total };
         }
-        // Alternative formats
-        if (typeof usage.input_tokens === "number" &&
-            typeof usage.output_tokens === "number") {
-            return {
-                input: usage.input_tokens || 0,
-                output: usage.output_tokens || 0,
-                total: typeof usage.total_tokens === "number"
-                    ? usage.total_tokens
-                    : usage.input_tokens + usage.output_tokens,
-            };
+        // Try OpenAI/Mistral format (promptTokens/completionTokens)
+        if (typeof usage.promptTokens === "number" ||
+            typeof usage.completionTokens === "number") {
+            const input = typeof usage.promptTokens === "number" ? usage.promptTokens : 0;
+            const output = typeof usage.completionTokens === "number" ? usage.completionTokens : 0;
+            const total = typeof usage.totalTokens === "number"
+                ? usage.totalTokens
+                : input + output;
+            return { input, output, total };
         }
-        // Generic tokens field
-        if (typeof usage.tokens === "number") {
-            return {
-                input: 0,
-                output: 0,
-                total: usage.tokens,
-            };
+        // Handle total-only case
+        if (typeof usage.totalTokens === "number") {
+            return { input: 0, output: 0, total: usage.totalTokens };
         }
     }
-    // Fallback: estimate from text length
-    const textLength = (typeof result.text === "string" ? result.text.length : 0) ||
-        (typeof result.content === "string" ? result.content.length : 0) ||
-        0;
-    const estimatedTokens = Math.ceil(textLength / 4); // ~4 chars per token
-    return {
-        input: 0,
-        output: estimatedTokens,
-        total: estimatedTokens,
-    };
+    // Fallback for edge cases
+    logger.debug("Token extraction failed: unknown usage format", { result });
+    return { input: 0, output: 0, total: 0 };
 }
 /**
  * Estimate cost based on provider, model, and token usage

package/dist/core/baseProvider.d.ts

@@ -447,5 +447,13 @@ export declare abstract class BaseProvider implements AIProvider {
      * Get timeout value in milliseconds
      */
     getTimeout(options: TextGenerationOptions | StreamOptions): number;
+    /**
+     * Utility method to chunk large prompts into smaller pieces
+     * @param prompt The prompt to chunk
+     * @param maxChunkSize Maximum size per chunk (default: 900,000 characters)
+     * @param overlap Overlap between chunks to maintain context (default: 100 characters)
+     * @returns Array of prompt chunks
+     */
+    static chunkPrompt(prompt: string, maxChunkSize?: number, overlap?: number): string[];
 }
 export {};