@juspay/neurolink 5.0.0 → 5.2.0

package/dist/providers/azureOpenAI.js

@@ -1,437 +0,0 @@
-/**
- * Azure OpenAI Provider
- *
- * Enterprise-grade OpenAI integration through Microsoft Azure.
- * Supports all OpenAI models with enhanced security and compliance.
- */
-import { AIProviderName } from "../core/types.js";
-import { logger } from "../utils/logger.js";
-import { createTimeoutController, TimeoutError, getDefaultTimeout, } from "../utils/timeout.js";
-import { DEFAULT_MAX_TOKENS } from "../core/constants.js";
-import { evaluateResponse } from "../core/evaluation.js";
-import { createAnalytics } from "../core/analytics.js";
-export class AzureOpenAIProvider {
-    name = AIProviderName.AZURE;
-    apiKey;
-    endpoint;
-    deploymentId;
-    apiVersion;
-    constructor() {
-        this.apiKey = this.getApiKey();
-        this.endpoint = this.getEndpoint();
-        this.deploymentId = this.getDeploymentId();
-        this.apiVersion =
-            process.env.AZURE_OPENAI_API_VERSION || "2024-02-15-preview";
-        logger.debug(`[AzureOpenAIProvider] Initialized with endpoint: ${this.endpoint}, deployment: ${this.deploymentId}`);
-    }
-    getApiKey() {
-        const apiKey = process.env.AZURE_OPENAI_API_KEY;
-        if (!apiKey) {
-            throw new Error("AZURE_OPENAI_API_KEY environment variable is required");
-        }
-        return apiKey;
-    }
-    getEndpoint() {
-        const endpoint = process.env.AZURE_OPENAI_ENDPOINT;
-        if (!endpoint) {
-            throw new Error("AZURE_OPENAI_ENDPOINT environment variable is required");
-        }
-        return endpoint.replace(/\/$/, ""); // Remove trailing slash
-    }
-    getDeploymentId() {
-        const deploymentId = process.env.AZURE_OPENAI_DEPLOYMENT_ID;
-        if (!deploymentId) {
-            throw new Error("AZURE_OPENAI_DEPLOYMENT_ID environment variable is required");
-        }
-        return deploymentId;
-    }
-    getApiUrl(stream = false) {
-        return `${this.endpoint}/openai/deployments/${this.deploymentId}/chat/completions?api-version=${this.apiVersion}`;
-    }
-    async makeRequest(body, stream = false, signal) {
-        const url = this.getApiUrl(stream);
-        const headers = {
-            "Content-Type": "application/json",
-            "api-key": this.apiKey,
-        };
-        logger.debug(`[AzureOpenAIProvider.makeRequest] ${stream ? "Streaming" : "Non-streaming"} request to deployment: ${this.deploymentId}`);
-        logger.debug(`[AzureOpenAIProvider.makeRequest] Max tokens: ${body.max_tokens || "default"}, Temperature: ${body.temperature || "default"}`);
-        const response = await fetch(url, {
-            method: "POST",
-            headers,
-            body: JSON.stringify(body),
-            signal, // Add abort signal for timeout support
-        });
-        if (!response.ok) {
-            const errorText = await response.text();
-            logger.error(`[AzureOpenAIProvider.makeRequest] API error ${response.status}: ${errorText}`);
-            throw new Error(`Azure OpenAI API error ${response.status}: ${errorText}`);
-        }
-        return response;
-    }
-    async generateText(optionsOrPrompt, schema) {
-        const functionTag = "AzureOpenAIProvider.generateText";
-        const provider = "azure";
-        const startTime = Date.now();
-        logger.debug(`[${functionTag}] Starting text generation`);
-        // Parse parameters with backward compatibility
-        const options = typeof optionsOrPrompt === "string"
-            ? { prompt: optionsOrPrompt }
-            : optionsOrPrompt;
-        const { prompt, temperature = 0.7, maxTokens = DEFAULT_MAX_TOKENS, systemPrompt = "You are a helpful AI assistant.", timeout = getDefaultTimeout(provider, "generate"), } = options;
-        logger.debug(`[${functionTag}] Prompt: "${prompt.substring(0, 100)}...", Temperature: ${temperature}, Max tokens: ${maxTokens}, Timeout: ${timeout}`);
-        const messages = [];
-        if (systemPrompt) {
-            messages.push({
-                role: "system",
-                content: systemPrompt,
-            });
-        }
-        messages.push({
-            role: "user",
-            content: prompt,
-        });
-        const requestBody = {
-            messages,
-            temperature,
-            max_tokens: maxTokens,
-        };
-        // Create timeout controller if timeout is specified
-        const timeoutController = createTimeoutController(timeout, provider, "generate");
-        try {
-            const response = await this.makeRequest(requestBody, false, timeoutController?.controller.signal);
-            const data = await response.json();
-            // Clean up timeout if successful
-            timeoutController?.cleanup();
-            logger.debug(`[${functionTag}] Success. Generated ${data.usage.completion_tokens} tokens`);
-            const content = data.choices[0]?.message?.content || "";
-            const result = {
-                content,
-                provider: this.name,
-                model: data.model,
-                usage: {
-                    promptTokens: data.usage.prompt_tokens,
-                    completionTokens: data.usage.completion_tokens,
-                    totalTokens: data.usage.total_tokens,
-                },
-                finishReason: data.choices[0]?.finish_reason || "stop",
-            };
-            // Add analytics if enabled
-            if (options.enableAnalytics) {
-                result.analytics = {
-                    provider: this.name,
-                    model: data.model,
-                    tokens: result.usage,
-                    responseTime: Date.now() - startTime,
-                    context: options.context,
-                };
-            }
-            // Add evaluation if enabled
-            if (options.enableEvaluation) {
-                result.evaluation = await evaluateResponse(options.prompt, content, options.context);
-            }
-            return result;
-        }
-        catch (error) {
-            // Always cleanup timeout
-            timeoutController?.cleanup();
-            // Log timeout errors specifically
-            if (error instanceof TimeoutError) {
-                logger.error(`[${functionTag}] Timeout error`, {
-                    provider,
-                    timeout: error.timeout,
-                    message: error.message,
-                });
-            }
-            else if (error?.name === "AbortError") {
-                // Convert AbortError to TimeoutError
-                const timeoutError = new TimeoutError(`${provider} generate operation timed out after ${timeout}`, timeoutController?.timeoutMs || 0, provider, "generate");
-                logger.error(`[${functionTag}] Timeout error`, {
-                    provider,
-                    timeout: timeoutController?.timeoutMs,
-                    message: timeoutError.message,
-                });
-                throw timeoutError;
-            }
-            else {
-                logger.error(`[${functionTag}] Error:`, error);
-            }
-            throw error;
-        }
-    }
-    async streamText(optionsOrPrompt, schema) {
-        const functionTag = "AzureOpenAIProvider.streamText";
-        const provider = "azure";
-        logger.debug(`[${functionTag}] Starting text streaming`);
-        // Parse parameters with backward compatibility
-        const options = typeof optionsOrPrompt === "string"
-            ? { prompt: optionsOrPrompt }
-            : optionsOrPrompt;
-        const { prompt, temperature = 0.7, maxTokens = DEFAULT_MAX_TOKENS, systemPrompt = "You are a helpful AI assistant.", timeout = getDefaultTimeout(provider, "stream"), } = options;
-        logger.debug(`[${functionTag}] Streaming prompt: "${prompt.substring(0, 100)}...", Timeout: ${timeout}`);
-        const messages = [];
-        if (systemPrompt) {
-            messages.push({
-                role: "system",
-                content: systemPrompt,
-            });
-        }
-        messages.push({
-            role: "user",
-            content: prompt,
-        });
-        const requestBody = {
-            messages,
-            temperature,
-            max_tokens: maxTokens,
-            stream: true,
-        };
-        // Create timeout controller if timeout is specified
-        const timeoutController = createTimeoutController(timeout, provider, "stream");
-        try {
-            const response = await this.makeRequest(requestBody, true, timeoutController?.controller.signal);
-            if (!response.body) {
-                throw new Error("No response body received");
-            }
-            // Return a StreamTextResult-like object with timeout signal
-            return {
-                textStream: this.createAsyncIterable(response.body, timeoutController?.controller.signal),
-                text: "",
-                usage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 },
-                finishReason: "stop",
-                // Store timeout controller for external cleanup if needed
-                _timeoutController: timeoutController,
-            };
-        }
-        catch (error) {
-            // Cleanup timeout on error
-            timeoutController?.cleanup();
-            // Log timeout errors specifically
-            if (error instanceof TimeoutError) {
-                logger.error(`[${functionTag}] Timeout error`, {
-                    provider,
-                    timeout: error.timeout,
-                    message: error.message,
-                });
-            }
-            else if (error?.name === "AbortError") {
-                // Convert AbortError to TimeoutError
-                const timeoutError = new TimeoutError(`${provider} stream operation timed out after ${timeout}`, timeoutController?.timeoutMs || 0, provider, "stream");
-                logger.error(`[${functionTag}] Timeout error`, {
-                    provider,
-                    timeout: timeoutController?.timeoutMs,
-                    message: timeoutError.message,
-                });
-                throw timeoutError;
-            }
-            else {
-                logger.error(`[${functionTag}] Error:`, error);
-            }
-            throw error;
-        }
-    }
-    async *createAsyncIterable(body, signal) {
-        const reader = body.getReader();
-        const decoder = new TextDecoder();
-        let buffer = "";
-        try {
-            while (true) {
-                // Check if aborted
-                if (signal?.aborted) {
-                    throw new Error("AbortError");
-                }
-                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 chunk
-                            if (chunk.choices?.[0]?.delta?.content) {
-                                yield chunk.choices[0].delta.content;
-                            }
-                        }
-                        catch (parseError) {
-                            logger.warn("[AzureOpenAIProvider.createAsyncIterable] Failed to parse chunk:", parseError);
-                            continue;
-                        }
-                    }
-                }
-            }
-        }
-        finally {
-            reader.releaseLock();
-        }
-    }
-    async *generateTextStream(optionsOrPrompt) {
-        logger.debug("[AzureOpenAIProvider.generateTextStream] Starting text streaming");
-        // Parse parameters with backward compatibility
-        const options = typeof optionsOrPrompt === "string"
-            ? { prompt: optionsOrPrompt }
-            : optionsOrPrompt;
-        const { prompt, temperature = 0.7, maxTokens = DEFAULT_MAX_TOKENS, systemPrompt = "You are a helpful AI assistant.", } = options;
-        logger.debug(`[AzureOpenAIProvider.generateTextStream] Streaming prompt: "${prompt.substring(0, 100)}..."`);
-        const messages = [];
-        if (systemPrompt) {
-            messages.push({
-                role: "system",
-                content: systemPrompt,
-            });
-        }
-        messages.push({
-            role: "user",
-            content: prompt,
-        });
-        const requestBody = {
-            messages,
-            temperature,
-            max_tokens: maxTokens,
-            stream: true,
-        };
-        try {
-            const response = await this.makeRequest(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 chunk
-                                if (chunk.choices?.[0]?.delta?.content) {
-                                    yield {
-                                        content: chunk.choices[0].delta.content,
-                                        provider: this.name,
-                                        model: chunk.model || this.deploymentId,
-                                    };
-                                }
-                            }
-                            catch (parseError) {
-                                logger.warn("[AzureOpenAIProvider.generateTextStream] Failed to parse chunk:", parseError);
-                                continue;
-                            }
-                        }
-                    }
-                }
-            }
-            finally {
-                reader.releaseLock();
-            }
-            logger.debug("[AzureOpenAIProvider.generateTextStream] Streaming completed");
-        }
-        catch (error) {
-            logger.error("[AzureOpenAIProvider.generateTextStream] Error:", error);
-            throw error;
-        }
-    }
-    async testConnection() {
-        logger.debug("[AzureOpenAIProvider.testConnection] Testing connection to Azure OpenAI");
-        const startTime = Date.now();
-        try {
-            await this.generateText({
-                prompt: "Hello",
-                maxTokens: 5,
-            });
-            const responseTime = Date.now() - startTime;
-            logger.debug(`[AzureOpenAIProvider.testConnection] Connection test successful (${responseTime}ms)`);
-            return {
-                success: true,
-                responseTime,
-            };
-        }
-        catch (error) {
-            const responseTime = Date.now() - startTime;
-            logger.error(`[AzureOpenAIProvider.testConnection] Connection test failed (${responseTime}ms):`, error);
-            return {
-                success: false,
-                error: error instanceof Error ? error.message : "Unknown error",
-                responseTime,
-            };
-        }
-    }
-    isConfigured() {
-        try {
-            this.getApiKey();
-            this.getEndpoint();
-            this.getDeploymentId();
-            return true;
-        }
-        catch {
-            return false;
-        }
-    }
-    getRequiredConfig() {
-        return [
-            "AZURE_OPENAI_API_KEY",
-            "AZURE_OPENAI_ENDPOINT",
-            "AZURE_OPENAI_DEPLOYMENT_ID",
-        ];
-    }
-    getOptionalConfig() {
-        return ["AZURE_OPENAI_API_VERSION"];
-    }
-    getModels() {
-        return [
-            "gpt-4",
-            "gpt-4-turbo",
-            "gpt-4-32k",
-            "gpt-35-turbo",
-            "gpt-35-turbo-16k",
-        ];
-    }
-    supportsStreaming() {
-        return true;
-    }
-    supportsSchema() {
-        return true; // Azure OpenAI supports JSON mode and function calling
-    }
-    getCapabilities() {
-        return [
-            "text-generation",
-            "streaming",
-            "conversation",
-            "system-prompts",
-            "json-mode",
-            "function-calling",
-            "enterprise-security",
-            "content-filtering",
-        ];
-    }
-    async generate(optionsOrPrompt, analysisSchema) {
-        return this.generateText(optionsOrPrompt, analysisSchema);
-    }
-    async gen(optionsOrPrompt, analysisSchema) {
-        return this.generateText(optionsOrPrompt, analysisSchema);
-    }
-}

