@juspay/neurolink 7.37.0 → 7.37.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (61) hide show
  1. package/CHANGELOG.md +6 -0
  2. package/dist/cli/commands/config.d.ts +18 -18
  3. package/dist/cli/factories/commandFactory.d.ts +24 -0
  4. package/dist/cli/factories/commandFactory.js +297 -245
  5. package/dist/core/baseProvider.d.ts +40 -3
  6. package/dist/core/baseProvider.js +689 -352
  7. package/dist/core/constants.d.ts +2 -30
  8. package/dist/core/constants.js +15 -43
  9. package/dist/factories/providerFactory.js +23 -6
  10. package/dist/index.d.ts +3 -2
  11. package/dist/index.js +4 -3
  12. package/dist/lib/core/baseProvider.d.ts +40 -3
  13. package/dist/lib/core/baseProvider.js +689 -352
  14. package/dist/lib/core/constants.d.ts +2 -30
  15. package/dist/lib/core/constants.js +15 -43
  16. package/dist/lib/factories/providerFactory.js +23 -6
  17. package/dist/lib/index.d.ts +3 -2
  18. package/dist/lib/index.js +4 -3
  19. package/dist/lib/mcp/externalServerManager.js +2 -2
  20. package/dist/lib/mcp/registry.js +2 -2
  21. package/dist/lib/mcp/servers/agent/directToolsServer.js +19 -10
  22. package/dist/lib/mcp/toolRegistry.js +4 -8
  23. package/dist/lib/neurolink.d.ts +62 -27
  24. package/dist/lib/neurolink.js +415 -719
  25. package/dist/lib/providers/amazonBedrock.js +2 -2
  26. package/dist/lib/providers/googleVertex.d.ts +3 -23
  27. package/dist/lib/providers/googleVertex.js +14 -342
  28. package/dist/lib/providers/openAI.d.ts +23 -0
  29. package/dist/lib/providers/openAI.js +313 -6
  30. package/dist/lib/providers/sagemaker/language-model.d.ts +2 -2
  31. package/dist/lib/sdk/toolRegistration.js +18 -1
  32. package/dist/lib/types/common.d.ts +98 -0
  33. package/dist/lib/types/streamTypes.d.ts +13 -6
  34. package/dist/lib/types/typeAliases.d.ts +3 -2
  35. package/dist/lib/utils/parameterValidation.js +6 -25
  36. package/dist/lib/utils/promptRedaction.js +4 -4
  37. package/dist/lib/utils/schemaConversion.d.ts +14 -0
  38. package/dist/lib/utils/schemaConversion.js +140 -0
  39. package/dist/lib/utils/transformationUtils.js +143 -5
  40. package/dist/mcp/externalServerManager.js +2 -2
  41. package/dist/mcp/registry.js +2 -2
  42. package/dist/mcp/servers/agent/directToolsServer.js +19 -10
  43. package/dist/mcp/toolRegistry.js +4 -8
  44. package/dist/neurolink.d.ts +62 -27
  45. package/dist/neurolink.js +415 -719
  46. package/dist/providers/amazonBedrock.js +2 -2
  47. package/dist/providers/googleVertex.d.ts +3 -23
  48. package/dist/providers/googleVertex.js +14 -342
  49. package/dist/providers/openAI.d.ts +23 -0
  50. package/dist/providers/openAI.js +313 -6
  51. package/dist/providers/sagemaker/language-model.d.ts +2 -2
  52. package/dist/sdk/toolRegistration.js +18 -1
  53. package/dist/types/common.d.ts +98 -0
  54. package/dist/types/streamTypes.d.ts +13 -6
  55. package/dist/types/typeAliases.d.ts +3 -2
  56. package/dist/utils/parameterValidation.js +6 -25
  57. package/dist/utils/promptRedaction.js +4 -4
  58. package/dist/utils/schemaConversion.d.ts +14 -0
  59. package/dist/utils/schemaConversion.js +140 -0
  60. package/dist/utils/transformationUtils.js +143 -5
  61. package/package.json +3 -2
