@brizz/sdk 0.1.13 → 0.1.15

package/README.md

@@ -313,6 +313,26 @@ await startSession('session-456', async (session) => {
 });
 ```
 
+### Accessing the Active Session
+
+Use `getActiveSession()` to retrieve the current session from anywhere within a `startSession` scope — no need to pass the session object through your call stack:
+
+```typescript
+import { startSession, getActiveSession } from '@brizz/sdk';
+
+function deepHelper() {
+  const session = getActiveSession();
+  session?.updateProperties({ step: 'helper' });
+}
+
+await startSession('session-123', async () => {
+  deepHelper(); // accesses session without it being passed as a parameter
+});
+
+// Outside a session, returns undefined
+getActiveSession(); // undefined
+```
+
 ### Function Wrapper Pattern
 
 For simpler cases where you just need to tag traces with a session ID:

package/dist/index.cjs

@@ -39,6 +39,7 @@ __export(src_exports, {
   callWithSessionId: () => callWithSessionId,
   detectRuntime: () => detectRuntime,
   emitEvent: () => emitEvent,
+  getActiveSession: () => getActiveSession,
   getLogLevel: () => getLogLevel,
   getMetricsExporter: () => getMetricsExporter,
   getMetricsReader: () => getMetricsReader,
@@ -368,7 +369,6 @@ var BRIZZ_SDK_LANGUAGE = "brizz.sdk.language";
 var SDK_LANGUAGE = "typescript";
 
 // src/internal/instrumentation/registry.ts
-var import_openinference_instrumentation_langchain = require("@arizeai/openinference-instrumentation-langchain");
 var import_instrumentation_anthropic2 = require("@traceloop/instrumentation-anthropic");
 var import_instrumentation_bedrock2 = require("@traceloop/instrumentation-bedrock");
 var import_instrumentation_chromadb2 = require("@traceloop/instrumentation-chromadb");
@@ -469,14 +469,19 @@ var InstrumentationRegistry = class _InstrumentationRegistry {
       }
     }
     if (this.manualModules?.langchain?.callbackManagerModule) {
-      try {
-        const lcInst = new import_openinference_instrumentation_langchain.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)}`
+          );
+        }
+      })();
     }
   }
 };
@@ -489,7 +494,7 @@ var import_sdk_logs2 = require("@opentelemetry/sdk-logs");
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.13";
+  return "0.1.15";
 }
 
 // src/internal/log/processors/log-processor.ts
@@ -1126,6 +1131,7 @@ var BRIZZ = "brizz";
 var PROPERTIES = "properties";
 var SESSION_ID = "session.id";
 var PROPERTIES_CONTEXT_KEY = (0, import_api2.createContextKey)(PROPERTIES);
+var SESSION_OBJECT_CONTEXT_KEY = (0, import_api2.createContextKey)("brizz.session.object");
 var SESSION_INPUT = "brizz.session.input";
 var SESSION_OUTPUT = "brizz.session.output";
 var SESSION_SPAN_NAME = "brizz.start_session";
@@ -1758,7 +1764,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);
@@ -1769,7 +1775,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);
@@ -1787,6 +1793,9 @@ var Session = class {
     }
   }
 };
+function getActiveSession() {
+  return import_api5.context.active().getValue(SESSION_OBJECT_CONTEXT_KEY);
+}
 function startSession(sessionId, callback, extraProperties) {
   const tracer = import_api5.trace.getTracer("@brizz/sdk");
   return tracer.startActiveSpan(SESSION_SPAN_NAME, (span) => {
@@ -1804,27 +1813,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: import_api5.SpanStatusCode.ERROR });
-            span.end();
-            throw error;
-          });
+      const sessionCtx = import_api5.context.active().setValue(SESSION_OBJECT_CONTEXT_KEY, session);
+      return import_api5.context.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: import_api5.SpanStatusCode.ERROR });
+              span.end();
+              throw error;
+            });
+          }
+          span.end();
+          return result;
+        } catch (error) {
+          span.recordException(error);
+          span.setStatus({ code: import_api5.SpanStatusCode.ERROR });
+          span.end();
+          throw error;
         }
-        span.end();
-        return result;
-      } catch (error) {
-        span.recordException(error);
-        span.setStatus({ code: import_api5.SpanStatusCode.ERROR });
-        span.end();
-        throw error;
-      }
+      });
     });
   });
 }
@@ -2084,6 +2096,7 @@ var init_exports = {};
   callWithSessionId,
   detectRuntime,
   emitEvent,
+  getActiveSession,
   getLogLevel,
   getMetricsExporter,
   getMetricsReader,