@juspay/neurolink 9.30.0 → 9.31.0

package/dist/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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";