@brizz/sdk 0.1.26 → 0.1.27

package/dist/index.cjs

@@ -305,7 +305,7 @@ var init_schemas = __esm({
 });
 
 // src/internal/semantic-conventions.ts
-var import_api3, BRIZZ, PROPERTIES, SESSION_ID, PROPERTIES_CONTEXT_KEY, SESSION_OBJECT_CONTEXT_KEY, SESSION_INPUT, SESSION_OUTPUT, SESSION_INPUT_CONTEXT, SESSION_OUTPUT_CONTEXT, SESSION_SPAN_NAME, SESSION_TITLE_SPAN_NAME, SESSION_TITLE_GENERATION, SESSION_TITLE;
+var import_api3, BRIZZ, PROPERTIES, SESSION_ID, PROPERTIES_CONTEXT_KEY, SESSION_OBJECT_CONTEXT_KEY, SESSION_INPUT, SESSION_OUTPUT, SESSION_INPUT_CONTEXT, SESSION_OUTPUT_CONTEXT, SESSION_SPAN_NAME, SESSION_TITLE_SPAN_NAME, SESSION_TITLE_GENERATION, SESSION_TITLE, INTERRUPT_TOOLS;
 var init_semantic_conventions2 = __esm({
   "src/internal/semantic-conventions.ts"() {
     "use strict";
@@ -323,6 +323,7 @@ var init_semantic_conventions2 = __esm({
     SESSION_TITLE_SPAN_NAME = "brizz.session_title";
     SESSION_TITLE_GENERATION = "session.title_generation";
     SESSION_TITLE = "brizz.session.title";
+    INTERRUPT_TOOLS = "brizz.internal.interrupt";
   }
 });
 
@@ -790,7 +791,7 @@ var init_protocol = __esm({
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.26";
+  return "0.1.27";
 }
 var init_version = __esm({
   "src/internal/version.ts"() {
@@ -972,6 +973,100 @@ var init_mcp = __esm({
   }
 });
 
+// src/internal/instrumentation/vercel-ai/interrupt.ts
+var interrupt_exports = {};
+__export(interrupt_exports, {
+  InterruptPropagator: () => InterruptPropagator,
+  _resetInterruptState: () => _resetInterruptState,
+  createInterruptIntegration: () => createInterruptIntegration
+});
+function isInnerLLMSpan(span) {
+  if (INNER_OPERATION_IDS.has(span.name)) {
+    return true;
+  }
+  const opId = span.attributes["ai.operationId"];
+  return typeof opId === "string" && INNER_OPERATION_IDS.has(opId);
+}
+function setPending(parentSpanId, value) {
+  pendingByParentSpanId.set(parentSpanId, value);
+  while (pendingByParentSpanId.size > MAX_PENDING_ENTRIES) {
+    const oldest = pendingByParentSpanId.keys().next().value;
+    if (oldest === void 0) {
+      break;
+    }
+    pendingByParentSpanId.delete(oldest);
+  }
+}
+function extractInterruptTools(tools) {
+  if (!tools || typeof tools !== "object") {
+    return [];
+  }
+  return Object.entries(tools).filter(
+    ([, t]) => !!t && typeof t === "object" && t.needsApproval
+  ).map(([name]) => name);
+}
+function createInterruptIntegration() {
+  const onStepStart = (event) => {
+    const names = extractInterruptTools(event.tools);
+    if (names.length === 0) {
+      return;
+    }
+    const value = JSON.stringify(names);
+    const span = import_api8.trace.getActiveSpan();
+    if (!span) {
+      return;
+    }
+    span.setAttribute(INTERRUPT_TOOLS, value);
+    setPending(span.spanContext().spanId, value);
+  };
+  return {
+    onStepStart
+  };
+}
+function _resetInterruptState() {
+  if (process.env["NODE_ENV"] !== "test") {
+    return;
+  }
+  pendingByParentSpanId.clear();
+}
+var import_api8, INNER_OPERATION_IDS, MAX_PENDING_ENTRIES, pendingByParentSpanId, InterruptPropagator;
+var init_interrupt = __esm({
+  "src/internal/instrumentation/vercel-ai/interrupt.ts"() {
+    "use strict";
+    import_api8 = require("@opentelemetry/api");
+    init_semantic_conventions2();
+    INNER_OPERATION_IDS = /* @__PURE__ */ new Set([
+      "ai.generateText.doGenerate",
+      "ai.streamText.doStream"
+    ]);
+    MAX_PENDING_ENTRIES = 1024;
+    pendingByParentSpanId = /* @__PURE__ */ new Map();
+    InterruptPropagator = class {
+      onStart(span, _parentContext) {
+        if (!isInnerLLMSpan(span)) {
+          return;
+        }
+        const parentSpanId = span.parentSpanContext?.spanId;
+        if (!parentSpanId) {
+          return;
+        }
+        const value = pendingByParentSpanId.get(parentSpanId);
+        if (!value) {
+          return;
+        }
+        span.setAttribute(INTERRUPT_TOOLS, value);
+        pendingByParentSpanId.delete(parentSpanId);
+      }
+      onEnd() {
+      }
+      async shutdown() {
+      }
+      async forceFlush() {
+      }
+    };
+  }
+});
+
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
@@ -1084,6 +1179,26 @@ function loadMCPInstrumentation() {
     }
   })();
 }
+function loadVercelAIInterruptIntegration() {
+  void (async () => {
+    try {
+      const ai = await import("ai");
+      if (typeof ai.registerTelemetryIntegration !== "function") {
+        logger.debug(
+          "AI SDK present but lacks registerTelemetryIntegration (needs ai@>=6); skipping interrupt bridge"
+        );
+        return;
+      }
+      const { createInterruptIntegration: createInterruptIntegration2 } = await Promise.resolve().then(() => (init_interrupt(), interrupt_exports));
+      ai.registerTelemetryIntegration(createInterruptIntegration2());
+      logger.debug("Auto-loaded Vercel AI interrupt bridge");
+    } catch (error) {
+      logger.debug(
+        `Vercel AI interrupt bridge not loaded (install ai@>=6 if you want HITL attribution): ${String(error)}`
+      );
+    }
+  })();
+}
 function autoInitializeInstrumentations() {
   if (autoInstrumentationsLoaded) {
     return;
@@ -1092,6 +1207,7 @@ function autoInitializeInstrumentations() {
     const nodeInstrumentations = loadNodeAutoInstrumentations();
     const genAIInstrumentations = loadGenAIInstrumentations();
     loadMCPInstrumentation();
+    loadVercelAIInterruptIntegration();
     autoInstrumentationsLoaded = true;
     logger.info(
       `Auto-initialization complete: ${nodeInstrumentations.length} node + ${genAIInstrumentations.length} GenAI instrumentations`
@@ -1406,6 +1522,9 @@ var InstrumentationRegistry = class _InstrumentationRegistry {
   }
 };
 
+// src/internal/sdk.ts
+init_interrupt();
+
 // src/internal/log/logging.ts
 var import_api_logs = require("@opentelemetry/api-logs");
 var import_exporter_logs_otlp_http = require("@opentelemetry/exporter-logs-otlp-http");
@@ -1415,7 +1534,7 @@ init_logger();
 init_version();
 
 // src/internal/log/processors/log-processor.ts
-var import_api8 = require("@opentelemetry/api");
+var import_api9 = require("@opentelemetry/api");
 var import_sdk_logs = require("@opentelemetry/sdk-logs");
 init_logger();
 
@@ -1847,7 +1966,7 @@ var BrizzSimpleLogRecordProcessor = class extends import_sdk_logs.SimpleLogRecor
     if (maskingConfig) {
       maskLog(logRecord, maskingConfig);
     }
-    const associationProperties = import_api8.context.active().getValue(PROPERTIES_CONTEXT_KEY);
+    const associationProperties = import_api9.context.active().getValue(PROPERTIES_CONTEXT_KEY);
     if (associationProperties) {
       for (const [key, value] of Object.entries(associationProperties)) {
         logRecord.setAttribute(`${BRIZZ}.${key}`, value);
@@ -1867,7 +1986,7 @@ var BrizzBatchLogRecordProcessor = class extends import_sdk_logs.BatchLogRecordP
     if (maskingConfig) {
       maskLog(logRecord, maskingConfig);
     }
-    const associationProperties = import_api8.context.active().getValue(PROPERTIES_CONTEXT_KEY);
+    const associationProperties = import_api9.context.active().getValue(PROPERTIES_CONTEXT_KEY);
     if (associationProperties) {
       for (const [key, value] of Object.entries(associationProperties)) {
         logRecord.setAttribute(`${BRIZZ}.${key}`, value);
@@ -2267,12 +2386,12 @@ var BrizzSpanExporter = class {
 };
 
 // src/internal/trace/processors/span-processor.ts
-var import_api9 = require("@opentelemetry/api");
+var import_api10 = require("@opentelemetry/api");
 var import_sdk_trace_base = require("@opentelemetry/sdk-trace-base");
 init_logger();
 init_semantic_conventions2();
 function applyContextAttributes(span) {
-  const sessionProperties = import_api9.context.active().getValue(PROPERTIES_CONTEXT_KEY);
+  const sessionProperties = import_api10.context.active().getValue(PROPERTIES_CONTEXT_KEY);
   if (sessionProperties) {
     for (const [key, value] of Object.entries(sessionProperties)) {
       span.setAttribute(`${BRIZZ}.${key}`, value);
@@ -2458,10 +2577,10 @@ function getSpanProcessor() {
 }
 
 // src/internal/trace/session.ts
-var import_api10 = require("@opentelemetry/api");
+var import_api11 = require("@opentelemetry/api");
 init_semantic_conventions2();
 function setCurrentSpanCustomProperties(properties) {
-  const current = import_api10.trace.getActiveSpan();
+  const current = import_api11.trace.getActiveSpan();
   if (!current || !current.isRecording()) {
     return;
   }
@@ -2473,11 +2592,11 @@ function setCurrentSpanCustomProperties(properties) {
   }
 }
 function callWithProperties(properties, fn, thisArg, ...args) {
-  const base = import_api10.context.active();
+  const base = import_api11.context.active();
   const prev = base.getValue(PROPERTIES_CONTEXT_KEY);
   const merged = prev ? { ...prev, ...properties } : properties;
   const next = base.setValue(PROPERTIES_CONTEXT_KEY, merged);
-  return import_api10.context.with(next, fn, thisArg, ...args);
+  return import_api11.context.with(next, fn, thisArg, ...args);
 }
 function withProperties(properties, fn, thisArg) {
   return function wrapped(...args) {
@@ -2576,12 +2695,12 @@ var SessionTitle = class {
   }
 };
 function getActiveSession() {
-  return import_api10.context.active().getValue(SESSION_OBJECT_CONTEXT_KEY);
+  return import_api11.context.active().getValue(SESSION_OBJECT_CONTEXT_KEY);
 }
 function startSession(sessionId, callback, extraProperties, options) {
   const isTitle = options?.mode === "title";
   const spanName = isTitle ? SESSION_TITLE_SPAN_NAME : SESSION_SPAN_NAME;
-  const tracer = import_api10.trace.getTracer("@brizz/sdk");
+  const tracer = import_api11.trace.getTracer("@brizz/sdk");
   return tracer.startActiveSpan(spanName, (span) => {
     span.setAttribute(`${BRIZZ}.${SESSION_ID}`, sessionId);
     if (extraProperties) {
@@ -2600,8 +2719,8 @@ function startSession(sessionId, callback, extraProperties, options) {
       }
     }
     return callWithProperties(contextProperties, () => {
-      const sessionCtx = import_api10.context.active().setValue(SESSION_OBJECT_CONTEXT_KEY, session);
-      return import_api10.context.with(sessionCtx, () => {
+      const sessionCtx = import_api11.context.active().setValue(SESSION_OBJECT_CONTEXT_KEY, session);
+      return import_api11.context.with(sessionCtx, () => {
         try {
           const result = callback(session);
           if (result && typeof result.then === "function") {
@@ -2610,7 +2729,7 @@ function startSession(sessionId, callback, extraProperties, options) {
               return value;
             }).catch((error) => {
               span.recordException(error);
-              span.setStatus({ code: import_api10.SpanStatusCode.ERROR });
+              span.setStatus({ code: import_api11.SpanStatusCode.ERROR });
               span.end();
               throw error;
             });
@@ -2619,7 +2738,7 @@ function startSession(sessionId, callback, extraProperties, options) {
           return result;
         } catch (error) {
           span.recordException(error);
-          span.setStatus({ code: import_api10.SpanStatusCode.ERROR });
+          span.setStatus({ code: import_api11.SpanStatusCode.ERROR });
           span.end();
           throw error;
         }
@@ -2629,7 +2748,7 @@ function startSession(sessionId, callback, extraProperties, options) {
 }
 function startSessionTitle(callback, options) {
   const resolvedSessionId = options?.sessionId ?? getActiveSession()?.sessionId;
-  const tracer = import_api10.trace.getTracer("@brizz/sdk");
+  const tracer = import_api11.trace.getTracer("@brizz/sdk");
   return tracer.startActiveSpan(SESSION_TITLE_SPAN_NAME, (span) => {
     if (resolvedSessionId) {
       span.setAttribute(`${BRIZZ}.${SESSION_ID}`, resolvedSessionId);
@@ -2648,7 +2767,7 @@ function startSessionTitle(callback, options) {
             return value;
           }).catch((error) => {
             span.recordException(error);
-            span.setStatus({ code: import_api10.SpanStatusCode.ERROR });
+            span.setStatus({ code: import_api11.SpanStatusCode.ERROR });
             span.end();
             throw error;
           });
@@ -2657,7 +2776,7 @@ function startSessionTitle(callback, options) {
         return result;
       } catch (error) {
         span.recordException(error);
-        span.setStatus({ code: import_api10.SpanStatusCode.ERROR });
+        span.setStatus({ code: import_api11.SpanStatusCode.ERROR });
         span.end();
         throw error;
       }
@@ -2749,7 +2868,7 @@ var _Brizz = class __Brizz {
       resourceAttributes["service.version"] = resolvedConfig.appVersion;
     }
     this._sdk = new import_sdk_node.NodeSDK({
-      spanProcessors: resolvedConfig.disableSpanExporter ? [] : [getSpanProcessor()],
+      spanProcessors: resolvedConfig.disableSpanExporter ? [] : [new InterruptPropagator(), getSpanProcessor()],
       metricReader: getMetricsReader(),
       resource: (0, import_resources3.resourceFromAttributes)(resourceAttributes),
       instrumentations: manualInstrumentations

package/dist/index.js

@@ -290,7 +290,7 @@ var init_schemas = __esm({
 
 // src/internal/semantic-conventions.ts
 import { createContextKey } from "@opentelemetry/api";
-var BRIZZ, PROPERTIES, SESSION_ID, PROPERTIES_CONTEXT_KEY, SESSION_OBJECT_CONTEXT_KEY, SESSION_INPUT, SESSION_OUTPUT, SESSION_INPUT_CONTEXT, SESSION_OUTPUT_CONTEXT, SESSION_SPAN_NAME, SESSION_TITLE_SPAN_NAME, SESSION_TITLE_GENERATION, SESSION_TITLE;
+var BRIZZ, PROPERTIES, SESSION_ID, PROPERTIES_CONTEXT_KEY, SESSION_OBJECT_CONTEXT_KEY, SESSION_INPUT, SESSION_OUTPUT, SESSION_INPUT_CONTEXT, SESSION_OUTPUT_CONTEXT, SESSION_SPAN_NAME, SESSION_TITLE_SPAN_NAME, SESSION_TITLE_GENERATION, SESSION_TITLE, INTERRUPT_TOOLS;
 var init_semantic_conventions2 = __esm({
   "src/internal/semantic-conventions.ts"() {
     "use strict";
@@ -307,6 +307,7 @@ var init_semantic_conventions2 = __esm({
     SESSION_TITLE_SPAN_NAME = "brizz.session_title";
     SESSION_TITLE_GENERATION = "session.title_generation";
     SESSION_TITLE = "brizz.session.title";
+    INTERRUPT_TOOLS = "brizz.internal.interrupt";
   }
 });
 
@@ -772,7 +773,7 @@ var init_protocol = __esm({
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.26";
+  return "0.1.27";
 }
 var init_version = __esm({
   "src/internal/version.ts"() {
@@ -954,6 +955,100 @@ var init_mcp = __esm({
   }
 });
 
+// src/internal/instrumentation/vercel-ai/interrupt.ts
+var interrupt_exports = {};
+__export(interrupt_exports, {
+  InterruptPropagator: () => InterruptPropagator,
+  _resetInterruptState: () => _resetInterruptState,
+  createInterruptIntegration: () => createInterruptIntegration
+});
+import { trace as trace3 } from "@opentelemetry/api";
+function isInnerLLMSpan(span) {
+  if (INNER_OPERATION_IDS.has(span.name)) {
+    return true;
+  }
+  const opId = span.attributes["ai.operationId"];
+  return typeof opId === "string" && INNER_OPERATION_IDS.has(opId);
+}
+function setPending(parentSpanId, value) {
+  pendingByParentSpanId.set(parentSpanId, value);
+  while (pendingByParentSpanId.size > MAX_PENDING_ENTRIES) {
+    const oldest = pendingByParentSpanId.keys().next().value;
+    if (oldest === void 0) {
+      break;
+    }
+    pendingByParentSpanId.delete(oldest);
+  }
+}
+function extractInterruptTools(tools) {
+  if (!tools || typeof tools !== "object") {
+    return [];
+  }
+  return Object.entries(tools).filter(
+    ([, t]) => !!t && typeof t === "object" && t.needsApproval
+  ).map(([name]) => name);
+}
+function createInterruptIntegration() {
+  const onStepStart = (event) => {
+    const names = extractInterruptTools(event.tools);
+    if (names.length === 0) {
+      return;
+    }
+    const value = JSON.stringify(names);
+    const span = trace3.getActiveSpan();
+    if (!span) {
+      return;
+    }
+    span.setAttribute(INTERRUPT_TOOLS, value);
+    setPending(span.spanContext().spanId, value);
+  };
+  return {
+    onStepStart
+  };
+}
+function _resetInterruptState() {
+  if (process.env["NODE_ENV"] !== "test") {
+    return;
+  }
+  pendingByParentSpanId.clear();
+}
+var INNER_OPERATION_IDS, MAX_PENDING_ENTRIES, pendingByParentSpanId, InterruptPropagator;
+var init_interrupt = __esm({
+  "src/internal/instrumentation/vercel-ai/interrupt.ts"() {
+    "use strict";
+    init_semantic_conventions2();
+    INNER_OPERATION_IDS = /* @__PURE__ */ new Set([
+      "ai.generateText.doGenerate",
+      "ai.streamText.doStream"
+    ]);
+    MAX_PENDING_ENTRIES = 1024;
+    pendingByParentSpanId = /* @__PURE__ */ new Map();
+    InterruptPropagator = class {
+      onStart(span, _parentContext) {
+        if (!isInnerLLMSpan(span)) {
+          return;
+        }
+        const parentSpanId = span.parentSpanContext?.spanId;
+        if (!parentSpanId) {
+          return;
+        }
+        const value = pendingByParentSpanId.get(parentSpanId);
+        if (!value) {
+          return;
+        }
+        span.setAttribute(INTERRUPT_TOOLS, value);
+        pendingByParentSpanId.delete(parentSpanId);
+      }
+      onEnd() {
+      }
+      async shutdown() {
+      }
+      async forceFlush() {
+      }
+    };
+  }
+});
+
 // src/internal/instrumentation/auto-init.ts
 init_logger();
 import { getNodeAutoInstrumentations } from "@opentelemetry/auto-instrumentations-node";
@@ -1034,6 +1129,26 @@ function loadMCPInstrumentation() {
     }
   })();
 }
+function loadVercelAIInterruptIntegration() {
+  void (async () => {
+    try {
+      const ai = await import("ai");
+      if (typeof ai.registerTelemetryIntegration !== "function") {
+        logger.debug(
+          "AI SDK present but lacks registerTelemetryIntegration (needs ai@>=6); skipping interrupt bridge"
+        );
+        return;
+      }
+      const { createInterruptIntegration: createInterruptIntegration2 } = await Promise.resolve().then(() => (init_interrupt(), interrupt_exports));
+      ai.registerTelemetryIntegration(createInterruptIntegration2());
+      logger.debug("Auto-loaded Vercel AI interrupt bridge");
+    } catch (error) {
+      logger.debug(
+        `Vercel AI interrupt bridge not loaded (install ai@>=6 if you want HITL attribution): ${String(error)}`
+      );
+    }
+  })();
+}
 function autoInitializeInstrumentations() {
   if (autoInstrumentationsLoaded) {
     return;
@@ -1042,6 +1157,7 @@ function autoInitializeInstrumentations() {
     const nodeInstrumentations = loadNodeAutoInstrumentations();
     const genAIInstrumentations = loadGenAIInstrumentations();
     loadMCPInstrumentation();
+    loadVercelAIInterruptIntegration();
     autoInstrumentationsLoaded = true;
     logger.info(
       `Auto-initialization complete: ${nodeInstrumentations.length} node + ${genAIInstrumentations.length} GenAI instrumentations`
@@ -1356,6 +1472,9 @@ var InstrumentationRegistry = class _InstrumentationRegistry {
   }
 };
 
+// src/internal/sdk.ts
+init_interrupt();
+
 // src/internal/log/logging.ts
 import { SeverityNumber } from "@opentelemetry/api-logs";
 import { OTLPLogExporter } from "@opentelemetry/exporter-logs-otlp-http";
@@ -2414,9 +2533,9 @@ function getSpanProcessor() {
 
 // src/internal/trace/session.ts
 init_semantic_conventions2();
-import { context as context5, trace as trace3, SpanStatusCode as SpanStatusCode3 } from "@opentelemetry/api";
+import { context as context5, trace as trace4, SpanStatusCode as SpanStatusCode3 } from "@opentelemetry/api";
 function setCurrentSpanCustomProperties(properties) {
-  const current = trace3.getActiveSpan();
+  const current = trace4.getActiveSpan();
   if (!current || !current.isRecording()) {
     return;
   }
@@ -2536,7 +2655,7 @@ function getActiveSession() {
 function startSession(sessionId, callback, extraProperties, options) {
   const isTitle = options?.mode === "title";
   const spanName = isTitle ? SESSION_TITLE_SPAN_NAME : SESSION_SPAN_NAME;
-  const tracer = trace3.getTracer("@brizz/sdk");
+  const tracer = trace4.getTracer("@brizz/sdk");
   return tracer.startActiveSpan(spanName, (span) => {
     span.setAttribute(`${BRIZZ}.${SESSION_ID}`, sessionId);
     if (extraProperties) {
@@ -2584,7 +2703,7 @@ function startSession(sessionId, callback, extraProperties, options) {
 }
 function startSessionTitle(callback, options) {
   const resolvedSessionId = options?.sessionId ?? getActiveSession()?.sessionId;
-  const tracer = trace3.getTracer("@brizz/sdk");
+  const tracer = trace4.getTracer("@brizz/sdk");
   return tracer.startActiveSpan(SESSION_TITLE_SPAN_NAME, (span) => {
     if (resolvedSessionId) {
       span.setAttribute(`${BRIZZ}.${SESSION_ID}`, resolvedSessionId);
@@ -2704,7 +2823,7 @@ var _Brizz = class __Brizz {
       resourceAttributes["service.version"] = resolvedConfig.appVersion;
     }
     this._sdk = new NodeSDK({
-      spanProcessors: resolvedConfig.disableSpanExporter ? [] : [getSpanProcessor()],
+      spanProcessors: resolvedConfig.disableSpanExporter ? [] : [new InterruptPropagator(), getSpanProcessor()],
       metricReader: getMetricsReader(),
       resource: resourceFromAttributes3(resourceAttributes),
       instrumentations: manualInstrumentations

package/dist/preload.cjs

@@ -471,6 +471,7 @@ var PROPERTIES = "properties";
 var SESSION_ID = "session.id";
 var PROPERTIES_CONTEXT_KEY = (0, import_api3.createContextKey)(PROPERTIES);
 var SESSION_OBJECT_CONTEXT_KEY = (0, import_api3.createContextKey)("brizz.session.object");
+var INTERRUPT_TOOLS = "brizz.internal.interrupt";
 
 // src/internal/instrumentation/mcp/session.ts
 function stampAndPropagateSession(span, sessionId, baseContext = import_api4.context.active()) {
@@ -908,7 +909,7 @@ function safeEnd(span) {
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.26";
+  return "0.1.27";
 }
 
 // src/internal/instrumentation/mcp/version.ts
@@ -1166,6 +1167,44 @@ var InstrumentationRegistry = class _InstrumentationRegistry {
   }
 };
 
+// src/internal/instrumentation/vercel-ai/interrupt.ts
+var import_api8 = require("@opentelemetry/api");
+var INNER_OPERATION_IDS = /* @__PURE__ */ new Set([
+  "ai.generateText.doGenerate",
+  "ai.streamText.doStream"
+]);
+function isInnerLLMSpan(span) {
+  if (INNER_OPERATION_IDS.has(span.name)) {
+    return true;
+  }
+  const opId = span.attributes["ai.operationId"];
+  return typeof opId === "string" && INNER_OPERATION_IDS.has(opId);
+}
+var pendingByParentSpanId = /* @__PURE__ */ new Map();
+var InterruptPropagator = class {
+  onStart(span, _parentContext) {
+    if (!isInnerLLMSpan(span)) {
+      return;
+    }
+    const parentSpanId = span.parentSpanContext?.spanId;
+    if (!parentSpanId) {
+      return;
+    }
+    const value = pendingByParentSpanId.get(parentSpanId);
+    if (!value) {
+      return;
+    }
+    span.setAttribute(INTERRUPT_TOOLS, value);
+    pendingByParentSpanId.delete(parentSpanId);
+  }
+  onEnd() {
+  }
+  async shutdown() {
+  }
+  async forceFlush() {
+  }
+};
+
 // src/internal/log/logging.ts
 var import_api_logs = require("@opentelemetry/api-logs");
 var import_exporter_logs_otlp_http = require("@opentelemetry/exporter-logs-otlp-http");
@@ -1173,7 +1212,7 @@ var import_resources = require("@opentelemetry/resources");
 var import_sdk_logs2 = require("@opentelemetry/sdk-logs");
 
 // src/internal/log/processors/log-processor.ts
-var import_api8 = require("@opentelemetry/api");
+var import_api9 = require("@opentelemetry/api");
 var import_sdk_logs = require("@opentelemetry/sdk-logs");
 
 // src/internal/masking/patterns.ts
@@ -1602,7 +1641,7 @@ var BrizzSimpleLogRecordProcessor = class extends import_sdk_logs.SimpleLogRecor
     if (maskingConfig) {
       maskLog(logRecord, maskingConfig);
     }
-    const associationProperties = import_api8.context.active().getValue(PROPERTIES_CONTEXT_KEY);
+    const associationProperties = import_api9.context.active().getValue(PROPERTIES_CONTEXT_KEY);
     if (associationProperties) {
       for (const [key, value] of Object.entries(associationProperties)) {
         logRecord.setAttribute(`${BRIZZ}.${key}`, value);
@@ -1622,7 +1661,7 @@ var BrizzBatchLogRecordProcessor = class extends import_sdk_logs.BatchLogRecordP
     if (maskingConfig) {
       maskLog(logRecord, maskingConfig);
     }
-    const associationProperties = import_api8.context.active().getValue(PROPERTIES_CONTEXT_KEY);
+    const associationProperties = import_api9.context.active().getValue(PROPERTIES_CONTEXT_KEY);
     if (associationProperties) {
       for (const [key, value] of Object.entries(associationProperties)) {
         logRecord.setAttribute(`${BRIZZ}.${key}`, value);
@@ -2009,10 +2048,10 @@ var BrizzSpanExporter = class {
 };
 
 // src/internal/trace/processors/span-processor.ts
-var import_api9 = require("@opentelemetry/api");
+var import_api10 = require("@opentelemetry/api");
 var import_sdk_trace_base = require("@opentelemetry/sdk-trace-base");
 function applyContextAttributes(span) {
-  const sessionProperties = import_api9.context.active().getValue(PROPERTIES_CONTEXT_KEY);
+  const sessionProperties = import_api10.context.active().getValue(PROPERTIES_CONTEXT_KEY);
   if (sessionProperties) {
     for (const [key, value] of Object.entries(sessionProperties)) {
       span.setAttribute(`${BRIZZ}.${key}`, value);
@@ -2195,7 +2234,7 @@ function getSpanProcessor() {
 }
 
 // src/internal/trace/session.ts
-var import_api10 = require("@opentelemetry/api");
+var import_api11 = require("@opentelemetry/api");
 
 // src/internal/sdk.ts
 var _Brizz = class __Brizz {
@@ -2280,7 +2319,7 @@ var _Brizz = class __Brizz {
       resourceAttributes["service.version"] = resolvedConfig.appVersion;
     }
     this._sdk = new import_sdk_node.NodeSDK({
-      spanProcessors: resolvedConfig.disableSpanExporter ? [] : [getSpanProcessor()],
+      spanProcessors: resolvedConfig.disableSpanExporter ? [] : [new InterruptPropagator(), getSpanProcessor()],
       metricReader: getMetricsReader(),
       resource: (0, import_resources3.resourceFromAttributes)(resourceAttributes),
       instrumentations: manualInstrumentations

package/dist/preload.js

@@ -454,6 +454,7 @@ var PROPERTIES = "properties";
 var SESSION_ID = "session.id";
 var PROPERTIES_CONTEXT_KEY = createContextKey(PROPERTIES);
 var SESSION_OBJECT_CONTEXT_KEY = createContextKey("brizz.session.object");
+var INTERRUPT_TOOLS = "brizz.internal.interrupt";
 
 // src/internal/instrumentation/mcp/session.ts
 function stampAndPropagateSession(span, sessionId, baseContext = context.active()) {
@@ -891,7 +892,7 @@ function safeEnd(span) {
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.26";
+  return "0.1.27";
 }
 
 // src/internal/instrumentation/mcp/version.ts
@@ -1149,6 +1150,44 @@ var InstrumentationRegistry = class _InstrumentationRegistry {
   }
 };
 
+// src/internal/instrumentation/vercel-ai/interrupt.ts
+import { trace as trace3 } from "@opentelemetry/api";
+var INNER_OPERATION_IDS = /* @__PURE__ */ new Set([
+  "ai.generateText.doGenerate",
+  "ai.streamText.doStream"
+]);
+function isInnerLLMSpan(span) {
+  if (INNER_OPERATION_IDS.has(span.name)) {
+    return true;
+  }
+  const opId = span.attributes["ai.operationId"];
+  return typeof opId === "string" && INNER_OPERATION_IDS.has(opId);
+}
+var pendingByParentSpanId = /* @__PURE__ */ new Map();
+var InterruptPropagator = class {
+  onStart(span, _parentContext) {
+    if (!isInnerLLMSpan(span)) {
+      return;
+    }
+    const parentSpanId = span.parentSpanContext?.spanId;
+    if (!parentSpanId) {
+      return;
+    }
+    const value = pendingByParentSpanId.get(parentSpanId);
+    if (!value) {
+      return;
+    }
+    span.setAttribute(INTERRUPT_TOOLS, value);
+    pendingByParentSpanId.delete(parentSpanId);
+  }
+  onEnd() {
+  }
+  async shutdown() {
+  }
+  async forceFlush() {
+  }
+};
+
 // src/internal/log/logging.ts
 import { SeverityNumber } from "@opentelemetry/api-logs";
 import { OTLPLogExporter } from "@opentelemetry/exporter-logs-otlp-http";
@@ -2183,7 +2222,7 @@ function getSpanProcessor() {
 }
 
 // src/internal/trace/session.ts
-import { context as context5, trace as trace3, SpanStatusCode as SpanStatusCode3 } from "@opentelemetry/api";
+import { context as context5, trace as trace4, SpanStatusCode as SpanStatusCode3 } from "@opentelemetry/api";
 
 // src/internal/sdk.ts
 var _Brizz = class __Brizz {
@@ -2268,7 +2307,7 @@ var _Brizz = class __Brizz {
       resourceAttributes["service.version"] = resolvedConfig.appVersion;
     }
     this._sdk = new NodeSDK({
-      spanProcessors: resolvedConfig.disableSpanExporter ? [] : [getSpanProcessor()],
+      spanProcessors: resolvedConfig.disableSpanExporter ? [] : [new InterruptPropagator(), getSpanProcessor()],
       metricReader: getMetricsReader(),
       resource: resourceFromAttributes3(resourceAttributes),
       instrumentations: manualInstrumentations

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@brizz/sdk",
-  "version": "0.1.26",
+  "version": "0.1.27",
   "type": "module",
   "description": "OpenTelemetry-based observability SDK for AI applications",
   "keywords": [
@@ -114,12 +114,16 @@
   },
   "peerDependencies": {
     "@langchain/core": "^0.3.0 || ^1.0.0",
+    "ai": "*",
     "typescript": ">=5.0.0"
   },
   "peerDependenciesMeta": {
     "@langchain/core": {
       "optional": true
     },
+    "ai": {
+      "optional": true
+    },
     "typescript": {
       "optional": true
     }