@brizz/sdk 0.1.14 → 0.1.16

package/dist/index.d.cts

@@ -130,11 +130,23 @@ declare class Session {
     private inputs;
     private outputs;
     constructor(sessionId: string, span: Span);
-    setInput(text: string): void;
-    setOutput(text: string): void;
+    setInput(text: string | null): void;
+    setOutput(text: string | null): void;
+    setTitle(title: string): void;
     updateProperties(properties: Record<string, AttributeValue>): void;
 }
-declare function startSession<T>(sessionId: string, callback: (session: Session) => T, extraProperties?: Record<string, AttributeValue>): T;
+declare class SessionTitle {
+    private readonly span;
+    constructor(span: Span);
+    setTitle(title: string): void;
+}
+declare function getActiveSession(): Session | undefined;
+declare function startSession<T>(sessionId: string, callback: (session: Session) => T, extraProperties?: Record<string, AttributeValue>, options?: {
+    mode?: 'default' | 'title';
+}): T;
+declare function startSessionTitle<T>(callback: (title: SessionTitle) => T, options?: {
+    sessionId?: string;
+}): T;
 
 declare function getMetricsExporter(): OTLPMetricExporter;
 declare function getMetricsReader(): MetricReader;
@@ -152,4 +164,4 @@ declare namespace init {
   export {  };
 }
 
-export { Brizz, DEFAULT_PII_PATTERNS, type IAttributesMaskingRule, type IBrizzInitializeOptions, type IEventMaskingConfig, type IEventMaskingRule, type IInstrumentModules, type ILogMaskingConfig, type ILogger, type IMaskingConfig, type ISpanMaskingConfig, LogLevel, type MaskingMode, type RuntimeInfo, Session, callWithProperties, callWithSessionId, detectRuntime, emitEvent, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, startSession, withProperties, withSessionId };
+export { Brizz, DEFAULT_PII_PATTERNS, type IAttributesMaskingRule, type IBrizzInitializeOptions, type IEventMaskingConfig, type IEventMaskingRule, type IInstrumentModules, type ILogMaskingConfig, type ILogger, type IMaskingConfig, type ISpanMaskingConfig, LogLevel, type MaskingMode, type RuntimeInfo, Session, SessionTitle, callWithProperties, callWithSessionId, detectRuntime, emitEvent, getActiveSession, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, startSession, startSessionTitle, withProperties, withSessionId };

package/dist/index.d.ts

@@ -130,11 +130,23 @@ declare class Session {
     private inputs;
     private outputs;
     constructor(sessionId: string, span: Span);
-    setInput(text: string): void;
-    setOutput(text: string): void;
+    setInput(text: string | null): void;
+    setOutput(text: string | null): void;
+    setTitle(title: string): void;
     updateProperties(properties: Record<string, AttributeValue>): void;
 }
-declare function startSession<T>(sessionId: string, callback: (session: Session) => T, extraProperties?: Record<string, AttributeValue>): T;
+declare class SessionTitle {
+    private readonly span;
+    constructor(span: Span);
+    setTitle(title: string): void;
+}
+declare function getActiveSession(): Session | undefined;
+declare function startSession<T>(sessionId: string, callback: (session: Session) => T, extraProperties?: Record<string, AttributeValue>, options?: {
+    mode?: 'default' | 'title';
+}): T;
+declare function startSessionTitle<T>(callback: (title: SessionTitle) => T, options?: {
+    sessionId?: string;
+}): T;
 
 declare function getMetricsExporter(): OTLPMetricExporter;
 declare function getMetricsReader(): MetricReader;
@@ -152,4 +164,4 @@ declare namespace init {
   export {  };
 }
 
-export { Brizz, DEFAULT_PII_PATTERNS, type IAttributesMaskingRule, type IBrizzInitializeOptions, type IEventMaskingConfig, type IEventMaskingRule, type IInstrumentModules, type ILogMaskingConfig, type ILogger, type IMaskingConfig, type ISpanMaskingConfig, LogLevel, type MaskingMode, type RuntimeInfo, Session, callWithProperties, callWithSessionId, detectRuntime, emitEvent, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, startSession, withProperties, withSessionId };
+export { Brizz, DEFAULT_PII_PATTERNS, type IAttributesMaskingRule, type IBrizzInitializeOptions, type IEventMaskingConfig, type IEventMaskingRule, type IInstrumentModules, type ILogMaskingConfig, type ILogger, type IMaskingConfig, type ISpanMaskingConfig, LogLevel, type MaskingMode, type RuntimeInfo, Session, SessionTitle, callWithProperties, callWithSessionId, detectRuntime, emitEvent, getActiveSession, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, startSession, startSessionTitle, withProperties, withSessionId };

package/dist/index.js

@@ -445,7 +445,7 @@ import {
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.14";
+  return "0.1.16";
 }
 
 // src/internal/log/processors/log-processor.ts
@@ -1082,9 +1082,13 @@ 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 SESSION_INPUT = "brizz.session.input";
 var SESSION_OUTPUT = "brizz.session.output";
 var SESSION_SPAN_NAME = "brizz.start_session";