@@ -0,0 +1,14 @@
1
+ import type { ZodUnknownSchema } from "../types/tools.js";
2
+ /**
3
+ * Convert Zod schema to JSON Schema format for Claude AI
4
+ */
5
+ export declare function convertZodToJsonSchema(zodSchema: ZodUnknownSchema): object;
6
+ /**
7
+ * Check if a value is a Zod schema
8
+ */
9
+ export declare function isZodSchema(value: unknown): boolean;
10
+ /**
11
+ * Convert JSON Schema to Zod schema format using official json-schema-to-zod library
12
+ * This ensures complete preservation of all schema structure and validation rules
13
+ */
14
+ export declare function convertJsonSchemaToZod(jsonSchema: Record<string, unknown>): ZodUnknownSchema;
@@ -0,0 +1,140 @@
1
+ import { zodToJsonSchema } from "zod-to-json-schema";
2
+ import { jsonSchemaToZod } from "json-schema-to-zod";
3
+ import { z } from "zod";
4
+ import { logger } from "./logger.js";
5
+ /**
6
+ * Convert Zod schema to JSON Schema format for Claude AI
7
+ */
8
+ export function convertZodToJsonSchema(zodSchema) {
9
+ try {
10
+ // Use a type assertion that bypasses the infinite recursion check
11
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
12
+ const jsonSchema = zodToJsonSchema(zodSchema, {
13
+ name: "ToolParameters",
14
+ target: "jsonSchema7",
15
+ errorMessages: true,
16
+ });
17
+ // CRITICAL FIX: Ensure schema always has a proper type field for Google Vertex AI
18
+ if (!jsonSchema.type) {
19
+ // Default to "object" type if not specified, as most tool schemas are objects
20
+ jsonSchema.type = "object";
21
+ // If no properties, ensure it's at least an empty object schema
22
+ if (!jsonSchema.properties) {
23
+ jsonSchema.properties = {};
24
+ }
25
+ logger.info(`[SCHEMA-TYPE-FIX] Added missing type field to JSON Schema`, {
26
+ originalType: undefined,
27
+ fixedType: jsonSchema.type,
28
+ hasProperties: !!jsonSchema.properties,
29
+ addedEmptyProperties: !jsonSchema.properties,
30
+ });
31
+ }
32
+ logger.debug("Converted Zod schema to JSON Schema", {
33
+ hasProperties: !!jsonSchema.properties,
34
+ propertiesCount: Object.keys(jsonSchema.properties || {}).length,
35
+ schemaType: jsonSchema.type,
36
+ hasTypeField: !!jsonSchema.type,
37
+ });
38
+ return jsonSchema;
39
+ }
40
+ catch (error) {
41
+ logger.warn("Failed to convert Zod schema to JSON Schema", {
42
+ error: error instanceof Error ? error.message : String(error),
43
+ });
44
+ // Return a valid empty object schema instead of empty object
45
+ return {
46
+ type: "object",
47
+ properties: {},
48
+ };
49
+ }
50
+ }
51
+ /**
52
+ * Check if a value is a Zod schema
53
+ */
54
+ export function isZodSchema(value) {
55
+ return !!(value &&
56
+ typeof value === "object" &&
57
+ "_def" in value &&
58
+ typeof value.parse === "function");
59
+ }
60
+ /**
61
+ * Convert JSON Schema to Zod schema format using official json-schema-to-zod library
62
+ * This ensures complete preservation of all schema structure and validation rules
63
+ */
64
+ export function convertJsonSchemaToZod(jsonSchema) {
65
+ const startTime = Date.now();
66
+ try {
67
+ // Handle empty or invalid schemas
68
+ if (!jsonSchema || typeof jsonSchema !== "object") {
69
+ logger.debug("🔍 [SCHEMA-CONVERSION] Invalid or empty JSON schema, using fallback");
70
+ return z.object({}).passthrough();
71
+ }
72
+ // Log detailed input schema for debugging
73
+ logger.debug("🔍 [SCHEMA-CONVERSION] ===== STARTING OFFICIAL LIBRARY CONVERSION =====");
74
+ logger.debug("🔍 [SCHEMA-CONVERSION] Input JSON Schema:", {
75
+ type: jsonSchema.type,
76
+ hasProperties: !!jsonSchema.properties,
77
+ propertiesCount: jsonSchema.properties
78
+ ? Object.keys(jsonSchema.properties).length
79
+ : 0,
80
+ requiredCount: Array.isArray(jsonSchema.required)
81
+ ? jsonSchema.required.length
82
+ : 0,
83
+ required: jsonSchema.required,
84
+ sampleProperties: jsonSchema.properties
85
+ ? Object.keys(jsonSchema.properties).slice(0, 5)
86
+ : [],
87
+ });
88
+ // Use official library to convert JSON Schema to Zod code
89
+ const zodCodeResult = jsonSchemaToZod(jsonSchema, {
90
+ module: "esm",
91
+ name: "schema",
92
+ });
93
+ logger.debug("🔍 [SCHEMA-CONVERSION] Generated Zod code:", {
94
+ codeLength: zodCodeResult.length,
95
+ codePreview: zodCodeResult.substring(0, 200) + "...",
96
+ });
97
+ // Extract the actual Zod schema expression from the generated code
98
+ // Generated code looks like: "import { z } from "zod"\n\nexport const schema = z.object({...})\n"
99
+ const schemaMatch = zodCodeResult.match(/export const schema = (z\..+?)(?:\n|$)/s);
100
+ if (!schemaMatch) {
101
+ throw new Error("Could not extract Zod schema from generated code");
102
+ }
103
+ const schemaExpression = schemaMatch[1].trim();
104
+ logger.debug("🔍 [SCHEMA-CONVERSION] Extracted schema expression:", {
105
+ expression: schemaExpression.substring(0, 300) + "...",
106
+ });
107
+ // Use Function constructor instead of eval for better scope control
108
+ const createZodSchema = new Function("z", `return ${schemaExpression}`);
109
+ const zodSchema = createZodSchema(z);
110
+ const conversionTime = Date.now() - startTime;
111
+ logger.debug("🔍 [SCHEMA-CONVERSION] ===== CONVERSION SUCCESSFUL =====", {
112
+ inputType: jsonSchema.type,
113
+ propertiesCount: jsonSchema.properties
114
+ ? Object.keys(jsonSchema.properties).length
115
+ : 0,
116
+ requiredCount: Array.isArray(jsonSchema.required)
117
+ ? jsonSchema.required.length
118
+ : 0,
119
+ conversionSuccess: true,
120
+ conversionTimeMs: conversionTime,
121
+ libraryUsed: "json-schema-to-zod-official",
122
+ zodSchemaType: zodSchema?.constructor?.name || "unknown",
123
+ });
124
+ return zodSchema;
125
+ }
126
+ catch (error) {
127
+ const conversionTime = Date.now() - startTime;
128
+ logger.warn("🚨 [SCHEMA-CONVERSION] Official library conversion failed, using passthrough fallback", {
129
+ error: error instanceof Error ? error.message : String(error),
130
+ errorType: error instanceof Error ? error.constructor.name : typeof error,
131
+ inputSchemaType: jsonSchema?.type,
132
+ inputSchemaKeys: jsonSchema && typeof jsonSchema === "object"
133
+ ? Object.keys(jsonSchema)
134
+ : [],
135
+ conversionTimeMs: conversionTime,
136
+ libraryUsed: "json-schema-to-zod-official-FAILED",
137
+ });
138
+ return z.object({}).passthrough();
139
+ }
140
+ }
@@ -2,6 +2,7 @@
2
2
  * Object Transformation Utilities
