@brizz/sdk 0.1.13 → 0.1.15

package/dist/index.d.cts

@@ -130,10 +130,11 @@ 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;
     updateProperties(properties: Record<string, AttributeValue>): void;
 }
+declare function getActiveSession(): Session | undefined;
 declare function startSession<T>(sessionId: string, callback: (session: Session) => T, extraProperties?: Record<string, AttributeValue>): T;
 
 declare function getMetricsExporter(): OTLPMetricExporter;
@@ -152,4 +153,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, callWithProperties, callWithSessionId, detectRuntime, emitEvent, getActiveSession, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, startSession, withProperties, withSessionId };

package/dist/index.d.ts

@@ -130,10 +130,11 @@ 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;
     updateProperties(properties: Record<string, AttributeValue>): void;
 }
+declare function getActiveSession(): Session | undefined;
 declare function startSession<T>(sessionId: string, callback: (session: Session) => T, extraProperties?: Record<string, AttributeValue>): T;
 
 declare function getMetricsExporter(): OTLPMetricExporter;
@@ -152,4 +153,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, callWithProperties, callWithSessionId, detectRuntime, emitEvent, getActiveSession, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, startSession, withProperties, withSessionId };

package/dist/index.js

@@ -318,7 +318,6 @@ var BRIZZ_SDK_LANGUAGE = "brizz.sdk.language";
 var SDK_LANGUAGE = "typescript";
 
 // src/internal/instrumentation/registry.ts
-import { LangChainInstrumentation } from "@arizeai/openinference-instrumentation-langchain";
 import { AnthropicInstrumentation as AnthropicInstrumentation2 } from "@traceloop/instrumentation-anthropic";
 import { BedrockInstrumentation as BedrockInstrumentation2 } from "@traceloop/instrumentation-bedrock";
 import { ChromaDBInstrumentation as ChromaDBInstrumentation2 } from "@traceloop/instrumentation-chromadb";
@@ -419,14 +418,19 @@ var InstrumentationRegistry = class _InstrumentationRegistry {
       }
     }
     if (this.manualModules?.langchain?.callbackManagerModule) {
-      try {
-        const lcInst = new LangChainInstrumentation();
-        lcInst.manuallyInstrument(this.manualModules.langchain.callbackManagerModule);
-        instrumentations.push(lcInst);
-        logger.debug("Manual instrumentation enabled for LangChain");
-      } catch (error) {
-        logger.error(`Failed to load LangChain instrumentation: ${String(error)}`);
-      }
+      const callbackManagerModule = this.manualModules.langchain.callbackManagerModule;
+      void (async () => {
+        try {
+          const { LangChainInstrumentation } = await import("@arizeai/openinference-instrumentation-langchain");
+          const lcInst = new LangChainInstrumentation();
+          lcInst.manuallyInstrument(callbackManagerModule);
+          logger.debug("Manual instrumentation enabled for LangChain");
+        } catch (error) {
+          logger.error(
+            `Failed to load LangChain instrumentation. Ensure @langchain/core is installed: ${String(error)}`
+          );
+        }
+      })();
     }
   }
 };
@@ -441,7 +445,7 @@ import {
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.13";
+  return "0.1.15";
 }
 
 // src/internal/log/processors/log-processor.ts
@@ -1078,6 +1082,7 @@ 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";
@@ -1713,7 +1718,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);
@@ -1724,7 +1729,7 @@ 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);
@@ -1742,6 +1747,9 @@ var Session = class {
     }
   }
 };
+function getActiveSession() {
+  return context3.active().getValue(SESSION_OBJECT_CONTEXT_KEY);
+}
 function startSession(sessionId, callback, extraProperties) {
   const tracer = trace.getTracer("@brizz/sdk");
   return tracer.startActiveSpan(SESSION_SPAN_NAME, (span) => {
@@ -1759,27 +1767,30 @@ function startSession(sessionId, callback, extraProperties) {
       }
     }
     return callWithProperties(contextProperties, () => {
-      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;
-          });
+      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;
         }
-        span.end();
-        return result;
-      } catch (error) {
-        span.recordException(error);
-        span.setStatus({ code: SpanStatusCode.ERROR });
-        span.end();
-        throw error;
-      }
+      });
     });
   });
 }
@@ -2038,6 +2049,7 @@ export {
   callWithSessionId,
   detectRuntime,
   emitEvent,
+  getActiveSession,
   getLogLevel,
   getMetricsExporter,
   getMetricsReader,