+var SESSION_TITLE_SPAN_NAME = "brizz.session_title";
+var SESSION_TITLE_GENERATION = "session.title_generation";
+var SESSION_TITLE = "brizz.session.title";
 
 // src/internal/log/processors/log-processor.ts
 var DEFAULT_LOG_MASKING_RULES = [
@@ -1717,7 +1721,7 @@ var Session = class {
    * Use when you need to track specific input data that differs from what's sent to the LLM.
    * Multiple calls accumulate in an array.
    *
-   * @param text - Text to append to session input
+   * @param text - Text to append to session input, or null to append null
    */
   setInput(text) {
     this.inputs.push(text);
@@ -1728,12 +1732,20 @@ var Session = class {
    * Use when you need to track specific output data that differs from what's received from the LLM.
    * Multiple calls accumulate in an array.
    *
-   * @param text - Text to append to session output
+   * @param text - Text to append to session output, or null to append null
    */
   setOutput(text) {
     this.outputs.push(text);
     this.span.setAttribute(SESSION_OUTPUT, JSON.stringify(this.outputs));
   }
+  /**
+   * Set the session title on the span.
+   *
+   * @param title - The generated title string
+   */
+  setTitle(title) {
+    this.span.setAttribute(SESSION_TITLE, title);
+  }
   /**
    * Update custom properties on the session span.
    * Properties are prefixed with 'brizz.'.
@@ -1746,9 +1758,28 @@ var Session = class {
     }
   }
 };
-function startSession(sessionId, callback, extraProperties) {
+var SessionTitle = class {
+  span;
+  constructor(span) {
+    this.span = span;
+  }
+  /**
+   * Set the generated title on the wrapper span.
+   *
+   * @param title - The generated title string
+   */
+  setTitle(title) {
+    this.span.setAttribute(SESSION_TITLE, title);
+  }
+};
+function getActiveSession() {
+  return context3.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 = trace.getTracer("@brizz/sdk");
-  return tracer.startActiveSpan(SESSION_SPAN_NAME, (span) => {
+  return tracer.startActiveSpan(spanName, (span) => {
     span.setAttribute(`${BRIZZ}.${SESSION_ID}`, sessionId);
     if (extraProperties) {
       for (const [key, value] of Object.entries(extraProperties)) {
@@ -1757,14 +1788,57 @@ function startSession(sessionId, callback, extraProperties) {
     }
     const session = new Session(sessionId, span);
     const contextProperties = { [SESSION_ID]: sessionId };
+    if (isTitle) {
+      contextProperties[SESSION_TITLE_GENERATION] = "true";
+    }
     if (extraProperties) {
       for (const [key, value] of Object.entries(extraProperties)) {
         contextProperties[key] = String(value);
       }
     }
     return callWithProperties(contextProperties, () => {
+      const sessionCtx = context3.active().setValue(SESSION_OBJECT_CONTEXT_KEY, session);
+      return context3.with(sessionCtx, () => {
+        try {
+          const result = callback(session);
+          if (result && typeof result.then === "function") {
+            return result.then((value) => {
+              span.end();
+              return value;
+            }).catch((error) => {
+              span.recordException(error);
+              span.setStatus({ code: SpanStatusCode.ERROR });
+              span.end();
+              throw error;
+            });
+          }
+          span.end();
+          return result;
+        } catch (error) {
+          span.recordException(error);
+          span.setStatus({ code: SpanStatusCode.ERROR });
+          span.end();
+          throw error;
+        }
+      });
+    });
+  });
+}
+function startSessionTitle(callback, options) {
+  const resolvedSessionId = options?.sessionId ?? getActiveSession()?.sessionId;
+  const tracer = trace.getTracer("@brizz/sdk");
+  return tracer.startActiveSpan(SESSION_TITLE_SPAN_NAME, (span) => {
+    if (resolvedSessionId) {
+      span.setAttribute(`${BRIZZ}.${SESSION_ID}`, resolvedSessionId);
+    }
+    const sessionTitle = new SessionTitle(span);
+    const properties = { [SESSION_TITLE_GENERATION]: "true" };
+    if (resolvedSessionId) {
+      properties[SESSION_ID] = resolvedSessionId;
+    }
+    return callWithProperties(properties, () => {
       try {
-        const result = callback(session);
+        const result = callback(sessionTitle);
         if (result && typeof result.then === "function") {
           return result.then((value) => {
             span.end();
@@ -2037,11 +2111,13 @@ export {
   DEFAULT_PII_PATTERNS,
   LogLevel,
   Session,
+  SessionTitle,
   SeverityNumber2 as SeverityNumber,
   callWithProperties,
   callWithSessionId,
   detectRuntime,
   emitEvent,
+  getActiveSession,
   getLogLevel,
   getMetricsExporter,
   getMetricsReader,
@@ -2053,6 +2129,7 @@ export {
   maskValue,
   setLogLevel,
   startSession,
+  startSessionTitle,
   withProperties,
   withSessionId
 };