neatlogs 1.0.6 → 1.0.8

package/dist/vertex-ai.mjs

@@ -0,0 +1,397 @@
+// src/vertex-ai.ts
+import { trace, context as otelContext, SpanStatusCode } from "@opentelemetry/api";
+var TRACER_NAME = "neatlogs.vertex_ai";
+var PROVIDER = "vertex_ai";
+var SYSTEM = "vertexai";
+function wrapVertexAI(client) {
+  return wrapNamespace(client, []);
+}
+function traceTool(name, fn) {
+  return async function tracedTool(args) {
+    const tracer = trace.getTracer(TRACER_NAME);
+    return tracer.startActiveSpan(
+      `tool.${name}`,
+      {
+        attributes: {
+          "neatlogs.span.kind": "TOOL",
+          "neatlogs.tool.name": name,
+          "input.value": safeStringify(args)
+        }
+      },
+      otelContext.active(),
+      async (span) => {
+        try {
+          const result = await fn(args);
+          span.setAttribute("output.value", safeStringify(result));
+          span.setStatus({ code: SpanStatusCode.OK });
+          return result;
+        } catch (err) {
+          recordError(span, err);
+          throw err;
+        } finally {
+          span.end();
+        }
+      }
+    );
+  };
+}
+function wrapNamespace(target, path) {
+  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 = [...path, String(prop)];
+      const pathStr = currentPath.join(".");
+      if (pathStr === "models.generateContent" && typeof value === "function") {
+        return tracedGenerateContent(value.bind(obj), false);
+      }
+      if (pathStr === "models.generateContentStream" && typeof value === "function") {
+        return tracedGenerateContent(value.bind(obj), true);
+      }
+      if (pathStr === "models.embedContent" && typeof value === "function") {
+        return tracedEmbedContent(value.bind(obj));
+      }
+      if (pathStr === "models.countTokens" && typeof value === "function") {
+        return tracedCountTokens(value.bind(obj));
+      }
+      if (value && typeof value === "object" && !Array.isArray(value) && isNamespace(currentPath)) {
+        return wrapNamespace(value, currentPath);
+      }
+      return value;
+    }
+  });
+}
+function isNamespace(path) {
+  if (path.length > 2) return false;
+  return ["models", "chats"].includes(path[path.length - 1]);
+}
+function wrapVertexAIChat(chat) {
+  const c = chat;
+  if (!c || c._neatlogsVertexPatched) return chat;
+  if (typeof c.sendMessage === "function") {
+    const orig = c.sendMessage.bind(c);
+    c.sendMessage = (params, ...rest) => tracedChatSend(orig, c, params, rest, false);
+  }
+  if (typeof c.sendMessageStream === "function") {
+    const orig = c.sendMessageStream.bind(c);
+    c.sendMessageStream = (params, ...rest) => tracedChatSend(orig, c, params, rest, true);
+  }
+  try {
+    Object.defineProperty(c, "_neatlogsVertexPatched", { value: true, enumerable: false, configurable: true });
+  } catch {
+    c._neatlogsVertexPatched = true;
+  }
+  return chat;
+}
+function tracedChatSend(original, chat, params, rest, isStream) {
+  const tracer = trace.getTracer(TRACER_NAME);
+  const model = chat?.model ?? chat?.modelVersion ?? "";
+  const span = tracer.startSpan("vertex_ai.chat.send_message", {
+    attributes: {
+      "neatlogs.span.kind": "LLM",
+      "neatlogs.llm.provider": PROVIDER,
+      "neatlogs.llm.system": SYSTEM,
+      "neatlogs.llm.model_name": model,
+      "neatlogs.llm.is_streaming": isStream
+    }
+  }, otelContext.active());
+  const message = params?.message ?? params;
+  span.setAttribute("neatlogs.llm.input_messages.0.role", "user");
+  span.setAttribute(
+    "neatlogs.llm.input_messages.0.content",
+    typeof message === "string" ? message : safeStringify(message)
+  );
+  const ctx = trace.setSpan(otelContext.active(), span);
+  const result = otelContext.with(ctx, () => original(params, ...rest));
+  return Promise.resolve(result).then(
+    (response) => {
+      if (isStream) return wrapStream(response, span);
+      finalizeResponse(span, response);
+      return response;
+    },
+    (err) => {
+      recordError(span, err);
+      throw err;
+    }
+  );
+}
+function tracedEmbedContent(original) {
+  return function(opts, ...rest) {
+    const tracer = trace.getTracer(TRACER_NAME);
+    const span = tracer.startSpan("vertex_ai.models.embed_content", {
+      attributes: {
+        "neatlogs.span.kind": "EMBEDDING",
+        "neatlogs.llm.provider": PROVIDER,
+        "neatlogs.embedding.model_name": opts?.model ?? "",
+        "neatlogs.embedding.text": safeStringify(opts?.contents ?? "")
+      }
+    }, otelContext.active());
+    const ctx = trace.setSpan(otelContext.active(), span);
+    const result = otelContext.with(ctx, () => original(opts, ...rest));
+    return Promise.resolve(result).then(
+      (response) => {
+        const embeddings = response?.embeddings;
+        if (Array.isArray(embeddings)) {
+          span.setAttribute("neatlogs.embedding.count", embeddings.length);
+          const vals = embeddings[0]?.values;
+          if (Array.isArray(vals)) span.setAttribute("neatlogs.embedding.dimensions", vals.length);
+        }
+        span.setStatus({ code: SpanStatusCode.OK });
+        span.end();
+        return response;
+      },
+      (err) => {
+        recordError(span, err);
+        throw err;
+      }
+    );
+  };
+}
+function tracedCountTokens(original) {
+  return function(opts, ...rest) {
+    const tracer = trace.getTracer(TRACER_NAME);
+    const span = tracer.startSpan("vertex_ai.models.count_tokens", {
+      attributes: {
+        "neatlogs.span.kind": "LLM",
+        "neatlogs.llm.provider": PROVIDER,
+        "neatlogs.llm.task": "count_tokens",
+        "neatlogs.llm.model_name": opts?.model ?? ""
+      }
+    }, otelContext.active());
+    const ctx = trace.setSpan(otelContext.active(), span);
+    const result = otelContext.with(ctx, () => original(opts, ...rest));
+    return Promise.resolve(result).then(
+      (response) => {
+        if (response?.totalTokens != null) span.setAttribute("neatlogs.llm.token_count.prompt", response.totalTokens);
+        span.setStatus({ code: SpanStatusCode.OK });
+        span.end();
+        return response;
+      },
+      (err) => {
+        recordError(span, err);
+        throw err;
+      }
+    );
+  };
+}
+function tracedGenerateContent(original, isStream) {
+  return function(opts, ...rest) {
+    const tracer = trace.getTracer(TRACER_NAME);
+    const model = opts?.model ?? "";
+    const span = tracer.startSpan("vertex_ai.models.generate_content", {
+      attributes: {
+        "neatlogs.span.kind": "LLM",
+        "neatlogs.llm.provider": PROVIDER,
+        "neatlogs.llm.system": SYSTEM,
+        "neatlogs.llm.model_name": model,
+        "neatlogs.llm.is_streaming": isStream
+      }
+    }, otelContext.active());
+    setInputAttributes(span, opts);
+    const ctx = trace.setSpan(otelContext.active(), span);
+    const result = otelContext.with(ctx, () => original(opts, ...rest));
+    return Promise.resolve(result).then(
+      (response) => {
+        if (isStream) {
+          return wrapStream(response, span);
+        }
+        finalizeResponse(span, response);
+        return response;
+      },
+      (err) => {
+        recordError(span, err);
+        throw err;
+      }
+    );
+  };
+}
+function setInputAttributes(span, opts) {
+  let idx = 0;
+  const config = opts?.config;
+  const systemInstruction = config?.systemInstruction ?? config?.system_instruction;
+  if (systemInstruction) {
+    span.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, "system");
+    span.setAttribute(
+      `neatlogs.llm.input_messages.${idx}.content`,
+      typeof systemInstruction === "string" ? systemInstruction : safeStringify(systemInstruction)
+    );
+    idx++;
+  }
+  const contents = opts?.contents;
+  if (typeof contents === "string") {
+    span.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, "user");
+    span.setAttribute(`neatlogs.llm.input_messages.${idx}.content`, contents);
+  } else if (Array.isArray(contents)) {
+    for (const item of contents) {
+      if (typeof item === "string") {
+        span.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, "user");
+        span.setAttribute(`neatlogs.llm.input_messages.${idx}.content`, item);
+        idx++;
+      } else if (item && typeof item === "object") {
+        const role = item.role ?? "user";
+        span.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);
+        }
+        span.setAttribute(
+          `neatlogs.llm.input_messages.${idx}.content`,
+          textParts.length ? textParts.join("\n") : safeStringify(parts)
+        );
+        idx++;
+      }
+    }
+  } else if (contents) {
+    span.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, "user");
+    span.setAttribute(`neatlogs.llm.input_messages.${idx}.content`, safeStringify(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) {
+        span.setAttribute(`neatlogs.llm.tools.${t}.name`, fn?.name ?? "");
+        if (fn?.description) span.setAttribute(`neatlogs.llm.tools.${t}.description`, fn.description);
+        if (fn?.parameters) span.setAttribute(`neatlogs.llm.tools.${t}.input_schema`, safeStringify(fn.parameters));
+        t++;
+      }
+    }
+  }
+  if (config) {
+    if (config.temperature != null) span.setAttribute("neatlogs.llm.temperature", config.temperature);
+    if (config.topP != null) span.setAttribute("neatlogs.llm.top_p", config.topP);
+    if (config.topK != null) span.setAttribute("neatlogs.llm.top_k", config.topK);
+    const maxTokens = config.maxOutputTokens ?? config.max_output_tokens;
+    if (maxTokens != null) span.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) {
+      span.setAttribute("neatlogs.llm.invocation_parameters", JSON.stringify(params));
+    }
+  }
+}
+function wrapStream(stream, span) {
+  const chunks = [];
+  const originalAsyncIterator = stream?.[Symbol.asyncIterator]?.bind(stream);
+  if (!originalAsyncIterator) {
+    finalizeStreamChunks(span, 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) {
+            finalizeStreamChunks(span, chunks);
+            return result;
+          }
+          chunks.push(result.value);
+          return result;
+        } catch (err) {
+          recordError(span, err);
+          throw err;
+        }
+      },
+      async return(value) {
+        finalizeStreamChunks(span, chunks);
+        return iterator.return?.(value) ?? { done: true, value: void 0 };
+      },
+      async throw(err) {
+        recordError(span, err);
+        return iterator.throw?.(err) ?? { done: true, value: void 0 };
+      }
+    };
+  };
+  return wrapped;
+}
+function finalizeStreamChunks(span, 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) {
+    span.setAttribute("neatlogs.llm.output_messages.0.role", "assistant");
+    span.setAttribute("neatlogs.llm.output_messages.0.content", fullText);
+  }
+  if (finishReason) span.setAttribute("neatlogs.llm.finish_reason", String(finishReason));
+  setUsage(span, usage);
+  span.setStatus({ code: SpanStatusCode.OK });
+  span.end();
+}
+function finalizeResponse(span, 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) {
+        span.setAttribute("neatlogs.llm.output_messages.0.thinking", part.text);
+      } else if (part?.functionCall) {
+        const fc = part.functionCall;
+        span.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.name`, fc?.name ?? "");
+        span.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.arguments`, safeStringify(fc?.args ?? {}));
+        toolIdx++;
+      }
+    }
+    if (candidate?.finishReason) span.setAttribute("neatlogs.llm.finish_reason", String(candidate.finishReason));
+  }
+  if (textParts.length) {
+    span.setAttribute("neatlogs.llm.output_messages.0.role", "assistant");
+    span.setAttribute("neatlogs.llm.output_messages.0.content", textParts.join(""));
+  }
+  setUsage(span, response?.usageMetadata);
+  span.setStatus({ code: SpanStatusCode.OK });
+  span.end();
+}
+function setUsage(span, usage) {
+  if (!usage) return;
+  if (usage.promptTokenCount != null) span.setAttribute("neatlogs.llm.token_count.prompt", usage.promptTokenCount);
+  if (usage.candidatesTokenCount != null) span.setAttribute("neatlogs.llm.token_count.completion", usage.candidatesTokenCount);
+  if (usage.totalTokenCount != null) span.setAttribute("neatlogs.llm.token_count.total", usage.totalTokenCount);
+  if (usage.cachedContentTokenCount != null) span.setAttribute("neatlogs.llm.token_count.cache_read", usage.cachedContentTokenCount);
+  if (usage.thoughtsTokenCount != null) span.setAttribute("neatlogs.llm.token_count.reasoning", usage.thoughtsTokenCount);
+}
+function safeStringify(value) {
+  try {
+    return typeof value === "string" ? value : JSON.stringify(value);
+  } catch {
+    return "";
+  }
+}
+function recordError(span, err) {
+  if (err instanceof Error) {
+    span.setStatus({ code: SpanStatusCode.ERROR, message: err.message });
+    span.recordException(err);
+  } else {
+    span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });
+  }
+  span.end();
+}
+export {
+  traceTool,
+  wrapVertexAI,
+  wrapVertexAIChat
+};
+//# sourceMappingURL=vertex-ai.mjs.map