3
3
  * Centralizes repeated object transformation patterns to improve code reuse and maintainability
4
4
  */
5
+ import { logger } from "./logger.js";
5
6
  // ============================================================================
6
7
  // TOOL EXECUTION TRANSFORMATIONS
7
8
  // ============================================================================
@@ -43,22 +44,109 @@ export function transformToolExecutions(toolExecutions) {
43
44
  if (!toolName) {
44
45
  toolName = `tool_execution_${index}`;
45
46
  }
47
+ // Enhanced tool parameter extraction with structured logging
48
+ const parameterExtractionData = {
49
+ executionIndex: index,
50
+ toolNameExtracted: toolName,
51
+ primarySources: {
52
+ hasInput: !!teRecord.input,
53
+ hasParameters: !!teRecord.parameters,
54
+ hasArgs: !!teRecord.args,
55
+ inputType: typeof teRecord.input,
56
+ parametersType: typeof teRecord.parameters,
57
+ argsType: typeof teRecord.args,
58
+ },
59
+ rawRecordKeys: Object.keys(teRecord),
60
+ };
61
+ logger.debug("[TransformationUtils] Tool parameter extraction analysis", parameterExtractionData);
46
62
  // Enhanced input extraction
47
63
  let input = teRecord.input ||
48
64
  teRecord.parameters ||
49
65
  teRecord.args ||
50
66
  {};
67
+ const primaryExtractionResult = {
68
+ keysCount: Object.keys(input).length,
69
+ keys: Object.keys(input),
70
+ extractionSuccessful: Object.keys(input).length > 0,
71
+ extractionSource: teRecord.input
72
+ ? "input"
73
+ : teRecord.parameters
74
+ ? "parameters"
75
+ : teRecord.args
76
+ ? "args"
77
+ : "empty",
78
+ };
79
+ logger.debug("[TransformationUtils] Primary parameter extraction result", primaryExtractionResult);
51
80
  // Extract input from nested toolCall if available
52
81
  if (Object.keys(input).length === 0 &&
53
82
  teRecord.toolCall &&
54
83
  typeof teRecord.toolCall === "object") {
55
84
  const toolCall = teRecord.toolCall;
85
+ const nestedExtractionData = {
86
+ reason: "Primary extraction failed, checking nested toolCall",
87
+ nestedSources: {
88
+ hasInput: !!toolCall.input,
89
+ hasParameters: !!toolCall.parameters,
90
+ hasArgs: !!toolCall.args,
91
+ },
92
+ toolCallKeys: Object.keys(toolCall),
93
+ };
94
+ logger.debug("[TransformationUtils] Nested parameter extraction attempt", nestedExtractionData);
56
95
  input =
57
96
  toolCall.input ||
58
97
  toolCall.parameters ||
59
98
  toolCall.args ||
60
99
  {};
100
+ const nestedExtractionResult = {
101
+ keysCount: Object.keys(input).length,
102
+ keys: Object.keys(input),
103
+ extractionSuccessful: Object.keys(input).length > 0,
104
+ extractionSource: toolCall.input
105
+ ? "toolCall.input"
106
+ : toolCall.parameters
107
+ ? "toolCall.parameters"
108
+ : toolCall.args
109
+ ? "toolCall.args"
110
+ : "empty",
111
+ };
112
+ logger.debug("[TransformationUtils] Nested parameter extraction result", nestedExtractionResult);
113
+ }
114
+ // Target tool parameter analysis for critical tools
115
+ if (toolName &&
116
+ (toolName.includes("SuccessRateSRByTime") ||
117
+ toolName.includes("juspay-analytics"))) {
118
+ const targetToolAnalysis = {
119
+ toolName,
120
+ inputKeys: Object.keys(input),
121
+ keysCount: Object.keys(input).length,
122
+ hasStartTime: "startTime" in input,
123
+ hasEndTime: "endTime" in input,
124
+ startTimeValue: input.startTime || "MISSING",
125
+ endTimeValue: input.endTime || "MISSING",
126
+ extractionStatus: Object.keys(input).length === 0 ? "FAILED" : "SUCCESS",
127
+ };
128
+ logger.debug("[TransformationUtils] Target tool parameter analysis", targetToolAnalysis);
129
+ if (Object.keys(input).length === 0) {
130
+ logger.error("[TransformationUtils] Critical: Target tool parameter extraction failed", {
131
+ toolName,
132
+ reason: "Both primary and nested extraction returned empty parameters",
133
+ impact: "AI response did not contain expected parameter structure",
134
+ });
135
+ }
61
136
  }
137
+ // Final parameter extraction summary
138
+ const finalExtractionSummary = {
139
+ toolName,
140
+ inputKeysCount: Object.keys(input).length,
141
+ inputKeys: Object.keys(input),
142
+ hasParameters: Object.keys(input).length > 0,
143
+ duration: teRecord.duration ??
144
+ teRecord.executionTime ??
145
+ teRecord.responseTime ??
146
+ 0,
147
+ hasOutput: !!(teRecord.output || teRecord.result || teRecord.response),
148
+ };
149
+ logger.debug("[TransformationUtils] Final parameter extraction result", finalExtractionSummary);
62
150
  // Enhanced output extraction with success indication
63
151
  const output = teRecord.output ||
64
152
  teRecord.result ||
@@ -169,14 +257,61 @@ export function transformAvailableTools(availableTools) {
169
257
  }
170
258
  return availableTools.map((tool) => {
171
259
  const toolRecord = tool;
260
+ let extractedParameters = {};
261
+ const inputSchema = toolRecord.inputSchema;
262
+ const directParameters = toolRecord.parameters;
263
+ const fallbackSchema = toolRecord.schema;
264
+ if (inputSchema && typeof inputSchema === "object") {
265
+ if (inputSchema.$ref && inputSchema.definitions) {
266
+ const definitions = inputSchema.definitions;
267
+ const refValue = inputSchema.$ref;
268
+ if (typeof refValue === "string") {
269
+ const refKey = refValue.replace("#/definitions/", "");
270
+ if (definitions[refKey] && typeof definitions[refKey] === "object") {
271
+ const resolvedSchema = definitions[refKey];
272
+ extractedParameters = {
273
+ type: resolvedSchema.type || "object",
274
+ properties: resolvedSchema.properties || {},
275
+ required: resolvedSchema.required || [],
276
+ ...resolvedSchema, // Include all schema metadata
277
+ };
278
+ }
279
+ }
280
+ }
281
+ else if (inputSchema.properties) {
282
+ extractedParameters = {
283
+ type: inputSchema.type || "object",
284
+ properties: inputSchema.properties,
285
+ required: inputSchema.required || [],
286
+ ...inputSchema,
287
+ };
288
+ }
289
+ else if (inputSchema.type === "object") {
290
+ extractedParameters = inputSchema;
291
+ }
292
+ else {
293
+ extractedParameters = inputSchema;
294
+ }
295
+ }
296
+ else if (directParameters && typeof directParameters === "object") {
297
+ extractedParameters = directParameters;
298
+ }
299
+ else if (fallbackSchema && typeof fallbackSchema === "object") {
300
+ extractedParameters = fallbackSchema;
301
+ }
302
+ if (!extractedParameters || typeof extractedParameters !== "object") {
303
+ extractedParameters = {};
304
+ }
305
+ if (extractedParameters &&
306
+ !extractedParameters.type &&
307
+ extractedParameters.properties) {
308
+ extractedParameters.type = "object";
309
+ }
172
310
  return {
173
311
  name: tool.name || "",
174
312
  description: tool.description || "",
175
313
  server: tool.server || "",
176
- parameters: toolRecord.inputSchema ||
177
- toolRecord.parameters ||
178
- toolRecord.schema ||
179
- {},
314
+ parameters: extractedParameters,
180
315
  };
181
316
  });
182
317
  }
