@fallom/trace 0.1.4 → 0.1.5

package/dist/index.d.mts

@@ -7,6 +7,7 @@
 interface SessionContext {
     configKey: string;
     sessionId: string;
+    customerId?: string;
 }
 /**
  * Initialize Fallom tracing. Auto-instruments all LLM calls.
@@ -42,34 +43,36 @@ declare function init$3(options?: {
  * Set the current session context.
  *
  * All subsequent LLM calls in this async context will be
- * automatically tagged with this configKey and sessionId.
+ * automatically tagged with this configKey, sessionId, and customerId.
  *
  * @param configKey - Your config name (e.g., "linkedin-agent")
  * @param sessionId - Your session/conversation ID
+ * @param customerId - Optional customer/user identifier for analytics
  *
  * @example
  * ```typescript
- * trace.setSession("linkedin-agent", sessionId);
- * await agent.run(message); // Automatically traced with session
+ * trace.setSession("linkedin-agent", sessionId, "user_123");
+ * await agent.run(message); // Automatically traced with session + customer
  * ```
  */
-declare function setSession(configKey: string, sessionId: string): void;
+declare function setSession(configKey: string, sessionId: string, customerId?: string): void;
 /**
  * Run a function with session context.
  * Use this to ensure session context propagates across async boundaries.
  *
  * @param configKey - Your config name
  * @param sessionId - Your session ID
+ * @param customerId - Optional customer/user identifier
  * @param fn - Function to run with session context
  *
  * @example
  * ```typescript
- * await trace.runWithSession("my-agent", sessionId, async () => {
+ * await trace.runWithSession("my-agent", sessionId, "user_123", async () => {
  *   await agent.run(message); // Has session context
  * });
  * ```
  */
-declare function runWithSession<T>(configKey: string, sessionId: string, fn: () => T): T;
+declare function runWithSession<T>(configKey: string, sessionId: string, customerIdOrFn: string | (() => T), fn?: () => T): T;
 /**
  * Get current session context, if any.
  */

package/dist/index.d.ts

@@ -7,6 +7,7 @@
 interface SessionContext {
     configKey: string;
     sessionId: string;
+    customerId?: string;
 }
 /**
  * Initialize Fallom tracing. Auto-instruments all LLM calls.
@@ -42,34 +43,36 @@ declare function init$3(options?: {
  * Set the current session context.
  *
  * All subsequent LLM calls in this async context will be
- * automatically tagged with this configKey and sessionId.
+ * automatically tagged with this configKey, sessionId, and customerId.
  *
  * @param configKey - Your config name (e.g., "linkedin-agent")
  * @param sessionId - Your session/conversation ID
+ * @param customerId - Optional customer/user identifier for analytics
  *
  * @example
  * ```typescript
- * trace.setSession("linkedin-agent", sessionId);
- * await agent.run(message); // Automatically traced with session
+ * trace.setSession("linkedin-agent", sessionId, "user_123");
+ * await agent.run(message); // Automatically traced with session + customer
  * ```
  */
-declare function setSession(configKey: string, sessionId: string): void;
+declare function setSession(configKey: string, sessionId: string, customerId?: string): void;
 /**
  * Run a function with session context.
  * Use this to ensure session context propagates across async boundaries.
  *
  * @param configKey - Your config name
  * @param sessionId - Your session ID
+ * @param customerId - Optional customer/user identifier
  * @param fn - Function to run with session context
  *
  * @example
  * ```typescript
- * await trace.runWithSession("my-agent", sessionId, async () => {
+ * await trace.runWithSession("my-agent", sessionId, "user_123", async () => {
  *   await agent.run(message); // Has session context
  * });
  * ```
  */
-declare function runWithSession<T>(configKey: string, sessionId: string, fn: () => T): T;
+declare function runWithSession<T>(configKey: string, sessionId: string, customerIdOrFn: string | (() => T), fn?: () => T): T;
 /**
  * Get current session context, if any.
  */

package/dist/index.js

@@ -930,7 +930,15 @@ var fallomSpanProcessor = {
     if (ctx) {
       span2.setAttribute("fallom.config_key", ctx.configKey);
       span2.setAttribute("fallom.session_id", ctx.sessionId);
-      log2("   Added session context:", ctx.configKey, ctx.sessionId);
+      if (ctx.customerId) {
+        span2.setAttribute("fallom.customer_id", ctx.customerId);
+      }
+      log2(
+        "   Added session context:",
+        ctx.configKey,
+        ctx.sessionId,
+        ctx.customerId
+      );
     } else {
       log2("   No session context available");
     }
@@ -1044,16 +1052,23 @@ async function tryAddInstrumentation(instrumentations, pkg, className) {
     log2(`   \u274C ${pkg} not installed`);
   }
 }
