@brizz/sdk 0.1.17 → 0.1.19

package/dist/index.d.cts

@@ -1,11 +1,13 @@
+import { AttributeValue, Span } from '@opentelemetry/api';
+export { AttributeValue } from '@opentelemetry/api';
 import { LogRecordExporter } from '@opentelemetry/sdk-logs';
 import { MetricReader } from '@opentelemetry/sdk-metrics';
-import { SpanExporter } from '@opentelemetry/sdk-trace-base';
+import { SpanExporter, ReadableSpan } from '@opentelemetry/sdk-trace-base';
+export { ReadableSpan } from '@opentelemetry/sdk-trace-base';
 import { LogBody, SeverityNumber } from '@opentelemetry/api-logs';
 export { SeverityNumber } from '@opentelemetry/api-logs';
 import { SpanProcessor } from '@opentelemetry/sdk-trace-node';
 import { OTLPMetricExporter } from '@opentelemetry/exporter-metrics-otlp-http';
-import { Span, AttributeValue } from '@opentelemetry/api';
 
 interface IInstrumentModules {
     openAI?: unknown;
@@ -83,6 +85,7 @@ declare function maskAttributes(attributes: Record<string, unknown>, rules: read
 
 interface IBrizzInitializeOptions {
     appName?: string;
+    appVersion?: string;
     baseUrl?: string;
     apiKey?: string;
     headers?: Record<string, string>;
@@ -96,6 +99,8 @@ interface IBrizzInitializeOptions {
     customLogExporter?: LogRecordExporter;
     customMetricReader?: MetricReader;
     disableSpanExporter?: boolean;
+    resourceAttributes?: Record<string, AttributeValue>;
+    beforeSendSpan?: (span: ReadableSpan) => boolean | Promise<boolean>;
 }
 declare class _Brizz {
     private static instance;
@@ -130,9 +135,11 @@ declare class Session {
     private readonly span;
     private inputs;
     private outputs;
+    private inputContexts;
+    private outputContexts;
     constructor(sessionId: string, span: Span);
-    setInput(text: string | null): void;
-    setOutput(text: string | null): void;
+    setInput(text: string | null, context?: Record<string, AttributeValue>): void;
+    setOutput(text: string | null, context?: Record<string, AttributeValue>): void;
     setTitle(title: string): void;
     updateProperties(properties: Record<string, AttributeValue>): void;
 }

package/dist/index.d.ts

@@ -1,11 +1,13 @@
+import { AttributeValue, Span } from '@opentelemetry/api';
+export { AttributeValue } from '@opentelemetry/api';
 import { LogRecordExporter } from '@opentelemetry/sdk-logs';
 import { MetricReader } from '@opentelemetry/sdk-metrics';
-import { SpanExporter } from '@opentelemetry/sdk-trace-base';
+import { SpanExporter, ReadableSpan } from '@opentelemetry/sdk-trace-base';
+export { ReadableSpan } from '@opentelemetry/sdk-trace-base';
 import { LogBody, SeverityNumber } from '@opentelemetry/api-logs';
 export { SeverityNumber } from '@opentelemetry/api-logs';
 import { SpanProcessor } from '@opentelemetry/sdk-trace-node';
 import { OTLPMetricExporter } from '@opentelemetry/exporter-metrics-otlp-http';
-import { Span, AttributeValue } from '@opentelemetry/api';
 
 interface IInstrumentModules {
     openAI?: unknown;
@@ -83,6 +85,7 @@ declare function maskAttributes(attributes: Record<string, unknown>, rules: read
 
 interface IBrizzInitializeOptions {
     appName?: string;
+    appVersion?: string;
     baseUrl?: string;
     apiKey?: string;
     headers?: Record<string, string>;
@@ -96,6 +99,8 @@ interface IBrizzInitializeOptions {
     customLogExporter?: LogRecordExporter;
     customMetricReader?: MetricReader;
     disableSpanExporter?: boolean;
+    resourceAttributes?: Record<string, AttributeValue>;
+    beforeSendSpan?: (span: ReadableSpan) => boolean | Promise<boolean>;
 }
 declare class _Brizz {
     private static instance;
@@ -130,9 +135,11 @@ declare class Session {
     private readonly span;
     private inputs;
     private outputs;
+    private inputContexts;
+    private outputContexts;
     constructor(sessionId: string, span: Span);
-    setInput(text: string | null): void;
-    setOutput(text: string | null): void;
+    setInput(text: string | null, context?: Record<string, AttributeValue>): void;
+    setOutput(text: string | null, context?: Record<string, AttributeValue>): void;
     setTitle(title: string): void;
     updateProperties(properties: Record<string, AttributeValue>): void;
 }

package/dist/index.js

@@ -170,7 +170,14 @@ var exceptionLogger = (error) => {
 };
 function loadNodeAutoInstrumentations() {
   try {
-    const nodeInstrumentations = getNodeAutoInstrumentations();
+    const nodeInstrumentations = 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 }
+    });
     registerInstrumentations({ instrumentations: nodeInstrumentations });
     return nodeInstrumentations;
   } catch (error) {
@@ -274,6 +281,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,
@@ -446,7 +454,7 @@ import {
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.17";
+  return "0.1.19";
 }
 
 // src/internal/log/processors/log-processor.ts
@@ -1086,6 +1094,8 @@ var PROPERTIES_CONTEXT_KEY = createContextKey(PROPERTIES);
 var SESSION_OBJECT_CONTEXT_KEY = 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";
@@ -1255,6 +1265,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
@@ -1262,6 +1273,9 @@ var LoggingModule = class _LoggingModule {
     if (config.environment) {
       resourceAttributes["deployment.environment"] = config.environment;
     }
+    if (config.appVersion) {
+      resourceAttributes["service.version"] = config.appVersion;
+    }
     const resource = resourceFromAttributes(resourceAttributes);
     logger.debug("Creating logger provider with resource");
     this.loggerProvider = new LoggerProvider({
@@ -1451,13 +1465,16 @@ function getMetricsReader() {
 import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-proto";
 
 // src/internal/trace/exporters/span-exporter.ts
+import { ExportResultCode } from "@opentelemetry/core";
 import { resourceFromAttributes as resourceFromAttributes2 } from "@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
@@ -1465,15 +1482,47 @@ var BrizzSpanExporter = class {
     if (config.environment) {
       resourceAttrs["deployment.environment"] = config.environment;
     }
+    if (config.appVersion) {
+      resourceAttrs["service.version"] = config.appVersion;
+    }
     this._brizzResource = resourceFromAttributes2(resourceAttrs);
+    this._beforeSendSpan = config.beforeSendSpan;
   }
   export(spans, resultCallback) {
+    if (spans.length === 0) {
+      resultCallback({ code: 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: 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();
@@ -1489,6 +1538,14 @@ import {
   BatchSpanProcessor,
   SimpleSpanProcessor
 } from "@opentelemetry/sdk-trace-base";
+function applyContextAttributes(span) {
+  const sessionProperties = context2.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",
@@ -1523,12 +1580,7 @@ var BrizzSimpleSpanProcessor = class extends SimpleSpanProcessor {
     if (maskingConfig) {
       maskSpan(span, maskingConfig);
     }
-    const associationProperties = context2.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);
   }
 };
@@ -1543,12 +1595,7 @@ var BrizzBatchSpanProcessor = class extends BatchSpanProcessor {
     if (maskingConfig) {
       maskSpan(span, maskingConfig);
     }
-    const associationProperties = context2.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);
   }
 };
@@ -1722,31 +1769,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.
@@ -1943,6 +2004,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
@@ -1950,6 +2012,9 @@ var _Brizz = class __Brizz {
     if (resolvedConfig.environment) {
       resourceAttributes["deployment.environment"] = resolvedConfig.environment;
     }
+    if (resolvedConfig.appVersion) {
+      resourceAttributes["service.version"] = resolvedConfig.appVersion;
+    }
     this._sdk = new NodeSDK({
       spanProcessors: resolvedConfig.disableSpanExporter ? [] : [getSpanProcessor()],
       metricReader: getMetricsReader(),