@brizz/sdk 0.1.17 → 0.1.19

package/README.md

@@ -287,10 +287,23 @@ const result = await startSession(
 
 **Session Methods:**
 - `session.updateProperties({ key: value })` - Add custom properties to the session span
-- `session.setInput(text)` - (Optional) Manually track input text
-- `session.setOutput(text)` - (Optional) Manually track output text
+- `session.setInput(text, context?)` - (Optional) Manually track input text; optional context bag attaches per-turn metadata rendered in the dashboard's Context panel
+- `session.setOutput(text, context?)` - (Optional) Manually track output text; optional context bag attaches per-turn metadata rendered in the dashboard's Context panel
 - `session.setTitle(text)` - Set a session title (typically used with `mode: 'title'`)
 
+**Per-turn context example:**
+
+```typescript
+await startSession('session-123', async (session) => {
+  session.setInput('Why is my bill high?', { selected_invoice: 'INV-9182' });
+  const response = await openai.chat.completions.create({ ... });
+  session.setOutput(response.choices[0].message.content, {
+    message_id: 'msg-42',
+    sources: ['doc-abc'],
+  });
+});
+```
+
 **When to use manual input/output tracking:**
 
 In most cases, Brizz automatically captures inputs and outputs from your LLM calls. Use `setInput`/`setOutput` for special scenarios:

package/dist/index.cjs

@@ -223,7 +223,14 @@ var exceptionLogger = (error) => {
 };
 function loadNodeAutoInstrumentations() {
   try {
-    const nodeInstrumentations = (0, import_auto_instrumentations_node.getNodeAutoInstrumentations)();
+    const nodeInstrumentations = (0, import_auto_instrumentations_node.getNodeAutoInstrumentations)({
+      // Disabled because @traceloop/instrumentation-openai wraps the same OpenAI
+      // prototype with richer span attributes (gen_ai.input.messages /
+      // gen_ai.output.messages). Leaving both enabled lets the official one win
+      // and route content to log events instead of span attributes, leaving the
+      // Brizz backend parser with no payload to read.
+      "@opentelemetry/instrumentation-openai": { enabled: false }
+    });
     (0, import_instrumentation.registerInstrumentations)({ instrumentations: nodeInstrumentations });
     return nodeInstrumentations;
   } catch (error) {
@@ -327,6 +334,7 @@ function resolveConfig(options) {
   const resolvedConfig = {
     ...options,
     appName: process.env["BRIZZ_APP_NAME"] || options.appName || "unknown-app",
+    appVersion: process.env["BRIZZ_APP_VERSION"] || options.appVersion,
     baseUrl: process.env["BRIZZ_BASE_URL"] || options.baseUrl || "https://telemetry.brizz.dev",
     headers: { ...options.headers },
     apiKey: process.env["BRIZZ_API_KEY"] || options.apiKey,
@@ -497,7 +505,7 @@ var import_sdk_logs2 = require("@opentelemetry/sdk-logs");
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.17";
+  return "0.1.19";
 }
 
 // src/internal/log/processors/log-processor.ts
@@ -1137,6 +1145,8 @@ var PROPERTIES_CONTEXT_KEY = (0, import_api2.createContextKey)(PROPERTIES);
 var SESSION_OBJECT_CONTEXT_KEY = (0, import_api2.createContextKey)("brizz.session.object");
 var SESSION_INPUT = "brizz.session.input";
 var SESSION_OUTPUT = "brizz.session.output";
+var SESSION_INPUT_CONTEXT = "brizz.session.input.context";
+var SESSION_OUTPUT_CONTEXT = "brizz.session.output.context";
 var SESSION_SPAN_NAME = "brizz.start_session";
 var SESSION_TITLE_SPAN_NAME = "brizz.session_title";
 var SESSION_TITLE_GENERATION = "session.title_generation";
@@ -1306,6 +1316,7 @@ var LoggingModule = class _LoggingModule {
       serviceName: config.appName
     });
     const resourceAttributes = {
+      ...config.resourceAttributes,
       "service.name": config.appName,
       [BRIZZ_SDK_VERSION]: getSDKVersion(),
       [BRIZZ_SDK_LANGUAGE]: SDK_LANGUAGE
@@ -1313,6 +1324,9 @@ var LoggingModule = class _LoggingModule {
     if (config.environment) {
       resourceAttributes["deployment.environment"] = config.environment;
     }
+    if (config.appVersion) {
+      resourceAttributes["service.version"] = config.appVersion;
+    }
     const resource = (0, import_resources.resourceFromAttributes)(resourceAttributes);
     logger.debug("Creating logger provider with resource");
     this.loggerProvider = new import_sdk_logs2.LoggerProvider({
@@ -1502,13 +1516,16 @@ function getMetricsReader() {
 var import_exporter_trace_otlp_proto = require("@opentelemetry/exporter-trace-otlp-proto");
 
 // src/internal/trace/exporters/span-exporter.ts
+var import_core = require("@opentelemetry/core");
 var import_resources2 = require("@opentelemetry/resources");
 var BrizzSpanExporter = class {
   _delegate;
   _brizzResource;
+  _beforeSendSpan;
   constructor(delegate, config) {
     this._delegate = delegate;
     const resourceAttrs = {
+      ...config.resourceAttributes,
       "service.name": config.appName,
       [BRIZZ_SDK_VERSION]: getSDKVersion(),
       [BRIZZ_SDK_LANGUAGE]: SDK_LANGUAGE
@@ -1516,15 +1533,47 @@ var BrizzSpanExporter = class {
     if (config.environment) {
       resourceAttrs["deployment.environment"] = config.environment;
     }
+    if (config.appVersion) {
+      resourceAttrs["service.version"] = config.appVersion;
+    }
     this._brizzResource = (0, import_resources2.resourceFromAttributes)(resourceAttrs);
+    this._beforeSendSpan = config.beforeSendSpan;
   }
   export(spans, resultCallback) {
+    if (spans.length === 0) {
+      resultCallback({ code: import_core.ExportResultCode.SUCCESS });
+      return;
+    }
     const patchedSpans = spans.map((span) => ({
       ...span,
       resource: span.resource.merge(this._brizzResource),
       spanContext: span.spanContext.bind(span)
     }));
-    this._delegate.export(patchedSpans, resultCallback);
+    const filter = this._beforeSendSpan;
+    if (!filter) {
+      this._delegate.export(patchedSpans, resultCallback);
+      return;
+    }
+    const verdicts = patchedSpans.map((span) => {
+      try {
+        return Promise.resolve(filter(span));
+      } catch (error) {
+        logger.warn("beforeSendSpan threw; span will be kept", { error });
+        return Promise.resolve(true);
+      }
+    });
+    Promise.all(verdicts).then((keep) => {
+      const filtered = patchedSpans.filter((_, i) => keep[i] !== false);
+      if (filtered.length === 0) {
+        resultCallback({ code: import_core.ExportResultCode.SUCCESS });
+        return;
+      }
+      this._delegate.export(filtered, resultCallback);
+      return;
+    }).catch((error) => {
+      logger.warn("beforeSendSpan rejected; all spans will be kept", { error });
+      this._delegate.export(patchedSpans, resultCallback);
+    });
   }
   async shutdown() {
     return this._delegate.shutdown();
@@ -1537,6 +1586,14 @@ var BrizzSpanExporter = class {
 // src/internal/trace/processors/span-processor.ts
 var import_api4 = require("@opentelemetry/api");
 var import_sdk_trace_base = require("@opentelemetry/sdk-trace-base");
+function applyContextAttributes(span) {
+  const sessionProperties = import_api4.context.active().getValue(PROPERTIES_CONTEXT_KEY);
+  if (sessionProperties) {
+    for (const [key, value] of Object.entries(sessionProperties)) {
+      span.setAttribute(`${BRIZZ}.${key}`, value);
+    }
+  }
+}
 var DEFAULT_MASKING_RULES = [
   {
     mode: "partial",
@@ -1571,12 +1628,7 @@ var BrizzSimpleSpanProcessor = class extends import_sdk_trace_base.SimpleSpanPro
     if (maskingConfig) {
       maskSpan(span, maskingConfig);
     }
-    const associationProperties = import_api4.context.active().getValue(PROPERTIES_CONTEXT_KEY);
-    if (associationProperties) {
-      for (const [key, value] of Object.entries(associationProperties)) {
-        span.setAttribute(`${BRIZZ}.${key}`, value);
-      }
-    }
+    applyContextAttributes(span);
     super.onStart(span, parentContext);
   }
 };
@@ -1591,12 +1643,7 @@ var BrizzBatchSpanProcessor = class extends import_sdk_trace_base.BatchSpanProce
     if (maskingConfig) {
       maskSpan(span, maskingConfig);
     }
-    const associationProperties = import_api4.context.active().getValue(PROPERTIES_CONTEXT_KEY);
-    if (associationProperties) {
-      for (const [key, value] of Object.entries(associationProperties)) {
-        span.setAttribute(`${BRIZZ}.${key}`, value);
-      }
-    }
+    applyContextAttributes(span);
     super.onStart(span, parentContext);
   }
 };
@@ -1770,31 +1817,45 @@ var Session = class {
   span;
   inputs = [];
   outputs = [];
+  inputContexts = [];
+  outputContexts = [];
   constructor(sessionId, span) {
     this.sessionId = sessionId;
     this.span = span;
   }
   /**
-   * (Optional) Append text to session input tracking.
-   * Use when you need to track specific input data that differs from what's sent to the LLM.
-   * Multiple calls accumulate in an array.
+   * Append a user turn to session input tracking.
    *
-   * @param text - Text to append to session input, or null to append null
+   * Each call appends one element to `brizz.session.input` (the text) AND one
+   * element to `brizz.session.input.context` (the per-turn context bag). The
+   * two arrays stay index-aligned — the ingestion pipeline zips them together
+   * so the i-th context bag lands on the i-th user_display conversation item's
+   * `span_attributes` map.
+   *
+   * @param text - Text to append (null becomes a null/"hide marker")
+   * @param context - Optional per-turn context bag to attach to this turn
    */
-  setInput(text) {
+  setInput(text, context4) {
     this.inputs.push(text);
+    this.inputContexts.push(context4 ?? {});
     this.span.setAttribute(SESSION_INPUT, JSON.stringify(this.inputs));
+    this.span.setAttribute(SESSION_INPUT_CONTEXT, JSON.stringify(this.inputContexts));
   }
   /**
-   * (Optional) Append text to session output tracking.
-   * Use when you need to track specific output data that differs from what's received from the LLM.
-   * Multiple calls accumulate in an array.
+   * Append an assistant turn to session output tracking.
+   *
+   * Symmetric to {@link setInput} — appends one text element and one
+   * context-bag element to the index-aligned `brizz.session.output` /
+   * `brizz.session.output.context` arrays.
    *
-   * @param text - Text to append to session output, or null to append null
+   * @param text - Text to append (null becomes a null/"hide marker")
+   * @param context - Optional per-turn context bag to attach to this turn
    */
-  setOutput(text) {
+  setOutput(text, context4) {
     this.outputs.push(text);
+    this.outputContexts.push(context4 ?? {});
     this.span.setAttribute(SESSION_OUTPUT, JSON.stringify(this.outputs));
+    this.span.setAttribute(SESSION_OUTPUT_CONTEXT, JSON.stringify(this.outputContexts));
   }
   /**
    * Set the session title on the span.
@@ -1991,6 +2052,7 @@ var _Brizz = class __Brizz {
     const registry = InstrumentationRegistry.getInstance();
     const manualInstrumentations = registry.getManualInstrumentations();
     const resourceAttributes = {
+      ...resolvedConfig.resourceAttributes,
       "service.name": resolvedConfig.appName,
       [BRIZZ_SDK_VERSION]: getSDKVersion(),
       [BRIZZ_SDK_LANGUAGE]: SDK_LANGUAGE
@@ -1998,6 +2060,9 @@ var _Brizz = class __Brizz {
     if (resolvedConfig.environment) {
       resourceAttributes["deployment.environment"] = resolvedConfig.environment;
     }
+    if (resolvedConfig.appVersion) {
+      resourceAttributes["service.version"] = resolvedConfig.appVersion;
+    }
     this._sdk = new import_sdk_node.NodeSDK({
       spanProcessors: resolvedConfig.disableSpanExporter ? [] : [getSpanProcessor()],
       metricReader: getMetricsReader(),