@juspay/neurolink 9.30.0 → 9.31.0

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

@@ -1,5 +1,5 @@
-import { type LanguageModel, type LanguageModelV1, type Schema } from "ai";
-import type { ZodType, ZodTypeDef } from "zod";
+import { type LanguageModel, type Schema } from "ai";
+import type { ZodType } from "zod";
 import { type AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { EnhancedGenerateResult, TextGenerationOptions } from "../types/generateTypes.js";
@@ -120,7 +120,7 @@ export declare class GoogleVertexProvider extends BaseProvider {
      * Validate stream options
      */
     private validateStreamOptionsOnly;
-    protected executeStream(options: StreamOptions, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<StreamResult>;
+    protected executeStream(options: StreamOptions, analysisSchema?: ZodType | Schema<unknown>): Promise<StreamResult>;
     /**
      * Create @google/genai client configured for Vertex AI
      */
@@ -193,7 +193,7 @@ export declare class GoogleVertexProvider extends BaseProvider {
      * @example
      * ```typescript
      * provider.resolveModelAlias("claude-sonnet-4-5"); // "claude-sonnet-4-5@20250929"
-     * provider.resolveModelAlias("gemini-3-pro");      // "gemini-3-pro-latest"
+     * provider.resolveModelAlias("gemini-3-pro");      // "gemini-3.1-pro-preview"
      * provider.resolveModelAlias("gemini-2.5-flash");  // "gemini-2.5-flash" (unchanged)
      * ```
      */
@@ -202,9 +202,9 @@ export declare class GoogleVertexProvider extends BaseProvider {
      * Create an Anthropic model instance using vertexAnthropic provider
      * Uses fresh vertex settings for each request with comprehensive validation
      * @param modelName Anthropic model name (e.g., 'claude-3-sonnet@20240229')
-     * @returns LanguageModelV1 instance or null if not available
+     * @returns LanguageModel instance or null if not available
      */
-    createAnthropicModel(modelName: string): Promise<LanguageModelV1 | null>;
+    createAnthropicModel(modelName: string): Promise<LanguageModel | null>;
     /**
      * Validate Vertex AI authentication configuration
      */

package/dist/lib/providers/googleVertex.js

@@ -2,7 +2,7 @@ import dns from "node:dns";
 import { createVertex, } from "@ai-sdk/google-vertex";
 import { createVertexAnthropic, } from "@ai-sdk/google-vertex/anthropic";
 import { SpanKind, SpanStatusCode, trace } from "@opentelemetry/api";
-import { embed, embedMany, Output, streamText, } from "ai";
+import { embed, embedMany, Output, stepCountIs, streamText, } from "ai";
 import fs from "fs";
 import os from "os";
 import path from "path";
@@ -23,6 +23,7 @@ import { convertZodToJsonSchema, inlineJsonSchema, } from "../utils/schemaConver
 import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../utils/timeout.js";
 import { estimateTokens } from "../utils/tokenEstimation.js";
 import { buildNativeConfig, buildNativeToolDeclarations, collectStreamChunks, collectStreamChunksIncremental, computeMaxSteps as computeMaxStepsShared, createTextChannel, executeNativeToolCalls, extractTextFromParts, handleMaxStepsTermination, pushModelResponseToHistory, sanitizeToolsForGemini, } from "./googleNativeGemini3.js";
+import { getModelId } from "./providerTypeUtils.js";
 // Import proper types for multimodal message handling
 // Keep-alive note: Node.js native fetch and undici (used by createProxyFetch)
 // handle HTTP keep-alive internally. The fetchWithRetry wrapper in proxyFetch.ts
@@ -183,7 +184,7 @@ const createVertexSettings = async (region) => {
     if (process.env.GOOGLE_APPLICATION_CREDENTIALS_NEUROLINK) {
         const credentialsPath = process.env.GOOGLE_APPLICATION_CREDENTIALS_NEUROLINK;
         // Check if the credentials file exists
-        let fileExists = false;
+        let fileExists;
         try {
             fileExists = fs.existsSync(credentialsPath);
         }
@@ -198,7 +199,7 @@ const createVertexSettings = async (region) => {
         if (process.env.GOOGLE_APPLICATION_CREDENTIALS) {
             const credentialsPath = process.env.GOOGLE_APPLICATION_CREDENTIALS;
             // Check if the credentials file exists
-            let fileExists = false;
+            let fileExists;
             try {
                 fileExists = fs.existsSync(credentialsPath);
             }
@@ -341,8 +342,8 @@ export const VERTEX_MODEL_ALIASES = {
     "claude-3-sonnet": "claude-3-sonnet-20240229",
     "claude-3-haiku": "claude-3-haiku-20240307",
     // Gemini shorthand aliases
-    "gemini-3-pro": "gemini-3-pro-latest",
-    "gemini-3-flash": "gemini-3-flash-latest",
+    "gemini-3-pro": "gemini-3.1-pro-preview",
+    "gemini-3-flash": "gemini-3-flash-preview",
 };
 /**
  * Google Vertex AI Provider v2 - BaseProvider Implementation
@@ -867,7 +868,7 @@ export class GoogleVertexProvider extends BaseProvider {
                     Object.keys(tools).length > 0 && {
                     tools,
                     toolChoice: "auto",
-                    maxSteps: options.maxSteps || DEFAULT_MAX_STEPS,
+                    stopWhen: stepCountIs(options.maxSteps || DEFAULT_MAX_STEPS),
                 }),
                 abortSignal: composeAbortSignals(options.abortSignal, timeoutController?.controller.signal),
                 experimental_telemetry: this.telemetryHandler.getTelemetryConfig(options),
@@ -925,7 +926,7 @@ export class GoogleVertexProvider extends BaseProvider {
                     if (!isAnthropic) {
                         delete streamOptions.tools;
                         delete streamOptions.toolChoice;
-                        delete streamOptions.maxSteps;
+                        delete streamOptions.stopWhen;
                     }
                     streamOptions = {
                         ...streamOptions,
@@ -945,7 +946,7 @@ export class GoogleVertexProvider extends BaseProvider {
                 kind: SpanKind.CLIENT,
                 attributes: {
                     "gen_ai.system": "vertex",
-                    "gen_ai.request.model": model.modelId || this.modelName || "unknown",
+                    "gen_ai.request.model": getModelId(model, this.modelName || "unknown"),
                 },
             });
             let result;
@@ -963,18 +964,16 @@ export class GoogleVertexProvider extends BaseProvider {
             }
             // Collect token usage and finish reason asynchronously when the stream completes,
             // then end the span. This avoids blocking the stream consumer.
-            result.usage
+            Promise.resolve(result.usage)
                 .then((usage) => {
-                streamSpan.setAttribute("gen_ai.usage.input_tokens", usage.promptTokens || 0);
-                streamSpan.setAttribute("gen_ai.usage.output_tokens", usage.completionTokens || 0);
+                streamSpan.setAttribute("gen_ai.usage.input_tokens", usage.inputTokens || 0);
+                streamSpan.setAttribute("gen_ai.usage.output_tokens", usage.outputTokens || 0);
                 const effectiveModel = options.model ||
-                    model.modelId ||
-                    this.modelName ||
-                    getDefaultVertexModel();
+                    getModelId(model, this.modelName || getDefaultVertexModel());
                 const cost = calculateCost(this.providerName, effectiveModel, {
-                    input: usage.promptTokens || 0,
-                    output: usage.completionTokens || 0,
-                    total: (usage.promptTokens || 0) + (usage.completionTokens || 0),
+                    input: usage.inputTokens || 0,
+                    output: usage.outputTokens || 0,
+                    total: (usage.inputTokens || 0) + (usage.outputTokens || 0),
                 });
                 if (cost && cost > 0) {
                     streamSpan.setAttribute("neurolink.cost", cost);
@@ -983,14 +982,14 @@ export class GoogleVertexProvider extends BaseProvider {
                 .catch(() => {
                 // Usage may not be available if the stream is aborted
             });
-            result.finishReason
+            Promise.resolve(result.finishReason)
                 .then((reason) => {
                 streamSpan.setAttribute("gen_ai.response.finish_reason", reason || "unknown");
             })
                 .catch(() => {
                 // Finish reason may not be available if the stream is aborted
             });
-            result.text
+            Promise.resolve(result.text)
                 .then(() => {
                 streamSpan.end();
             })
@@ -1001,8 +1000,15 @@ export class GoogleVertexProvider extends BaseProvider {
                 });
                 streamSpan.end();
             });
-            // Defer timeout cleanup until the stream completes or errors
-            result.text.finally(() => timeoutController?.cleanup());
+            // Defer timeout cleanup until the stream completes or errors.
+            // Guard against NoOutputGeneratedError becoming an unhandled rejection.
+            Promise.resolve(result.text)
+                .catch((err) => {
+                logger.debug("Stream text promise rejected (expected for empty streams)", {
+                    error: err instanceof Error ? err.message : String(err),
+                });
+            })
+                .finally(() => timeoutController?.cleanup());
             // Transform string stream to content object stream using BaseProvider method
             const transformedStream = this.createTextStream(result);
             // Track tool calls and results for streaming
@@ -1789,7 +1795,7 @@ export class GoogleVertexProvider extends BaseProvider {
      * @example
      * ```typescript
      * provider.resolveModelAlias("claude-sonnet-4-5"); // "claude-sonnet-4-5@20250929"
-     * provider.resolveModelAlias("gemini-3-pro");      // "gemini-3-pro-latest"
+     * provider.resolveModelAlias("gemini-3-pro");      // "gemini-3.1-pro-preview"
      * provider.resolveModelAlias("gemini-2.5-flash");  // "gemini-2.5-flash" (unchanged)
      * ```
      */
@@ -1800,7 +1806,7 @@ export class GoogleVertexProvider extends BaseProvider {
      * Create an Anthropic model instance using vertexAnthropic provider
      * Uses fresh vertex settings for each request with comprehensive validation
      * @param modelName Anthropic model name (e.g., 'claude-3-sonnet@20240229')
-     * @returns LanguageModelV1 instance or null if not available
+     * @returns LanguageModel instance or null if not available
      */
     async createAnthropicModel(modelName) {
         const validationId = `anthropic-validation-${Date.now()}-${Math.random().toString(36).substring(2, 11)}`;
@@ -2885,9 +2891,9 @@ export class GoogleVertexProvider extends BaseProvider {
     getModelSuggestions(requestedModel) {
         const availableModels = {
             google: [
-                "gemini-3-pro-preview-11-2025",
-                "gemini-3-pro-latest",
-                "gemini-3-pro-preview",
+                "gemini-3.1-pro-preview",
+                "gemini-3.1-flash-lite-preview",
+                "gemini-3-flash-preview",
                 "gemini-2.5-pro",
                 "gemini-2.5-flash",
                 "gemini-2.5-flash-lite",

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

@@ -1,5 +1,5 @@
-import type { ZodType, ZodTypeDef } from "zod";
-import { type Schema, type LanguageModelV1 } from "ai";
+import type { ZodType } from "zod";
+import { type Schema, type LanguageModel } from "ai";
 import { AIProviderName } from "../constants/enums.js";
 import type { StreamOptions, StreamResult } from "../types/streamTypes.js";
 import { BaseProvider } from "../core/baseProvider.js";
@@ -36,7 +36,7 @@ export declare class HuggingFaceProvider extends BaseProvider {
      * @returns true for supported models, false for unsupported models
      */
     supportsTools(): boolean;
-    protected executeStream(options: StreamOptions, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<StreamResult>;
+    protected executeStream(options: StreamOptions, analysisSchema?: ZodType | Schema<unknown>): Promise<StreamResult>;
     /**
      * Prepare stream options with HuggingFace-specific enhancements
      * Handles tool calling optimizations and model-specific formatting
@@ -74,6 +74,6 @@ export declare class HuggingFaceProvider extends BaseProvider {
     /**
      * Returns the Vercel AI SDK model instance for HuggingFace
      */
-    getAISDKModel(): LanguageModelV1;
+    getAISDKModel(): LanguageModel;
 }
 export default HuggingFaceProvider;

package/dist/lib/providers/huggingFace.js

@@ -1,5 +1,5 @@
 import { createOpenAI } from "@ai-sdk/openai";
-import { streamText, } from "ai";
+import { NoOutputGeneratedError, stepCountIs, streamText, } from "ai";
 import { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import { logger } from "../utils/logger.js";
@@ -131,8 +131,8 @@ export class HuggingFaceProvider extends BaseProvider {
                 model: this.model,
                 messages: messages,
                 temperature: options.temperature,
-                maxTokens: options.maxTokens, // No default limit - unlimited unless specified
-                maxSteps: options.maxSteps || DEFAULT_MAX_STEPS,
+                maxOutputTokens: options.maxTokens, // No default limit - unlimited unless specified
+                stopWhen: stepCountIs(options.maxSteps || DEFAULT_MAX_STEPS),
                 tools: (shouldUseTools
                     ? streamOptions.tools || allTools
                     : {}),
@@ -153,8 +153,18 @@ export class HuggingFaceProvider extends BaseProvider {
             timeoutController?.cleanup();
             // Transform stream to match StreamResult interface with enhanced tool call parsing
             const transformedStream = async function* () {
-                for await (const chunk of result.textStream) {
-                    yield { content: chunk };
+                try {
+                    for await (const chunk of result.textStream) {
+                        yield { content: chunk };
+                    }
+                }
+                catch (streamError) {
+                    // AI SDK v6 throws NoOutputGeneratedError when the stream produced no output.
+                    if (NoOutputGeneratedError.isInstance(streamError)) {
+                        logger.warn("HuggingFace: Stream produced no output (NoOutputGeneratedError)");
+                        return;
+                    }
+                    throw streamError;
                 }
             };
             return {

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

@@ -1,5 +1,5 @@
-import { type LanguageModelV1, type Schema } from "ai";
-import type { ZodType, ZodTypeDef } from "zod";
+import { type LanguageModel, type Schema } from "ai";
+import type { ZodType } from "zod";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { StreamOptions, StreamResult } from "../types/streamTypes.js";
@@ -18,7 +18,7 @@ export declare class LiteLLMProvider extends BaseProvider {
     /**
      * Returns the Vercel AI SDK model instance for LiteLLM
      */
-    protected getAISDKModel(): LanguageModelV1;
+    protected getAISDKModel(): LanguageModel;
     formatProviderError(error: unknown): Error;
     /**
      * LiteLLM supports tools for compatible models
@@ -28,7 +28,7 @@ export declare class LiteLLMProvider extends BaseProvider {
      * Provider-specific streaming implementation
      * Note: This is only used when tools are disabled
      */
-    protected executeStream(options: StreamOptions, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<StreamResult>;
+    protected executeStream(options: StreamOptions, analysisSchema?: ZodType | Schema<unknown>): Promise<StreamResult>;
     /**
      * Generate an embedding for a single text input
      * Uses the LiteLLM proxy with OpenAI-compatible embedding API

package/dist/lib/providers/litellm.js

@@ -1,6 +1,6 @@
 import { createOpenAI } from "@ai-sdk/openai";
 import { SpanKind, SpanStatusCode, trace } from "@opentelemetry/api";
-import { Output, streamText, } from "ai";
+import { NoOutputGeneratedError, Output, streamText, } from "ai";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
 import { streamAnalyticsCollector } from "../core/streamAnalytics.js";
@@ -11,6 +11,7 @@ import { logger } from "../utils/logger.js";
 import { calculateCost } from "../utils/pricing.js";
 import { getProviderModel } from "../utils/providerConfig.js";
 import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../utils/timeout.js";
+import { getModelId } from "./providerTypeUtils.js";
 const streamTracer = trace.getTracer("neurolink.provider.litellm");
 // Configuration helpers
 const getLiteLLMConfig = () => {
@@ -58,9 +59,7 @@ export class LiteLLMProvider extends BaseProvider {
             apiKey: config.apiKey,
             fetch: createProxyFetch(),
         });
-        this.model = customOpenAI(this.modelName || getDefaultLiteLLMModel(), {
-            structuredOutputs: false,
-        });
+        this.model = customOpenAI(this.modelName || getDefaultLiteLLMModel());
         logger.debug("LiteLLM Provider initialized", {
             modelName: this.modelName,
             provider: this.providerName,
@@ -216,7 +215,7 @@ export class LiteLLMProvider extends BaseProvider {
                 kind: SpanKind.CLIENT,
                 attributes: {
                     "gen_ai.system": "litellm",
-                    "gen_ai.request.model": model.modelId || this.modelName || "unknown",
+                    "gen_ai.request.model": getModelId(model, this.modelName || "unknown"),
                 },
             });
             let result;
@@ -235,14 +234,14 @@ export class LiteLLMProvider extends BaseProvider {
             }
             // Collect token usage, cost, and finish reason asynchronously when the stream completes,
             // then end the span. This avoids blocking the stream consumer.
-            result.usage
+            Promise.resolve(result.usage)
                 .then((usage) => {
-                streamSpan.setAttribute("gen_ai.usage.input_tokens", usage.promptTokens || 0);
-                streamSpan.setAttribute("gen_ai.usage.output_tokens", usage.completionTokens || 0);
+                streamSpan.setAttribute("gen_ai.usage.input_tokens", usage.inputTokens || 0);
+                streamSpan.setAttribute("gen_ai.usage.output_tokens", usage.outputTokens || 0);
                 const cost = calculateCost(this.providerName, this.modelName, {
-                    input: usage.promptTokens || 0,
-                    output: usage.completionTokens || 0,
-                    total: (usage.promptTokens || 0) + (usage.completionTokens || 0),
+                    input: usage.inputTokens || 0,
+                    output: usage.outputTokens || 0,
+                    total: (usage.inputTokens || 0) + (usage.outputTokens || 0),
                 });
                 if (cost && cost > 0) {
                     streamSpan.setAttribute("neurolink.cost", cost);
@@ -251,14 +250,14 @@ export class LiteLLMProvider extends BaseProvider {
                 .catch(() => {
                 // Usage may not be available if the stream is aborted
             });
-            result.finishReason
+            Promise.resolve(result.finishReason)
                 .then((reason) => {
                 streamSpan.setAttribute("gen_ai.response.finish_reason", reason || "unknown");
             })
                 .catch(() => {
                 // Finish reason may not be available if the stream is aborted
             });
-            result.text
+            Promise.resolve(result.text)
                 .then(() => {
                 streamSpan.end();
             })
@@ -273,40 +272,53 @@ export class LiteLLMProvider extends BaseProvider {
             // Transform stream to content object stream using fullStream (handles both text and tool calls)
             // Note: fullStream includes tool results, textStream only has text
             const transformedStream = (async function* () {
-                // Try fullStream first (handles both text and tool calls), fallback to textStream
-                const streamToUse = result.fullStream || result.textStream;
-                for await (const chunk of streamToUse) {
-                    // Handle different chunk types from fullStream
-                    if (chunk && typeof chunk === "object") {
-                        // Check for error chunks first (critical error handling)
-                        if ("type" in chunk && chunk.type === "error") {
-                            const errorChunk = chunk;
-                            logger.error(`LiteLLM: Error chunk received:`, {
-                                errorType: errorChunk.type,
-                                errorDetails: errorChunk.error,
-                            });
-                            throw new Error(`LiteLLM streaming error: ${errorChunk.error?.message || "Unknown error"}`);
-                        }
-                        if ("textDelta" in chunk) {
-                            // Text delta from fullStream
-                            const textDelta = chunk.textDelta;
-                            if (textDelta) {
-                                yield { content: textDelta };
+                try {
+                    // Try fullStream first (handles both text and tool calls), fallback to textStream
+                    const streamToUse = result.fullStream || result.textStream;
+                    for await (const chunk of streamToUse) {
+                        // Handle different chunk types from fullStream
+                        if (chunk && typeof chunk === "object") {
+                            // Check for error chunks first (critical error handling)
+                            if ("type" in chunk && chunk.type === "error") {
+                                const errorChunk = chunk;
+                                logger.error(`LiteLLM: Error chunk received:`, {
+                                    errorType: errorChunk.type,
+                                    errorDetails: errorChunk.error,
+                                });
+                                throw new Error(`LiteLLM streaming error: ${errorChunk.error?.message || "Unknown error"}`);
+                            }
+                            if ("textDelta" in chunk) {
+                                // Text delta from fullStream
+                                const textDelta = chunk.textDelta;
+                                if (textDelta) {
+                                    yield { content: textDelta };
+                                }
+                            }
+                            else if ("type" in chunk &&
+                                chunk.type === "tool-call" &&
+                                "toolCallId" in chunk) {
+                                // Tool call event - log for debugging
+                                const toolCallId = String(chunk.toolCallId);
+                                const toolName = "toolName" in chunk ? String(chunk.toolName) : "unknown";
+                                logger.debug("LiteLLM: Tool call", {
+                                    toolCallId,
+                                    toolName,
+                                });
                             }
                         }
-                        else if (chunk.type === "tool-call-streaming-start") {
-                            // Tool call streaming start event - log for debugging
-                            const toolCall = chunk;
-                            logger.debug("LiteLLM: Tool call streaming start", {
-                                toolCallId: toolCall.toolCallId,
-                                toolName: toolCall.toolName,
-                            });
+                        else if (typeof chunk === "string") {
+                            // Direct string chunk from textStream fallback
+                            yield { content: chunk };
                         }
                     }
-                    else if (typeof chunk === "string") {
-                        // Direct string chunk from textStream fallback
-                        yield { content: chunk };
+                }
+                catch (streamError) {
+                    // AI SDK v6 throws NoOutputGeneratedError when the stream produced no output.
+                    if (NoOutputGeneratedError.isInstance(streamError)) {
+                        logger.warn("LiteLLM: Stream produced no output (NoOutputGeneratedError)");
+                        return;
                     }
+                    throw streamError;
                 }
             })();
             // Create analytics promise that resolves after stream completion

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

@@ -1,4 +1,4 @@
-import { type LanguageModelV1 } from "ai";
+import { type LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { StreamOptions, StreamResult } from "../types/streamTypes.js";
@@ -16,7 +16,7 @@ export declare class MistralProvider extends BaseProvider {
     /**
      * Returns the Vercel AI SDK model instance for Mistral
      */
-    getAISDKModel(): LanguageModelV1;
+    getAISDKModel(): LanguageModel;
     formatProviderError(error: unknown): Error;
     /**
      * Validate provider configuration

package/dist/lib/providers/mistral.js

@@ -1,5 +1,5 @@
 import { createMistral } from "@ai-sdk/mistral";
-import { streamText } from "ai";
+import { stepCountIs, streamText } from "ai";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
 import { streamAnalyticsCollector } from "../core/streamAnalytics.js";
@@ -7,6 +7,7 @@ import { createProxyFetch } from "../proxy/proxyFetch.js";
 import { logger } from "../utils/logger.js";
 import { createMistralConfig, getProviderModel, validateApiKey, } from "../utils/providerConfig.js";
 import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../utils/timeout.js";
+import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
 // Configuration helpers - now using consolidated utility
 const getMistralApiKey = () => {
     return validateApiKey(createMistralConfig());
@@ -59,9 +60,9 @@ export class MistralProvider extends BaseProvider {
                 model,
                 messages: messages,
                 temperature: options.temperature,
-                maxTokens: options.maxTokens, // No default limit - unlimited unless specified
+                maxOutputTokens: options.maxTokens, // No default limit - unlimited unless specified
                 tools,
-                maxSteps: options.maxSteps || DEFAULT_MAX_STEPS,
+                stopWhen: stepCountIs(options.maxSteps || DEFAULT_MAX_STEPS),
                 toolChoice: shouldUseTools ? "auto" : "none",
                 abortSignal: composeAbortSignals(options.abortSignal, timeoutController?.controller.signal),
                 experimental_telemetry: this.telemetryHandler.getTelemetryConfig(options),
@@ -78,7 +79,7 @@ export class MistralProvider extends BaseProvider {
             // Transform string stream to content object stream using BaseProvider method
             const transformedStream = this.createTextStream(result);
             // Create analytics promise that resolves after stream completion
-            const analyticsPromise = streamAnalyticsCollector.createAnalytics(this.providerName, this.modelName, result, Date.now() - startTime, {
+            const analyticsPromise = streamAnalyticsCollector.createAnalytics(this.providerName, this.modelName, toAnalyticsStreamResult(result), Date.now() - startTime, {
                 requestId: `mistral-stream-${Date.now()}`,
                 streamingMode: true,
             });

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

@@ -1,4 +1,4 @@
-import type { LanguageModelV1, Schema } from "ai";
+import type { LanguageModel, Schema } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { StreamOptions, StreamResult } from "../types/streamTypes.js";
@@ -22,10 +22,13 @@ export declare class OllamaProvider extends BaseProvider {
     protected getProviderName(): AIProviderName;
     protected getDefaultModel(): string;
     /**
-     * Returns the Vercel AI SDK model instance for Ollama
-     * The OllamaLanguageModel implements LanguageModelV1 interface properly
+     * Returns the Vercel AI SDK model instance for Ollama.
+     *
+     * OllamaLanguageModel implements OllamaAsLanguageModel which is structurally
+     * compatible with LanguageModelV2 (specificationVersion "v2", modelId, provider,
+     * supportedUrls, doGenerate, doStream).
      */
-    protected getAISDKModel(): LanguageModelV1;
+    protected getAISDKModel(): LanguageModel;
     /**
      * Ollama Tool Calling Support (Enhanced 2025)
      *

package/dist/lib/providers/ollama.js

@@ -42,14 +42,25 @@ const getOllamaTimeout = () => {
 };
 // Create proxy-aware fetch instance
 const proxyFetch = createProxyFetch();
-// Custom LanguageModelV1 implementation for Ollama
+// Custom LanguageModel implementation for Ollama
 class OllamaLanguageModel {
-    specificationVersion = "v1";
+    /**
+     * Specification version for the AI SDK LanguageModel interface.
+     * Uses "v2" for structural compatibility with AI SDK v6's `LanguageModelV2`.
+     * The AI SDK checks this field to determine which interface version to use.
+     */
+    specificationVersion = "v2";
     provider = "ollama";
     modelId;
     maxTokens;
     supportsStreaming = true;
     defaultObjectGenerationMode = "json";
+    /**
+     * Supported URL patterns by media type.
+     * Ollama runs locally and does not natively download URLs, so this is empty.
+     * Required by the LanguageModelV2 interface.
+     */
+    supportedUrls = {};
     baseUrl;
     timeout;
     constructor(modelId, baseUrl, timeout) {
@@ -472,11 +483,15 @@ export class OllamaProvider extends BaseProvider {
         return getDefaultOllamaModel();
     }
     /**
-     * Returns the Vercel AI SDK model instance for Ollama
-     * The OllamaLanguageModel implements LanguageModelV1 interface properly
+     * Returns the Vercel AI SDK model instance for Ollama.
+     *
+     * OllamaLanguageModel implements OllamaAsLanguageModel which is structurally
+     * compatible with LanguageModelV2 (specificationVersion "v2", modelId, provider,
+     * supportedUrls, doGenerate, doStream).
      */
     getAISDKModel() {
-        return this.ollamaModel;
+        const model = this.ollamaModel;
+        return model;
     }
     /**
      * Ollama Tool Calling Support (Enhanced 2025)
@@ -1174,9 +1189,16 @@ export class OllamaProvider extends BaseProvider {
         const result = {};
         for (const [name, tool] of Object.entries(aiTools)) {
             if ("description" in tool && tool.description) {
+                // AI SDK v6 uses `inputSchema`; legacy tools may still use `parameters`
+                const toolSchema = "inputSchema" in tool
+                    ? tool.inputSchema
+                    : "parameters" in tool
+                        ? tool
+                            .parameters
+                        : undefined;
                 result[name] = {
                     description: tool.description,
-                    parameters: "parameters" in tool ? tool.parameters : undefined,
+                    parameters: toolSchema,
                     execute: async (params) => {
                         if ("execute" in tool && tool.execute) {
                             const result = await tool.execute(params, {
@@ -1282,7 +1304,7 @@ export class OllamaProvider extends BaseProvider {
         for (const call of toolCalls) {
             logger.debug(`[OllamaProvider] Executing tool: ${call.function.name}`);
             // Parse arguments
-            let args = {};
+            let args;
             try {
                 args = JSON.parse(call.function.arguments);
             }
@@ -1488,7 +1510,7 @@ export class OllamaProvider extends BaseProvider {
         }
         catch (error) {
             if (error instanceof Error && error.message.includes("ECONNREFUSED")) {
-                throw new Error(`❌ Ollama Service Not Running\n\nCannot connect to Ollama service.\n\n🔧 Start Ollama:\n1. Run: ollama serve\n2. Or start Ollama app\n3. Verify: curl ${this.baseUrl}/api/version`);
+                throw new Error(`❌ Ollama Service Not Running\n\nCannot connect to Ollama service.\n\n🔧 Start Ollama:\n1. Run: ollama serve\n2. Or start Ollama app\n3. Verify: curl ${this.baseUrl}/api/version`, { cause: error });
             }
             throw error;
         }

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

@@ -1,4 +1,4 @@
-import { type LanguageModelV1 } from "ai";
+import { type LanguageModel } from "ai";
 import { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeuroLink } from "../neurolink.js";
@@ -25,7 +25,7 @@ export declare class OpenAIProvider extends BaseProvider {
     /**
      * Returns the Vercel AI SDK model instance for OpenAI
      */
-    getAISDKModel(): LanguageModelV1;
+    getAISDKModel(): LanguageModel;
     /**
      * OpenAI-specific tool validation and filtering
      * Filters out tools that might cause streaming issues