neatlogs 1.1.4 → 1.1.5

package/dist/index.cjs

@@ -66,7 +66,8 @@ __export(index_exports, {
   trace: () => trace2,
   traceToolAnthropic: () => traceTool2,
   traceToolAzureOpenAI: () => traceTool3,
-  traceToolBedrock: () => traceTool5,
+  traceToolBedrock: () => traceTool6,
+  traceToolGoogleGenAI: () => traceTool5,
   traceToolOpenAI: () => traceTool,
   traceToolVertexAI: () => traceTool4,
   updatePrompt: () => updatePrompt,
@@ -76,6 +77,8 @@ __export(index_exports, {
   wrapBedrock: () => wrapBedrock,
   wrapCallModel: () => wrapCallModel,
   wrapClaudeAgentSDK: () => wrapClaudeAgentSDK,
+  wrapGoogleGenAI: () => wrapGoogleGenAI,
+  wrapGoogleGenAIChat: () => wrapGoogleGenAIChat,
   wrapMastra: () => wrapMastra,
   wrapOpenAI: () => wrapOpenAI,
   wrapOpenRouterAgent: () => wrapOpenRouterAgent,
@@ -5820,7 +5823,7 @@ async function removeTag(name, tag) {
 }
 
 // src/version.ts
-var __version__ = "1.1.4";
+var __version__ = "1.1.5";
 
 // src/init.ts
 var logger13 = getLogger();
@@ -7881,10 +7884,14 @@ function recordError4(span2, err) {
   span2.end();
 }
 
-// src/bedrock.ts
+// src/google-genai.ts
 var import_api15 = require("@opentelemetry/api");
-var TRACER_NAME7 = "neatlogs.bedrock";
-var PROVIDER3 = "bedrock";
+var TRACER_NAME7 = "neatlogs.google_genai";
+var PROVIDER3 = "google";
+var SYSTEM3 = "google_genai";
+function wrapGoogleGenAI(client) {
+  return wrapNamespace5(client, []);
+}
 function traceTool5(name, fn) {
   return async function tracedTool(args) {
     const tracer = getProviderTracer(TRACER_NAME7);
@@ -7914,6 +7921,394 @@ function traceTool5(name, fn) {
     );
   };
 }
+function wrapNamespace5(target, path3) {
+  return new Proxy(target, {
+    get(obj, prop, receiver) {
+      const value = Reflect.get(obj, prop, receiver);
+      if (typeof prop === "symbol" || String(prop).startsWith("_")) return value;
+      const currentPath = [...path3, String(prop)];
+      const pathStr = currentPath.join(".");
+      if (pathStr === "models.generateContent" && typeof value === "function") {
+        return tracedGenerateContent2(value.bind(obj), false);
+      }
+      if (pathStr === "models.generateContentStream" && typeof value === "function") {
+        return tracedGenerateContent2(value.bind(obj), true);
+      }
+      if (pathStr === "models.embedContent" && typeof value === "function") {
+        return tracedEmbedContent2(value.bind(obj));
+      }
+      if (pathStr === "models.countTokens" && typeof value === "function") {
+        return tracedCountTokens2(value.bind(obj));
+      }
+      if (value && typeof value === "object" && !Array.isArray(value) && isNamespace5(currentPath)) {
+        return wrapNamespace5(value, currentPath);
+      }
+      return value;
+    }
+  });
+}
+function isNamespace5(path3) {
+  if (path3.length > 2) return false;
+  return ["models", "chats"].includes(path3[path3.length - 1]);
+}
+function wrapGoogleGenAIChat(chat) {
+  const c = chat;
+  if (!c || c._neatlogsGoogleGenAIPatched) return chat;
+  if (typeof c.sendMessage === "function") {
+    const orig = c.sendMessage.bind(c);
+    c.sendMessage = (params, ...rest) => tracedChatSend2(orig, c, params, rest, false);
+  }
+  if (typeof c.sendMessageStream === "function") {
+    const orig = c.sendMessageStream.bind(c);
+    c.sendMessageStream = (params, ...rest) => tracedChatSend2(orig, c, params, rest, true);
+  }
+  try {
+    Object.defineProperty(c, "_neatlogsGoogleGenAIPatched", { value: true, enumerable: false, configurable: true });
+  } catch {
+    c._neatlogsGoogleGenAIPatched = true;
+  }
+  return chat;
+}
+function tracedChatSend2(original, chat, params, rest, isStream) {
+  const tracer = getProviderTracer(TRACER_NAME7);
+  const model = chat?.model ?? chat?.modelVersion ?? "";
+  const span2 = tracer.startSpan("google_genai.chat.send_message", {
+    attributes: {
+      "neatlogs.span.kind": "LLM",
+      "neatlogs.llm.provider": PROVIDER3,
+      "neatlogs.llm.system": SYSTEM3,
+      "neatlogs.llm.model_name": model,
+      "neatlogs.llm.is_streaming": isStream
+    }
+  }, import_api15.context.active());
+  const message = params?.message ?? params;
+  span2.setAttribute("neatlogs.llm.input_messages.0.role", "user");
+  span2.setAttribute(
+    "neatlogs.llm.input_messages.0.content",
+    typeof message === "string" ? message : safeStringify7(message)
+  );
+  const ctx = import_api15.trace.setSpan(import_api15.context.active(), span2);
+  const result = import_api15.context.with(ctx, () => original(params, ...rest));
+  return Promise.resolve(result).then(
+    (response) => {
+      if (isStream) return wrapStream2(response, span2);
+      finalizeResponse2(span2, response);
+      return response;
+    },
+    (err) => {
+      recordError5(span2, err);
+      throw err;
+    }
+  );
+}
+function tracedEmbedContent2(original) {
+  return function(opts, ...rest) {
+    const tracer = getProviderTracer(TRACER_NAME7);
+    const span2 = tracer.startSpan("google_genai.models.embed_content", {
+      attributes: {
+        "neatlogs.span.kind": "EMBEDDING",
+        "neatlogs.llm.provider": PROVIDER3,
+        "neatlogs.embedding.model_name": opts?.model ?? "",
+        "neatlogs.embedding.text": safeStringify7(opts?.contents ?? "")
+      }
+    }, import_api15.context.active());
+    const ctx = import_api15.trace.setSpan(import_api15.context.active(), span2);
+    const result = import_api15.context.with(ctx, () => original(opts, ...rest));
+    return Promise.resolve(result).then(
+      (response) => {
+        const embeddings = response?.embeddings;
+        if (Array.isArray(embeddings)) {
+          span2.setAttribute("neatlogs.embedding.count", embeddings.length);
+          const vals = embeddings[0]?.values;
+          if (Array.isArray(vals)) span2.setAttribute("neatlogs.embedding.dimensions", vals.length);
+        }
+        span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+        span2.end();
+        return response;
+      },
+      (err) => {
+        recordError5(span2, err);
+        throw err;
+      }
+    );
+  };
+}
+function tracedCountTokens2(original) {
+  return function(opts, ...rest) {
+    const tracer = getProviderTracer(TRACER_NAME7);
+    const span2 = tracer.startSpan("google_genai.models.count_tokens", {
+      attributes: {
+        "neatlogs.span.kind": "LLM",
+        "neatlogs.llm.provider": PROVIDER3,
+        "neatlogs.llm.task": "count_tokens",
+        "neatlogs.llm.model_name": opts?.model ?? ""
+      }
+    }, import_api15.context.active());
+    const ctx = import_api15.trace.setSpan(import_api15.context.active(), span2);
+    const result = import_api15.context.with(ctx, () => original(opts, ...rest));
+    return Promise.resolve(result).then(
+      (response) => {
+        if (response?.totalTokens != null) span2.setAttribute("neatlogs.llm.token_count.prompt", response.totalTokens);
+        span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+        span2.end();
+        return response;
+      },
+      (err) => {
+        recordError5(span2, err);
+        throw err;
+      }
+    );
+  };
+}
+function tracedGenerateContent2(original, isStream) {
+  return function(opts, ...rest) {
+    const tracer = getProviderTracer(TRACER_NAME7);
+    const model = opts?.model ?? "";
+    const span2 = tracer.startSpan("google_genai.models.generate_content", {
+      attributes: {
+        "neatlogs.span.kind": "LLM",
+        "neatlogs.llm.provider": PROVIDER3,
+        "neatlogs.llm.system": SYSTEM3,
+        "neatlogs.llm.model_name": model,
+        "neatlogs.llm.is_streaming": isStream
+      }
+    }, import_api15.context.active());
+    setInputAttributes2(span2, opts);
+    const ctx = import_api15.trace.setSpan(import_api15.context.active(), span2);
+    const result = import_api15.context.with(ctx, () => original(opts, ...rest));
+    return Promise.resolve(result).then(
+      (response) => {
+        if (isStream) {
+          return wrapStream2(response, span2);
+        }
+        finalizeResponse2(span2, response);
+        return response;
+      },
+      (err) => {
+        recordError5(span2, err);
+        throw err;
+      }
+    );
+  };
+}
+function setInputAttributes2(span2, opts) {
+  let idx = 0;
+  const config = opts?.config;
+  const systemInstruction = config?.systemInstruction ?? config?.system_instruction;
+  if (systemInstruction) {
+    span2.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, "system");
+    span2.setAttribute(
+      `neatlogs.llm.input_messages.${idx}.content`,
+      typeof systemInstruction === "string" ? systemInstruction : safeStringify7(systemInstruction)
+    );
+    idx++;
+  }
+  const contents = opts?.contents;
+  if (typeof contents === "string") {
+    span2.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, "user");
+    span2.setAttribute(`neatlogs.llm.input_messages.${idx}.content`, contents);
+  } else if (Array.isArray(contents)) {
+    for (const item of contents) {
+      if (typeof item === "string") {
+        span2.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, "user");
+        span2.setAttribute(`neatlogs.llm.input_messages.${idx}.content`, item);
+        idx++;
+      } else if (item && typeof item === "object") {
+        const role = item.role ?? "user";
+        span2.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, role);
+        const parts = item.parts ?? [];
+        const textParts = [];
+        for (const part of parts) {
+          if (typeof part === "string") textParts.push(part);
+          else if (part?.text) textParts.push(part.text);
+        }
+        span2.setAttribute(
+          `neatlogs.llm.input_messages.${idx}.content`,
+          textParts.length ? textParts.join("\n") : safeStringify7(parts)
+        );
+        idx++;
+      }
+    }
+  } else if (contents) {
+    span2.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, "user");
+    span2.setAttribute(`neatlogs.llm.input_messages.${idx}.content`, safeStringify7(contents));
+  }
+  const tools = config?.tools;
+  if (Array.isArray(tools)) {
+    let t = 0;
+    for (const tool of tools) {
+      const decls = tool?.functionDeclarations ?? tool?.function_declarations ?? [];
+      for (const fn of decls) {
+        span2.setAttribute(`neatlogs.llm.tools.${t}.name`, fn?.name ?? "");
+        if (fn?.description) span2.setAttribute(`neatlogs.llm.tools.${t}.description`, fn.description);
+        if (fn?.parameters) span2.setAttribute(`neatlogs.llm.tools.${t}.input_schema`, safeStringify7(fn.parameters));
+        t++;
+      }
+    }
+  }
+  if (config) {
+    if (config.temperature != null) span2.setAttribute("neatlogs.llm.temperature", config.temperature);
+    if (config.topP != null) span2.setAttribute("neatlogs.llm.top_p", config.topP);
+    if (config.topK != null) span2.setAttribute("neatlogs.llm.top_k", config.topK);
+    const maxTokens = config.maxOutputTokens ?? config.max_output_tokens;
+    if (maxTokens != null) span2.setAttribute("neatlogs.llm.max_tokens", maxTokens);
+    const params = {};
+    if (config.temperature != null) params.temperature = config.temperature;
+    if (config.topP != null) params.top_p = config.topP;
+    if (config.topK != null) params.top_k = config.topK;
+    if (maxTokens != null) params.max_tokens = maxTokens;
+    if (config.frequencyPenalty != null) params.frequency_penalty = config.frequencyPenalty;
+    if (config.presencePenalty != null) params.presence_penalty = config.presencePenalty;
+    if (Object.keys(params).length > 0) {
+      span2.setAttribute("neatlogs.llm.invocation_parameters", JSON.stringify(params));
+    }
+  }
+}
+function wrapStream2(stream, span2) {
+  const chunks = [];
+  const originalAsyncIterator = stream?.[Symbol.asyncIterator]?.bind(stream);
+  if (!originalAsyncIterator) {
+    finalizeStreamChunks4(span2, chunks);
+    return stream;
+  }
+  const wrapped = Object.create(Object.getPrototypeOf(stream));
+  Object.assign(wrapped, stream);
+  wrapped[Symbol.asyncIterator] = function() {
+    const iterator = originalAsyncIterator();
+    return {
+      async next() {
+        try {
+          const result = await iterator.next();
+          if (result.done) {
+            finalizeStreamChunks4(span2, chunks);
+            return result;
+          }
+          chunks.push(result.value);
+          return result;
+        } catch (err) {
+          recordError5(span2, err);
+          throw err;
+        }
+      },
+      async return(value) {
+        finalizeStreamChunks4(span2, chunks);
+        return iterator.return?.(value) ?? { done: true, value: void 0 };
+      },
+      async throw(err) {
+        recordError5(span2, err);
+        return iterator.throw?.(err) ?? { done: true, value: void 0 };
+      }
+    };
+  };
+  return wrapped;
+}
+function finalizeStreamChunks4(span2, chunks) {
+  const textParts = [];
+  let finishReason = "";
+  let usage = null;
+  for (const chunk of chunks) {
+    for (const candidate of chunk?.candidates ?? []) {
+      for (const part of candidate?.content?.parts ?? []) {
+        if (part?.text && !part?.thought) textParts.push(part.text);
+      }
+      if (candidate?.finishReason) finishReason = candidate.finishReason;
+    }
+    if (chunk?.usageMetadata) usage = chunk.usageMetadata;
+  }
+  const fullText = textParts.join("");
+  if (fullText) {
+    span2.setAttribute("neatlogs.llm.output_messages.0.role", "assistant");
+    span2.setAttribute("neatlogs.llm.output_messages.0.content", fullText);
+  }
+  if (finishReason) span2.setAttribute("neatlogs.llm.finish_reason", String(finishReason));
+  setUsage2(span2, usage);
+  span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+  span2.end();
+}
+function finalizeResponse2(span2, response) {
+  const textParts = [];
+  let toolIdx = 0;
+  for (const candidate of response?.candidates ?? []) {
+    for (const part of candidate?.content?.parts ?? []) {
+      if (part?.text && !part?.thought) {
+        textParts.push(part.text);
+      } else if (part?.thought && part?.text) {
+        span2.setAttribute("neatlogs.llm.output_messages.0.thinking", part.text);
+      } else if (part?.functionCall) {
+        const fc = part.functionCall;
+        span2.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.name`, fc?.name ?? "");
+        span2.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.arguments`, safeStringify7(fc?.args ?? {}));
+        toolIdx++;
+      }
+    }
+    if (candidate?.finishReason) span2.setAttribute("neatlogs.llm.finish_reason", String(candidate.finishReason));
+  }
+  if (textParts.length) {
+    span2.setAttribute("neatlogs.llm.output_messages.0.role", "assistant");
+    span2.setAttribute("neatlogs.llm.output_messages.0.content", textParts.join(""));
+  }
+  setUsage2(span2, response?.usageMetadata);
+  span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+  span2.end();
+}
+function setUsage2(span2, usage) {
+  if (!usage) return;
+  if (usage.promptTokenCount != null) span2.setAttribute("neatlogs.llm.token_count.prompt", usage.promptTokenCount);
+  if (usage.candidatesTokenCount != null) span2.setAttribute("neatlogs.llm.token_count.completion", usage.candidatesTokenCount);
+  if (usage.totalTokenCount != null) span2.setAttribute("neatlogs.llm.token_count.total", usage.totalTokenCount);
+  if (usage.cachedContentTokenCount != null) span2.setAttribute("neatlogs.llm.token_count.cache_read", usage.cachedContentTokenCount);
+  if (usage.thoughtsTokenCount != null) span2.setAttribute("neatlogs.llm.token_count.reasoning", usage.thoughtsTokenCount);
+}
+function safeStringify7(value) {
+  try {
+    return typeof value === "string" ? value : JSON.stringify(value);
+  } catch {
+    return "";
+  }
+}
+function recordError5(span2, err) {
+  if (err instanceof Error) {
+    span2.setStatus({ code: import_api15.SpanStatusCode.ERROR, message: err.message });
+    span2.recordException(err);
+  } else {
+    span2.setStatus({ code: import_api15.SpanStatusCode.ERROR, message: String(err) });
+  }
+  span2.end();
+}
+
+// src/bedrock.ts
+var import_api16 = require("@opentelemetry/api");
+var TRACER_NAME8 = "neatlogs.bedrock";
+var PROVIDER4 = "bedrock";
+function traceTool6(name, fn) {
+  return async function tracedTool(args) {
+    const tracer = getProviderTracer(TRACER_NAME8);
+    return tracer.startActiveSpan(
+      `tool.${name}`,
+      {
+        attributes: {
+          "neatlogs.span.kind": "TOOL",
+          "neatlogs.tool.name": name,
+          "input.value": safeStringify8(args)
+        }
+      },
+      import_api16.context.active(),
+      async (span2) => {
+        try {
+          const result = await fn(args);
+          span2.setAttribute("output.value", safeStringify8(result));
+          span2.setStatus({ code: import_api16.SpanStatusCode.OK });
+          return result;
+        } catch (err) {
+          recordError6(span2, err);
+          throw err;
+        } finally {
+          span2.end();
+        }
+      }
+    );
+  };
+}
 function wrapBedrock(client) {
   const c = client;
   if (c._neatlogsBedrockPatched) return client;
@@ -7948,15 +8343,15 @@ function vendorFromModel(modelId) {
   return vendor || "bedrock";
 }
 function startSpan(name, modelId, isStream) {
-  return getProviderTracer(TRACER_NAME7).startSpan(name, {
+  return getProviderTracer(TRACER_NAME8).startSpan(name, {
     attributes: {
       "neatlogs.span.kind": "LLM",
-      "neatlogs.llm.provider": PROVIDER3,
+      "neatlogs.llm.provider": PROVIDER4,
       "neatlogs.llm.system": vendorFromModel(modelId),
       "neatlogs.llm.model_name": String(modelId ?? ""),
       "neatlogs.llm.is_streaming": isStream
     }
-  }, import_api15.context.active());
+  }, import_api16.context.active());
 }
 function tracedConverse(originalSend, command, input, rest, isStream) {
   const span2 = startSpan(
@@ -7965,8 +8360,8 @@ function tracedConverse(originalSend, command, input, rest, isStream) {
     isStream
   );
   setConverseInput(span2, input);
-  const ctx = import_api15.trace.setSpan(import_api15.context.active(), span2);
-  const promise = import_api15.context.with(ctx, () => originalSend(command, ...rest));
+  const ctx = import_api16.trace.setSpan(import_api16.context.active(), span2);
+  const promise = import_api16.context.with(ctx, () => originalSend(command, ...rest));
   return promise.then(
     (response) => {
       if (isStream) {
@@ -7976,7 +8371,7 @@ function tracedConverse(originalSend, command, input, rest, isStream) {
       return response;
     },
     (err) => {
-      recordError5(span2, err);
+      recordError6(span2, err);
       throw err;
     }
   );
@@ -8005,19 +8400,19 @@ function setConverseInput(span2, input) {
     const spec = tools[i]?.toolSpec ?? {};
     if (spec.name) span2.setAttribute(`neatlogs.llm.tools.${i}.name`, spec.name);
     if (spec.description) span2.setAttribute(`neatlogs.llm.tools.${i}.description`, spec.description);
-    if (spec.inputSchema) span2.setAttribute(`neatlogs.llm.tools.${i}.input_schema`, safeStringify7(spec.inputSchema));
+    if (spec.inputSchema) span2.setAttribute(`neatlogs.llm.tools.${i}.input_schema`, safeStringify8(spec.inputSchema));
   }
 }
 function converseBlocksToText(content) {
   if (typeof content === "string") return content;
-  if (!Array.isArray(content)) return safeStringify7(content);
+  if (!Array.isArray(content)) return safeStringify8(content);
   const parts = [];
   for (const block of content) {
     if (block && typeof block === "object") {
       if ("text" in block) parts.push(String(block.text));
-      else if ("toolResult" in block) parts.push(safeStringify7(block.toolResult));
-      else if ("toolUse" in block) parts.push(safeStringify7(block.toolUse));
-      else parts.push(safeStringify7(block));
+      else if ("toolResult" in block) parts.push(safeStringify8(block.toolResult));
+      else if ("toolUse" in block) parts.push(safeStringify8(block.toolUse));
+      else parts.push(safeStringify8(block));
     } else {
       parts.push(String(block));
     }
@@ -8036,7 +8431,7 @@ function finalizeConverse(span2, response) {
       const tu = block.toolUse;
       span2.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.id`, String(tu?.toolUseId ?? ""));
       span2.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.name`, String(tu?.name ?? ""));
-      span2.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.arguments`, safeStringify7(tu?.input ?? {}));
+      span2.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.arguments`, safeStringify8(tu?.input ?? {}));
       toolIdx++;
     }
   }
@@ -8046,7 +8441,7 @@ function finalizeConverse(span2, response) {
   }
   if (response?.stopReason) span2.setAttribute("neatlogs.llm.finish_reason", String(response.stopReason));
   setConverseUsage(span2, response?.usage);
-  span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+  span2.setStatus({ code: import_api16.SpanStatusCode.OK });
   span2.end();
 }
 function setConverseUsage(span2, usage) {
@@ -8060,7 +8455,7 @@ function setConverseUsage(span2, usage) {
 function wrapConverseStream(response, span2) {
   const stream = response?.stream;
   if (!stream || typeof stream[Symbol.asyncIterator] !== "function") {
-    span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+    span2.setStatus({ code: import_api16.SpanStatusCode.OK });
     span2.end();
     return response;
   }
@@ -8099,7 +8494,7 @@ function wrapConverseStream(response, span2) {
             if (ev?.metadata?.usage) usage = ev.metadata.usage;
             return result;
           } catch (err) {
-            recordError5(span2, err);
+            recordError6(span2, err);
             throw err;
           }
         },
@@ -8128,7 +8523,7 @@ function finalizeConverseStream(span2, textParts, toolCalls, finishReason, usage
   }
   if (finishReason) span2.setAttribute("neatlogs.llm.finish_reason", String(finishReason));
   setConverseUsage(span2, usage);
-  span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+  span2.setStatus({ code: import_api16.SpanStatusCode.OK });
   span2.end();
 }
 function isEmbeddingModel(modelId) {
@@ -8140,16 +8535,16 @@ function tracedInvokeModel(originalSend, command, input, rest, isStream) {
   const bodyIn = decodeBody(input?.body);
   let span2;
   if (isEmbedding) {
-    span2 = getProviderTracer(TRACER_NAME7).startSpan("bedrock.invoke_model", {
+    span2 = getProviderTracer(TRACER_NAME8).startSpan("bedrock.invoke_model", {
       attributes: {
         "neatlogs.span.kind": "EMBEDDING",
-        "neatlogs.llm.provider": PROVIDER3,
+        "neatlogs.llm.provider": PROVIDER4,
         "neatlogs.embedding.model_name": String(input?.modelId ?? "")
       }
-    }, import_api15.context.active());
+    }, import_api16.context.active());
     const text = bodyIn?.inputText ?? bodyIn?.texts ?? bodyIn?.input_text;
     if (text) {
-      span2.setAttribute("neatlogs.embedding.text", typeof text === "string" ? text : safeStringify7(text));
+      span2.setAttribute("neatlogs.embedding.text", typeof text === "string" ? text : safeStringify8(text));
     }
   } else {
     span2 = startSpan(
@@ -8159,8 +8554,8 @@ function tracedInvokeModel(originalSend, command, input, rest, isStream) {
     );
     setInvokeInput(span2, vendor, bodyIn);
   }
-  const ctx = import_api15.trace.setSpan(import_api15.context.active(), span2);
-  const promise = import_api15.context.with(ctx, () => originalSend(command, ...rest));
+  const ctx = import_api16.trace.setSpan(import_api16.context.active(), span2);
+  const promise = import_api16.context.with(ctx, () => originalSend(command, ...rest));
   return promise.then(
     (response) => {
       if (isStream) {
@@ -8173,13 +8568,13 @@ function tracedInvokeModel(originalSend, command, input, rest, isStream) {
           finalizeInvoke(span2, vendor, decodeBody(response?.body));
         }
       } catch {
-        span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+        span2.setStatus({ code: import_api16.SpanStatusCode.OK });
         span2.end();
       }
       return response;
     },
     (err) => {
-      recordError5(span2, err);
+      recordError6(span2, err);
       throw err;
     }
   );
@@ -8198,7 +8593,7 @@ function finalizeInvokeEmbedding(span2, body) {
     span2.setAttribute("neatlogs.llm.token_count.prompt", body.inputTextTokenCount);
     span2.setAttribute("neatlogs.embedding.token_count", body.inputTextTokenCount);
   }
-  span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+  span2.setStatus({ code: import_api16.SpanStatusCode.OK });
   span2.end();
 }
 function decodeBody(body) {
@@ -8220,7 +8615,7 @@ function setInvokeInput(span2, vendor, body) {
     span2.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, "system");
     span2.setAttribute(
       `neatlogs.llm.input_messages.${idx}.content`,
-      typeof body.system === "string" ? body.system : safeStringify7(body.system)
+      typeof body.system === "string" ? body.system : safeStringify8(body.system)
     );
     idx++;
   }
@@ -8229,7 +8624,7 @@ function setInvokeInput(span2, vendor, body) {
       span2.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, msg?.role ?? "user");
       span2.setAttribute(
         `neatlogs.llm.input_messages.${idx}.content`,
-        typeof msg?.content === "string" ? msg.content : safeStringify7(msg?.content)
+        typeof msg?.content === "string" ? msg.content : safeStringify8(msg?.content)
       );
       idx++;
     }
@@ -8264,7 +8659,7 @@ function finalizeInvoke(span2, vendor, body) {
         if (b?.type === "tool_use") {
           span2.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.id`, String(b.id ?? ""));
           span2.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.name`, String(b.name ?? ""));
-          span2.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.arguments`, safeStringify7(b.input ?? {}));
+          span2.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.arguments`, safeStringify8(b.input ?? {}));
           toolIdx++;
         }
       }
@@ -8306,13 +8701,13 @@ function finalizeInvoke(span2, vendor, body) {
     span2.setAttribute("neatlogs.llm.token_count.total", promptTokens + completionTokens);
   }
   if (finishReason) span2.setAttribute("neatlogs.llm.finish_reason", String(finishReason));
-  span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+  span2.setStatus({ code: import_api16.SpanStatusCode.OK });
   span2.end();
 }
 function wrapInvokeStream(response, span2, vendor) {
   const body = response?.body;
   if (!body || typeof body[Symbol.asyncIterator] !== "function") {
-    span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+    span2.setStatus({ code: import_api16.SpanStatusCode.OK });
     span2.end();
     return response;
   }
@@ -8330,7 +8725,7 @@ function wrapInvokeStream(response, span2, vendor) {
     if (promptTokens != null) span2.setAttribute("neatlogs.llm.token_count.prompt", promptTokens);
     if (completionTokens != null) span2.setAttribute("neatlogs.llm.token_count.completion", completionTokens);
     if (finishReason) span2.setAttribute("neatlogs.llm.finish_reason", String(finishReason));
-    span2.setStatus({ code: import_api15.SpanStatusCode.OK });
+    span2.setStatus({ code: import_api16.SpanStatusCode.OK });
     span2.end();
   };
   const wrappedBody = {
@@ -8360,7 +8755,7 @@ function wrapInvokeStream(response, span2, vendor) {
             }
             return result;
           } catch (err) {
-            recordError5(span2, err);
+            recordError6(span2, err);
             throw err;
           }
         },
@@ -8374,26 +8769,26 @@ function wrapInvokeStream(response, span2, vendor) {
   response.body = wrappedBody;
   return response;
 }
-function safeStringify7(value) {
+function safeStringify8(value) {
   try {
     return typeof value === "string" ? value : JSON.stringify(value);
   } catch {
     return "";
   }
 }
-function recordError5(span2, err) {
+function recordError6(span2, err) {
   if (err instanceof Error) {
-    span2.setStatus({ code: import_api15.SpanStatusCode.ERROR, message: err.message });
+    span2.setStatus({ code: import_api16.SpanStatusCode.ERROR, message: err.message });
     span2.recordException(err);
   } else {
-    span2.setStatus({ code: import_api15.SpanStatusCode.ERROR, message: String(err) });
+    span2.setStatus({ code: import_api16.SpanStatusCode.ERROR, message: String(err) });
   }
   span2.end();
 }
 
 // src/langchain.ts
-var import_api16 = require("@opentelemetry/api");
-var TRACER_NAME8 = "neatlogs.langchain";
+var import_api17 = require("@opentelemetry/api");
+var TRACER_NAME9 = "neatlogs.langchain";
 function langchainHandler(opts) {
   return new NeatlogsCallbackHandler(opts);
 }
@@ -8417,10 +8812,10 @@ var NeatlogsCallbackHandler = class {
   _startCtx(parentRunId, runId, kind) {
     const parentSpan = parentRunId ? this._spans.get(parentRunId) : void 0;
     if (parentSpan) {
-      return import_api16.trace.setSpan(import_api16.context.active(), parentSpan);
+      return import_api17.trace.setSpan(import_api17.context.active(), parentSpan);
     }
-    const tracer = import_api16.trace.getTracer(TRACER_NAME8);
-    const { root, ctx } = maybeOpenAutoRoot(tracer, kind, import_api16.context.active());
+    const tracer = import_api17.trace.getTracer(TRACER_NAME9);
+    const { root, ctx } = maybeOpenAutoRoot(tracer, kind, import_api17.context.active());
     if (root) this._autoRoots.set(runId, root);
     return ctx;
   }
@@ -8433,7 +8828,7 @@ var NeatlogsCallbackHandler = class {
   }
   // --- Chain/Graph callbacks ---
   async handleChainStart(serialized, inputs, runId, parentRunId, tags, metadata) {
-    const tracer = import_api16.trace.getTracer(TRACER_NAME8);
+    const tracer = import_api17.trace.getTracer(TRACER_NAME9);
     const name = serialized?.name ?? serialized?.id?.at(-1) ?? "chain";
     const parentCtx = this._startCtx(parentRunId, runId, "chain");
     const attrs = {
@@ -8441,7 +8836,7 @@ var NeatlogsCallbackHandler = class {
       "neatlogs.chain.name": name
     };
     if (this._workflowName) attrs["neatlogs.workflow.name"] = this._workflowName;
-    if (inputs) attrs["input.value"] = safeStringify8(inputs);
+    if (inputs) attrs["input.value"] = safeStringify9(inputs);
     if (tags?.length) attrs["neatlogs.tags"] = tags.join(",");
     const span2 = tracer.startSpan(`langchain.chain.${name}`, { attributes: attrs }, parentCtx);
     this._spans.set(runId, span2);
@@ -8449,8 +8844,8 @@ var NeatlogsCallbackHandler = class {
   async handleChainEnd(outputs, runId) {
     const span2 = this._spans.get(runId);
     if (!span2) return;
-    if (outputs) span2.setAttribute("output.value", safeStringify8(outputs));
-    span2.setStatus({ code: import_api16.SpanStatusCode.OK });
+    if (outputs) span2.setAttribute("output.value", safeStringify9(outputs));
+    span2.setStatus({ code: import_api17.SpanStatusCode.OK });
     span2.end();
     this._spans.delete(runId);
     this._endAutoRoot(runId);
@@ -8458,7 +8853,7 @@ var NeatlogsCallbackHandler = class {
   async handleChainError(error, runId) {
     const span2 = this._spans.get(runId);
     if (!span2) return;
-    span2.setStatus({ code: import_api16.SpanStatusCode.ERROR, message: error.message });
+    span2.setStatus({ code: import_api17.SpanStatusCode.ERROR, message: error.message });
     span2.recordException(error);
     span2.end();
     this._spans.delete(runId);
@@ -8466,7 +8861,7 @@ var NeatlogsCallbackHandler = class {
   }
   // --- LLM callbacks ---
   async handleLLMStart(serialized, prompts, runId, parentRunId, extraParams) {
-    const tracer = import_api16.trace.getTracer(TRACER_NAME8);
+    const tracer = import_api17.trace.getTracer(TRACER_NAME9);
     const model = serialized?.kwargs?.model_name ?? serialized?.kwargs?.model ?? serialized?.name ?? "";
     const parentCtx = this._startCtx(parentRunId, runId, "llm");
     const attrs = {
@@ -8488,7 +8883,7 @@ var NeatlogsCallbackHandler = class {
     this._spans.set(runId, span2);
   }
   async handleChatModelStart(serialized, messages, runId, parentRunId, extraParams) {
-    const tracer = import_api16.trace.getTracer(TRACER_NAME8);
+    const tracer = import_api17.trace.getTracer(TRACER_NAME9);
     const model = serialized?.kwargs?.model_name ?? serialized?.kwargs?.model ?? serialized?.name ?? "";
     const parentCtx = this._startCtx(parentRunId, runId, "llm");
     const attrs = {
@@ -8500,7 +8895,7 @@ var NeatlogsCallbackHandler = class {
     for (let i = 0; i < flatMessages.length; i++) {
       const msg = flatMessages[i];
       const role = msg?.role ?? msg?._getType?.() ?? msg?.constructor?.name ?? "unknown";
-      const content = typeof msg?.content === "string" ? msg.content : safeStringify8(msg?.content);
+      const content = typeof msg?.content === "string" ? msg.content : safeStringify9(msg?.content);
       attrs[`neatlogs.llm.input_messages.${i}.role`] = mapRole(role);
       attrs[`neatlogs.llm.input_messages.${i}.content`] = content;
     }
@@ -8531,7 +8926,7 @@ var NeatlogsCallbackHandler = class {
             const tc = toolCalls[k];
             span2.setAttribute(`neatlogs.llm.tool_calls.${k}.id`, tc.id ?? "");
             span2.setAttribute(`neatlogs.llm.tool_calls.${k}.name`, tc.name ?? tc.function?.name ?? "");
-            span2.setAttribute(`neatlogs.llm.tool_calls.${k}.arguments`, tc.args ? safeStringify8(tc.args) : tc.function?.arguments ?? "");
+            span2.setAttribute(`neatlogs.llm.tool_calls.${k}.arguments`, tc.args ? safeStringify9(tc.args) : tc.function?.arguments ?? "");
           }
         }
       }
@@ -8548,7 +8943,7 @@ var NeatlogsCallbackHandler = class {
         span2.setAttribute("neatlogs.llm.token_count.total", usage.totalTokens ?? usage.total_tokens);
       }
     }
-    span2.setStatus({ code: import_api16.SpanStatusCode.OK });
+    span2.setStatus({ code: import_api17.SpanStatusCode.OK });
     span2.end();
     this._spans.delete(runId);
     this._endAutoRoot(runId);
@@ -8558,7 +8953,7 @@ var NeatlogsCallbackHandler = class {
   async handleLLMError(error, runId) {
     const span2 = this._spans.get(runId);
     if (!span2) return;
-    span2.setStatus({ code: import_api16.SpanStatusCode.ERROR, message: error.message });
+    span2.setStatus({ code: import_api17.SpanStatusCode.ERROR, message: error.message });
     span2.recordException(error);
     span2.end();
     this._spans.delete(runId);
@@ -8566,7 +8961,7 @@ var NeatlogsCallbackHandler = class {
   }
   // --- Tool callbacks ---
   async handleToolStart(serialized, input, runId, parentRunId, _tags, _metadata, runName, toolCallId) {
-    const tracer = import_api16.trace.getTracer(TRACER_NAME8);
+    const tracer = import_api17.trace.getTracer(TRACER_NAME9);
     const name = serialized?.name || runName || (Array.isArray(serialized?.id) ? serialized.id[serialized.id.length - 1] : void 0) || "tool";
     const parentCtx = this._startCtx(parentRunId, runId, "tool");
     const attrs = {
@@ -8582,7 +8977,7 @@ var NeatlogsCallbackHandler = class {
     const span2 = this._spans.get(runId);
     if (!span2) return;
     span2.setAttribute("output.value", String(output));
-    span2.setStatus({ code: import_api16.SpanStatusCode.OK });
+    span2.setStatus({ code: import_api17.SpanStatusCode.OK });
     span2.end();
     this._spans.delete(runId);
     this._endAutoRoot(runId);
@@ -8590,7 +8985,7 @@ var NeatlogsCallbackHandler = class {
   async handleToolError(error, runId) {
     const span2 = this._spans.get(runId);
     if (!span2) return;
-    span2.setStatus({ code: import_api16.SpanStatusCode.ERROR, message: error.message });
+    span2.setStatus({ code: import_api17.SpanStatusCode.ERROR, message: error.message });
     span2.recordException(error);
     span2.end();
     this._spans.delete(runId);
@@ -8598,7 +8993,7 @@ var NeatlogsCallbackHandler = class {
   }
   // --- Retriever callbacks ---
   async handleRetrieverStart(serialized, query, runId, parentRunId) {
-    const tracer = import_api16.trace.getTracer(TRACER_NAME8);
+    const tracer = import_api17.trace.getTracer(TRACER_NAME9);
     const name = serialized?.name ?? "retriever";
     const parentCtx = this._startCtx(parentRunId, runId, "retriever");
     const attrs = {
@@ -8621,7 +9016,7 @@ var NeatlogsCallbackHandler = class {
         }
       }
     }
-    span2.setStatus({ code: import_api16.SpanStatusCode.OK });
+    span2.setStatus({ code: import_api17.SpanStatusCode.OK });
     span2.end();
     this._spans.delete(runId);
     this._endAutoRoot(runId);
@@ -8629,7 +9024,7 @@ var NeatlogsCallbackHandler = class {
   async handleRetrieverError(error, runId) {
     const span2 = this._spans.get(runId);
     if (!span2) return;
-    span2.setStatus({ code: import_api16.SpanStatusCode.ERROR, message: error.message });
+    span2.setStatus({ code: import_api17.SpanStatusCode.ERROR, message: error.message });
     span2.recordException(error);
     span2.end();
     this._spans.delete(runId);
@@ -8654,7 +9049,7 @@ function mapRole(role) {
   if (r === "tool" || r === "toolmessage") return "tool";
   return role;
 }
-function safeStringify8(value) {
+function safeStringify9(value) {
   try {
     return typeof value === "string" ? value : JSON.stringify(value);
   } catch {
@@ -8730,7 +9125,7 @@ function setOutput(span2, isTool, out) {
   } else {
     span2.setAttribute("neatlogs.llm.output_messages.0.role", "assistant");
     span2.setAttribute("neatlogs.llm.output_messages.0.content", out);
-    span2.setAttribute("neatlogs.llm.output", safeStringify9({ role: "assistant", content: out }));
+    span2.setAttribute("neatlogs.llm.output", safeStringify10({ role: "assistant", content: out }));
   }
 }
 function appendInputMessage(span2, role, content) {
@@ -8741,7 +9136,7 @@ function appendInputMessage(span2, role, content) {
   const existing = span2.__neatlogs_in_msgs ?? [];
   existing.push({ role, content });
   span2.__neatlogs_in_msgs = existing;
-  const blob = safeStringify9({ messages: existing });
+  const blob = safeStringify10({ messages: existing });
   span2.setAttribute("input.value", blob);
   span2.setAttribute("neatlogs.llm.input", blob);
 }
@@ -8784,15 +9179,15 @@ function flattenBlocks(val) {
     const o = item;
     if (typeof o.text === "string") out.push(o.text);
     else if (typeof o.content === "string") out.push(o.content);
-    else if (o.toolUse) out.push(`${o.toolUse.name ?? "tool"}(${safeStringify9(o.toolUse.input ?? {})})`);
+    else if (o.toolUse) out.push(`${o.toolUse.name ?? "tool"}(${safeStringify10(o.toolUse.input ?? {})})`);
     else if (o.toolResult) out.push(flattenBlocks(o.toolResult.content ?? o.toolResult));
     else if (Array.isArray(o.parts)) out.push(flattenBlocks(o.parts));
     else if (Array.isArray(o.content)) out.push(flattenBlocks(o.content));
-    else out.push(safeStringify9(o));
+    else out.push(safeStringify10(o));
   }
   return out.filter(Boolean).join("\n");
 }
-function safeStringify9(value) {
+function safeStringify10(value) {
   if (typeof value === "string") return value;
   try {
     return JSON.stringify(value) ?? "";
@@ -8802,8 +9197,8 @@ function safeStringify9(value) {
 }
 
 // src/openai-agents.ts
-var import_api17 = require("@opentelemetry/api");
-var TRACER_NAME9 = "neatlogs.openai_agents";
+var import_api18 = require("@opentelemetry/api");
+var TRACER_NAME10 = "neatlogs.openai_agents";
 function openaiAgentsProcessor() {
   return new NeatlogsTraceProcessor();
 }
@@ -8812,13 +9207,13 @@ var NeatlogsTraceProcessor = class {
   _startTimes = /* @__PURE__ */ new Map();
   // The @openai/agents SDK passes a Trace object: { traceId, name, groupId, metadata }.
   onTraceStart(traceData) {
-    const tracer = import_api17.trace.getTracer(TRACER_NAME9);
+    const tracer = import_api18.trace.getTracer(TRACER_NAME10);
     const attrs = { "neatlogs.span.kind": "WORKFLOW" };
     const workflowName = traceData?.name ?? traceData?.workflow_name;
     if (workflowName) attrs["neatlogs.workflow.name"] = workflowName;
     const traceId = traceData?.traceId ?? traceData?.trace_id;
     if (traceId) attrs["neatlogs.agent.trace_id"] = String(traceId);
-    const span2 = tracer.startSpan("openai_agents.trace", { attributes: attrs }, import_api17.context.active());
+    const span2 = tracer.startSpan("openai_agents.trace", { attributes: attrs }, import_api18.context.active());
     const key = String(traceId ?? `trace_${Date.now()}`);
     this._spans.set(key, span2);
     this._startTimes.set(key, Date.now());
@@ -8831,7 +9226,7 @@ var NeatlogsTraceProcessor = class {
     if (startTime) {
       span2.setAttribute("neatlogs.metrics.duration_ms", Date.now() - startTime);
     }
-    span2.setStatus({ code: import_api17.SpanStatusCode.OK });
+    span2.setStatus({ code: import_api18.SpanStatusCode.OK });
     span2.end();
     this._spans.delete(key);
     this._startTimes.delete(key);
@@ -8839,13 +9234,13 @@ var NeatlogsTraceProcessor = class {
   // The SDK passes a Span object: { type, spanId, traceId, parentId?, spanData: {...} }.
   // The meaningful payload (type, name, input, output, usage, ...) lives in `spanData`.
   onSpanStart(span2) {
-    const tracer = import_api17.trace.getTracer(TRACER_NAME9);
+    const tracer = import_api18.trace.getTracer(TRACER_NAME10);
     const data = span2?.spanData ?? span2 ?? {};
     const spanType = data?.type ?? data?.span_type ?? span2?.type ?? "";
     const spanId = String(span2?.spanId ?? span2?.span_id ?? data?.span_id ?? `span_${Date.now()}`);
     const parentKey = String(span2?.parentId ?? span2?.traceId ?? span2?.trace_id ?? "");
     const parentSpan = this._spans.get(parentKey);
-    const parentCtx = parentSpan ? import_api17.trace.setSpan(import_api17.context.active(), parentSpan) : import_api17.context.active();
+    const parentCtx = parentSpan ? import_api18.trace.setSpan(import_api18.context.active(), parentSpan) : import_api18.context.active();
     let otelSpan;
     if (spanType === "agent" || spanType === "agent_run") {
       const agentName = data?.name ?? data?.agent_name ?? "agent";
@@ -8871,7 +9266,7 @@ var NeatlogsTraceProcessor = class {
           const role = typeof msg === "object" ? msg.role ?? "" : "";
           const content = typeof msg === "object" ? msg.content ?? "" : String(msg);
           if (role) attrs[`neatlogs.llm.input_messages.${i}.role`] = role;
-          if (content) attrs[`neatlogs.llm.input_messages.${i}.content`] = typeof content === "string" ? content : safeStringify10(content);
+          if (content) attrs[`neatlogs.llm.input_messages.${i}.content`] = typeof content === "string" ? content : safeStringify11(content);
         }
       }
       otelSpan = tracer.startSpan("openai_agents.generation", { attributes: attrs }, parentCtx);
@@ -8883,7 +9278,7 @@ var NeatlogsTraceProcessor = class {
       };
       const toolInput = data?.input ?? data?.arguments;
       if (toolInput !== void 0) {
-        attrs["input.value"] = typeof toolInput === "string" ? toolInput : safeStringify10(toolInput);
+        attrs["input.value"] = typeof toolInput === "string" ? toolInput : safeStringify11(toolInput);
       }
       otelSpan = tracer.startSpan(`openai_agents.tool.${toolName}`, { attributes: attrs }, parentCtx);
     } else if (spanType === "handoff") {
@@ -8920,13 +9315,13 @@ var NeatlogsTraceProcessor = class {
         for (let i = 0; i < toolCalls.length; i++) {
           const tc = toolCalls[i];
           otelSpan.setAttribute(`neatlogs.llm.tool_calls.${i}.name`, tc.name ?? "");
-          otelSpan.setAttribute(`neatlogs.llm.tool_calls.${i}.arguments`, typeof tc.arguments === "string" ? tc.arguments : safeStringify10(tc.arguments ?? {}));
+          otelSpan.setAttribute(`neatlogs.llm.tool_calls.${i}.arguments`, typeof tc.arguments === "string" ? tc.arguments : safeStringify11(tc.arguments ?? {}));
           if (tc.callId ?? tc.id) otelSpan.setAttribute(`neatlogs.llm.tool_calls.${i}.id`, tc.callId ?? tc.id);
         }
       } else if (outputItems?.content) {
         otelSpan.setAttribute("neatlogs.llm.output_messages.0.role", "assistant");
         const c = outputItems.content;
-        otelSpan.setAttribute("neatlogs.llm.output_messages.0.content", typeof c === "string" ? c : safeStringify10(c));
+        otelSpan.setAttribute("neatlogs.llm.output_messages.0.content", typeof c === "string" ? c : safeStringify11(c));
       }
       const usage = data?.usage ?? resp?.usage;
       if (usage) {
@@ -8943,24 +9338,24 @@ var NeatlogsTraceProcessor = class {
     } else if (spanType === "function" || spanType === "tool" || spanType === "tool_call") {
       const output = data?.output ?? data?.result;
       if (output != null) {
-        otelSpan.setAttribute("output.value", typeof output === "string" ? output : safeStringify10(output));
+        otelSpan.setAttribute("output.value", typeof output === "string" ? output : safeStringify11(output));
       }
     } else if (spanType === "agent" || spanType === "agent_run") {
       const output = data?.output;
       if (output != null) {
-        otelSpan.setAttribute("output.value", typeof output === "string" ? output : safeStringify10(output));
+        otelSpan.setAttribute("output.value", typeof output === "string" ? output : safeStringify11(output));
       }
     }
     const error = data?.error ?? span2?.error;
     if (error) {
       if (error instanceof Error) {
-        otelSpan.setStatus({ code: import_api17.SpanStatusCode.ERROR, message: error.message });
+        otelSpan.setStatus({ code: import_api18.SpanStatusCode.ERROR, message: error.message });
         otelSpan.recordException(error);
       } else {
-        otelSpan.setStatus({ code: import_api17.SpanStatusCode.ERROR, message: String(error) });
+        otelSpan.setStatus({ code: import_api18.SpanStatusCode.ERROR, message: String(error) });
       }
     } else {
-      otelSpan.setStatus({ code: import_api17.SpanStatusCode.OK });
+      otelSpan.setStatus({ code: import_api18.SpanStatusCode.OK });
     }
     if (startTime) {
       otelSpan.setAttribute("neatlogs.llm.metrics.duration_ms", Date.now() - startTime);
@@ -8971,7 +9366,7 @@ var NeatlogsTraceProcessor = class {
   }
   shutdown() {
     for (const [, span2] of this._spans) {
-      span2.setStatus({ code: import_api17.SpanStatusCode.ERROR, message: "Processor shutdown before span completed" });
+      span2.setStatus({ code: import_api18.SpanStatusCode.ERROR, message: "Processor shutdown before span completed" });
       span2.end();
     }
     this._spans.clear();
@@ -8980,7 +9375,7 @@ var NeatlogsTraceProcessor = class {
   forceFlush() {
   }
 };
-function safeStringify10(value) {
+function safeStringify11(value) {
   try {
     return typeof value === "string" ? value : JSON.stringify(value);
   } catch {
@@ -8989,8 +9384,8 @@ function safeStringify10(value) {
 }
 
 // src/mastra-wrap.ts
-var import_api18 = require("@opentelemetry/api");
-var TRACER_NAME10 = "neatlogs.mastra";
+var import_api19 = require("@opentelemetry/api");
+var TRACER_NAME11 = "neatlogs.mastra";
 var PATCH_FLAG2 = "_neatlogs_patched";
 function wrapMastra(entity) {
   if (!entity || entity[PATCH_FLAG2]) return entity;
@@ -9056,14 +9451,14 @@ function patchAgentMethod(agent, method, streaming) {
     }
     if (streaming) attrs["neatlogs.llm.is_streaming"] = true;
     if (streaming) {
-      const tracer = import_api18.trace.getTracer(TRACER_NAME10);
-      const span2 = tracer.startSpan(`mastra.agent.${agentName}`, { attributes: attrs }, import_api18.context.active());
-      const ctx = import_api18.trace.setSpan(import_api18.context.active(), span2);
+      const tracer = import_api19.trace.getTracer(TRACER_NAME11);
+      const span2 = tracer.startSpan(`mastra.agent.${agentName}`, { attributes: attrs }, import_api19.context.active());
+      const ctx = import_api19.trace.setSpan(import_api19.context.active(), span2);
       try {
-        const result = await import_api18.context.with(ctx, () => orig(input, opts));
+        const result = await import_api19.context.with(ctx, () => orig(input, opts));
         return wrapStreamingOutput(result, span2, ctx);
       } catch (err) {
-        recordError6(span2, err);
+        recordError7(span2, err);
         span2.end();
         throw err;
       }
@@ -9106,7 +9501,7 @@ function patchModelInPlace(model) {
       if (provider) attrs["neatlogs.llm.provider"] = provider;
       if (isStream) attrs["neatlogs.llm.is_streaming"] = true;
       const promptInput = callOpts?.prompt ?? callOpts?.messages;
-      if (promptInput !== void 0) attrs["input.value"] = safeStringify11(promptInput);
+      if (promptInput !== void 0) attrs["input.value"] = safeStringify12(promptInput);
       captureInvocationParams(attrs, callOpts);
       return withActiveSpan(`mastra.llm.${modelId || "model"}.${fn}`, attrs, async (span2) => {
         const result = await orig(callOpts);
@@ -9140,12 +9535,12 @@ function patchToolExecute(tool, key) {
     const attrs = {
       "neatlogs.span.kind": "TOOL",
       "neatlogs.tool.name": toolName,
-      "input.value": safeStringify11(params)
+      "input.value": safeStringify12(params)
     };
     if (tool.description) attrs["neatlogs.tool.description"] = String(tool.description);
     return withActiveSpan(`mastra.tool.${toolName}`, attrs, async (span2) => {
       const result = await orig(params, options);
-      span2.setAttribute("output.value", safeStringify11(result));
+      span2.setAttribute("output.value", safeStringify12(result));
       return result;
     });
   };
@@ -9171,13 +9566,13 @@ function patchRunMethod(run, method, workflowName) {
       "neatlogs.workflow.name": workflowName
     };
     if (startOpts?.inputData !== void 0) {
-      attrs["input.value"] = safeStringify11(startOpts.inputData);
+      attrs["input.value"] = safeStringify12(startOpts.inputData);
     }
     return withActiveSpan(`mastra.workflow.${workflowName}`, attrs, async (span2) => {
       const result = await orig(startOpts);
-      if (result?.status) span2.setAttribute("neatlogs.metadata", safeStringify11({ status: result.status }));
+      if (result?.status) span2.setAttribute("neatlogs.metadata", safeStringify12({ status: result.status }));
       if (result?.result !== void 0) {
-        span2.setAttribute("output.value", safeStringify11(result.result));
+        span2.setAttribute("output.value", safeStringify12(result.result));
       }
       return result;
     });
@@ -9198,14 +9593,14 @@ function patchVectorOp(vector, op, kind) {
       "neatlogs.span.kind": kind,
       "neatlogs.db.system": dbName,
       "neatlogs.db.operation": op,
-      "input.value": safeStringify11(params)
+      "input.value": safeStringify12(params)
     };
     const indexName = params?.indexName;
     if (indexName) attrs["neatlogs.vectordb.index_name"] = String(indexName);
     if (kind === "RETRIEVER" && params?.topK != null) attrs["neatlogs.retriever.top_k"] = params.topK;
     return withActiveSpan(`mastra.vector.${op}`, attrs, async (span2) => {
       const result = await orig(params);
-      span2.setAttribute("output.value", safeStringify11(result));
+      span2.setAttribute("output.value", safeStringify12(result));
       return result;
     });
   };
@@ -9219,11 +9614,11 @@ function patchMemory(memory) {
       const attrs = {
         "neatlogs.span.kind": "CHAIN",
         "neatlogs.db.operation": op,
-        "input.value": safeStringify11(args?.[0])
+        "input.value": safeStringify12(args?.[0])
       };
       return withActiveSpan(`mastra.memory.${op}`, attrs, async (span2) => {
         const result = await orig(...args);
-        span2.setAttribute("output.value", safeStringify11(result));
+        span2.setAttribute("output.value", safeStringify12(result));
         return result;
       });
     };
@@ -9264,7 +9659,7 @@ function wrapRootMastra(mastra) {
 }
 function wrapStreamingOutput(output, span2, ctx) {
   if (!output) {
-    span2.setStatus({ code: import_api18.SpanStatusCode.OK });
+    span2.setStatus({ code: import_api19.SpanStatusCode.OK });
     span2.end();
     return output;
   }
@@ -9272,8 +9667,8 @@ function wrapStreamingOutput(output, span2, ctx) {
   const endOnce = (err) => {
     if (ended) return;
     ended = true;
-    if (err) recordError6(span2, err);
-    else span2.setStatus({ code: import_api18.SpanStatusCode.OK });
+    if (err) recordError7(span2, err);
+    else span2.setStatus({ code: import_api19.SpanStatusCode.OK });
     span2.end();
   };
   if (output.text && typeof output.text.then === "function") {
@@ -9364,7 +9759,7 @@ function rebindStreamContext(output, ctx) {
         const origIterator = value[Symbol.asyncIterator].bind(value);
         return {
           [Symbol.asyncIterator]() {
-            return import_api18.context.with(ctx, () => origIterator());
+            return import_api19.context.with(ctx, () => origIterator());
           }
         };
       }
@@ -9406,7 +9801,7 @@ function finalizeModelResult(span2, result) {
       const tc = toolCalls[i];
       setToolCall(span2, i, tc.toolName, tc.input ?? tc.args, tc.toolCallId);
     }
-    span2.setAttribute("output.value", safeStringify11(result.content));
+    span2.setAttribute("output.value", safeStringify12(result.content));
   }
   if (result.usage) recordUsage(span2, result.usage);
   span2.setAttribute("neatlogs.llm.stop_reason", normalizeFinishReason(result.finishReason));
@@ -9414,7 +9809,7 @@ function finalizeModelResult(span2, result) {
 function normalizeFinishReason(fr) {
   if (fr == null) return "";
   if (typeof fr === "string") return fr;
-  return String(fr.unified ?? fr.reason ?? fr.type ?? safeStringify11(fr));
+  return String(fr.unified ?? fr.reason ?? fr.type ?? safeStringify12(fr));
 }
 function tokenValue(v) {
   if (v == null) return void 0;
@@ -9443,22 +9838,22 @@ function captureInvocationParams(attrs, callOpts) {
     }
   }
   if (Array.isArray(callOpts.stopSequences) && callOpts.stopSequences.length) {
-    attrs["neatlogs.llm.stop_sequences"] = safeStringify11(callOpts.stopSequences);
+    attrs["neatlogs.llm.stop_sequences"] = safeStringify12(callOpts.stopSequences);
     invocation.stopSequences = callOpts.stopSequences;
   }
   if (Array.isArray(callOpts.tools)) {
     for (let i = 0; i < callOpts.tools.length; i++) {
-      attrs[`neatlogs.llm.tools.${i}`] = safeStringify11(callOpts.tools[i]);
+      attrs[`neatlogs.llm.tools.${i}`] = safeStringify12(callOpts.tools[i]);
     }
     invocation.toolChoice = callOpts.toolChoice;
   }
   if (Object.keys(invocation).length) {
-    attrs["neatlogs.llm.invocation_parameters"] = safeStringify11(invocation);
+    attrs["neatlogs.llm.invocation_parameters"] = safeStringify12(invocation);
   }
 }
 function setToolCall(span2, i, name, args, id) {
   span2.setAttribute(`neatlogs.llm.tool_calls.${i}.name`, name ?? "");
-  span2.setAttribute(`neatlogs.llm.tool_calls.${i}.arguments`, safeStringify11(args ?? {}));
+  span2.setAttribute(`neatlogs.llm.tool_calls.${i}.arguments`, safeStringify12(args ?? {}));
   if (id) span2.setAttribute(`neatlogs.llm.tool_calls.${i}.id`, id);
 }
 function recordUsage(span2, usage) {
@@ -9474,16 +9869,16 @@ function recordUsage(span2, usage) {
   }
 }
 function withActiveSpan(name, attrs, fn) {
-  const tracer = import_api18.trace.getTracer(TRACER_NAME10);
-  const span2 = tracer.startSpan(name, { attributes: attrs }, import_api18.context.active());
-  const ctx = import_api18.trace.setSpan(import_api18.context.active(), span2);
-  return import_api18.context.with(ctx, async () => {
+  const tracer = import_api19.trace.getTracer(TRACER_NAME11);
+  const span2 = tracer.startSpan(name, { attributes: attrs }, import_api19.context.active());
+  const ctx = import_api19.trace.setSpan(import_api19.context.active(), span2);
+  return import_api19.context.with(ctx, async () => {
     try {
       const result = await fn(span2);
-      span2.setStatus({ code: import_api18.SpanStatusCode.OK });
+      span2.setStatus({ code: import_api19.SpanStatusCode.OK });
       return result;
     } catch (err) {
-      recordError6(span2, err);
+      recordError7(span2, err);
       throw err;
     } finally {
       span2.end();
@@ -9503,9 +9898,9 @@ function extractModelId(model) {
   return model.modelId ?? model.name ?? "";
 }
 function toInputValue(input) {
-  return typeof input === "string" ? input : safeStringify11(input);
+  return typeof input === "string" ? input : safeStringify12(input);
 }
-function safeStringify11(value) {
+function safeStringify12(value) {
   if (typeof value === "string") return value;
   try {
     return JSON.stringify(value) ?? "";
@@ -9513,25 +9908,25 @@ function safeStringify11(value) {
     return "";
   }
 }
-function recordError6(span2, err) {
+function recordError7(span2, err) {
   if (err instanceof Error) {
-    span2.setStatus({ code: import_api18.SpanStatusCode.ERROR, message: err.message });
+    span2.setStatus({ code: import_api19.SpanStatusCode.ERROR, message: err.message });
     span2.recordException(err);
   } else {
-    span2.setStatus({ code: import_api18.SpanStatusCode.ERROR, message: String(err) });
+    span2.setStatus({ code: import_api19.SpanStatusCode.ERROR, message: String(err) });
   }
 }
 
 // src/pi-agent.ts
-var import_api19 = require("@opentelemetry/api");
-var TRACER_NAME11 = "neatlogs.pi-agent";
+var import_api20 = require("@opentelemetry/api");
+var TRACER_NAME12 = "neatlogs.pi-agent";
 var PATCH_FLAG3 = "_neatlogs_patched";
 function piAgentHooks(agent) {
   if (!agent || agent[PATCH_FLAG3]) return agent;
   const a = agent;
   if (typeof a.subscribe !== "function") return agent;
   const state = { toolSpans: /* @__PURE__ */ new Map(), inputMessages: [] };
-  const tracer = import_api19.trace.getTracer(TRACER_NAME11);
+  const tracer = import_api20.trace.getTracer(TRACER_NAME12);
   a.subscribe((event) => {
     try {
       handleEvent(tracer, state, event);
@@ -9547,10 +9942,10 @@ function handleEvent(tracer, state, event) {
       const span2 = tracer.startSpan(
         "pi_agent.run",
         { attributes: { "neatlogs.span.kind": "AGENT" } },
-        import_api19.context.active()
+        import_api20.context.active()
       );
       state.agentSpan = span2;
-      state.agentCtx = import_api19.trace.setSpan(import_api19.context.active(), span2);
+      state.agentCtx = import_api20.trace.setSpan(import_api20.context.active(), span2);
       state.inputMessages = [];
       break;
     }
@@ -9569,14 +9964,14 @@ function handleEvent(tracer, state, event) {
       break;
     }
     case "tool_execution_start": {
-      const parent = state.agentCtx ?? import_api19.context.active();
+      const parent = state.agentCtx ?? import_api20.context.active();
       const span2 = tracer.startSpan(
         `pi_agent.tool.${event.toolName ?? "tool"}`,
         {
           attributes: {
             "neatlogs.span.kind": "TOOL",
             ...event.toolName ? { "neatlogs.tool.name": String(event.toolName) } : {},
-            ...event.args !== void 0 ? { "input.value": safeStringify12(event.args) } : {}
+            ...event.args !== void 0 ? { "input.value": safeStringify13(event.args) } : {}
           }
         },
         parent
@@ -9588,13 +9983,13 @@ function handleEvent(tracer, state, event) {
       const span2 = event.toolCallId ? state.toolSpans.get(event.toolCallId) : void 0;
       if (!span2) return;
       if (event.result !== void 0) {
-        span2.setAttribute("output.value", safeStringify12(event.result));
+        span2.setAttribute("output.value", safeStringify13(event.result));
       }
       if (event.isError) {
-        span2.setStatus({ code: import_api19.SpanStatusCode.ERROR });
+        span2.setStatus({ code: import_api20.SpanStatusCode.ERROR });
         span2.setAttribute("neatlogs.tool.is_error", true);
       } else {
-        span2.setStatus({ code: import_api19.SpanStatusCode.OK });
+        span2.setStatus({ code: import_api20.SpanStatusCode.OK });
       }
       span2.end();
       if (event.toolCallId) state.toolSpans.delete(event.toolCallId);
@@ -9613,7 +10008,7 @@ function handleEvent(tracer, state, event) {
         if (firstUser) state.agentSpan.setAttribute("input.value", firstUser.content);
         const finalText = lastAssistantText(event.messages);
         if (finalText) state.agentSpan.setAttribute("output.value", finalText);
-        state.agentSpan.setStatus({ code: import_api19.SpanStatusCode.OK });
+        state.agentSpan.setStatus({ code: import_api20.SpanStatusCode.OK });
         state.agentSpan.end();
         state.agentSpan = void 0;
         state.agentCtx = void 0;
@@ -9635,11 +10030,11 @@ function emitLlmSpan(tracer, state, msg) {
       attrs[`neatlogs.llm.input_messages.${i}.role`] = m.role;
       attrs[`neatlogs.llm.input_messages.${i}.content`] = m.content;
     });
-    attrs["neatlogs.llm.input"] = safeStringify12({ messages: inMsgs });
-    attrs["input.value"] = safeStringify12({ messages: inMsgs });
+    attrs["neatlogs.llm.input"] = safeStringify13({ messages: inMsgs });
+    attrs["input.value"] = safeStringify13({ messages: inMsgs });
   }
   const { text, toolCalls } = splitAssistantContent(msg.content);
-  const outText = text || toolCalls.map((tc) => `${tc.name}(${safeStringify12(tc.arguments)})`).join("\n");
+  const outText = text || toolCalls.map((tc) => `${tc.name}(${safeStringify13(tc.arguments)})`).join("\n");
   if (outText || toolCalls.length) {
     attrs["neatlogs.llm.output_messages.0.role"] = "assistant";
     attrs["neatlogs.llm.output_messages.0.content"] = outText || "";
@@ -9649,11 +10044,11 @@ function emitLlmSpan(tracer, state, msg) {
       toolCalls.forEach((tc, j) => {
         if (tc.name) attrs[`neatlogs.llm.tool_calls.${j}.name`] = tc.name;
         if (tc.arguments !== void 0)
-          attrs[`neatlogs.llm.tool_calls.${j}.arguments`] = safeStringify12(tc.arguments);
+          attrs[`neatlogs.llm.tool_calls.${j}.arguments`] = safeStringify13(tc.arguments);
         if (tc.id) attrs[`neatlogs.llm.tool_calls.${j}.id`] = String(tc.id);
       });
     }
-    attrs["neatlogs.llm.output"] = safeStringify12(outBlob);
+    attrs["neatlogs.llm.output"] = safeStringify13(outBlob);
     attrs["output.value"] = outText || "";
   }
   const usage = msg.usage;
@@ -9665,13 +10060,13 @@ function emitLlmSpan(tracer, state, msg) {
     if (usage.cacheRead) attrs["neatlogs.llm.token_count.cache_read"] = usage.cacheRead;
     if (usage.cacheWrite) attrs["neatlogs.llm.token_count.cache_write"] = usage.cacheWrite;
   }
-  const parent = state.agentCtx ?? import_api19.context.active();
+  const parent = state.agentCtx ?? import_api20.context.active();
   const span2 = tracer.startSpan(
     `pi_agent.llm.${msg.model || "model"}`,
     { attributes: attrs },
     parent
   );
-  span2.setStatus({ code: import_api19.SpanStatusCode.OK });
+  span2.setStatus({ code: import_api20.SpanStatusCode.OK });
   span2.end();
 }
 function splitAssistantContent(content) {
@@ -9699,7 +10094,7 @@ function messageText(msg) {
     if (typeof block === "string") parts.push(block);
     else if (block && typeof block === "object") {
       if (typeof block.text === "string") parts.push(block.text);
-      else if (block.type === "toolCall") parts.push(`${block.name ?? "tool"}(${safeStringify12(block.arguments)})`);
+      else if (block.type === "toolCall") parts.push(`${block.name ?? "tool"}(${safeStringify13(block.arguments)})`);
     }
   }
   return parts.join("");
@@ -9722,7 +10117,7 @@ function markPatched2(e) {
     e[PATCH_FLAG3] = true;
   }
 }
-function safeStringify12(value) {
+function safeStringify13(value) {
   if (typeof value === "string") return value;
   try {
     return JSON.stringify(value) ?? "";
@@ -9732,8 +10127,8 @@ function safeStringify12(value) {
 }
 
 // src/claude-agent-sdk.ts
-var import_api20 = require("@opentelemetry/api");
-var TRACER_NAME12 = "neatlogs.claude_agent_sdk";
+var import_api21 = require("@opentelemetry/api");
+var TRACER_NAME13 = "neatlogs.claude_agent_sdk";
 var ROOT_SCOPE = "__root__";
 function wrapClaudeAgentSDK(sdk, options = {}) {
   if (!sdk || typeof sdk.query !== "function") return sdk;
@@ -9753,13 +10148,13 @@ function wrapClaudeAgentSDK(sdk, options = {}) {
 }
 function wrapQuery(original, options) {
   return function(params, ...rest) {
-    const tracer = import_api20.trace.getTracer(TRACER_NAME12);
+    const tracer = import_api21.trace.getTracer(TRACER_NAME13);
     const workflowName = options.workflowName ?? "claude_agent.query";
     const promptRef = { text: "" };
     const agentSpan = tracer.startSpan(
       "claude_agent.query",
       { attributes: { "neatlogs.span.kind": "AGENT", "neatlogs.workflow.name": workflowName } },
-      import_api20.context.active()
+      import_api21.context.active()
     );
     const promptText = extractPromptText(params?.prompt);
     if (promptText) {
@@ -9768,8 +10163,8 @@ function wrapQuery(original, options) {
     } else if (params && isAsyncIterable(params.prompt)) {
       params = { ...params, prompt: tapPromptStream(params.prompt, promptRef) };
     }
-    const agentCtx = import_api20.trace.setSpan(import_api20.context.active(), agentSpan);
-    const queryObj = import_api20.context.with(agentCtx, () => original(params, ...rest));
+    const agentCtx = import_api21.trace.setSpan(import_api21.context.active(), agentSpan);
+    const queryObj = import_api21.context.with(agentCtx, () => original(params, ...rest));
     return instrumentQueryIterable(queryObj, agentSpan, agentCtx, tracer, promptRef);
   };
 }
@@ -9788,7 +10183,7 @@ async function* tapPromptStream(prompt, ref) {
 function instrumentQueryIterable(queryObj, agentSpan, agentCtx, tracer, promptRef) {
   const originalAsyncIterator = queryObj?.[Symbol.asyncIterator]?.bind(queryObj);
   if (!originalAsyncIterator) {
-    agentSpan.setStatus({ code: import_api20.SpanStatusCode.OK });
+    agentSpan.setStatus({ code: import_api21.SpanStatusCode.OK });
     agentSpan.end();
     return queryObj;
   }
@@ -9823,9 +10218,9 @@ function instrumentQueryIterable(queryObj, agentSpan, agentCtx, tracer, promptRe
     }
     if (rootScope.finalText) agentSpan.setAttribute("output.value", rootScope.finalText);
     if (status === "error") {
-      recordError7(agentSpan, err);
+      recordError8(agentSpan, err);
     } else {
-      agentSpan.setStatus({ code: import_api20.SpanStatusCode.OK });
+      agentSpan.setStatus({ code: import_api21.SpanStatusCode.OK });
       agentSpan.end();
     }
   };
@@ -9836,7 +10231,7 @@ function instrumentQueryIterable(queryObj, agentSpan, agentCtx, tracer, promptRe
     return {
       async next() {
         try {
-          const result = await import_api20.context.with(agentCtx, () => iterator.next());
+          const result = await import_api21.context.with(agentCtx, () => iterator.next());
           if (result.done) {
             finalizeAgent("ok");
             return result;
@@ -9867,7 +10262,7 @@ function instrumentQueryIterable(queryObj, agentSpan, agentCtx, tracer, promptRe
 function closeScope(scope, status) {
   try {
     if (scope.finalText) scope.span.setAttribute("output.value", scope.finalText);
-    scope.span.setStatus({ code: status === "ok" ? import_api20.SpanStatusCode.OK : import_api20.SpanStatusCode.ERROR });
+    scope.span.setStatus({ code: status === "ok" ? import_api21.SpanStatusCode.OK : import_api21.SpanStatusCode.ERROR });
     scope.span.end();
   } catch {
   }
@@ -9882,7 +10277,7 @@ function getScope(tracer, state, msg) {
     flushAssistantTurn(tracer, root, state);
   }
   const parentToolSpan = state.toolSpans.get(parentId);
-  const parentCtx = parentToolSpan ? import_api20.trace.setSpan(import_api20.context.active(), parentToolSpan) : state.scopes.get(ROOT_SCOPE).ctx;
+  const parentCtx = parentToolSpan ? import_api21.trace.setSpan(import_api21.context.active(), parentToolSpan) : state.scopes.get(ROOT_SCOPE).ctx;
   const subType = msg?.subagent_type ? String(msg.subagent_type) : "subagent";
   const attrs = {
     "neatlogs.span.kind": "AGENT",
@@ -9892,7 +10287,7 @@ function getScope(tracer, state, msg) {
   const span2 = tracer.startSpan(`claude_agent.subagent.${subType}`, { attributes: attrs }, parentCtx);
   const scope = {
     span: span2,
-    ctx: import_api20.trace.setSpan(import_api20.context.active(), span2),
+    ctx: import_api21.trace.setSpan(import_api21.context.active(), span2),
     inputMessages: msg?.task_description ? [{ role: "user", content: String(msg.task_description) }] : [],
     assistantBuffer: null,
     finalText: "",
@@ -9947,7 +10342,7 @@ function handleMessage(tracer, state, msg, finalizeAgent) {
         rootScope.span.setAttribute("neatlogs.conversation.id", String(msg.session_id));
       }
       const usage = msg.usage;
-      if (usage) setUsage2(rootScope.span, usage);
+      if (usage) setUsage3(rootScope.span, usage);
       if (msg.total_cost_usd != null) rootScope.span.setAttribute("neatlogs.agent.cost_usd", msg.total_cost_usd);
       if (msg.num_turns != null) rootScope.span.setAttribute("neatlogs.agent.num_turns", msg.num_turns);
       if (msg.is_error) rootScope.span.setAttribute("neatlogs.agent.is_error", true);
@@ -9996,10 +10391,10 @@ function flushAssistantTurn(tracer, scope, state) {
     attrs[`neatlogs.llm.input_messages.${i}.content`] = m.content;
   });
   if (scope.inputMessages.length) {
-    attrs["input.value"] = safeStringify13({ messages: scope.inputMessages });
+    attrs["input.value"] = safeStringify14({ messages: scope.inputMessages });
   }
   const outText = buf.textParts.join("");
-  const outValue = outText || buf.toolCalls.map((tc) => `${tc.name}(${safeStringify13(tc.input ?? {})})`).join("\n");
+  const outValue = outText || buf.toolCalls.map((tc) => `${tc.name}(${safeStringify14(tc.input ?? {})})`).join("\n");
   if (outValue) {
     attrs["neatlogs.llm.output_messages.0.role"] = "assistant";
     attrs["neatlogs.llm.output_messages.0.content"] = outValue;
@@ -10011,17 +10406,17 @@ function flushAssistantTurn(tracer, scope, state) {
   buf.toolCalls.forEach((tc, j) => {
     attrs[`neatlogs.llm.tool_calls.${j}.id`] = tc.id;
     attrs[`neatlogs.llm.tool_calls.${j}.name`] = tc.name;
-    attrs[`neatlogs.llm.tool_calls.${j}.arguments`] = safeStringify13(tc.input ?? {});
+    attrs[`neatlogs.llm.tool_calls.${j}.arguments`] = safeStringify14(tc.input ?? {});
   });
   if (buf.stopReason) attrs["neatlogs.llm.finish_reason"] = String(buf.stopReason);
   const span2 = tracer.startSpan(`claude_agent.llm.${model || "model"}`, { attributes: attrs }, scope.ctx);
-  if (buf.usage) setUsage2(span2, buf.usage);
-  span2.setStatus({ code: import_api20.SpanStatusCode.OK });
+  if (buf.usage) setUsage3(span2, buf.usage);
+  span2.setStatus({ code: import_api21.SpanStatusCode.OK });
   span2.end();
   if (outText) scope.finalText = outText;
   const turnParts = [];
   if (outText) turnParts.push(outText);
-  for (const tc of buf.toolCalls) turnParts.push(`[tool_call ${tc.name} ${safeStringify13(tc.input ?? {})}]`);
+  for (const tc of buf.toolCalls) turnParts.push(`[tool_call ${tc.name} ${safeStringify14(tc.input ?? {})}]`);
   if (turnParts.length) scope.inputMessages.push({ role: "assistant", content: turnParts.join("\n") });
   for (const tc of buf.toolCalls) {
     const toolSpan = tracer.startSpan(
@@ -10031,7 +10426,7 @@ function flushAssistantTurn(tracer, scope, state) {
           "neatlogs.span.kind": "TOOL",
           "neatlogs.tool.name": String(tc.name ?? ""),
           ...tc.id ? { "neatlogs.tool_call.id": String(tc.id) } : {},
-          "input.value": safeStringify13(tc.input ?? {})
+          "input.value": safeStringify14(tc.input ?? {})
         }
       },
       scope.ctx
@@ -10045,16 +10440,16 @@ function closeToolSpansFromUser(state, scope, msg) {
     if (block?.type !== "tool_result") continue;
     const id = block.tool_use_id ?? "";
     const out = block.content;
-    const outText = typeof out === "string" ? out : safeStringify13(out);
+    const outText = typeof out === "string" ? out : safeStringify14(out);
     if (outText) scope.inputMessages.push({ role: "tool", content: outText });
     const span2 = state.toolSpans.get(id);
     if (!span2) continue;
     span2.setAttribute("output.value", outText);
     if (block.is_error) {
-      span2.setStatus({ code: import_api20.SpanStatusCode.ERROR });
+      span2.setStatus({ code: import_api21.SpanStatusCode.ERROR });
       span2.setAttribute("neatlogs.tool.is_error", true);
     } else {
-      span2.setStatus({ code: import_api20.SpanStatusCode.OK });
+      span2.setStatus({ code: import_api21.SpanStatusCode.OK });
     }
     span2.end();
     state.toolSpans.delete(id);
@@ -10073,7 +10468,7 @@ function userMessageText(msg) {
   }
   return parts.join("\n");
 }
-function setUsage2(span2, usage) {
+function setUsage3(span2, usage) {
   if (!usage) return;
   if (usage.input_tokens != null) span2.setAttribute("neatlogs.llm.token_count.prompt", usage.input_tokens);
   if (usage.output_tokens != null) span2.setAttribute("neatlogs.llm.token_count.completion", usage.output_tokens);
@@ -10091,7 +10486,7 @@ function extractPromptText(prompt) {
   if (typeof prompt === "string") return prompt;
   return "";
 }
-function safeStringify13(value) {
+function safeStringify14(value) {
   if (typeof value === "string") return value;
   try {
     return JSON.stringify(value) ?? "";
@@ -10099,20 +10494,20 @@ function safeStringify13(value) {
     return "";
   }
 }
-function recordError7(span2, err) {
+function recordError8(span2, err) {
   if (err instanceof Error) {
-    span2.setStatus({ code: import_api20.SpanStatusCode.ERROR, message: err.message });
+    span2.setStatus({ code: import_api21.SpanStatusCode.ERROR, message: err.message });
     span2.recordException(err);
   } else {
-    span2.setStatus({ code: import_api20.SpanStatusCode.ERROR, message: String(err) });
+    span2.setStatus({ code: import_api21.SpanStatusCode.ERROR, message: String(err) });
   }
   span2.end();
 }
 
 // src/openrouter-agent.ts
-var import_api21 = require("@opentelemetry/api");
-var TRACER_NAME13 = "neatlogs.openrouter_agent";
-var PROVIDER4 = "openrouter";
+var import_api22 = require("@opentelemetry/api");
+var TRACER_NAME14 = "neatlogs.openrouter_agent";
+var PROVIDER5 = "openrouter";
 function wrapOpenRouterAgent(client) {
   const c = client;
   if (!c || c._neatlogsWrapped) return client;
@@ -10129,30 +10524,30 @@ function wrapOpenRouterAgent(client) {
 function wrapCallModel(callModel) {
   return function(clientArg, opts, ...rest) {
     const span2 = startLlmSpan(opts);
-    const ctx = import_api21.trace.setSpan(import_api21.context.active(), span2);
-    const result = import_api21.context.with(ctx, () => callModel.call(this, clientArg, opts, ...rest));
+    const ctx = import_api22.trace.setSpan(import_api22.context.active(), span2);
+    const result = import_api22.context.with(ctx, () => callModel.call(this, clientArg, opts, ...rest));
     return instrumentModelResult(result, span2);
   };
 }
 function tracedCallModel(original) {
   return function(opts, ...rest) {
     const span2 = startLlmSpan(opts);
-    const ctx = import_api21.trace.setSpan(import_api21.context.active(), span2);
-    const result = import_api21.context.with(ctx, () => original(opts, ...rest));
+    const ctx = import_api22.trace.setSpan(import_api22.context.active(), span2);
+    const result = import_api22.context.with(ctx, () => original(opts, ...rest));
     return instrumentModelResult(result, span2);
   };
 }
 function startLlmSpan(opts) {
-  const tracer = getProviderTracer(TRACER_NAME13);
+  const tracer = getProviderTracer(TRACER_NAME14);
   const model = opts?.model ?? "";
   const span2 = tracer.startSpan("openrouter.call_model", {
     attributes: {
       "neatlogs.span.kind": "LLM",
-      "neatlogs.llm.provider": PROVIDER4,
-      "neatlogs.llm.system": PROVIDER4,
+      "neatlogs.llm.provider": PROVIDER5,
+      "neatlogs.llm.system": PROVIDER5,
       "neatlogs.llm.model_name": model
     }
-  }, import_api21.context.active());
+  }, import_api22.context.active());
   const messages = Array.isArray(opts?.messages) ? opts.messages : Array.isArray(opts?.input) ? opts.input : [];
   if (messages.length) {
     messages.forEach((msg, i) => {
@@ -10160,10 +10555,10 @@ function startLlmSpan(opts) {
       const content = msg?.content;
       span2.setAttribute(
         `neatlogs.llm.input_messages.${i}.content`,
-        typeof content === "string" ? content : safeStringify14(content)
+        typeof content === "string" ? content : safeStringify15(content)
       );
     });
-    span2.setAttribute("input.value", safeStringify14({ messages }));
+    span2.setAttribute("input.value", safeStringify15({ messages }));
   } else if (typeof opts?.input === "string") {
     span2.setAttribute("neatlogs.llm.input_messages.0.role", "user");
     span2.setAttribute("neatlogs.llm.input_messages.0.content", opts.input);
@@ -10204,7 +10599,7 @@ function startLlmSpan(opts) {
 }
 function instrumentModelResult(result, span2) {
   if (!result || typeof result !== "object" && typeof result !== "function") {
-    span2.setStatus({ code: import_api21.SpanStatusCode.OK });
+    span2.setStatus({ code: import_api22.SpanStatusCode.OK });
     span2.end();
     return result;
   }
@@ -10215,14 +10610,14 @@ function instrumentModelResult(result, span2) {
     try {
       finalizeLlm(span2, resolved);
     } catch {
-      span2.setStatus({ code: import_api21.SpanStatusCode.OK });
+      span2.setStatus({ code: import_api22.SpanStatusCode.OK });
       span2.end();
     }
   };
   const finalizeError = (err) => {
     if (finalized) return;
     finalized = true;
-    recordError8(span2, err);
+    recordError9(span2, err);
   };
   return new Proxy(result, {
     get(obj, prop, receiver) {
@@ -10308,7 +10703,7 @@ function finalizeLlm(span2, result) {
       span2.setAttribute(`neatlogs.llm.tool_calls.${j}.id`, tc?.id ?? "");
       span2.setAttribute(`neatlogs.llm.tool_calls.${j}.name`, tc?.function?.name ?? tc?.name ?? "");
       const args = tc?.function?.arguments ?? tc?.arguments;
-      span2.setAttribute(`neatlogs.llm.tool_calls.${j}.arguments`, typeof args === "string" ? args : safeStringify14(args ?? {}));
+      span2.setAttribute(`neatlogs.llm.tool_calls.${j}.arguments`, typeof args === "string" ? args : safeStringify15(args ?? {}));
     });
   }
   const model = result?.model ?? result?.response?.model;
@@ -10318,7 +10713,7 @@ function finalizeLlm(span2, result) {
   const finishReason = result?.finishReason ?? result?.finish_reason ?? result?.choices?.[0]?.finish_reason;
   if (finishReason) span2.setAttribute("neatlogs.llm.finish_reason", String(finishReason));
   setOpenResponsesUsage(span2, result);
-  span2.setStatus({ code: import_api21.SpanStatusCode.OK });
+  span2.setStatus({ code: import_api22.SpanStatusCode.OK });
   span2.end();
 }
 function extractOpenResponsesText(result) {
@@ -10334,7 +10729,7 @@ function extractOpenResponsesText(result) {
   }
   return parts.length ? parts.join("") : void 0;
 }
-function safeStringify14(value) {
+function safeStringify15(value) {
   if (typeof value === "string") return value;
   try {
     return JSON.stringify(value) ?? "";
@@ -10342,12 +10737,12 @@ function safeStringify14(value) {
     return "";
   }
 }
-function recordError8(span2, err) {
+function recordError9(span2, err) {
   if (err instanceof Error) {
-    span2.setStatus({ code: import_api21.SpanStatusCode.ERROR, message: err.message });
+    span2.setStatus({ code: import_api22.SpanStatusCode.ERROR, message: err.message });
     span2.recordException(err);
   } else {
-    span2.setStatus({ code: import_api21.SpanStatusCode.ERROR, message: String(err) });
+    span2.setStatus({ code: import_api22.SpanStatusCode.ERROR, message: String(err) });
   }
   span2.end();
 }
@@ -10504,7 +10899,7 @@ var protoRoot = import_protobufjs.default.Root.fromJSON(OTLP_PROTO_JSON);
 var ExportTraceServiceRequest = protoRoot.lookupType(
   "opentelemetry.proto.collector.trace.v1.ExportTraceServiceRequest"
 );
-var SpanStatusCode16 = { UNSET: 0, OK: 1, ERROR: 2 };
+var SpanStatusCode17 = { UNSET: 0, OK: 1, ERROR: 2 };
 function randomBytes(length) {
   const out = new Uint8Array(length);
   const crypto = globalThis.crypto;
@@ -10711,7 +11106,7 @@ var NeatlogsOpencodePlugin = async (_ctx) => {
         startTimeUnixNano: st.rootSpan.startNano,
         endTimeUnixNano: nowNanoString(),
         attributes: attrs,
-        status: { code: SpanStatusCode16.OK }
+        status: { code: SpanStatusCode17.OK }
       });
       const m = nowNanoString();
       shipper.enqueue({
@@ -10798,18 +11193,18 @@ var NeatlogsOpencodePlugin = async (_ctx) => {
           { key: "neatlogs.conversation.id", value: { stringValue: sessionID } }
         ];
         if (start.args !== void 0) {
-          setIO(attrs, "TOOL", safeStringify15(start.args), void 0);
-          push(attrs, attrStr("neatlogs.tool.input", safeStringify15(start.args)));
+          setIO(attrs, "TOOL", safeStringify16(start.args), void 0);
+          push(attrs, attrStr("neatlogs.tool.input", safeStringify16(start.args)));
         }
         if (result?.title) push(attrs, attrStr("neatlogs.tool.title", String(result.title)));
         const out = result?.output ?? result?.result;
         if (out !== void 0) {
-          const o = typeof out === "string" ? out : safeStringify15(out);
+          const o = typeof out === "string" ? out : safeStringify16(out);
           setIO(attrs, "TOOL", void 0, o);
           push(attrs, attrStr("neatlogs.tool.output", o));
         }
         if (result?.metadata !== void 0) {
-          push(attrs, attrStr("neatlogs.tool.metadata", safeStringify15(result.metadata)));
+          push(attrs, attrStr("neatlogs.tool.metadata", safeStringify16(result.metadata)));
         }
         shipper.enqueue({
           traceId: st.traceId,
@@ -10820,7 +11215,7 @@ var NeatlogsOpencodePlugin = async (_ctx) => {
           startTimeUnixNano: start.startNano,
           endTimeUnixNano: nowNanoString(),
           attributes: attrs,
-          status: { code: SpanStatusCode16.OK }
+          status: { code: SpanStatusCode17.OK }
         });
       } catch {
       }
@@ -10907,10 +11302,10 @@ function emitLlmSpan2(shipper, st, info, sessionID) {
   if (toolCalls.length) {
     toolCalls.forEach((tc, j) => {
       push(attrs, attrStr(`neatlogs.llm.tool_calls.${j}.name`, tc.name));
-      push(attrs, attrStr(`neatlogs.llm.tool_calls.${j}.arguments`, safeStringify15(tc.input ?? {})));
+      push(attrs, attrStr(`neatlogs.llm.tool_calls.${j}.arguments`, safeStringify16(tc.input ?? {})));
     });
     if (!outText) {
-      const rendered = toolCalls.map((tc) => `\u2192 ${tc.name}(${safeStringify15(tc.input ?? {})})`).join("\n");
+      const rendered = toolCalls.map((tc) => `\u2192 ${tc.name}(${safeStringify16(tc.input ?? {})})`).join("\n");
       push(attrs, attrStr("neatlogs.llm.output_messages.0.role", "assistant"));
       push(attrs, attrStr("neatlogs.llm.output_messages.0.content", rendered));
       setIO(attrs, "LLM", void 0, rendered);
@@ -10939,7 +11334,7 @@ function emitLlmSpan2(shipper, st, info, sessionID) {
     startTimeUnixNano: startNano,
     endTimeUnixNano: nowNanoString(),
     attributes: attrs,
-    status: { code: SpanStatusCode16.OK }
+    status: { code: SpanStatusCode17.OK }
   });
   if (info?.id) {
     st.outputParts.delete(String(info.id));
@@ -10975,7 +11370,7 @@ function messageText2(info) {
   }
   return out.join("");
 }
-function safeStringify15(value) {
+function safeStringify16(value) {
   if (typeof value === "string") return value;
   try {
     return JSON.stringify(value) ?? "";
@@ -11072,6 +11467,7 @@ function bindTemplates(llm, systemTpl, userTpl, compiledVars) {
   traceToolAnthropic,
   traceToolAzureOpenAI,
   traceToolBedrock,
+  traceToolGoogleGenAI,
   traceToolOpenAI,
   traceToolVertexAI,
   updatePrompt,
@@ -11081,6 +11477,8 @@ function bindTemplates(llm, systemTpl, userTpl, compiledVars) {
   wrapBedrock,
   wrapCallModel,
   wrapClaudeAgentSDK,
+  wrapGoogleGenAI,
+  wrapGoogleGenAIChat,
   wrapMastra,
   wrapOpenAI,
   wrapOpenRouterAgent,