@juspay/neurolink 7.54.0 → 8.0.0

package/dist/lib/services/server/ai/observability/instrumentation.d.ts

@@ -55,3 +55,20 @@ export declare function getLangfuseHealthStatus(): {
         release: string;
     } | undefined;
 };
+/**
+ * Set user and session context for Langfuse spans in the current async context
+ *
+ * Merges the provided context with existing AsyncLocalStorage context. If a callback is provided,
+ * the context is scoped to that callback execution. Without a callback, the context applies to
+ * the current execution context and its children.
+ *
+ * Uses AsyncLocalStorage to properly scope context per request, avoiding race conditions
+ * in concurrent scenarios.
+ *
+ * @param context - Object containing optional userId and/or sessionId to merge with existing context
+ * @param callback - Optional callback to run within the context scope. If omitted, context applies to current execution
+ */
+export declare function setLangfuseContext(context: {
+    userId?: string | null;
+    sessionId?: string | null;
+}, callback?: () => void | Promise<void>): Promise<void>;

package/dist/lib/services/server/ai/observability/instrumentation.js

@@ -10,13 +10,38 @@ import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
 import { LangfuseSpanProcessor } from "@langfuse/otel";
 import { ATTR_SERVICE_NAME, ATTR_SERVICE_VERSION, } from "@opentelemetry/semantic-conventions";
 import { resourceFromAttributes } from "@opentelemetry/resources";
+import { AsyncLocalStorage } from "async_hooks";
 import { logger } from "../../../../utils/logger.js";
 const LOG_PREFIX = "[OpenTelemetry]";
+const contextStorage = new AsyncLocalStorage();
 let tracerProvider = null;
 let langfuseProcessor = null;
 let isInitialized = false;
 let isCredentialsValid = false;
 let currentConfig = null;
+/**
+ * Span processor that enriches spans with user and session context from AsyncLocalStorage
+ */
+class ContextEnricher {
+    onStart(span) {
+        const context = contextStorage.getStore();
+        const userId = context?.userId ?? currentConfig?.userId;
+        const sessionId = context?.sessionId ?? currentConfig?.sessionId;
+        if (userId) {
+            span.setAttribute("user.id", userId);
+        }
+        if (sessionId) {
+            span.setAttribute("session.id", sessionId);
+        }
+    }
+    onEnd() { }
+    shutdown() {
+        return Promise.resolve();
+    }
+    forceFlush() {
+        return Promise.resolve();
+    }
+}
 /**
  * Initialize OpenTelemetry with Langfuse span processor
  *
@@ -46,7 +71,6 @@ export function initializeOpenTelemetry(config) {
     try {
         currentConfig = config;
         isCredentialsValid = true;
-        // Create Langfuse span processor with configuration
         langfuseProcessor = new LangfuseSpanProcessor({
             publicKey: config.publicKey,
             secretKey: config.secretKey,
@@ -54,18 +78,15 @@ export function initializeOpenTelemetry(config) {
             environment: config.environment || "dev",
             release: config.release || "v1.0.0",
         });
-        // Create resource with service metadata (v2.x API)
         const resource = resourceFromAttributes({
             [ATTR_SERVICE_NAME]: "neurolink",
             [ATTR_SERVICE_VERSION]: config.release || "v1.0.0",
             "deployment.environment": config.environment || "dev",
         });
-        // Initialize tracer provider with span processor and resource
         tracerProvider = new NodeTracerProvider({
             resource,
-            spanProcessors: [langfuseProcessor],
+            spanProcessors: [new ContextEnricher(), langfuseProcessor],
         });
-        // Register provider globally so Vercel AI SDK can use it
         tracerProvider.register();
         isInitialized = true;
         logger.info(`${LOG_PREFIX} Initialized with Langfuse span processor`, {
@@ -168,4 +189,32 @@ export function getLangfuseHealthStatus() {
             : undefined,
     };
 }
+/**
+ * Set user and session context for Langfuse spans in the current async context
+ *
+ * Merges the provided context with existing AsyncLocalStorage context. If a callback is provided,
+ * the context is scoped to that callback execution. Without a callback, the context applies to
+ * the current execution context and its children.
+ *
+ * Uses AsyncLocalStorage to properly scope context per request, avoiding race conditions
+ * in concurrent scenarios.
+ *
+ * @param context - Object containing optional userId and/or sessionId to merge with existing context
+ * @param callback - Optional callback to run within the context scope. If omitted, context applies to current execution
+ */
+export async function setLangfuseContext(context, callback) {
+    const currentContext = contextStorage.getStore() || {};
+    const newContext = {
+        userId: context.userId !== undefined ? context.userId : currentContext.userId,
+        sessionId: context.sessionId !== undefined
+            ? context.sessionId
+            : currentContext.sessionId,
+    };
+    if (callback) {
+        return await contextStorage.run(newContext, callback);
+    }
+    else {
+        contextStorage.enterWith(newContext);
+    }
+}
 //# sourceMappingURL=instrumentation.js.map

