@fallom/trace 0.1.11 → 0.2.0

package/dist/index.js

@@ -20,249 +20,248 @@ var __copyProps = (to, from, except, desc) => {
 };
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
-// src/prompts.ts
-var prompts_exports = {};
-__export(prompts_exports, {
-  clearPromptContext: () => clearPromptContext,
+// src/models.ts
+var models_exports = {};
+__export(models_exports, {
   get: () => get,
-  getAB: () => getAB,
-  getPromptContext: () => getPromptContext,
-  init: () => init
+  init: () => init2
 });
-function log(msg) {
-  if (debugMode) {
-    console.log(`[Fallom Prompts] ${msg}`);
+function log4(msg) {
+  if (debugMode2) {
+    console.log(`[Fallom] ${msg}`);
   }
 }
-function init(options = {}) {
-  apiKey = options.apiKey || process.env.FALLOM_API_KEY || null;
-  baseUrl = options.baseUrl || process.env.FALLOM_PROMPTS_URL || process.env.FALLOM_BASE_URL || "https://prompts.fallom.com";
-  initialized = true;
-  if (!apiKey) {
+function init2(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";
+  initialized2 = true;
+  if (!apiKey2) {
     return;
   }
-  fetchAll().catch(() => {
+  fetchConfigs().catch(() => {
   });
   if (!syncInterval) {
     syncInterval = setInterval(() => {
-      fetchAll().catch(() => {
+      fetchConfigs().catch(() => {
       });
     }, 3e4);
     syncInterval.unref();
   }
 }
 function ensureInit() {
-  if (!initialized) {
+  if (!initialized2) {
     try {
-      init();
+      init2();
     } catch {
     }
   }
 }
-async function fetchAll() {
-  await Promise.all([fetchPrompts(), fetchPromptABTests()]);
-}
-async function fetchPrompts(timeout = SYNC_TIMEOUT) {
-  if (!apiKey) return;
+async function fetchConfigs(timeout = SYNC_TIMEOUT) {
+  if (!apiKey2) {
+    log4("_fetchConfigs: No API key, skipping");
+    return;
+  }
   try {
+    log4(`Fetching configs from ${baseUrl2}/configs`);
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), timeout);
-    const resp = await fetch(`${baseUrl}/prompts`, {
-      headers: { Authorization: `Bearer ${apiKey}` },
+    const resp = await fetch(`${baseUrl2}/configs`, {
+      headers: { Authorization: `Bearer ${apiKey2}` },
       signal: controller.signal
     });
     clearTimeout(timeoutId);
+    log4(`Response status: ${resp.status}`);
     if (resp.ok) {
       const data = await resp.json();
-      for (const p of data.prompts || []) {
-        if (!promptCache.has(p.key)) {
-          promptCache.set(p.key, { versions: /* @__PURE__ */ new Map(), current: null });
+      const configs = data.configs || [];
+      log4(`Got ${configs.length} configs: ${configs.map((c) => c.key)}`);
+      for (const c of configs) {
+        const key = c.key;
+        const version = c.version || 1;
+        log4(`Config '${key}' v${version}: ${JSON.stringify(c.variants)}`);
+        if (!configCache.has(key)) {
+          configCache.set(key, { versions: /* @__PURE__ */ new Map(), latest: null });
         }
-        const cached = promptCache.get(p.key);
-        cached.versions.set(p.version, {
-          systemPrompt: p.system_prompt,
-          userTemplate: p.user_template
-        });
-        cached.current = p.version;
+        const cached = configCache.get(key);
+        cached.versions.set(version, c);
+        cached.latest = version;
       }
+    } else {
+      log4(`Fetch failed: ${resp.statusText}`);
     }
-  } catch {
+  } catch (e) {
+    log4(`Fetch exception: ${e}`);
   }
 }
-async function fetchPromptABTests(timeout = SYNC_TIMEOUT) {
-  if (!apiKey) return;
+async function fetchSpecificVersion(configKey, version, timeout = SYNC_TIMEOUT) {
+  if (!apiKey2) return null;
   try {
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), timeout);
-    const resp = await fetch(`${baseUrl}/prompt-ab-tests`, {
-      headers: { Authorization: `Bearer ${apiKey}` },
-      signal: controller.signal
-    });
+    const resp = await fetch(
+      `${baseUrl2}/configs/${configKey}/version/${version}`,
+      {
+        headers: { Authorization: `Bearer ${apiKey2}` },
+        signal: controller.signal
+      }
+    );
     clearTimeout(timeoutId);
     if (resp.ok) {
-      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;
+      const config = await resp.json();
+      if (!configCache.has(configKey)) {
+        configCache.set(configKey, { versions: /* @__PURE__ */ new Map(), latest: null });
       }
+      configCache.get(configKey).versions.set(version, config);
+      return config;
     }
   } catch {
   }
+  return null;
 }
-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;
-  debugMode = debug;
-  ensureInit();
-  log(`get() called: promptKey=${promptKey}`);
-  let promptData = promptCache.get(promptKey);
-  if (!promptData) {
-    log("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.`
-    );
-  }
-  const targetVersion = version ?? 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
-  });
-  log(`\u2705 Got prompt: ${promptKey} v${targetVersion}`);
-  return {
-    key: promptKey,
-    version: targetVersion,
-    system,
-    user
-  };
-}
-async function getAB(abTestKey, sessionId, options = {}) {
-  const { variables, debug = false } = options;
-  debugMode = debug;
+async function get(configKey, sessionId, options = {}) {
+  const { version, fallback, debug = false } = options;
+  debugMode2 = debug;
   ensureInit();
-  log(`getAB() called: abTestKey=${abTestKey}, sessionId=${sessionId}`);
-  let abData = promptABCache.get(abTestKey);
-  if (!abData) {
-    log("Not in cache, fetching...");
-    await fetchPromptABTests(SYNC_TIMEOUT);
-    abData = promptABCache.get(abTestKey);
-  }
-  if (!abData) {
-    throw new Error(
-      `Prompt A/B test '${abTestKey}' not found. Check that it exists in your Fallom dashboard.`
-    );
-  }
-  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;
-  log(`A/B test '${abTestKey}' has ${variants?.length ?? 0} variants`);
-  log(`Version data: ${JSON.stringify(versionData, null, 2)}`);
-  if (!variants || variants.length === 0) {
-    throw new Error(
-      `Prompt A/B test '${abTestKey}' has no variants configured.`
+  log4(
+    `get() called: configKey=${configKey}, sessionId=${sessionId}, fallback=${fallback}`
+  );
+  try {
+    let configData = configCache.get(configKey);
+    log4(
+      `Cache lookup for '${configKey}': ${configData ? "found" : "not found"}`
     );
-  }
-  const hashBytes = (0, import_crypto.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;
+    if (!configData) {
+      log4("Not in cache, fetching...");
+      await fetchConfigs(SYNC_TIMEOUT);
+      configData = configCache.get(configKey);
+      log4(
+        `After fetch, cache lookup: ${configData ? "found" : "still not found"}`
+      );
     }
-  }
-  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.`
+    if (!configData) {
+      log4(`Config not found, using fallback: ${fallback}`);
+      if (fallback) {
+        console.warn(
+          `[Fallom WARNING] Config '${configKey}' not found, using fallback model: ${fallback}`
+        );
+        return returnModel(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 returnModel(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 returnModel(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);
+    log4(
+      `Config found! Version: ${configVersion}, Variants: ${JSON.stringify(
+        variants
+      )}`
     );
+    const hashBytes = (0, import_crypto.createHash)("md5").update(sessionId).digest();
+    const hashVal = hashBytes.readUInt32BE(0) % 1e6;
+    log4(`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;
+      log4(
+        `Variant ${v.model}: weight=${v.weight}%, range=${oldCumulative}-${cumulative}, hash=${hashVal}, match=${hashVal < cumulative}`
+      );
+      if (hashVal < cumulative) {
+        assignedModel = v.model;
+        break;
+      }
+    }
+    log4(`\u2705 Assigned model: ${assignedModel}`);
+    return returnModel(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 returnModel(configKey, sessionId, fallback, 0);
+    }
+    throw e;
   }
-  const targetVersion = promptVersion ?? promptData.current;
-  const content = promptData.versions.get(targetVersion);
-  if (!content) {
-    throw new Error(
-      `Prompt '${promptKey}' version ${targetVersion} not found.`
-    );
+}
+function returnModel(configKey, sessionId, model, version) {
+  if (version > 0) {
+    recordSession(configKey, version, sessionId, model).catch(() => {
+    });
   }
-  const system = replaceVariables(content.systemPrompt, variables);
-  const user = replaceVariables(content.userTemplate, variables);
-  setPromptContext({
-    promptKey,
-    promptVersion: targetVersion,
-    abTestKey,
-    variantIndex: selectedIndex
-  });
-  log(
-    `\u2705 Got prompt from A/B: ${promptKey} v${targetVersion} (variant ${selectedIndex})`
-  );
-  return {
-    key: promptKey,
-    version: targetVersion,
-    system,
-    user,
-    abTestKey,
-    variantIndex: selectedIndex
-  };
+  return model;
 }
-function clearPromptContext() {
-  promptContext = null;
+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 {
+  }
 }
-var import_crypto, apiKey, baseUrl, initialized, syncInterval, debugMode, promptCache, promptABCache, promptContext, SYNC_TIMEOUT;
-var init_prompts = __esm({
-  "src/prompts.ts"() {
+var import_crypto, apiKey2, baseUrl2, initialized2, syncInterval, debugMode2, configCache, SYNC_TIMEOUT, RECORD_TIMEOUT;
+var init_models = __esm({
+  "src/models.ts"() {
     "use strict";
     import_crypto = require("crypto");
-    apiKey = null;
-    baseUrl = "https://prompts.fallom.com";
-    initialized = false;
+    apiKey2 = null;
+    baseUrl2 = "https://configs.fallom.com";
+    initialized2 = false;
     syncInterval = null;
-    debugMode = false;
-    promptCache = /* @__PURE__ */ new Map();
-    promptABCache = /* @__PURE__ */ new Map();
-    promptContext = null;
+    debugMode2 = false;
+    configCache = /* @__PURE__ */ new Map();
     SYNC_TIMEOUT = 2e3;
+    RECORD_TIMEOUT = 1e3;
   }
 });
 
@@ -270,11 +269,13 @@ var init_prompts = __esm({
 var index_exports = {};
 __export(index_exports, {
   FallomExporter: () => FallomExporter,
+  FallomSession: () => FallomSession,
   clearMastraPrompt: () => clearMastraPrompt,
   default: () => index_default,
   init: () => init4,
   models: () => models_exports,
   prompts: () => prompts_exports,
+  session: () => session,
   setMastraPrompt: () => setMastraPrompt,
   setMastraPromptAB: () => setMastraPromptAB,
   trace: () => trace_exports
@@ -284,19 +285,13 @@ module.exports = __toCommonJS(index_exports);
 // src/trace.ts
 var trace_exports = {};
 __export(trace_exports, {
-  clearSession: () => clearSession,
-  getSession: () => getSession,
-  init: () => init2,
-  runWithSession: () => runWithSession,
-  setSession: () => setSession,
-  shutdown: () => shutdown,
-  span: () => span,
-  wrapAISDK: () => wrapAISDK,
-  wrapAnthropic: () => wrapAnthropic,
-  wrapGoogleAI: () => wrapGoogleAI,
-  wrapMastraAgent: () => wrapMastraAgent,
-  wrapOpenAI: () => wrapOpenAI
+  FallomSession: () => FallomSession,
+  init: () => init,
+  session: () => session,
+  shutdown: () => shutdown
 });
+
+// src/trace/core.ts
 var import_async_hooks = require("async_hooks");
 var import_sdk_node = require("@opentelemetry/sdk-node");
 var import_exporter_trace_otlp_http = require("@opentelemetry/exporter-trace-otlp-http");
@@ -898,40 +893,39 @@ var Resource = (
   })()
 );
 
-// src/trace.ts
-var sessionStorage = new import_async_hooks.AsyncLocalStorage();
-var fallbackSession = null;
-var apiKey2 = null;
-var baseUrl2 = "https://traces.fallom.com";
-var initialized2 = false;
+// src/trace/core.ts
+var traceContextStorage = new import_async_hooks.AsyncLocalStorage();
+var fallbackTraceContext = null;
+var apiKey = null;
+var baseUrl = "https://traces.fallom.com";
+var initialized = false;
 var captureContent = true;
-var debugMode2 = false;
+var debugMode = false;
 var sdk = null;
-function log2(...args) {
-  if (debugMode2) console.log("[Fallom]", ...args);
+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) {
-    log2("\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);
-      }
-      log2(
-        "   Added session context:",
-        ctx.configKey,
-        ctx.sessionId,
-        ctx.customerId
-      );
-    } else {
-      log2("   No session context available");
-    }
+  onStart(span, _parentContext) {
+    log("\u{1F4CD} Span started:", span.name || "unknown");
   },
-  onEnd(span2) {
-    log2("\u2705 Span ended:", span2.name, "duration:", span2.duration);
+  onEnd(span) {
+    log("\u2705 Span ended:", span.name, "duration:", span.duration);
   },
   shutdown() {
     return Promise.resolve();
@@ -940,47 +934,6 @@ var fallomSpanProcessor = {
     return Promise.resolve();
   }
 };
-async function init2(options = {}) {
-  if (initialized2) return;
-  debugMode2 = options.debug ?? false;
-  log2("\u{1F680} Initializing Fallom tracing...");
-  apiKey2 = options.apiKey || process.env.FALLOM_API_KEY || null;
-  baseUrl2 = 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 (!apiKey2) {
-    throw new Error(
-      "No API key provided. Set FALLOM_API_KEY environment variable or pass apiKey parameter."
-    );
-  }
-  initialized2 = true;
-  log2("\u{1F4E1} Exporter URL:", `${baseUrl2}/v1/traces`);
-  const exporter = new import_exporter_trace_otlp_http.OTLPTraceExporter({
-    url: `${baseUrl2}/v1/traces`,
-    headers: {
-      Authorization: `Bearer ${apiKey2}`
-    }
-  });
-  const instrumentations = await getInstrumentations();
-  log2("\u{1F527} Loaded instrumentations:", instrumentations.length);
-  sdk = new import_sdk_node.NodeSDK({
-    resource: new Resource({
-      "service.name": "fallom-traced-app"
-    }),
-    traceExporter: exporter,
-    spanProcessor: fallomSpanProcessor,
-    instrumentations
-  });
-  sdk.start();
-  log2("\u2705 SDK started");
-  process.on("SIGTERM", () => {
-    sdk?.shutdown().catch(console.error);
-  });
-}
 async function getInstrumentations() {
   const instrumentations = [];
   await tryAddInstrumentation(
@@ -1028,82 +981,97 @@ async function tryAddInstrumentation(instrumentations, pkg, className) {
       instrumentations.push(
         new InstrumentationClass({ traceContent: captureContent })
       );
-      log2(`   \u2705 Loaded ${pkg}`);
+      log(`   \u2705 Loaded ${pkg}`);
     } else {
-      log2(
+      log(
         `   \u26A0\uFE0F ${pkg} loaded but ${className} not found. Available:`,
         Object.keys(mod)
       );
-    }
-  } catch (e) {
-    log2(`   \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);
+    }
+  } catch {
+    log(`   \u274C ${pkg} not installed`);
   }
-  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 (!initialized2) {
-    throw new Error("Fallom not initialized. Call trace.init() first.");
+async function init(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 import_exporter_trace_otlp_http.OTLPTraceExporter({
+    url: `${baseUrl}/v1/traces`,
+    headers: {
+      Authorization: `Bearer ${apiKey}`
+    }
+  });
+  const instrumentations = await getInstrumentations();
+  log("\u{1F527} Loaded instrumentations:", instrumentations.length);
+  sdk = new import_sdk_node.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(`${baseUrl2}/spans`, {
+    const response = await fetch(url, {
       method: "POST",
       headers: {
-        Authorization: `Bearer ${apiKey2}`,
+        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();
-    initialized2 = 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 = {};
@@ -1131,70 +1099,39 @@ 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 import_async_hooks.AsyncLocalStorage();
-var fallbackTraceContext = null;
-async function sendTrace(trace) {
-  const url = `${baseUrl2}/v1/traces`;
-  log2("\u{1F4E4} Sending trace to:", url);
-  log2("   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 ${apiKey2}`,
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(trace),
-      signal: controller.signal
-    });
-    clearTimeout(timeoutId);
-    if (!response.ok) {
-      const text = await response.text();
-      log2("\u274C Trace send failed:", response.status, text);
-    } else {
-      log2("\u2705 Trace sent:", trace.name, trace.model);
-    }
-  } catch (err) {
-    log2("\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 || !initialized2) {
+    if (!isInitialized()) {
       return originalCreate(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
-      promptCtx = getPromptContext2();
-    } 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,
         response?.id
-      ) : void 0;
+      ) : {};
+      if (response?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(response.usage);
+      }
+      if (response?.choices?.[0]?.finish_reason) {
+        attributes["gen_ai.response.finish_reason"] = response.choices[0].finish_reason;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1212,17 +1149,13 @@ function wrapOpenAI(client) {
         prompt_tokens: response?.usage?.prompt_tokens,
         completion_tokens: response?.usage?.completion_tokens,
         total_tokens: response?.usage?.total_tokens,
-        attributes,
-        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,
@@ -1246,11 +1179,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;
@@ -1258,37 +1187,40 @@ 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 || !initialized2) {
+    if (!isInitialized()) {
       return originalCreate(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
-      promptCtx = getPromptContext2();
-    } 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,
         response?.id
-      ) : void 0;
-      if (attributes && params?.system) {
+      ) : {};
+      if (params?.system) {
         attributes["gen_ai.system_prompt"] = params.system;
       }
+      if (response?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(response.usage);
+      }
+      if (response?.stop_reason) {
+        attributes["gen_ai.response.finish_reason"] = response.stop_reason;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1306,17 +1238,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,
-        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,
@@ -1343,11 +1271,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;
@@ -1355,24 +1279,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 || !initialized2) {
+    if (!isInitialized()) {
       return originalGenerate(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
-      promptCtx = getPromptContext2();
-    } 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();
@@ -1380,7 +1301,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];
@@ -1399,6 +1320,13 @@ function wrapGoogleAI(model) {
           attributes["gen_ai.completion.0.content"] = outputText;
         }
       }
+      if (usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(usage);
+      }
+      const candidate = result?.candidates?.[0];
+      if (candidate?.finishReason) {
+        attributes["gen_ai.response.finish_reason"] = candidate.finishReason;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1416,11 +1344,7 @@ function wrapGoogleAI(model) {
         prompt_tokens: usage?.promptTokenCount,
         completion_tokens: usage?.candidatesTokenCount,
         total_tokens: usage?.totalTokenCount,
-        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: Object.keys(attributes).length > 0 ? attributes : void 0
       }).catch(() => {
       });
       return response;
@@ -1428,7 +1352,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];
@@ -1452,11 +1376,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;
@@ -1464,39 +1384,70 @@ function wrapGoogleAI(model) {
   };
   return model;
 }
-function wrapAISDK(ai) {
-  const aiModule = ai;
-  return {
-    generateText: createGenerateTextWrapper(aiModule),
-    streamText: createStreamTextWrapper(aiModule),
-    generateObject: aiModule.generateObject ? createGenerateObjectWrapper(aiModule) : void 0,
-    streamObject: aiModule.streamObject ? createStreamObjectWrapper(aiModule) : void 0
-  };
+
+// 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);
+  let promptTokens = isValidNumber(usage?.promptTokens) ? usage.promptTokens : void 0;
+  let completionTokens = isValidNumber(usage?.completionTokens) ? usage.completionTokens : void 0;
+  let totalTokens = isValidNumber(usage?.totalTokens) ? usage.totalTokens : void 0;
+  let cost;
+  const orUsage = result?.experimental_providerMetadata?.openrouter?.usage;
+  if (orUsage) {
+    if (promptTokens === void 0 && isValidNumber(orUsage.promptTokens)) {
+      promptTokens = orUsage.promptTokens;
+    }
+    if (completionTokens === void 0 && isValidNumber(orUsage.completionTokens)) {
+      completionTokens = orUsage.completionTokens;
+    }
+    if (totalTokens === void 0 && isValidNumber(orUsage.totalTokens)) {
+      totalTokens = orUsage.totalTokens;
+    }
+    if (isValidNumber(orUsage.cost)) {
+      cost = orUsage.cost;
+    }
+  }
+  if (totalTokens === void 0 && (promptTokens !== void 0 || completionTokens !== void 0)) {
+    totalTokens = (promptTokens ?? 0) + (completionTokens ?? 0);
+  }
+  return { promptTokens, completionTokens, totalTokens, cost };
 }
-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 || !initialized2) {
+    if (!isInitialized()) {
       return aiModule.generateText(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
-      promptCtx = getPromptContext2();
-    } 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 (debug || isDebugMode()) {
+        console.log(
+          "\n\u{1F50D} [Fallom Debug] generateText result keys:",
+          Object.keys(result || {})
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.usage:",
+          JSON.stringify(result?.usage, null, 2)
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.experimental_providerMetadata:",
+          JSON.stringify(result?.experimental_providerMetadata, 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 (params?.prompt) {
@@ -1517,6 +1468,18 @@ function createGenerateTextWrapper(aiModule) {
           attributes["gen_ai.response.id"] = result.response.id;
         }
       }
+      if (result?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(result.usage);
+      }
+      if (result?.experimental_providerMetadata) {
+        attributes["fallom.raw.providerMetadata"] = JSON.stringify(
+          result.experimental_providerMetadata
+        );
+      }
+      if (result?.finishReason) {
+        attributes["gen_ai.response.finish_reason"] = result.finishReason;
+      }
+      const usage = extractUsageFromResult(result);
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1531,14 +1494,10 @@ function createGenerateTextWrapper(aiModule) {
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "OK",
-        prompt_tokens: result?.usage?.promptTokens,
-        completion_tokens: result?.usage?.completionTokens,
-        total_tokens: result?.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
+        prompt_tokens: usage.promptTokens,
+        completion_tokens: usage.completionTokens,
+        total_tokens: usage.totalTokens,
+        attributes: captureContent2 ? attributes : void 0
       }).catch(() => {
       });
       return result;
@@ -1559,44 +1518,58 @@ 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 || !initialized2) {
+    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: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
-      promptCtx = getPromptContext2();
-    } catch {
-    }
     if (result?.usage) {
-      result.usage.then((usage) => {
+      result.usage.then(async (rawUsage) => {
         const endTime = Date.now();
-        log2("\u{1F4CA} streamText usage:", JSON.stringify(usage, null, 2));
+        if (debug || isDebugMode()) {
+          console.log(
+            "\n\u{1F50D} [Fallom Debug] 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 {
+            providerMetadata = await providerMetadata;
+          } catch {
+            providerMetadata = void 0;
+          }
+        }
+        const usage = extractUsageFromResult(
+          { experimental_providerMetadata: providerMetadata },
+          rawUsage
+        );
         const attributes = {};
-        if (captureContent) {
+        if (captureContent2) {
           attributes["gen_ai.request.model"] = modelId;
           if (params?.prompt) {
             attributes["gen_ai.prompt.0.role"] = "user";
@@ -1606,6 +1579,12 @@ function createStreamTextWrapper(aiModule) {
         if (firstTokenTime) {
           attributes["gen_ai.time_to_first_token_ms"] = firstTokenTime - startTime;
         }
+        if (rawUsage) {
+          attributes["fallom.raw.usage"] = JSON.stringify(rawUsage);
+        }
+        if (providerMetadata) {
+          attributes["fallom.raw.providerMetadata"] = JSON.stringify(providerMetadata);
+        }
         const tracePayload = {
           config_key: ctx.configKey,
           session_id: ctx.sessionId,
@@ -1620,15 +1599,11 @@ function createStreamTextWrapper(aiModule) {
           end_time: new Date(endTime).toISOString(),
           duration_ms: endTime - startTime,
           status: "OK",
-          prompt_tokens: usage?.promptTokens,
-          completion_tokens: usage?.completionTokens,
-          total_tokens: usage?.totalTokens,
+          prompt_tokens: usage.promptTokens,
+          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(() => {
         });
@@ -1649,11 +1624,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(() => {
         });
       });
@@ -1681,30 +1652,37 @@ 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 || !initialized2) {
+    if (!isInitialized()) {
       return aiModule.generateObject(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
-      promptCtx = getPromptContext2();
-    } 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 (debug || isDebugMode()) {
+        console.log(
+          "\n\u{1F50D} [Fallom Debug] generateObject result keys:",
+          Object.keys(result || {})
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.usage:",
+          JSON.stringify(result?.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) {
@@ -1714,6 +1692,18 @@ function createGenerateObjectWrapper(aiModule) {
           );
         }
       }
+      if (result?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(result.usage);
+      }
+      if (result?.experimental_providerMetadata) {
+        attributes["fallom.raw.providerMetadata"] = JSON.stringify(
+          result.experimental_providerMetadata
+        );
+      }
+      if (result?.finishReason) {
+        attributes["gen_ai.response.finish_reason"] = result.finishReason;
+      }
+      const usage = extractUsageFromResult(result);
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1728,14 +1718,10 @@ function createGenerateObjectWrapper(aiModule) {
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "OK",
-        prompt_tokens: result?.usage?.promptTokens,
-        completion_tokens: result?.usage?.completionTokens,
-        total_tokens: result?.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
+        prompt_tokens: usage.promptTokens,
+        completion_tokens: usage.completionTokens,
+        total_tokens: usage.totalTokens,
+        attributes: captureContent2 ? attributes : void 0
       }).catch(() => {
       });
       return result;
@@ -1756,50 +1742,70 @@ 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);
-    log2("\u{1F50D} streamObject result keys:", Object.keys(result || {}));
-    if (!ctx || !initialized2) {
+    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: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
-      promptCtx = getPromptContext2();
-    } catch {
-    }
     if (result?.usage) {
-      result.usage.then((usage) => {
+      result.usage.then(async (rawUsage) => {
         const endTime = Date.now();
-        log2("\u{1F4CA} streamObject usage:", JSON.stringify(usage, null, 2));
+        if (debug || isDebugMode()) {
+          console.log(
+            "\n\u{1F50D} [Fallom Debug] 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 {
+            providerMetadata = await providerMetadata;
+          } catch {
+            providerMetadata = void 0;
+          }
+        }
+        const usage = extractUsageFromResult(
+          { experimental_providerMetadata: providerMetadata },
+          rawUsage
+        );
         const attributes = {};
-        if (captureContent) {
+        if (captureContent2) {
           attributes["gen_ai.request.model"] = modelId;
         }
         if (firstTokenTime) {
           attributes["gen_ai.time_to_first_token_ms"] = firstTokenTime - startTime;
         }
+        if (rawUsage) {
+          attributes["fallom.raw.usage"] = JSON.stringify(rawUsage);
+        }
+        if (providerMetadata) {
+          attributes["fallom.raw.providerMetadata"] = JSON.stringify(providerMetadata);
+        }
         sendTrace({
           config_key: ctx.configKey,
           session_id: ctx.sessionId,
@@ -1814,14 +1820,10 @@ function createStreamObjectWrapper(aiModule) {
           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: captureContent ? attributes : void 0,
-          prompt_key: promptCtx?.promptKey,
-          prompt_version: promptCtx?.promptVersion,
-          prompt_ab_test_key: promptCtx?.abTestKey,
-          prompt_variant_index: promptCtx?.variantIndex
+          prompt_tokens: usage.promptTokens,
+          completion_tokens: usage.completionTokens,
+          total_tokens: usage.totalTokens,
+          attributes: captureContent2 ? attributes : void 0
         }).catch(() => {
         });
       }).catch((error) => {
@@ -1840,11 +1842,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(() => {
         });
       });
@@ -1855,7 +1853,7 @@ function createStreamObjectWrapper(aiModule) {
         for await (const chunk of originalStream) {
           if (!firstTokenTime) {
             firstTokenTime = Date.now();
-            log2("\u23F1\uFE0F Time to first token:", firstTokenTime - startTime, "ms");
+            log3("\u23F1\uFE0F Time to first token:", firstTokenTime - startTime, "ms");
           }
           yield chunk;
         }
@@ -1872,20 +1870,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 || !initialized2) {
+    if (!isInitialized()) {
       return originalGenerate(...args);
     }
-    let promptCtx = null;
-    try {
-      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
-      promptCtx = getPromptContext2();
-    } catch {
-    }
     const traceId = generateHexId(32);
     const spanId = generateHexId(16);
     const startTime = Date.now();
@@ -1957,11 +1962,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(() => {
       });
@@ -1980,11 +1981,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(() => {
       });
@@ -1994,38 +1991,216 @@ 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: get3 } = await Promise.resolve().then(() => (init_models(), models_exports));
+    return get3(configKey, this.ctx.sessionId, opts);
+  }
+  /**
+   * Wrap a Vercel AI SDK model to trace all calls.
+   */
+  traceModel(model) {
+    const ctx = this.ctx;
+    const tracedModel = Object.create(model);
+    if (model.doGenerate) {
+      const originalDoGenerate = model.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 (model.doStream) {
+      const originalDoStream = model.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/index.ts
+init_models();
+
+// src/prompts.ts
+var prompts_exports = {};
+__export(prompts_exports, {
+  clearPromptContext: () => clearPromptContext,
   get: () => get2,
+  getAB: () => getAB,
+  getPromptContext: () => getPromptContext,
   init: () => init3
 });
 var import_crypto2 = require("crypto");
 var apiKey3 = null;
-var baseUrl3 = "https://configs.fallom.com";
+var baseUrl3 = "https://prompts.fallom.com";
 var initialized3 = false;
 var syncInterval2 = null;
 var debugMode3 = false;
-var configCache = /* @__PURE__ */ new Map();
+var promptCache = /* @__PURE__ */ new Map();
+var promptABCache = /* @__PURE__ */ new Map();
+var promptContext = null;
 var SYNC_TIMEOUT2 = 2e3;
-var RECORD_TIMEOUT = 1e3;
-function log3(msg) {
+function log5(msg) {
   if (debugMode3) {
-    console.log(`[Fallom] ${msg}`);
+    console.log(`[Fallom Prompts] ${msg}`);
   }
 }
 function init3(options = {}) {
   apiKey3 = options.apiKey || process.env.FALLOM_API_KEY || null;
-  baseUrl3 = options.baseUrl || process.env.FALLOM_CONFIGS_URL || process.env.FALLOM_BASE_URL || "https://configs.fallom.com";
+  baseUrl3 = options.baseUrl || process.env.FALLOM_PROMPTS_URL || process.env.FALLOM_BASE_URL || "https://prompts.fallom.com";
   initialized3 = true;
   if (!apiKey3) {
     return;
   }
-  fetchConfigs().catch(() => {
+  fetchAll().catch(() => {
   });
   if (!syncInterval2) {
     syncInterval2 = setInterval(() => {
-      fetchConfigs().catch(() => {
+      fetchAll().catch(() => {
       });
     }, 3e4);
     syncInterval2.unref();
@@ -2039,231 +2214,221 @@ function ensureInit2() {
     }
   }
 }
-async function fetchConfigs(timeout = SYNC_TIMEOUT2) {
-  if (!apiKey3) {
-    log3("_fetchConfigs: No API key, skipping");
-    return;
+async function fetchAll() {
+  await Promise.all([fetchPrompts(), fetchPromptABTests()]);
+}
+async function fetchPrompts(timeout = SYNC_TIMEOUT2) {
+  if (!apiKey3) return;
+  try {
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), timeout);
+    const resp = await fetch(`${baseUrl3}/prompts`, {
+      headers: { Authorization: `Bearer ${apiKey3}` },
+      signal: controller.signal
+    });
+    clearTimeout(timeoutId);
+    if (resp.ok) {
+      const data = await resp.json();
+      for (const p of data.prompts || []) {
+        if (!promptCache.has(p.key)) {
+          promptCache.set(p.key, { versions: /* @__PURE__ */ new Map(), current: null });
+        }
+        const cached = promptCache.get(p.key);
+        cached.versions.set(p.version, {
+          systemPrompt: p.system_prompt,
+          userTemplate: p.user_template
+        });
+        cached.current = p.version;
+      }
+    }
+  } catch {
   }
+}
+async function fetchPromptABTests(timeout = SYNC_TIMEOUT2) {
+  if (!apiKey3) return;
   try {
-    log3(`Fetching configs from ${baseUrl3}/configs`);
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), timeout);
-    const resp = await fetch(`${baseUrl3}/configs`, {
+    const resp = await fetch(`${baseUrl3}/prompt-ab-tests`, {
       headers: { Authorization: `Bearer ${apiKey3}` },
       signal: controller.signal
     });
     clearTimeout(timeoutId);
-    log3(`Response status: ${resp.status}`);
     if (resp.ok) {
       const data = await resp.json();
-      const configs = data.configs || [];
-      log3(`Got ${configs.length} configs: ${configs.map((c) => c.key)}`);
-      for (const c of configs) {
-        const key = c.key;
-        const version = c.version || 1;
-        log3(`Config '${key}' v${version}: ${JSON.stringify(c.variants)}`);
-        if (!configCache.has(key)) {
-          configCache.set(key, { versions: /* @__PURE__ */ new Map(), latest: null });
+      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 = configCache.get(key);
-        cached.versions.set(version, c);
-        cached.latest = version;
+        const cached = promptABCache.get(t.key);
+        cached.versions.set(t.version, { variants: t.variants });
+        cached.current = t.version;
       }
-    } else {
-      log3(`Fetch failed: ${resp.statusText}`);
     }
-  } catch (e) {
-    log3(`Fetch exception: ${e}`);
+  } catch {
   }
 }
-async function fetchSpecificVersion(configKey, version, timeout = SYNC_TIMEOUT2) {
-  if (!apiKey3) return null;
-  try {
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), timeout);
-    const resp = await fetch(
-      `${baseUrl3}/configs/${configKey}/version/${version}`,
-      {
-        headers: { Authorization: `Bearer ${apiKey3}` },
-        signal: controller.signal
-      }
+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 get2(promptKey, options = {}) {
+  const { variables, version, debug = false } = options;
+  debugMode3 = debug;
+  ensureInit2();
+  log5(`get() called: promptKey=${promptKey}`);
+  let promptData = promptCache.get(promptKey);
+  if (!promptData) {
+    log5("Not in cache, fetching...");
+    await fetchPrompts(SYNC_TIMEOUT2);
+    promptData = promptCache.get(promptKey);
+  }
+  if (!promptData) {
+    throw new Error(
+      `Prompt '${promptKey}' not found. Check that it exists in your Fallom dashboard.`
     );
-    clearTimeout(timeoutId);
-    if (resp.ok) {
-      const config = await resp.json();
-      if (!configCache.has(configKey)) {
-        configCache.set(configKey, { versions: /* @__PURE__ */ new Map(), latest: null });
-      }
-      configCache.get(configKey).versions.set(version, config);
-      return config;
-    }
-  } catch {
   }
-  return null;
+  const targetVersion = version ?? 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
+  });
+  log5(`\u2705 Got prompt: ${promptKey} v${targetVersion}`);
+  return {
+    key: promptKey,
+    version: targetVersion,
+    system,
+    user
+  };
 }
-async function get2(configKey, sessionId, options = {}) {
-  const { version, fallback, debug = false } = options;
+async function getAB(abTestKey, sessionId, options = {}) {
+  const { variables, debug = false } = options;
   debugMode3 = debug;
   ensureInit2();
-  log3(
-    `get() called: configKey=${configKey}, sessionId=${sessionId}, fallback=${fallback}`
-  );
-  try {
-    let configData = configCache.get(configKey);
-    log3(
-      `Cache lookup for '${configKey}': ${configData ? "found" : "not found"}`
+  log5(`getAB() called: abTestKey=${abTestKey}, sessionId=${sessionId}`);
+  let abData = promptABCache.get(abTestKey);
+  if (!abData) {
+    log5("Not in cache, fetching...");
+    await fetchPromptABTests(SYNC_TIMEOUT2);
+    abData = promptABCache.get(abTestKey);
+  }
+  if (!abData) {
+    throw new Error(
+      `Prompt A/B test '${abTestKey}' not found. Check that it exists in your Fallom dashboard.`
     );
-    if (!configData) {
-      log3("Not in cache, fetching...");
-      await fetchConfigs(SYNC_TIMEOUT2);
-      configData = configCache.get(configKey);
-      log3(
-        `After fetch, cache lookup: ${configData ? "found" : "still not found"}`
-      );
-    }
-    if (!configData) {
-      log3(`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_TIMEOUT2) || 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);
-    log3(
-      `Config found! Version: ${configVersion}, Variants: ${JSON.stringify(
-        variants
-      )}`
+  }
+  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;
+  log5(`A/B test '${abTestKey}' has ${variants?.length ?? 0} variants`);
+  log5(`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 = (0, import_crypto2.createHash)("md5").update(sessionId).digest();
-    const hashVal = hashBytes.readUInt32BE(0) % 1e6;
-    log3(`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;
-      log3(
-        `Variant ${v.model}: weight=${v.weight}%, range=${oldCumulative}-${cumulative}, hash=${hashVal}, match=${hashVal < cumulative}`
-      );
-      if (hashVal < cumulative) {
-        assignedModel = v.model;
-        break;
-      }
-    }
-    log3(`\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);
+  }
+  const hashBytes = (0, import_crypto2.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;
     }
-    throw e;
   }
-}
-function returnWithTrace(configKey, sessionId, model, version) {
-  try {
-    setSession(configKey, sessionId);
-  } catch {
+  const promptKey = selectedVariant.prompt_key;
+  const promptVersion = selectedVariant.prompt_version;
+  let promptData = promptCache.get(promptKey);
+  if (!promptData) {
+    await fetchPrompts(SYNC_TIMEOUT2);
+    promptData = promptCache.get(promptKey);
   }
-  if (version > 0) {
-    recordSession(configKey, version, sessionId, model).catch(() => {
-    });
+  if (!promptData) {
+    throw new Error(
+      `Prompt '${promptKey}' (from A/B test '${abTestKey}') not found.`
+    );
   }
-  return model;
-}
-async function recordSession(configKey, version, sessionId, model) {
-  if (!apiKey3) return;
-  try {
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), RECORD_TIMEOUT);
-    await fetch(`${baseUrl3}/sessions`, {
-      method: "POST",
-      headers: {
-        Authorization: `Bearer ${apiKey3}`,
-        "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 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
+  });
+  log5(
+    `\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/index.ts
-init_prompts();
 
 // src/init.ts
-init_prompts();
+init_models();
 async function init4(options = {}) {
   const tracesUrl = options.tracesUrl || process.env.FALLOM_TRACES_URL || "https://traces.fallom.com";
   const configsUrl = options.configsUrl || process.env.FALLOM_CONFIGS_URL || "https://configs.fallom.com";
   const promptsUrl = options.promptsUrl || process.env.FALLOM_PROMPTS_URL || "https://prompts.fallom.com";
-  await init2({
+  await init({
     apiKey: options.apiKey,
     baseUrl: tracesUrl,
     captureContent: options.captureContent,
     debug: options.debug
   });
-  init3({
+  init2({
     apiKey: options.apiKey,
     baseUrl: configsUrl
   });
-  init({
+  init3({
     apiKey: options.apiKey,
     baseUrl: promptsUrl
   });
 }
 
 // src/mastra.ts
-var import_core2 = require("@opentelemetry/core");
+var import_core12 = require("@opentelemetry/core");
 var promptContext2 = {};
 function setMastraPrompt(promptKey, version) {
   promptContext2 = {
@@ -2290,9 +2455,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."
@@ -2309,15 +2478,15 @@ var FallomExporter = class {
    */
   export(spans, resultCallback) {
     if (spans.length === 0) {
-      resultCallback({ code: import_core2.ExportResultCode.SUCCESS });
+      resultCallback({ code: import_core12.ExportResultCode.SUCCESS });
       return;
     }
     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")
             )
           )
@@ -2326,11 +2495,11 @@ var FallomExporter = class {
     }
     const exportPromise = this.sendSpans(spans).then(() => {
       this.log("Export successful");
-      resultCallback({ code: import_core2.ExportResultCode.SUCCESS });
+      resultCallback({ code: import_core12.ExportResultCode.SUCCESS });
     }).catch((error) => {
       console.error("[FallomExporter] Export failed:", error);
       resultCallback({
-        code: import_core2.ExportResultCode.FAILED,
+        code: import_core12.ExportResultCode.FAILED,
         error: error instanceof Error ? error : new Error(String(error))
       });
     });
@@ -2353,20 +2522,19 @@ 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 (promptContext2.promptKey) {
       headers["X-Fallom-Prompt-Key"] = promptContext2.promptKey;
@@ -2403,12 +2571,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) {
@@ -2423,7 +2591,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))
           }
         ]
       });
@@ -2433,21 +2601,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 || {})
@@ -2498,20 +2666,23 @@ var FallomExporter = class {
 };
 
 // src/index.ts
-init_prompts();
+init_models();
 var index_default = {
   init: init4,
   trace: trace_exports,
   models: models_exports,
-  prompts: prompts_exports
+  prompts: prompts_exports,
+  session
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   FallomExporter,
+  FallomSession,
   clearMastraPrompt,
   init,
   models,
   prompts,
+  session,
   setMastraPrompt,
   setMastraPromptAB,
   trace