@fallom/trace 0.1.5 → 0.1.10

package/dist/index.mjs

@@ -2,7 +2,7 @@ import {
   __export,
   init,
   prompts_exports
-} from "./chunk-IGJD7GBO.mjs";
+} from "./chunk-6MSTRIK4.mjs";
 
 // src/trace.ts
 var trace_exports = {};
@@ -14,6 +14,7 @@ __export(trace_exports, {
   setSession: () => setSession,
   shutdown: () => shutdown,
   span: () => span,
+  wrapAISDK: () => wrapAISDK,
   wrapAnthropic: () => wrapAnthropic,
   wrapGoogleAI: () => wrapGoogleAI,
   wrapOpenAI: () => wrapOpenAI
@@ -649,7 +650,7 @@ var Resource = (
 var sessionStorage = new AsyncLocalStorage();
 var fallbackSession = null;
 var apiKey = null;
-var baseUrl = "https://spans.fallom.com";
+var baseUrl = "https://traces.fallom.com";
 var initialized = false;
 var captureContent = true;
 var debugMode = false;
@@ -692,7 +693,7 @@ async function init2(options = {}) {
   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_BASE_URL || "https://spans.fallom.com";
+  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;
@@ -852,11 +853,47 @@ async function shutdown() {
     initialized = 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 AsyncLocalStorage();
+var fallbackTraceContext = null;
 async function sendTrace(trace) {
+  const url = `${baseUrl}/v1/traces`;
+  log("\u{1F4E4} Sending trace to:", url);
+  log("   Session:", trace.session_id, "Config:", trace.config_key);
   try {
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), 5e3);
-    await fetch(`${baseUrl}/v1/traces`, {
+    const response = await fetch(url, {
       method: "POST",
       headers: {
         Authorization: `Bearer ${apiKey}`,
@@ -866,8 +903,14 @@ async function sendTrace(trace) {
       signal: controller.signal
     });
     clearTimeout(timeoutId);
-    log("\u{1F4E4} Trace sent:", trace.name, trace.model);
-  } 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);
   }
 }
 function wrapOpenAI(client) {
@@ -881,20 +924,34 @@ function wrapOpenAI(client) {
     }
     let promptCtx = null;
     try {
-      const { getPromptContext } = await import("./prompts-67DJ33I4.mjs");
+      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
       promptCtx = getPromptContext();
     } 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(),
@@ -903,8 +960,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,
@@ -914,17 +970,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,
@@ -945,20 +1015,37 @@ function wrapAnthropic(client) {
     }
     let promptCtx = null;
     try {
-      const { getPromptContext } = await import("./prompts-67DJ33I4.mjs");
+      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
       promptCtx = getPromptContext();
     } 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(),
@@ -967,8 +1054,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,
@@ -978,17 +1064,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,
@@ -1009,22 +1112,51 @@ function wrapGoogleAI(model) {
     }
     let promptCtx = null;
     try {
-      const { getPromptContext } = await import("./prompts-67DJ33I4.mjs");
+      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
       promptCtx = getPromptContext();
     } 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,
@@ -1032,8 +1164,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,
@@ -1043,17 +1174,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,
@@ -1065,6 +1212,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 || !initialized) {
+      return aiModule.generateText(...args);
+    }
+    let promptCtx = null;
+    try {
+      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
+      promptCtx = getPromptContext();
+    } 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 || !initialized) {
+      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 } = await import("./prompts-VAN5E3L4.mjs");
+      promptCtx = getPromptContext();
+    } catch {
+    }
+    if (result?.usage) {
+      result.usage.then((usage) => {
+        const endTime = Date.now();
+        log("\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();
+        log("\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();
+            log("\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 || !initialized) {
+      return aiModule.generateObject(...args);
+    }
+    let promptCtx = null;
+    try {
+      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
+      promptCtx = getPromptContext();
+    } 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);
+    log("\u{1F50D} streamObject result keys:", Object.keys(result || {}));
+    if (!ctx || !initialized) {
+      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 } = await import("./prompts-VAN5E3L4.mjs");
+      promptCtx = getPromptContext();
+    } catch {
+    }
+    if (result?.usage) {
+      result.usage.then((usage) => {
+        const endTime = Date.now();
+        log("\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();
+            log("\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 = {};
@@ -1074,7 +1629,7 @@ __export(models_exports, {
 });
 import { createHash } from "crypto";
 var apiKey2 = null;
-var baseUrl2 = "https://spans.fallom.com";
+var baseUrl2 = "https://configs.fallom.com";
 var initialized2 = false;
 var syncInterval = null;
 var debugMode2 = false;
@@ -1088,7 +1643,7 @@ function log2(msg) {
 }
 function init3(options = {}) {
   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_CONFIGS_URL || process.env.FALLOM_BASE_URL || "https://configs.fallom.com";
   initialized2 = true;
   if (!apiKey2) {
     return;
@@ -1177,20 +1732,28 @@ async function get(configKey, sessionId, options = {}) {
   const { version, fallback, debug = false } = options;
   debugMode2 = debug;
   ensureInit();
-  log2(`get() called: configKey=${configKey}, sessionId=${sessionId}, fallback=${fallback}`);
+  log2(
+    `get() called: configKey=${configKey}, sessionId=${sessionId}, fallback=${fallback}`
+  );
   try {
     let configData = configCache.get(configKey);
-    log2(`Cache lookup for '${configKey}': ${configData ? "found" : "not found"}`);
+    log2(
+      `Cache lookup for '${configKey}': ${configData ? "found" : "not found"}`
+    );
     if (!configData) {
       log2("Not in cache, fetching...");
       await fetchConfigs(SYNC_TIMEOUT);
       configData = configCache.get(configKey);
-      log2(`After fetch, cache lookup: ${configData ? "found" : "still not found"}`);
+      log2(
+        `After fetch, cache lookup: ${configData ? "found" : "still not found"}`
+      );
     }
     if (!configData) {
       log2(`Config not found, using fallback: ${fallback}`);
       if (fallback) {
-        console.warn(`[Fallom WARNING] Config '${configKey}' not found, using fallback model: ${fallback}`);
+        console.warn(
+          `[Fallom WARNING] Config '${configKey}' not found, using fallback model: ${fallback}`
+        );
         return returnWithTrace(configKey, sessionId, fallback, 0);
       }
       throw new Error(
@@ -1206,7 +1769,9 @@ async function get(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.`);
@@ -1217,7 +1782,9 @@ async function get(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.`);
@@ -1226,7 +1793,11 @@ async function get(configKey, sessionId, options = {}) {
     const variantsRaw = config.variants;
     const configVersion = config.version || targetVersion;
     const variants = Array.isArray(variantsRaw) ? variantsRaw : Object.values(variantsRaw);
-    log2(`Config found! Version: ${configVersion}, Variants: ${JSON.stringify(variants)}`);
+    log2(
+      `Config found! Version: ${configVersion}, Variants: ${JSON.stringify(
+        variants
+      )}`
+    );
     const hashBytes = createHash("md5").update(sessionId).digest();
     const hashVal = hashBytes.readUInt32BE(0) % 1e6;
     log2(`Session hash: ${hashVal} (out of 1,000,000)`);
@@ -1235,7 +1806,9 @@ async function get(configKey, sessionId, options = {}) {
     for (const v of variants) {
       const oldCumulative = cumulative;
       cumulative += v.weight * 1e4;
-      log2(`Variant ${v.model}: weight=${v.weight}%, range=${oldCumulative}-${cumulative}, hash=${hashVal}, match=${hashVal < cumulative}`);
+      log2(
+        `Variant ${v.model}: weight=${v.weight}%, range=${oldCumulative}-${cumulative}, hash=${hashVal}, match=${hashVal < cumulative}`
+      );
       if (hashVal < cumulative) {
         assignedModel = v.model;
         break;
@@ -1248,7 +1821,9 @@ async function get(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;
@@ -1291,20 +1866,22 @@ async function recordSession(configKey, version, sessionId, model) {
 
 // src/init.ts
 async function init4(options = {}) {
-  const baseUrl3 = 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: baseUrl3,
+    baseUrl: tracesUrl,
     captureContent: options.captureContent,
     debug: options.debug
   });
   init3({
     apiKey: options.apiKey,
-    baseUrl: baseUrl3
+    baseUrl: configsUrl
   });
   init({
     apiKey: options.apiKey,
-    baseUrl: baseUrl3
+    baseUrl: promptsUrl
   });
 }