package/dist/lib/types/observability.d.ts

@@ -24,6 +24,10 @@ export type LangfuseConfig = {
     environment?: string;
     /** Release/version identifier */
     release?: string;
+    /** Optional default user id to attach to spans */
+    userId?: string;
+    /** Optional default session id to attach to spans */
+    sessionId?: string;
 };
 /**
  * OpenTelemetry configuration

package/dist/lib/utils/fileDetector.js

@@ -3,7 +3,7 @@
  * Centralized file detection for all multimodal file types
  * Uses multi-strategy approach for reliable type identification
  */
-import { request } from "undici";
+import { request, getGlobalDispatcher, interceptors } from "undici";
 import { readFile, stat } from "fs/promises";
 import { logger } from "./logger.js";
 import { CSVProcessor } from "./csvProcessor.js";
@@ -146,10 +146,10 @@ export class FileDetector {
         const maxSize = options?.maxSize || 10 * 1024 * 1024;
         const timeout = options?.timeout || 30000;
         const response = await request(url, {
+            dispatcher: getGlobalDispatcher().compose(interceptors.redirect({ maxRedirections: 5 })),
             method: "GET",
             headersTimeout: timeout,
             bodyTimeout: timeout,
-            maxRedirections: 5,
         });
         if (response.statusCode !== 200) {
             throw new Error(`HTTP ${response.statusCode}`);
@@ -271,10 +271,10 @@ class MimeTypeStrategy {
         }
         try {
             const response = await request(input, {
+                dispatcher: getGlobalDispatcher().compose(interceptors.redirect({ maxRedirections: 5 })),
                 method: "HEAD",
                 headersTimeout: 5000,
                 bodyTimeout: 5000,
-                maxRedirections: 5,
             });
             const contentType = response.headers["content-type"] || "";
             const type = this.mimeToFileType(contentType);

package/dist/lib/utils/messageBuilder.js

@@ -8,7 +8,7 @@ import { ProviderImageAdapter, MultimodalLogger, } from "../adapters/providerIma
 import { logger } from "./logger.js";
 import { FileDetector } from "./fileDetector.js";
 import { PDFProcessor } from "./pdfProcessor.js";
-import { request } from "undici";
+import { request, getGlobalDispatcher, interceptors } from "undici";
 import { readFileSync, existsSync } from "fs";
 /**
  * Type guard for validating message roles
@@ -562,10 +562,10 @@ function isInternetUrl(input) {
 async function downloadImageFromUrl(url) {
     try {
         const response = await request(url, {
+            dispatcher: getGlobalDispatcher().compose(interceptors.redirect({ maxRedirections: 5 })),
             method: "GET",
             headersTimeout: 10000, // 10 second timeout for headers
-            bodyTimeout: 30000, // 30 second timeout for body
-            maxRedirections: 5,
+            bodyTimeout: 30000, // 30 second timeout for body,
         });
         if (response.statusCode !== 200) {
             throw new Error(`HTTP ${response.statusCode}: Failed to download image from ${url}`);

package/dist/mcp/mcpClientFactory.js

@@ -20,9 +20,6 @@ export class MCPClientFactory {
         version: "1.0.0",
     };
     static DEFAULT_CAPABILITIES = {
-        tools: {},
-        resources: {},
-        prompts: {},
         sampling: {},
         roots: {
             listChanged: false,

package/dist/neurolink.d.ts

@@ -50,6 +50,10 @@ export declare class NeuroLink {
     private hitlManager?;
     private mem0Instance?;
     private mem0Config?;
+    /**
+     * Extract and set Langfuse context from options with proper async scoping
+     */
+    private setLangfuseContextFromOptions;
     /**
      * Simple sync config setup for mem0
      */