@@ -246,7 +381,10 @@ export function transformSchemaToParameterDescription(schema) {
246
381
  .map(([key, value]) => {
247
382
  const typedValue = value;
248
383
  const required = requiredParams.has(key) ? " (required)" : "";
249
- return ` - ${key}: ${typedValue.type || "unknown"}${required}`;
384
+ const description = typedValue.description
385
+ ? ` - ${typedValue.description}`
386
+ : "";
387
+ return ` - ${key}: ${typedValue.type || "unknown"}${required}${description}`;
250
388
  })
251
389
  .join("\n");
252
390
  }
@@ -89,9 +89,9 @@ export class ExternalServerManager extends EventEmitter {
89
89
  enablePerformanceMonitoring: config.enablePerformanceMonitoring ?? true,
90
90
  logLevel: config.logLevel ?? "info",
91
91
  };
92
- // Enable main tool registry integration by default
92
+ // Disable main tool registry integration by default to prevent automatic tool execution
93
93
  this.enableMainRegistryIntegration =
94
- options.enableMainRegistryIntegration ?? true;
94
+ options.enableMainRegistryIntegration ?? false;
95
95
  // Initialize tool discovery service
96
96
  this.toolDiscovery = new ToolDiscoveryService();
97
97
  // Forward tool discovery events
