@juspay/neurolink 1.5.3 → 1.9.0

package/dist/providers/mistralAI.d.ts

@@ -0,0 +1,32 @@
+import type { ZodType, ZodTypeDef } from "zod";
+import { type StreamTextResult, type ToolSet, type Schema, type GenerateTextResult } from "ai";
+import type { AIProvider, TextGenerationOptions, StreamTextOptions } from "../core/types.js";
+export declare class MistralAI implements AIProvider {
+    private modelName;
+    private client;
+    /**
+     * Initializes a new instance of MistralAI
+     * @param modelName - Optional model name to override the default from config
+     */
+    constructor(modelName?: string | null);
+    /**
+     * Gets the appropriate model instance
+     * @private
+     */
+    private getModel;
+    /**
+     * 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>;
+}
+export default MistralAI;

package/dist/providers/mistralAI.js

@@ -0,0 +1,223 @@
+import { createMistral } from "@ai-sdk/mistral";
+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 getMistralApiKey = () => {
+    const apiKey = process.env.MISTRAL_API_KEY;
+    if (!apiKey) {
+        throw new Error("MISTRAL_API_KEY environment variable is not set");
+    }
+    return apiKey;
+};
+const getMistralModelId = () => {
+    return process.env.MISTRAL_MODEL || "mistral-small";
+};
+const hasValidAuth = () => {
+    return !!process.env.MISTRAL_API_KEY;
+};
+// Lazy initialization cache
+let _mistralClient = null;
+function getMistralClient() {
+    if (!_mistralClient) {
+        const apiKey = getMistralApiKey();
+        _mistralClient = createMistral({
+            apiKey,
+            baseURL: process.env.MISTRAL_ENDPOINT || "https://api.mistral.ai/v1",
+        });
+    }
+    return _mistralClient;
+}
+// Mistral AI class with enhanced error handling
+export class MistralAI {
+    modelName;
+    client;
+    /**
+     * Initializes a new instance of MistralAI
+     * @param modelName - Optional model name to override the default from config
+     */
+    constructor(modelName) {
+        const functionTag = "MistralAI.constructor";
+        this.modelName = modelName || getMistralModelId();
+        try {
+            this.client = getMistralClient();
+            logger.debug(`[${functionTag}] Initialization started`, {
+                modelName: this.modelName,
+                hasApiKey: hasValidAuth(),
+            });
+            logger.debug(`[${functionTag}] Initialization completed`, {
+                modelName: this.modelName,
+                success: true,
+            });
+        }
+        catch (err) {
+            logger.error(`[${functionTag}] Initialization failed`, {
+                message: "Error in initializing Mistral AI",
+                modelName: this.modelName,
+                error: err instanceof Error ? err.message : String(err),
+                stack: err instanceof Error ? err.stack : undefined,
+            });
+            throw err;
+        }
+    }
+    /**
+     * Gets the appropriate model instance
+     * @private
+     */
+    getModel() {
+        logger.debug("MistralAI.getModel - Mistral AI model selected", {
+            modelName: this.modelName,
+        });
+        return this.client(this.modelName);
+    }
+    /**
+     * 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
+     */
+    async streamText(optionsOrPrompt, analysisSchema) {
+        const functionTag = "MistralAI.streamText";
+        const provider = "mistral";
+        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,
+                hasSchema: !!finalSchema,
+            });
+            const model = this.getModel();
+            const streamOptions = {
+                model: 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,
+                        error: errorMessage,
+                        stack: errorStack,
+                        promptLength: prompt.length,
+                        chunkCount,
+                    });
+                },
+                onFinish: (event) => {
+                    logger.debug(`[${functionTag}] Stream text finished`, {
+                        provider,
+                        modelName: this.modelName,
+                        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,
+                });
+            }
+            const result = streamText(streamOptions);
+            return result;
+        }
+        catch (err) {
+            logger.error(`[${functionTag}] Exception`, {
+                provider,
+                modelName: this.modelName,
+                message: "Error in streaming text",
+                err: String(err),
+                promptLength: typeof optionsOrPrompt === "string"
+                    ? optionsOrPrompt.length
+                    : optionsOrPrompt.prompt.length,
+            });
+            throw err; // Re-throw error to trigger fallback
+        }
+    }
+    /**
+     * 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
+     */
+    async generateText(optionsOrPrompt, analysisSchema) {
+        const functionTag = "MistralAI.generateText";
+        const provider = "mistral";
+        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 request started`, {
+                provider,
+                modelName: this.modelName,
+                promptLength: prompt.length,
+                temperature,
+                maxTokens,
+            });
+            const model = this.getModel();
+            const generateOptions = {
+                model: 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
+        }
+    }
+}
+export default MistralAI;

package/dist/providers/ollama.d.ts

@@ -0,0 +1,51 @@
+/**
+ * Ollama Provider for NeuroLink
+ *
+ * Local AI model deployment and management using Ollama.
+ * Provides offline AI capabilities with local model hosting.
+ *
+ * Features:
+ * - Local model deployment (privacy-first)
+ * - Model management (download, list, remove)
+ * - Health checking and service validation
+ * - Streaming and non-streaming text generation
+ */
+import type { AIProvider, TextGenerationOptions, StreamTextOptions } from "../core/types.js";
+import type { GenerateTextResult, StreamTextResult, ToolSet } from "ai";
+import type { ZodType, ZodTypeDef } from "zod";
+import type { Schema } from "ai";
+export declare class Ollama implements AIProvider {
+    private baseUrl;
+    private modelName;
+    private timeout;
+    constructor(modelName?: string);
+    /**
+     * Gets the appropriate model instance
+     * @private
+     */
+    private getModel;
+    /**
+     * Health check - verify Ollama service is running and accessible
+     */
+    checkHealth(): Promise<boolean>;
+    /**
+     * List available models on the Ollama instance
+     */
+    listModels(): Promise<string[]>;
+    /**
+     * Check if a specific model is available
+     */
+    isModelAvailable(modelName: string): Promise<boolean>;
+    /**
+     * Pull/download a model to the local Ollama instance
+     */
+    pullModel(modelName: string): Promise<void>;
+    /**
+     * Generate text using Ollama local models
+     */
+    generateText(optionsOrPrompt: TextGenerationOptions | string, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<GenerateTextResult<ToolSet, unknown> | null>;
+    /**
+     * Generate streaming text using Ollama local models
+     */
+    streamText(optionsOrPrompt: StreamTextOptions | string, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<StreamTextResult<ToolSet, unknown> | null>;
+}