@fallom/trace 0.1.3 → 0.1.5

package/dist/index.mjs

@@ -1,15 +1,15 @@
-var __defProp = Object.defineProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
+import {
+  __export,
+  init,
+  prompts_exports
+} from "./chunk-IGJD7GBO.mjs";
 
 // src/trace.ts
 var trace_exports = {};
 __export(trace_exports, {
   clearSession: () => clearSession,
   getSession: () => getSession,
-  init: () => init,
+  init: () => init2,
   runWithSession: () => runWithSession,
   setSession: () => setSession,
   shutdown: () => shutdown,
@@ -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");
     }
@@ -679,7 +687,7 @@ var fallomSpanProcessor = {
     return Promise.resolve();
   }
 };
-async function init(options = {}) {
+async function init2(options = {}) {
   if (initialized) return;
   debugMode = options.debug ?? false;
   log("\u{1F680} Initializing Fallom tracing...");
@@ -722,13 +730,41 @@ async function init(options = {}) {
 }
 async function getInstrumentations() {
   const instrumentations = [];
-  await tryAddInstrumentation(instrumentations, "@traceloop/instrumentation-openai", "OpenAIInstrumentation");
-  await tryAddInstrumentation(instrumentations, "@traceloop/instrumentation-anthropic", "AnthropicInstrumentation");
-  await tryAddInstrumentation(instrumentations, "@traceloop/instrumentation-cohere", "CohereInstrumentation");
-  await tryAddInstrumentation(instrumentations, "@traceloop/instrumentation-bedrock", "BedrockInstrumentation");
-  await tryAddInstrumentation(instrumentations, "@traceloop/instrumentation-google-generativeai", "GoogleGenerativeAIInstrumentation");
-  await tryAddInstrumentation(instrumentations, "@traceloop/instrumentation-azure", "AzureOpenAIInstrumentation");
-  await tryAddInstrumentation(instrumentations, "@traceloop/instrumentation-vertexai", "VertexAIInstrumentation");
+  await tryAddInstrumentation(
+    instrumentations,
+    "@traceloop/instrumentation-openai",
+    "OpenAIInstrumentation"
+  );
+  await tryAddInstrumentation(
+    instrumentations,
+    "@traceloop/instrumentation-anthropic",
+    "AnthropicInstrumentation"
+  );
+  await tryAddInstrumentation(
+    instrumentations,
+    "@traceloop/instrumentation-cohere",
+    "CohereInstrumentation"
+  );
+  await tryAddInstrumentation(
+    instrumentations,
+    "@traceloop/instrumentation-bedrock",
+    "BedrockInstrumentation"
+  );
+  await tryAddInstrumentation(
+    instrumentations,
+    "@traceloop/instrumentation-google-generativeai",
+    "GoogleGenerativeAIInstrumentation"
+  );
+  await tryAddInstrumentation(
+    instrumentations,
+    "@traceloop/instrumentation-azure",
+    "AzureOpenAIInstrumentation"
+  );
+  await tryAddInstrumentation(
+    instrumentations,
+    "@traceloop/instrumentation-vertexai",
+    "VertexAIInstrumentation"
+  );
   return instrumentations;
 }
 async function tryAddInstrumentation(instrumentations, pkg, className) {
@@ -736,25 +772,37 @@ async function tryAddInstrumentation(instrumentations, pkg, className) {
     const mod = await import(pkg);
     const InstrumentationClass = mod[className] || mod.default?.[className];
     if (InstrumentationClass) {
-      instrumentations.push(new InstrumentationClass({ traceContent: captureContent }));
+      instrumentations.push(
+        new InstrumentationClass({ traceContent: captureContent })
+      );
       log(`   \u2705 Loaded ${pkg}`);
     } else {
-      log(`   \u26A0\uFE0F ${pkg} loaded but ${className} not found. Available:`, Object.keys(mod));
+      log(
+        `   \u26A0\uFE0F ${pkg} loaded but ${className} not found. Available:`,
+        Object.keys(mod)
+      );
     }
   } catch (e) {
     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;
@@ -823,12 +871,20 @@ async function sendTrace(trace) {
   }
 }
 function wrapOpenAI(client) {
-  const originalCreate = client.chat.completions.create.bind(client.chat.completions);
+  const originalCreate = client.chat.completions.create.bind(
+    client.chat.completions
+  );
   client.chat.completions.create = async function(...args) {
     const ctx = sessionStorage.getStore() || fallbackSession;
     if (!ctx || !initialized) {
       return originalCreate(...args);
     }
+    let promptCtx = null;
+    try {
+      const { getPromptContext } = await import("./prompts-67DJ33I4.mjs");
+      promptCtx = getPromptContext();
+    } catch {
+    }
     const params = args[0] || {};
     const startTime = Date.now();
     try {
@@ -837,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(),
@@ -847,7 +904,11 @@ function wrapOpenAI(client) {
         completion_tokens: response?.usage?.completion_tokens,
         total_tokens: response?.usage?.total_tokens,
         input: captureContent ? JSON.stringify(params?.messages) : void 0,
-        output: captureContent ? response?.choices?.[0]?.message?.content : void 0
+        output: captureContent ? response?.choices?.[0]?.message?.content : void 0,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
       }).catch(() => {
       });
       return response;
@@ -856,13 +917,18 @@ 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(),
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "ERROR",
-        error_message: error?.message
+        error_message: error?.message,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
       }).catch(() => {
       });
       throw error;
@@ -877,6 +943,12 @@ function wrapAnthropic(client) {
     if (!ctx || !initialized) {
       return originalCreate(...args);
     }
+    let promptCtx = null;
+    try {
+      const { getPromptContext } = await import("./prompts-67DJ33I4.mjs");
+      promptCtx = getPromptContext();
+    } catch {
+    }
     const params = args[0] || {};
     const startTime = Date.now();
     try {
@@ -885,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(),
@@ -895,7 +968,11 @@ function wrapAnthropic(client) {
         completion_tokens: response?.usage?.output_tokens,
         total_tokens: (response?.usage?.input_tokens || 0) + (response?.usage?.output_tokens || 0),
         input: captureContent ? JSON.stringify(params?.messages) : void 0,
-        output: captureContent ? response?.content?.[0]?.text : void 0
+        output: captureContent ? response?.content?.[0]?.text : void 0,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
       }).catch(() => {
       });
       return response;
@@ -904,13 +981,18 @@ 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(),
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "ERROR",
-        error_message: error?.message
+        error_message: error?.message,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
       }).catch(() => {
       });
       throw error;
@@ -925,6 +1007,12 @@ function wrapGoogleAI(model) {
     if (!ctx || !initialized) {
       return originalGenerate(...args);
     }
+    let promptCtx = null;
+    try {
+      const { getPromptContext } = await import("./prompts-67DJ33I4.mjs");
+      promptCtx = getPromptContext();
+    } catch {
+    }
     const startTime = Date.now();
     try {
       const response = await originalGenerate(...args);
@@ -934,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(),
@@ -944,7 +1033,11 @@ function wrapGoogleAI(model) {
         completion_tokens: usage?.candidatesTokenCount,
         total_tokens: usage?.totalTokenCount,
         input: captureContent ? JSON.stringify(args[0]) : void 0,
-        output: captureContent ? result?.text?.() : void 0
+        output: captureContent ? result?.text?.() : void 0,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
       }).catch(() => {
       });
       return response;
@@ -953,13 +1046,18 @@ 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(),
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "ERROR",
-        error_message: error?.message
+        error_message: error?.message,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
       }).catch(() => {
       });
       throw error;
@@ -972,7 +1070,7 @@ function wrapGoogleAI(model) {
 var models_exports = {};
 __export(models_exports, {
   get: () => get,
-  init: () => init2
+  init: () => init3
 });
 import { createHash } from "crypto";
 var apiKey2 = null;
@@ -988,7 +1086,7 @@ function log2(msg) {
     console.log(`[Fallom] ${msg}`);
   }
 }
-function init2(options = {}) {
+function init3(options = {}) {
   apiKey2 = options.apiKey || process.env.FALLOM_API_KEY || null;
   baseUrl2 = options.baseUrl || process.env.FALLOM_BASE_URL || "https://spans.fallom.com";
   initialized2 = true;
@@ -1008,7 +1106,7 @@ function init2(options = {}) {
 function ensureInit() {
   if (!initialized2) {
     try {
-      init2();
+      init3();
     } catch {
     }
   }
@@ -1192,15 +1290,19 @@ async function recordSession(configKey, version, sessionId, model) {
 }
 
 // src/init.ts
-async function init3(options = {}) {
+async function init4(options = {}) {
   const baseUrl3 = options.baseUrl || process.env.FALLOM_BASE_URL || "https://spans.fallom.com";
-  await init({
+  await init2({
     apiKey: options.apiKey,
     baseUrl: baseUrl3,
     captureContent: options.captureContent,
     debug: options.debug
   });
-  init2({
+  init3({
+    apiKey: options.apiKey,
+    baseUrl: baseUrl3
+  });
+  init({
     apiKey: options.apiKey,
     baseUrl: baseUrl3
   });
@@ -1208,13 +1310,15 @@ async function init3(options = {}) {
 
 // src/index.ts
 var index_default = {
-  init: init3,
+  init: init4,
   trace: trace_exports,
-  models: models_exports
+  models: models_exports,
+  prompts: prompts_exports
 };
 export {
   index_default as default,
-  init3 as init,
+  init4 as init,
   models_exports as models,
+  prompts_exports as prompts,
   trace_exports as trace
 };

package/dist/prompts-67DJ33I4.mjs

@@ -0,0 +1,14 @@
+import {
+  clearPromptContext,
+  get,
+  getAB,
+  getPromptContext,
+  init
+} from "./chunk-IGJD7GBO.mjs";
+export {
+  clearPromptContext,
+  get,
+  getAB,
+  getPromptContext,
+  init
+};

package/dist/prompts-ODF4KO2E.mjs

@@ -0,0 +1,14 @@
+import {
+  clearPromptContext,
+  get,
+  getAB,
+  getPromptContext,
+  init
+} from "./chunk-VNUUS74T.mjs";
+export {
+  clearPromptContext,
+  get,
+  getAB,
+  getPromptContext,
+  init
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fallom/trace",
-  "version": "0.1.3",
+  "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",