@fallom/trace 0.1.5 → 0.1.10

package/dist/index.js

@@ -36,7 +36,7 @@ function log(msg) {
 }
 function init(options = {}) {
   apiKey = options.apiKey || process.env.FALLOM_API_KEY || null;
-  baseUrl = options.baseUrl || process.env.FALLOM_BASE_URL || "https://spans.fallom.com";
+  baseUrl = options.baseUrl || process.env.FALLOM_PROMPTS_URL || process.env.FALLOM_BASE_URL || "https://prompts.fallom.com";
   initialized = true;
   if (!apiKey) {
     return;
@@ -187,6 +187,13 @@ async function getAB(abTestKey, sessionId, options = {}) {
     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.`
+    );
+  }
   const hashBytes = (0, import_crypto.createHash)("md5").update(sessionId).digest();
   const hashVal = hashBytes.readUInt32BE(0) % 1e6;
   let cumulative = 0;
@@ -248,7 +255,7 @@ var init_prompts = __esm({
     "use strict";
     import_crypto = require("crypto");
     apiKey = null;
-    baseUrl = "https://spans.fallom.com";
+    baseUrl = "https://prompts.fallom.com";
     initialized = false;
     syncInterval = null;
     debugMode = false;
@@ -280,6 +287,7 @@ __export(trace_exports, {
   setSession: () => setSession,
   shutdown: () => shutdown,
   span: () => span,
+  wrapAISDK: () => wrapAISDK,
   wrapAnthropic: () => wrapAnthropic,
   wrapGoogleAI: () => wrapGoogleAI,
   wrapOpenAI: () => wrapOpenAI
@@ -915,7 +923,7 @@ var Resource = (
 var sessionStorage = new import_async_hooks.AsyncLocalStorage();
 var fallbackSession = null;
 var apiKey2 = null;
-var baseUrl2 = "https://spans.fallom.com";
+var baseUrl2 = "https://traces.fallom.com";
 var initialized2 = false;
 var captureContent = true;
 var debugMode2 = false;
@@ -958,7 +966,7 @@ async function init2(options = {}) {
   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_BASE_URL || "https://spans.fallom.com";
+  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;
@@ -1118,11 +1126,47 @@ async function shutdown() {
     initialized2 = false;
   }
 }
+function messagesToOtelAttributes(messages, completion, model, responseId) {
+  const attrs = {};
+  if (model) {
+    attrs["gen_ai.request.model"] = model;
+    attrs["gen_ai.response.model"] = model;
+  }
+  if (responseId) {
+    attrs["gen_ai.response.id"] = responseId;
+  }
+  if (messages) {
+    messages.forEach((msg, i) => {
+      attrs[`gen_ai.prompt.${i}.role`] = msg.role;
+      attrs[`gen_ai.prompt.${i}.content`] = typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content);
+    });
+  }
+  if (completion) {
+    attrs["gen_ai.completion.0.role"] = completion.role;
+    attrs["gen_ai.completion.0.content"] = typeof completion.content === "string" ? completion.content : JSON.stringify(completion.content);
+    if (completion.tool_calls) {
+      attrs["gen_ai.completion.0.tool_calls"] = JSON.stringify(
+        completion.tool_calls
+      );
+    }
+  }
+  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);
-    await fetch(`${baseUrl2}/v1/traces`, {
+    const response = await fetch(url, {
       method: "POST",
       headers: {
         Authorization: `Bearer ${apiKey2}`,
@@ -1132,8 +1176,14 @@ async function sendTrace(trace) {
       signal: controller.signal
     });
     clearTimeout(timeoutId);
-    log2("\u{1F4E4} Trace sent:", trace.name, trace.model);
-  } catch {
+    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) {
@@ -1151,16 +1201,30 @@ function wrapOpenAI(client) {
       promptCtx = getPromptContext2();
     } catch {
     }
+    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceId = traceCtx?.traceId || generateHexId(32);
+    const spanId = generateHexId(16);
+    const parentSpanId = traceCtx?.parentSpanId;
     const params = args[0] || {};
     const startTime = Date.now();
     try {
       const response = await originalCreate(...args);
       const endTime = Date.now();
+      const attributes = captureContent ? messagesToOtelAttributes(
+        params?.messages,
+        response?.choices?.[0]?.message,
+        response?.model || params?.model,
+        response?.id
+      ) : void 0;
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
         customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
         name: "chat.completions.create",
+        kind: "llm",
         model: response?.model || params?.model,
         start_time: new Date(startTime).toISOString(),
         end_time: new Date(endTime).toISOString(),
@@ -1169,8 +1233,7 @@ function wrapOpenAI(client) {
         prompt_tokens: response?.usage?.prompt_tokens,
         completion_tokens: response?.usage?.completion_tokens,
         total_tokens: response?.usage?.total_tokens,
-        input: captureContent ? JSON.stringify(params?.messages) : void 0,
-        output: captureContent ? response?.choices?.[0]?.message?.content : void 0,
+        attributes,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1180,17 +1243,31 @@ function wrapOpenAI(client) {
       return response;
     } catch (error) {
       const endTime = Date.now();
+      const attributes = captureContent ? messagesToOtelAttributes(
+        params?.messages,
+        void 0,
+        params?.model,
+        void 0
+      ) : void 0;
+      if (attributes) {
+        attributes["error.message"] = error?.message;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
         customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
         name: "chat.completions.create",
+        kind: "llm",
         model: params?.model,
         start_time: new Date(startTime).toISOString(),
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "ERROR",
         error_message: error?.message,
+        attributes,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1215,16 +1292,33 @@ function wrapAnthropic(client) {
       promptCtx = getPromptContext2();
     } catch {
     }
+    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceId = traceCtx?.traceId || generateHexId(32);
+    const spanId = generateHexId(16);
+    const parentSpanId = traceCtx?.parentSpanId;
     const params = args[0] || {};
     const startTime = Date.now();
     try {
       const response = await originalCreate(...args);
       const endTime = Date.now();
+      const attributes = captureContent ? messagesToOtelAttributes(
+        params?.messages,
+        { role: "assistant", content: response?.content?.[0]?.text || "" },
+        response?.model || params?.model,
+        response?.id
+      ) : void 0;
+      if (attributes && params?.system) {
+        attributes["gen_ai.system_prompt"] = params.system;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
         customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
         name: "messages.create",
+        kind: "llm",
         model: response?.model || params?.model,
         start_time: new Date(startTime).toISOString(),
         end_time: new Date(endTime).toISOString(),
@@ -1233,8 +1327,7 @@ 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),
-        input: captureContent ? JSON.stringify(params?.messages) : void 0,
-        output: captureContent ? response?.content?.[0]?.text : void 0,
+        attributes,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1244,17 +1337,34 @@ function wrapAnthropic(client) {
       return response;
     } catch (error) {
       const endTime = Date.now();
+      const attributes = captureContent ? messagesToOtelAttributes(
+        params?.messages,
+        void 0,
+        params?.model,
+        void 0
+      ) : void 0;
+      if (attributes) {
+        attributes["error.message"] = error?.message;
+        if (params?.system) {
+          attributes["gen_ai.system_prompt"] = params.system;
+        }
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
         customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
         name: "messages.create",
+        kind: "llm",
         model: params?.model,
         start_time: new Date(startTime).toISOString(),
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "ERROR",
         error_message: error?.message,
+        attributes,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1279,18 +1389,47 @@ function wrapGoogleAI(model) {
       promptCtx = getPromptContext2();
     } catch {
     }
+    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceId = traceCtx?.traceId || generateHexId(32);
+    const spanId = generateHexId(16);
+    const parentSpanId = traceCtx?.parentSpanId;
     const startTime = Date.now();
     try {
       const response = await originalGenerate(...args);
       const endTime = Date.now();
       const result = response?.response;
       const usage = result?.usageMetadata;
+      const modelName = model?.model || "gemini";
+      const attributes = {};
+      if (captureContent) {
+        attributes["gen_ai.request.model"] = modelName;
+        attributes["gen_ai.response.model"] = modelName;
+        const input = args[0];
+        if (typeof input === "string") {
+          attributes["gen_ai.prompt.0.role"] = "user";
+          attributes["gen_ai.prompt.0.content"] = input;
+        } else if (input?.contents) {
+          input.contents.forEach((content, i) => {
+            attributes[`gen_ai.prompt.${i}.role`] = content.role || "user";
+            attributes[`gen_ai.prompt.${i}.content`] = content.parts?.[0]?.text || JSON.stringify(content.parts);
+          });
+        }
+        const outputText = result?.text?.();
+        if (outputText) {
+          attributes["gen_ai.completion.0.role"] = "assistant";
+          attributes["gen_ai.completion.0.content"] = outputText;
+        }
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
         customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
         name: "generateContent",
-        model: model?.model || "gemini",
+        kind: "llm",
+        model: modelName,
         start_time: new Date(startTime).toISOString(),
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
@@ -1298,8 +1437,7 @@ function wrapGoogleAI(model) {
         prompt_tokens: usage?.promptTokenCount,
         completion_tokens: usage?.candidatesTokenCount,
         total_tokens: usage?.totalTokenCount,
-        input: captureContent ? JSON.stringify(args[0]) : void 0,
-        output: captureContent ? result?.text?.() : void 0,
+        attributes: captureContent ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1309,17 +1447,33 @@ function wrapGoogleAI(model) {
       return response;
     } catch (error) {
       const endTime = Date.now();
+      const modelName = model?.model || "gemini";
+      const attributes = {};
+      if (captureContent) {
+        attributes["gen_ai.request.model"] = modelName;
+        attributes["error.message"] = error?.message;
+        const input = args[0];
+        if (typeof input === "string") {
+          attributes["gen_ai.prompt.0.role"] = "user";
+          attributes["gen_ai.prompt.0.content"] = input;
+        }
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
         customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
         name: "generateContent",
-        model: model?.model || "gemini",
+        kind: "llm",
+        model: modelName,
         start_time: new Date(startTime).toISOString(),
         end_time: new Date(endTime).toISOString(),
         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,
@@ -1331,6 +1485,414 @@ 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
+  };
+}
+function createGenerateTextWrapper(aiModule) {
+  return async (...args) => {
+    const ctx = sessionStorage.getStore() || fallbackSession;
+    if (!ctx || !initialized2) {
+      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 traceId = traceCtx?.traceId || generateHexId(32);
+    const spanId = generateHexId(16);
+    const parentSpanId = traceCtx?.parentSpanId;
+    const params = args[0] || {};
+    const startTime = Date.now();
+    try {
+      const result = await aiModule.generateText(...args);
+      const endTime = Date.now();
+      const modelId = result?.response?.modelId || params?.model?.modelId || String(params?.model || "unknown");
+      const attributes = {};
+      if (captureContent) {
+        attributes["gen_ai.request.model"] = modelId;
+        attributes["gen_ai.response.model"] = modelId;
+        if (params?.prompt) {
+          attributes["gen_ai.prompt.0.role"] = "user";
+          attributes["gen_ai.prompt.0.content"] = params.prompt;
+        }
+        if (params?.messages) {
+          params.messages.forEach((msg, i) => {
+            attributes[`gen_ai.prompt.${i}.role`] = msg.role;
+            attributes[`gen_ai.prompt.${i}.content`] = typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content);
+          });
+        }
+        if (result?.text) {
+          attributes["gen_ai.completion.0.role"] = "assistant";
+          attributes["gen_ai.completion.0.content"] = result.text;
+        }
+        if (result?.response?.id) {
+          attributes["gen_ai.response.id"] = result.response.id;
+        }
+      }
+      sendTrace({
+        config_key: ctx.configKey,
+        session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: 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: 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
+      }).catch(() => {
+      });
+      return result;
+    } catch (error) {
+      const endTime = Date.now();
+      const modelId = params?.model?.modelId || String(params?.model || "unknown");
+      sendTrace({
+        config_key: ctx.configKey,
+        session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
+        name: "generateText",
+        kind: "llm",
+        model: modelId,
+        start_time: new Date(startTime).toISOString(),
+        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
+      }).catch(() => {
+      });
+      throw error;
+    }
+  };
+}
+function createStreamTextWrapper(aiModule) {
+  return async (...args) => {
+    const ctx = sessionStorage.getStore() || fallbackSession;
+    const params = args[0] || {};
+    const startTime = Date.now();
+    const result = await aiModule.streamText(...args);
+    if (!ctx || !initialized2) {
+      return result;
+    }
+    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    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) => {
+        const endTime = Date.now();
+        log2("\u{1F4CA} streamText usage:", JSON.stringify(usage, null, 2));
+        const attributes = {};
+        if (captureContent) {
+          attributes["gen_ai.request.model"] = modelId;
+          if (params?.prompt) {
+            attributes["gen_ai.prompt.0.role"] = "user";
+            attributes["gen_ai.prompt.0.content"] = params.prompt;
+          }
+        }
+        if (firstTokenTime) {
+          attributes["gen_ai.time_to_first_token_ms"] = firstTokenTime - startTime;
+        }
+        const tracePayload = {
+          config_key: ctx.configKey,
+          session_id: ctx.sessionId,
+          customer_id: ctx.customerId,
+          trace_id: traceId,
+          span_id: spanId,
+          parent_span_id: parentSpanId,
+          name: "streamText",
+          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,
+          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
+        };
+        sendTrace(tracePayload).catch(() => {
+        });
+      }).catch((error) => {
+        const endTime = Date.now();
+        log2("\u274C streamText error:", error?.message);
+        sendTrace({
+          config_key: ctx.configKey,
+          session_id: ctx.sessionId,
+          customer_id: ctx.customerId,
+          trace_id: traceId,
+          span_id: spanId,
+          parent_span_id: parentSpanId,
+          name: "streamText",
+          kind: "llm",
+          model: modelId,
+          start_time: new Date(startTime).toISOString(),
+          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
+        }).catch(() => {
+        });
+      });
+    }
+    if (result?.textStream) {
+      const originalTextStream = result.textStream;
+      const wrappedTextStream = (async function* () {
+        for await (const chunk of originalTextStream) {
+          if (!firstTokenTime) {
+            firstTokenTime = Date.now();
+            log2("\u23F1\uFE0F Time to first token:", firstTokenTime - startTime, "ms");
+          }
+          yield chunk;
+        }
+      })();
+      return new Proxy(result, {
+        get(target, prop) {
+          if (prop === "textStream") {
+            return wrappedTextStream;
+          }
+          return target[prop];
+        }
+      });
+    }
+    return result;
+  };
+}
+function createGenerateObjectWrapper(aiModule) {
+  return async (...args) => {
+    const ctx = sessionStorage.getStore() || fallbackSession;
+    if (!ctx || !initialized2) {
+      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 traceId = traceCtx?.traceId || generateHexId(32);
+    const spanId = generateHexId(16);
+    const parentSpanId = traceCtx?.parentSpanId;
+    const params = args[0] || {};
+    const startTime = Date.now();
+    try {
+      const result = await aiModule.generateObject(...args);
+      const endTime = Date.now();
+      const modelId = result?.response?.modelId || params?.model?.modelId || String(params?.model || "unknown");
+      const attributes = {};
+      if (captureContent) {
+        attributes["gen_ai.request.model"] = modelId;
+        attributes["gen_ai.response.model"] = modelId;
+        if (result?.object) {
+          attributes["gen_ai.completion.0.role"] = "assistant";
+          attributes["gen_ai.completion.0.content"] = JSON.stringify(
+            result.object
+          );
+        }
+      }
+      sendTrace({
+        config_key: ctx.configKey,
+        session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
+        name: "generateObject",
+        kind: "llm",
+        model: modelId,
+        start_time: new Date(startTime).toISOString(),
+        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
+      }).catch(() => {
+      });
+      return result;
+    } catch (error) {
+      const endTime = Date.now();
+      const modelId = params?.model?.modelId || String(params?.model || "unknown");
+      sendTrace({
+        config_key: ctx.configKey,
+        session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
+        name: "generateObject",
+        kind: "llm",
+        model: modelId,
+        start_time: new Date(startTime).toISOString(),
+        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
+      }).catch(() => {
+      });
+      throw error;
+    }
+  };
+}
+function createStreamObjectWrapper(aiModule) {
+  return async (...args) => {
+    const ctx = sessionStorage.getStore() || fallbackSession;
+    const params = args[0] || {};
+    const startTime = Date.now();
+    const result = await aiModule.streamObject(...args);
+    log2("\u{1F50D} streamObject result keys:", Object.keys(result || {}));
+    if (!ctx || !initialized2) {
+      return result;
+    }
+    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    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) => {
+        const endTime = Date.now();
+        log2("\u{1F4CA} streamObject usage:", JSON.stringify(usage, null, 2));
+        const attributes = {};
+        if (captureContent) {
+          attributes["gen_ai.request.model"] = modelId;
+        }
+        if (firstTokenTime) {
+          attributes["gen_ai.time_to_first_token_ms"] = firstTokenTime - startTime;
+        }
+        sendTrace({
+          config_key: ctx.configKey,
+          session_id: ctx.sessionId,
+          customer_id: ctx.customerId,
+          trace_id: traceId,
+          span_id: spanId,
+          parent_span_id: parentSpanId,
+          name: "streamObject",
+          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: captureContent ? attributes : void 0,
+          prompt_key: promptCtx?.promptKey,
+          prompt_version: promptCtx?.promptVersion,
+          prompt_ab_test_key: promptCtx?.abTestKey,
+          prompt_variant_index: promptCtx?.variantIndex
+        }).catch(() => {
+        });
+      }).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: parentSpanId,
+          name: "streamObject",
+          kind: "llm",
+          model: modelId,
+          start_time: new Date(startTime).toISOString(),
+          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
+        }).catch(() => {
+        });
+      });
+    }
+    if (result?.partialObjectStream) {
+      const originalStream = result.partialObjectStream;
+      const wrappedStream = (async function* () {
+        for await (const chunk of originalStream) {
+          if (!firstTokenTime) {
+            firstTokenTime = Date.now();
+            log2("\u23F1\uFE0F Time to first token:", firstTokenTime - startTime, "ms");
+          }
+          yield chunk;
+        }
+      })();
+      return new Proxy(result, {
+        get(target, prop) {
+          if (prop === "partialObjectStream") {
+            return wrappedStream;
+          }
+          return target[prop];
+        }
+      });
+    }
+    return result;
+  };
+}
 
 // src/models.ts
 var models_exports = {};
@@ -1340,7 +1902,7 @@ __export(models_exports, {
 });
 var import_crypto2 = require("crypto");
 var apiKey3 = null;
-var baseUrl3 = "https://spans.fallom.com";
+var baseUrl3 = "https://configs.fallom.com";
 var initialized3 = false;
 var syncInterval2 = null;
 var debugMode3 = false;
@@ -1354,7 +1916,7 @@ function log3(msg) {
 }
 function init3(options = {}) {
   apiKey3 = options.apiKey || process.env.FALLOM_API_KEY || null;
-  baseUrl3 = options.baseUrl || process.env.FALLOM_BASE_URL || "https://spans.fallom.com";
+  baseUrl3 = options.baseUrl || process.env.FALLOM_CONFIGS_URL || process.env.FALLOM_BASE_URL || "https://configs.fallom.com";
   initialized3 = true;
   if (!apiKey3) {
     return;
@@ -1443,20 +2005,28 @@ async function get2(configKey, sessionId, options = {}) {
   const { version, fallback, debug = false } = options;
   debugMode3 = debug;
   ensureInit2();
-  log3(`get() called: configKey=${configKey}, sessionId=${sessionId}, fallback=${fallback}`);
+  log3(
+    `get() called: configKey=${configKey}, sessionId=${sessionId}, fallback=${fallback}`
+  );
   try {
     let configData = configCache.get(configKey);
-    log3(`Cache lookup for '${configKey}': ${configData ? "found" : "not found"}`);
+    log3(
+      `Cache lookup for '${configKey}': ${configData ? "found" : "not found"}`
+    );
     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"}`);
+      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}`);
+        console.warn(
+          `[Fallom WARNING] Config '${configKey}' not found, using fallback model: ${fallback}`
+        );
         return returnWithTrace(configKey, sessionId, fallback, 0);
       }
       throw new Error(
@@ -1472,7 +2042,9 @@ async function get2(configKey, sessionId, options = {}) {
       }
       if (!config) {
         if (fallback) {
-          console.warn(`[Fallom WARNING] Config '${configKey}' version ${version} not found, using fallback: ${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.`);
@@ -1483,7 +2055,9 @@ async function get2(configKey, sessionId, options = {}) {
       config = configData.versions.get(targetVersion);
       if (!config) {
         if (fallback) {
-          console.warn(`[Fallom WARNING] Config '${configKey}' has no cached version, using fallback: ${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.`);
@@ -1492,7 +2066,11 @@ async function get2(configKey, sessionId, options = {}) {
     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)}`);
+    log3(
+      `Config found! Version: ${configVersion}, Variants: ${JSON.stringify(
+        variants
+      )}`
+    );
     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)`);
@@ -1501,7 +2079,9 @@ async function get2(configKey, sessionId, options = {}) {
     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}`);
