@fallom/trace 0.1.12 → 0.2.1

package/dist/index.mjs

@@ -1,25 +1,19 @@
 import {
   __export,
   init,
-  prompts_exports
-} from "./chunk-6MSTRIK4.mjs";
+  models_exports
+} from "./chunk-W6M2RQ3W.mjs";
 
 // src/trace.ts
 var trace_exports = {};
 __export(trace_exports, {
-  clearSession: () => clearSession,
-  getSession: () => getSession,
+  FallomSession: () => FallomSession,
   init: () => init2,
-  runWithSession: () => runWithSession,
-  setSession: () => setSession,
-  shutdown: () => shutdown,
-  span: () => span,
-  wrapAISDK: () => wrapAISDK,
-  wrapAnthropic: () => wrapAnthropic,
-  wrapGoogleAI: () => wrapGoogleAI,
-  wrapMastraAgent: () => wrapMastraAgent,
-  wrapOpenAI: () => wrapOpenAI
+  session: () => session,
+  shutdown: () => shutdown
 });
+
+// src/trace/core.ts
 import { AsyncLocalStorage } from "async_hooks";
 import { NodeSDK } from "@opentelemetry/sdk-node";
 import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-http";
@@ -621,9 +615,9 @@ var Resource = (
   })()
 );
 
-// src/trace.ts
-var sessionStorage = new AsyncLocalStorage();
-var fallbackSession = null;
+// src/trace/core.ts
+var traceContextStorage = new AsyncLocalStorage();
+var fallbackTraceContext = null;
 var apiKey = null;
 var baseUrl = "https://traces.fallom.com";
 var initialized = false;
@@ -633,28 +627,27 @@ var sdk = null;
 function log(...args) {
   if (debugMode) console.log("[Fallom]", ...args);
 }