package/dist/providers/googleAIStudio.d.ts

@@ -1,49 +0,0 @@
-import type { ZodType, ZodTypeDef } from "zod";
-import { type StreamTextResult, type ToolSet, type Schema, type GenerateTextResult, type LanguageModelV1 } from "ai";
-import type { AIProvider, TextGenerationOptions, StreamTextOptions, EnhancedGenerateTextResult } from "../core/types.js";
-export declare class GoogleAIStudio implements AIProvider {
-    private modelName;
-    /**
-     * Initializes a new instance of GoogleAIStudio
-     * @param modelName - Optional model name to override the default from config
-     */
-    constructor(modelName?: string | null);
-    /**
-     * Gets the appropriate model instance
-     * Made public to support FunctionCallingProvider integration
-     */
-    getModel(): LanguageModelV1;
-    /**
-     * Expose model property for FunctionCallingProvider
-     * This allows the enhanced provider to access the underlying model
-     */
-    get model(): LanguageModelV1;
-    /**
-     * Processes text using streaming approach with enhanced error handling callbacks
-     * @param prompt - The input text prompt to analyze
-     * @param analysisSchema - Optional Zod schema or Schema object for output validation
-     * @returns Promise resolving to StreamTextResult or null if operation fails
-     */
-    streamText(optionsOrPrompt: StreamTextOptions | string, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<StreamTextResult<ToolSet, unknown> | null>;
-    /**
-     * Processes text using non-streaming approach with optional schema validation
-     * @param prompt - The input text prompt to analyze
-     * @param analysisSchema - Optional Zod schema or Schema object for output validation
-     * @returns Promise resolving to GenerateTextResult or null if operation fails
-     */
-    generateText(optionsOrPrompt: TextGenerationOptions | string, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<GenerateTextResult<ToolSet, unknown> | null>;
-    /**
-     * Alias for generateText() - CLI-SDK consistency
-     * @param optionsOrPrompt - TextGenerationOptions object or prompt string
-     * @param analysisSchema - Optional schema for output validation
-     * @returns Promise resolving to GenerateTextResult or null
-     */
-    generate(optionsOrPrompt: TextGenerationOptions | string, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<EnhancedGenerateTextResult | null>;
-    /**
-     * Short alias for generateText() - CLI-SDK consistency
-     * @param optionsOrPrompt - TextGenerationOptions object or prompt string
-     * @param analysisSchema - Optional schema for output validation
-     * @returns Promise resolving to GenerateTextResult or null
-     */
-    gen(optionsOrPrompt: TextGenerationOptions | string, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<EnhancedGenerateTextResult | null>;
-}