neatlogs 1.0.6 → 1.0.8

package/dist/azure-openai.cjs

@@ -0,0 +1,344 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/azure-openai.ts
+var azure_openai_exports = {};
+__export(azure_openai_exports, {
+  traceTool: () => traceTool,
+  wrapAzureOpenAI: () => wrapAzureOpenAI
+});
+module.exports = __toCommonJS(azure_openai_exports);
+var import_api = require("@opentelemetry/api");
+var TRACER_NAME = "neatlogs.azure_openai";
+var PROVIDER = "azure";
+var SYSTEM = "azure";
+function wrapAzureOpenAI(client) {
+  return wrapNamespace(client, []);
+}
+function traceTool(name, fn) {
+  return async function tracedTool(args) {
+    const tracer = import_api.trace.getTracer(TRACER_NAME);
+    return tracer.startActiveSpan(
+      `tool.${name}`,
+      {
+        attributes: {
+          "neatlogs.span.kind": "TOOL",
+          "neatlogs.tool.name": name,
+          "input.value": safeStringify(args)
+        }
+      },
+      import_api.context.active(),
+      async (span) => {
+        try {
+          const result = await fn(args);
+          span.setAttribute("output.value", safeStringify(result));
+          span.setStatus({ code: import_api.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 === "chat.completions.create" && typeof value === "function") {
+        return tracedChatCompletionsCreate(value.bind(obj));
+      }
+      if (pathStr === "responses.create" && typeof value === "function") {
+        return tracedResponsesCreate(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 > 3) return false;
+  const key = path[path.length - 1];
+  return ["chat", "completions", "responses", "beta"].includes(key);
+}
+function tracedChatCompletionsCreate(original) {
+  return function(opts, ...rest) {
+    const tracer = import_api.trace.getTracer(TRACER_NAME);
+    const model = opts?.model ?? "";
+    const messages = opts?.messages ?? [];
+    const isStream = opts?.stream === true;
+    const span = tracer.startSpan("azure_openai.chat.completions.create", {
+      attributes: {
+        "neatlogs.span.kind": "LLM",
+        "neatlogs.llm.provider": PROVIDER,
+        "neatlogs.llm.system": SYSTEM,
+        "neatlogs.llm.model_name": model,
+        "neatlogs.llm.is_streaming": isStream
+      }
+    }, import_api.context.active());
+    for (let i = 0; i < messages.length; i++) {
+      const msg = messages[i];
+      span.setAttribute(`neatlogs.llm.input_messages.${i}.role`, msg.role ?? "");
+      if (typeof msg.content === "string") {
+        span.setAttribute(`neatlogs.llm.input_messages.${i}.content`, msg.content);
+      } else if (msg.content) {
+        span.setAttribute(`neatlogs.llm.input_messages.${i}.content`, safeStringify(msg.content));
+      }
+      if (msg.tool_call_id) {
+        span.setAttribute(`neatlogs.llm.input_messages.${i}.tool_call_id`, msg.tool_call_id);
+      }
+    }
+    if (opts?.tools) {
+      for (let i = 0; i < opts.tools.length; i++) {
+        const fn = opts.tools[i]?.function ?? {};
+        span.setAttribute(`neatlogs.llm.tools.${i}.name`, fn.name ?? "");
+        if (fn.description) span.setAttribute(`neatlogs.llm.tools.${i}.description`, fn.description);
+        if (fn.parameters) span.setAttribute(`neatlogs.llm.tools.${i}.input_schema`, safeStringify(fn.parameters));
+      }
+    }
+    setInvocationParams(span, opts);
+    if (isStream) {
+      opts = { ...opts };
+      const streamOpts = opts.stream_options ?? {};
+      if (!streamOpts.include_usage) {
+        opts.stream_options = { ...streamOpts, include_usage: true };
+      }
+    }
+    const ctx = import_api.trace.setSpan(import_api.context.active(), span);
+    const promise = import_api.context.with(ctx, () => original(opts, ...rest));
+    return promise.then(
+      (response) => {
+        if (isStream) {
+          return wrapAsyncIterableStream(response, span);
+        }
+        finalizeChatResponse(span, response);
+        return response;
+      },
+      (err) => {
+        recordError(span, err);
+        throw err;
+      }
+    );
+  };
+}
+function tracedResponsesCreate(original) {
+  return function(opts, ...rest) {
+    const tracer = import_api.trace.getTracer(TRACER_NAME);
+    const model = opts?.model ?? "";
+    const span = tracer.startSpan("azure_openai.responses.create", {
+      attributes: {
+        "neatlogs.span.kind": "LLM",
+        "neatlogs.llm.provider": PROVIDER,
+        "neatlogs.llm.system": SYSTEM,
+        "neatlogs.llm.model_name": model,
+        "input.value": safeStringify(opts?.input ?? "")
+      }
+    }, import_api.context.active());
+    const ctx = import_api.trace.setSpan(import_api.context.active(), span);
+    const promise = import_api.context.with(ctx, () => original(opts, ...rest));
+    return promise.then(
+      (response) => {
+        if (response?.output_text) {
+          span.setAttribute("neatlogs.llm.output_messages.0.role", "assistant");
+          span.setAttribute("neatlogs.llm.output_messages.0.content", response.output_text);
+        }
+        if (response?.model) span.setAttribute("neatlogs.llm.model_name", response.model);
+        if (response?.usage) {
+          if (response.usage.input_tokens != null) span.setAttribute("neatlogs.llm.token_count.prompt", response.usage.input_tokens);
+          if (response.usage.output_tokens != null) span.setAttribute("neatlogs.llm.token_count.completion", response.usage.output_tokens);
+        }
+        span.setStatus({ code: import_api.SpanStatusCode.OK });
+        span.end();
+        return response;
+      },
+      (err) => {
+        recordError(span, err);
+        throw err;
+      }
+    );
+  };
+}
+function wrapAsyncIterableStream(stream, span) {
+  const chunks = [];
+  const originalAsyncIterator = stream?.[Symbol.asyncIterator]?.bind(stream);
+  if (!originalAsyncIterator) {
+    span.setStatus({ code: import_api.SpanStatusCode.OK });
+    span.end();
+    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 = [];
+  const toolCallsAcc = {};
+  let finishReason = "";
+  let model = "";
+  let usage = null;
+  for (const chunk of chunks) {
+    if (!chunk?.choices?.length) {
+      if (chunk?.usage) usage = chunk.usage;
+      continue;
+    }
+    const choice = chunk.choices[0];
+    const delta = choice?.delta;
+    if (delta?.content) textParts.push(delta.content);
+    if (delta?.tool_calls) {
+      for (const tc of delta.tool_calls) {
+        const idx = tc.index ?? 0;
+        if (!toolCallsAcc[idx]) toolCallsAcc[idx] = { id: "", name: "", arguments: "" };
+        if (tc.id) toolCallsAcc[idx].id = tc.id;
+        if (tc.function?.name) toolCallsAcc[idx].name = tc.function.name;
+        if (tc.function?.arguments) toolCallsAcc[idx].arguments += tc.function.arguments;
+      }
+    }
+    if (choice?.finish_reason) finishReason = choice.finish_reason;
+    if (chunk?.model) model = chunk.model;
+  }
+  const fullText = textParts.join("");
+  if (fullText) {
+    span.setAttribute("neatlogs.llm.output_messages.0.role", "assistant");
+    span.setAttribute("neatlogs.llm.output_messages.0.content", fullText);
+  }
+  let j = 0;
+  for (const tc of Object.values(toolCallsAcc)) {
+    span.setAttribute(`neatlogs.llm.tool_calls.${j}.id`, tc.id);
+    span.setAttribute(`neatlogs.llm.tool_calls.${j}.name`, tc.name);
+    span.setAttribute(`neatlogs.llm.tool_calls.${j}.arguments`, tc.arguments);
+    j++;
+  }
+  if (model) span.setAttribute("neatlogs.llm.model_name", model);
+  if (finishReason) span.setAttribute("neatlogs.llm.finish_reason", finishReason);
+  if (usage) {
+    if (usage.prompt_tokens != null) span.setAttribute("neatlogs.llm.token_count.prompt", usage.prompt_tokens);
+    if (usage.completion_tokens != null) span.setAttribute("neatlogs.llm.token_count.completion", usage.completion_tokens);
+    if (usage.total_tokens != null) span.setAttribute("neatlogs.llm.token_count.total", usage.total_tokens);
+    if (usage.prompt_tokens_details?.cached_tokens != null) {
+      span.setAttribute("neatlogs.llm.token_count.cache_read", usage.prompt_tokens_details.cached_tokens);
+    }
+    if (usage.completion_tokens_details?.reasoning_tokens != null) {
+      span.setAttribute("neatlogs.llm.token_count.reasoning", usage.completion_tokens_details.reasoning_tokens);
+    }
+  }
+  span.setStatus({ code: import_api.SpanStatusCode.OK });
+  span.end();
+}
+function finalizeChatResponse(span, response) {
+  const choices = response?.choices ?? [];
+  for (let i = 0; i < choices.length; i++) {
+    const message = choices[i]?.message;
+    if (!message) continue;
+    span.setAttribute(`neatlogs.llm.output_messages.${i}.role`, "assistant");
+    if (message.content) {
+      span.setAttribute(`neatlogs.llm.output_messages.${i}.content`, message.content);
+    }
+    if (message.tool_calls) {
+      for (let j = 0; j < message.tool_calls.length; j++) {
+        const tc = message.tool_calls[j];
+        span.setAttribute(`neatlogs.llm.tool_calls.${j}.id`, tc.id ?? "");
+        span.setAttribute(`neatlogs.llm.tool_calls.${j}.name`, tc.function?.name ?? "");
+        span.setAttribute(`neatlogs.llm.tool_calls.${j}.arguments`, tc.function?.arguments ?? "");
+      }
+    }
+    if (choices[i].finish_reason) {
+      span.setAttribute("neatlogs.llm.finish_reason", choices[i].finish_reason);
+    }
+  }
+  const usage = response?.usage;
+  if (usage) {
+    if (usage.prompt_tokens != null) span.setAttribute("neatlogs.llm.token_count.prompt", usage.prompt_tokens);
+    if (usage.completion_tokens != null) span.setAttribute("neatlogs.llm.token_count.completion", usage.completion_tokens);
+    if (usage.total_tokens != null) span.setAttribute("neatlogs.llm.token_count.total", usage.total_tokens);
+    if (usage.prompt_tokens_details?.cached_tokens != null) {
+      span.setAttribute("neatlogs.llm.token_count.cache_read", usage.prompt_tokens_details.cached_tokens);
+    }
+    if (usage.completion_tokens_details?.reasoning_tokens != null) {
+      span.setAttribute("neatlogs.llm.token_count.reasoning", usage.completion_tokens_details.reasoning_tokens);
+    }
+  }
+  if (response?.model) span.setAttribute("neatlogs.llm.model_name", response.model);
+  span.setStatus({ code: import_api.SpanStatusCode.OK });
+  span.end();
+}
+function setInvocationParams(span, opts) {
+  if (opts?.temperature != null) span.setAttribute("neatlogs.llm.temperature", opts.temperature);
+  if (opts?.top_p != null) span.setAttribute("neatlogs.llm.top_p", opts.top_p);
+  if (opts?.max_tokens != null) span.setAttribute("neatlogs.llm.max_tokens", opts.max_tokens);
+  if (opts?.frequency_penalty != null) span.setAttribute("neatlogs.llm.frequency_penalty", opts.frequency_penalty);
+  if (opts?.presence_penalty != null) span.setAttribute("neatlogs.llm.presence_penalty", opts.presence_penalty);
+  if (opts?.stop) span.setAttribute("neatlogs.llm.stop_sequences", safeStringify(opts.stop));
+}
+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: import_api.SpanStatusCode.ERROR, message: err.message });
+    span.recordException(err);
+  } else {
+    span.setStatus({ code: import_api.SpanStatusCode.ERROR, message: String(err) });
+  }
+  span.end();
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  traceTool,
+  wrapAzureOpenAI
+});
+//# sourceMappingURL=azure-openai.cjs.map

package/dist/azure-openai.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/azure-openai.ts"],"sourcesContent":["/**\n * Neatlogs Azure OpenAI wrapper — ES6 Proxy-based.\n *\n * Azure OpenAI is accessed through the `openai` SDK's `AzureOpenAI` client,\n * which exposes the same `chat.completions.create()` shape as `OpenAI`. This\n * wrapper traces those calls with provider `azure` so Azure traffic is\n * distinguished from direct OpenAI traffic.\n *\n * Usage:\n *   import { wrapAzureOpenAI } from 'neatlogs/azure-openai';\n *   import { AzureOpenAI } from 'openai';\n *   const client = wrapAzureOpenAI(new AzureOpenAI({ endpoint, apiKey, apiVersion }));\n *   await client.chat.completions.create({ model: 'gpt-4o', messages: [...] });\n *\n * Intercepts:\n *   - client.chat.completions.create() — non-streaming and streaming\n *   - client.responses.create()        — Responses API\n */\n\nimport { trace, context as otelContext, SpanStatusCode, type Span } from '@opentelemetry/api';\n\nconst TRACER_NAME = 'neatlogs.azure_openai';\nconst PROVIDER = 'azure';\nconst SYSTEM = 'azure';\n\nexport function wrapAzureOpenAI<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\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 === 'chat.completions.create' && typeof value === 'function') {\n        return tracedChatCompletionsCreate(value.bind(obj));\n      }\n      if (pathStr === 'responses.create' && typeof value === 'function') {\n        return tracedResponsesCreate(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 > 3) return false;\n  const key = path[path.length - 1];\n  return ['chat', 'completions', 'responses', 'beta'].includes(key);\n}\n\n// ---------------------------------------------------------------------------\n// chat.completions.create — non-streaming + streaming\n// ---------------------------------------------------------------------------\n\nfunction tracedChatCompletionsCreate(original: (...args: any[]) => any) {\n  return function (opts: any, ...rest: any[]): any {\n    const tracer = trace.getTracer(TRACER_NAME);\n    const model = opts?.model ?? '';\n    const messages: any[] = opts?.messages ?? [];\n    const isStream = opts?.stream === true;\n\n    const span = tracer.startSpan('azure_openai.chat.completions.create', {\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    for (let i = 0; i < messages.length; i++) {\n      const msg = messages[i];\n      span.setAttribute(`neatlogs.llm.input_messages.${i}.role`, msg.role ?? '');\n      if (typeof msg.content === 'string') {\n        span.setAttribute(`neatlogs.llm.input_messages.${i}.content`, msg.content);\n      } else if (msg.content) {\n        span.setAttribute(`neatlogs.llm.input_messages.${i}.content`, safeStringify(msg.content));\n      }\n      if (msg.tool_call_id) {\n        span.setAttribute(`neatlogs.llm.input_messages.${i}.tool_call_id`, msg.tool_call_id);\n      }\n    }\n\n    if (opts?.tools) {\n      for (let i = 0; i < opts.tools.length; i++) {\n        const fn = opts.tools[i]?.function ?? {};\n        span.setAttribute(`neatlogs.llm.tools.${i}.name`, fn.name ?? '');\n        if (fn.description) span.setAttribute(`neatlogs.llm.tools.${i}.description`, fn.description);\n        if (fn.parameters) span.setAttribute(`neatlogs.llm.tools.${i}.input_schema`, safeStringify(fn.parameters));\n      }\n    }\n\n    setInvocationParams(span, opts);\n\n    if (isStream) {\n      opts = { ...opts };\n      const streamOpts = opts.stream_options ?? {};\n      if (!streamOpts.include_usage) {\n        opts.stream_options = { ...streamOpts, include_usage: true };\n      }\n    }\n\n    const ctx = trace.setSpan(otelContext.active(), span);\n    const promise = otelContext.with(ctx, () => original(opts, ...rest));\n\n    return promise.then(\n      (response: any) => {\n        if (isStream) {\n          return wrapAsyncIterableStream(response, span);\n        }\n        finalizeChatResponse(span, response);\n        return response;\n      },\n      (err: any) => {\n        recordError(span, err);\n        throw err;\n      },\n    );\n  };\n}\n\n// ---------------------------------------------------------------------------\n// responses.create\n// ---------------------------------------------------------------------------\n\nfunction tracedResponsesCreate(original: (...args: any[]) => any) {\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('azure_openai.responses.create', {\n      attributes: {\n        'neatlogs.span.kind': 'LLM',\n        'neatlogs.llm.provider': PROVIDER,\n        'neatlogs.llm.system': SYSTEM,\n        'neatlogs.llm.model_name': model,\n        'input.value': safeStringify(opts?.input ?? ''),\n      },\n    }, otelContext.active());\n\n    const ctx = trace.setSpan(otelContext.active(), span);\n    const promise = otelContext.with(ctx, () => original(opts, ...rest));\n\n    return promise.then(\n      (response: any) => {\n        if (response?.output_text) {\n          span.setAttribute('neatlogs.llm.output_messages.0.role', 'assistant');\n          span.setAttribute('neatlogs.llm.output_messages.0.content', response.output_text);\n        }\n        if (response?.model) span.setAttribute('neatlogs.llm.model_name', response.model);\n        if (response?.usage) {\n          if (response.usage.input_tokens != null) span.setAttribute('neatlogs.llm.token_count.prompt', response.usage.input_tokens);\n          if (response.usage.output_tokens != null) span.setAttribute('neatlogs.llm.token_count.completion', response.usage.output_tokens);\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\n// ---------------------------------------------------------------------------\n// Streaming support\n// ---------------------------------------------------------------------------\n\nfunction wrapAsyncIterableStream(stream: any, span: Span): any {\n  const chunks: any[] = [];\n  const originalAsyncIterator = stream?.[Symbol.asyncIterator]?.bind(stream);\n\n  if (!originalAsyncIterator) {\n    span.setStatus({ code: SpanStatusCode.OK });\n    span.end();\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  const toolCallsAcc: Record<number, { id: string; name: string; arguments: string }> = {};\n  let finishReason = '';\n  let model = '';\n  let usage: any = null;\n\n  for (const chunk of chunks) {\n    if (!chunk?.choices?.length) {\n      if (chunk?.usage) usage = chunk.usage;\n      continue;\n    }\n    const choice = chunk.choices[0];\n    const delta = choice?.delta;\n    if (delta?.content) textParts.push(delta.content);\n    if (delta?.tool_calls) {\n      for (const tc of delta.tool_calls) {\n        const idx = tc.index ?? 0;\n        if (!toolCallsAcc[idx]) toolCallsAcc[idx] = { id: '', name: '', arguments: '' };\n        if (tc.id) toolCallsAcc[idx].id = tc.id;\n        if (tc.function?.name) toolCallsAcc[idx].name = tc.function.name;\n        if (tc.function?.arguments) toolCallsAcc[idx].arguments += tc.function.arguments;\n      }\n    }\n    if (choice?.finish_reason) finishReason = choice.finish_reason;\n    if (chunk?.model) model = chunk.model;\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\n  let j = 0;\n  for (const tc of Object.values(toolCallsAcc)) {\n    span.setAttribute(`neatlogs.llm.tool_calls.${j}.id`, tc.id);\n    span.setAttribute(`neatlogs.llm.tool_calls.${j}.name`, tc.name);\n    span.setAttribute(`neatlogs.llm.tool_calls.${j}.arguments`, tc.arguments);\n    j++;\n  }\n\n  if (model) span.setAttribute('neatlogs.llm.model_name', model);\n  if (finishReason) span.setAttribute('neatlogs.llm.finish_reason', finishReason);\n\n  if (usage) {\n    if (usage.prompt_tokens != null) span.setAttribute('neatlogs.llm.token_count.prompt', usage.prompt_tokens);\n    if (usage.completion_tokens != null) span.setAttribute('neatlogs.llm.token_count.completion', usage.completion_tokens);\n    if (usage.total_tokens != null) span.setAttribute('neatlogs.llm.token_count.total', usage.total_tokens);\n    if (usage.prompt_tokens_details?.cached_tokens != null) {\n      span.setAttribute('neatlogs.llm.token_count.cache_read', usage.prompt_tokens_details.cached_tokens);\n    }\n    if (usage.completion_tokens_details?.reasoning_tokens != null) {\n      span.setAttribute('neatlogs.llm.token_count.reasoning', usage.completion_tokens_details.reasoning_tokens);\n    }\n  }\n\n  span.setStatus({ code: SpanStatusCode.OK });\n  span.end();\n}\n\n// ---------------------------------------------------------------------------\n// Non-streaming response finalization\n// ---------------------------------------------------------------------------\n\nfunction finalizeChatResponse(span: Span, response: any): void {\n  const choices = response?.choices ?? [];\n  for (let i = 0; i < choices.length; i++) {\n    const message = choices[i]?.message;\n    if (!message) continue;\n\n    span.setAttribute(`neatlogs.llm.output_messages.${i}.role`, 'assistant');\n    if (message.content) {\n      span.setAttribute(`neatlogs.llm.output_messages.${i}.content`, message.content);\n    }\n    if (message.tool_calls) {\n      for (let j = 0; j < message.tool_calls.length; j++) {\n        const tc = message.tool_calls[j];\n        span.setAttribute(`neatlogs.llm.tool_calls.${j}.id`, tc.id ?? '');\n        span.setAttribute(`neatlogs.llm.tool_calls.${j}.name`, tc.function?.name ?? '');\n        span.setAttribute(`neatlogs.llm.tool_calls.${j}.arguments`, tc.function?.arguments ?? '');\n      }\n    }\n    if (choices[i].finish_reason) {\n      span.setAttribute('neatlogs.llm.finish_reason', choices[i].finish_reason);\n    }\n  }\n\n  const usage = response?.usage;\n  if (usage) {\n    if (usage.prompt_tokens != null) span.setAttribute('neatlogs.llm.token_count.prompt', usage.prompt_tokens);\n    if (usage.completion_tokens != null) span.setAttribute('neatlogs.llm.token_count.completion', usage.completion_tokens);\n    if (usage.total_tokens != null) span.setAttribute('neatlogs.llm.token_count.total', usage.total_tokens);\n    if (usage.prompt_tokens_details?.cached_tokens != null) {\n      span.setAttribute('neatlogs.llm.token_count.cache_read', usage.prompt_tokens_details.cached_tokens);\n    }\n    if (usage.completion_tokens_details?.reasoning_tokens != null) {\n      span.setAttribute('neatlogs.llm.token_count.reasoning', usage.completion_tokens_details.reasoning_tokens);\n    }\n  }\n\n  if (response?.model) span.setAttribute('neatlogs.llm.model_name', response.model);\n\n  span.setStatus({ code: SpanStatusCode.OK });\n  span.end();\n}\n\n// ---------------------------------------------------------------------------\n// Utilities\n// ---------------------------------------------------------------------------\n\nfunction setInvocationParams(span: Span, opts: any): void {\n  if (opts?.temperature != null) span.setAttribute('neatlogs.llm.temperature', opts.temperature);\n  if (opts?.top_p != null) span.setAttribute('neatlogs.llm.top_p', opts.top_p);\n  if (opts?.max_tokens != null) span.setAttribute('neatlogs.llm.max_tokens', opts.max_tokens);\n  if (opts?.frequency_penalty != null) span.setAttribute('neatlogs.llm.frequency_penalty', opts.frequency_penalty);\n  if (opts?.presence_penalty != null) span.setAttribute('neatlogs.llm.presence_penalty', opts.presence_penalty);\n  if (opts?.stop) span.setAttribute('neatlogs.llm.stop_sequences', safeStringify(opts.stop));\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":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAmBA,iBAAyE;AAEzE,IAAM,cAAc;AACpB,IAAM,WAAW;AACjB,IAAM,SAAS;AAER,SAAS,gBAAkC,QAAc;AAC9D,SAAO,cAAc,QAAQ,CAAC,CAAC;AACjC;AAUO,SAAS,UACd,MACA,IACmC;AACnC,SAAO,eAAe,WAAW,MAA+B;AAC9D,UAAM,SAAS,iBAAM,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,WAAAA,QAAY,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,0BAAe,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;AAEA,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,6BAA6B,OAAO,UAAU,YAAY;AACxE,eAAO,4BAA4B,MAAM,KAAK,GAAG,CAAC;AAAA,MACpD;AACA,UAAI,YAAY,sBAAsB,OAAO,UAAU,YAAY;AACjE,eAAO,sBAAsB,MAAM,KAAK,GAAG,CAAC;AAAA,MAC9C;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,QAAM,MAAM,KAAK,KAAK,SAAS,CAAC;AAChC,SAAO,CAAC,QAAQ,eAAe,aAAa,MAAM,EAAE,SAAS,GAAG;AAClE;AAMA,SAAS,4BAA4B,UAAmC;AACtE,SAAO,SAAU,SAAc,MAAkB;AAC/C,UAAM,SAAS,iBAAM,UAAU,WAAW;AAC1C,UAAM,QAAQ,MAAM,SAAS;AAC7B,UAAM,WAAkB,MAAM,YAAY,CAAC;AAC3C,UAAM,WAAW,MAAM,WAAW;AAElC,UAAM,OAAO,OAAO,UAAU,wCAAwC;AAAA,MACpE,YAAY;AAAA,QACV,sBAAsB;AAAA,QACtB,yBAAyB;AAAA,QACzB,uBAAuB;AAAA,QACvB,2BAA2B;AAAA,QAC3B,6BAA6B;AAAA,MAC/B;AAAA,IACF,GAAG,WAAAA,QAAY,OAAO,CAAC;AAEvB,aAAS,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;AACxC,YAAM,MAAM,SAAS,CAAC;AACtB,WAAK,aAAa,+BAA+B,CAAC,SAAS,IAAI,QAAQ,EAAE;AACzE,UAAI,OAAO,IAAI,YAAY,UAAU;AACnC,aAAK,aAAa,+BAA+B,CAAC,YAAY,IAAI,OAAO;AAAA,MAC3E,WAAW,IAAI,SAAS;AACtB,aAAK,aAAa,+BAA+B,CAAC,YAAY,cAAc,IAAI,OAAO,CAAC;AAAA,MAC1F;AACA,UAAI,IAAI,cAAc;AACpB,aAAK,aAAa,+BAA+B,CAAC,iBAAiB,IAAI,YAAY;AAAA,MACrF;AAAA,IACF;AAEA,QAAI,MAAM,OAAO;AACf,eAAS,IAAI,GAAG,IAAI,KAAK,MAAM,QAAQ,KAAK;AAC1C,cAAM,KAAK,KAAK,MAAM,CAAC,GAAG,YAAY,CAAC;AACvC,aAAK,aAAa,sBAAsB,CAAC,SAAS,GAAG,QAAQ,EAAE;AAC/D,YAAI,GAAG,YAAa,MAAK,aAAa,sBAAsB,CAAC,gBAAgB,GAAG,WAAW;AAC3F,YAAI,GAAG,WAAY,MAAK,aAAa,sBAAsB,CAAC,iBAAiB,cAAc,GAAG,UAAU,CAAC;AAAA,MAC3G;AAAA,IACF;AAEA,wBAAoB,MAAM,IAAI;AAE9B,QAAI,UAAU;AACZ,aAAO,EAAE,GAAG,KAAK;AACjB,YAAM,aAAa,KAAK,kBAAkB,CAAC;AAC3C,UAAI,CAAC,WAAW,eAAe;AAC7B,aAAK,iBAAiB,EAAE,GAAG,YAAY,eAAe,KAAK;AAAA,MAC7D;AAAA,IACF;AAEA,UAAM,MAAM,iBAAM,QAAQ,WAAAA,QAAY,OAAO,GAAG,IAAI;AACpD,UAAM,UAAU,WAAAA,QAAY,KAAK,KAAK,MAAM,SAAS,MAAM,GAAG,IAAI,CAAC;AAEnE,WAAO,QAAQ;AAAA,MACb,CAAC,aAAkB;AACjB,YAAI,UAAU;AACZ,iBAAO,wBAAwB,UAAU,IAAI;AAAA,QAC/C;AACA,6BAAqB,MAAM,QAAQ;AACnC,eAAO;AAAA,MACT;AAAA,MACA,CAAC,QAAa;AACZ,oBAAY,MAAM,GAAG;AACrB,cAAM;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACF;AAMA,SAAS,sBAAsB,UAAmC;AAChE,SAAO,SAAU,SAAc,MAAkB;AAC/C,UAAM,SAAS,iBAAM,UAAU,WAAW;AAC1C,UAAM,QAAQ,MAAM,SAAS;AAE7B,UAAM,OAAO,OAAO,UAAU,iCAAiC;AAAA,MAC7D,YAAY;AAAA,QACV,sBAAsB;AAAA,QACtB,yBAAyB;AAAA,QACzB,uBAAuB;AAAA,QACvB,2BAA2B;AAAA,QAC3B,eAAe,cAAc,MAAM,SAAS,EAAE;AAAA,MAChD;AAAA,IACF,GAAG,WAAAA,QAAY,OAAO,CAAC;AAEvB,UAAM,MAAM,iBAAM,QAAQ,WAAAA,QAAY,OAAO,GAAG,IAAI;AACpD,UAAM,UAAU,WAAAA,QAAY,KAAK,KAAK,MAAM,SAAS,MAAM,GAAG,IAAI,CAAC;AAEnE,WAAO,QAAQ;AAAA,MACb,CAAC,aAAkB;AACjB,YAAI,UAAU,aAAa;AACzB,eAAK,aAAa,uCAAuC,WAAW;AACpE,eAAK,aAAa,0CAA0C,SAAS,WAAW;AAAA,QAClF;AACA,YAAI,UAAU,MAAO,MAAK,aAAa,2BAA2B,SAAS,KAAK;AAChF,YAAI,UAAU,OAAO;AACnB,cAAI,SAAS,MAAM,gBAAgB,KAAM,MAAK,aAAa,mCAAmC,SAAS,MAAM,YAAY;AACzH,cAAI,SAAS,MAAM,iBAAiB,KAAM,MAAK,aAAa,uCAAuC,SAAS,MAAM,aAAa;AAAA,QACjI;AACA,aAAK,UAAU,EAAE,MAAM,0BAAe,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,wBAAwB,QAAa,MAAiB;AAC7D,QAAM,SAAgB,CAAC;AACvB,QAAM,wBAAwB,SAAS,OAAO,aAAa,GAAG,KAAK,MAAM;AAEzE,MAAI,CAAC,uBAAuB;AAC1B,SAAK,UAAU,EAAE,MAAM,0BAAe,GAAG,CAAC;AAC1C,SAAK,IAAI;AACT,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,QAAM,eAAgF,CAAC;AACvF,MAAI,eAAe;AACnB,MAAI,QAAQ;AACZ,MAAI,QAAa;AAEjB,aAAW,SAAS,QAAQ;AAC1B,QAAI,CAAC,OAAO,SAAS,QAAQ;AAC3B,UAAI,OAAO,MAAO,SAAQ,MAAM;AAChC;AAAA,IACF;AACA,UAAM,SAAS,MAAM,QAAQ,CAAC;AAC9B,UAAM,QAAQ,QAAQ;AACtB,QAAI,OAAO,QAAS,WAAU,KAAK,MAAM,OAAO;AAChD,QAAI,OAAO,YAAY;AACrB,iBAAW,MAAM,MAAM,YAAY;AACjC,cAAM,MAAM,GAAG,SAAS;AACxB,YAAI,CAAC,aAAa,GAAG,EAAG,cAAa,GAAG,IAAI,EAAE,IAAI,IAAI,MAAM,IAAI,WAAW,GAAG;AAC9E,YAAI,GAAG,GAAI,cAAa,GAAG,EAAE,KAAK,GAAG;AACrC,YAAI,GAAG,UAAU,KAAM,cAAa,GAAG,EAAE,OAAO,GAAG,SAAS;AAC5D,YAAI,GAAG,UAAU,UAAW,cAAa,GAAG,EAAE,aAAa,GAAG,SAAS;AAAA,MACzE;AAAA,IACF;AACA,QAAI,QAAQ,cAAe,gBAAe,OAAO;AACjD,QAAI,OAAO,MAAO,SAAQ,MAAM;AAAA,EAClC;AAEA,QAAM,WAAW,UAAU,KAAK,EAAE;AAClC,MAAI,UAAU;AACZ,SAAK,aAAa,uCAAuC,WAAW;AACpE,SAAK,aAAa,0CAA0C,QAAQ;AAAA,EACtE;AAEA,MAAI,IAAI;AACR,aAAW,MAAM,OAAO,OAAO,YAAY,GAAG;AAC5C,SAAK,aAAa,2BAA2B,CAAC,OAAO,GAAG,EAAE;AAC1D,SAAK,aAAa,2BAA2B,CAAC,SAAS,GAAG,IAAI;AAC9D,SAAK,aAAa,2BAA2B,CAAC,cAAc,GAAG,SAAS;AACxE;AAAA,EACF;AAEA,MAAI,MAAO,MAAK,aAAa,2BAA2B,KAAK;AAC7D,MAAI,aAAc,MAAK,aAAa,8BAA8B,YAAY;AAE9E,MAAI,OAAO;AACT,QAAI,MAAM,iBAAiB,KAAM,MAAK,aAAa,mCAAmC,MAAM,aAAa;AACzG,QAAI,MAAM,qBAAqB,KAAM,MAAK,aAAa,uCAAuC,MAAM,iBAAiB;AACrH,QAAI,MAAM,gBAAgB,KAAM,MAAK,aAAa,kCAAkC,MAAM,YAAY;AACtG,QAAI,MAAM,uBAAuB,iBAAiB,MAAM;AACtD,WAAK,aAAa,uCAAuC,MAAM,sBAAsB,aAAa;AAAA,IACpG;AACA,QAAI,MAAM,2BAA2B,oBAAoB,MAAM;AAC7D,WAAK,aAAa,sCAAsC,MAAM,0BAA0B,gBAAgB;AAAA,IAC1G;AAAA,EACF;AAEA,OAAK,UAAU,EAAE,MAAM,0BAAe,GAAG,CAAC;AAC1C,OAAK,IAAI;AACX;AAMA,SAAS,qBAAqB,MAAY,UAAqB;AAC7D,QAAM,UAAU,UAAU,WAAW,CAAC;AACtC,WAAS,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK;AACvC,UAAM,UAAU,QAAQ,CAAC,GAAG;AAC5B,QAAI,CAAC,QAAS;AAEd,SAAK,aAAa,gCAAgC,CAAC,SAAS,WAAW;AACvE,QAAI,QAAQ,SAAS;AACnB,WAAK,aAAa,gCAAgC,CAAC,YAAY,QAAQ,OAAO;AAAA,IAChF;AACA,QAAI,QAAQ,YAAY;AACtB,eAAS,IAAI,GAAG,IAAI,QAAQ,WAAW,QAAQ,KAAK;AAClD,cAAM,KAAK,QAAQ,WAAW,CAAC;AAC/B,aAAK,aAAa,2BAA2B,CAAC,OAAO,GAAG,MAAM,EAAE;AAChE,aAAK,aAAa,2BAA2B,CAAC,SAAS,GAAG,UAAU,QAAQ,EAAE;AAC9E,aAAK,aAAa,2BAA2B,CAAC,cAAc,GAAG,UAAU,aAAa,EAAE;AAAA,MAC1F;AAAA,IACF;AACA,QAAI,QAAQ,CAAC,EAAE,eAAe;AAC5B,WAAK,aAAa,8BAA8B,QAAQ,CAAC,EAAE,aAAa;AAAA,IAC1E;AAAA,EACF;AAEA,QAAM,QAAQ,UAAU;AACxB,MAAI,OAAO;AACT,QAAI,MAAM,iBAAiB,KAAM,MAAK,aAAa,mCAAmC,MAAM,aAAa;AACzG,QAAI,MAAM,qBAAqB,KAAM,MAAK,aAAa,uCAAuC,MAAM,iBAAiB;AACrH,QAAI,MAAM,gBAAgB,KAAM,MAAK,aAAa,kCAAkC,MAAM,YAAY;AACtG,QAAI,MAAM,uBAAuB,iBAAiB,MAAM;AACtD,WAAK,aAAa,uCAAuC,MAAM,sBAAsB,aAAa;AAAA,IACpG;AACA,QAAI,MAAM,2BAA2B,oBAAoB,MAAM;AAC7D,WAAK,aAAa,sCAAsC,MAAM,0BAA0B,gBAAgB;AAAA,IAC1G;AAAA,EACF;AAEA,MAAI,UAAU,MAAO,MAAK,aAAa,2BAA2B,SAAS,KAAK;AAEhF,OAAK,UAAU,EAAE,MAAM,0BAAe,GAAG,CAAC;AAC1C,OAAK,IAAI;AACX;AAMA,SAAS,oBAAoB,MAAY,MAAiB;AACxD,MAAI,MAAM,eAAe,KAAM,MAAK,aAAa,4BAA4B,KAAK,WAAW;AAC7F,MAAI,MAAM,SAAS,KAAM,MAAK,aAAa,sBAAsB,KAAK,KAAK;AAC3E,MAAI,MAAM,cAAc,KAAM,MAAK,aAAa,2BAA2B,KAAK,UAAU;AAC1F,MAAI,MAAM,qBAAqB,KAAM,MAAK,aAAa,kCAAkC,KAAK,iBAAiB;AAC/G,MAAI,MAAM,oBAAoB,KAAM,MAAK,aAAa,iCAAiC,KAAK,gBAAgB;AAC5G,MAAI,MAAM,KAAM,MAAK,aAAa,+BAA+B,cAAc,KAAK,IAAI,CAAC;AAC3F;AAEA,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,0BAAe,OAAO,SAAS,IAAI,QAAQ,CAAC;AACnE,SAAK,gBAAgB,GAAG;AAAA,EAC1B,OAAO;AACL,SAAK,UAAU,EAAE,MAAM,0BAAe,OAAO,SAAS,OAAO,GAAG,EAAE,CAAC;AAAA,EACrE;AACA,OAAK,IAAI;AACX;","names":["otelContext"]}

package/dist/azure-openai.d.ts

@@ -0,0 +1,30 @@
+/**
+ * Neatlogs Azure OpenAI wrapper — ES6 Proxy-based.
+ *
+ * Azure OpenAI is accessed through the `openai` SDK's `AzureOpenAI` client,
+ * which exposes the same `chat.completions.create()` shape as `OpenAI`. This
+ * wrapper traces those calls with provider `azure` so Azure traffic is
+ * distinguished from direct OpenAI traffic.
+ *
+ * Usage:
+ *   import { wrapAzureOpenAI } from 'neatlogs/azure-openai';
+ *   import { AzureOpenAI } from 'openai';
+ *   const client = wrapAzureOpenAI(new AzureOpenAI({ endpoint, apiKey, apiVersion }));
+ *   await client.chat.completions.create({ model: 'gpt-4o', messages: [...] });
+ *
+ * Intercepts:
+ *   - client.chat.completions.create() — non-streaming and streaming
+ *   - client.responses.create()        — Responses API
+ */
+declare function wrapAzureOpenAI<T extends object>(client: T): T;
+/**
+ * Wrap a tool/function implementation to emit TOOL spans when executed.
+ *
+ * Usage:
+ *   const getWeather = traceTool('get_weather', async (args: { city: string }) => {
+ *     return `Weather in ${args.city}: sunny`;
+ *   });
+ */
+declare function traceTool<TArgs = any, TResult = any>(name: string, fn: (args: TArgs) => TResult | Promise<TResult>): (args: TArgs) => Promise<TResult>;
+
+export { traceTool, wrapAzureOpenAI };