@juspay/neurolink 1.5.1 → 1.5.2

package/dist/lib/providers/amazonBedrock.js

@@ -0,0 +1,256 @@
+import { createAmazonBedrock } from '@ai-sdk/amazon-bedrock';
+import { streamText, generateText, Output } from 'ai';
+import { logger } from '../utils/logger.js';
+// Default system context
+const DEFAULT_SYSTEM_CONTEXT = {
+    systemPrompt: 'You are a helpful AI assistant.'
+};
+// Configuration helpers
+const getBedrockModelId = () => {
+    return process.env.BEDROCK_MODEL ||
+        process.env.BEDROCK_MODEL_ID ||
+        'arn:aws:bedrock:us-east-2:225681119357:inference-profile/us.anthropic.claude-3-7-sonnet-20250219-v1:0';
+};
+const getAWSAccessKeyId = () => {
+    const keyId = process.env.AWS_ACCESS_KEY_ID;
+    if (!keyId) {
+        throw new Error('AWS_ACCESS_KEY_ID environment variable is not set');
+    }
+    return keyId;
+};
+const getAWSSecretAccessKey = () => {
+    const secretKey = process.env.AWS_SECRET_ACCESS_KEY;
+    if (!secretKey) {
+        throw new Error('AWS_SECRET_ACCESS_KEY environment variable is not set');
+    }
+    return secretKey;
+};
+const getAWSRegion = () => {
+    return process.env.AWS_REGION || 'us-east-2';
+};
+const getAWSSessionToken = () => {
+    return process.env.AWS_SESSION_TOKEN;
+};
+const getAppEnvironment = () => {
+    return process.env.PUBLIC_APP_ENVIRONMENT || 'dev';
+};
+// Amazon Bedrock class with enhanced error handling using createAmazonBedrock
+export class AmazonBedrock {
+    modelName;
+    model;
+    bedrock;
+    constructor(modelName) {
+        const functionTag = 'AmazonBedrock.constructor';
+        this.modelName = modelName || getBedrockModelId();
+        try {
+            logger.debug(`[${functionTag}] Function called`, {
+                modelName: this.modelName,
+                envBedrockModel: process.env.BEDROCK_MODEL,
+                envBedrockModelId: process.env.BEDROCK_MODEL_ID,
+                fallbackModel: 'arn:aws:bedrock:us-east-2:225681119357:inference-profile/us.anthropic.claude-3-7-sonnet-20250219-v1:0'
+            });
+            // Configure AWS credentials for custom Bedrock instance
+            const awsConfig = {
+                accessKeyId: getAWSAccessKeyId(),
+                secretAccessKey: getAWSSecretAccessKey(),
+                region: getAWSRegion()
+            };
+            logger.debug(`[${functionTag}] AWS config validation`, {
+                hasAccessKeyId: !!awsConfig.accessKeyId,
+                hasSecretAccessKey: !!awsConfig.secretAccessKey,
+                region: awsConfig.region || 'MISSING',
+                accessKeyIdLength: awsConfig.accessKeyId?.length || 0,
+                hasSessionToken: !!process.env.AWS_SESSION_TOKEN
+            });
+            // Add session token for development environment
+            if (getAppEnvironment() === 'dev') {
+                const sessionToken = getAWSSessionToken();
+                if (sessionToken) {
+                    awsConfig.sessionToken = sessionToken;
+                    logger.debug(`[${functionTag}] Session token added`, {
+                        environment: 'dev'
+                    });
+                }
+                else {
+                    logger.warn(`[${functionTag}] Session token missing`, {
+                        environment: 'dev'
+                    });
+                }
+            }
+            logger.debug(`[${functionTag}] AWS config created`, {
+                region: awsConfig.region,
+                hasSessionToken: !!awsConfig.sessionToken
+            });
+            logger.debug(`[${functionTag}] Bedrock provider creating`, {
+                modelName: this.modelName
+            });
+            // Create custom Bedrock provider instance with environment-based configuration
+            this.bedrock = createAmazonBedrock(awsConfig);
+            logger.debug(`[${functionTag}] Bedrock provider initialized`, {
+                modelName: this.modelName
+            });
+            logger.debug(`[${functionTag}] Model instance creating`, {
+                modelName: this.modelName
+            });
+            this.model = this.bedrock(this.modelName);
+            logger.debug(`[${functionTag}] Model instance created`, {
+                modelName: this.modelName
+            });
+            logger.debug(`[${functionTag}] Function result`, {
+                modelName: this.modelName,
+                region: awsConfig.region,
+                hasSessionToken: !!awsConfig.sessionToken,
+                success: true
+            });
+            logger.debug(`[${functionTag}] Initialization completed`, {
+                modelName: this.modelName,
+                region: awsConfig.region,
+                hasSessionToken: !!awsConfig.sessionToken
+            });
+        }
+        catch (err) {
+            logger.error(`[${functionTag}] Initialization failed`, {
+                message: 'Error in initializing Amazon Bedrock',
+                modelName: this.modelName,
+                region: getAWSRegion(),
+                error: err instanceof Error ? err.message : String(err),
+                stack: err instanceof Error ? err.stack : undefined
+            });
+            throw err;
+        }
+    }
+    async streamText(optionsOrPrompt, analysisSchema) {
+        const functionTag = 'AmazonBedrock.streamText';
+        const provider = 'bedrock';
+        let chunkCount = 0;
+        try {
+            // Parse parameters - support both string and options object
+            const options = typeof optionsOrPrompt === 'string'
+                ? { prompt: optionsOrPrompt }
+                : optionsOrPrompt;
+            const { prompt, temperature = 0.7, maxTokens = 500, systemPrompt = DEFAULT_SYSTEM_CONTEXT.systemPrompt, schema } = options;
+            // Use schema from options or fallback parameter
+            const finalSchema = schema || analysisSchema;
+            logger.debug(`[${functionTag}] Stream request started`, {
+                provider,
+                modelName: this.modelName,
+                promptLength: prompt.length,
+                temperature,
+                maxTokens
+            });
+            const streamOptions = {
+                model: this.model,
+                prompt: prompt,
+                system: systemPrompt,
+                temperature,
+                maxTokens,
+                onError: (event) => {
+                    const error = event.error;
+                    const errorMessage = error instanceof Error ? error.message : String(error);
+                    const errorStack = error instanceof Error ? error.stack : undefined;
+                    logger.error(`[${functionTag}] Stream text error`, {
+                        provider,
+                        modelName: this.modelName,
+                        region: getAWSRegion(),
+                        error: errorMessage,
+                        stack: errorStack,
+                        promptLength: prompt.length,
+                        chunkCount
+                    });
+                },
+                onFinish: (event) => {
+                    logger.debug(`[${functionTag}] Stream text finished`, {
+                        provider,
+                        modelName: this.modelName,
+                        region: getAWSRegion(),
+                        finishReason: event.finishReason,
+                        usage: event.usage,
+                        totalChunks: chunkCount,
+                        promptLength: prompt.length,
+                        responseLength: event.text?.length || 0
+                    });
+                },
+                onChunk: (event) => {
+                    chunkCount++;
+                    logger.debug(`[${functionTag}] Stream text chunk`, {
+                        provider,
+                        modelName: this.modelName,
+                        chunkNumber: chunkCount,
+                        chunkLength: event.chunk.text?.length || 0,
+                        chunkType: event.chunk.type
+                    });
+                }
+            };
+            if (finalSchema) {
+                streamOptions.experimental_output = Output.object({ schema: finalSchema });
+            }
+            // Direct streamText call - let the real error bubble up
+            const result = streamText(streamOptions);
+            logger.debug(`[${functionTag}] Stream text call successful`, {
+                provider,
+                modelName: this.modelName,
+                promptLength: prompt.length
+            });
+            return result;
+        }
+        catch (err) {
+            logger.error(`[${functionTag}] Exception`, {
+                provider,
+                modelName: this.modelName,
+                region: getAWSRegion(),
+                message: 'Error in streaming text',
+                err: String(err)
+            });
+            throw err; // Re-throw error to trigger fallback
+        }
+    }
+    async generateText(optionsOrPrompt, analysisSchema) {
+        const functionTag = 'AmazonBedrock.generateText';
+        const provider = 'bedrock';
+        try {
+            // Parse parameters - support both string and options object
+            const options = typeof optionsOrPrompt === 'string'
+                ? { prompt: optionsOrPrompt }
+                : optionsOrPrompt;
+            const { prompt, temperature = 0.7, maxTokens = 500, systemPrompt = DEFAULT_SYSTEM_CONTEXT.systemPrompt, schema } = options;
+            // Use schema from options or fallback parameter
+            const finalSchema = schema || analysisSchema;
+            logger.debug(`[${functionTag}] Generate text started`, {
+                provider,
+                modelName: this.modelName,
+                region: getAWSRegion(),
+                promptLength: prompt.length,
+                temperature,
+                maxTokens
+            });
+            const generateOptions = {
+                model: this.model,
+                prompt: prompt,
+                system: systemPrompt,
+                temperature,
+                maxTokens
+            };
+            if (finalSchema) {
+                generateOptions.experimental_output = Output.object({ schema: finalSchema });
+            }
+            const result = await generateText(generateOptions);
+            logger.debug(`[${functionTag}] Generate text completed`, {
+                provider,
+                modelName: this.modelName,
+                usage: result.usage,
+                finishReason: result.finishReason,
+                responseLength: result.text?.length || 0
+            });
+            return result;
+        }
+        catch (err) {
+            logger.error(`[${functionTag}] Exception`, {
+                provider,
+                modelName: this.modelName,
+                message: 'Error in generating text',
+                err: String(err)
+            });
+            throw err; // Re-throw error to trigger fallback instead of returning null
+        }
+    }
+}