@@ -13,7 +13,7 @@ export class MCPRegistry {
13
13
  */
14
14
  register(plugin) {
15
15
  this.plugins.set(plugin.metadata.name, plugin);
16
- registryLogger.info(`Registered plugin: ${plugin.metadata.name}`);
16
+ registryLogger.debug(`Registered plugin: ${plugin.metadata.name}`);
17
17
  }
18
18
  /**
19
19
  * Unregister a plugin
@@ -21,7 +21,7 @@ export class MCPRegistry {
21
21
  unregister(name) {
22
22
  const removed = this.plugins.delete(name);
23
23
  if (removed) {
24
- registryLogger.info(`Unregistered plugin: ${name}`);
24
+ registryLogger.debug(`Unregistered plugin: ${name}`);
25
25
  }
26
26
  return removed;
27
27
  }
@@ -46,8 +46,24 @@ if (!shouldDisableBuiltinTools()) {
46
46
  execute: async (params, context) => {
47
47
  const startTime = Date.now();
48
48
  try {
49
- logger.debug(`[Direct Tools] Executing ${toolName} with params:`, params);
50
- // Execute the direct tool
49
+ if (Object.keys(params || {}).length === 0) {
50
+ return {
51
+ success: false,
52
+ data: null,
53
+ error: "Tool execution blocked: Empty parameters not allowed during startup",
54
+ usage: {
55
+ executionTime: Date.now() - startTime,
56
+ },
57
+ metadata: {
58
+ toolName,
59
+ serverId: "neurolink-direct",
60
+ sessionId: context.sessionId,
61
+ blocked: true,
62
+ reason: "empty_parameters_startup_prevention",
63
+ timestamp: Date.now(),
64
+ },
65
+ };
66
+ }
51
67
  if (!execute || typeof execute !== "function") {
52
68
  throw new Error(`Tool ${toolName} has no execute function`);
53
69
  }
@@ -104,7 +120,7 @@ if (!shouldDisableBuiltinTools()) {
104
120
  });
105
121
  }
106
122
  else {
107
- logger.info("[Direct Tools] Built-in tools disabled via configuration");
123
+ logger.debug("Built-in tools disabled via configuration");
108
124
  }
109
125
  /**
110
126
  * Get tool category based on tool name
@@ -126,10 +142,3 @@ function getToolCategory(toolName) {
126
142
  return "utility";
127
143
  }
128
144
  }
129
- // Log successful registration or disable status
130
- if (!shouldDisableBuiltinTools()) {
131
- logger.info(`[Direct Tools] Registered ${Object.keys(directAgentTools).length} direct tools`);
132
- }
133
- else {
134
- logger.info("[Direct Tools] 0 direct tools registered (disabled via environment variable)");
135
- }
@@ -16,13 +16,9 @@ export class MCPToolRegistry extends MCPRegistry {
16
16
  builtInServerInfos = []; // DIRECT storage for MCPServerInfo
17
17
  constructor() {
18
18
  super();
19
- // 🔧 CONDITIONAL: Only auto-register direct tools if not disabled via configuration
20
19
  if (!shouldDisableBuiltinTools()) {
21
20
  this.registerDirectTools();
22
21
  }
23
- else {
24
- registryLogger.debug("Built-in direct tools disabled via configuration");
25
- }
26
22
  }
27
23
  /**
28
24
  * Register all direct tools from directAgentTools
@@ -100,7 +96,6 @@ export class MCPToolRegistry extends MCPRegistry {
100
96
  _finalContext = serverConfigOrContext;
101
97
  }
102
98
  const serverId = serverInfo.id;
103
- registryLogger.info(`Registering MCPServerInfo directly: ${serverId}`);
104
99
  // Use MCPServerInfo.tools array directly - ZERO conversions!
105
100
  const toolsObject = {};
106
101
  for (const tool of serverInfo.tools) {
@@ -133,7 +128,6 @@ export class MCPToolRegistry extends MCPRegistry {
133
128
  this.register(plugin);
134
129
  // Use MCPServerInfo.tools array directly - ZERO conversions!
135
130
  const tools = serverInfo.tools;
136
- registryLogger.debug(`Registering ${tools.length} tools for server ${serverId}:`, tools.map((t) => t.name));
137
131
  for (const tool of tools) {
138
132
  // For custom tools, use just the tool name to avoid redundant serverId.toolName format
139
133
  // For other tools, use fully-qualified serverId.toolName to avoid collisions
@@ -166,7 +160,7 @@ export class MCPToolRegistry extends MCPRegistry {
166
160
  serverId: serverInfo.id,
167
161
  }),
168
162
  });
169
- registryLogger.debug(`Registered tool '${tool.name}' with execute function:`, typeof tool.execute);
163
+ // Tool registered successfully
170
164
  }
171
165
  // Store MCPServerInfo directly - NO recreation needed!
172
166
  if (tools.length > 0) {
@@ -211,9 +205,11 @@ export class MCPToolRegistry extends MCPRegistry {
211
205
  async executeTool(toolName, args, context) {
212
206
  const startTime = Date.now();
213
207
  try {
214
- registryLogger.info(`Executing tool: ${toolName}`);
208
+ registryLogger.info(`🔧 [TOOL_EXECUTION] Starting execution: ${toolName}`);
209
+ registryLogger.info(`🔧 [TOOL_EXECUTION] Starting execution: ${toolName}`, { args, context });
215
210
  // Try to find the tool by fully-qualified name first
216
211
  let tool = this.tools.get(toolName);
212
+ registryLogger.info(`🔍 [TOOL_LOOKUP] Direct lookup result for '${toolName}':`, !!tool);
217
213
  // If not found, search for tool by name across all entries (for backward compatibility)
218
214
  let toolId = toolName;
219
215
  if (!tool) {
@@ -10,9 +10,9 @@ import type { GenerateOptions, GenerateResult } from "./types/generateTypes.js";
10
10
  import type { StreamOptions, StreamResult } from "./types/streamTypes.js";
11
11
  import type { MCPServerInfo, MCPExecutableTool } from "./types/mcpTypes.js";
12
12
  import type { ToolInfo } from "./mcp/contracts/mcpContract.js";
13
+ import type { NeuroLinkEvents, TypedEventEmitter, ToolExecutionContext, ToolExecutionSummary } from "./types/common.js";
13
14
  import type { JsonObject } from "./types/common.js";
14
15
  import type { BatchOperationResult } from "./types/typeAliases.js";
15
- import { EventEmitter } from "events";
16
16
  import type { ConversationMemoryConfig, ChatMessage } from "./types/conversation.js";
17
17
  import type { ExternalMCPServerInstance, ExternalMCPOperationResult, ExternalMCPToolInfo } from "./types/externalMcp.js";
18
18
  export interface ProviderStatus {
@@ -49,6 +49,9 @@ export declare class NeuroLink {
49
49
  private readonly toolCacheDuration;
50
50
  private toolCircuitBreakers;
51
51
  private toolExecutionMetrics;
52
+ private currentStreamToolExecutions;
53
+ private toolExecutionHistory;
54
+ private activeToolExecutions;
52
55
  /**
53
56
  * Helper method to emit tool end event in a consistent way
54
57
  * Used by executeTool in both success and error paths
@@ -63,6 +66,11 @@ export declare class NeuroLink {
63
66
  private conversationMemoryNeedsInit;
64
67
  private conversationMemoryConfig?;
65
68
  private enableOrchestration;
69
+ /**
70
+ * Context storage for tool execution
71
+ * This context will be merged with any runtime context passed by the AI model
72
+ */
73
+ private toolExecutionContext?;
66
74
  /**
67
75
  * Creates a new NeuroLink instance for AI text generation with MCP tool integration.
68
76
  *
@@ -101,10 +109,6 @@ export declare class NeuroLink {
101
109
  conversationMemory?: Partial<ConversationMemoryConfig>;
102
110
  enableOrchestration?: boolean;
103
111
  });
104
- /**
105
- * Log constructor start with comprehensive environment analysis
106
- */
107
- private logConstructorStart;
108
112
  /**
109
113
  * Initialize provider registry with security settings
110
114
  */
@@ -130,14 +134,6 @@ export declare class NeuroLink {
130
134
  * Uses isolated async context to prevent hanging
131
135
  */
132
136
  private initializeMCP;
133
- /**
134
- * Log MCP initialization start
135
- */
136
- private logMCPInitStart;
137
- /**
138
- * Log MCP already initialized
139
- */
140
- private logMCPAlreadyInitialized;
141
137
  /**
142
138
  * Import performance manager with error handling
143
139
  */
@@ -293,10 +289,6 @@ export declare class NeuroLink {
293
289
  * Simplified approach without domain detection - relies on tool registry
294
290
  */
295
291
  private detectAndExecuteTools;
296
- /**
297
- * Enhance prompt with tool results (domain-agnostic)
298
- */
299
- private enhancePromptWithToolResults;
300
292
  /**
301
293
  * BACKWARD COMPATIBILITY: Legacy streamText method
302
294
  * Internally calls stream() and converts result format
@@ -354,14 +346,6 @@ export declare class NeuroLink {
354
346
  * @throws {Error} When conversation memory operations fail (if enabled)
355
347
  */
356
348
  stream(options: StreamOptions): Promise<StreamResult>;
357
- /**
358
- * Log stream entry point with comprehensive analysis
359
- */
360
- private logStreamEntryPoint;
361
- /**
362
- * Log performance baseline
363
- */
364
- private logPerformanceBaseline;
365
349
  /**
366
350
  * Validate stream input with comprehensive error reporting
367
351
  */
@@ -564,13 +548,58 @@ export declare class NeuroLink {
564
548
  * @see {@link NeuroLink.stream} for events related to streaming
565
549
  * @see {@link NeuroLink.executeTool} for events related to tool execution
566
550
  */
567
- getEventEmitter(): EventEmitter<[never]>;
551
+ getEventEmitter(): TypedEventEmitter<NeuroLinkEvents>;
552
+ /**
553
+ * Emit tool start event with execution tracking
554
+ * @param toolName - Name of the tool being executed
555
+ * @param input - Input parameters for the tool
556
+ * @param startTime - Timestamp when execution started
557
+ * @returns executionId for tracking this specific execution
558
+ */
559
+ emitToolStart(toolName: string, input: unknown, startTime?: number): string;
560
+ /**
561
+ * Emit tool end event with execution summary
562
+ * @param toolName - Name of the tool that finished
563
+ * @param result - Result from the tool execution
564
+ * @param error - Error message if execution failed
565
+ * @param startTime - When execution started
566
+ * @param endTime - When execution finished
567
+ * @param executionId - Optional execution ID for tracking
568
+ */
569
+ emitToolEnd(toolName: string, result?: unknown, error?: string, startTime?: number, endTime?: number, executionId?: string): void;
570
+ /**
571
+ * Get current tool execution contexts for stream metadata
572
+ */
573
+ getCurrentToolExecutions(): ToolExecutionContext[];
574
+ /**
575
+ * Get tool execution history
576
+ */
577
+ getToolExecutionHistory(): ToolExecutionSummary[];
578
+ /**
579
+ * Clear current stream tool executions (called at stream start)
580
+ */
581
+ clearCurrentStreamExecutions(): void;
568
582
  /**
569
583
  * Register a custom tool that will be available to all AI providers
570
584
  * @param name - Unique name for the tool
571
585
  * @param tool - Tool in MCPExecutableTool format (unified MCP protocol type)
572
586
  */
573
587
  registerTool(name: string, tool: MCPExecutableTool): void;
588
+ /**
589
+ * Set the context that will be passed to tools during execution
590
+ * This context will be merged with any runtime context passed by the AI model
591
+ * @param context - Context object containing session info, tokens, shop data, etc.
592
+ */
593
+ setToolContext(context: Record<string, unknown>): void;
594
+ /**
595
+ * Get the current tool execution context
596
+ * @returns Current context or undefined if not set
597
+ */
598
+ getToolContext(): Record<string, unknown> | undefined;
599
+ /**
600
+ * Clear the tool execution context
601
+ */
602
+ clearToolContext(): void;
574
603
  /**
575
604
  * Register multiple tools at once - Supports both object and array formats
576
605
  * @param tools - Object mapping tool names to MCPExecutableTool format OR Array of tools with names
@@ -621,13 +650,19 @@ export declare class NeuroLink {
621
650
  * Supports both custom tools and MCP server tools with timeout, retry, and circuit breaker patterns
622
651
  * @param toolName - Name of the tool to execute
623
652
  * @param params - Parameters to pass to the tool
624
- * @param options - Execution options
653
+ * @param options - Execution options including optional authentication context
625
654
  * @returns Tool execution result
626
655
  */
627
656
  executeTool<T = unknown>(toolName: string, params?: unknown, options?: {
628
657
  timeout?: number;
629
658
  maxRetries?: number;
630
659
  retryDelayMs?: number;
660
+ authContext?: {
661
+ userId?: string;
662
+ sessionId?: string;
663
+ user?: Record<string, unknown>;
664
+ [key: string]: unknown;
665
+ };
631
666
  }): Promise<T>;
632
667
  /**
633
668
  * Internal tool execution method (extracted for better error handling)