package/dist/vertex-ai.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/vertex-ai.ts"],"sourcesContent":["/**\n * Neatlogs Vertex AI wrapper — ES6 Proxy-based.\n *\n * Vertex AI is accessed through the same `@google/genai` SDK as Gemini, but in\n * Vertex mode (`new GoogleGenAI({ vertexai: true, project, location })`). This\n * wrapper traces those clients with provider `vertex_ai` so Vertex traffic is\n * distinguished from Google AI Studio (Gemini) traffic.\n *\n * Usage:\n *   import { wrapVertexAI } from 'neatlogs/vertex-ai';\n *   import { GoogleGenAI } from '@google/genai';\n *   const client = wrapVertexAI(new GoogleGenAI({ vertexai: true, project: 'p', location: 'us-central1' }));\n *   const res = await client.models.generateContent({ model: 'gemini-2.0-flash', contents: 'Hello' });\n *\n * Intercepts:\n *   - client.models.generateContent()        — non-streaming\n *   - client.models.generateContentStream()  — streaming (async iterable)\n *   - client.models.embedContent()           — embeddings\n *   - client.models.countTokens()            — token counting\n *   - chat.sendMessage() / sendMessageStream() — chat sessions\n */\n\nimport { trace, context as otelContext, SpanStatusCode, type Span } from '@opentelemetry/api';\n\nconst TRACER_NAME = 'neatlogs.vertex_ai';\nconst PROVIDER = 'vertex_ai';\nconst SYSTEM = 'vertexai';\n\n// ---------------------------------------------------------------------------\n// Public API\n// ---------------------------------------------------------------------------\n\nexport function wrapVertexAI<T extends object>(client: T): T {\n  return wrapNamespace(client, []) as T;\n}\n\n/**\n * Wrap a tool/function implementation to emit TOOL spans when executed.\n *\n * Usage:\n *   const getWeather = traceTool('get_weather', async (args: { city: string }) => {\n *     return `Weather in ${args.city}: sunny`;\n *   });\n */\nexport function traceTool<TArgs = any, TResult = any>(\n  name: string,\n  fn: (args: TArgs) => TResult | Promise<TResult>,\n): (args: TArgs) => Promise<TResult> {\n  return async function tracedTool(args: TArgs): Promise<TResult> {\n    const tracer = trace.getTracer(TRACER_NAME);\n    return tracer.startActiveSpan(\n      `tool.${name}`,\n      {\n        attributes: {\n          'neatlogs.span.kind': 'TOOL',\n          'neatlogs.tool.name': name,\n          'input.value': safeStringify(args),\n        },\n      },\n      otelContext.active(),\n      async (span) => {\n        try {\n          const result = await fn(args);\n          span.setAttribute('output.value', safeStringify(result));\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (err) {\n          recordError(span, err);\n          throw err;\n        } finally {\n          span.end();\n        }\n      },\n    );\n  };\n}\n\n// ---------------------------------------------------------------------------\n// Proxy wrapping\n// ---------------------------------------------------------------------------\n\nfunction wrapNamespace(target: any, path: string[]): any {\n  return new Proxy(target, {\n    get(obj, prop, receiver) {\n      const value = Reflect.get(obj, prop, receiver);\n      if (typeof prop === 'symbol' || String(prop).startsWith('_')) return value;\n\n      const currentPath = [...path, String(prop)];\n      const pathStr = currentPath.join('.');\n\n      if (pathStr === 'models.generateContent' && typeof value === 'function') {\n        return tracedGenerateContent(value.bind(obj), false);\n      }\n      if (pathStr === 'models.generateContentStream' && typeof value === 'function') {\n        return tracedGenerateContent(value.bind(obj), true);\n      }\n      if (pathStr === 'models.embedContent' && typeof value === 'function') {\n        return tracedEmbedContent(value.bind(obj));\n      }\n      if (pathStr === 'models.countTokens' && typeof value === 'function') {\n        return tracedCountTokens(value.bind(obj));\n      }\n\n      if (value && typeof value === 'object' && !Array.isArray(value) && isNamespace(currentPath)) {\n        return wrapNamespace(value, currentPath);\n      }\n\n      return value;\n    },\n  });\n}\n\nfunction isNamespace(path: string[]): boolean {\n  if (path.length > 2) return false;\n  return ['models', 'chats'].includes(path[path.length - 1]);\n}\n\n/**\n * Wrap a chat session returned by `client.chats.create(...)`. The `@google/genai`\n * Chat object exposes `sendMessage` / `sendMessageStream`; wrap them so each turn\n * emits an LLM span. Returns the same chat instance.\n */\nexport function wrapVertexAIChat<T extends object>(chat: T): T {\n  const c = chat as any;\n  if (!c || c._neatlogsVertexPatched) return chat;\n\n  if (typeof c.sendMessage === 'function') {\n    const orig = c.sendMessage.bind(c);\n    c.sendMessage = (params: any, ...rest: any[]) => tracedChatSend(orig, c, params, rest, false);\n  }\n  if (typeof c.sendMessageStream === 'function') {\n    const orig = c.sendMessageStream.bind(c);\n    c.sendMessageStream = (params: any, ...rest: any[]) => tracedChatSend(orig, c, params, rest, true);\n  }\n\n  try {\n    Object.defineProperty(c, '_neatlogsVertexPatched', { value: true, enumerable: false, configurable: true });\n  } catch {\n    c._neatlogsVertexPatched = true;\n  }\n  return chat;\n}\n\nfunction tracedChatSend(\n  original: (...a: any[]) => any,\n  chat: any,\n  params: any,\n  rest: any[],\n  isStream: boolean,\n): any {\n  const tracer = trace.getTracer(TRACER_NAME);\n  const model = chat?.model ?? chat?.modelVersion ?? '';\n  const span = tracer.startSpan('vertex_ai.chat.send_message', {\n    attributes: {\n      'neatlogs.span.kind': 'LLM',\n      'neatlogs.llm.provider': PROVIDER,\n      'neatlogs.llm.system': SYSTEM,\n      'neatlogs.llm.model_name': model,\n      'neatlogs.llm.is_streaming': isStream,\n    },\n  }, otelContext.active());\n\n  // The chat sendMessage param shape is { message } (string or Part[]).\n  const message = params?.message ?? params;\n  span.setAttribute('neatlogs.llm.input_messages.0.role', 'user');\n  span.setAttribute(\n    'neatlogs.llm.input_messages.0.content',\n    (typeof message === 'string' ? message : safeStringify(message)),\n  );\n\n  const ctx = trace.setSpan(otelContext.active(), span);\n  const result = otelContext.with(ctx, () => original(params, ...rest));\n\n  return Promise.resolve(result).then(\n    (response: any) => {\n      if (isStream) return wrapStream(response, span);\n      finalizeResponse(span, response);\n      return response;\n    },\n    (err: any) => {\n      recordError(span, err);\n      throw err;\n    },\n  );\n}\n\nfunction tracedEmbedContent(original: (...a: any[]) => any) {\n  return function (opts: any, ...rest: any[]): any {\n    const tracer = trace.getTracer(TRACER_NAME);\n    const span = tracer.startSpan('vertex_ai.models.embed_content', {\n      attributes: {\n        'neatlogs.span.kind': 'EMBEDDING',\n        'neatlogs.llm.provider': PROVIDER,\n        'neatlogs.embedding.model_name': opts?.model ?? '',\n        'neatlogs.embedding.text': safeStringify(opts?.contents ?? ''),\n      },\n    }, otelContext.active());\n\n    const ctx = trace.setSpan(otelContext.active(), span);\n    const result = otelContext.with(ctx, () => original(opts, ...rest));\n\n    return Promise.resolve(result).then(\n      (response: any) => {\n        const embeddings = response?.embeddings;\n        if (Array.isArray(embeddings)) {\n          span.setAttribute('neatlogs.embedding.count', embeddings.length);\n          const vals = embeddings[0]?.values;\n          if (Array.isArray(vals)) span.setAttribute('neatlogs.embedding.dimensions', vals.length);\n        }\n        span.setStatus({ code: SpanStatusCode.OK });\n        span.end();\n        return response;\n      },\n      (err: any) => {\n        recordError(span, err);\n        throw err;\n      },\n    );\n  };\n}\n\nfunction tracedCountTokens(original: (...a: any[]) => any) {\n  return function (opts: any, ...rest: any[]): any {\n    const tracer = trace.getTracer(TRACER_NAME);\n    const span = tracer.startSpan('vertex_ai.models.count_tokens', {\n      attributes: {\n        'neatlogs.span.kind': 'LLM',\n        'neatlogs.llm.provider': PROVIDER,\n        'neatlogs.llm.task': 'count_tokens',\n        'neatlogs.llm.model_name': opts?.model ?? '',\n      },\n    }, otelContext.active());\n\n    const ctx = trace.setSpan(otelContext.active(), span);\n    const result = otelContext.with(ctx, () => original(opts, ...rest));\n\n    return Promise.resolve(result).then(\n      (response: any) => {\n        if (response?.totalTokens != null) span.setAttribute('neatlogs.llm.token_count.prompt', response.totalTokens);\n        span.setStatus({ code: SpanStatusCode.OK });\n        span.end();\n        return response;\n      },\n      (err: any) => {\n        recordError(span, err);\n        throw err;\n      },\n    );\n  };\n}\n\n// ---------------------------------------------------------------------------\n// generateContent / generateContentStream\n// ---------------------------------------------------------------------------\n\nfunction tracedGenerateContent(original: (...args: any[]) => any, isStream: boolean) {\n  return function (opts: any, ...rest: any[]): any {\n    const tracer = trace.getTracer(TRACER_NAME);\n    const model = opts?.model ?? '';\n\n    const span = tracer.startSpan('vertex_ai.models.generate_content', {\n      attributes: {\n        'neatlogs.span.kind': 'LLM',\n        'neatlogs.llm.provider': PROVIDER,\n        'neatlogs.llm.system': SYSTEM,\n        'neatlogs.llm.model_name': model,\n        'neatlogs.llm.is_streaming': isStream,\n      },\n    }, otelContext.active());\n\n    setInputAttributes(span, opts);\n\n    const ctx = trace.setSpan(otelContext.active(), span);\n    const result = otelContext.with(ctx, () => original(opts, ...rest));\n\n    return Promise.resolve(result).then(\n      (response: any) => {\n        if (isStream) {\n          return wrapStream(response, span);\n        }\n        finalizeResponse(span, response);\n        return response;\n      },\n      (err: any) => {\n        recordError(span, err);\n        throw err;\n      },\n    );\n  };\n}\n\nfunction setInputAttributes(span: Span, opts: any): void {\n  let idx = 0;\n\n  const config = opts?.config;\n  const systemInstruction = config?.systemInstruction ?? config?.system_instruction;\n  if (systemInstruction) {\n    span.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, 'system');\n    span.setAttribute(\n      `neatlogs.llm.input_messages.${idx}.content`,\n      typeof systemInstruction === 'string' ? systemInstruction : safeStringify(systemInstruction),\n    );\n    idx++;\n  }\n\n  const contents = opts?.contents;\n  if (typeof contents === 'string') {\n    span.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, 'user');\n    span.setAttribute(`neatlogs.llm.input_messages.${idx}.content`, contents);\n  } else if (Array.isArray(contents)) {\n    for (const item of contents) {\n      if (typeof item === 'string') {\n        span.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, 'user');\n        span.setAttribute(`neatlogs.llm.input_messages.${idx}.content`, item);\n        idx++;\n      } else if (item && typeof item === 'object') {\n        const role = item.role ?? 'user';\n        span.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, role);\n        const parts = item.parts ?? [];\n        const textParts: string[] = [];\n        for (const part of parts) {\n          if (typeof part === 'string') textParts.push(part);\n          else if (part?.text) textParts.push(part.text);\n        }\n        span.setAttribute(\n          `neatlogs.llm.input_messages.${idx}.content`,\n          textParts.length ? textParts.join('\\n') : safeStringify(parts),\n        );\n        idx++;\n      }\n    }\n  } else if (contents) {\n    span.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, 'user');\n    span.setAttribute(`neatlogs.llm.input_messages.${idx}.content`, safeStringify(contents));\n  }\n\n  // Tools (function declarations)\n  const tools = config?.tools;\n  if (Array.isArray(tools)) {\n    let t = 0;\n    for (const tool of tools) {\n      const decls = tool?.functionDeclarations ?? tool?.function_declarations ?? [];\n      for (const fn of decls) {\n        span.setAttribute(`neatlogs.llm.tools.${t}.name`, fn?.name ?? '');\n        if (fn?.description) span.setAttribute(`neatlogs.llm.tools.${t}.description`, fn.description);\n        if (fn?.parameters) span.setAttribute(`neatlogs.llm.tools.${t}.input_schema`, safeStringify(fn.parameters));\n        t++;\n      }\n    }\n  }\n\n  // Invocation params\n  if (config) {\n    if (config.temperature != null) span.setAttribute('neatlogs.llm.temperature', config.temperature);\n    if (config.topP != null) span.setAttribute('neatlogs.llm.top_p', config.topP);\n    if (config.topK != null) span.setAttribute('neatlogs.llm.top_k', config.topK);\n    const maxTokens = config.maxOutputTokens ?? config.max_output_tokens;\n    if (maxTokens != null) span.setAttribute('neatlogs.llm.max_tokens', maxTokens);\n\n    // Blob the backend reads for metadata.model_settings (the UI source).\n    const params: Record<string, any> = {};\n    if (config.temperature != null) params.temperature = config.temperature;\n    if (config.topP != null) params.top_p = config.topP;\n    if (config.topK != null) params.top_k = config.topK;\n    if (maxTokens != null) params.max_tokens = maxTokens;\n    if (config.frequencyPenalty != null) params.frequency_penalty = config.frequencyPenalty;\n    if (config.presencePenalty != null) params.presence_penalty = config.presencePenalty;\n    if (Object.keys(params).length > 0) {\n      span.setAttribute('neatlogs.llm.invocation_parameters', JSON.stringify(params));\n    }\n  }\n}\n\n// ---------------------------------------------------------------------------\n// Streaming support — @google/genai stream is an async iterable of chunks\n// ---------------------------------------------------------------------------\n\nfunction wrapStream(stream: any, span: Span): any {\n  const chunks: any[] = [];\n  const originalAsyncIterator = stream?.[Symbol.asyncIterator]?.bind(stream);\n\n  if (!originalAsyncIterator) {\n    finalizeStreamChunks(span, chunks);\n    return stream;\n  }\n\n  const wrapped = Object.create(Object.getPrototypeOf(stream));\n  Object.assign(wrapped, stream);\n\n  wrapped[Symbol.asyncIterator] = function () {\n    const iterator = originalAsyncIterator();\n    return {\n      async next(): Promise<IteratorResult<any>> {\n        try {\n          const result = await iterator.next();\n          if (result.done) {\n            finalizeStreamChunks(span, chunks);\n            return result;\n          }\n          chunks.push(result.value);\n          return result;\n        } catch (err) {\n          recordError(span, err);\n          throw err;\n        }\n      },\n      async return(value?: any): Promise<IteratorResult<any>> {\n        finalizeStreamChunks(span, chunks);\n        return iterator.return?.(value) ?? { done: true, value: undefined };\n      },\n      async throw(err?: any): Promise<IteratorResult<any>> {\n        recordError(span, err);\n        return iterator.throw?.(err) ?? { done: true, value: undefined };\n      },\n    };\n  };\n\n  return wrapped;\n}\n\nfunction finalizeStreamChunks(span: Span, chunks: any[]): void {\n  const textParts: string[] = [];\n  let finishReason = '';\n  let usage: any = null;\n\n  for (const chunk of chunks) {\n    for (const candidate of chunk?.candidates ?? []) {\n      for (const part of candidate?.content?.parts ?? []) {\n        if (part?.text && !part?.thought) textParts.push(part.text);\n      }\n      if (candidate?.finishReason) finishReason = candidate.finishReason;\n    }\n    if (chunk?.usageMetadata) usage = chunk.usageMetadata;\n  }\n\n  const fullText = textParts.join('');\n  if (fullText) {\n    span.setAttribute('neatlogs.llm.output_messages.0.role', 'assistant');\n    span.setAttribute('neatlogs.llm.output_messages.0.content', fullText);\n  }\n  if (finishReason) span.setAttribute('neatlogs.llm.finish_reason', String(finishReason));\n  setUsage(span, usage);\n\n  span.setStatus({ code: SpanStatusCode.OK });\n  span.end();\n}\n\n// ---------------------------------------------------------------------------\n// Non-streaming response finalization\n// ---------------------------------------------------------------------------\n\nfunction finalizeResponse(span: Span, response: any): void {\n  const textParts: string[] = [];\n  let toolIdx = 0;\n\n  for (const candidate of response?.candidates ?? []) {\n    for (const part of candidate?.content?.parts ?? []) {\n      if (part?.text && !part?.thought) {\n        textParts.push(part.text);\n      } else if (part?.thought && part?.text) {\n        span.setAttribute('neatlogs.llm.output_messages.0.thinking', part.text);\n      } else if (part?.functionCall) {\n        const fc = part.functionCall;\n        span.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.name`, fc?.name ?? '');\n        span.setAttribute(`neatlogs.llm.tool_calls.${toolIdx}.arguments`, safeStringify(fc?.args ?? {}));\n        toolIdx++;\n      }\n    }\n    if (candidate?.finishReason) span.setAttribute('neatlogs.llm.finish_reason', String(candidate.finishReason));\n  }\n\n  if (textParts.length) {\n    span.setAttribute('neatlogs.llm.output_messages.0.role', 'assistant');\n    span.setAttribute('neatlogs.llm.output_messages.0.content', textParts.join(''));\n  }\n\n  setUsage(span, response?.usageMetadata);\n\n  span.setStatus({ code: SpanStatusCode.OK });\n  span.end();\n}\n\nfunction setUsage(span: Span, usage: any): void {\n  if (!usage) return;\n  if (usage.promptTokenCount != null) span.setAttribute('neatlogs.llm.token_count.prompt', usage.promptTokenCount);\n  if (usage.candidatesTokenCount != null) span.setAttribute('neatlogs.llm.token_count.completion', usage.candidatesTokenCount);\n  if (usage.totalTokenCount != null) span.setAttribute('neatlogs.llm.token_count.total', usage.totalTokenCount);\n  if (usage.cachedContentTokenCount != null) span.setAttribute('neatlogs.llm.token_count.cache_read', usage.cachedContentTokenCount);\n  if (usage.thoughtsTokenCount != null) span.setAttribute('neatlogs.llm.token_count.reasoning', usage.thoughtsTokenCount);\n}\n\n// ---------------------------------------------------------------------------\n// Utilities\n// ---------------------------------------------------------------------------\n\nfunction safeStringify(value: unknown): string {\n  try {\n    return typeof value === 'string' ? value : JSON.stringify(value);\n  } catch {\n    return '';\n  }\n}\n\nfunction recordError(span: Span, err: unknown): void {\n  if (err instanceof Error) {\n    span.setStatus({ code: SpanStatusCode.ERROR, message: err.message });\n    span.recordException(err);\n  } else {\n    span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });\n  }\n  span.end();\n}\n"],"mappings":";AAsBA,SAAS,OAAO,WAAW,aAAa,sBAAiC;AAEzE,IAAM,cAAc;AACpB,IAAM,WAAW;AACjB,IAAM,SAAS;AAMR,SAAS,aAA+B,QAAc;AAC3D,SAAO,cAAc,QAAQ,CAAC,CAAC;AACjC;AAUO,SAAS,UACd,MACA,IACmC;AACnC,SAAO,eAAe,WAAW,MAA+B;AAC9D,UAAM,SAAS,MAAM,UAAU,WAAW;AAC1C,WAAO,OAAO;AAAA,MACZ,QAAQ,IAAI;AAAA,MACZ;AAAA,QACE,YAAY;AAAA,UACV,sBAAsB;AAAA,UACtB,sBAAsB;AAAA,UACtB,eAAe,cAAc,IAAI;AAAA,QACnC;AAAA,MACF;AAAA,MACA,YAAY,OAAO;AAAA,MACnB,OAAO,SAAS;AACd,YAAI;AACF,gBAAM,SAAS,MAAM,GAAG,IAAI;AAC5B,eAAK,aAAa,gBAAgB,cAAc,MAAM,CAAC;AACvD,eAAK,UAAU,EAAE,MAAM,eAAe,GAAG,CAAC;AAC1C,iBAAO;AAAA,QACT,SAAS,KAAK;AACZ,sBAAY,MAAM,GAAG;AACrB,gBAAM;AAAA,QACR,UAAE;AACA,eAAK,IAAI;AAAA,QACX;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AAMA,SAAS,cAAc,QAAa,MAAqB;AACvD,SAAO,IAAI,MAAM,QAAQ;AAAA,IACvB,IAAI,KAAK,MAAM,UAAU;AACvB,YAAM,QAAQ,QAAQ,IAAI,KAAK,MAAM,QAAQ;AAC7C,UAAI,OAAO,SAAS,YAAY,OAAO,IAAI,EAAE,WAAW,GAAG,EAAG,QAAO;AAErE,YAAM,cAAc,CAAC,GAAG,MAAM,OAAO,IAAI,CAAC;AAC1C,YAAM,UAAU,YAAY,KAAK,GAAG;AAEpC,UAAI,YAAY,4BAA4B,OAAO,UAAU,YAAY;AACvE,eAAO,sBAAsB,MAAM,KAAK,GAAG,GAAG,KAAK;AAAA,MACrD;AACA,UAAI,YAAY,kCAAkC,OAAO,UAAU,YAAY;AAC7E,eAAO,sBAAsB,MAAM,KAAK,GAAG,GAAG,IAAI;AAAA,MACpD;AACA,UAAI,YAAY,yBAAyB,OAAO,UAAU,YAAY;AACpE,eAAO,mBAAmB,MAAM,KAAK,GAAG,CAAC;AAAA,MAC3C;AACA,UAAI,YAAY,wBAAwB,OAAO,UAAU,YAAY;AACnE,eAAO,kBAAkB,MAAM,KAAK,GAAG,CAAC;AAAA,MAC1C;AAEA,UAAI,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK,KAAK,YAAY,WAAW,GAAG;AAC3F,eAAO,cAAc,OAAO,WAAW;AAAA,MACzC;AAEA,aAAO;AAAA,IACT;AAAA,EACF,CAAC;AACH;AAEA,SAAS,YAAY,MAAyB;AAC5C,MAAI,KAAK,SAAS,EAAG,QAAO;AAC5B,SAAO,CAAC,UAAU,OAAO,EAAE,SAAS,KAAK,KAAK,SAAS,CAAC,CAAC;AAC3D;AAOO,SAAS,iBAAmC,MAAY;AAC7D,QAAM,IAAI;AACV,MAAI,CAAC,KAAK,EAAE,uBAAwB,QAAO;AAE3C,MAAI,OAAO,EAAE,gBAAgB,YAAY;AACvC,UAAM,OAAO,EAAE,YAAY,KAAK,CAAC;AACjC,MAAE,cAAc,CAAC,WAAgB,SAAgB,eAAe,MAAM,GAAG,QAAQ,MAAM,KAAK;AAAA,EAC9F;AACA,MAAI,OAAO,EAAE,sBAAsB,YAAY;AAC7C,UAAM,OAAO,EAAE,kBAAkB,KAAK,CAAC;AACvC,MAAE,oBAAoB,CAAC,WAAgB,SAAgB,eAAe,MAAM,GAAG,QAAQ,MAAM,IAAI;AAAA,EACnG;AAEA,MAAI;AACF,WAAO,eAAe,GAAG,0BAA0B,EAAE,OAAO,MAAM,YAAY,OAAO,cAAc,KAAK,CAAC;AAAA,EAC3G,QAAQ;AACN,MAAE,yBAAyB;AAAA,EAC7B;AACA,SAAO;AACT;AAEA,SAAS,eACP,UACA,MACA,QACA,MACA,UACK;AACL,QAAM,SAAS,MAAM,UAAU,WAAW;AAC1C,QAAM,QAAQ,MAAM,SAAS,MAAM,gBAAgB;AACnD,QAAM,OAAO,OAAO,UAAU,+BAA+B;AAAA,IAC3D,YAAY;AAAA,MACV,sBAAsB;AAAA,MACtB,yBAAyB;AAAA,MACzB,uBAAuB;AAAA,MACvB,2BAA2B;AAAA,MAC3B,6BAA6B;AAAA,IAC/B;AAAA,EACF,GAAG,YAAY,OAAO,CAAC;AAGvB,QAAM,UAAU,QAAQ,WAAW;AACnC,OAAK,aAAa,sCAAsC,MAAM;AAC9D,OAAK;AAAA,IACH;AAAA,IACC,OAAO,YAAY,WAAW,UAAU,cAAc,OAAO;AAAA,EAChE;AAEA,QAAM,MAAM,MAAM,QAAQ,YAAY,OAAO,GAAG,IAAI;AACpD,QAAM,SAAS,YAAY,KAAK,KAAK,MAAM,SAAS,QAAQ,GAAG,IAAI,CAAC;AAEpE,SAAO,QAAQ,QAAQ,MAAM,EAAE;AAAA,IAC7B,CAAC,aAAkB;AACjB,UAAI,SAAU,QAAO,WAAW,UAAU,IAAI;AAC9C,uBAAiB,MAAM,QAAQ;AAC/B,aAAO;AAAA,IACT;AAAA,IACA,CAAC,QAAa;AACZ,kBAAY,MAAM,GAAG;AACrB,YAAM;AAAA,IACR;AAAA,EACF;AACF;AAEA,SAAS,mBAAmB,UAAgC;AAC1D,SAAO,SAAU,SAAc,MAAkB;AAC/C,UAAM,SAAS,MAAM,UAAU,WAAW;AAC1C,UAAM,OAAO,OAAO,UAAU,kCAAkC;AAAA,MAC9D,YAAY;AAAA,QACV,sBAAsB;AAAA,QACtB,yBAAyB;AAAA,QACzB,iCAAiC,MAAM,SAAS;AAAA,QAChD,2BAA2B,cAAc,MAAM,YAAY,EAAE;AAAA,MAC/D;AAAA,IACF,GAAG,YAAY,OAAO,CAAC;AAEvB,UAAM,MAAM,MAAM,QAAQ,YAAY,OAAO,GAAG,IAAI;AACpD,UAAM,SAAS,YAAY,KAAK,KAAK,MAAM,SAAS,MAAM,GAAG,IAAI,CAAC;AAElE,WAAO,QAAQ,QAAQ,MAAM,EAAE;AAAA,MAC7B,CAAC,aAAkB;AACjB,cAAM,aAAa,UAAU;AAC7B,YAAI,MAAM,QAAQ,UAAU,GAAG;AAC7B,eAAK,aAAa,4BAA4B,WAAW,MAAM;AAC/D,gBAAM,OAAO,WAAW,CAAC,GAAG;AAC5B,cAAI,MAAM,QAAQ,IAAI,EAAG,MAAK,aAAa,iCAAiC,KAAK,MAAM;AAAA,QACzF;AACA,aAAK,UAAU,EAAE,MAAM,eAAe,GAAG,CAAC;AAC1C,aAAK,IAAI;AACT,eAAO;AAAA,MACT;AAAA,MACA,CAAC,QAAa;AACZ,oBAAY,MAAM,GAAG;AACrB,cAAM;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,kBAAkB,UAAgC;AACzD,SAAO,SAAU,SAAc,MAAkB;AAC/C,UAAM,SAAS,MAAM,UAAU,WAAW;AAC1C,UAAM,OAAO,OAAO,UAAU,iCAAiC;AAAA,MAC7D,YAAY;AAAA,QACV,sBAAsB;AAAA,QACtB,yBAAyB;AAAA,QACzB,qBAAqB;AAAA,QACrB,2BAA2B,MAAM,SAAS;AAAA,MAC5C;AAAA,IACF,GAAG,YAAY,OAAO,CAAC;AAEvB,UAAM,MAAM,MAAM,QAAQ,YAAY,OAAO,GAAG,IAAI;AACpD,UAAM,SAAS,YAAY,KAAK,KAAK,MAAM,SAAS,MAAM,GAAG,IAAI,CAAC;AAElE,WAAO,QAAQ,QAAQ,MAAM,EAAE;AAAA,MAC7B,CAAC,aAAkB;AACjB,YAAI,UAAU,eAAe,KAAM,MAAK,aAAa,mCAAmC,SAAS,WAAW;AAC5G,aAAK,UAAU,EAAE,MAAM,eAAe,GAAG,CAAC;AAC1C,aAAK,IAAI;AACT,eAAO;AAAA,MACT;AAAA,MACA,CAAC,QAAa;AACZ,oBAAY,MAAM,GAAG;AACrB,cAAM;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACF;AAMA,SAAS,sBAAsB,UAAmC,UAAmB;AACnF,SAAO,SAAU,SAAc,MAAkB;AAC/C,UAAM,SAAS,MAAM,UAAU,WAAW;AAC1C,UAAM,QAAQ,MAAM,SAAS;AAE7B,UAAM,OAAO,OAAO,UAAU,qCAAqC;AAAA,MACjE,YAAY;AAAA,QACV,sBAAsB;AAAA,QACtB,yBAAyB;AAAA,QACzB,uBAAuB;AAAA,QACvB,2BAA2B;AAAA,QAC3B,6BAA6B;AAAA,MAC/B;AAAA,IACF,GAAG,YAAY,OAAO,CAAC;AAEvB,uBAAmB,MAAM,IAAI;AAE7B,UAAM,MAAM,MAAM,QAAQ,YAAY,OAAO,GAAG,IAAI;AACpD,UAAM,SAAS,YAAY,KAAK,KAAK,MAAM,SAAS,MAAM,GAAG,IAAI,CAAC;AAElE,WAAO,QAAQ,QAAQ,MAAM,EAAE;AAAA,MAC7B,CAAC,aAAkB;AACjB,YAAI,UAAU;AACZ,iBAAO,WAAW,UAAU,IAAI;AAAA,QAClC;AACA,yBAAiB,MAAM,QAAQ;AAC/B,eAAO;AAAA,MACT;AAAA,MACA,CAAC,QAAa;AACZ,oBAAY,MAAM,GAAG;AACrB,cAAM;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,mBAAmB,MAAY,MAAiB;AACvD,MAAI,MAAM;AAEV,QAAM,SAAS,MAAM;AACrB,QAAM,oBAAoB,QAAQ,qBAAqB,QAAQ;AAC/D,MAAI,mBAAmB;AACrB,SAAK,aAAa,+BAA+B,GAAG,SAAS,QAAQ;AACrE,SAAK;AAAA,MACH,+BAA+B,GAAG;AAAA,MAClC,OAAO,sBAAsB,WAAW,oBAAoB,cAAc,iBAAiB;AAAA,IAC7F;AACA;AAAA,EACF;AAEA,QAAM,WAAW,MAAM;AACvB,MAAI,OAAO,aAAa,UAAU;AAChC,SAAK,aAAa,+BAA+B,GAAG,SAAS,MAAM;AACnE,SAAK,aAAa,+BAA+B,GAAG,YAAY,QAAQ;AAAA,EAC1E,WAAW,MAAM,QAAQ,QAAQ,GAAG;AAClC,eAAW,QAAQ,UAAU;AAC3B,UAAI,OAAO,SAAS,UAAU;AAC5B,aAAK,aAAa,+BAA+B,GAAG,SAAS,MAAM;AACnE,aAAK,aAAa,+BAA+B,GAAG,YAAY,IAAI;AACpE;AAAA,MACF,WAAW,QAAQ,OAAO,SAAS,UAAU;AAC3C,cAAM,OAAO,KAAK,QAAQ;AAC1B,aAAK,aAAa,+BAA+B,GAAG,SAAS,IAAI;AACjE,cAAM,QAAQ,KAAK,SAAS,CAAC;AAC7B,cAAM,YAAsB,CAAC;AAC7B,mBAAW,QAAQ,OAAO;AACxB,cAAI,OAAO,SAAS,SAAU,WAAU,KAAK,IAAI;AAAA,mBACxC,MAAM,KAAM,WAAU,KAAK,KAAK,IAAI;AAAA,QAC/C;AACA,aAAK;AAAA,UACH,+BAA+B,GAAG;AAAA,UAClC,UAAU,SAAS,UAAU,KAAK,IAAI,IAAI,cAAc,KAAK;AAAA,QAC/D;AACA;AAAA,MACF;AAAA,IACF;AAAA,EACF,WAAW,UAAU;AACnB,SAAK,aAAa,+BAA+B,GAAG,SAAS,MAAM;AACnE,SAAK,aAAa,+BAA+B,GAAG,YAAY,cAAc,QAAQ,CAAC;AAAA,EACzF;AAGA,QAAM,QAAQ,QAAQ;AACtB,MAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,QAAI,IAAI;AACR,eAAW,QAAQ,OAAO;AACxB,YAAM,QAAQ,MAAM,wBAAwB,MAAM,yBAAyB,CAAC;AAC5E,iBAAW,MAAM,OAAO;AACtB,aAAK,aAAa,sBAAsB,CAAC,SAAS,IAAI,QAAQ,EAAE;AAChE,YAAI,IAAI,YAAa,MAAK,aAAa,sBAAsB,CAAC,gBAAgB,GAAG,WAAW;AAC5F,YAAI,IAAI,WAAY,MAAK,aAAa,sBAAsB,CAAC,iBAAiB,cAAc,GAAG,UAAU,CAAC;AAC1G;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAGA,MAAI,QAAQ;AACV,QAAI,OAAO,eAAe,KAAM,MAAK,aAAa,4BAA4B,OAAO,WAAW;AAChG,QAAI,OAAO,QAAQ,KAAM,MAAK,aAAa,sBAAsB,OAAO,IAAI;AAC5E,QAAI,OAAO,QAAQ,KAAM,MAAK,aAAa,sBAAsB,OAAO,IAAI;AAC5E,UAAM,YAAY,OAAO,mBAAmB,OAAO;AACnD,QAAI,aAAa,KAAM,MAAK,aAAa,2BAA2B,SAAS;AAG7E,UAAM,SAA8B,CAAC;AACrC,QAAI,OAAO,eAAe,KAAM,QAAO,cAAc,OAAO;AAC5D,QAAI,OAAO,QAAQ,KAAM,QAAO,QAAQ,OAAO;AAC/C,QAAI,OAAO,QAAQ,KAAM,QAAO,QAAQ,OAAO;AAC/C,QAAI,aAAa,KAAM,QAAO,aAAa;AAC3C,QAAI,OAAO,oBAAoB,KAAM,QAAO,oBAAoB,OAAO;AACvE,QAAI,OAAO,mBAAmB,KAAM,QAAO,mBAAmB,OAAO;AACrE,QAAI,OAAO,KAAK,MAAM,EAAE,SAAS,GAAG;AAClC,WAAK,aAAa,sCAAsC,KAAK,UAAU,MAAM,CAAC;AAAA,IAChF;AAAA,EACF;AACF;AAMA,SAAS,WAAW,QAAa,MAAiB;AAChD,QAAM,SAAgB,CAAC;AACvB,QAAM,wBAAwB,SAAS,OAAO,aAAa,GAAG,KAAK,MAAM;AAEzE,MAAI,CAAC,uBAAuB;AAC1B,yBAAqB,MAAM,MAAM;AACjC,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,OAAO,OAAO,OAAO,eAAe,MAAM,CAAC;AAC3D,SAAO,OAAO,SAAS,MAAM;AAE7B,UAAQ,OAAO,aAAa,IAAI,WAAY;AAC1C,UAAM,WAAW,sBAAsB;AACvC,WAAO;AAAA,MACL,MAAM,OAAqC;AACzC,YAAI;AACF,gBAAM,SAAS,MAAM,SAAS,KAAK;AACnC,cAAI,OAAO,MAAM;AACf,iCAAqB,MAAM,MAAM;AACjC,mBAAO;AAAA,UACT;AACA,iBAAO,KAAK,OAAO,KAAK;AACxB,iBAAO;AAAA,QACT,SAAS,KAAK;AACZ,sBAAY,MAAM,GAAG;AACrB,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,MACA,MAAM,OAAO,OAA2C;AACtD,6BAAqB,MAAM,MAAM;AACjC,eAAO,SAAS,SAAS,KAAK,KAAK,EAAE,MAAM,MAAM,OAAO,OAAU;AAAA,MACpE;AAAA,MACA,MAAM,MAAM,KAAyC;AACnD,oBAAY,MAAM,GAAG;AACrB,eAAO,SAAS,QAAQ,GAAG,KAAK,EAAE,MAAM,MAAM,OAAO,OAAU;AAAA,MACjE;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;AAEA,SAAS,qBAAqB,MAAY,QAAqB;AAC7D,QAAM,YAAsB,CAAC;AAC7B,MAAI,eAAe;AACnB,MAAI,QAAa;AAEjB,aAAW,SAAS,QAAQ;AAC1B,eAAW,aAAa,OAAO,cAAc,CAAC,GAAG;AAC/C,iBAAW,QAAQ,WAAW,SAAS,SAAS,CAAC,GAAG;AAClD,YAAI,MAAM,QAAQ,CAAC,MAAM,QAAS,WAAU,KAAK,KAAK,IAAI;AAAA,MAC5D;AACA,UAAI,WAAW,aAAc,gBAAe,UAAU;AAAA,IACxD;AACA,QAAI,OAAO,cAAe,SAAQ,MAAM;AAAA,EAC1C;AAEA,QAAM,WAAW,UAAU,KAAK,EAAE;AAClC,MAAI,UAAU;AACZ,SAAK,aAAa,uCAAuC,WAAW;AACpE,SAAK,aAAa,0CAA0C,QAAQ;AAAA,EACtE;AACA,MAAI,aAAc,MAAK,aAAa,8BAA8B,OAAO,YAAY,CAAC;AACtF,WAAS,MAAM,KAAK;AAEpB,OAAK,UAAU,EAAE,MAAM,eAAe,GAAG,CAAC;AAC1C,OAAK,IAAI;AACX;AAMA,SAAS,iBAAiB,MAAY,UAAqB;AACzD,QAAM,YAAsB,CAAC;AAC7B,MAAI,UAAU;AAEd,aAAW,aAAa,UAAU,cAAc,CAAC,GAAG;AAClD,eAAW,QAAQ,WAAW,SAAS,SAAS,CAAC,GAAG;AAClD,UAAI,MAAM,QAAQ,CAAC,MAAM,SAAS;AAChC,kBAAU,KAAK,KAAK,IAAI;AAAA,MAC1B,WAAW,MAAM,WAAW,MAAM,MAAM;AACtC,aAAK,aAAa,2CAA2C,KAAK,IAAI;AAAA,MACxE,WAAW,MAAM,cAAc;AAC7B,cAAM,KAAK,KAAK;AAChB,aAAK,aAAa,2BAA2B,OAAO,SAAS,IAAI,QAAQ,EAAE;AAC3E,aAAK,aAAa,2BAA2B,OAAO,cAAc,cAAc,IAAI,QAAQ,CAAC,CAAC,CAAC;AAC/F;AAAA,MACF;AAAA,IACF;AACA,QAAI,WAAW,aAAc,MAAK,aAAa,8BAA8B,OAAO,UAAU,YAAY,CAAC;AAAA,EAC7G;AAEA,MAAI,UAAU,QAAQ;AACpB,SAAK,aAAa,uCAAuC,WAAW;AACpE,SAAK,aAAa,0CAA0C,UAAU,KAAK,EAAE,CAAC;AAAA,EAChF;AAEA,WAAS,MAAM,UAAU,aAAa;AAEtC,OAAK,UAAU,EAAE,MAAM,eAAe,GAAG,CAAC;AAC1C,OAAK,IAAI;AACX;AAEA,SAAS,SAAS,MAAY,OAAkB;AAC9C,MAAI,CAAC,MAAO;AACZ,MAAI,MAAM,oBAAoB,KAAM,MAAK,aAAa,mCAAmC,MAAM,gBAAgB;AAC/G,MAAI,MAAM,wBAAwB,KAAM,MAAK,aAAa,uCAAuC,MAAM,oBAAoB;AAC3H,MAAI,MAAM,mBAAmB,KAAM,MAAK,aAAa,kCAAkC,MAAM,eAAe;AAC5G,MAAI,MAAM,2BAA2B,KAAM,MAAK,aAAa,uCAAuC,MAAM,uBAAuB;AACjI,MAAI,MAAM,sBAAsB,KAAM,MAAK,aAAa,sCAAsC,MAAM,kBAAkB;AACxH;AAMA,SAAS,cAAc,OAAwB;AAC7C,MAAI;AACF,WAAO,OAAO,UAAU,WAAW,QAAQ,KAAK,UAAU,KAAK;AAAA,EACjE,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEA,SAAS,YAAY,MAAY,KAAoB;AACnD,MAAI,eAAe,OAAO;AACxB,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,SAAS,IAAI,QAAQ,CAAC;AACnE,SAAK,gBAAgB,GAAG;AAAA,EAC1B,OAAO;AACL,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,SAAS,OAAO,GAAG,EAAE,CAAC;AAAA,EACrE;AACA,OAAK,IAAI;AACX;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "neatlogs",
-  "version": "1.0.6",
+  "version": "1.0.8",
   "description": "AI agent debugging, collaboration, and trace observability. Built for teams using CrewAI, OpenAI, and more.",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",