+      log3(
+        `Variant ${v.model}: weight=${v.weight}%, range=${oldCumulative}-${cumulative}, hash=${hashVal}, match=${hashVal < cumulative}`
+      );
       if (hashVal < cumulative) {
         assignedModel = v.model;
         break;
@@ -1514,7 +2094,9 @@ async function get2(configKey, sessionId, options = {}) {
       throw e;
     }
     if (fallback) {
-      console.warn(`[Fallom WARNING] Error getting model for '${configKey}': ${e}. Using fallback: ${fallback}`);
+      console.warn(
+        `[Fallom WARNING] Error getting model for '${configKey}': ${e}. Using fallback: ${fallback}`
+      );
       return returnWithTrace(configKey, sessionId, fallback, 0);
     }
     throw e;
@@ -1561,20 +2143,22 @@ init_prompts();
 // src/init.ts
 init_prompts();
 async function init4(options = {}) {
-  const baseUrl4 = options.baseUrl || process.env.FALLOM_BASE_URL || "https://spans.fallom.com";
+  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({
     apiKey: options.apiKey,
-    baseUrl: baseUrl4,
+    baseUrl: tracesUrl,
     captureContent: options.captureContent,
     debug: options.debug
   });
   init3({
     apiKey: options.apiKey,
-    baseUrl: baseUrl4
+    baseUrl: configsUrl
   });
   init({
     apiKey: options.apiKey,
-    baseUrl: baseUrl4
+    baseUrl: promptsUrl
   });
 }