@brizz/sdk 0.1.25 → 0.1.27

package/dist/index.js

@@ -164,31 +164,9 @@ var init_logger = __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 init_semantic_conventions = __esm({
-  "src/internal/semantic-conventions.ts"() {
-    "use strict";
-    BRIZZ = "brizz";
-    PROPERTIES = "properties";
-    SESSION_ID = "session.id";
-    PROPERTIES_CONTEXT_KEY = createContextKey(PROPERTIES);
-    SESSION_OBJECT_CONTEXT_KEY = createContextKey("brizz.session.object");
-    SESSION_INPUT = "brizz.session.input";
-    SESSION_OUTPUT = "brizz.session.output";
-    SESSION_INPUT_CONTEXT = "brizz.session.input.context";
-    SESSION_OUTPUT_CONTEXT = "brizz.session.output.context";
-    SESSION_SPAN_NAME = "brizz.start_session";
-    SESSION_TITLE_SPAN_NAME = "brizz.session_title";
-    SESSION_TITLE_GENERATION = "session.title_generation";
-    SESSION_TITLE = "brizz.session.title";
-  }
-});
-
 // src/internal/instrumentation/mcp/semantic-conventions.ts
 var MCP_TOOL_NAME, MCP_TOOL_ARGUMENTS, MCP_TOOL_RESULT, MCP_COMPONENT_TYPE, MCP_COMPONENT_TOOL, MCP_COMPONENT_TOOL_SCHEMA, MCP_TOOL_SCHEMA_PARAMETERS, MCP_TOOL_SCHEMA_OUTPUT, MCP_TOOL_DESCRIPTION, SPAN_NAME_TOOL_REGISTER, MCP_METHOD_NAME, MCP_REQUEST_ID, MCP_SESSION_ID, MCP_PROTOCOL_VERSION, MCP_RESOURCE_URI, RPC_SYSTEM, RPC_SYSTEM_MCP, RPC_RESPONSE_STATUS_CODE, GEN_AI_TOOL_NAME, GEN_AI_PROMPT_NAME, GEN_AI_OPERATION_NAME, GEN_AI_OPERATION_EXECUTE_TOOL, NETWORK_TRANSPORT, ERROR_TYPE, ERROR_TYPE_TOOL, JSONRPC_REQUEST_ID, SPAN_NAME_TOOLS_CALL, MAX_ATTRIBUTE_LENGTH, TRUNCATION_SUFFIX, METHOD_TOOLS_CALL, METHOD_TOOLS_LIST, METHOD_RESOURCES_READ, METHOD_PROMPTS_GET, METHOD_INITIALIZE;