-function setSession(configKey, sessionId) {
+function setSession(configKey, sessionId, customerId) {
   const store = sessionStorage.getStore();
   if (store) {
     store.configKey = configKey;
     store.sessionId = sessionId;
+    store.customerId = customerId;
   }
-  fallbackSession = { configKey, sessionId };
+  fallbackSession = { configKey, sessionId, customerId };
 }
-function runWithSession(configKey, sessionId, fn) {
-  return sessionStorage.run({ configKey, sessionId }, fn);
+function runWithSession(configKey, sessionId, customerIdOrFn, fn) {
+  if (typeof customerIdOrFn === "function") {
+    return sessionStorage.run({ configKey, sessionId }, customerIdOrFn);
+  }
+  return sessionStorage.run(
+    { configKey, sessionId, customerId: customerIdOrFn },
+    fn
+  );
 }
 function getSession() {
   return sessionStorage.getStore() || fallbackSession || void 0;
@@ -1144,6 +1159,7 @@ function wrapOpenAI(client) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "chat.completions.create",
         model: response?.model || params?.model,
         start_time: new Date(startTime).toISOString(),
@@ -1167,6 +1183,7 @@ function wrapOpenAI(client) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "chat.completions.create",
         model: params?.model,
         start_time: new Date(startTime).toISOString(),
@@ -1206,6 +1223,7 @@ function wrapAnthropic(client) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "messages.create",
         model: response?.model || params?.model,
         start_time: new Date(startTime).toISOString(),
@@ -1229,6 +1247,7 @@ function wrapAnthropic(client) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "messages.create",
         model: params?.model,
         start_time: new Date(startTime).toISOString(),
@@ -1269,6 +1288,7 @@ function wrapGoogleAI(model) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "generateContent",
         model: model?.model || "gemini",
         start_time: new Date(startTime).toISOString(),
@@ -1292,6 +1312,7 @@ function wrapGoogleAI(model) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "generateContent",
         model: model?.model || "gemini",
         start_time: new Date(startTime).toISOString(),

package/dist/index.mjs

@@ -664,7 +664,15 @@ var fallomSpanProcessor = {
     if (ctx) {
       span2.setAttribute("fallom.config_key", ctx.configKey);
       span2.setAttribute("fallom.session_id", ctx.sessionId);
-      log("   Added session context:", ctx.configKey, ctx.sessionId);
+      if (ctx.customerId) {
+        span2.setAttribute("fallom.customer_id", ctx.customerId);
+      }
+      log(
+        "   Added session context:",
+        ctx.configKey,
+        ctx.sessionId,
+        ctx.customerId
+      );
     } else {
       log("   No session context available");
     }
@@ -778,16 +786,23 @@ async function tryAddInstrumentation(instrumentations, pkg, className) {
     log(`   \u274C ${pkg} not installed`);
   }
 }
-function setSession(configKey, sessionId) {
+function setSession(configKey, sessionId, customerId) {
   const store = sessionStorage.getStore();
   if (store) {
     store.configKey = configKey;
     store.sessionId = sessionId;
+    store.customerId = customerId;
   }
-  fallbackSession = { configKey, sessionId };
+  fallbackSession = { configKey, sessionId, customerId };
 }
-function runWithSession(configKey, sessionId, fn) {
-  return sessionStorage.run({ configKey, sessionId }, fn);
+function runWithSession(configKey, sessionId, customerIdOrFn, fn) {
+  if (typeof customerIdOrFn === "function") {
+    return sessionStorage.run({ configKey, sessionId }, customerIdOrFn);
+  }
+  return sessionStorage.run(
+    { configKey, sessionId, customerId: customerIdOrFn },
+    fn
+  );
 }
 function getSession() {
   return sessionStorage.getStore() || fallbackSession || void 0;
@@ -878,6 +893,7 @@ function wrapOpenAI(client) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "chat.completions.create",
         model: response?.model || params?.model,
         start_time: new Date(startTime).toISOString(),
@@ -901,6 +917,7 @@ function wrapOpenAI(client) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "chat.completions.create",
         model: params?.model,
         start_time: new Date(startTime).toISOString(),
@@ -940,6 +957,7 @@ function wrapAnthropic(client) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "messages.create",
         model: response?.model || params?.model,
         start_time: new Date(startTime).toISOString(),
@@ -963,6 +981,7 @@ function wrapAnthropic(client) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "messages.create",
         model: params?.model,
         start_time: new Date(startTime).toISOString(),
@@ -1003,6 +1022,7 @@ function wrapGoogleAI(model) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "generateContent",
         model: model?.model || "gemini",
         start_time: new Date(startTime).toISOString(),
@@ -1026,6 +1046,7 @@ function wrapGoogleAI(model) {
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
         name: "generateContent",
         model: model?.model || "gemini",
         start_time: new Date(startTime).toISOString(),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fallom/trace",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "Model A/B testing and tracing for LLM applications. Zero latency, production-ready.",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",