@juspay/neurolink 5.0.0 → 5.2.0

package/dist/providers/agent-enhanced-provider.js

@@ -1,614 +0,0 @@
-/**
- * Agent-Enhanced Provider for NeuroLink CLI
- * Integrates direct tools with AI providers for true agent functionality
- */
-import { generateText, streamText, tool, } from "ai";
-import { google } from "@ai-sdk/google";
-import { openai } from "@ai-sdk/openai";
-import { anthropic } from "@ai-sdk/anthropic";
-import { directAgentTools, getToolsForCategory, } from "../agent/direct-tools.js";
-import { StreamingEnhancer, StreamingMonitor, } from "../utils/streaming-utils.js";
-import { UnifiedMCPSystem } from "../mcp/unified-mcp.js";
-import { mcpLogger } from "../mcp/logging.js";
-import { parseTimeout } from "../utils/timeout.js";
-import { evaluateResponse } from "../core/evaluation.js";
-import { createAnalytics } from "../core/analytics.js";
-import { logger } from "../utils/logger.js";
-/**
- * Agent-Enhanced Provider Class
- * Provides AI generation with tool calling capabilities
- * Now includes MCP tool integration alongside direct tools
- */
-export class AgentEnhancedProvider {
-    config;
-    model;
-    resolvedModelName = "default";
-    mcpSystem = null;
-    mcpInitialized = false;
-    mcpInitializing = false;
-    mcpInitFailed = false;
-    constructor(config) {
-        this.config = {
-            maxSteps: 5,
-            toolCategory: "all",
-            enableTools: true,
-            enableMCP: true,
-            mcpDiscoveryOptions: {
-                autoDiscover: true,
-                searchPaths: [process.cwd()],
-                configFiles: [".mcp-config.json", ".mcp-servers.json"],
-            },
-            ...config,
-        };
-        // Initialize the AI model based on provider and store resolved model name
-        this.model = this.createModel();
-        // Initialize MCP registry if enabled
-        if (this.config.enableMCP) {
-            this.initializeMCP();
-        }
-    }
-    createModel() {
-        const { provider, model } = this.config;
-        switch (provider) {
-            case "google-ai":
-                this.resolvedModelName =
-                    model || process.env.GOOGLE_AI_MODEL || "gemini-2.5-flash";
-                return google(this.resolvedModelName);
-            case "openai":
-                this.resolvedModelName = model || process.env.OPENAI_MODEL || "gpt-4o";
-                return openai(this.resolvedModelName);
-            case "anthropic":
-                this.resolvedModelName =
-                    model || process.env.ANTHROPIC_MODEL || "claude-3-5-sonnet-20241022";
-                return anthropic(this.resolvedModelName);
-            default:
-                throw new Error(`Unsupported provider: ${provider}`);
-        }
-    }
-    /**
-     * Initialize MCP registry with auto-discovery
-     */
-    async initializeMCP() {
-        if (this.mcpInitializing || this.mcpInitFailed) {
-            return;
-        }
-        this.mcpInitializing = true;
-        try {
-            mcpLogger.info("[AgentEnhancedProvider] Initializing MCP integration...");
-            this.mcpSystem = new UnifiedMCPSystem({
-                configPath: this.config.mcpDiscoveryOptions?.configFiles?.[0] ||
-                    ".mcp-config.json",
-                enableExternalServers: true,
-                enableInternalServers: true,
-                autoInitialize: false,
-            });
-            // ADD TIMEOUT to prevent hanging forever
-            const initPromise = this.mcpSystem.initialize();
-            const timeoutPromise = new Promise((_, reject) => setTimeout(() => reject(new Error("MCP initialization timeout after 15 seconds")), this.config.mcpInitTimeoutMs || 15000));
-            await Promise.race([initPromise, timeoutPromise]);
-            this.mcpInitialized = true;
-            mcpLogger.info("[AgentEnhancedProvider] MCP integration initialized successfully");
-        }
-        catch (error) {
-            mcpLogger.error("[AgentEnhancedProvider] Failed to initialize MCP:", error);
-            this.mcpSystem = null;
-            this.mcpInitialized = false;
-            this.mcpInitFailed = true;
-            // Don't throw - continue with direct tools only
-        }
-        finally {
-            this.mcpInitializing = false;
-        }
-    }
-    /**
-     * Get combined tools: direct tools + MCP tools
-     */
-    async getCombinedTools() {
-        const directTools = this.config.enableTools
-            ? getToolsForCategory(this.config.toolCategory)
-            : {};
-        // If MCP is disabled or failed to initialize, return only direct tools
-        if (!this.config.enableMCP || !this.mcpSystem) {
-            return directTools;
-        }
-        // Get MCP tools if available
-        const mcpTools = {};
-        try {
-            // Skip if MCP failed to initialize or is still initializing
-            if (this.mcpInitFailed ||
-                this.mcpInitializing ||
-                !this.mcpInitialized ||
-                !this.mcpSystem) {
-                return directTools;
-            }
-            const mcpToolInfos = await this.mcpSystem.listTools();
-            // Convert MCP tools to AI SDK format
-            for (const toolInfo of mcpToolInfos) {
-                const toolKey = `mcp_${toolInfo.name}`;
-                mcpTools[toolKey] = {
-                    description: toolInfo.description || `MCP tool: ${toolInfo.name}`,
-                    parameters: toolInfo.inputSchema || {},
-                    execute: async (args) => {
-                        let timeoutId;
-                        try {
-                            // Create timeout controller for tool execution if configured
-                            const toolTimeout = this.config.toolExecutionTimeout;
-                            const toolAbortController = toolTimeout
-                                ? new AbortController()
-                                : undefined;
-                            if (toolAbortController && toolTimeout) {
-                                const timeoutMs = typeof toolTimeout === "string"
-                                    ? parseTimeout(toolTimeout)
-                                    : toolTimeout;
-                                timeoutId = setTimeout(() => {
-                                    toolAbortController.abort();
-                                }, timeoutMs);
-                            }
-                            const context = {
-                                sessionId: "cli-session",
-                                userId: "cli-user",
-                                secureFS: {
-                                    readFile: async (path, encoding) => {
-                                        const fs = await import("fs/promises");
-                                        return encoding
-                                            ? fs.readFile(path, {
-                                                encoding: encoding,
-                                            })
-                                            : fs.readFile(path);
-                                    },
-                                    writeFile: async (path, content) => {
-                                        const fs = await import("fs/promises");
-                                        await fs.writeFile(path, content);
-                                    },
-                                    readdir: async (path) => {
-                                        const fs = await import("fs/promises");
-                                        return fs.readdir(path);
-                                    },
-                                    stat: async (path) => {
-                                        const fs = await import("fs/promises");
-                                        return fs.stat(path);
-                                    },
-                                    mkdir: async (path, options) => {
-                                        const fs = await import("fs/promises");
-                                        await fs.mkdir(path, options);
-                                    },
-                                    exists: async (path) => {
-                                        const fs = await import("fs/promises");
-                                        try {
-                                            await fs.access(path);
-                                            return true;
-                                        }
-                                        catch {
-                                            return false;
-                                        }
-                                    },
-                                },
-                                path: {
-                                    join: (...paths) => {
-                                        const path = require("path");
-                                        return path.join(...paths);
-                                    },
-                                    resolve: (...paths) => {
-                                        const path = require("path");
-                                        return path.resolve(...paths);
-                                    },
-                                    relative: (from, to) => {
-                                        const path = require("path");
-                                        return path.relative(from, to);
-                                    },
-                                    dirname: (path) => {
-                                        const pathLib = require("path");
-                                        return pathLib.dirname(path);
-                                    },
-                                    basename: (path, ext) => {
-                                        const pathLib = require("path");
-                                        return pathLib.basename(path, ext);
-                                    },
-                                },
-                                grantedPermissions: ["read", "write", "execute"],
-                                log: (level, message, data) => {
-                                    const logFn = mcpLogger[level];
-                                    if (typeof logFn === "function") {
-                                        if (data) {
-                                            logFn(`${message} ${JSON.stringify(data)}`);
-                                        }
-                                        else {
-                                            logFn(message);
-                                        }
-                                    }
-                                },
-                            };
-                            const toolPromise = this.mcpSystem.executeTool(toolInfo.name, args, context);
-                            let result;
-                            if (toolAbortController) {
-                                // Race between tool execution and timeout
-                                result = await Promise.race([
-                                    toolPromise,
-                                    new Promise((_, reject) => {
-                                        toolAbortController.signal.addEventListener("abort", () => {
-                                            reject(new Error(`Tool ${toolInfo.name} timed out after ${this.config.toolExecutionTimeout}`));
-                                        });
-                                    }),
-                                ]);
-                            }
-                            else {
-                                result = await toolPromise;
-                            }
-                            // Clear timeout if successful
-                            if (timeoutId) {
-                                clearTimeout(timeoutId);
-                            }
-                            return result.data || result;
-                        }
-                        catch (error) {
-                            // Clear timeout on error
-                            if (timeoutId) {
-                                clearTimeout(timeoutId);
-                            }
-                            mcpLogger.error(`MCP tool ${toolInfo.name} execution failed:`, error);
-                            throw error;
-                        }
-                    },
-                };
-            }
-            mcpLogger.info(`[AgentEnhancedProvider] Loaded ${Object.keys(mcpTools).length} MCP tools`);
-        }
-        catch (error) {
-            mcpLogger.error("[AgentEnhancedProvider] Failed to load MCP tools:", error);
-        }
-        return { ...directTools, ...mcpTools };
-    }
-    async generateText(optionsOrPrompt) {
-        const startTime = Date.now();
-        const options = typeof optionsOrPrompt === "string"
-            ? { prompt: optionsOrPrompt }
-            : optionsOrPrompt;
-        const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt, schema, timeout, } = options;
-        // Get combined tools (direct + MCP) if enabled
-        const tools = this.config.enableTools ? await this.getCombinedTools() : {};
-        const log = (msg, data) => {
-            mcpLogger.info(`[AgentEnhancedProvider] ${msg}`, data ? JSON.stringify(data, null, 2) : "");
-        };
-        log("Starting text generation", {
-            prompt: prompt.substring(0, 100),
-            toolsCount: Object.keys(tools).length,
-            maxSteps: this.config.maxSteps,
-        });
-        try {
-            // Parse timeout if provided
-            let abortSignal;
-            if (timeout) {
-                const timeoutMs = typeof timeout === "string" ? parseTimeout(timeout) : timeout;
-                if (timeoutMs !== undefined) {
-                    abortSignal = AbortSignal.timeout(timeoutMs);
-                }
-            }
-            // The AI SDK with maxSteps automatically handles tool calling and result integration
-            const result = await generateText({
-                model: this.model,
-                prompt: systemPrompt
-                    ? `System: ${systemPrompt}\n\nUser: ${prompt}`
-                    : prompt,
-                tools,
-                maxSteps: this.config.maxSteps, // This enables automatic tool calling
-                temperature,
-                maxTokens,
-                toolChoice: this.shouldForceToolUsage(prompt) ? "required" : "auto",
-                abortSignal, // Pass abort signal for timeout support
-            });
-            log("Generation completed", {
-                text: result.text?.substring(0, 200),
-                finishReason: result.finishReason,
-                toolCallsCount: result.toolCalls?.length || 0,
-                toolResultsCount: result.toolResults?.length || 0,
-                stepsCount: result.steps?.length || 0,
-            });
-            // Check if tools were called but no final text was generated
-            if (result.finishReason === "tool-calls" &&
-                !result.text &&
-                result.toolResults?.length > 0) {
-                log("Tools called but no final text generated, creating summary response");
-                try {
-                    // Extract tool results and create a summary prompt
-                    let toolResultsSummary = "";
-                    if (result.toolResults) {
-                        for (const toolResult of result.toolResults) {
-                            const resultData = toolResult.result || toolResult;
-                            // Try to extract meaningful data from the result
-                            if (typeof resultData === "object" && resultData !== null) {
-                                if (resultData.success && resultData.items) {
-                                    // This looks like a filesystem listing
-                                    toolResultsSummary += `Directory listing for ${resultData.path}:\n`;
-                                    for (const item of resultData.items) {
-                                        toolResultsSummary += `- ${item.name} (${item.type})\n`;
-                                    }
-                                }
-                                else {
-                                    toolResultsSummary += JSON.stringify(resultData, null, 2);
-                                }
-                            }
-                            else {
-                                toolResultsSummary += String(resultData);
-                            }
-                            toolResultsSummary += "\n\n";
-                        }
-                    }
-                    log("Tool results extracted", {
-                        summaryLength: toolResultsSummary.length,
-                        preview: toolResultsSummary.substring(0, 200),
-                    });
-                    // Create a simple, direct summary
-                    const finalText = `Based on the user request "${prompt}", here's what I found:\n\n${toolResultsSummary}`;
-                    log("Final text created", {
-                        textLength: finalText.length,
-                        preview: finalText.substring(0, 200),
-                    });
-                    // Return result with the formatted text
-                    return {
-                        ...result,
-                        text: finalText,
-                        finishReason: "stop",
-                    };
-                }
-                catch (error) {
-                    log("Error in summary generation", {
-                        error: error instanceof Error ? error.message : String(error),
-                    });
-                    // Fallback: return raw tool results
-                    const fallbackText = `Tool execution completed. Raw results: ${JSON.stringify(result.toolResults, null, 2)}`;
-                    return {
-                        ...result,
-                        text: fallbackText,
-                        finishReason: "stop",
-                    };
-                }
-            }
-            // Add analytics if enabled
-            if (options.enableAnalytics) {
-                result.analytics = createAnalytics(this.config.provider, this.resolvedModelName, result, Date.now() - startTime, options.context);
-            }
-            // Add evaluation if enabled
-            if (options.enableEvaluation) {
-                result.evaluation = await evaluateResponse(prompt, result.text, options.context);
-            }
-            // Return the full result - the AI SDK has already handled tool execution and integration
-            return result;
-        }
-        catch (error) {
-            console.error("[AgentEnhancedProvider] generateText error:", error);
-            throw error;
-        }
-    }
-    async streamText(optionsOrPrompt) {
-        const options = typeof optionsOrPrompt === "string"
-            ? { prompt: optionsOrPrompt }
-            : optionsOrPrompt;
-        const { prompt, temperature = 0.7, maxTokens = 1000, systemPrompt, timeout, 
-        // Phase 2: Enhanced streaming options
-        enableProgressTracking, progressCallback, includeStreamingMetadata, streamingBufferSize, enableStreamingHeaders, customStreamingConfig, } = options;
-        // Phase 2.1: Setup streaming enhancements
-        const streamId = `agent_stream_${Date.now()}`;
-        const streamingConfig = StreamingEnhancer.createStreamingConfig(options);
-        if (enableProgressTracking) {
-            StreamingMonitor.registerStream(streamId);
-        }
-        // Get combined tools (direct + MCP) if enabled
-        const tools = this.config.enableTools ? await this.getCombinedTools() : {};
-        try {
-            // Parse timeout if provided
-            let abortSignal;
-            if (timeout) {
-                const timeoutMs = typeof timeout === "string" ? parseTimeout(timeout) : timeout;
-                if (timeoutMs !== undefined) {
-                    abortSignal = AbortSignal.timeout(timeoutMs);
-                }
-            }
-            const result = await streamText({
-                model: this.model,
-                prompt: systemPrompt
-                    ? `System: ${systemPrompt}\n\nUser: ${prompt}`
-                    : prompt,
-                tools,
-                maxSteps: this.config.maxSteps,
-                temperature,
-                maxTokens,
-                toolChoice: this.shouldForceToolUsage(prompt) ? "required" : "auto",
-                abortSignal, // Pass abort signal for timeout support
-            });
-            // Phase 2.1: Apply streaming enhancements if enabled
-            if (streamingConfig.progressTracking && result.textStream) {
-                const enhancedCallback = streamingConfig.callback
-                    ? (progress) => {
-                        StreamingMonitor.updateStream(streamId, progress);
-                        streamingConfig.callback(progress);
-                        if (progress.phase === "complete") {
-                            StreamingMonitor.completeStream(streamId);
-                        }
-                    }
-                    : undefined;
-                // Enhance the stream with progress tracking
-                const enhancedStream = StreamingEnhancer.addProgressTracking(result.textStream, enhancedCallback, { streamId, bufferSize: streamingConfig.bufferSize });
-                // Return enhanced result with tracking
-                return {
-                    ...result,
-                    textStream: enhancedStream,
-                    // Phase 2.1: Add streaming metadata
-                    streamingMetadata: streamingConfig.metadata
-                        ? {
-                            streamId,
-                            provider: this.getProviderName(),
-                            model: this.getModelName(),
-                            enabledFeatures: {
-                                progressTracking: true,
-                                metadata: streamingConfig.metadata,
-                                headers: streamingConfig.headers,
-                            },
-                        }
-                        : undefined,
-                };
-            }
-            return result;
-        }
-        catch (error) {
-            console.error("[AgentEnhancedProvider] streamText error:", error);
-            throw error;
-        }
-    }
-    /**
-     * Determine if we should force tool usage based on prompt patterns
-     */
-    shouldForceToolUsage(prompt) {
-        const forceToolPatterns = [
-            /what time is it/i,
-            /current time/i,
-            /list files/i,
-            /read file/i,
-            /directory/i,
-            /calculate/i,
-            /math/i,
-            /search for/i,
-            /find files/i,
-        ];
-        return forceToolPatterns.some((pattern) => pattern.test(prompt));
-    }
-    getCapabilities() {
-        return [
-            "text-generation",
-            "streaming",
-            "tool-calling",
-            "agent-functionality",
-        ];
-    }
-    getProviderName() {
-        return `agent-${this.config.provider}`;
-    }
-    getModelName() {
-        return this.config.model || `default-${this.config.provider}-model`;
-    }
-    /**
-     * Test agent functionality
-     */
-    async testAgentCapabilities() {
-        const testPrompts = [
-            "What time is it right now?",
-            "List files in current directory",
-            "Calculate 15 * 7",
-            "What is the square root of 144?",
-        ];
-        const results = [];
-        let successCount = 0;
-        for (const prompt of testPrompts) {
-            try {
-                logger.debug(`Testing: "${prompt}"`);
-                const result = await this.generateText(prompt);
-                if (!result) {
-                    results.push({
-                        prompt,
-                        success: false,
-                        error: "No result returned from generateText",
-                    });
-                    logger.warn(`❌ No result returned`);
-                    continue;
-                }
-                const toolsCalled = result.toolCalls?.length || 0;
-                const success = toolsCalled > 0;
-                if (success) {
-                    successCount++;
-                }
-                results.push({
-                    prompt,
-                    success,
-                    toolsCalled,
-                    response: result.text.substring(0, 100) + "...",
-                });
-                logger.debug(`✅ Tools called: ${toolsCalled}, Response: ${result.text.substring(0, 50)}...`);
-            }
-            catch (error) {
-                results.push({
-                    prompt,
-                    success: false,
-                    error: error instanceof Error ? error.message : String(error),
-                });
-                logger.error(`❌ Error: ${error}`);
-            }
-        }
-        return {
-            success: successCount > 0,
-            results,
-        };
-    }
-    /**
-     * Create agent-enhanced provider factory
-     */
-    static createAgent(config) {
-        return new AgentEnhancedProvider(config);
-    }
-    /**
-     * Create multiple agent providers for comparison
-     */
-    static createMultiProviderAgents() {
-        const providers = {};
-        // Only create providers that have API keys configured
-        if (process.env.GOOGLE_AI_API_KEY) {
-            providers["google-ai"] = new AgentEnhancedProvider({
-                provider: "google-ai",
-            });
-        }
-        if (process.env.OPENAI_API_KEY) {
-            providers["openai"] = new AgentEnhancedProvider({ provider: "openai" });
-        }
-        if (process.env.ANTHROPIC_API_KEY) {
-            providers["anthropic"] = new AgentEnhancedProvider({
-                provider: "anthropic",
-            });
-        }
-        return providers;
-    }
-    /**
-     * Alias for generateText() - CLI-SDK consistency
-     */
-    async generate(optionsOrPrompt, analysisSchema) {
-        return this.generateText(optionsOrPrompt);
-    }
-    /**
-     * Short alias for generateText() - CLI-SDK consistency
-     */
-    async gen(optionsOrPrompt, analysisSchema) {
-        return this.generateText(optionsOrPrompt);
-    }
-}
-/**
- * Helper function to create agent provider
- */
-export function createAgentProvider(provider, options) {
-    return new AgentEnhancedProvider({
-        provider,
-        ...options,
-    });
-}
-/**
- * Test all available agent providers
- */
-export async function testAllAgentProviders() {
-    logger.info("🧪 Testing All Agent Providers\n");
-    const providers = AgentEnhancedProvider.createMultiProviderAgents();
-    if (Object.keys(providers).length === 0) {
-        logger.warn("❌ No API keys found. Please configure at least one provider.");
-        return;
-    }
-    for (const [name, provider] of Object.entries(providers)) {
-        logger.info(`\n🔬 Testing ${name.toUpperCase()} Agent Provider:`);
-        try {
-            const testResult = await provider.testAgentCapabilities();
-            if (testResult.success) {
-                logger.info(`✅ ${name} agent provider working correctly`);
-            }
-            else {
-                logger.warn(`❌ ${name} agent provider failed tests`);
-            }
-        }
-        catch (error) {
-            logger.error(`❌ ${name} provider error:`, error);
-        }
-    }
-}

package/dist/providers/amazonBedrock.d.ts

@@ -1,19 +0,0 @@
-import type { ZodType, ZodTypeDef } from "zod";
-import { type StreamTextResult, type ToolSet, type Schema, type GenerateTextResult } from "ai";
-import type { AIProvider, TextGenerationOptions, StreamTextOptions, EnhancedGenerateTextResult } from "../core/types.js";
-export declare class AmazonBedrock implements AIProvider {
-    private modelName;
-    private model;
-    private bedrock;
-    constructor(modelName?: string | null);
-    streamText(optionsOrPrompt: StreamTextOptions | string, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<StreamTextResult<ToolSet, unknown> | null>;
-    generateText(optionsOrPrompt: TextGenerationOptions | string, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<GenerateTextResult<ToolSet, unknown> | null>;
-    /**
-     * Alias for generateText() - CLI-SDK consistency
-     */
-    generate(optionsOrPrompt: TextGenerationOptions | string, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<EnhancedGenerateTextResult | null>;
-    /**
-     * Short alias for generateText() - CLI-SDK consistency
-     */
-    gen(optionsOrPrompt: TextGenerationOptions | string, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<EnhancedGenerateTextResult | null>;
-}