@@ -46,6 +46,66 @@
         "default": "./dist/anthropic.cjs"
       }
     },
+    "./azure-openai": {
+      "import": {
+        "types": "./dist/azure-openai.d.ts",
+        "default": "./dist/azure-openai.mjs"
+      },
+      "require": {
+        "types": "./dist/azure-openai.d.ts",
+        "default": "./dist/azure-openai.cjs"
+      }
+    },
+    "./vertex-ai": {
+      "import": {
+        "types": "./dist/vertex-ai.d.ts",
+        "default": "./dist/vertex-ai.mjs"
+      },
+      "require": {
+        "types": "./dist/vertex-ai.d.ts",
+        "default": "./dist/vertex-ai.cjs"
+      }
+    },
+    "./bedrock": {
+      "import": {
+        "types": "./dist/bedrock.d.ts",
+        "default": "./dist/bedrock.mjs"
+      },
+      "require": {
+        "types": "./dist/bedrock.d.ts",
+        "default": "./dist/bedrock.cjs"
+      }
+    },
+    "./claude-agent-sdk": {
+      "import": {
+        "types": "./dist/claude-agent-sdk.d.ts",
+        "default": "./dist/claude-agent-sdk.mjs"
+      },
+      "require": {
+        "types": "./dist/claude-agent-sdk.d.ts",
+        "default": "./dist/claude-agent-sdk.cjs"
+      }
+    },
+    "./openrouter-agent": {
+      "import": {
+        "types": "./dist/openrouter-agent.d.ts",
+        "default": "./dist/openrouter-agent.mjs"
+      },
+      "require": {
+        "types": "./dist/openrouter-agent.d.ts",
+        "default": "./dist/openrouter-agent.cjs"
+      }
+    },
+    "./opencode": {
+      "import": {
+        "types": "./dist/opencode-plugin.d.ts",
+        "default": "./dist/opencode-plugin.mjs"
+      },
+      "require": {
+        "types": "./dist/opencode-plugin.d.ts",
+        "default": "./dist/opencode-plugin.cjs"
+      }
+    },
     "./langchain": {
       "import": {
         "types": "./dist/langchain.d.ts",
@@ -95,6 +155,16 @@
         "types": "./dist/pi-agent.d.ts",
         "default": "./dist/pi-agent.cjs"
       }
+    },
+    "./browser": {
+      "import": {
+        "types": "./dist/browser.d.ts",
+        "default": "./dist/browser.mjs"
+      },
+      "require": {
+        "types": "./dist/browser.d.ts",
+        "default": "./dist/browser.cjs"
+      }
     }
   },
   "files": [
@@ -118,6 +188,10 @@
     "example:langgraph": "tsx examples/langgraph_multiagent/main.ts",
     "example:marketing": "tsx examples/marketing_strategy_demo/main.ts",
     "example:reasoning": "tsx examples/reasoning_model_workflow/main.ts",
+    "example:azure-openai": "tsx examples/sdk_examples/azure_openai_basic/main.ts",
+    "example:vertex-ai": "tsx examples/sdk_examples/vertex_ai_basic/main.ts",
+    "example:bedrock": "tsx examples/sdk_examples/bedrock_basic/main.ts",
+    "example:openrouter": "tsx examples/sdk_examples/openrouter_agent_basic/main.ts",
     "test:examples:original7": "node scripts/run-original7-examples.mjs",
     "verify:logs": "node scripts/verify-original7-logs.mjs"
   },
@@ -163,6 +237,7 @@
     "@langchain/google-genai": "^0.2.18",
     "@langchain/langgraph": "^0.3.12",
     "@langchain/openai": "^0.6.17",
+    "@openrouter/agent": "^0.7.1",
     "@types/node": "^20.0.0",
     "ai": "^6.0.190",
     "dotenv": "^17.4.2",