@juspay/neurolink 1.2.3 → 1.2.4

package/dist/core/factory.js

@@ -1,4 +1,4 @@
-import { GoogleVertexAI, AmazonBedrock, OpenAI } from '../providers/index.js';
+import { GoogleVertexAI, AmazonBedrock, OpenAI, AnthropicProvider, AzureOpenAIProvider } from '../providers/index.js';
 import { getBestProvider } from '../utils/providerUtils.js';
 const componentIdentifier = 'aiProviderFactory';
 /**
@@ -34,8 +34,16 @@ export class AIProviderFactory {
                 case 'gpt':
                     provider = new OpenAI(modelName);
                     break;
+                case 'anthropic':
+                case 'claude':
+                    provider = new AnthropicProvider();
+                    break;
+                case 'azure':
+                case 'azure-openai':
+                    provider = new AzureOpenAIProvider();
+                    break;
                 default:
-                    throw new Error(`Unknown provider: ${providerName}. Supported providers: vertex, bedrock, openai`);
+                    throw new Error(`Unknown provider: ${providerName}. Supported providers: vertex, bedrock, openai, anthropic, azure`);
             }
             console.log(`[${functionTag}] Provider creation succeeded`, {
                 providerName,

package/dist/core/types.d.ts

@@ -6,7 +6,9 @@ import type { StreamTextResult, ToolSet, Schema, GenerateTextResult } from 'ai';
 export declare enum AIProviderName {
     BEDROCK = "bedrock",
     OPENAI = "openai",
-    VERTEX = "vertex"
+    VERTEX = "vertex",
+    ANTHROPIC = "anthropic",
+    AZURE = "azure"
 }
 /**
  * Supported Models for Amazon Bedrock

package/dist/core/types.js

@@ -6,6 +6,8 @@ export var AIProviderName;
     AIProviderName["BEDROCK"] = "bedrock";
     AIProviderName["OPENAI"] = "openai";
     AIProviderName["VERTEX"] = "vertex";
+    AIProviderName["ANTHROPIC"] = "anthropic";
+    AIProviderName["AZURE"] = "azure";
 })(AIProviderName || (AIProviderName = {}));
 /**
  * Supported Models for Amazon Bedrock

package/dist/index.d.ts

@@ -10,7 +10,7 @@ import { AIProviderFactory } from './core/factory.js';
 export { AIProviderFactory };
 export type { AIProvider, AIProviderName, ProviderConfig, StreamingOptions, ProviderAttempt, SupportedModelName } from './core/types.js';
 export { BedrockModels, OpenAIModels, VertexModels, DEFAULT_PROVIDER_CONFIGS } from './core/types.js';
-export { GoogleVertexAI, AmazonBedrock, OpenAI } from './providers/index.js';
+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 { getBestProvider, getAvailableProviders, isValidProvider } from './utils/providerUtils.js';

package/dist/index.js

@@ -12,7 +12,7 @@ export { AIProviderFactory };
 // Model enums
 export { BedrockModels, OpenAIModels, VertexModels, DEFAULT_PROVIDER_CONFIGS } from './core/types.js';
 // Provider exports
-export { GoogleVertexAI, AmazonBedrock, OpenAI } from './providers/index.js';
+export { GoogleVertexAI, AmazonBedrock, OpenAI, AnthropicProvider, AzureOpenAIProvider } from './providers/index.js';
 export { PROVIDERS, AVAILABLE_PROVIDERS } from './providers/index.js';
 // Utility exports
 export { getBestProvider, getAvailableProviders, isValidProvider } from './utils/providerUtils.js';

package/dist/neurolink.d.ts

@@ -7,7 +7,7 @@
 import type { AIProviderName } from './core/types.js';
 export interface TextGenerationOptions {
     prompt: string;
-    provider?: 'openai' | 'bedrock' | 'vertex' | 'auto';
+    provider?: 'openai' | 'bedrock' | 'vertex' | 'anthropic' | 'azure' | 'auto';
     temperature?: number;
     maxTokens?: number;
     systemPrompt?: string;
@@ -15,7 +15,7 @@ export interface TextGenerationOptions {
 }
 export interface StreamTextOptions {
     prompt: string;
-    provider?: 'openai' | 'bedrock' | 'vertex' | 'auto';
+    provider?: 'openai' | 'bedrock' | 'vertex' | 'anthropic' | 'azure' | 'auto';
     temperature?: number;
     maxTokens?: number;
     systemPrompt?: string;

package/dist/neurolink.js

@@ -34,7 +34,7 @@ export class NeuroLink {
             }
             const responseTime = Date.now() - startTime;
             return {
-                content: result.text,
+                content: result.content || result.text || '',
                 provider: providerName,
                 usage: result.usage,
                 responseTime

package/dist/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/providers/anthropic.js

@@ -0,0 +1,307 @@
+/**
+ * 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';
+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';
+        console.log(`[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
+        };
+        console.log(`[AnthropicProvider.makeRequest] ${stream ? 'Streaming' : 'Non-streaming'} request to ${url}`);
+        console.log(`[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();
+            console.error(`[AnthropicProvider.makeRequest] API error ${response.status}: ${errorText}`);
+            throw new Error(`Anthropic API error ${response.status}: ${errorText}`);
+        }
+        return response;
+    }
+    async generateText(optionsOrPrompt, schema) {
+        console.log('[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;
+        console.log(`[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();
+            console.log(`[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) {
+            console.error('[AnthropicProvider.generateText] Error:', error);
+            throw error;
+        }
+    }
+    async streamText(optionsOrPrompt, schema) {
+        console.log('[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;
+        console.log(`[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) {
+            console.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) {
+                            console.warn('[AnthropicProvider.createAsyncIterable] Failed to parse chunk:', parseError);
+                            continue;
+                        }
+                    }
+                }
+            }
+        }
+        finally {
+            reader.releaseLock();
+        }
+    }
+    async *generateTextStream(optionsOrPrompt) {
+        console.log('[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;
+        console.log(`[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) {
+                                console.warn('[AnthropicProvider.generateTextStream] Failed to parse chunk:', parseError);
+                                continue;
+                            }
+                        }
+                    }
+                }
+            }
+            finally {
+                reader.releaseLock();
+            }
+            console.log('[AnthropicProvider.generateTextStream] Streaming completed');
+        }
+        catch (error) {
+            console.error('[AnthropicProvider.generateTextStream] Error:', error);
+            throw error;
+        }
+    }
+    async testConnection() {
+        console.log('[AnthropicProvider.testConnection] Testing connection to Anthropic API');
+        const startTime = Date.now();
+        try {
+            await this.generateText({
+                prompt: 'Hello',
+                maxTokens: 5
+            });
+            const responseTime = Date.now() - startTime;
+            console.log(`[AnthropicProvider.testConnection] Connection test successful (${responseTime}ms)`);
+            return {
+                success: true,
+                responseTime
+            };
+        }
+        catch (error) {
+            const responseTime = Date.now() - startTime;
+            console.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/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[];
+}