@juspay/neurolink 4.1.1 → 4.2.0

package/dist/cli/index.js

@@ -291,12 +291,11 @@ const cli = yargs(args)
     exitProcess(); // Default exit
 })
     // Generate Text Command
-    .command(["generate-text <prompt>", "generate <prompt>", "gen <prompt>"], "Generate text using AI providers", (yargsInstance) => yargsInstance
-    .usage("Usage: $0 generate-text <prompt> [options]")
+    .command(["generate-text [prompt]", "generate [prompt]", "gen [prompt]"], "Generate text using AI providers", (yargsInstance) => yargsInstance
+    .usage("Usage: $0 generate-text [prompt] [options]")
     .positional("prompt", {
     type: "string",
-    description: "Text prompt for AI generation",
-    demandOption: true,
+    description: "Text prompt for AI generation (or read from stdin)",
 })
     .option("provider", {
     choices: [
@@ -384,6 +383,53 @@ const cli = yargs(args)
     .example('$0 generate-text "Write a story" --provider openai', "Use specific provider")
     .example('$0 generate-text "What time is it?"', "Use with natural tool integration (default)")
     .example('$0 generate-text "Hello world" --disable-tools', "Use without tool integration"), async (argv) => {
+    // SOLUTION 1: Handle stdin input if no prompt provided
+    if (!argv.prompt && !process.stdin.isTTY) {
+        // Read from stdin
+        let stdinData = "";
+        process.stdin.setEncoding("utf8");
+        for await (const chunk of process.stdin) {
+            stdinData += chunk;
+        }
+        argv.prompt = stdinData.trim();
+        if (!argv.prompt) {
+            throw new Error("No input received from stdin");
+        }
+    }
+    else if (!argv.prompt) {
+        throw new Error('Prompt required. Use: neurolink generate "your prompt" or echo "prompt" | neurolink generate');
+    }
+    // SOLUTION 2: Parameter validation
+    const errors = [];
+    // Validate max-tokens
+    if (argv.maxTokens !== undefined) {
+        if (!Number.isInteger(argv.maxTokens) || argv.maxTokens < 1) {
+            errors.push(`max-tokens must be a positive integer >= 1, got: ${argv.maxTokens}`);
+        }
+        if (argv.maxTokens > 100000) {
+            errors.push(`max-tokens too large (>100000), got: ${argv.maxTokens}`);
+        }
+    }
+    // Validate temperature
+    if (argv.temperature !== undefined) {
+        if (typeof argv.temperature !== "number" ||
+            argv.temperature < 0 ||
+            argv.temperature > 1) {
+            errors.push(`temperature must be between 0.0 and 1.0, got: ${argv.temperature}`);
+        }
+    }
+    // Validate timeout
+    if (argv.timeout !== undefined) {
+        if (!Number.isInteger(argv.timeout) || argv.timeout < 1) {
+            errors.push(`timeout must be a positive integer >= 1 second, got: ${argv.timeout}`);
+        }
+        if (argv.timeout > 600) {
+            errors.push(`timeout too large (>600s), got: ${argv.timeout}s`);
+        }
+    }
+    if (errors.length > 0) {
+        throw new Error(`Parameter validation failed:\n${errors.map((e) => `  • ${e}`).join("\n")}\n\nUse --help for valid parameter ranges.`);
+    }
     // Check if generate-text was used specifically (for deprecation warning)
     const usedCommand = argv._[0];
     if (usedCommand === "generate-text" && !argv.quiet) {
@@ -591,12 +637,11 @@ const cli = yargs(args)
     }
 })
     // Stream Text Command
-    .command("stream <prompt>", "Stream text generation in real-time", (yargsInstance) => yargsInstance
-    .usage("Usage: $0 stream <prompt> [options]")
+    .command("stream [prompt]", "Stream text generation in real-time", (yargsInstance) => yargsInstance
+    .usage("Usage: $0 stream [prompt] [options]")
     .positional("prompt", {
     type: "string",
-    description: "Text prompt for streaming",
-    demandOption: true,
+    description: "Text prompt for streaming (or read from stdin)",
 })
     .option("provider", {
     choices: [
@@ -668,6 +713,22 @@ const cli = yargs(args)
     .example('$0 stream "Tell me a story"', "Stream a story in real-time")
     .example('$0 stream "What time is it?"', "Stream with natural tool integration (default)")
     .example('$0 stream "Tell me a story" --disable-tools', "Stream without tool integration"), async (argv) => {
+    // SOLUTION 1: Handle stdin input if no prompt provided
+    if (!argv.prompt && !process.stdin.isTTY) {
+        // Read from stdin
+        let stdinData = "";
+        process.stdin.setEncoding("utf8");
+        for await (const chunk of process.stdin) {
+            stdinData += chunk;
+        }
+        argv.prompt = stdinData.trim();
+        if (!argv.prompt) {
+            throw new Error("No input received from stdin");
+        }
+    }
+    else if (!argv.prompt) {
+        throw new Error('Prompt required. Use: neurolink stream "your prompt" or echo "prompt" | neurolink stream');
+    }
     // Default mode: Simple streaming message
     // Debug mode: More detailed information
     if (!argv.quiet && !argv.debug) {

package/dist/core/factory.js

@@ -1,5 +1,5 @@
 import { GoogleVertexAI, AmazonBedrock, OpenAI, AnthropicProvider, AzureOpenAIProvider, GoogleAIStudio, HuggingFace, Ollama, MistralAI, } from "../providers/index.js";
-import { getBestProviderSync } from "../utils/providerUtils.js";
+import { getBestProvider } from "../utils/providerUtils.js";
 import { logger } from "../utils/logger.js";
 import { dynamicModelProvider } from "./dynamic-models.js";
 const componentIdentifier = "aiProviderFactory";
@@ -240,7 +240,7 @@ export class AIProviderFactory {
     static async createBestProvider(requestedProvider, modelName, enableMCP = true) {
         const functionTag = "AIProviderFactory.createBestProvider";
         try {
-            const bestProvider = getBestProviderSync(requestedProvider);
+            const bestProvider = await getBestProvider(requestedProvider);
             logger.debug(`[${functionTag}] Best provider selected`, {
                 requestedProvider: requestedProvider || "auto",
                 selectedProvider: bestProvider,

package/dist/index.d.ts

@@ -13,7 +13,7 @@ export { BedrockModels, OpenAIModels, VertexModels, DEFAULT_PROVIDER_CONFIGS, }
 export { GoogleVertexAI, AmazonBedrock, OpenAI, AnthropicProvider, AzureOpenAIProvider, } from "./providers/index.js";
 export type { ProviderName } from "./providers/index.js";
 export { PROVIDERS, AVAILABLE_PROVIDERS } from "./providers/index.js";
-export { getBestProviderSync as getBestProvider, getAvailableProviders, isValidProvider, } from "./utils/providerUtils.js";
+export { getBestProvider, getAvailableProviders, isValidProvider, } from "./utils/providerUtils.js";
 export { NeuroLink } from "./neurolink.js";
 export type { TextGenerationOptions, StreamTextOptions, TextGenerationResult, } from "./neurolink.js";
 export declare const VERSION = "1.0.0";

package/dist/index.js

@@ -15,7 +15,7 @@ export { BedrockModels, OpenAIModels, VertexModels, DEFAULT_PROVIDER_CONFIGS, }
 export { GoogleVertexAI, AmazonBedrock, OpenAI, AnthropicProvider, AzureOpenAIProvider, } from "./providers/index.js";
 export { PROVIDERS, AVAILABLE_PROVIDERS } from "./providers/index.js";
 // Utility exports
-export { getBestProviderSync as getBestProvider, getAvailableProviders, isValidProvider, } from "./utils/providerUtils.js";
+export { getBestProvider, getAvailableProviders, isValidProvider, } from "./utils/providerUtils.js";
 // Main NeuroLink wrapper class
 export { NeuroLink } from "./neurolink.js";
 // Version

package/dist/lib/core/factory.js

@@ -1,5 +1,5 @@
 import { GoogleVertexAI, AmazonBedrock, OpenAI, AnthropicProvider, AzureOpenAIProvider, GoogleAIStudio, HuggingFace, Ollama, MistralAI, } from "../providers/index.js";
-import { getBestProviderSync } from "../utils/providerUtils.js";
+import { getBestProvider } from "../utils/providerUtils.js";
 import { logger } from "../utils/logger.js";
 const componentIdentifier = "aiProviderFactory";
 /**
@@ -239,7 +239,7 @@ export class AIProviderFactory {
     static async createBestProvider(requestedProvider, modelName, enableMCP = true) {
         const functionTag = "AIProviderFactory.createBestProvider";
         try {
-            const bestProvider = getBestProviderSync(requestedProvider);
+            const bestProvider = await getBestProvider(requestedProvider);
             logger.debug(`[${functionTag}] Best provider selected`, {
                 requestedProvider: requestedProvider || "auto",
                 selectedProvider: bestProvider,

package/dist/lib/index.d.ts

@@ -13,7 +13,7 @@ export { BedrockModels, OpenAIModels, VertexModels, DEFAULT_PROVIDER_CONFIGS, }
 export { GoogleVertexAI, AmazonBedrock, OpenAI, AnthropicProvider, AzureOpenAIProvider, } from "./providers/index.js";
 export type { ProviderName } from "./providers/index.js";
 export { PROVIDERS, AVAILABLE_PROVIDERS } from "./providers/index.js";
-export { getBestProviderSync as getBestProvider, getAvailableProviders, isValidProvider, } from "./utils/providerUtils.js";
+export { getBestProvider, getAvailableProviders, isValidProvider, } from "./utils/providerUtils.js";
 export { NeuroLink } from "./neurolink.js";
 export type { TextGenerationOptions, StreamTextOptions, TextGenerationResult, } from "./neurolink.js";
 export declare const VERSION = "1.0.0";

package/dist/lib/index.js

@@ -15,7 +15,7 @@ export { BedrockModels, OpenAIModels, VertexModels, DEFAULT_PROVIDER_CONFIGS, }
 export { GoogleVertexAI, AmazonBedrock, OpenAI, AnthropicProvider, AzureOpenAIProvider, } from "./providers/index.js";
 export { PROVIDERS, AVAILABLE_PROVIDERS } from "./providers/index.js";
 // Utility exports
-export { getBestProviderSync as getBestProvider, getAvailableProviders, isValidProvider, } from "./utils/providerUtils.js";
+export { getBestProvider, getAvailableProviders, isValidProvider, } from "./utils/providerUtils.js";
 // Main NeuroLink wrapper class
 export { NeuroLink } from "./neurolink.js";
 // Version

package/dist/lib/mcp/context-manager.d.ts

@@ -79,6 +79,12 @@ export declare class ContextManager {
      * @param updates Partial context updates
      */
     updateContext(sessionId: string, updates: Partial<NeuroLinkExecutionContext>): void;
+    /**
+     * Store context directly (used when session ID changes)
+     *
+     * @param context Complete execution context to store
+     */
+    storeContext(context: NeuroLinkExecutionContext): void;
     /**
      * Remove context from active tracking
      *

package/dist/lib/mcp/context-manager.js

@@ -196,6 +196,14 @@ export class ContextManager {
             this.activeContexts.set(sessionId, updatedContext);
         }
     }
+    /**
+     * Store context directly (used when session ID changes)
+     *
+     * @param context Complete execution context to store
+     */
+    storeContext(context) {
+        this.activeContexts.set(context.sessionId, context);
+    }
     /**
      * Remove context from active tracking
      *

package/dist/lib/mcp/contracts/mcpContract.d.ts

@@ -66,6 +66,7 @@ export interface McpMetadata {
     author?: string;
     homepage?: string;
     repository?: string;
+    category?: string;
 }
 /**
  * Tool definition schema

package/dist/lib/mcp/external-client.js

@@ -40,6 +40,10 @@ export class ExternalMCPClient extends EventEmitter {
                 stdio: ["pipe", "pipe", "pipe"],
                 env: { ...process.env, ...this.config.env },
             });
+            // 🔧 FIX: Register for cleanup
+            process.once("beforeExit", () => this.disconnect());
+            process.once("SIGINT", () => this.disconnect());
+            process.once("SIGTERM", () => this.disconnect());
             if (!this.process.stdout || !this.process.stdin || !this.process.stderr) {
                 throw new Error("Failed to create stdio pipes");
             }
@@ -279,7 +283,7 @@ export class ExternalMCPClient extends EventEmitter {
             pending.reject(new Error("Connection closed"));
         }
         this.pendingRequests.clear();
-        // Close process
+        // 🔧 FIX: Enhanced process cleanup
         if (this.process) {
             this.process.kill("SIGTERM");
             // Force kill after timeout
@@ -287,7 +291,7 @@ export class ExternalMCPClient extends EventEmitter {
                 if (this.process && !this.process.killed) {
                     this.process.kill("SIGKILL");
                 }
-            }, 5000);
+            }, 1000); // Reduced from 5000ms to 1000ms
             this.process = null;
         }
         this.isConnected = false;

package/dist/lib/mcp/initialize.d.ts

@@ -3,10 +3,11 @@
  * Automatically registers built-in NeuroLink servers with the default registry
  * Ensures built-in tools are always available without manual configuration
  */
+import { type MCPToolRegistry } from "./tool-registry.js";
 /**
  * Initialize NeuroLink MCP system by registering built-in servers
  */
-export declare function initializeNeuroLinkMCP(): Promise<void>;
+export declare function initializeNeuroLinkMCP(targetRegistry?: MCPToolRegistry): Promise<void>;
 /**
  * Get initialization status
  */

package/dist/lib/mcp/initialize.js

@@ -3,13 +3,13 @@
  * Automatically registers built-in NeuroLink servers with the default registry
  * Ensures built-in tools are always available without manual configuration
  */
-import { toolRegistry, defaultToolRegistry } from "./tool-registry.js";
+import { toolRegistry, defaultToolRegistry, } from "./tool-registry.js";
 import { mcpLogger } from "./logging.js";
 let isInitialized = false;
 /**
  * Initialize NeuroLink MCP system by registering built-in servers
  */
-export async function initializeNeuroLinkMCP() {
+export async function initializeNeuroLinkMCP(targetRegistry) {
     if (isInitialized) {
         return;
     }
@@ -18,14 +18,15 @@ export async function initializeNeuroLinkMCP() {
         // Import utility server dynamically to avoid circular dependencies
         // Note: AI core server temporarily disabled due to circular dependency issues
         const { utilityServer } = await import("./servers/utilities/utility-server.js");
-        // Register built-in NeuroLink servers with default registry
-        await toolRegistry.registerServer(utilityServer.id, utilityServer);
-        mcpLogger.debug("Registered neurolink-utility server with built-in tools");
+        // Register built-in NeuroLink servers with specified registry (or default)
+        const registry = targetRegistry || toolRegistry;
+        await registry.registerServer(utilityServer.id, utilityServer);
+        mcpLogger.debug(`Registered neurolink-utility server with built-in tools in ${targetRegistry ? "target" : "default"} registry`);
         // TODO: Re-enable AI core server once circular dependencies are resolved
         // const { aiCoreServer } = await import('./servers/ai-providers/ai-core-server.js');
-        // await toolRegistry.registerServer(aiCoreServer.id, aiCoreServer);
+        // await registry.registerServer(aiCoreServer.id, aiCoreServer);
         // mcpLogger.debug('Registered neurolink-ai-core server with AI tools');
-        const stats = await toolRegistry.getStats();
+        const stats = await registry.getStats();
         mcpLogger.info(`Initialization complete: ${stats.totalServers} server, ${stats.totalTools} tools available`);
         isInitialized = true;
     }

package/dist/lib/mcp/orchestrator.js

@@ -70,7 +70,14 @@ export class MCPOrchestrator {
             // Create new session with options
             session = await this.sessionManager.createSession(context, options.sessionOptions);
             // Update context with new session ID
+            const oldSessionId = context.sessionId;
             context.sessionId = session.id;
+            // Remove old context and store updated context in context manager
+            if (oldSessionId) {
+                this.contextManager.removeContext(oldSessionId);
+            }
+            // Store the updated context with the new session ID
+            this.contextManager.storeContext(context);
         }
         if (process.env.NEUROLINK_DEBUG === "true") {
             console.log(`[Orchestrator] Executing tool '${toolName}' in session ${context.sessionId}`);
@@ -80,6 +87,8 @@ export class MCPOrchestrator {
         const semaphoreKey = `tool:${toolName}`;
         const semaphoreResult = await this.semaphoreManager.acquire(semaphoreKey, async () => {
             try {
+                // Add tool to the execution chain
+                this.contextManager.addToToolChain(context, toolName);
                 // Execute tool through registry
                 const result = await this.registry.executeTool(toolName, params, context);
                 if (process.env.NEUROLINK_DEBUG === "true") {

package/dist/lib/mcp/registry.d.ts

@@ -15,7 +15,7 @@ export interface McpRegistry {
  * Maintains backward compatibility with existing code
  */
 export declare class MCPRegistry implements McpRegistry {
-    private plugins;
+    plugins: Map<string, DiscoveredMcp<Record<string, unknown>>>;
     /**
      * Register a plugin
      */

package/dist/lib/mcp/servers/ai-providers/ai-analysis-tools.js

@@ -5,7 +5,7 @@
  */
 import { z } from "zod";
 import { AIProviderFactory } from "../../../core/factory.js";
-import { getBestProviderSync as getBestProvider, getAvailableProviders, } from "../../../utils/providerUtils.js";
+import { getBestProvider, getAvailableProviders, } from "../../../utils/providerUtils.js";
 /**
  * Input Schemas for AI Analysis Tools
  */

package/dist/lib/mcp/servers/ai-providers/ai-core-server.js

@@ -6,7 +6,7 @@
 import { z } from "zod";
 import { createMCPServer } from "../../factory.js";
 import { AIProviderFactory } from "../../../core/factory.js";
-import { getBestProviderSync as getBestProvider, getAvailableProviders, } from "../../../utils/providerUtils.js";
+import { getBestProvider, getAvailableProviders, } from "../../../utils/providerUtils.js";
 import { logger } from "../../../utils/logger.js";
 import { analyzeAIUsageTool, benchmarkProviderPerformanceTool, optimizePromptParametersTool, } from "./ai-analysis-tools.js";
 import { generateTestCasesTool, refactorCodeTool, generateDocumentationTool, debugAIOutputTool, } from "./ai-workflow-tools.js";
@@ -85,7 +85,7 @@ aiCoreServer.registerTool({
         try {
             logger.debug(`[AI-Core] Starting text generation: "${params.prompt.substring(0, 50)}..."`);
             // Use existing AIProviderFactory with best provider selection
-            const selectedProvider = params.provider || getBestProvider(params.provider);
+            const selectedProvider = params.provider || (await getBestProvider(params.provider));
             const provider = await AIProviderFactory.createBestProvider(selectedProvider);
             // Generate text using existing NeuroLink patterns
             const result = await provider.generateText({
@@ -157,7 +157,7 @@ aiCoreServer.registerTool({
             logger.debug(`[AI-Core] Selecting provider with requirements:`, params.requirements);
             // Use existing provider selection logic
             const availableProviders = getAvailableProviders();
-            const selectedProvider = getBestProvider(params.preferred);
+            const selectedProvider = await getBestProvider(params.preferred);
             // Get provider capabilities
             const getProviderCapabilities = (provider) => ({
                 multimodal: provider === "openai" ||

package/dist/lib/mcp/servers/ai-providers/ai-workflow-tools.d.ts

@@ -30,13 +30,13 @@ export declare const workflowToolSchemas: {
         includeAsyncTests: z.ZodDefault<z.ZodBoolean>;
     }, "strip", z.ZodTypeAny, {
         codeFunction: string;
-        testTypes: ("integration" | "unit" | "performance" | "edge-cases" | "security")[];
+        testTypes: ("integration" | "performance" | "unit" | "edge-cases" | "security")[];
         framework: "jest" | "mocha" | "vitest" | "pytest" | "unittest" | "rspec";
         coverageTarget: number;
         includeAsyncTests: boolean;
     }, {
         codeFunction: string;
-        testTypes?: ("integration" | "unit" | "performance" | "edge-cases" | "security")[] | undefined;
+        testTypes?: ("integration" | "performance" | "unit" | "edge-cases" | "security")[] | undefined;
         framework?: "jest" | "mocha" | "vitest" | "pytest" | "unittest" | "rspec" | undefined;
         coverageTarget?: number | undefined;
         includeAsyncTests?: boolean | undefined;

package/dist/lib/mcp/servers/ai-providers/ai-workflow-tools.js

@@ -4,7 +4,7 @@
  */
 import { z } from "zod";
 import { AIProviderFactory } from "../../../core/factory.js";
-import { getBestProviderSync as getBestProvider } from "../../../utils/providerUtils.js";
+import { getBestProvider } from "../../../utils/providerUtils.js";
 import { DEFAULT_MAX_TOKENS } from "../../../core/constants.js";
 // Tool-specific schemas with comprehensive validation
 const generateTestCasesSchema = z.object({

package/dist/lib/mcp/session-manager.js

@@ -94,7 +94,7 @@ export class SessionManager {
      * @returns Session if found and not expired
      */
     getSession(sessionId, extend = true) {
-        let session = this.sessions.get(sessionId) || null;
+        const session = this.sessions.get(sessionId) || null;
         // For sync compatibility, skip disk loading in getSession
         // Use async getSessionAsync() for full persistence features
         if (!session) {

package/dist/lib/mcp/session-persistence.js

@@ -60,7 +60,7 @@ export class SessionPersistence {
             ...session,
             state: Array.from(session.state.entries()),
         };
-        let fileData = {
+        const fileData = {
             session: sessionData,
             checksum: "",
             version: "1.0",

package/dist/lib/mcp/tool-registry.d.ts

@@ -20,23 +20,37 @@ export interface ToolExecutionOptions {
 }
 export declare class MCPToolRegistry extends MCPRegistry {
     private tools;
+    private toolImpls;
     private toolExecutionStats;
     /**
      * Register a server with its tools (updated signature)
      */
-    registerServer(serverId: string, serverConfig?: unknown, context?: ExecutionContext): Promise<void>;
+    registerServer(serverOrId: string | any, serverConfig?: unknown, context?: ExecutionContext): Promise<void>;
     /**
      * Execute a tool with enhanced context
      */
     executeTool<T = unknown>(toolName: string, args?: unknown, context?: ExecutionContext): Promise<T>;
     /**
-     * List all available tools (updated signature)
+     * List all available tools (updated signature with filtering)
      */
-    listTools(context?: ExecutionContext): Promise<ToolInfo[]>;
+    listTools(): Promise<ToolInfo[]>;
+    listTools(context: ExecutionContext): Promise<ToolInfo[]>;
+    listTools(filter: {
+        category?: string;
+        serverId?: string;
+        serverCategory?: string;
+        permissions?: string[];
+        context?: ExecutionContext;
+    }): Promise<ToolInfo[]>;
     /**
-     * Get tool information
+     * Get tool information with server details
      */
-    getToolInfo(toolName: string): ToolInfo | undefined;
+    getToolInfo(toolName: string): {
+        tool: ToolInfo;
+        server: {
+            id: string;
+        };
+    } | undefined;
     /**
      * Update execution statistics
      */
@@ -70,13 +84,19 @@ export declare class MCPToolRegistry extends MCPRegistry {
      */
     getToolCount(): number;
     /**
-     * Get statistics (alias for getExecutionStats)
+     * Get comprehensive statistics
      */
-    getStats(): Record<string, {
-        count: number;
-        averageTime: number;
-        totalTime: number;
-    }>;
+    getStats(): {
+        totalServers: number;
+        totalTools: number;
+        serversByCategory: Record<string, number>;
+        toolsByCategory: Record<string, number>;
+        executionStats: Record<string, {
+            count: number;
+            averageTime: number;
+            totalTime: number;
+        }>;
+    };
     /**
      * Unregister a server
      */