@brizz/sdk 0.1.7 → 0.1.9

package/dist/index.d.cts

@@ -5,6 +5,7 @@ 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;
@@ -15,9 +16,7 @@ interface IInstrumentModules {
     bedrock?: unknown;
     pinecone?: unknown;
     langchain?: {
-        chains?: unknown;
-        agents?: unknown;
-        tools?: unknown;
+        callbackManagerModule?: unknown;
     };
     llamaindex?: unknown;
     chromadb?: unknown;
@@ -125,6 +124,17 @@ declare function withProperties<F extends (this: any, ...args: any[]) => any>(pr
 }, fn: F, thisArg?: ThisParameterType<F>): F;
 declare function withSessionId<F extends (this: any, ...args: any[]) => any>(sessionId: string, fn: F, thisArg?: ThisParameterType<F>): F;
 declare function callWithSessionId<A extends unknown[], F extends (...args: A) => ReturnType<F>>(sessionId: string, fn: F, thisArg?: ThisParameterType<F>, ...args: A): ReturnType<F>;
+declare class Session {
+    readonly sessionId: string;
+    private readonly span;
+    private inputs;
+    private outputs;
+    constructor(sessionId: string, span: Span);
+    setInput(text: string): void;
+    setOutput(text: string): void;
+    updateProperties(properties: Record<string, AttributeValue>): void;
+}
+declare function startSession<T>(sessionId: string, callback: (session: Session) => T): T;
 
 declare function getMetricsExporter(): OTLPMetricExporter;
 declare function getMetricsReader(): MetricReader;
@@ -142,4 +152,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, callWithProperties, callWithSessionId, detectRuntime, emitEvent, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, 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, callWithProperties, callWithSessionId, detectRuntime, emitEvent, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, startSession, withProperties, withSessionId };

package/dist/index.d.ts

@@ -5,6 +5,7 @@ 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;
@@ -15,9 +16,7 @@ interface IInstrumentModules {
     bedrock?: unknown;
     pinecone?: unknown;
     langchain?: {
-        chains?: unknown;
-        agents?: unknown;
-        tools?: unknown;
+        callbackManagerModule?: unknown;
     };
     llamaindex?: unknown;
     chromadb?: unknown;
@@ -125,6 +124,17 @@ declare function withProperties<F extends (this: any, ...args: any[]) => any>(pr
 }, fn: F, thisArg?: ThisParameterType<F>): F;
 declare function withSessionId<F extends (this: any, ...args: any[]) => any>(sessionId: string, fn: F, thisArg?: ThisParameterType<F>): F;
 declare function callWithSessionId<A extends unknown[], F extends (...args: A) => ReturnType<F>>(sessionId: string, fn: F, thisArg?: ThisParameterType<F>, ...args: A): ReturnType<F>;
+declare class Session {
+    readonly sessionId: string;
+    private readonly span;
+    private inputs;
+    private outputs;
+    constructor(sessionId: string, span: Span);
+    setInput(text: string): void;
+    setOutput(text: string): void;
+    updateProperties(properties: Record<string, AttributeValue>): void;
+}
+declare function startSession<T>(sessionId: string, callback: (session: Session) => T): T;
 
 declare function getMetricsExporter(): OTLPMetricExporter;
 declare function getMetricsReader(): MetricReader;
@@ -142,4 +152,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, callWithProperties, callWithSessionId, detectRuntime, emitEvent, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, 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, callWithProperties, callWithSessionId, detectRuntime, emitEvent, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, startSession, withProperties, withSessionId };

package/dist/index.js

@@ -434,7 +434,7 @@ import {
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.7";
+  return "0.1.9";
 }
 
 // src/internal/log/processors/log-processor.ts
@@ -1069,6 +1069,9 @@ var BRIZZ = "brizz";
 var PROPERTIES = "properties";
 var SESSION_ID = "session.id";
 var PROPERTIES_CONTEXT_KEY = createContextKey(PROPERTIES);
+var SESSION_INPUT = "brizz.session.input";
+var SESSION_OUTPUT = "brizz.session.output";
+var SESSION_SPAN_NAME = "brizz.start_session";
 
 // src/internal/log/processors/log-processor.ts
 var DEFAULT_LOG_MASKING_RULES = [
@@ -1628,7 +1631,7 @@ function getSpanProcessor() {
 }
 
 // src/internal/trace/session.ts
-import { context as context3 } from "@opentelemetry/api";
+import { context as context3, trace, SpanStatusCode } from "@opentelemetry/api";
 function callWithProperties(properties, fn, thisArg, ...args) {
   const base = context3.active();
   const prev = base.getValue(PROPERTIES_CONTEXT_KEY);
@@ -1652,6 +1655,79 @@ function withSessionId(sessionId, fn, thisArg) {
 function callWithSessionId(sessionId, fn, thisArg, ...args) {
   return callWithProperties({ [SESSION_ID]: sessionId }, fn, thisArg, ...args);
 }
+var Session = class {
+  sessionId;
+  span;
+  inputs = [];
+  outputs = [];
+  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.
+   *
+   * @param text - Text to append to session input
+   */
+  setInput(text) {
+    this.inputs.push(text);
+    this.span.setAttribute(SESSION_INPUT, JSON.stringify(this.inputs));
+  }
+  /**
+   * (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.
+   *
+   * @param text - Text to append to session output
+   */
+  setOutput(text) {
+    this.outputs.push(text);
+    this.span.setAttribute(SESSION_OUTPUT, JSON.stringify(this.outputs));
+  }
+  /**
+   * Update custom properties on the session span.
+   * Properties are prefixed with 'brizz.'.
+   *
+   * @param properties - Key-value properties to set on the session
+   */
+  updateProperties(properties) {
+    for (const [key, value] of Object.entries(properties)) {
+      this.span.setAttribute(`${BRIZZ}.${key}`, value);
+    }
+  }
+};
+function startSession(sessionId, callback) {
+  const tracer = trace.getTracer("@brizz/sdk");
+  return tracer.startActiveSpan(SESSION_SPAN_NAME, (span) => {
+    span.setAttribute(`${BRIZZ}.${SESSION_ID}`, sessionId);
+    const session = new Session(sessionId, span);
+    return callWithProperties({ [SESSION_ID]: sessionId }, () => {
+      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;
+      }
+    });
+  });
+}
 
 // src/internal/sdk.ts
 var _Brizz = class __Brizz {
@@ -1901,6 +1977,7 @@ export {
   Brizz,
   DEFAULT_PII_PATTERNS,
   LogLevel,
+  Session,
   SeverityNumber2 as SeverityNumber,
   callWithProperties,
   callWithSessionId,
@@ -1916,6 +1993,7 @@ export {
   maskAttributes,
   maskValue,
   setLogLevel,
+  startSession,
   withProperties,
   withSessionId
 };