package/dist/lib/providers/anthropic.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Anthropic AI Provider (Direct API)
+ *
+ * Direct integration with Anthropic's Claude models via their native API.
+ * Supports Claude 3.5 Sonnet, Claude 3.5 Haiku, and Claude 3 Opus.
+ */
+import type { AIProvider, TextGenerationOptions, StreamTextOptions } from '../core/types.js';
+import { AIProviderName } from '../core/types.js';
+export declare class AnthropicProvider implements AIProvider {
+    readonly name: AIProviderName;
+    private apiKey;
+    private baseURL;
+    private defaultModel;
+    constructor();
+    private getApiKey;
+    private getModel;
+    private makeRequest;
+    generateText(optionsOrPrompt: TextGenerationOptions | string, schema?: any): Promise<any>;
+    streamText(optionsOrPrompt: StreamTextOptions | string, schema?: any): Promise<any>;
+    private createAsyncIterable;
+    generateTextStream(optionsOrPrompt: StreamTextOptions | string): AsyncGenerator<any, void, unknown>;
+    testConnection(): Promise<{
+        success: boolean;
+        error?: string;
+        responseTime?: number;
+    }>;
+    isConfigured(): boolean;
+    getRequiredConfig(): string[];
+    getOptionalConfig(): string[];
+    getModels(): string[];
+    supportsStreaming(): boolean;
+    supportsSchema(): boolean;
+    getCapabilities(): string[];
+}

