@juspay/neurolink 5.1.0 → 5.2.0

package/dist/cli/factories/command-factory.js

@@ -74,6 +74,7 @@ export class CLICommandFactory {
                     type: "string",
                     choices: ["text", "structured", "json"],
                     default: "text",
+                    alias: "format",
                 })
                     .option("debug", {
                     describe: "Enable debug output",
@@ -84,6 +85,52 @@ export class CLICommandFactory {
             handler: async (argv) => await CLICommandFactory.executeGenerate(argv),
         };
     }
+    /**
+     * Execute provider status command
+     */
+    async executeProviderStatus(argv) {
+        if (argv.verbose && !argv.quiet) {
+            console.log(chalk.yellow("ℹ️ Verbose mode enabled. Displaying detailed status.\n"));
+        }
+        const spinner = argv.quiet
+            ? null
+            : ora("🔍 Checking AI provider status...\n").start();
+        try {
+            // Use SDK's provider diagnostic method instead of manual testing
+            const sdk = new NeuroLink();
+            const results = await sdk.getProviderStatus();
+            if (spinner) {
+                const working = results.filter((r) => r.status === "working").length;
+                const configured = results.filter((r) => r.configured).length;
+                spinner.succeed(`Provider check complete: ${working}/${configured} providers working`);
+            }
+            // Display results
+            for (const result of results) {
+                const status = result.status === "working"
+                    ? chalk.green("✅ Working")
+                    : result.status === "failed"
+                        ? chalk.red("❌ Failed")
+                        : chalk.gray("⚪ Not configured");
+                const time = result.responseTime ? ` (${result.responseTime}ms)` : "";
+                const model = result.model ? ` [${result.model}]` : "";
+                console.log(`${result.provider}: ${status}${time}${model}`);
+                if (argv.verbose && result.error) {
+                    console.log(`  Error: ${chalk.red(result.error)}`);
+                }
+            }
+            if (argv.verbose && !argv.quiet) {
+                console.log(chalk.blue("\n📋 Detailed Results:"));
+                console.log(JSON.stringify(results, null, 2));
+            }
+        }
+        catch (error) {
+            if (spinner) {
+                spinner.fail("Provider status check failed");
+            }
+            console.error(chalk.red("Error checking provider status:"), error);
+            process.exit(1);
+        }
+    }
     /**
      * Execute the generate command
      */
@@ -91,9 +138,10 @@ export class CLICommandFactory {
         const spinner = ora("Generating content...").start();
         try {
             const sdk = new NeuroLink();
+            const outputFormat = argv.outputFormat || argv.format || "text";
             const result = await sdk.generate({
                 input: { text: argv.input },
-                output: { format: argv.outputFormat },
+                output: { format: outputFormat },
                 provider: argv.provider,
                 model: argv.model,
                 temperature: argv.temperature,
@@ -104,8 +152,14 @@ export class CLICommandFactory {
                 enableEvaluation: argv.enableEvaluation,
             });
             spinner.succeed("Content generated successfully!");
-            console.log("\n" + chalk.cyan("Generated Content:"));
-            console.log(result.content);
+            // Handle different output formats
+            if (outputFormat === "json") {
+                console.log(JSON.stringify(result, null, 2));
+            }
+            else {
+                console.log("\n" + chalk.cyan("Generated Content:"));
+                console.log(result.content);
+            }
             if (argv.debug) {
                 console.log("\n" + chalk.yellow("Debug Information:"));
                 console.log("Provider:", result.provider);

package/dist/cli/index.js

@@ -22,7 +22,6 @@ import fs from "fs";
 import { addMCPCommands } from "./commands/mcp.js";
 import { addOllamaCommands } from "./commands/ollama.js";
 import { CLICommandFactory } from "./factories/command-factory.js";
-import { AgentEnhancedProvider } from "../lib/providers/agent-enhanced-provider.js";
 import { logger } from "../lib/utils/logger.js";
 /**
  * Helper functions for displaying analytics and evaluation results
@@ -193,6 +192,19 @@ function handleError(error, context) {
     }
     process.exit(1);
 }
+// Initialize MCP system for CLI with manual config enabled
+async function initializeCLI() {
+    // Import and configure for CLI mode
+    const { ProviderRegistry } = await import("../lib/factories/provider-registry.js");
+    const { initializeNeuroLinkMCP } = await import("../lib/mcp/initialize.js");
+    // Enable manual MCP only for CLI
+    ProviderRegistry.setOptions({
+        enableManualMCP: true,
+    });
+    // Initialize MCP system
+    await initializeNeuroLinkMCP();
+    logger.debug("CLI initialized with manual MCP support enabled");
+}
 // Initialize SDK
 const sdk = new NeuroLink();
 // Manual pre-validation for unknown flags
@@ -303,6 +315,7 @@ const cli = yargs(args)
         "openai",
         "bedrock",
         "vertex",
+        "google-vertex",
         "anthropic",
         "azure",
         "google-ai",
@@ -432,7 +445,8 @@ const cli = yargs(args)
     }
     // Command is now the primary generate method
     let originalConsole = {};
-    if (argv.format === "json" && !argv.quiet) {
+    if ((argv.format === "json" || argv.outputFormat === "json") &&
+        !argv.quiet) {
         // Suppress only if not quiet, as quiet implies no spinners anyway
         originalConsole = { ...console };
         Object.keys(originalConsole).forEach((key) => {
@@ -441,7 +455,7 @@ const cli = yargs(args)
             }
         });
     }
-    const spinner = argv.format === "json" || argv.quiet
+    const spinner = argv.outputFormat === "json" || argv.format === "json" || argv.quiet
         ? null
         : ora("🤖 Generating text...").start();
     try {
@@ -452,8 +466,6 @@ const cli = yargs(args)
                 reject(new Error(`CLI operation timed out after ${cliTimeout / 1000} seconds. Use --timeout to adjust.`));
             }, cliTimeout);
         });
-        // Use AgentEnhancedProvider when tools are enabled, otherwise use standard SDK
-        let generatePromise;
         // Parse context if provided
         let contextObj;
         if (argv.context) {
@@ -464,60 +476,26 @@ const cli = yargs(args)
                 throw new Error("Invalid JSON provided for --context option");
             }
         }
-        if (argv.disableTools === true) {
-            // Tools disabled - use standard SDK
-            generatePromise = sdk.generate({
-                input: { text: argv.prompt },
-                provider: argv.provider === "auto"
-                    ? undefined
-                    : argv.provider,
-                model: argv.model,
-                temperature: argv.temperature,
-                maxTokens: argv.maxTokens,
-                systemPrompt: argv.system,
-                timeout: argv.timeout,
-                // NEW: Analytics and evaluation support
-                enableAnalytics: argv.enableAnalytics,
-                enableEvaluation: argv.enableEvaluation,
-                context: contextObj,
-                // NEW: Lighthouse-compatible domain-aware evaluation
-                evaluationDomain: argv.evaluationDomain,
-                toolUsageContext: argv.toolUsageContext,
-            });
-        }
-        else {
-            // Tools enabled - use AgentEnhancedProvider for tool calling capabilities
-            // Map provider to supported AgentEnhancedProvider types
-            const supportedProvider = (() => {
-                switch (argv.provider) {
-                    case "openai":
-                    case "anthropic":
-                    case "google-ai":
-                        return argv.provider;
-                    case "auto":
-                    default:
-                        return "google-ai"; // Default to google-ai for best tool support
-                }
-            })();
-            const agentProvider = new AgentEnhancedProvider({
-                provider: supportedProvider,
-                model: argv.model, // Use specified model or default
-                toolCategory: "all", // Enable all tool categories
-            });
-            generatePromise = agentProvider.generate({
-                prompt: argv.prompt,
-                temperature: argv.temperature,
-                maxTokens: argv.maxTokens, // Respect user's token limit - no artificial caps
-                systemPrompt: argv.system,
-                // NEW: Analytics and evaluation support
-                enableAnalytics: argv.enableAnalytics,
-                enableEvaluation: argv.enableEvaluation,
-                context: contextObj,
-                // NEW: Lighthouse-compatible domain-aware evaluation
-                evaluationDomain: argv.evaluationDomain,
-                toolUsageContext: argv.toolUsageContext,
-            });
-        }
+        // Use standard SDK for all generation - tools are now built into BaseProvider
+        const generatePromise = sdk.generate({
+            input: { text: argv.prompt },
+            provider: argv.provider === "auto"
+                ? undefined
+                : argv.provider,
+            model: argv.model,
+            temperature: argv.temperature,
+            maxTokens: argv.maxTokens,
+            systemPrompt: argv.system,
+            timeout: argv.timeout,
+            disableTools: argv.disableTools === true, // Tools are enabled by default
+            // NEW: Analytics and evaluation support
+            enableAnalytics: argv.enableAnalytics,
+            enableEvaluation: argv.enableEvaluation,
+            context: contextObj,
+            // NEW: Lighthouse-compatible domain-aware evaluation
+            evaluationDomain: argv.evaluationDomain,
+            toolUsageContext: argv.toolUsageContext,
+        });
         // Wrap generation with master timeout to prevent infinite hangs
         const result = await Promise.race([generatePromise, timeoutPromise]);
         if (argv.format === "json" && originalConsole.log) {
@@ -533,7 +511,8 @@ const cli = yargs(args)
             completionTokens: 0,
             totalTokens: 0,
         };
-        if (argv.format === "json") {
+        if (argv.format === "json" || argv.outputFormat === "json") {
+            // CLI debug removed - analytics and evaluation now working correctly
             const jsonOutput = {
                 content: responseText,
                 provider: typedResult?.provider || argv.provider,
@@ -645,6 +624,7 @@ const cli = yargs(args)
         "openai",
         "bedrock",
         "vertex",
+        "google-vertex",
         "anthropic",
         "azure",
         "google-ai",
@@ -659,6 +639,10 @@ const cli = yargs(args)
     type: "number",
     default: 0.7,
     description: "Creativity level",
+})
+    .option("max-tokens", {
+    type: "number",
+    description: "Maximum number of tokens to generate",
 })
     .option("timeout", {
     type: "string",
@@ -744,77 +728,30 @@ const cli = yargs(args)
                 throw new Error("Invalid JSON provided for --context option");
             }
         }
-        let stream;
-        if (argv.disableTools === true) {
-            // Tools disabled - use standard SDK
-            stream = await sdk.stream({
-                input: { text: argv.prompt },
-                provider: argv.provider === "auto"
-                    ? undefined
-                    : argv.provider,
-                model: argv.model,
-                temperature: argv.temperature,
-                timeout: argv.timeout,
-                // NEW: Analytics and evaluation support
-                enableAnalytics: argv.enableAnalytics,
-                enableEvaluation: argv.enableEvaluation,
-                context: contextObj,
-            });
-        }
-        else {
-            // Tools enabled - use AgentEnhancedProvider for streaming tool calls
-            // Map provider to supported AgentEnhancedProvider types
-            const supportedProvider = (() => {
-                switch (argv.provider) {
-                    case "openai":
-                    case "anthropic":
-                    case "google-ai":
-                        return argv.provider;
-                    case "auto":
-                    default:
-                        return "google-ai"; // Default to google-ai for best tool support
-                }
-            })();
-            const agentProvider = new AgentEnhancedProvider({
-                provider: supportedProvider,
-                model: argv.model, // Use specified model or default
-                toolCategory: "all", // Enable all tool categories
-            });
-            // Note: AgentEnhancedProvider doesn't support streaming with tools yet
-            // Fall back to generate for now
-            const result = await agentProvider.generate({
-                prompt: argv.prompt,
-                temperature: argv.temperature,
-                // NEW: Analytics and evaluation support
-                enableAnalytics: argv.enableAnalytics,
-                enableEvaluation: argv.enableEvaluation,
-                context: contextObj,
-            });
-            // Simulate streaming by outputting the result
-            const text = result?.content || "";
-            const CHUNK_SIZE = 10;
-            const DELAY_MS = 50;
-            for (let i = 0; i < text.length; i += CHUNK_SIZE) {
-                process.stdout.write(text.slice(i, i + CHUNK_SIZE));
-                await new Promise((resolve) => setTimeout(resolve, DELAY_MS)); // Small delay
-            }
-            if (!argv.quiet) {
-                process.stdout.write("\n");
-            }
-            // Show analytics if enabled
-            // Show analytics and evaluation if enabled
-            displayAnalyticsAndEvaluation(result, argv);
-            return; // Exit early for agent mode
-        }
+        // Use standard SDK streaming - tools are handled automatically
+        const sdk = new NeuroLink();
+        const stream = await sdk.stream({
+            input: { text: argv.prompt },
+            provider: argv.provider === "auto"
+                ? undefined
+                : argv.provider,
+            model: argv.model,
+            temperature: argv.temperature,
+            timeout: argv.timeout,
+            disableTools: argv.disableTools === true, // Tools are enabled by default
+            // NEW: Analytics and evaluation support
+            enableAnalytics: argv.enableAnalytics,
+            enableEvaluation: argv.enableEvaluation,
+            context: contextObj,
+        });
+        // Process the stream
         for await (const chunk of stream.stream) {
             process.stdout.write(chunk.content);
-            // In debug mode, interleaved logging would appear here
-            // (SDK logs are controlled by NEUROLINK_DEBUG set in middleware)
         }
         if (!argv.quiet) {
             process.stdout.write("\n");
-        } // Ensure newline after stream
-        // Exit successfully
+        }
+        // Clean exit for tools-disabled streaming
         process.exit(0);
     }
     catch (error) {
@@ -844,6 +781,7 @@ const cli = yargs(args)
         "openai",
         "bedrock",
         "vertex",
+        "google-vertex",
         "anthropic",
         "azure",
         "google-ai",
@@ -983,6 +921,7 @@ const cli = yargs(args)
             "openai",
             "bedrock",
             "vertex",
+            "google-vertex",
             "anthropic",
             "azure",
             "google-ai",
@@ -1134,19 +1073,24 @@ const cli = yargs(args)
         .example("$0 provider status", "Check all providers");
 })
     // Status command alias
-    .command("status", "Check AI provider connectivity and performance (alias for provider status)", (yargsConfig) => yargsConfig.example("$0 status", "Quick provider status check"), async (argv) => {
-    // Simply redirect to provider status
-    process.argv = [
-        process.argv[0],
-        process.argv[1],
-        "provider",
-        "status",
-        ...process.argv.slice(3),
-    ];
-    const { hideBin } = await import("yargs/helpers");
-    const redirectedCli = yargs(hideBin(process.argv));
-    // Re-run with provider status
-    await cli.parse("provider status");
+    .command("status", "Check AI provider connectivity and performance (alias for provider status)", (yargsConfig) => yargsConfig
+    .usage("Usage: $0 status [options]")
+    .option("verbose", {
+    type: "boolean",
+    alias: "v",
+    description: "Show detailed information",
+})
+    .option("quiet", {
+    type: "boolean",
+    alias: "q",
+    description: "Suppress non-essential output",
+})
+    .example("$0 status", "Quick provider status check")
+    .example("$0 status --verbose", "Show detailed status information"), async (argv) => {
+    // Direct implementation instead of redirect to avoid recursion
+    const { CLICommandFactory } = await import("./factories/command-factory.js");
+    const commandFactory = new CLICommandFactory();
+    await commandFactory.executeProviderStatus(argv);
 })
     // Configuration Command Group
     .command("config <subcommand>", "Manage NeuroLink configuration", (yargsConfig) => {
@@ -1212,7 +1156,7 @@ const cli = yargs(args)
     .example("$0 get-best-provider --format json", "Show in JSON format"), async (argv) => {
     try {
         const { getBestProvider } = await import("../lib/utils/providerUtils-fixed.js");
-        const bestProvider = getBestProvider();
+        const bestProvider = await getBestProvider();
         if (argv.format === "json") {
             process.stdout.write(JSON.stringify({ provider: bestProvider }, null, 2) + "\n");
         }
@@ -1245,6 +1189,9 @@ addOllamaCommands(cli);
 // Execute CLI
 (async () => {
     try {
+        // Initialize CLI with manual MCP support
+        await initializeCLI();
+        // Parse and execute commands
         await cli.parse();
     }
     catch (error) {