@brizz/sdk 0.1.18 → 0.1.19

package/dist/index.d.cts

@@ -135,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

@@ -135,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

@@ -454,7 +454,7 @@ import {
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.18";
+  return "0.1.19";
 }
 
 // src/internal/log/processors/log-processor.ts
@@ -1094,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";
@@ -1536,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",
@@ -1570,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);
   }
 };
@@ -1590,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);
   }
 };
@@ -1769,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.
+   *
+   * 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 to session input, 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
    */
-  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.