package/dist/lib/providers/anthropic.js

@@ -0,0 +1,308 @@
+/**
+ * Anthropic AI Provider (Direct API)
+ *
+ * Direct integration with Anthropic's Claude models via their native API.
+ * Supports Claude 3.5 Sonnet, Claude 3.5 Haiku, and Claude 3 Opus.
+ */
+import { AIProviderName } from '../core/types.js';
+import { logger } from '../utils/logger.js';
+export class AnthropicProvider {
+    name = AIProviderName.ANTHROPIC;
+    apiKey;
+    baseURL;
+    defaultModel;
+    constructor() {
+        this.apiKey = this.getApiKey();
+        this.baseURL = process.env.ANTHROPIC_BASE_URL || 'https://api.anthropic.com';
+        this.defaultModel = process.env.ANTHROPIC_MODEL || 'claude-3-5-sonnet-20241022';
+        logger.debug(`[AnthropicProvider] Initialized with model: ${this.defaultModel}`);
+    }
+    getApiKey() {
+        const apiKey = process.env.ANTHROPIC_API_KEY;
+        if (!apiKey) {
+            throw new Error('ANTHROPIC_API_KEY environment variable is required');
+        }
+        return apiKey;
+    }
+    getModel() {
+        return this.defaultModel;
+    }
+    async makeRequest(endpoint, body, stream = false) {
+        const url = `${this.baseURL}/v1/${endpoint}`;
+        const headers = {
+            'Content-Type': 'application/json',
+            'x-api-key': this.apiKey,
+            'anthropic-version': '2023-06-01',
+            'anthropic-dangerous-direct-browser-access': 'true' // Required for browser usage
+        };
+        logger.debug(`[AnthropicProvider.makeRequest] ${stream ? 'Streaming' : 'Non-streaming'} request to ${url}`);
+        logger.debug(`[AnthropicProvider.makeRequest] Model: ${body.model}, Max tokens: ${body.max_tokens}`);
+        const response = await fetch(url, {
+            method: 'POST',
+            headers,
+            body: JSON.stringify(body)
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            logger.error(`[AnthropicProvider.makeRequest] API error ${response.status}: ${errorText}`);
+            throw new Error(`Anthropic API error ${response.status}: ${errorText}`);
+        }
+        return response;
+    }
+    async generateText(optionsOrPrompt, schema) {
+        logger.debug('[AnthropicProvider.generateText] Starting text generation');
+        // Parse parameters with backward compatibility
+        const options = typeof optionsOrPrompt === 'string'
+            ? { prompt: optionsOrPrompt }
+            : optionsOrPrompt;
+        const { prompt, temperature = 0.7, maxTokens = 500, systemPrompt = 'You are Claude, an AI assistant created by Anthropic. You are helpful, harmless, and honest.' } = options;
+        logger.debug(`[AnthropicProvider.generateText] Prompt: "${prompt.substring(0, 100)}...", Temperature: ${temperature}, Max tokens: ${maxTokens}`);
+        const requestBody = {
+            model: this.getModel(),
+            max_tokens: maxTokens,
+            messages: [
+                {
+                    role: 'user',
+                    content: prompt
+                }
+            ],
+            temperature,
+            system: systemPrompt
+        };
+        try {
+            const response = await this.makeRequest('messages', requestBody);
+            const data = await response.json();
+            logger.debug(`[AnthropicProvider.generateText] Success. Generated ${data.usage.output_tokens} tokens`);
+            const content = data.content.map(block => block.text).join('');
+            return {
+                content,
+                provider: this.name,
+                model: data.model,
+                usage: {
+                    promptTokens: data.usage.input_tokens,
+                    completionTokens: data.usage.output_tokens,
+                    totalTokens: data.usage.input_tokens + data.usage.output_tokens
+                },
+                finishReason: data.stop_reason
+            };
+        }
+        catch (error) {
+            logger.error('[AnthropicProvider.generateText] Error:', error);
+            throw error;
+        }
+    }
+    async streamText(optionsOrPrompt, schema) {
+        logger.debug('[AnthropicProvider.streamText] Starting text streaming');
+        // Parse parameters with backward compatibility
+        const options = typeof optionsOrPrompt === 'string'
+            ? { prompt: optionsOrPrompt }
+            : optionsOrPrompt;
+        const { prompt, temperature = 0.7, maxTokens = 500, systemPrompt = 'You are Claude, an AI assistant created by Anthropic. You are helpful, harmless, and honest.' } = options;
+        logger.debug(`[AnthropicProvider.streamText] Streaming prompt: "${prompt.substring(0, 100)}..."`);
+        const requestBody = {
+            model: this.getModel(),
+            max_tokens: maxTokens,
+            messages: [
+                {
+                    role: 'user',
+                    content: prompt
+                }
+            ],
+            temperature,
+            system: systemPrompt,
+            stream: true
+        };
+        try {
+            const response = await this.makeRequest('messages', requestBody, true);
+            if (!response.body) {
+                throw new Error('No response body received');
+            }
+            // Return a StreamTextResult-like object
+            return {
+                textStream: this.createAsyncIterable(response.body),
+                text: '',
+                usage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 },
+                finishReason: 'end_turn'
+            };
+        }
+        catch (error) {
+            logger.error('[AnthropicProvider.streamText] Error:', error);
+            throw error;
+        }
+    }
+    async *createAsyncIterable(body) {
+        const reader = body.getReader();
+        const decoder = new TextDecoder();
+        let buffer = '';
+        try {
+            while (true) {
+                const { done, value } = await reader.read();
+                if (done)
+                    break;
+                buffer += decoder.decode(value, { stream: true });
+                const lines = buffer.split('\n');
+                buffer = lines.pop() || '';
+                for (const line of lines) {
+                    if (line.trim() === '')
+                        continue;
+                    if (line.startsWith('data: ')) {
+                        const data = line.slice(6);
+                        if (data.trim() === '[DONE]')
+                            continue;
+                        try {
+                            const chunk = JSON.parse(data);
+                            // Extract text content from different chunk types
+                            if (chunk.type === 'content_block_delta' && chunk.delta?.text) {
+                                yield chunk.delta.text;
+                            }
+                        }
+                        catch (parseError) {
+                            logger.warn('[AnthropicProvider.createAsyncIterable] Failed to parse chunk:', parseError);
+                            continue;
+                        }
+                    }
+                }
+            }
+        }
+        finally {
+            reader.releaseLock();
+        }
+    }
+    async *generateTextStream(optionsOrPrompt) {
+        logger.debug('[AnthropicProvider.generateTextStream] Starting text streaming');
+        // Parse parameters with backward compatibility
+        const options = typeof optionsOrPrompt === 'string'
+            ? { prompt: optionsOrPrompt }
+            : optionsOrPrompt;
+        const { prompt, temperature = 0.7, maxTokens = 500, systemPrompt = 'You are Claude, an AI assistant created by Anthropic. You are helpful, harmless, and honest.' } = options;
+        logger.debug(`[AnthropicProvider.generateTextStream] Streaming prompt: "${prompt.substring(0, 100)}..."`);
+        const requestBody = {
+            model: this.getModel(),
+            max_tokens: maxTokens,
+            messages: [
+                {
+                    role: 'user',
+                    content: prompt
+                }
+            ],
+            temperature,
+            system: systemPrompt,
+            stream: true
+        };
+        try {
+            const response = await this.makeRequest('messages', requestBody, true);
+            if (!response.body) {
+                throw new Error('No response body received');
+            }
+            const reader = response.body.getReader();
+            const decoder = new TextDecoder();
+            let buffer = '';
+            try {
+                while (true) {
+                    const { done, value } = await reader.read();
+                    if (done)
+                        break;
+                    buffer += decoder.decode(value, { stream: true });
+                    const lines = buffer.split('\n');
+                    buffer = lines.pop() || '';
+                    for (const line of lines) {
+                        if (line.trim() === '')
+                            continue;
+                        if (line.startsWith('data: ')) {
+                            const data = line.slice(6);
+                            if (data.trim() === '[DONE]')
+                                continue;
+                            try {
+                                const chunk = JSON.parse(data);
+                                // Extract text content from different chunk types
+                                if (chunk.type === 'content_block_delta' && chunk.delta?.text) {
+                                    yield {
+                                        content: chunk.delta.text,
+                                        provider: this.name,
+                                        model: this.getModel()
+                                    };
+                                }
+                            }
+                            catch (parseError) {
+                                logger.warn('[AnthropicProvider.generateTextStream] Failed to parse chunk:', parseError);
+                                continue;
+                            }
+                        }
+                    }
+                }
+            }
+            finally {
+                reader.releaseLock();
+            }
+            logger.debug('[AnthropicProvider.generateTextStream] Streaming completed');
+        }
+        catch (error) {
+            logger.error('[AnthropicProvider.generateTextStream] Error:', error);
+            throw error;
+        }
+    }
+    async testConnection() {
+        logger.debug('[AnthropicProvider.testConnection] Testing connection to Anthropic API');
+        const startTime = Date.now();
+        try {
+            await this.generateText({
+                prompt: 'Hello',
+                maxTokens: 5
+            });
+            const responseTime = Date.now() - startTime;
+            logger.debug(`[AnthropicProvider.testConnection] Connection test successful (${responseTime}ms)`);
+            return {
+                success: true,
+                responseTime
+            };
+        }
+        catch (error) {
+            const responseTime = Date.now() - startTime;
+            logger.error(`[AnthropicProvider.testConnection] Connection test failed (${responseTime}ms):`, error);
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : 'Unknown error',
+                responseTime
+            };
+        }
+    }
+    isConfigured() {
+        try {
+            this.getApiKey();
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    getRequiredConfig() {
+        return ['ANTHROPIC_API_KEY'];
+    }
+    getOptionalConfig() {
+        return ['ANTHROPIC_MODEL', 'ANTHROPIC_BASE_URL'];
+    }
+    getModels() {
+        return [
+            'claude-3-5-sonnet-20241022',
+            'claude-3-5-haiku-20241022',
+            'claude-3-opus-20240229',
+            'claude-3-sonnet-20240229',
+            'claude-3-haiku-20240307'
+        ];
+    }
+    supportsStreaming() {
+        return true;
+    }
+    supportsSchema() {
+        return false; // Anthropic doesn't have native JSON schema support like OpenAI
+    }
+    getCapabilities() {
+        return [
+            'text-generation',
+            'streaming',
+            'conversation',
+            'system-prompts',
+            'long-context' // Claude models support up to 200k tokens
+        ];
+    }
+}

package/dist/lib/providers/azureOpenAI.d.ts

@@ -0,0 +1,37 @@
+/**
+ * Azure OpenAI Provider
+ *
+ * Enterprise-grade OpenAI integration through Microsoft Azure.
+ * Supports all OpenAI models with enhanced security and compliance.
+ */
+import type { AIProvider, TextGenerationOptions, StreamTextOptions } from '../core/types.js';
+import { AIProviderName } from '../core/types.js';
+export declare class AzureOpenAIProvider implements AIProvider {
+    readonly name: AIProviderName;
+    private apiKey;
+    private endpoint;
+    private deploymentId;
+    private apiVersion;
+    constructor();
+    private getApiKey;
+    private getEndpoint;
+    private getDeploymentId;
+    private getApiUrl;
+    private makeRequest;
+    generateText(optionsOrPrompt: TextGenerationOptions | string, schema?: any): Promise<any>;
+    streamText(optionsOrPrompt: StreamTextOptions | string, schema?: any): Promise<any>;
+    private createAsyncIterable;
+    generateTextStream(optionsOrPrompt: StreamTextOptions | string): AsyncGenerator<any, void, unknown>;
+    testConnection(): Promise<{
+        success: boolean;
+        error?: string;
+        responseTime?: number;
+    }>;
+    isConfigured(): boolean;
+    getRequiredConfig(): string[];
+    getOptionalConfig(): string[];
+    getModels(): string[];
+    supportsStreaming(): boolean;
+    supportsSchema(): boolean;
+    getCapabilities(): string[];
+}