@juspay/neurolink 7.47.0 → 7.47.2

package/CHANGELOG.md

@@ -1,3 +1,15 @@
+## [7.47.2](https://github.com/juspay/neurolink/compare/v7.47.1...v7.47.2) (2025-09-26)
+
+### Bug Fixes
+
+- **(timestamp):** Incorrect timestamps being stored in redis ([2d66232](https://github.com/juspay/neurolink/commit/2d6623275bc4c1f5986957d476ddcf3933ba61e4))
+
+## [7.47.1](https://github.com/juspay/neurolink/compare/v7.47.0...v7.47.1) (2025-09-26)
+
+### Bug Fixes
+
+- **(tools):** Unregistered tools getting called ([45fd67a](https://github.com/juspay/neurolink/commit/45fd67af418b5e458ce6a261a7891234a8d489b8))
+
 ## [7.47.0](https://github.com/juspay/neurolink/compare/v7.46.0...v7.47.0) (2025-09-25)
 
 ### Features

package/dist/core/baseProvider.d.ts

@@ -227,7 +227,7 @@ export declare abstract class BaseProvider implements AIProvider {
     /**
      * Check if tool executions should be stored and handle storage
      */
-    protected handleToolExecutionStorage(toolCalls: unknown[], toolResults: unknown[], options: TextGenerationOptions | StreamOptions): Promise<void>;
+    protected handleToolExecutionStorage(toolCalls: unknown[], toolResults: unknown[], options: TextGenerationOptions | StreamOptions, currentTime: Date): Promise<void>;
     /**
      * Utility method to chunk large prompts into smaller pieces
      * @param prompt The prompt to chunk

package/dist/core/baseProvider.js

@@ -300,7 +300,7 @@ export class BaseProvider {
             onStepFinish: ({ toolCalls, toolResults }) => {
                 logger.info("Tool execution completed", { toolResults, toolCalls });
                 // Handle tool execution storage
-                this.handleToolExecutionStorage(toolCalls, toolResults, options).catch((error) => {
+                this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
                     logger.warn("[BaseProvider] Failed to store tool executions", {
                         provider: this.providerName,
                         error: error instanceof Error ? error.message : String(error),
@@ -1499,7 +1499,7 @@ export class BaseProvider {
     /**
      * Check if tool executions should be stored and handle storage
      */
-    async handleToolExecutionStorage(toolCalls, toolResults, options) {
+    async handleToolExecutionStorage(toolCalls, toolResults, options, currentTime) {
         // Check if tools are not empty
         const hasToolData = (toolCalls && toolCalls.length > 0) ||
             (toolResults && toolResults.length > 0);
@@ -1515,7 +1515,7 @@ export class BaseProvider {
         const userId = options.context?.userId ||
             options.userId;
         try {
-            await this.neurolink.storeToolExecutions(sessionId, userId, toolCalls, toolResults);
+            await this.neurolink.storeToolExecutions(sessionId, userId, toolCalls, toolResults, currentTime);
         }
         catch (error) {
             logger.warn("[BaseProvider] Failed to store tool executions", {

package/dist/core/conversationMemoryManager.d.ts

@@ -16,7 +16,7 @@ export declare class ConversationMemoryManager {
      * Store a conversation turn for a session
      * ULTRA-OPTIMIZED: Direct ChatMessage[] storage with zero conversion overhead
      */
-    storeConversationTurn(sessionId: string, userId: string | undefined, userMessage: string, aiResponse: string): Promise<void>;
+    storeConversationTurn(sessionId: string, userId: string | undefined, userMessage: string, aiResponse: string, _startTimeStamp: Date | undefined): Promise<void>;
     /**
      * Build context messages for AI prompt injection (ULTRA-OPTIMIZED)
      * Returns pre-stored message array with zero conversion overhead

package/dist/core/conversationMemoryManager.js

@@ -36,7 +36,7 @@ export class ConversationMemoryManager {
      * Store a conversation turn for a session
      * ULTRA-OPTIMIZED: Direct ChatMessage[] storage with zero conversion overhead
      */
-    async storeConversationTurn(sessionId, userId, userMessage, aiResponse) {
+    async storeConversationTurn(sessionId, userId, userMessage, aiResponse, _startTimeStamp) {
         await this.ensureInitialized();
         try {
             // Get or create session

package/dist/core/redisConversationMemoryManager.d.ts

@@ -60,11 +60,11 @@ export declare class RedisConversationMemoryManager {
         result?: unknown;
         error?: string;
         [key: string]: unknown;
-    }>): Promise<void>;
+    }>, currentTime?: Date): Promise<void>;
     /**
      * Store a conversation turn for a session
      */
-    storeConversationTurn(sessionId: string, userId: string | undefined, userMessage: string, aiResponse: string): Promise<void>;
+    storeConversationTurn(sessionId: string, userId: string | undefined, userMessage: string, aiResponse: string, startTimeStamp: Date | undefined): Promise<void>;
     /**
      * Build context messages for AI prompt injection
      */

package/dist/core/redisConversationMemoryManager.js

@@ -156,7 +156,7 @@ export class RedisConversationMemoryManager {
     /**
      * Store tool execution data for a session (temporarily to avoid race conditions)
      */
-    async storeToolExecution(sessionId, userId, toolCalls, toolResults) {
+    async storeToolExecution(sessionId, userId, toolCalls, toolResults, currentTime) {
         logger.debug("[RedisConversationMemoryManager] Storing tool execution temporarily", {
             sessionId,
             userId,
@@ -168,8 +168,14 @@ export class RedisConversationMemoryManager {
             const pendingKey = `${sessionId}:${normalizedUserId}`;
             // Store tool execution data temporarily to prevent race conditions
             const pendingData = {
-                toolCalls: toolCalls || [],
-                toolResults: toolResults || [],
+                toolCalls: (toolCalls || []).map((call) => ({
+                    ...call,
+                    timestamp: currentTime,
+                })),
+                toolResults: (toolResults || []).map((result) => ({
+                    ...result,
+                    timestamp: currentTime,
+                })),
                 timestamp: Date.now(),
             };
             // Check if there's existing pending data and merge
@@ -214,7 +220,7 @@ export class RedisConversationMemoryManager {
     /**
      * Store a conversation turn for a session
      */
-    async storeConversationTurn(sessionId, userId, userMessage, aiResponse) {
+    async storeConversationTurn(sessionId, userId, userMessage, aiResponse, startTimeStamp) {
         logger.debug("[RedisConversationMemoryManager] Storing conversation turn", {
             sessionId,
             userId,
@@ -294,8 +300,8 @@ export class RedisConversationMemoryManager {
                     title: "New Conversation", // Temporary title until generated
                     sessionId,
                     userId: normalizedUserId,
-                    createdAt: currentTime,
-                    updatedAt: currentTime,
+                    createdAt: startTimeStamp?.toISOString() || currentTime,
+                    updatedAt: startTimeStamp?.toISOString() || currentTime,
                     messages: [],
                 };
             }
@@ -312,7 +318,7 @@ export class RedisConversationMemoryManager {
             // Add new messages to conversation history with new format
             const userMsg = {
                 id: this.generateMessageId(conversation),
-                timestamp: this.generateTimestamp(),
+                timestamp: startTimeStamp?.toISOString() || this.generateTimestamp(),
                 role: "user",
                 content: userMessage,
             };
@@ -868,7 +874,7 @@ User message: "${userMessage}`;
             toolCallMap.set(toolCallId, toolName);
             const toolCallMessage = {
                 id: this.generateMessageId(conversation),
-                timestamp: this.generateTimestamp(),
+                timestamp: toolCall.timestamp?.toISOString() || this.generateTimestamp(),
                 role: "tool_call",
                 content: "", // Can be empty for tool calls
                 tool: toolName,
@@ -885,7 +891,7 @@ User message: "${userMessage}`;
             const toolName = toolCallMap.get(toolCallId) || "unknown";
             const toolResultMessage = {
                 id: this.generateMessageId(conversation),
-                timestamp: this.generateTimestamp(),
+                timestamp: toolResult.timestamp?.toISOString() || this.generateTimestamp(),
                 role: "tool_result",
                 content: "", // Can be empty for tool results
                 tool: toolName, // Now correctly extracted from tool call mapping

package/dist/factories/providerRegistry.js

@@ -87,7 +87,7 @@ export class ProviderRegistry {
             // Register Amazon SageMaker provider
             ProviderFactory.registerProvider(AIProviderName.SAGEMAKER, async (modelName, _providerName, _sdk, region) => {
                 const { AmazonSageMakerProvider } = await import("../providers/amazonSagemaker.js");
-                return new AmazonSageMakerProvider(modelName, region);
+                return new AmazonSageMakerProvider(modelName, undefined, region);
             }, process.env.SAGEMAKER_MODEL || "sagemaker-model", ["sagemaker", "aws-sagemaker"]);
             logger.debug("All providers registered successfully");
             this.registered = true;

package/dist/lib/core/baseProvider.d.ts

@@ -227,7 +227,7 @@ export declare abstract class BaseProvider implements AIProvider {
     /**
      * Check if tool executions should be stored and handle storage
      */
-    protected handleToolExecutionStorage(toolCalls: unknown[], toolResults: unknown[], options: TextGenerationOptions | StreamOptions): Promise<void>;
+    protected handleToolExecutionStorage(toolCalls: unknown[], toolResults: unknown[], options: TextGenerationOptions | StreamOptions, currentTime: Date): Promise<void>;
     /**
      * Utility method to chunk large prompts into smaller pieces
      * @param prompt The prompt to chunk

package/dist/lib/core/baseProvider.js

@@ -300,7 +300,7 @@ export class BaseProvider {
             onStepFinish: ({ toolCalls, toolResults }) => {
                 logger.info("Tool execution completed", { toolResults, toolCalls });
                 // Handle tool execution storage
-                this.handleToolExecutionStorage(toolCalls, toolResults, options).catch((error) => {
+                this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
                     logger.warn("[BaseProvider] Failed to store tool executions", {
                         provider: this.providerName,
                         error: error instanceof Error ? error.message : String(error),
@@ -1499,7 +1499,7 @@ export class BaseProvider {
     /**
      * Check if tool executions should be stored and handle storage
      */
-    async handleToolExecutionStorage(toolCalls, toolResults, options) {
+    async handleToolExecutionStorage(toolCalls, toolResults, options, currentTime) {
         // Check if tools are not empty
         const hasToolData = (toolCalls && toolCalls.length > 0) ||
             (toolResults && toolResults.length > 0);
@@ -1515,7 +1515,7 @@ export class BaseProvider {
         const userId = options.context?.userId ||
             options.userId;
         try {
-            await this.neurolink.storeToolExecutions(sessionId, userId, toolCalls, toolResults);
+            await this.neurolink.storeToolExecutions(sessionId, userId, toolCalls, toolResults, currentTime);
         }
         catch (error) {
             logger.warn("[BaseProvider] Failed to store tool executions", {

package/dist/lib/core/conversationMemoryManager.d.ts

@@ -16,7 +16,7 @@ export declare class ConversationMemoryManager {
      * Store a conversation turn for a session
      * ULTRA-OPTIMIZED: Direct ChatMessage[] storage with zero conversion overhead
      */
-    storeConversationTurn(sessionId: string, userId: string | undefined, userMessage: string, aiResponse: string): Promise<void>;
+    storeConversationTurn(sessionId: string, userId: string | undefined, userMessage: string, aiResponse: string, _startTimeStamp: Date | undefined): Promise<void>;
     /**
      * Build context messages for AI prompt injection (ULTRA-OPTIMIZED)
      * Returns pre-stored message array with zero conversion overhead

package/dist/lib/core/conversationMemoryManager.js

@@ -36,7 +36,7 @@ export class ConversationMemoryManager {
      * Store a conversation turn for a session
      * ULTRA-OPTIMIZED: Direct ChatMessage[] storage with zero conversion overhead
      */
-    async storeConversationTurn(sessionId, userId, userMessage, aiResponse) {
+    async storeConversationTurn(sessionId, userId, userMessage, aiResponse, _startTimeStamp) {
         await this.ensureInitialized();
         try {
             // Get or create session

package/dist/lib/core/redisConversationMemoryManager.d.ts

@@ -60,11 +60,11 @@ export declare class RedisConversationMemoryManager {
         result?: unknown;
         error?: string;
         [key: string]: unknown;
-    }>): Promise<void>;
+    }>, currentTime?: Date): Promise<void>;
     /**
      * Store a conversation turn for a session
      */
-    storeConversationTurn(sessionId: string, userId: string | undefined, userMessage: string, aiResponse: string): Promise<void>;
+    storeConversationTurn(sessionId: string, userId: string | undefined, userMessage: string, aiResponse: string, startTimeStamp: Date | undefined): Promise<void>;
     /**
      * Build context messages for AI prompt injection
      */

package/dist/lib/core/redisConversationMemoryManager.js

@@ -156,7 +156,7 @@ export class RedisConversationMemoryManager {
     /**
      * Store tool execution data for a session (temporarily to avoid race conditions)
      */
-    async storeToolExecution(sessionId, userId, toolCalls, toolResults) {
+    async storeToolExecution(sessionId, userId, toolCalls, toolResults, currentTime) {
         logger.debug("[RedisConversationMemoryManager] Storing tool execution temporarily", {
             sessionId,
             userId,
@@ -168,8 +168,14 @@ export class RedisConversationMemoryManager {
             const pendingKey = `${sessionId}:${normalizedUserId}`;
             // Store tool execution data temporarily to prevent race conditions
             const pendingData = {
-                toolCalls: toolCalls || [],
-                toolResults: toolResults || [],
+                toolCalls: (toolCalls || []).map((call) => ({
+                    ...call,
+                    timestamp: currentTime,
+                })),
+                toolResults: (toolResults || []).map((result) => ({
+                    ...result,
+                    timestamp: currentTime,
+                })),
                 timestamp: Date.now(),
             };
             // Check if there's existing pending data and merge
@@ -214,7 +220,7 @@ export class RedisConversationMemoryManager {
     /**
      * Store a conversation turn for a session
      */
-    async storeConversationTurn(sessionId, userId, userMessage, aiResponse) {
+    async storeConversationTurn(sessionId, userId, userMessage, aiResponse, startTimeStamp) {
         logger.debug("[RedisConversationMemoryManager] Storing conversation turn", {
             sessionId,
             userId,
@@ -294,8 +300,8 @@ export class RedisConversationMemoryManager {
                     title: "New Conversation", // Temporary title until generated
                     sessionId,
                     userId: normalizedUserId,
-                    createdAt: currentTime,
-                    updatedAt: currentTime,
+                    createdAt: startTimeStamp?.toISOString() || currentTime,
+                    updatedAt: startTimeStamp?.toISOString() || currentTime,
                     messages: [],
                 };
             }
@@ -312,7 +318,7 @@ export class RedisConversationMemoryManager {
             // Add new messages to conversation history with new format
             const userMsg = {
                 id: this.generateMessageId(conversation),
-                timestamp: this.generateTimestamp(),
+                timestamp: startTimeStamp?.toISOString() || this.generateTimestamp(),
                 role: "user",
                 content: userMessage,
             };
@@ -868,7 +874,7 @@ User message: "${userMessage}`;
             toolCallMap.set(toolCallId, toolName);
             const toolCallMessage = {
                 id: this.generateMessageId(conversation),
-                timestamp: this.generateTimestamp(),
+                timestamp: toolCall.timestamp?.toISOString() || this.generateTimestamp(),
                 role: "tool_call",
                 content: "", // Can be empty for tool calls
                 tool: toolName,
@@ -885,7 +891,7 @@ User message: "${userMessage}`;
             const toolName = toolCallMap.get(toolCallId) || "unknown";
             const toolResultMessage = {
                 id: this.generateMessageId(conversation),
-                timestamp: this.generateTimestamp(),
+                timestamp: toolResult.timestamp?.toISOString() || this.generateTimestamp(),
                 role: "tool_result",
                 content: "", // Can be empty for tool results
                 tool: toolName, // Now correctly extracted from tool call mapping

package/dist/lib/factories/providerRegistry.js

@@ -87,7 +87,7 @@ export class ProviderRegistry {
             // Register Amazon SageMaker provider
             ProviderFactory.registerProvider(AIProviderName.SAGEMAKER, async (modelName, _providerName, _sdk, region) => {
                 const { AmazonSageMakerProvider } = await import("../providers/amazonSagemaker.js");
-                return new AmazonSageMakerProvider(modelName, region);
+                return new AmazonSageMakerProvider(modelName, undefined, region);
             }, process.env.SAGEMAKER_MODEL || "sagemaker-model", ["sagemaker", "aws-sagemaker"]);
             logger.debug("All providers registered successfully");
             this.registered = true;

package/dist/lib/neurolink.d.ts

@@ -6,6 +6,7 @@
  * Uses real MCP infrastructure for tool discovery and execution.
  */
 import type { TextGenerationOptions, TextGenerationResult } from "./types/index.js";
+import { MCPToolRegistry } from "./mcp/toolRegistry.js";
 import type { GenerateOptions, GenerateResult } from "./types/generateTypes.js";
 import type { StreamOptions, StreamResult } from "./types/streamTypes.js";
 import type { MCPServerInfo, MCPExecutableTool } from "./types/mcpTypes.js";
@@ -46,6 +47,7 @@ export interface MCPStatus {
 export declare class NeuroLink {
     private mcpInitialized;
     private emitter;
+    private toolRegistry;
     private autoDiscoveredServerInfos;
     private externalServerManager;
     private toolCache;
@@ -99,6 +101,7 @@ export declare class NeuroLink {
      * @param config.hitl.dangerousActions - Keywords that trigger confirmation (default: ['delete', 'remove', 'drop'])
      * @param config.hitl.timeout - Confirmation timeout in milliseconds (default: 30000)
      * @param config.hitl.allowArgumentModification - Allow users to modify tool parameters (default: true)
+     * @param config.toolRegistry - Optional tool registry instance for advanced use cases (default: new MCPToolRegistry())
      *
      * @example
      * ```typescript
@@ -139,6 +142,7 @@ export declare class NeuroLink {
         conversationMemory?: Partial<ConversationMemoryConfig>;
         enableOrchestration?: boolean;
         hitl?: HITLConfig;
+        toolRegistry?: MCPToolRegistry;
     });
     /**
      * Initialize provider registry with security settings
@@ -919,6 +923,7 @@ export declare class NeuroLink {
      * @param userId - User identifier (optional)
      * @param toolCalls - Array of tool calls
      * @param toolResults - Array of tool results
+     * @param currentTime - Date when the tool execution occurred (optional)
      * @returns Promise resolving when storage is complete
      */
     storeToolExecutions(sessionId: string, userId: string | undefined, toolCalls: Array<{
@@ -931,7 +936,7 @@ export declare class NeuroLink {
         result?: unknown;
         error?: string;
         [key: string]: unknown;
-    }>): Promise<void>;
+    }>, currentTime?: Date): Promise<void>;
     /**
      * Check if tool execution storage is available
      * @returns boolean indicating if Redis storage is configured and available