@juspay/neurolink 9.30.0 → 9.31.1

package/dist/lib/core/modules/MessageBuilder.js

@@ -8,7 +8,7 @@
  * - Building messages from text generation options
  * - Building messages from stream options
  * - Multimodal input detection
- * - Message format conversion (to CoreMessage[])
+ * - Message format conversion (to ModelMessage[])
  *
  * @module core/modules/MessageBuilder
  */
@@ -129,8 +129,7 @@ export class MessageBuilder {
                     return {
                         role: msg.role,
                         content: msg.content.map((item) => {
-                            const itemProviderOptions = item
-                                .providerOptions;
+                            const itemProviderOptions = item.providerOptions;
                             if (item.type === "text") {
                                 return {
                                     type: "text",
@@ -169,7 +168,7 @@ export class MessageBuilder {
      * with automatic multimodal detection, eliminating code duplication
      *
      * @param options - Stream options or text generation options
-     * @returns Promise resolving to CoreMessage array ready for AI SDK
+     * @returns Promise resolving to ModelMessage array ready for AI SDK
      */
     async buildMessagesForStream(options) {
         return withSpan({
@@ -238,8 +237,7 @@ export class MessageBuilder {
                     return {
                         role: msg.role,
                         content: msg.content.map((item) => {
-                            const itemProviderOptions = item
-                                .providerOptions;
+                            const itemProviderOptions = item.providerOptions;
                             if (item.type === "text") {
                                 return {
                                     type: "text",

package/dist/lib/core/modules/StreamHandler.js

@@ -19,6 +19,7 @@ import { validateStreamOptions as validateStreamOpts, ValidationError, createVal
 import { STEP_LIMITS } from "../constants.js";
 import { createAnalytics } from "../analytics.js";
 import { nanoid } from "nanoid";
+import { NoOutputGeneratedError } from "ai";
 /**
  * StreamHandler class - Handles streaming operations for AI providers
  */
@@ -89,20 +90,33 @@ export class StreamHandler {
             let totalBytes = 0;
             const streamStart = Date.now();
             let firstChunkTime;
-            for await (const chunk of result.textStream) {
-                chunkCount++;
-                totalBytes += chunk.length;
-                if (!firstChunkTime) {
-                    firstChunkTime = Date.now();
-                    const activeSpan = trace.getSpan(otelContext.active());
-                    if (activeSpan) {
-                        activeSpan.addEvent("stream.first_chunk", {
-                            "stream.ttfc_ms": firstChunkTime - streamStart,
-                            "stream.provider": providerName,
-                        });
+            try {
+                for await (const chunk of result.textStream) {
+                    chunkCount++;
+                    totalBytes += chunk.length;
+                    if (!firstChunkTime) {
+                        firstChunkTime = Date.now();
+                        const activeSpan = trace.getSpan(otelContext.active());
+                        if (activeSpan) {
+                            activeSpan.addEvent("stream.first_chunk", {
+                                "stream.ttfc_ms": firstChunkTime - streamStart,
+                                "stream.provider": providerName,
+                            });
+                        }
                     }
+                    yield { content: chunk };
+                }
+            }
+            catch (error) {
+                // AI SDK v6 throws NoOutputGeneratedError when the stream produces no output
+                // (e.g. empty response, model refusal with no text). Treat as an empty stream
+                // rather than crashing the process with an unhandled rejection.
+                if (NoOutputGeneratedError.isInstance(error)) {
+                    logger.warn(`${providerName}: Stream produced no output (NoOutputGeneratedError), returning empty stream`);
+                }
+                else {
+                    throw error;
                 }
-                yield { content: chunk };
             }
             // Record completion metrics on the active span
             const activeSpan = trace.getSpan(otelContext.active());

package/dist/lib/core/modules/TelemetryHandler.d.ts

@@ -36,9 +36,8 @@ export declare class TelemetryHandler {
      * Record performance metrics for a generation
      */
     recordPerformanceMetrics(usage: {
-        promptTokens: number;
-        completionTokens: number;
-        totalTokens: number;
+        inputTokens: number | undefined;
+        outputTokens: number | undefined;
     } | undefined, responseTime: number): Promise<void>;
     /**
      * Calculate actual cost based on token usage and provider configuration.
@@ -53,9 +52,8 @@ export declare class TelemetryHandler {
      * on Vertex AI ($0.000060 vs $0.106895 for the same request).
      */
     calculateActualCost(usage: {
-        promptTokens?: number;
-        completionTokens?: number;
-        totalTokens?: number;
+        inputTokens?: number | undefined;
+        outputTokens?: number | undefined;
     }): Promise<number>;
     /**
      * Create telemetry configuration for Vercel AI SDK experimental_telemetry

package/dist/lib/core/modules/TelemetryHandler.js

@@ -75,18 +75,19 @@ export class TelemetryHandler {
      */
     async recordPerformanceMetrics(usage, responseTime) {
         try {
-            const actualCost = await this.calculateActualCost(usage || { promptTokens: 0, completionTokens: 0, totalTokens: 0 });
+            const totalTokens = (usage?.inputTokens || 0) + (usage?.outputTokens || 0);
+            const actualCost = await this.calculateActualCost(usage || { inputTokens: 0, outputTokens: 0 });
             recordProviderPerformanceFromMetrics(this.providerName, {
                 responseTime,
-                tokensGenerated: usage?.totalTokens || 0,
+                tokensGenerated: totalTokens,
                 cost: actualCost,
                 success: true,
             });
             // Wire TelemetryService metrics so OTEL counters/histograms are populated
-            TelemetryService.getInstance().recordAIRequest(this.providerName, this.modelName, usage?.totalTokens || 0, responseTime, actualCost > 0 ? actualCost : undefined);
+            TelemetryService.getInstance().recordAIRequest(this.providerName, this.modelName, totalTokens, responseTime, actualCost > 0 ? actualCost : undefined);
             logger.debug(`Performance recorded for ${this.providerName}`, {
                 responseTime: `${responseTime}ms`,
-                tokens: usage?.totalTokens || 0,
+                tokens: totalTokens,
                 estimatedCost: `$${actualCost.toFixed(6)}`,
             });
         }
@@ -108,8 +109,8 @@ export class TelemetryHandler {
      */
     async calculateActualCost(usage) {
         try {
-            const promptTokens = usage?.promptTokens || 0;
-            const completionTokens = usage?.completionTokens || 0;
+            const promptTokens = usage?.inputTokens || 0;
+            const completionTokens = usage?.outputTokens || 0;
             // Try the per-model pricing table first (includes correct rates for
             // Claude on Vertex, cache token rates, etc.)
             if (hasPricing(this.providerName, this.modelName)) {

package/dist/lib/core/modules/ToolsManager.d.ts

@@ -15,18 +15,8 @@
  * @module core/modules/ToolsManager
  */
 import type { Tool } from "ai";
-import { z } from "zod";
-import type { AIProviderName } from "../../types/index.js";
+import type { AIProviderName, ToolUtilities } from "../../types/index.js";
 import type { NeuroLink } from "../../neurolink.js";
-/**
- * Utility functions interface for ToolsManager
- */
-export type ToolUtilities = {
-    isZodSchema?: (schema: unknown) => boolean;
-    convertToolResult?: (result: unknown) => Promise<unknown>;
-    createPermissiveZodSchema?: () => z.ZodSchema;
-    fixSchemaForOpenAIStrictMode?: (schema: Record<string, unknown>) => Record<string, unknown>;
-};
 /**
  * ToolsManager class - Handles all tool management operations
  */
@@ -37,7 +27,7 @@ export declare class ToolsManager {
     private readonly utilities?;
     protected mcpTools?: Record<string, Tool>;
     protected customTools?: Map<string, unknown>;
-    protected toolExecutor?: (toolName: string, params: unknown) => Promise<unknown>;
+    protected toolExecutor?: (toolName: string, params: unknown, options?: Record<string, unknown>) => Promise<unknown>;
     protected sessionId?: string;
     protected userId?: string;
     constructor(providerName: AIProviderName, directTools: Record<string, unknown>, neurolink?: NeuroLink | undefined, utilities?: ToolUtilities | undefined);

package/dist/lib/core/modules/ToolsManager.js

@@ -65,7 +65,7 @@ export class ToolsManager {
         try {
             // Store custom tools for use in getAllTools()
             this.customTools = sdk.customTools;
-            this.toolExecutor = sdk.executeTool;
+            this.toolExecutor = sdk.executeTool.bind(sdk);
             logger.debug(`[${functionTag}] Setting up tool executor for provider`, {
                 providerName: this.providerName,
                 availableCustomTools: sdk.customTools.size,
@@ -307,7 +307,7 @@ export class ToolsManager {
             }
             return createAISDKTool({
                 description: toolInfo.description || `Tool ${toolName}`,
-                parameters: finalSchema,
+                inputSchema: finalSchema, // AI SDK v6 uses inputSchema (not parameters)
                 execute: async (params) => {
                     const customToolSpan = tracers.sdk.startSpan("neurolink.tools.execute_custom", {
                         attributes: {
@@ -321,7 +321,23 @@ export class ToolsManager {
                         // Route through NeuroLink.executeTool() when available for MCP enhancement support
                         // (cache, middleware, annotations, circuit breaker, routing)
                         if (this.toolExecutor) {
-                            const result = await this.toolExecutor(toolName, params);
+                            // Per-tool timeout and retries flow through the customTools map
+                            // (set at registration via ToolRegistrationOptions).
+                            // The execute wrapper in registerTool already enforces timeouts,
+                            // but we also forward them to toolExecutor for MCP-level handling.
+                            const toolTimeoutMs = toolInfo.timeoutMs;
+                            const toolMaxRetries = toolInfo.maxRetries;
+                            const hasRegistrationOptions = toolTimeoutMs !== undefined || toolMaxRetries !== undefined;
+                            const result = await this.toolExecutor(toolName, params, hasRegistrationOptions
+                                ? {
+                                    ...(toolTimeoutMs !== undefined && {
+                                        timeout: toolTimeoutMs,
+                                    }),
+                                    ...(toolMaxRetries !== undefined && {
+                                        maxRetries: toolMaxRetries,
+                                    }),
+                                }
+                                : undefined);
                             const convertedResult = this.utilities?.convertToolResult
                                 ? await this.utilities.convertToolResult(result)
                                 : result;
@@ -425,7 +441,7 @@ export class ToolsManager {
             }
             return createAISDKTool({
                 description: tool.description || `External MCP tool ${tool.name}`,
-                parameters: finalSchema,
+                inputSchema: finalSchema, // AI SDK v6 uses inputSchema (not parameters)
                 execute: async (params) => {
                     // Emit tool start event
                     if (this.neurolink?.getEventEmitter) {

package/dist/lib/core/modules/Utilities.js

@@ -226,7 +226,9 @@ export class Utilities {
     createPermissiveZodSchema() {
         // Create a permissive record that accepts any object structure
         // This allows all parameters to pass through without validation issues
-        return z.record(z.unknown()).transform((data) => {
+        return z
+            .record(z.string(), z.unknown())
+            .transform((data) => {
             // Return the data as-is to preserve all parameter information
             return data;
         });

package/dist/lib/core/redisConversationMemoryManager.js

@@ -242,7 +242,7 @@ export class RedisConversationMemoryManager {
         try {
             const userSessionsKey = getUserSessionsKey(this.redisConfig, userId);
             const sessions = await this.redisClient.sMembers(userSessionsKey);
-            return sessions;
+            return Array.from(sessions).map(String);
         }
         catch (error) {
             logger.error("[RedisConversationMemoryManager] Failed to get user sessions", {
@@ -284,7 +284,7 @@ export class RedisConversationMemoryManager {
         try {
             const userSessionsKey = getUserSessionsKey(this.redisConfig, userId);
             const result = await this.redisClient.sRem(userSessionsKey, sessionId);
-            return result > 0;
+            return Number(result) > 0;
         }
         catch (error) {
             logger.error("[RedisConversationMemoryManager] Failed to remove session from user set", {
@@ -1091,7 +1091,7 @@ User message: "${userMessage}"`;
             try {
                 const redisKey = getSessionKey(this.redisConfig, sessionId, userId);
                 const result = await this.redisClient.del(redisKey);
-                if (result > 0) {
+                if (Number(result) > 0) {
                     // Remove session from user's session set
                     if (userId) {
                         await this.removeUserSession(userId, sessionId);

package/dist/lib/core/streamAnalytics.js

@@ -1,6 +1,7 @@
 import { createAnalytics } from "./analytics.js";
 import { logger } from "../utils/logger.js";
 import { extractTokenUsage, createEmptyTokenUsage, } from "../utils/tokenUtils.js";
+import { NoOutputGeneratedError } from "ai";
 /**
  * Base implementation for collecting analytics from Vercel AI SDK stream results
  */
@@ -22,7 +23,12 @@ export class BaseStreamAnalyticsCollector {
             return extractTokenUsage(usage);
         }
         catch (error) {
-            logger.warn("Failed to collect usage from stream result", { error });
+            if (NoOutputGeneratedError.isInstance(error)) {
+                logger.debug("No output generated from stream — returning empty usage");
+            }
+            else {
+                logger.warn("Failed to collect usage from stream result", { error });
+            }
             return createEmptyTokenUsage();
         }
     }
@@ -45,11 +51,17 @@ export class BaseStreamAnalyticsCollector {
             };
         }
         catch (error) {
-            logger.warn("Failed to collect metadata from stream result", { error });
-            const finishReason = await result.finishReason.catch(() => "error");
+            if (NoOutputGeneratedError.isInstance(error)) {
+                logger.debug("No output generated from stream — returning default metadata");
+            }
+            else {
+                logger.warn("Failed to collect metadata from stream result", {
+                    error,
+                });
+            }
             return {
                 timestamp: Date.now(),
-                finishReason: finishReason,
+                finishReason: "error",
             };
         }
     }
@@ -63,12 +75,14 @@ export class BaseStreamAnalyticsCollector {
                 this.collectUsage(result),
                 this.collectMetadata(result),
             ]);
-            // Get final text content and finish reason
+            // Get final text content and finish reason.
+            // Guard each promise individually: AI SDK v6 rejects all of these with
+            // NoOutputGeneratedError when the stream produced no output.
             const [content, finishReason, toolResults, toolCalls] = await Promise.all([
-                result.text,
-                result.finishReason,
-                result.toolResults || Promise.resolve([]),
-                result.toolCalls || Promise.resolve([]),
+                Promise.resolve(result.text).catch(() => ""),
+                Promise.resolve(result.finishReason).catch(() => "error"),
+                Promise.resolve(result.toolResults || []).catch(() => []),
+                Promise.resolve(result.toolCalls || []).catch(() => []),
             ]);
             // Create comprehensive analytics
             return createAnalytics(provider, model, {

package/dist/lib/evaluation/contextBuilder.d.ts

@@ -1,9 +1,9 @@
 /**
  * @file Implements the ContextBuilder class for creating rich evaluation context.
  */
+import type { LanguageModelV3CallOptions } from "@ai-sdk/provider";
 import type { EnhancedEvaluationContext, EvaluationResult } from "../types/evaluationTypes.js";
 import type { GenerateResult } from "../types/generateTypes.js";
-import type { LanguageModelV1CallOptions } from "ai";
 /**
  * Builds the enhanced context required for a RAGAS-style evaluation.
  * This class gathers data from the generation options and results to create a
@@ -20,7 +20,7 @@ export declare class ContextBuilder {
      * @param result The `GenerateResult` from the provider.
      * @returns An `EnhancedEvaluationContext` object ready for evaluation.
      */
-    buildContext(options: LanguageModelV1CallOptions, result: GenerateResult): EnhancedEvaluationContext;
+    buildContext(options: LanguageModelV3CallOptions, result: GenerateResult): EnhancedEvaluationContext;
     /**
      * Records the result of an evaluation and increments the internal attempt counter.
      * This is used to build up the `previousEvaluations` array for subsequent retries.

package/dist/lib/evaluation/contextBuilder.js

@@ -45,7 +45,7 @@ export class ContextBuilder {
             model: result.model || "unknown",
             generationParams: {
                 temperature: options.temperature,
-                maxTokens: options.maxTokens,
+                maxTokens: options.maxOutputTokens,
                 systemPrompt: systemPrompt || undefined,
             },
             toolExecutions,
@@ -89,7 +89,7 @@ export class ContextBuilder {
      */
     analyzeQuery(query) {
         const lowerCaseQuery = query.toLowerCase();
-        let type = "unknown";
+        let type;
         if (lowerCaseQuery.startsWith("what") ||
             lowerCaseQuery.startsWith("how") ||
             lowerCaseQuery.startsWith("why")) {

package/dist/lib/evaluation/index.d.ts

@@ -1,9 +1,9 @@
 /**
  * @file This file exports the main Evaluator class, which serves as the central entry point for the evaluation system.
  */
+import type { LanguageModelV3CallOptions } from "@ai-sdk/provider";
 import type { GenerateResult } from "../types/generateTypes.js";
 import type { EvaluationConfig } from "../types/evaluationTypes.js";
-import type { LanguageModelV1CallOptions } from "ai";
 import type { AutoEvaluationConfig } from "../types/middlewareTypes.js";
 import type { EvaluationData } from "../types/evaluation.js";
 /**
@@ -24,7 +24,7 @@ export declare class Evaluator {
      * @param result The `GenerateResult` from the provider.
      * @returns A promise that resolves to the `EvaluationResult`.
      */
-    evaluate(options: LanguageModelV1CallOptions, result: GenerateResult, threshold: number, config: AutoEvaluationConfig): Promise<EvaluationData>;
+    evaluate(options: LanguageModelV3CallOptions, result: GenerateResult, threshold: number, config: AutoEvaluationConfig): Promise<EvaluationData>;
     /**
      * Performs evaluation using the RAGAS-style model-based evaluator.
      *

package/dist/lib/evaluation/retryManager.js

@@ -54,7 +54,7 @@ export class RetryManager {
      * @returns A new, enhanced prompt string.
      */
     buildRetryPrompt(originalPrompt, feedback, attemptNumber) {
-        let instruction = "";
+        let instruction;
         switch (attemptNumber) {
             case 2: // First retry
                 instruction = `The previous response was not satisfactory. Please improve it based on the following feedback: "${feedback}".`;

package/dist/lib/factories/providerFactory.js

@@ -77,7 +77,7 @@ export class ProviderFactory {
                         result = new registration.constructor(model, resolvedProviderName, sdk, region);
                     }
                     catch (constructorError) {
-                        throw new Error(`Both factory function and constructor failed. Factory error: ${factoryError}. Constructor error: ${constructorError}`);
+                        throw new Error(`Both factory function and constructor failed. Factory error: ${factoryError}. Constructor error: ${constructorError}`, { cause: constructorError });
                     }
                 }
                 else {
@@ -88,7 +88,7 @@ export class ProviderFactory {
         }
         catch (error) {
             logger.error(`Failed to create provider ${resolvedProviderName}:`, error);
-            throw new Error(`Failed to create provider ${resolvedProviderName}: ${error}`);
+            throw new Error(`Failed to create provider ${resolvedProviderName}: ${error}`, { cause: error });
         }
     }
     /**

package/dist/lib/features/ppt/constants.js

@@ -372,7 +372,7 @@ const ADVANCED_MODEL_PATTERNS = [
     // OpenAI - multimodal with json-mode (openai, azure providers)
     "gpt-4o", // multimodal, highest quality, json-mode capable
     // Google - has advanced-reasoning (vertex, google-ai providers)
-    "gemini-3-pro", // advanced-reasoning, highest quality
+    "gemini-3.1-pro-preview", // advanced-reasoning, highest quality
     "gemini-2.5-pro", // advanced-reasoning, highest quality
 ];
 /**

package/dist/lib/features/ppt/presentationOrchestrator.js

@@ -11,9 +11,13 @@
  *
  * @module presentation/presentationOrchestrator
  */
-// Import pptxgenjs using dynamic import for ESM compatibility
-import PptxGenJSModule from "pptxgenjs";
-const PptxGenJS = PptxGenJSModule;
+// ESM/CJS interop: pptxgenjs v4 may double-wrap the default export under tsx/esbuild
+import PptxGenJSImport from "pptxgenjs";
+const PptxGenJSResolved = typeof PptxGenJSImport === "function"
+    ? PptxGenJSImport
+    : PptxGenJSImport
+        .default;
+const PptxGenJS = PptxGenJSResolved;
 import * as fs from "fs/promises";
 import { PPTError, PPT_ERROR_CODES } from "./types.js";
 import { generateContentPlan, postProcessPlan } from "./contentPlanner.js";

package/dist/lib/features/ppt/slideGenerator.d.ts

@@ -12,7 +12,8 @@
  *
  * @module presentation/slideGenerator
  */
-import PptxGenJS from "pptxgenjs";
+import PptxGenJSImport from "pptxgenjs";
+declare const PptxGenJS: typeof PptxGenJSImport;
 import type { SlideSchema, CompleteSlide, PptxPresentation, PptxSlide, SlideGeneratorConfig, SlideGenerationBatchResult } from "./types.js";
 /**
  * Generates individual slides with content, images, and proper layouts

package/dist/lib/features/ppt/slideGenerator.js

@@ -12,7 +12,12 @@
  *
  * @module presentation/slideGenerator
  */
-import PptxGenJS from "pptxgenjs";
+import PptxGenJSImport from "pptxgenjs";
+// ESM/CJS interop: pptxgenjs v4 may double-wrap the default export under tsx/esbuild
+const PptxGenJS = typeof PptxGenJSImport === "function"
+    ? PptxGenJSImport
+    : PptxGenJSImport
+        .default;
 import pLimit from "p-limit";
 import * as fs from "fs";
 import { SLIDE_DIMENSIONS } from "./types.js";