+function getTraceContextStorage() {
+  return traceContextStorage;
+}
+function getFallbackTraceContext() {
+  return fallbackTraceContext;
+}
+function isInitialized() {
+  return initialized;
+}
+function shouldCaptureContent() {
+  return captureContent;
+}
+function isDebugMode() {
+  return debugMode;
+}
 var fallomSpanProcessor = {
-  onStart(span2, _parentContext) {
-    log("\u{1F4CD} Span started:", span2.name || "unknown");
-    const ctx = sessionStorage.getStore() || fallbackSession;
-    if (ctx) {
-      span2.setAttribute("fallom.config_key", ctx.configKey);
-      span2.setAttribute("fallom.session_id", 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");
-    }
+  onStart(span, _parentContext) {
+    log("\u{1F4CD} Span started:", span.name || "unknown");
   },
-  onEnd(span2) {
-    log("\u2705 Span ended:", span2.name, "duration:", span2.duration);
+  onEnd(span) {
+    log("\u2705 Span ended:", span.name, "duration:", span.duration);
   },
   shutdown() {
     return Promise.resolve();
@@ -663,47 +656,6 @@ var fallomSpanProcessor = {
     return Promise.resolve();
   }
 };
-async function init2(options = {}) {
-  if (initialized) return;
-  debugMode = options.debug ?? false;
-  log("\u{1F680} Initializing Fallom tracing...");
-  apiKey = options.apiKey || process.env.FALLOM_API_KEY || null;
-  baseUrl = options.baseUrl || process.env.FALLOM_TRACES_URL || process.env.FALLOM_BASE_URL || "https://traces.fallom.com";
-  const envCapture = process.env.FALLOM_CAPTURE_CONTENT?.toLowerCase();
-  if (envCapture === "false" || envCapture === "0" || envCapture === "no") {
-    captureContent = false;
-  } else {
-    captureContent = options.captureContent ?? true;
-  }
-  if (!apiKey) {
-    throw new Error(
-      "No API key provided. Set FALLOM_API_KEY environment variable or pass apiKey parameter."
-    );
-  }
-  initialized = true;
-  log("\u{1F4E1} Exporter URL:", `${baseUrl}/v1/traces`);
-  const exporter = new OTLPTraceExporter({
-    url: `${baseUrl}/v1/traces`,
-    headers: {
-      Authorization: `Bearer ${apiKey}`
-    }
-  });
-  const instrumentations = await getInstrumentations();
-  log("\u{1F527} Loaded instrumentations:", instrumentations.length);
-  sdk = new NodeSDK({
-    resource: new Resource({
-      "service.name": "fallom-traced-app"
-    }),
-    traceExporter: exporter,
-    spanProcessor: fallomSpanProcessor,
-    instrumentations
-  });
-  sdk.start();
-  log("\u2705 SDK started");
-  process.on("SIGTERM", () => {
-    sdk?.shutdown().catch(console.error);
-  });
-}
 async function getInstrumentations() {
   const instrumentations = [];
   await tryAddInstrumentation(
@@ -758,75 +710,90 @@ async function tryAddInstrumentation(instrumentations, pkg, className) {
         Object.keys(mod)
       );
     }
-  } catch (e) {
+  } catch {
     log(`   \u274C ${pkg} not installed`);
   }
 }
-function setSession(configKey, sessionId, customerId) {
-  const store = sessionStorage.getStore();
-  if (store) {
-    store.configKey = configKey;
-    store.sessionId = sessionId;
-    store.customerId = customerId;
-  }
-  fallbackSession = { configKey, sessionId, customerId };
-}
-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;
-}
-function clearSession() {
-  fallbackSession = null;
-}
-function span(data, options = {}) {
-  if (!initialized) {
-    throw new Error("Fallom not initialized. Call trace.init() first.");
+async function init2(options = {}) {
+  if (initialized) return;
+  debugMode = options.debug ?? false;
+  log("\u{1F680} Initializing Fallom tracing...");
+  apiKey = options.apiKey || process.env.FALLOM_API_KEY || null;
+  baseUrl = options.baseUrl || process.env.FALLOM_TRACES_URL || process.env.FALLOM_BASE_URL || "https://traces.fallom.com";
+  const envCapture = process.env.FALLOM_CAPTURE_CONTENT?.toLowerCase();
+  if (envCapture === "false" || envCapture === "0" || envCapture === "no") {
+    captureContent = false;
+  } else {
+    captureContent = options.captureContent ?? true;
   }
-  const ctx = sessionStorage.getStore() || fallbackSession;
-  const configKey = options.configKey || ctx?.configKey;
-  const sessionId = options.sessionId || ctx?.sessionId;
-  if (!configKey || !sessionId) {
+  if (!apiKey) {
     throw new Error(
-      "No session context. Either call setSession() first, or pass configKey and sessionId explicitly."
+      "No API key provided. Set FALLOM_API_KEY environment variable or pass apiKey parameter."
     );
   }
-  sendSpan(configKey, sessionId, data).catch(() => {
+  initialized = true;
+  log("\u{1F4E1} Exporter URL:", `${baseUrl}/v1/traces`);
+  const exporter = new OTLPTraceExporter({
+    url: `${baseUrl}/v1/traces`,
+    headers: {
+      Authorization: `Bearer ${apiKey}`
+    }
+  });
+  const instrumentations = await getInstrumentations();
+  log("\u{1F527} Loaded instrumentations:", instrumentations.length);
+  sdk = new NodeSDK({
+    resource: new Resource({
+      "service.name": "fallom-traced-app"
+    }),
+    traceExporter: exporter,
+    spanProcessor: fallomSpanProcessor,
+    instrumentations
+  });
+  sdk.start();
+  log("\u2705 SDK started");
+  process.on("SIGTERM", () => {
+    sdk?.shutdown().catch(console.error);
   });
 }
-async function sendSpan(configKey, sessionId, data) {
+async function shutdown() {
+  if (sdk) {
+    await sdk.shutdown();
+    initialized = false;
+  }
+}
+async function sendTrace(trace) {
+  const url = `${baseUrl}/v1/traces`;
+  log("\u{1F4E4} Sending trace to:", url);
+  log("   Session:", trace.session_id, "Config:", trace.config_key);
   try {
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), 5e3);
-    await fetch(`${baseUrl}/spans`, {
+    const response = await fetch(url, {
       method: "POST",
       headers: {
         Authorization: `Bearer ${apiKey}`,
         "Content-Type": "application/json"
       },
-      body: JSON.stringify({
-        config_key: configKey,
-        session_id: sessionId,
-        data
-      }),
+      body: JSON.stringify(trace),
       signal: controller.signal
     });
     clearTimeout(timeoutId);
-  } catch {
+    if (!response.ok) {
+      const text = await response.text();
+      log("\u274C Trace send failed:", response.status, text);
+    } else {
+      log("\u2705 Trace sent:", trace.name, trace.model);
+    }
+  } catch (err) {
+    log("\u274C Trace send error:", err instanceof Error ? err.message : err);
   }
 }
-async function shutdown() {
-  if (sdk) {
-    await sdk.shutdown();
-    initialized = false;
-  }
+
+// src/trace/utils.ts
+function generateHexId(length) {
+  const bytes = new Uint8Array(length / 2);
+  crypto.getRandomValues(bytes);
+  return Array.from(bytes).map((b) => b.toString(16).padStart(2, "0")).join("");
 }
 function messagesToOtelAttributes(messages, completion, model, responseId) {
   const attrs = {};
@@ -854,65 +821,28 @@ function messagesToOtelAttributes(messages, completion, model, responseId) {
   }
   return attrs;
 }
-function generateHexId(length) {
-  const bytes = new Uint8Array(length / 2);
-  crypto.getRandomValues(bytes);
-  return Array.from(bytes).map((b) => b.toString(16).padStart(2, "0")).join("");
-}
-var traceContextStorage = new AsyncLocalStorage();
-var fallbackTraceContext = null;
-async function sendTrace(trace) {
-  const url = `${baseUrl}/v1/traces`;
-  log("\u{1F4E4} Sending trace to:", url);
-  log("   Session:", trace.session_id, "Config:", trace.config_key);
-  try {
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), 5e3);
-    const response = await fetch(url, {
-      method: "POST",
-      headers: {
-        Authorization: `Bearer ${apiKey}`,
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(trace),
-      signal: controller.signal
-    });
-    clearTimeout(timeoutId);
-    if (!response.ok) {
-      const text = await response.text();
-      log("\u274C Trace send failed:", response.status, text);
-    } else {
-      log("\u2705 Trace sent:", trace.name, trace.model);
-    }
-  } catch (err) {
-    log("\u274C Trace send error:", err instanceof Error ? err.message : err);
-  }
-}
-function wrapOpenAI(client) {
+
+// src/trace/wrappers/openai.ts
+function wrapOpenAI(client, sessionCtx) {
   const originalCreate = client.chat.completions.create.bind(
     client.chat.completions
   );
+  const ctx = sessionCtx;
   client.chat.completions.create = async function(...args) {
-    const ctx = sessionStorage.getStore() || fallbackSession;
-    if (!ctx || !initialized) {
+    if (!isInitialized()) {
       return originalCreate(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
-      promptCtx = getPromptContext();
-    } catch {
-    }
-    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
     const traceId = traceCtx?.traceId || generateHexId(32);
     const spanId = generateHexId(16);
     const parentSpanId = traceCtx?.parentSpanId;
     const params = args[0] || {};
     const startTime = Date.now();
+    const captureContent2 = shouldCaptureContent();
     try {
       const response = await originalCreate(...args);
       const endTime = Date.now();
-      const attributes = captureContent ? messagesToOtelAttributes(
+      const attributes = captureContent2 ? messagesToOtelAttributes(
         params?.messages,
         response?.choices?.[0]?.message,
         response?.model || params?.model,
@@ -941,17 +871,13 @@ function wrapOpenAI(client) {
         prompt_tokens: response?.usage?.prompt_tokens,
         completion_tokens: response?.usage?.completion_tokens,
         total_tokens: response?.usage?.total_tokens,
-        attributes: Object.keys(attributes).length > 0 ? attributes : void 0,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        attributes: Object.keys(attributes).length > 0 ? attributes : void 0
       }).catch(() => {
       });
       return response;
     } catch (error) {
       const endTime = Date.now();
-      const attributes = captureContent ? messagesToOtelAttributes(
+      const attributes = captureContent2 ? messagesToOtelAttributes(
         params?.messages,
         void 0,
         params?.model,
@@ -975,11 +901,7 @@ function wrapOpenAI(client) {
         duration_ms: endTime - startTime,
         status: "ERROR",
         error_message: error?.message,
-        attributes,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        attributes
       }).catch(() => {
       });
       throw error;
@@ -987,29 +909,26 @@ function wrapOpenAI(client) {
   };
   return client;
 }
-function wrapAnthropic(client) {
+
+// src/trace/wrappers/anthropic.ts
+function wrapAnthropic(client, sessionCtx) {
   const originalCreate = client.messages.create.bind(client.messages);
+  const ctx = sessionCtx;
   client.messages.create = async function(...args) {
-    const ctx = sessionStorage.getStore() || fallbackSession;
-    if (!ctx || !initialized) {
+    if (!isInitialized()) {
       return originalCreate(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
-      promptCtx = getPromptContext();
-    } catch {
-    }
-    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
     const traceId = traceCtx?.traceId || generateHexId(32);
     const spanId = generateHexId(16);
     const parentSpanId = traceCtx?.parentSpanId;
     const params = args[0] || {};
     const startTime = Date.now();
+    const captureContent2 = shouldCaptureContent();
     try {
       const response = await originalCreate(...args);
       const endTime = Date.now();
-      const attributes = captureContent ? messagesToOtelAttributes(
+      const attributes = captureContent2 ? messagesToOtelAttributes(
         params?.messages,
         { role: "assistant", content: response?.content?.[0]?.text || "" },
         response?.model || params?.model,
@@ -1041,17 +960,13 @@ function wrapAnthropic(client) {
         prompt_tokens: response?.usage?.input_tokens,
         completion_tokens: response?.usage?.output_tokens,
         total_tokens: (response?.usage?.input_tokens || 0) + (response?.usage?.output_tokens || 0),
-        attributes: Object.keys(attributes).length > 0 ? attributes : void 0,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        attributes: Object.keys(attributes).length > 0 ? attributes : void 0
       }).catch(() => {
       });
       return response;
     } catch (error) {
       const endTime = Date.now();
-      const attributes = captureContent ? messagesToOtelAttributes(
+      const attributes = captureContent2 ? messagesToOtelAttributes(
         params?.messages,
         void 0,
         params?.model,
@@ -1078,11 +993,7 @@ function wrapAnthropic(client) {
         duration_ms: endTime - startTime,
         status: "ERROR",
         error_message: error?.message,
-        attributes,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        attributes
       }).catch(() => {
       });
       throw error;
@@ -1090,24 +1001,21 @@ function wrapAnthropic(client) {
   };
   return client;
 }
-function wrapGoogleAI(model) {
+
+// src/trace/wrappers/google-ai.ts
+function wrapGoogleAI(model, sessionCtx) {
   const originalGenerate = model.generateContent.bind(model);
+  const ctx = sessionCtx;
   model.generateContent = async function(...args) {
-    const ctx = sessionStorage.getStore() || fallbackSession;
-    if (!ctx || !initialized) {
+    if (!isInitialized()) {
       return originalGenerate(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
-      promptCtx = getPromptContext();
-    } catch {
-    }
-    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
     const traceId = traceCtx?.traceId || generateHexId(32);
     const spanId = generateHexId(16);
     const parentSpanId = traceCtx?.parentSpanId;
     const startTime = Date.now();
+    const captureContent2 = shouldCaptureContent();
     try {
       const response = await originalGenerate(...args);
       const endTime = Date.now();
@@ -1115,7 +1023,7 @@ function wrapGoogleAI(model) {
       const usage = result?.usageMetadata;
       const modelName = model?.model || "gemini";
       const attributes = {};
-      if (captureContent) {
+      if (captureContent2) {
         attributes["gen_ai.request.model"] = modelName;
         attributes["gen_ai.response.model"] = modelName;
         const input = args[0];
@@ -1158,11 +1066,7 @@ function wrapGoogleAI(model) {
         prompt_tokens: usage?.promptTokenCount,
         completion_tokens: usage?.candidatesTokenCount,
         total_tokens: usage?.totalTokenCount,
-        attributes: Object.keys(attributes).length > 0 ? attributes : void 0,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        attributes: Object.keys(attributes).length > 0 ? attributes : void 0
       }).catch(() => {
       });
       return response;
@@ -1170,7 +1074,7 @@ function wrapGoogleAI(model) {
       const endTime = Date.now();
       const modelName = model?.model || "gemini";
       const attributes = {};
-      if (captureContent) {
+      if (captureContent2) {
         attributes["gen_ai.request.model"] = modelName;
         attributes["error.message"] = error?.message;
         const input = args[0];
@@ -1194,11 +1098,7 @@ function wrapGoogleAI(model) {
         duration_ms: endTime - startTime,
         status: "ERROR",
         error_message: error?.message,
-        attributes: captureContent ? attributes : void 0,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        attributes: captureContent2 ? attributes : void 0
       }).catch(() => {
       });
       throw error;
@@ -1206,7 +1106,8 @@ function wrapGoogleAI(model) {
   };
   return model;
 }
-var aiSdkDebug = false;
+
+// src/trace/wrappers/vercel-ai/utils.ts
 function extractUsageFromResult(result, directUsage) {
   let usage = directUsage ?? result?.usage;
   const isValidNumber = (v) => v !== null && v !== void 0 && !Number.isNaN(v);
@@ -1234,38 +1135,25 @@ function extractUsageFromResult(result, directUsage) {
   }
   return { promptTokens, completionTokens, totalTokens, cost };
 }
-function wrapAISDK(ai, options) {
-  const aiModule = ai;
-  aiSdkDebug = options?.debug ?? false;
-  return {
-    generateText: createGenerateTextWrapper(aiModule),
-    streamText: createStreamTextWrapper(aiModule),
-    generateObject: aiModule.generateObject ? createGenerateObjectWrapper(aiModule) : void 0,
-    streamObject: aiModule.streamObject ? createStreamObjectWrapper(aiModule) : void 0
-  };
-}
-function createGenerateTextWrapper(aiModule) {
+
+// src/trace/wrappers/vercel-ai/generate-text.ts
+function createGenerateTextWrapper(aiModule, sessionCtx, debug = false) {
+  const ctx = sessionCtx;
   return async (...args) => {
-    const ctx = sessionStorage.getStore() || fallbackSession;
-    if (!ctx || !initialized) {
+    if (!isInitialized()) {
       return aiModule.generateText(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
-      promptCtx = getPromptContext();
-    } catch {
-    }
-    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
     const traceId = traceCtx?.traceId || generateHexId(32);
     const spanId = generateHexId(16);
     const parentSpanId = traceCtx?.parentSpanId;
     const params = args[0] || {};
     const startTime = Date.now();
+    const captureContent2 = shouldCaptureContent();
     try {
       const result = await aiModule.generateText(...args);
       const endTime = Date.now();
-      if (aiSdkDebug) {
+      if (debug || isDebugMode()) {
         console.log(
           "\n\u{1F50D} [Fallom Debug] generateText result keys:",
           Object.keys(result || {})
@@ -1274,14 +1162,6 @@ function createGenerateTextWrapper(aiModule) {
           "\u{1F50D} [Fallom Debug] result.usage:",
           JSON.stringify(result?.usage, null, 2)
         );
-        console.log(
-          "\u{1F50D} [Fallom Debug] result.response keys:",
-          Object.keys(result?.response || {})
-        );
-        console.log(
-          "\u{1F50D} [Fallom Debug] result.response.usage:",
-          JSON.stringify(result?.response?.usage, null, 2)
-        );
         console.log(
           "\u{1F50D} [Fallom Debug] result.experimental_providerMetadata:",
           JSON.stringify(result?.experimental_providerMetadata, null, 2)
@@ -1289,7 +1169,7 @@ function createGenerateTextWrapper(aiModule) {
       }
       const modelId = result?.response?.modelId || params?.model?.modelId || String(params?.model || "unknown");
       const attributes = {};
-      if (captureContent) {
+      if (captureContent2) {
         attributes["gen_ai.request.model"] = modelId;
         attributes["gen_ai.response.model"] = modelId;
         if (params?.prompt) {
@@ -1339,11 +1219,7 @@ function createGenerateTextWrapper(aiModule) {
         prompt_tokens: usage.promptTokens,
         completion_tokens: usage.completionTokens,
         total_tokens: usage.totalTokens,
-        attributes: captureContent ? attributes : void 0,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        attributes: captureContent2 ? attributes : void 0
       }).catch(() => {
       });
       return result;
@@ -1364,52 +1240,44 @@ function createGenerateTextWrapper(aiModule) {
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "ERROR",
-        error_message: error?.message,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        error_message: error?.message
       }).catch(() => {
       });
       throw error;
     }
   };
 }
-function createStreamTextWrapper(aiModule) {
+
+// src/trace/wrappers/vercel-ai/stream-text.ts
+function log2(...args) {
+  if (isDebugMode()) console.log("[Fallom]", ...args);
+}
+function createStreamTextWrapper(aiModule, sessionCtx, debug = false) {
+  const ctx = sessionCtx;
   return async (...args) => {
-    const ctx = sessionStorage.getStore() || fallbackSession;
     const params = args[0] || {};
     const startTime = Date.now();
+    const captureContent2 = shouldCaptureContent();
     const result = await aiModule.streamText(...args);
-    if (!ctx || !initialized) {
+    if (!isInitialized()) {
       return result;
     }
-    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
     const traceId = traceCtx?.traceId || generateHexId(32);
     const spanId = generateHexId(16);
     const parentSpanId = traceCtx?.parentSpanId;
     let firstTokenTime = null;
     const modelId = params?.model?.modelId || String(params?.model || "unknown");
-    let promptCtx = null;
-    try {
-      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
-      promptCtx = getPromptContext();
-    } catch {
-    }
     if (result?.usage) {
       result.usage.then(async (rawUsage) => {
         const endTime = Date.now();
-        if (aiSdkDebug) {
+        if (debug || isDebugMode()) {
           console.log(
             "\n\u{1F50D} [Fallom Debug] streamText usage:",
             JSON.stringify(rawUsage, null, 2)
           );
-          console.log(
-            "\u{1F50D} [Fallom Debug] streamText result keys:",
-            Object.keys(result || {})
-          );
         }
-        log("\u{1F4CA} streamText usage:", JSON.stringify(rawUsage, null, 2));
+        log2("\u{1F4CA} streamText usage:", JSON.stringify(rawUsage, null, 2));
         let providerMetadata = result?.experimental_providerMetadata;
         if (providerMetadata && typeof providerMetadata.then === "function") {
           try {
@@ -1423,7 +1291,7 @@ function createStreamTextWrapper(aiModule) {
           rawUsage
         );
         const attributes = {};
-        if (captureContent) {
+        if (captureContent2) {
           attributes["gen_ai.request.model"] = modelId;
           if (params?.prompt) {
             attributes["gen_ai.prompt.0.role"] = "user";
@@ -1457,17 +1325,13 @@ function createStreamTextWrapper(aiModule) {
           completion_tokens: usage.completionTokens,
           total_tokens: usage.totalTokens,
           time_to_first_token_ms: firstTokenTime ? firstTokenTime - startTime : void 0,
-          attributes: captureContent ? attributes : void 0,
-          prompt_key: promptCtx?.promptKey,
-          prompt_version: promptCtx?.promptVersion,
-          prompt_ab_test_key: promptCtx?.abTestKey,
-          prompt_variant_index: promptCtx?.variantIndex
+          attributes: captureContent2 ? attributes : void 0
         };
         sendTrace(tracePayload).catch(() => {
         });
       }).catch((error) => {
         const endTime = Date.now();
-        log("\u274C streamText error:", error?.message);
+        log2("\u274C streamText error:", error?.message);
         sendTrace({
           config_key: ctx.configKey,
           session_id: ctx.sessionId,
@@ -1482,11 +1346,7 @@ function createStreamTextWrapper(aiModule) {
           end_time: new Date(endTime).toISOString(),
           duration_ms: endTime - startTime,
           status: "ERROR",
-          error_message: error?.message,
-          prompt_key: promptCtx?.promptKey,
-          prompt_version: promptCtx?.promptVersion,
-          prompt_ab_test_key: promptCtx?.abTestKey,
-          prompt_variant_index: promptCtx?.variantIndex
+          error_message: error?.message
         }).catch(() => {
         });
       });
@@ -1497,7 +1357,7 @@ function createStreamTextWrapper(aiModule) {
         for await (const chunk of originalTextStream) {
           if (!firstTokenTime) {
             firstTokenTime = Date.now();
-            log("\u23F1\uFE0F Time to first token:", firstTokenTime - startTime, "ms");
+            log2("\u23F1\uFE0F Time to first token:", firstTokenTime - startTime, "ms");
           }
           yield chunk;
         }
@@ -1514,28 +1374,25 @@ function createStreamTextWrapper(aiModule) {
     return result;
   };
 }
-function createGenerateObjectWrapper(aiModule) {
+
+// src/trace/wrappers/vercel-ai/generate-object.ts
+function createGenerateObjectWrapper(aiModule, sessionCtx, debug = false) {
+  const ctx = sessionCtx;
   return async (...args) => {
-    const ctx = sessionStorage.getStore() || fallbackSession;
-    if (!ctx || !initialized) {
+    if (!isInitialized()) {
       return aiModule.generateObject(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
-      promptCtx = getPromptContext();
-    } catch {
-    }
-    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
     const traceId = traceCtx?.traceId || generateHexId(32);
     const spanId = generateHexId(16);
     const parentSpanId = traceCtx?.parentSpanId;
     const params = args[0] || {};
     const startTime = Date.now();
+    const captureContent2 = shouldCaptureContent();
     try {
       const result = await aiModule.generateObject(...args);
       const endTime = Date.now();
-      if (aiSdkDebug) {
+      if (debug || isDebugMode()) {
         console.log(
           "\n\u{1F50D} [Fallom Debug] generateObject result keys:",
           Object.keys(result || {})
@@ -1544,18 +1401,10 @@ function createGenerateObjectWrapper(aiModule) {
           "\u{1F50D} [Fallom Debug] result.usage:",
           JSON.stringify(result?.usage, null, 2)
         );
-        console.log(
-          "\u{1F50D} [Fallom Debug] result.response keys:",
-          Object.keys(result?.response || {})
-        );
-        console.log(
-          "\u{1F50D} [Fallom Debug] result.response.usage:",
-          JSON.stringify(result?.response?.usage, null, 2)
-        );
       }
       const modelId = result?.response?.modelId || params?.model?.modelId || String(params?.model || "unknown");
       const attributes = {};
-      if (captureContent) {
+      if (captureContent2) {
         attributes["gen_ai.request.model"] = modelId;
         attributes["gen_ai.response.model"] = modelId;
         if (result?.object) {
@@ -1594,11 +1443,7 @@ function createGenerateObjectWrapper(aiModule) {
         prompt_tokens: usage.promptTokens,
         completion_tokens: usage.completionTokens,
         total_tokens: usage.totalTokens,
-        attributes: captureContent ? attributes : void 0,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        attributes: captureContent2 ? attributes : void 0
       }).catch(() => {
       });
       return result;
@@ -1619,53 +1464,45 @@ function createGenerateObjectWrapper(aiModule) {
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "ERROR",
-        error_message: error?.message,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        error_message: error?.message
       }).catch(() => {
       });
       throw error;
     }
   };
 }
-function createStreamObjectWrapper(aiModule) {
+
+// src/trace/wrappers/vercel-ai/stream-object.ts
+function log3(...args) {
+  if (isDebugMode()) console.log("[Fallom]", ...args);
+}
+function createStreamObjectWrapper(aiModule, sessionCtx, debug = false) {
+  const ctx = sessionCtx;
   return async (...args) => {
-    const ctx = sessionStorage.getStore() || fallbackSession;
     const params = args[0] || {};
     const startTime = Date.now();
+    const captureContent2 = shouldCaptureContent();
     const result = await aiModule.streamObject(...args);
-    log("\u{1F50D} streamObject result keys:", Object.keys(result || {}));
-    if (!ctx || !initialized) {
+    log3("\u{1F50D} streamObject result keys:", Object.keys(result || {}));
+    if (!isInitialized()) {
       return result;
     }
-    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
     const traceId = traceCtx?.traceId || generateHexId(32);
     const spanId = generateHexId(16);
     const parentSpanId = traceCtx?.parentSpanId;
     let firstTokenTime = null;
     const modelId = params?.model?.modelId || String(params?.model || "unknown");
-    let promptCtx = null;
-    try {
-      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
-      promptCtx = getPromptContext();
-    } catch {
-    }
     if (result?.usage) {
       result.usage.then(async (rawUsage) => {
         const endTime = Date.now();
-        if (aiSdkDebug) {
+        if (debug || isDebugMode()) {
           console.log(
             "\n\u{1F50D} [Fallom Debug] streamObject usage:",
             JSON.stringify(rawUsage, null, 2)
           );
-          console.log(
-            "\u{1F50D} [Fallom Debug] streamObject result keys:",
-            Object.keys(result || {})
-          );
         }
-        log("\u{1F4CA} streamObject usage:", JSON.stringify(rawUsage, null, 2));
+        log3("\u{1F4CA} streamObject usage:", JSON.stringify(rawUsage, null, 2));
         let providerMetadata = result?.experimental_providerMetadata;
         if (providerMetadata && typeof providerMetadata.then === "function") {
           try {
@@ -1679,7 +1516,7 @@ function createStreamObjectWrapper(aiModule) {
           rawUsage
         );
         const attributes = {};
-        if (captureContent) {
+        if (captureContent2) {
           attributes["gen_ai.request.model"] = modelId;
         }
         if (firstTokenTime) {
@@ -1708,11 +1545,7 @@ function createStreamObjectWrapper(aiModule) {
           prompt_tokens: usage.promptTokens,
           completion_tokens: usage.completionTokens,
           total_tokens: usage.totalTokens,
-          attributes: captureContent ? attributes : void 0,
-          prompt_key: promptCtx?.promptKey,
-          prompt_version: promptCtx?.promptVersion,
-          prompt_ab_test_key: promptCtx?.abTestKey,
-          prompt_variant_index: promptCtx?.variantIndex
+          attributes: captureContent2 ? attributes : void 0
         }).catch(() => {
         });
       }).catch((error) => {
@@ -1731,11 +1564,7 @@ function createStreamObjectWrapper(aiModule) {
           end_time: new Date(endTime).toISOString(),
           duration_ms: endTime - startTime,
           status: "ERROR",
-          error_message: error?.message,
-          prompt_key: promptCtx?.promptKey,
-          prompt_version: promptCtx?.promptVersion,
-          prompt_ab_test_key: promptCtx?.abTestKey,
-          prompt_variant_index: promptCtx?.variantIndex
+          error_message: error?.message
         }).catch(() => {
         });
       });
@@ -1746,7 +1575,7 @@ function createStreamObjectWrapper(aiModule) {
         for await (const chunk of originalStream) {
           if (!firstTokenTime) {
             firstTokenTime = Date.now();
-            log("\u23F1\uFE0F Time to first token:", firstTokenTime - startTime, "ms");
+            log3("\u23F1\uFE0F Time to first token:", firstTokenTime - startTime, "ms");
           }
           yield chunk;
         }
@@ -1763,20 +1592,27 @@ function createStreamObjectWrapper(aiModule) {
     return result;
   };
 }
-function wrapMastraAgent(agent) {
+
+// src/trace/wrappers/vercel-ai/index.ts
+function wrapAISDK(ai, sessionCtx, options) {
+  const debug = options?.debug ?? false;
+  return {
+    generateText: createGenerateTextWrapper(ai, sessionCtx, debug),
+    streamText: createStreamTextWrapper(ai, sessionCtx, debug),
+    generateObject: ai.generateObject ? createGenerateObjectWrapper(ai, sessionCtx, debug) : void 0,
+    streamObject: ai.streamObject ? createStreamObjectWrapper(ai, sessionCtx, debug) : void 0
+  };
+}
+
+// src/trace/wrappers/mastra.ts
+function wrapMastraAgent(agent, sessionCtx) {
   const originalGenerate = agent.generate.bind(agent);
   const agentName = agent.name || "MastraAgent";
+  const ctx = sessionCtx;
   agent.generate = async function(...args) {
-    const ctx = sessionStorage.getStore() || fallbackSession;
-    if (!ctx || !initialized) {
+    if (!isInitialized()) {
       return originalGenerate(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
-      promptCtx = getPromptContext();
-    } catch {
-    }
     const traceId = generateHexId(32);
     const spanId = generateHexId(16);
     const startTime = Date.now();
@@ -1848,11 +1684,7 @@ function wrapMastraAgent(agent) {
         prompt_tokens: result?.usage?.promptTokens,
         completion_tokens: result?.usage?.completionTokens,
         total_tokens: result?.usage?.totalTokens,
-        attributes,
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        attributes
       };
       sendTrace(traceData).catch(() => {
       });
@@ -1871,11 +1703,7 @@ function wrapMastraAgent(agent) {
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "ERROR",
-        error_message: error instanceof Error ? error.message : String(error),
-        prompt_key: promptCtx?.promptKey,
-        prompt_version: promptCtx?.promptVersion,
-        prompt_ab_test_key: promptCtx?.abTestKey,
-        prompt_variant_index: promptCtx?.variantIndex
+        error_message: error instanceof Error ? error.message : String(error)
       };
       sendTrace(traceData).catch(() => {
       });
@@ -1885,38 +1713,215 @@ function wrapMastraAgent(agent) {
   return agent;
 }
 
-// src/models.ts
-var models_exports = {};
-__export(models_exports, {
+// src/trace/session.ts
+var FallomSession = class {
+  constructor(options) {
+    this.ctx = {
+      configKey: options.configKey,
+      sessionId: options.sessionId,
+      customerId: options.customerId
+    };
+  }
+  /** Get the session context. */
+  getContext() {
+    return { ...this.ctx };
+  }
+  /**
+   * Get model assignment for this session (A/B testing).
+   */
+  async getModel(configKeyOrOptions, options) {
+    let configKey;
+    let opts;
+    if (typeof configKeyOrOptions === "string") {
+      configKey = configKeyOrOptions;
+      opts = options || {};
+    } else {
+      configKey = this.ctx.configKey;
+      opts = configKeyOrOptions || {};
+    }
+    const { get: get2 } = await import("./models-JKMOBZUO.mjs");
+    return get2(configKey, this.ctx.sessionId, opts);
+  }
+  /**
+   * Wrap a Vercel AI SDK model to trace all calls (PostHog style).
+   * Returns the same model type with tracing injected.
+   */
+  traceModel(model) {
+    const ctx = this.ctx;
+    const tracedModel = Object.create(model);
+    const m = model;
+    if (m.doGenerate) {
+      const originalDoGenerate = m.doGenerate.bind(model);
+      tracedModel.doGenerate = async function(...args) {
+        if (!isInitialized()) return originalDoGenerate(...args);
+        const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
+        const traceId = traceCtx?.traceId || generateHexId(32);
+        const spanId = generateHexId(16);
+        const startTime = Date.now();
+        try {
+          const result = await originalDoGenerate(...args);
+          const endTime = Date.now();
+          const modelId = model.modelId || "unknown";
+          const usage = result?.usage || result?.rawResponse?.usage;
+          sendTrace({
+            config_key: ctx.configKey,
+            session_id: ctx.sessionId,
+            customer_id: ctx.customerId,
+            trace_id: traceId,
+            span_id: spanId,
+            parent_span_id: traceCtx?.parentSpanId,
+            name: "generateText",
+            kind: "llm",
+            model: modelId,
+            start_time: new Date(startTime).toISOString(),
+            end_time: new Date(endTime).toISOString(),
+            duration_ms: endTime - startTime,
+            status: "OK",
+            prompt_tokens: usage?.promptTokens,
+            completion_tokens: usage?.completionTokens,
+            total_tokens: usage?.totalTokens,
+            attributes: shouldCaptureContent() && usage ? { "fallom.raw.usage": JSON.stringify(usage) } : void 0
+          }).catch(() => {
+          });
+          return result;
+        } catch (error) {
+          const endTime = Date.now();
+          sendTrace({
+            config_key: ctx.configKey,
+            session_id: ctx.sessionId,
+            customer_id: ctx.customerId,
+            trace_id: traceId,
+            span_id: spanId,
+            parent_span_id: traceCtx?.parentSpanId,
+            name: "generateText",
+            kind: "llm",
+            model: model.modelId || "unknown",
+            start_time: new Date(startTime).toISOString(),
+            end_time: new Date(endTime).toISOString(),
+            duration_ms: endTime - startTime,
+            status: "ERROR",
+            error_message: error instanceof Error ? error.message : String(error)
+          }).catch(() => {
+          });
+          throw error;
+        }
+      };
+    }
+    if (m.doStream) {
+      const originalDoStream = m.doStream.bind(model);
+      tracedModel.doStream = async function(...args) {
+        if (!isInitialized()) return originalDoStream(...args);
+        const traceCtx = getTraceContextStorage().getStore() || getFallbackTraceContext();
+        const traceId = traceCtx?.traceId || generateHexId(32);
+        const spanId = generateHexId(16);
+        const startTime = Date.now();
+        const modelId = model.modelId || "unknown";
+        try {
+          const result = await originalDoStream(...args);
+          sendTrace({
+            config_key: ctx.configKey,
+            session_id: ctx.sessionId,
+            customer_id: ctx.customerId,
+            trace_id: traceId,
+            span_id: spanId,
+            parent_span_id: traceCtx?.parentSpanId,
+            name: "streamText",
+            kind: "llm",
+            model: modelId,
+            start_time: new Date(startTime).toISOString(),
+            end_time: new Date(Date.now()).toISOString(),
+            duration_ms: Date.now() - startTime,
+            status: "OK",
+            is_streaming: true
+          }).catch(() => {
+          });
+          return result;
+        } catch (error) {
+          sendTrace({
+            config_key: ctx.configKey,
+            session_id: ctx.sessionId,
+            customer_id: ctx.customerId,
+            trace_id: traceId,
+            span_id: spanId,
+            parent_span_id: traceCtx?.parentSpanId,
+            name: "streamText",
+            kind: "llm",
+            model: modelId,
+            start_time: new Date(startTime).toISOString(),
+            end_time: new Date(Date.now()).toISOString(),
+            duration_ms: Date.now() - startTime,
+            status: "ERROR",
+            error_message: error instanceof Error ? error.message : String(error),
+            is_streaming: true
+          }).catch(() => {
+          });
+          throw error;
+        }
+      };
+    }
+    return tracedModel;
+  }
+  /** Wrap OpenAI client. Delegates to shared wrapper. */
+  wrapOpenAI(client) {
+    return wrapOpenAI(client, this.ctx);
+  }
+  /** Wrap Anthropic client. Delegates to shared wrapper. */
+  wrapAnthropic(client) {
+    return wrapAnthropic(client, this.ctx);
+  }
+  /** Wrap Google AI model. Delegates to shared wrapper. */
+  wrapGoogleAI(model) {
+    return wrapGoogleAI(model, this.ctx);
+  }
+  /** Wrap Vercel AI SDK. Delegates to shared wrapper. */
+  wrapAISDK(ai, options) {
+    return wrapAISDK(ai, this.ctx, options);
+  }
+  /** Wrap Mastra agent. Delegates to shared wrapper. */
+  wrapMastraAgent(agent) {
+    return wrapMastraAgent(agent, this.ctx);
+  }
+};
+function session(options) {
+  return new FallomSession(options);
+}
+
+// src/prompts.ts
+var prompts_exports = {};
+__export(prompts_exports, {
+  clearPromptContext: () => clearPromptContext,
   get: () => get,
+  getAB: () => getAB,
+  getPromptContext: () => getPromptContext,
   init: () => init3
 });
 import { createHash } from "crypto";
 var apiKey2 = null;
-var baseUrl2 = "https://configs.fallom.com";
+var baseUrl2 = "https://prompts.fallom.com";
 var initialized2 = false;
 var syncInterval = null;
 var debugMode2 = false;
-var configCache = /* @__PURE__ */ new Map();
+var promptCache = /* @__PURE__ */ new Map();
+var promptABCache = /* @__PURE__ */ new Map();
+var promptContext = null;
 var SYNC_TIMEOUT = 2e3;
-var RECORD_TIMEOUT = 1e3;
-function log2(msg) {
+function log4(msg) {
   if (debugMode2) {
-    console.log(`[Fallom] ${msg}`);
+    console.log(`[Fallom Prompts] ${msg}`);
   }
 }
 function init3(options = {}) {
   apiKey2 = options.apiKey || process.env.FALLOM_API_KEY || null;
-  baseUrl2 = options.baseUrl || process.env.FALLOM_CONFIGS_URL || process.env.FALLOM_BASE_URL || "https://configs.fallom.com";
+  baseUrl2 = options.baseUrl || process.env.FALLOM_PROMPTS_URL || process.env.FALLOM_BASE_URL || "https://prompts.fallom.com";
   initialized2 = true;
   if (!apiKey2) {
     return;
   }
-  fetchConfigs().catch(() => {
+  fetchAll().catch(() => {
   });
   if (!syncInterval) {
     syncInterval = setInterval(() => {
-      fetchConfigs().catch(() => {
+      fetchAll().catch(() => {
       });
     }, 3e4);
     syncInterval.unref();
@@ -1930,202 +1935,195 @@ function ensureInit() {
     }
   }
 }
-async function fetchConfigs(timeout = SYNC_TIMEOUT) {
-  if (!apiKey2) {
-    log2("_fetchConfigs: No API key, skipping");
-    return;
-  }
+async function fetchAll() {
+  await Promise.all([fetchPrompts(), fetchPromptABTests()]);
+}
+async function fetchPrompts(timeout = SYNC_TIMEOUT) {
+  if (!apiKey2) return;
   try {
-    log2(`Fetching configs from ${baseUrl2}/configs`);
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), timeout);
-    const resp = await fetch(`${baseUrl2}/configs`, {
+    const resp = await fetch(`${baseUrl2}/prompts`, {
       headers: { Authorization: `Bearer ${apiKey2}` },
       signal: controller.signal
     });
     clearTimeout(timeoutId);
-    log2(`Response status: ${resp.status}`);
     if (resp.ok) {
       const data = await resp.json();
-      const configs = data.configs || [];
-      log2(`Got ${configs.length} configs: ${configs.map((c) => c.key)}`);
-      for (const c of configs) {
-        const key = c.key;
-        const version = c.version || 1;
-        log2(`Config '${key}' v${version}: ${JSON.stringify(c.variants)}`);
-        if (!configCache.has(key)) {
-          configCache.set(key, { versions: /* @__PURE__ */ new Map(), latest: null });
+      for (const p of data.prompts || []) {
+        if (!promptCache.has(p.key)) {
+          promptCache.set(p.key, { versions: /* @__PURE__ */ new Map(), current: null });
         }
-        const cached = configCache.get(key);
-        cached.versions.set(version, c);
-        cached.latest = version;
+        const cached = promptCache.get(p.key);
+        cached.versions.set(p.version, {
+          systemPrompt: p.system_prompt,
+          userTemplate: p.user_template
+        });
+        cached.current = p.version;
       }
-    } else {
-      log2(`Fetch failed: ${resp.statusText}`);
     }
-  } catch (e) {
-    log2(`Fetch exception: ${e}`);
+  } catch {
   }
 }
-async function fetchSpecificVersion(configKey, version, timeout = SYNC_TIMEOUT) {
-  if (!apiKey2) return null;
+async function fetchPromptABTests(timeout = SYNC_TIMEOUT) {
+  if (!apiKey2) return;
   try {
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), timeout);
-    const resp = await fetch(
-      `${baseUrl2}/configs/${configKey}/version/${version}`,
-      {
-        headers: { Authorization: `Bearer ${apiKey2}` },
-        signal: controller.signal
-      }
-    );
+    const resp = await fetch(`${baseUrl2}/prompt-ab-tests`, {
+      headers: { Authorization: `Bearer ${apiKey2}` },
+      signal: controller.signal
+    });
     clearTimeout(timeoutId);
     if (resp.ok) {
-      const config = await resp.json();
-      if (!configCache.has(configKey)) {
-        configCache.set(configKey, { versions: /* @__PURE__ */ new Map(), latest: null });
+      const data = await resp.json();
+      for (const t of data.prompt_ab_tests || []) {
+        if (!promptABCache.has(t.key)) {
+          promptABCache.set(t.key, { versions: /* @__PURE__ */ new Map(), current: null });
+        }
+        const cached = promptABCache.get(t.key);
+        cached.versions.set(t.version, { variants: t.variants });
+        cached.current = t.version;
       }
-      configCache.get(configKey).versions.set(version, config);
-      return config;
     }
   } catch {
   }
-  return null;
 }
-async function get(configKey, sessionId, options = {}) {
-  const { version, fallback, debug = false } = options;
+function replaceVariables(template, variables) {
+  if (!variables) return template;
+  return template.replace(/\{\{(\s*\w+\s*)\}\}/g, (match, varName) => {
+    const key = varName.trim();
+    return key in variables ? String(variables[key]) : match;
+  });
+}
+function setPromptContext(ctx) {
+  promptContext = ctx;
+}
+function getPromptContext() {
+  const ctx = promptContext;
+  promptContext = null;
+  return ctx;
+}
+async function get(promptKey, options = {}) {
+  const { variables, version, debug = false } = options;
   debugMode2 = debug;
   ensureInit();
-  log2(
-    `get() called: configKey=${configKey}, sessionId=${sessionId}, fallback=${fallback}`
-  );
-  try {
-    let configData = configCache.get(configKey);
-    log2(
-      `Cache lookup for '${configKey}': ${configData ? "found" : "not found"}`
+  log4(`get() called: promptKey=${promptKey}`);
+  let promptData = promptCache.get(promptKey);
+  if (!promptData) {
+    log4("Not in cache, fetching...");
+    await fetchPrompts(SYNC_TIMEOUT);
+    promptData = promptCache.get(promptKey);
+  }
+  if (!promptData) {
+    throw new Error(
+      `Prompt '${promptKey}' not found. Check that it exists in your Fallom dashboard.`
     );
-    if (!configData) {
-      log2("Not in cache, fetching...");
-      await fetchConfigs(SYNC_TIMEOUT);
-      configData = configCache.get(configKey);
-      log2(
-        `After fetch, cache lookup: ${configData ? "found" : "still not found"}`
-      );
-    }
-    if (!configData) {
-      log2(`Config not found, using fallback: ${fallback}`);
-      if (fallback) {
-        console.warn(
-          `[Fallom WARNING] Config '${configKey}' not found, using fallback model: ${fallback}`
-        );
-        return returnWithTrace(configKey, sessionId, fallback, 0);
-      }
-      throw new Error(
-        `Config '${configKey}' not found. Check that it exists in your Fallom dashboard.`
-      );
-    }
-    let config;
-    let targetVersion;
-    if (version !== void 0) {
-      config = configData.versions.get(version);
-      if (!config) {
-        config = await fetchSpecificVersion(configKey, version, SYNC_TIMEOUT) || void 0;
-      }
-      if (!config) {
-        if (fallback) {
-          console.warn(
-            `[Fallom WARNING] Config '${configKey}' version ${version} not found, using fallback: ${fallback}`
-          );
-          return returnWithTrace(configKey, sessionId, fallback, 0);
-        }
-        throw new Error(`Config '${configKey}' version ${version} not found.`);
-      }
-      targetVersion = version;
-    } else {
-      targetVersion = configData.latest;
-      config = configData.versions.get(targetVersion);
-      if (!config) {
-        if (fallback) {
-          console.warn(
-            `[Fallom WARNING] Config '${configKey}' has no cached version, using fallback: ${fallback}`
-          );
-          return returnWithTrace(configKey, sessionId, fallback, 0);
-        }
-        throw new Error(`Config '${configKey}' has no cached version.`);
-      }
-    }
-    const variantsRaw = config.variants;
-    const configVersion = config.version || targetVersion;
-    const variants = Array.isArray(variantsRaw) ? variantsRaw : Object.values(variantsRaw);
-    log2(
-      `Config found! Version: ${configVersion}, Variants: ${JSON.stringify(
-        variants
-      )}`
+  }
+  const targetVersion = version ?? promptData.current;
+  const content = promptData.versions.get(targetVersion);
+  if (!content) {
+    throw new Error(
+      `Prompt '${promptKey}' version ${targetVersion} not found.`
     );
-    const hashBytes = createHash("md5").update(sessionId).digest();
-    const hashVal = hashBytes.readUInt32BE(0) % 1e6;
-    log2(`Session hash: ${hashVal} (out of 1,000,000)`);
-    let cumulative = 0;
-    let assignedModel = variants[variants.length - 1].model;
-    for (const v of variants) {
-      const oldCumulative = cumulative;
-      cumulative += v.weight * 1e4;
-      log2(
-        `Variant ${v.model}: weight=${v.weight}%, range=${oldCumulative}-${cumulative}, hash=${hashVal}, match=${hashVal < cumulative}`
-      );
-      if (hashVal < cumulative) {
-        assignedModel = v.model;
-        break;
-      }
-    }
-    log2(`\u2705 Assigned model: ${assignedModel}`);
-    return returnWithTrace(configKey, sessionId, assignedModel, configVersion);
-  } catch (e) {
-    if (e instanceof Error && e.message.includes("not found")) {
-      throw e;
-    }
-    if (fallback) {
-      console.warn(
-        `[Fallom WARNING] Error getting model for '${configKey}': ${e}. Using fallback: ${fallback}`
-      );
-      return returnWithTrace(configKey, sessionId, fallback, 0);
-    }
-    throw e;
   }
+  const system = replaceVariables(content.systemPrompt, variables);
+  const user = replaceVariables(content.userTemplate, variables);
+  setPromptContext({
+    promptKey,
+    promptVersion: targetVersion
+  });
+  log4(`\u2705 Got prompt: ${promptKey} v${targetVersion}`);
+  return {
+    key: promptKey,
+    version: targetVersion,
+    system,
+    user
+  };
 }
-function returnWithTrace(configKey, sessionId, model, version) {
-  try {
-    setSession(configKey, sessionId);
-  } catch {
+async function getAB(abTestKey, sessionId, options = {}) {
+  const { variables, debug = false } = options;
+  debugMode2 = debug;
+  ensureInit();
+  log4(`getAB() called: abTestKey=${abTestKey}, sessionId=${sessionId}`);
+  let abData = promptABCache.get(abTestKey);
+  if (!abData) {
+    log4("Not in cache, fetching...");
+    await fetchPromptABTests(SYNC_TIMEOUT);
+    abData = promptABCache.get(abTestKey);
   }
-  if (version > 0) {
-    recordSession(configKey, version, sessionId, model).catch(() => {
-    });
+  if (!abData) {
+    throw new Error(
+      `Prompt A/B test '${abTestKey}' not found. Check that it exists in your Fallom dashboard.`
+    );
   }
-  return model;
-}
-async function recordSession(configKey, version, sessionId, model) {
-  if (!apiKey2) return;
-  try {
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), RECORD_TIMEOUT);
-    await fetch(`${baseUrl2}/sessions`, {
-      method: "POST",
-      headers: {
-        Authorization: `Bearer ${apiKey2}`,
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify({
-        config_key: configKey,
-        config_version: version,
-        session_id: sessionId,
-        assigned_model: model
-      }),
-      signal: controller.signal
-    });
-    clearTimeout(timeoutId);
-  } catch {
+  const currentVersion = abData.current;
+  const versionData = abData.versions.get(currentVersion);
+  if (!versionData) {
+    throw new Error(`Prompt A/B test '${abTestKey}' has no current version.`);
+  }
+  const { variants } = versionData;
+  log4(`A/B test '${abTestKey}' has ${variants?.length ?? 0} variants`);
+  log4(`Version data: ${JSON.stringify(versionData, null, 2)}`);
+  if (!variants || variants.length === 0) {
+    throw new Error(
+      `Prompt A/B test '${abTestKey}' has no variants configured.`
+    );
   }
+  const hashBytes = createHash("md5").update(sessionId).digest();
+  const hashVal = hashBytes.readUInt32BE(0) % 1e6;
+  let cumulative = 0;
+  let selectedVariant = variants[variants.length - 1];
+  let selectedIndex = variants.length - 1;
+  for (let i = 0; i < variants.length; i++) {
+    cumulative += variants[i].weight * 1e4;
+    if (hashVal < cumulative) {
+      selectedVariant = variants[i];
+      selectedIndex = i;
+      break;
+    }
+  }
+  const promptKey = selectedVariant.prompt_key;
+  const promptVersion = selectedVariant.prompt_version;
+  let promptData = promptCache.get(promptKey);
+  if (!promptData) {
+    await fetchPrompts(SYNC_TIMEOUT);
+    promptData = promptCache.get(promptKey);
+  }
+  if (!promptData) {
+    throw new Error(
+      `Prompt '${promptKey}' (from A/B test '${abTestKey}') not found.`
+    );
+  }
+  const targetVersion = promptVersion ?? promptData.current;
+  const content = promptData.versions.get(targetVersion);
+  if (!content) {
+    throw new Error(
+      `Prompt '${promptKey}' version ${targetVersion} not found.`
+    );
+  }
+  const system = replaceVariables(content.systemPrompt, variables);
+  const user = replaceVariables(content.userTemplate, variables);
+  setPromptContext({
+    promptKey,
+    promptVersion: targetVersion,
+    abTestKey,
+    variantIndex: selectedIndex
+  });
+  log4(
+    `\u2705 Got prompt from A/B: ${promptKey} v${targetVersion} (variant ${selectedIndex})`
+  );
+  return {
+    key: promptKey,
+    version: targetVersion,
+    system,
+    user,
+    abTestKey,
+    variantIndex: selectedIndex
+  };
+}
+function clearPromptContext() {
+  promptContext = null;
 }
 
 // src/init.ts
@@ -2139,11 +2137,11 @@ async function init4(options = {}) {
     captureContent: options.captureContent,
     debug: options.debug
   });
-  init3({
+  init({
     apiKey: options.apiKey,
     baseUrl: configsUrl
   });
-  init({
+  init3({
     apiKey: options.apiKey,
     baseUrl: promptsUrl
   });
@@ -2151,9 +2149,9 @@ async function init4(options = {}) {
 
 // src/mastra.ts
 import { ExportResultCode } from "@opentelemetry/core";
-var promptContext = {};
+var promptContext2 = {};
 function setMastraPrompt(promptKey, version) {
-  promptContext = {
+  promptContext2 = {
     promptKey,
     promptVersion: version,
     promptAbTestKey: void 0,
@@ -2161,7 +2159,7 @@ function setMastraPrompt(promptKey, version) {
   };
 }
 function setMastraPromptAB(abTestKey, variantIndex) {
-  promptContext = {
+  promptContext2 = {
     promptKey: void 0,
     promptVersion: void 0,
     promptAbTestKey: abTestKey,
@@ -2169,7 +2167,7 @@ function setMastraPromptAB(abTestKey, variantIndex) {
   };
 }
 function clearMastraPrompt() {
-  promptContext = {};
+  promptContext2 = {};
 }
 var FallomExporter = class {
   constructor(options = {}) {
@@ -2177,9 +2175,13 @@ var FallomExporter = class {
     this.apiKey = options.apiKey ?? process.env.FALLOM_API_KEY ?? "";
     this.baseUrl = options.baseUrl ?? "https://traces.fallom.com";
     this.debug = options.debug ?? false;
-    console.log("[FallomExporter] Constructor called, debug:", this.debug);
-    console.log("[FallomExporter] API key present:", !!this.apiKey);
-    console.log("[FallomExporter] Base URL:", this.baseUrl);
+    this.session = options.session;
+    if (this.debug) {
+      console.log("[FallomExporter] Constructor called");
+      console.log("[FallomExporter] API key present:", !!this.apiKey);
+      console.log("[FallomExporter] Base URL:", this.baseUrl);
+      console.log("[FallomExporter] Session:", this.session);
+    }
     if (!this.apiKey) {
       console.warn(
         "[FallomExporter] No API key provided. Set FALLOM_API_KEY env var or pass apiKey option."
@@ -2201,10 +2203,10 @@ var FallomExporter = class {
     }
     this.log(`Exporting ${spans.length} spans...`);
     if (this.debug) {
-      for (const span2 of spans) {
-        this.log(`  - ${span2.name}`, {
+      for (const span of spans) {
+        this.log(`  - ${span.name}`, {
           attributes: Object.fromEntries(
-            Object.entries(span2.attributes).filter(
+            Object.entries(span.attributes).filter(
               ([k]) => k.startsWith("gen_ai") || k.startsWith("llm")
             )
           )
@@ -2240,33 +2242,32 @@ var FallomExporter = class {
    * Send spans to Fallom's OTLP endpoint.
    */
   async sendSpans(spans) {
-    const session = getSession();
     const resourceSpans = this.spansToOtlpJson(spans);
     const headers = {
       "Content-Type": "application/json",
       Authorization: `Bearer ${this.apiKey}`
     };
-    if (session?.configKey) {
-      headers["X-Fallom-Config-Key"] = session.configKey;
+    if (this.session?.configKey) {
+      headers["X-Fallom-Config-Key"] = this.session.configKey;
     }
-    if (session?.sessionId) {
-      headers["X-Fallom-Session-Id"] = session.sessionId;
+    if (this.session?.sessionId) {
+      headers["X-Fallom-Session-Id"] = this.session.sessionId;
     }
-    if (session?.customerId) {
-      headers["X-Fallom-Customer-Id"] = session.customerId;
+    if (this.session?.customerId) {
+      headers["X-Fallom-Customer-Id"] = this.session.customerId;
     }
-    if (promptContext.promptKey) {
-      headers["X-Fallom-Prompt-Key"] = promptContext.promptKey;
+    if (promptContext2.promptKey) {
+      headers["X-Fallom-Prompt-Key"] = promptContext2.promptKey;
     }
-    if (promptContext.promptVersion !== void 0) {
-      headers["X-Fallom-Prompt-Version"] = String(promptContext.promptVersion);
+    if (promptContext2.promptVersion !== void 0) {
+      headers["X-Fallom-Prompt-Version"] = String(promptContext2.promptVersion);
     }
-    if (promptContext.promptAbTestKey) {
-      headers["X-Fallom-Prompt-AB-Test"] = promptContext.promptAbTestKey;
+    if (promptContext2.promptAbTestKey) {
+      headers["X-Fallom-Prompt-AB-Test"] = promptContext2.promptAbTestKey;
     }
-    if (promptContext.promptVariantIndex !== void 0) {
+    if (promptContext2.promptVariantIndex !== void 0) {
       headers["X-Fallom-Prompt-Variant"] = String(
-        promptContext.promptVariantIndex
+        promptContext2.promptVariantIndex
       );
     }
     const endpoint = `${this.baseUrl}/v1/traces`;
@@ -2290,12 +2291,12 @@ var FallomExporter = class {
    */
   spansToOtlpJson(spans) {
     const resourceMap = /* @__PURE__ */ new Map();
-    for (const span2 of spans) {
-      const resourceKey = JSON.stringify(span2.resource.attributes);
+    for (const span of spans) {
+      const resourceKey = JSON.stringify(span.resource.attributes);
       if (!resourceMap.has(resourceKey)) {
         resourceMap.set(resourceKey, []);
       }
-      resourceMap.get(resourceKey).push(span2);
+      resourceMap.get(resourceKey).push(span);
     }
     const resourceSpans = [];
     for (const [_resourceKey, resourceSpanList] of resourceMap) {
@@ -2310,7 +2311,7 @@ var FallomExporter = class {
               name: firstSpan.instrumentationLibrary.name,
               version: firstSpan.instrumentationLibrary.version
             },
-            spans: resourceSpanList.map((span2) => this.spanToOtlp(span2))
+            spans: resourceSpanList.map((span) => this.spanToOtlp(span))
           }
         ]
       });
@@ -2320,21 +2321,21 @@ var FallomExporter = class {
   /**
    * Convert a single span to OTLP format.
    */
-  spanToOtlp(span2) {
+  spanToOtlp(span) {
     return {
-      traceId: span2.spanContext().traceId,
-      spanId: span2.spanContext().spanId,
-      parentSpanId: span2.parentSpanId,
-      name: span2.name,
-      kind: span2.kind,
-      startTimeUnixNano: this.hrTimeToNanos(span2.startTime),
-      endTimeUnixNano: this.hrTimeToNanos(span2.endTime),
-      attributes: this.attributesToOtlp(span2.attributes),
+      traceId: span.spanContext().traceId,
+      spanId: span.spanContext().spanId,
+      parentSpanId: span.parentSpanId,
+      name: span.name,
+      kind: span.kind,
+      startTimeUnixNano: this.hrTimeToNanos(span.startTime),
+      endTimeUnixNano: this.hrTimeToNanos(span.endTime),
+      attributes: this.attributesToOtlp(span.attributes),
       status: {
-        code: span2.status.code,
-        message: span2.status.message
+        code: span.status.code,
+        message: span.status.message
       },
-      events: span2.events.map((event) => ({
+      events: span.events.map((event) => ({
         timeUnixNano: this.hrTimeToNanos(event.time),
         name: event.name,
         attributes: this.attributesToOtlp(event.attributes || {})
@@ -2389,15 +2390,18 @@ var index_default = {
   init: init4,
   trace: trace_exports,
   models: models_exports,
-  prompts: prompts_exports
+  prompts: prompts_exports,
+  session
 };
 export {
   FallomExporter,
+  FallomSession,
   clearMastraPrompt,
   index_default as default,
   init4 as init,
   models_exports as models,
   prompts_exports as prompts,
+  session,
   setMastraPrompt,
   setMastraPromptAB,
   trace_exports as trace