@juspay/neurolink 7.47.0 → 7.47.2

package/dist/lib/neurolink.js

@@ -18,7 +18,7 @@ import { mcpLogger } from "./utils/logger.js";
 import { SYSTEM_LIMITS } from "./core/constants.js";
 import { NANOSECOND_TO_MS_DIVISOR, TOOL_TIMEOUTS, RETRY_ATTEMPTS, RETRY_DELAYS, CIRCUIT_BREAKER, CIRCUIT_BREAKER_RESET_MS, MEMORY_THRESHOLDS, PROVIDER_TIMEOUTS, PERFORMANCE_THRESHOLDS, } from "./constants/index.js";
 import pLimit from "p-limit";
-import { toolRegistry } from "./mcp/toolRegistry.js";
+import { MCPToolRegistry } from "./mcp/toolRegistry.js";
 import { logger } from "./utils/logger.js";
 import { getBestProvider } from "./utils/providerUtils.js";
 import { ProviderRegistry } from "./factories/providerRegistry.js";
@@ -45,6 +45,7 @@ import { isZodSchema } from "./utils/schemaConversion.js";
 export class NeuroLink {
     mcpInitialized = false;
     emitter = new EventEmitter();
+    toolRegistry;
     autoDiscoveredServerInfos = [];
     // External MCP server management
     externalServerManager;
@@ -140,6 +141,7 @@ export 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
@@ -177,6 +179,7 @@ export class NeuroLink {
      * @throws {Error} When HITL configuration is invalid (if enabled)
      */
     constructor(config) {
+        this.toolRegistry = config?.toolRegistry || new MCPToolRegistry();
         // Initialize orchestration setting
         this.enableOrchestration = config?.enableOrchestration ?? false;
         // Read tool cache duration from environment variables, with a default
@@ -278,7 +281,7 @@ export class NeuroLink {
                 // Initialize HITL manager
                 this.hitlManager = new HITLManager(config.hitl);
                 // Inject HITL manager into tool registry
-                toolRegistry.setHITLManager(this.hitlManager);
+                this.toolRegistry.setHITLManager(this.hitlManager);
                 // Inject HITL manager into external server manager
                 this.externalServerManager.setHITLManager(this.hitlManager);
                 // Set up HITL event forwarding to main emitter
@@ -627,7 +630,7 @@ export class NeuroLink {
                 mcpLogger.debug("Direct tools server are disabled via environment variable.");
             }
             else {
-                await toolRegistry.registerServer("neurolink-direct", directToolsServer);
+                await this.toolRegistry.registerServer("neurolink-direct", directToolsServer);
                 mcpLogger.debug("[NeuroLink] Direct tools server registered successfully", {
                     serverId: "neurolink-direct",
                 });
@@ -1225,13 +1228,13 @@ export class NeuroLink {
             await this.initializeConversationMemoryForGeneration(generateInternalId, generateInternalStartTime, generateInternalHrTimeStart);
             const mcpResult = await this.attemptMCPGeneration(options, generateInternalId, generateInternalStartTime, generateInternalHrTimeStart, functionTag);
             if (mcpResult) {
-                await storeConversationTurn(this.conversationMemory, options, mcpResult);
+                await storeConversationTurn(this.conversationMemory, options, mcpResult, new Date(generateInternalStartTime));
                 this.emitter.emit("response:end", mcpResult.content || "");
                 return mcpResult;
             }
             const directResult = await this.directProviderGeneration(options);
             logger.debug(`[${functionTag}] Direct generation successful`);
-            await storeConversationTurn(this.conversationMemory, options, directResult);
+            await storeConversationTurn(this.conversationMemory, options, directResult, new Date(generateInternalStartTime));
             this.emitter.emit("response:end", directResult.content || "");
             this.emitter.emit("message", `Text generation completed successfully`);
             return directResult;
@@ -1371,7 +1374,7 @@ export class NeuroLink {
                     mcpInitialized: this.mcpInitialized,
                     mcpComponents: {
                         hasExternalServerManager: !!this.externalServerManager,
-                        hasToolRegistry: !!toolRegistry,
+                        hasToolRegistry: !!this.toolRegistry,
                         hasProviderRegistry: !!AIProviderFactory,
                     },
                     fallbackReason: "MCP_NOT_INITIALIZED",
@@ -1800,7 +1803,7 @@ export class NeuroLink {
                         try {
                             await self.conversationMemory.storeConversationTurn(enhancedOptions.context
                                 ?.sessionId, enhancedOptions.context
-                                ?.userId, originalPrompt ?? "", accumulatedContent);
+                                ?.userId, originalPrompt ?? "", accumulatedContent, new Date(startTime));
                             logger.debug("Stream conversation turn stored", {
                                 sessionId: enhancedOptions.context
                                     ?.sessionId,
@@ -2005,7 +2008,7 @@ export class NeuroLink {
                         const sessionId = enhancedOptions?.context?.sessionId;
                         const userId = enhancedOptions?.context
                             ?.userId;
-                        await self.conversationMemory.storeConversationTurn(sessionId || options.context?.sessionId, userId || options.context?.userId, originalPrompt ?? "", fallbackAccumulatedContent);
+                        await self.conversationMemory.storeConversationTurn(sessionId || options.context?.sessionId, userId || options.context?.userId, originalPrompt ?? "", fallbackAccumulatedContent, new Date(startTime));
                         logger.debug("Fallback stream conversation turn stored", {
                             sessionId: sessionId || options.context?.sessionId,
                             userInputLength: originalPrompt?.length ?? 0,
@@ -2403,7 +2406,7 @@ export class NeuroLink {
             // SMART DEFAULTS: Use utility to eliminate boilerplate creation
             const mcpServerInfo = createCustomToolServerInfo(name, convertedTool);
             // Register with toolRegistry using MCPServerInfo directly
-            toolRegistry.registerServer(mcpServerInfo);
+            this.toolRegistry.registerServer(mcpServerInfo);
             // Emit tool registration success event
             this.emitter.emit("tools-register:end", {
                 toolName: name,
@@ -2475,7 +2478,7 @@ export class NeuroLink {
     unregisterTool(name) {
         this.invalidateToolCache(); // Invalidate cache when a tool is unregistered
         const serverId = `custom-tool-${name}`;
-        const removed = toolRegistry.unregisterServer(serverId);
+        const removed = this.toolRegistry.unregisterServer(serverId);
         if (removed) {
             logger.info(`Unregistered custom tool: ${name}`);
         }
@@ -2487,7 +2490,7 @@ export class NeuroLink {
      */
     getCustomTools() {
         // Get tools from toolRegistry with smart category detection
-        const customTools = toolRegistry.getToolsByCategory(detectCategory({ isCustomTool: true }));
+        const customTools = this.toolRegistry.getToolsByCategory(detectCategory({ isCustomTool: true }));
         const toolMap = new Map();
         for (const tool of customTools) {
             const effectiveSchema = tool.inputSchema || tool.parameters;
@@ -2545,7 +2548,7 @@ export class NeuroLink {
                         hasShopId: !!executionContext.shopId,
                         sessionId: executionContext.sessionId,
                     });
-                    return await toolRegistry.executeTool(tool.name, params, executionContext);
+                    return await this.toolRegistry.executeTool(tool.name, params, executionContext);
                 },
             });
         }
@@ -2566,7 +2569,7 @@ export class NeuroLink {
                 serverInfo.tools = [];
             }
             // ZERO CONVERSIONS: Pass MCPServerInfo directly to toolRegistry
-            await toolRegistry.registerServer(serverInfo);
+            await this.toolRegistry.registerServer(serverInfo);
             mcpLogger.info(`[NeuroLink] Successfully registered in-memory server: ${serverId}`, {
                 category: serverInfo.metadata?.category,
                 provider: serverInfo.metadata?.provider,
@@ -2584,7 +2587,7 @@ export class NeuroLink {
      */
     getInMemoryServers() {
         // Get in-memory servers from toolRegistry
-        const serverInfos = toolRegistry.getBuiltInServerInfos();
+        const serverInfos = this.toolRegistry.getBuiltInServerInfos();
         const serverMap = new Map();
         for (const serverInfo of serverInfos) {
             if (detectCategory({
@@ -2603,7 +2606,7 @@ export class NeuroLink {
      */
     getInMemoryServerInfos() {
         // Get in-memory servers from centralized tool registry
-        const allServers = toolRegistry.getBuiltInServerInfos();
+        const allServers = this.toolRegistry.getBuiltInServerInfos();
         return allServers.filter((server) => detectCategory({
             existingCategory: server.metadata?.category,
             serverId: server.id,
@@ -2855,7 +2858,7 @@ export class NeuroLink {
                 storedContextKeys: Object.keys(storedContext),
                 finalContextKeys: Object.keys(context),
             });
-            const result = (await toolRegistry.executeTool(toolName, params, context));
+            const result = (await this.toolRegistry.executeTool(toolName, params, context));
             // ADD: Check if result indicates a failure and emit error event
             if (result &&
                 typeof result === "object" &&
@@ -2905,9 +2908,9 @@ export class NeuroLink {
             getAllToolsHrTimeStart: getAllToolsHrTimeStart.toString(),
             // 🔧 Tool registry state
             toolRegistryState: {
-                hasToolRegistry: !!toolRegistry,
+                hasToolRegistry: !!this.toolRegistry,
                 toolRegistrySize: 0, // Not accessible as size property
-                toolRegistryType: toolRegistry?.constructor?.name || "NOT_SET",
+                toolRegistryType: this.toolRegistry?.constructor?.name || "NOT_SET",
                 hasExternalServerManager: !!this.externalServerManager,
                 externalServerManagerType: this.externalServerManager?.constructor?.name || "NOT_SET",
             },
@@ -2926,7 +2929,7 @@ export class NeuroLink {
             // Optimized: Collect all tools with minimal object creation
             const allTools = new Map();
             // 1. Add MCP server tools (built-in direct tools)
-            const mcpToolsRaw = await toolRegistry.listTools();
+            const mcpToolsRaw = await this.toolRegistry.listTools();
             for (const tool of mcpToolsRaw) {
                 if (!allTools.has(tool.name)) {
                     const optimizedTool = optimizeToolForCollection(tool, {
@@ -2936,7 +2939,7 @@ export class NeuroLink {
                 }
             }
             // 2. Add custom tools from this NeuroLink instance
-            const customToolsRaw = toolRegistry.getToolsByCategory(detectCategory({ isCustomTool: true }));
+            const customToolsRaw = this.toolRegistry.getToolsByCategory(detectCategory({ isCustomTool: true }));
             for (const tool of customToolsRaw) {
                 if (!allTools.has(tool.name)) {
                     const optimizedTool = optimizeToolForCollection(tool, {
@@ -2952,7 +2955,7 @@ export class NeuroLink {
                 }
             }
             // 3. Add tools from in-memory MCP servers
-            const inMemoryToolsRaw = toolRegistry.getToolsByCategory("in-memory");
+            const inMemoryToolsRaw = this.toolRegistry.getToolsByCategory("in-memory");
             for (const tool of inMemoryToolsRaw) {
                 if (!allTools.has(tool.name)) {
                     const optimizedTool = optimizeToolForCollection(tool, {
@@ -3231,13 +3234,13 @@ export class NeuroLink {
             // Initialize MCP if not already initialized (loads external servers from config)
             await this.initializeMCP();
             // Get built-in tools
-            const allTools = await toolRegistry.listTools();
+            const allTools = await this.toolRegistry.listTools();
             // Get external MCP server statistics
             const externalStats = this.externalServerManager.getStatistics();
             // DIRECT RETURNS - ZERO conversion
             const externalMCPServers = this.externalServerManager.listServers();
             const inMemoryServerInfos = this.getInMemoryServerInfos();
-            const builtInServerInfos = toolRegistry.getBuiltInServerInfos();
+            const builtInServerInfos = this.toolRegistry.getBuiltInServerInfos();
             const autoDiscoveredServerInfos = this.getAutoDiscoveredServerInfos();
             // Calculate totals
             const totalServers = externalMCPServers.length +
@@ -3255,7 +3258,7 @@ export class NeuroLink {
                 autoDiscoveredCount: autoDiscoveredServerInfos.length,
                 totalTools,
                 autoDiscoveredServers: autoDiscoveredServerInfos,
-                customToolsCount: toolRegistry.getToolsByCategory(detectCategory({ isCustomTool: true })).length,
+                customToolsCount: this.toolRegistry.getToolsByCategory(detectCategory({ isCustomTool: true })).length,
                 inMemoryServersCount: inMemoryServerInfos.length,
                 externalMCPServersCount: externalMCPServers.length,
                 externalMCPConnectedCount: externalStats.connectedServers,
@@ -3271,7 +3274,7 @@ export class NeuroLink {
                 autoDiscoveredCount: 0,
                 totalTools: 0,
                 autoDiscoveredServers: [],
-                customToolsCount: toolRegistry.getToolsByCategory(detectCategory({ isCustomTool: true })).length,
+                customToolsCount: this.toolRegistry.getToolsByCategory(detectCategory({ isCustomTool: true })).length,
                 inMemoryServersCount: 0,
                 externalMCPServersCount: 0,
                 externalMCPConnectedCount: 0,
@@ -3290,7 +3293,7 @@ export class NeuroLink {
         return [
             ...this.externalServerManager.listServers(), // Direct return
             ...this.getInMemoryServerInfos(), // Direct return
-            ...toolRegistry.getBuiltInServerInfos(), // Direct return
+            ...this.toolRegistry.getBuiltInServerInfos(), // Direct return
             ...this.getAutoDiscoveredServerInfos(), // Direct return
         ];
     }
@@ -3303,7 +3306,7 @@ export class NeuroLink {
         try {
             // Test built-in tools
             if (serverId === "neurolink-direct") {
-                const tools = await toolRegistry.listTools();
+                const tools = await this.toolRegistry.listTools();
                 return tools.length > 0;
             }
             // Test in-memory servers
@@ -3480,7 +3483,7 @@ export class NeuroLink {
         const tools = {};
         let healthyCount = 0;
         // Get all tool names from toolRegistry
-        const allTools = await toolRegistry.listTools();
+        const allTools = await this.toolRegistry.listTools();
         const allToolNames = new Set(allTools.map((tool) => tool.name));
         for (const toolName of allToolNames) {
             const metrics = this.toolExecutionMetrics.get(toolName);
@@ -3630,9 +3633,10 @@ export 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
      */
-    async storeToolExecutions(sessionId, userId, toolCalls, toolResults) {
+    async storeToolExecutions(sessionId, userId, toolCalls, toolResults, currentTime) {
         // Check if tools are not empty
         const hasToolData = (toolCalls && toolCalls.length > 0) ||
             (toolResults && toolResults.length > 0);
@@ -3648,7 +3652,7 @@ export class NeuroLink {
         const redisMemory = this
             .conversationMemory;
         try {
-            await redisMemory.storeToolExecution(sessionId, userId, toolCalls, toolResults);
+            await redisMemory.storeToolExecution(sessionId, userId, toolCalls, toolResults, currentTime);
         }
         catch (error) {
             logger.warn("Failed to store tool executions", {
@@ -3908,7 +3912,7 @@ export class NeuroLink {
         try {
             const externalTools = this.externalServerManager.getServerTools(serverId);
             for (const tool of externalTools) {
-                toolRegistry.removeTool(tool.name);
+                this.toolRegistry.removeTool(tool.name);
                 mcpLogger.debug(`[NeuroLink] Unregistered external MCP tool from main registry: ${tool.name}`);
             }
         }
@@ -3921,7 +3925,7 @@ export class NeuroLink {
      */
     unregisterExternalMCPToolFromRegistry(toolName) {
         try {
-            toolRegistry.removeTool(toolName);
+            this.toolRegistry.removeTool(toolName);
             mcpLogger.debug(`[NeuroLink] Unregistered external MCP tool from main registry: ${toolName}`);
         }
         catch (error) {
@@ -3979,7 +3983,7 @@ export class NeuroLink {
         try {
             const externalTools = this.externalServerManager.getAllTools();
             for (const tool of externalTools) {
-                toolRegistry.removeTool(tool.name);
+                this.toolRegistry.removeTool(tool.name);
             }
             mcpLogger.debug(`[NeuroLink] Unregistered ${externalTools.length} external MCP tools from main registry`);
         }

package/dist/lib/providers/anthropic.js

@@ -105,7 +105,7 @@ export class AnthropicProvider extends BaseProvider {
                 toolChoice: shouldUseTools ? "auto" : "none",
                 abortSignal: timeoutController?.controller.signal,
                 onStepFinish: ({ toolCalls, toolResults }) => {
-                    this.handleToolExecutionStorage(toolCalls, toolResults, options).catch((error) => {
+                    this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
                         logger.warn("[AnthropicProvider] Failed to store tool executions", {
                             provider: this.providerName,
                             error: error instanceof Error ? error.message : String(error),

package/dist/lib/providers/anthropicBaseProvider.js

@@ -72,7 +72,7 @@ export class AnthropicProviderV2 extends BaseProvider {
                 toolChoice: "auto",
                 abortSignal: timeoutController?.controller.signal,
                 onStepFinish: ({ toolCalls, toolResults }) => {
-                    this.handleToolExecutionStorage(toolCalls, toolResults, options).catch((error) => {
+                    this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
                         logger.warn("[AnthropicBaseProvider] Failed to store tool executions", {
                             provider: this.providerName,
                             error: error instanceof Error ? error.message : String(error),

package/dist/lib/providers/azureOpenai.js

@@ -157,7 +157,7 @@ export class AzureOpenAIProvider extends BaseProvider {
                 tools,
                 toolChoice: shouldUseTools ? "auto" : "none",
                 onStepFinish: ({ toolCalls, toolResults }) => {
-                    this.handleToolExecutionStorage(toolCalls, toolResults, options).catch((error) => {
+                    this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
                         logger.warn("[AzureOpenaiProvider] Failed to store tool executions", {
                             provider: this.providerName,
                             error: error instanceof Error ? error.message : String(error),

package/dist/lib/providers/googleAiStudio.js

@@ -136,7 +136,7 @@ export class GoogleAIStudioProvider extends BaseProvider {
                 toolChoice: shouldUseTools ? "auto" : "none",
                 abortSignal: timeoutController?.controller.signal,
                 onStepFinish: ({ toolCalls, toolResults }) => {
-                    this.handleToolExecutionStorage(toolCalls, toolResults, options).catch((error) => {
+                    this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
                         logger.warn("[GoogleAiStudioProvider] Failed to store tool executions", {
                             provider: this.providerName,
                             error: error instanceof Error ? error.message : String(error),

package/dist/lib/providers/googleVertex.js

@@ -682,7 +682,7 @@ export class GoogleVertexProvider extends 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("[GoogleVertexProvider] Failed to store tool executions", {
                             provider: this.providerName,
                             error: error instanceof Error ? error.message : String(error),

package/dist/lib/providers/huggingFace.js

@@ -124,7 +124,7 @@ export class HuggingFaceProvider extends BaseProvider {
                 toolChoice: streamOptions.toolChoice, // Tool choice handled by prepareStreamOptions
                 abortSignal: timeoutController?.controller.signal,
                 onStepFinish: ({ toolCalls, toolResults }) => {
-                    this.handleToolExecutionStorage(toolCalls, toolResults, options).catch((error) => {
+                    this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
                         logger.warn("[HuggingFaceProvider] Failed to store tool executions", {
                             provider: this.providerName,
                             error: error instanceof Error ? error.message : String(error),

package/dist/lib/providers/litellm.js

@@ -132,7 +132,7 @@ export class LiteLLMProvider extends BaseProvider {
                 toolChoice: "auto",
                 abortSignal: timeoutController?.controller.signal,
                 onStepFinish: ({ toolCalls, toolResults }) => {
-                    this.handleToolExecutionStorage(toolCalls, toolResults, options).catch((error) => {
+                    this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
                         logger.warn("LiteLLMProvider] Failed to store tool executions", {
                             provider: this.providerName,
                             error: error instanceof Error ? error.message : String(error),

package/dist/lib/providers/mistral.js

@@ -61,7 +61,7 @@ export class MistralProvider extends BaseProvider {
                 toolChoice: shouldUseTools ? "auto" : "none",
                 abortSignal: timeoutController?.controller.signal,
                 onStepFinish: ({ toolCalls, toolResults }) => {
-                    this.handleToolExecutionStorage(toolCalls, toolResults, options).catch((error) => {
+                    this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
                         logger.warn("[MistralProvider] Failed to store tool executions", {
                             provider: this.providerName,
                             error: error instanceof Error ? error.message : String(error),

package/dist/lib/providers/openAI.js

@@ -276,7 +276,7 @@ export class OpenAIProvider extends 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("[OpenAIProvider] Failed to store tool executions", {
                             provider: this.providerName,
                             error: error instanceof Error ? error.message : String(error),

package/dist/lib/providers/openaiCompatible.js

@@ -168,7 +168,7 @@ export class OpenAICompatibleProvider extends BaseProvider {
                 toolChoice: "auto",
                 abortSignal: timeoutController?.controller.signal,
                 onStepFinish: ({ toolCalls, toolResults }) => {
-                    this.handleToolExecutionStorage(toolCalls, toolResults, options).catch((error) => {
+                    this.handleToolExecutionStorage(toolCalls, toolResults, options, new Date()).catch((error) => {
                         logger.warn("[OpenAiCompatibleProvider] Failed to store tool executions", {
                             provider: this.providerName,
                             error: error instanceof Error ? error.message : String(error),

package/dist/lib/utils/conversationMemory.d.ts

@@ -19,4 +19,4 @@ export declare function getConversationMessages(conversationMemory: Conversation
  * Store conversation turn for future context
  * Saves user messages and AI responses for conversation memory
  */
-export declare function storeConversationTurn(conversationMemory: ConversationMemoryManager | RedisConversationMemoryManager | null | undefined, originalOptions: TextGenerationOptions, result: TextGenerationResult): Promise<void>;
+export declare function storeConversationTurn(conversationMemory: ConversationMemoryManager | RedisConversationMemoryManager | null | undefined, originalOptions: TextGenerationOptions, result: TextGenerationResult, startTimeStamp?: Date | undefined): Promise<void>;

package/dist/lib/utils/conversationMemory.js

@@ -71,7 +71,7 @@ export async function getConversationMessages(conversationMemory, options) {
  * Store conversation turn for future context
  * Saves user messages and AI responses for conversation memory
  */
-export async function storeConversationTurn(conversationMemory, originalOptions, result) {
+export async function storeConversationTurn(conversationMemory, originalOptions, result, startTimeStamp) {
     logger.debug("[conversationMemoryUtils] storeConversationTurn called", {
         hasMemory: !!conversationMemory,
         memoryType: conversationMemory?.constructor?.name || "NONE",
@@ -102,7 +102,7 @@ export async function storeConversationTurn(conversationMemory, originalOptions,
     const userMessage = originalOptions.originalPrompt || originalOptions.prompt || "";
     const aiResponse = result.content;
     try {
-        await conversationMemory.storeConversationTurn(sessionId, userId, userMessage, aiResponse);
+        await conversationMemory.storeConversationTurn(sessionId, userId, userMessage, aiResponse, startTimeStamp);
         logger.debug("[conversationMemoryUtils] Conversation turn stored successfully", {
             sessionId,
             userId,

package/dist/lib/utils/conversationMemoryUtils.d.ts

@@ -18,7 +18,7 @@ export declare function getConversationMessages(conversationMemory: Conversation
  * Store conversation turn for future context
  * Saves user messages and AI responses for conversation memory
  */
-export declare function storeConversationTurn(conversationMemory: ConversationMemoryManager | undefined, originalOptions: TextGenerationOptions, result: TextGenerationResult): Promise<void>;
+export declare function storeConversationTurn(conversationMemory: ConversationMemoryManager | undefined, originalOptions: TextGenerationOptions, result: TextGenerationResult, startTimeStamp?: Date | undefined): Promise<void>;
 /**
  * Check if Redis is available for conversation memory
  */

package/dist/lib/utils/conversationMemoryUtils.js

@@ -48,7 +48,7 @@ export async function getConversationMessages(conversationMemory, options) {
  * Store conversation turn for future context
  * Saves user messages and AI responses for conversation memory
  */
-export async function storeConversationTurn(conversationMemory, originalOptions, result) {
+export async function storeConversationTurn(conversationMemory, originalOptions, result, startTimeStamp) {
     if (!conversationMemory || !originalOptions.context) {
         return;
     }
@@ -59,7 +59,7 @@ export async function storeConversationTurn(conversationMemory, originalOptions,
         return;
     }
     try {
-        await conversationMemory.storeConversationTurn(sessionId, userId, originalOptions.originalPrompt || originalOptions.prompt || "", result.content);
+        await conversationMemory.storeConversationTurn(sessionId, userId, originalOptions.originalPrompt || originalOptions.prompt || "", result.content, startTimeStamp);
         logger.debug("Conversation turn stored", {
             sessionId,
             userId,

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