-var init_semantic_conventions2 = __esm({
+var init_semantic_conventions = __esm({
   "src/internal/instrumentation/mcp/semantic-conventions.ts"() {
     "use strict";
     MCP_TOOL_NAME = "mcp.tool.name";
@@ -283,7 +261,6 @@ function stampSchemaAttributes(span, toolName, transportSessionId, tool) {
   span.setAttribute(RPC_SYSTEM, RPC_SYSTEM_MCP);
   span.setAttribute(MCP_COMPONENT_TYPE, MCP_COMPONENT_TOOL_SCHEMA);
   span.setAttribute(MCP_SESSION_ID, transportSessionId);
-  span.setAttribute(`${BRIZZ}.${SESSION_ID}`, transportSessionId);
   span.setAttribute(MCP_TOOL_NAME, toolName);
   span.setAttribute(MCP_TOOL_SCHEMA_PARAMETERS, parameters);
   span.setAttribute(MCP_TOOL_SCHEMA_OUTPUT, outputSchema);
@@ -307,11 +284,33 @@ var init_schemas = __esm({
     "use strict";
     init_logger();
     init_semantic_conventions();
-    init_semantic_conventions2();
     _MAX_SCHEMA_ATTR_BYTES = 4e3;
   }
 });
 
+// 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, INTERRUPT_TOOLS;
+var init_semantic_conventions2 = __esm({
+  "src/internal/semantic-conventions.ts"() {
+    "use strict";
+    BRIZZ = "brizz";
+    PROPERTIES = "properties";
+    SESSION_ID = "session.id";
+    PROPERTIES_CONTEXT_KEY = createContextKey(PROPERTIES);
+    SESSION_OBJECT_CONTEXT_KEY = createContextKey("brizz.session.object");
+    SESSION_INPUT = "brizz.session.input";
+    SESSION_OUTPUT = "brizz.session.output";
+    SESSION_INPUT_CONTEXT = "brizz.session.input.context";
+    SESSION_OUTPUT_CONTEXT = "brizz.session.output.context";
+    SESSION_SPAN_NAME = "brizz.start_session";
+    SESSION_TITLE_SPAN_NAME = "brizz.session_title";
+    SESSION_TITLE_GENERATION = "session.title_generation";
+    SESSION_TITLE = "brizz.session.title";
+    INTERRUPT_TOOLS = "brizz.internal.interrupt";
+  }
+});
+
 // src/internal/instrumentation/mcp/session.ts
 import { context } from "@opentelemetry/api";
 function stampAndPropagateSession(span, sessionId, baseContext = context.active()) {
@@ -343,7 +342,7 @@ var init_session = __esm({
   "src/internal/instrumentation/mcp/session.ts"() {
     "use strict";
     init_logger();
-    init_semantic_conventions();
+    init_semantic_conventions2();
   }
 });
 
@@ -540,7 +539,7 @@ function normalizeTransport(ctorName) {
 var init_attributes = __esm({
   "src/internal/instrumentation/mcp/patches/attributes.ts"() {
     "use strict";
-    init_semantic_conventions2();
+    init_semantic_conventions();
   }
 });
 
@@ -765,7 +764,7 @@ var init_protocol = __esm({
     "use strict";
     init_logger();
     init_schemas();
-    init_semantic_conventions2();
+    init_semantic_conventions();
     init_session();
     init_attributes();
     PATCHED_FLAG = /* @__PURE__ */ Symbol("brizz.mcp.protocol-patched");
@@ -774,7 +773,7 @@ var init_protocol = __esm({
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.25";
+  return "0.1.27";
 }
 var init_version = __esm({
   "src/internal/version.ts"() {
@@ -956,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";
@@ -1036,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;
@@ -1044,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`
@@ -1358,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";
@@ -1782,7 +1899,7 @@ function maskAttributes(attributes, rules, outputOriginalValue = false) {
 }
 
 // src/internal/log/processors/log-processor.ts
-init_semantic_conventions();
+init_semantic_conventions2();
 var DEFAULT_LOG_MASKING_RULES = [
   {
     mode: "partial",
@@ -2227,7 +2344,7 @@ import {
   BatchSpanProcessor,
   SimpleSpanProcessor
 } from "@opentelemetry/sdk-trace-base";
-init_semantic_conventions();
+init_semantic_conventions2();
 function applyContextAttributes(span) {
   const sessionProperties = context4.active().getValue(PROPERTIES_CONTEXT_KEY);
   if (sessionProperties) {
@@ -2415,10 +2532,10 @@ function getSpanProcessor() {
 }
 
 // src/internal/trace/session.ts
-init_semantic_conventions();
-import { context as context5, trace as trace3, SpanStatusCode as SpanStatusCode3 } from "@opentelemetry/api";
+init_semantic_conventions2();
+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;
   }
@@ -2538,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) {
@@ -2586,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);
@@ -2706,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
@@ -2901,4 +3018,3 @@ export {
   withProperties,
   withSessionId
 };
-//# sourceMappingURL=index.js.map

package/dist/preload.cjs

@@ -350,15 +350,7 @@ var import_instrumentation = require("@opentelemetry/instrumentation");
 var import_api6 = require("@opentelemetry/api");
 
 // src/internal/instrumentation/mcp/schemas.ts
-var import_api3 = require("@opentelemetry/api");
-
-// src/internal/semantic-conventions.ts
 var import_api2 = require("@opentelemetry/api");
-var BRIZZ = "brizz";
-var PROPERTIES = "properties";
-var SESSION_ID = "session.id";
-var PROPERTIES_CONTEXT_KEY = (0, import_api2.createContextKey)(PROPERTIES);
-var SESSION_OBJECT_CONTEXT_KEY = (0, import_api2.createContextKey)("brizz.session.object");
 
 // src/internal/instrumentation/mcp/semantic-conventions.ts
 var MCP_TOOL_NAME = "mcp.tool.name";
@@ -428,11 +420,11 @@ function emitSchemaSpansFromListResponse(result, transportSessionId, tracer) {
       continue;
     }
     const span = tracer.startSpan(`${SPAN_NAME_TOOL_REGISTER} ${name}`, {
-      kind: import_api3.SpanKind.INTERNAL
+      kind: import_api2.SpanKind.INTERNAL
     });
     try {
       stampSchemaAttributes(span, name, transportSessionId, tool);
-      span.setStatus({ code: import_api3.SpanStatusCode.OK });
+      span.setStatus({ code: import_api2.SpanStatusCode.OK });
     } finally {
       span.end();
     }
@@ -451,7 +443,6 @@ function stampSchemaAttributes(span, toolName, transportSessionId, tool) {
   span.setAttribute(RPC_SYSTEM, RPC_SYSTEM_MCP);
   span.setAttribute(MCP_COMPONENT_TYPE, MCP_COMPONENT_TOOL_SCHEMA);
   span.setAttribute(MCP_SESSION_ID, transportSessionId);
-  span.setAttribute(`${BRIZZ}.${SESSION_ID}`, transportSessionId);
   span.setAttribute(MCP_TOOL_NAME, toolName);
   span.setAttribute(MCP_TOOL_SCHEMA_PARAMETERS, parameters);
   span.setAttribute(MCP_TOOL_SCHEMA_OUTPUT, outputSchema);
@@ -472,6 +463,17 @@ function extractTools(result) {
 
 // src/internal/instrumentation/mcp/session.ts
 var import_api4 = require("@opentelemetry/api");
+
+// src/internal/semantic-conventions.ts
+var import_api3 = require("@opentelemetry/api");
+var BRIZZ = "brizz";
+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()) {
   if (!sessionId) {
     return { context: baseContext, sessionId: null };
@@ -907,7 +909,7 @@ function safeEnd(span) {
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.25";
+  return "0.1.27";
 }
 
 // src/internal/instrumentation/mcp/version.ts
@@ -1165,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");
@@ -1172,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
@@ -1601,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);
@@ -1621,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);
@@ -2008,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);
@@ -2194,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 {
@@ -2279,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
@@ -2543,4 +2583,3 @@ try {
 } catch (error) {
   logger.warn("Failed to auto-initialize SDK:", { error });
 }
-//# sourceMappingURL=preload.cjs.map

package/dist/preload.js

@@ -335,14 +335,6 @@ import { context as context2, propagation, SpanKind as SpanKind2, trace } from "
 // src/internal/instrumentation/mcp/schemas.ts
 import { SpanKind, SpanStatusCode } from "@opentelemetry/api";
 
-// src/internal/semantic-conventions.ts
-import { createContextKey } from "@opentelemetry/api";
-var BRIZZ = "brizz";
-var PROPERTIES = "properties";
-var SESSION_ID = "session.id";
-var PROPERTIES_CONTEXT_KEY = createContextKey(PROPERTIES);
-var SESSION_OBJECT_CONTEXT_KEY = createContextKey("brizz.session.object");
-
 // src/internal/instrumentation/mcp/semantic-conventions.ts
 var MCP_TOOL_NAME = "mcp.tool.name";
 var MCP_TOOL_ARGUMENTS = "mcp.tool.arguments";
@@ -434,7 +426,6 @@ function stampSchemaAttributes(span, toolName, transportSessionId, tool) {
   span.setAttribute(RPC_SYSTEM, RPC_SYSTEM_MCP);
   span.setAttribute(MCP_COMPONENT_TYPE, MCP_COMPONENT_TOOL_SCHEMA);
   span.setAttribute(MCP_SESSION_ID, transportSessionId);
-  span.setAttribute(`${BRIZZ}.${SESSION_ID}`, transportSessionId);
   span.setAttribute(MCP_TOOL_NAME, toolName);
   span.setAttribute(MCP_TOOL_SCHEMA_PARAMETERS, parameters);
   span.setAttribute(MCP_TOOL_SCHEMA_OUTPUT, outputSchema);
@@ -455,6 +446,17 @@ function extractTools(result) {
 
 // src/internal/instrumentation/mcp/session.ts
 import { context } from "@opentelemetry/api";
+
+// src/internal/semantic-conventions.ts
+import { createContextKey } from "@opentelemetry/api";
+var BRIZZ = "brizz";
+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()) {
   if (!sessionId) {
     return { context: baseContext, sessionId: null };
@@ -890,7 +892,7 @@ function safeEnd(span) {
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.25";
+  return "0.1.27";
 }
 
 // src/internal/instrumentation/mcp/version.ts
@@ -1148,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";
@@ -2182,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 {
@@ -2267,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
@@ -2530,4 +2570,3 @@ try {
 } catch (error) {
   logger.warn("Failed to auto-initialize SDK:", { error });
 }
-//# sourceMappingURL=preload.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@brizz/sdk",
-  "version": "0.1.25",
+  "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
     }