@brizz/sdk 0.1.25 → 0.1.27

package/dist/index.cjs

@@ -180,31 +180,9 @@ var init_logger = __esm({
   }
 });
 
-// src/internal/semantic-conventions.ts
-var import_api2, 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";
-    import_api2 = require("@opentelemetry/api");
-    BRIZZ = "brizz";
-    PROPERTIES = "properties";
-    SESSION_ID = "session.id";
-    PROPERTIES_CONTEXT_KEY = (0, import_api2.createContextKey)(PROPERTIES);
-    SESSION_OBJECT_CONTEXT_KEY = (0, import_api2.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";
@@ -275,11 +253,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();
     }
@@ -298,7 +276,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);
@@ -316,18 +293,40 @@ function extractTools(result) {
     (t) => t !== null && typeof t === "object"
   );
 }
-var import_api3, _MAX_SCHEMA_ATTR_BYTES;
+var import_api2, _MAX_SCHEMA_ATTR_BYTES;
 var init_schemas = __esm({
   "src/internal/instrumentation/mcp/schemas.ts"() {
     "use strict";
-    import_api3 = require("@opentelemetry/api");
+    import_api2 = require("@opentelemetry/api");
     init_logger();
     init_semantic_conventions();
-    init_semantic_conventions2();
     _MAX_SCHEMA_ATTR_BYTES = 4e3;
   }
 });
 
+// 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, INTERRUPT_TOOLS;
+var init_semantic_conventions2 = __esm({
+  "src/internal/semantic-conventions.ts"() {
+    "use strict";
+    import_api3 = require("@opentelemetry/api");
+    BRIZZ = "brizz";
+    PROPERTIES = "properties";
+    SESSION_ID = "session.id";
+    PROPERTIES_CONTEXT_KEY = (0, import_api3.createContextKey)(PROPERTIES);
+    SESSION_OBJECT_CONTEXT_KEY = (0, import_api3.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
 function stampAndPropagateSession(span, sessionId, baseContext = import_api4.context.active()) {
   if (!sessionId) {
@@ -360,7 +359,7 @@ var init_session = __esm({
     "use strict";
     import_api4 = require("@opentelemetry/api");
     init_logger();
-    init_semantic_conventions();
+    init_semantic_conventions2();
   }
 });
 
@@ -558,7 +557,7 @@ var init_attributes = __esm({
   "src/internal/instrumentation/mcp/patches/attributes.ts"() {
     "use strict";
     import_api5 = require("@opentelemetry/api");
-    init_semantic_conventions2();
+    init_semantic_conventions();
   }
 });
 
@@ -783,7 +782,7 @@ var init_protocol = __esm({
     import_api6 = require("@opentelemetry/api");
     init_logger();
     init_schemas();
-    init_semantic_conventions2();
+    init_semantic_conventions();
     init_session();
     init_attributes();
     PATCHED_FLAG = /* @__PURE__ */ Symbol("brizz.mcp.protocol-patched");
@@ -792,7 +791,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"() {
@@ -974,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, {
@@ -1086,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;
@@ -1094,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`
@@ -1408,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");
@@ -1417,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();
 
@@ -1830,7 +1947,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",
@@ -1849,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);
@@ -1869,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);
@@ -2269,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_conventions();
+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);
@@ -2460,10 +2577,10 @@ function getSpanProcessor() {
 }
 
 // src/internal/trace/session.ts
-var import_api10 = require("@opentelemetry/api");
-init_semantic_conventions();
+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;
   }
@@ -2475,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) {
@@ -2578,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) {
@@ -2602,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") {
@@ -2612,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;
             });
@@ -2621,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;
         }
@@ -2631,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);
@@ -2650,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;
           });
@@ -2659,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;
       }
@@ -2751,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
@@ -2947,4 +3064,3 @@ var init_exports = {};
   withProperties,
   withSessionId
 });
-//# sourceMappingURL=index.cjs.map