neatlogs 1.0.4 → 1.0.6

package/dist/openai.mjs

@@ -0,0 +1,316 @@
+// src/openai.ts
+import { trace, context as otelContext, SpanStatusCode } from "@opentelemetry/api";
+var TRACER_NAME = "neatlogs.openai";
+function wrapOpenAI(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 === "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 = trace.getTracer(TRACER_NAME);
+    const model = opts?.model ?? "";
+    const messages = opts?.messages ?? [];
+    const isStream = opts?.stream === true;
+    const span = tracer.startSpan("openai.chat.completions.create", {
+      attributes: {
+        "neatlogs.span.kind": "LLM",
+        "neatlogs.llm.provider": "openai",
+        "neatlogs.llm.system": "openai",
+        "neatlogs.llm.model_name": model,
+        "neatlogs.llm.is_streaming": isStream
+      }
+    }, otelContext.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 = trace.setSpan(otelContext.active(), span);
+    const promise = otelContext.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 = trace.getTracer(TRACER_NAME);
+    const model = opts?.model ?? "";
+    const span = tracer.startSpan("openai.responses.create", {
+      attributes: {
+        "neatlogs.span.kind": "LLM",
+        "neatlogs.llm.provider": "openai",
+        "neatlogs.llm.system": "openai",
+        "neatlogs.llm.model_name": model,
+        "input.value": safeStringify(opts?.input ?? "")
+      }
+    }, otelContext.active());
+    const ctx = trace.setSpan(otelContext.active(), span);
+    const promise = otelContext.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: 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: 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: 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: 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: SpanStatusCode.ERROR, message: err.message });
+    span.recordException(err);
+  } else {
+    span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });
+  }
+  span.end();
+}
+export {
+  traceTool,
+  wrapOpenAI
+};
+//# sourceMappingURL=openai.mjs.map

package/dist/openai.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/openai.ts"],"sourcesContent":["/**\n * Neatlogs OpenAI wrapper — ES6 Proxy-based.\n *\n * Usage:\n *   import { wrapOpenAI, traceTool } from 'neatlogs/openai';\n *   import OpenAI from 'openai';\n *   const client = wrapOpenAI(new OpenAI());\n *\n * Intercepts:\n *   - client.chat.completions.create() — non-streaming and streaming\n *   - client.responses.create() — Responses API\n *\n * Also exports traceTool() to wrap user-defined tool functions with TOOL spans.\n */\n\nimport { trace, context as otelContext, SpanStatusCode, type Span } from '@opentelemetry/api';\n\nconst TRACER_NAME = 'neatlogs.openai';\n\n// ---------------------------------------------------------------------------\n// Public API\n// ---------------------------------------------------------------------------\n\nexport function wrapOpenAI<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 === '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('openai.chat.completions.create', {\n      attributes: {\n        'neatlogs.span.kind': 'LLM',\n        'neatlogs.llm.provider': 'openai',\n        'neatlogs.llm.system': 'openai',\n        'neatlogs.llm.model_name': model,\n        'neatlogs.llm.is_streaming': isStream,\n      },\n    }, otelContext.active());\n\n    // Input messages\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    // Tools\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    // Invocation parameters\n    setInvocationParams(span, opts);\n\n    // Force stream_options.include_usage for streaming\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('openai.responses.create', {\n      attributes: {\n        'neatlogs.span.kind': 'LLM',\n        'neatlogs.llm.provider': 'openai',\n        'neatlogs.llm.system': 'openai',\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":";AAeA,SAAS,OAAO,WAAW,aAAa,sBAAiC;AAEzE,IAAM,cAAc;AAMb,SAAS,WAA6B,QAAc;AACzD,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,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,MAAM,UAAU,WAAW;AAC1C,UAAM,QAAQ,MAAM,SAAS;AAC7B,UAAM,WAAkB,MAAM,YAAY,CAAC;AAC3C,UAAM,WAAW,MAAM,WAAW;AAElC,UAAM,OAAO,OAAO,UAAU,kCAAkC;AAAA,MAC9D,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;AAGvB,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;AAGA,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;AAGA,wBAAoB,MAAM,IAAI;AAG9B,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,MAAM,QAAQ,YAAY,OAAO,GAAG,IAAI;AACpD,UAAM,UAAU,YAAY,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,MAAM,UAAU,WAAW;AAC1C,UAAM,QAAQ,MAAM,SAAS;AAE7B,UAAM,OAAO,OAAO,UAAU,2BAA2B;AAAA,MACvD,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,YAAY,OAAO,CAAC;AAEvB,UAAM,MAAM,MAAM,QAAQ,YAAY,OAAO,GAAG,IAAI;AACpD,UAAM,UAAU,YAAY,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,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,wBAAwB,QAAa,MAAiB;AAC7D,QAAM,SAAgB,CAAC;AACvB,QAAM,wBAAwB,OAAO,OAAO,aAAa,GAAG,KAAK,MAAM;AAEvE,MAAI,CAAC,uBAAuB;AAC1B,SAAK,UAAU,EAAE,MAAM,eAAe,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,eAAe,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,eAAe,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,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/dist/pi-agent.cjs

@@ -0,0 +1,237 @@
+"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/pi-agent.ts
+var pi_agent_exports = {};
+__export(pi_agent_exports, {
+  piAgentHooks: () => piAgentHooks
+});
+module.exports = __toCommonJS(pi_agent_exports);
+var import_api = require("@opentelemetry/api");
+var TRACER_NAME = "neatlogs.pi-agent";
+var PATCH_FLAG = "_neatlogs_patched";
+function piAgentHooks(agent) {
+  if (!agent || agent[PATCH_FLAG]) return agent;
+  const a = agent;
+  if (typeof a.subscribe !== "function") return agent;
+  const state = { toolSpans: /* @__PURE__ */ new Map(), inputMessages: [] };
+  const tracer = import_api.trace.getTracer(TRACER_NAME);
+  a.subscribe((event) => {
+    try {
+      handleEvent(tracer, state, event);
+    } catch {
+    }
+  });
+  markPatched(a);
+  return agent;
+}
+function handleEvent(tracer, state, event) {
+  switch (event.type) {
+    case "agent_start": {
+      const span = tracer.startSpan(
+        "pi_agent.run",
+        { attributes: { "neatlogs.span.kind": "AGENT" } },
+        import_api.context.active()
+      );
+      state.agentSpan = span;
+      state.agentCtx = import_api.trace.setSpan(import_api.context.active(), span);
+      state.inputMessages = [];
+      break;
+    }
+    case "message_end": {
+      const msg = event.message;
+      if (!msg) return;
+      if (msg.role === "assistant") {
+        emitLlmSpan(tracer, state, msg);
+        const { text } = splitAssistantContent(msg.content);
+        if (text) state.inputMessages.push({ role: "assistant", content: text });
+      } else {
+        const role = msg.role === "toolResult" ? "tool" : String(msg.role || "user");
+        const content = messageText(msg);
+        if (content) state.inputMessages.push({ role, content });
+      }
+      break;
+    }
+    case "tool_execution_start": {
+      const parent = state.agentCtx ?? import_api.context.active();
+      const span = 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": safeStringify(event.args).slice(0, 1e4) } : {}
+          }
+        },
+        parent
+      );
+      if (event.toolCallId) state.toolSpans.set(event.toolCallId, span);
+      break;
+    }
+    case "tool_execution_end": {
+      const span = event.toolCallId ? state.toolSpans.get(event.toolCallId) : void 0;
+      if (!span) return;
+      if (event.result !== void 0) {
+        span.setAttribute("output.value", safeStringify(event.result).slice(0, 1e4));
+      }
+      if (event.isError) {
+        span.setStatus({ code: import_api.SpanStatusCode.ERROR });
+        span.setAttribute("neatlogs.tool.is_error", true);
+      } else {
+        span.setStatus({ code: import_api.SpanStatusCode.OK });
+      }
+      span.end();
+      if (event.toolCallId) state.toolSpans.delete(event.toolCallId);
+      break;
+    }
+    case "agent_end": {
+      for (const ts of state.toolSpans.values()) {
+        try {
+          ts.end();
+        } catch {
+        }
+      }
+      state.toolSpans.clear();
+      if (state.agentSpan) {
+        const firstUser = state.inputMessages.find((m) => m.role === "user");
+        if (firstUser) state.agentSpan.setAttribute("input.value", firstUser.content.slice(0, 1e4));
+        const finalText = lastAssistantText(event.messages);
+        if (finalText) state.agentSpan.setAttribute("output.value", finalText.slice(0, 1e4));
+        state.agentSpan.setStatus({ code: import_api.SpanStatusCode.OK });
+        state.agentSpan.end();
+        state.agentSpan = void 0;
+        state.agentCtx = void 0;
+      }
+      break;
+    }
+    default:
+      break;
+  }
+}
+function emitLlmSpan(tracer, state, msg) {
+  const attrs = { "neatlogs.span.kind": "LLM" };
+  if (msg.model) attrs["neatlogs.llm.model_name"] = String(msg.model);
+  if (msg.provider) attrs["neatlogs.llm.provider"] = String(msg.provider);
+  if (msg.stopReason) attrs["neatlogs.llm.stop_reason"] = String(msg.stopReason);
+  const inMsgs = state.inputMessages;
+  if (inMsgs.length) {
+    inMsgs.forEach((m, i) => {
+      attrs[`neatlogs.llm.input_messages.${i}.role`] = m.role;
+      attrs[`neatlogs.llm.input_messages.${i}.content`] = m.content.slice(0, 1e4);
+    });
+    attrs["neatlogs.llm.input"] = safeStringify({ messages: inMsgs }).slice(0, 2e4);
+    attrs["input.value"] = safeStringify({ messages: inMsgs }).slice(0, 1e4);
+  }
+  const { text, toolCalls } = splitAssistantContent(msg.content);
+  const outText = text || toolCalls.map((tc) => `${tc.name}(${safeStringify(tc.arguments)})`).join("\n");
+  if (outText || toolCalls.length) {
+    attrs["neatlogs.llm.output_messages.0.role"] = "assistant";
+    attrs["neatlogs.llm.output_messages.0.content"] = (outText || "").slice(0, 1e4);
+    const outBlob = { role: "assistant", content: outText || "" };
+    if (toolCalls.length) {
+      outBlob.tool_calls = toolCalls.map((tc) => ({ name: tc.name, arguments: tc.arguments }));
+      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`] = safeStringify(tc.arguments);
+        if (tc.id) attrs[`neatlogs.llm.tool_calls.${j}.id`] = String(tc.id);
+      });
+    }
+    attrs["neatlogs.llm.output"] = safeStringify(outBlob).slice(0, 2e4);
+    attrs["output.value"] = (outText || "").slice(0, 1e4);
+  }
+  const usage = msg.usage;
+  if (usage) {
+    if (usage.input != null) attrs["neatlogs.llm.token_count.prompt"] = usage.input;
+    if (usage.output != null) attrs["neatlogs.llm.token_count.completion"] = usage.output;
+    const total = usage.totalTokens ?? (usage.input ?? 0) + (usage.output ?? 0);
+    if (total) attrs["neatlogs.llm.token_count.total"] = total;
+    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_api.context.active();
+  const span = tracer.startSpan(
+    `pi_agent.llm.${msg.model || "model"}`,
+    { attributes: attrs },
+    parent
+  );
+  span.setStatus({ code: import_api.SpanStatusCode.OK });
+  span.end();
+}
+function splitAssistantContent(content) {
+  const texts = [];
+  const toolCalls = [];
+  if (Array.isArray(content)) {
+    for (const block of content) {
+      if (!block || typeof block !== "object") continue;
+      if (block.type === "text" && typeof block.text === "string") texts.push(block.text);
+      else if (block.type === "toolCall")
+        toolCalls.push(block);
+    }
+  } else if (typeof content === "string") {
+    texts.push(content);
+  }
+  return { text: texts.join(""), toolCalls };
+}
+function messageText(msg) {
+  if (!msg) return "";
+  const c = msg.content;
+  if (typeof c === "string") return c;
+  if (!Array.isArray(c)) return "";
+  const parts = [];
+  for (const block of c) {
+    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"}(${safeStringify(block.arguments)})`);
+    }
+  }
+  return parts.join("");
+}
+function lastAssistantText(messages) {
+  if (!Array.isArray(messages)) return "";
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const m = messages[i];
+    if (m && m.role === "assistant") {
+      const { text } = splitAssistantContent(m.content);
+      if (text) return text;
+    }
+  }
+  return "";
+}
+function markPatched(e) {
+  try {
+    Object.defineProperty(e, PATCH_FLAG, { value: true, enumerable: false, configurable: true });
+  } catch {
+    e[PATCH_FLAG] = true;
+  }
+}
+function safeStringify(value) {
+  if (typeof value === "string") return value;
+  try {
+    return JSON.stringify(value) ?? "";
+  } catch {
+    return "";
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  piAgentHooks
+});
+//# sourceMappingURL=pi-agent.cjs.map

package/dist/pi-agent.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/pi-agent.ts"],"sourcesContent":["/**\n * Neatlogs Pi Agent integration.\n *\n * Usage:\n *   import { init } from 'neatlogs';\n *   import { piAgentHooks } from 'neatlogs/pi-agent';\n *   import { Agent } from '@mariozechner/pi-agent-core';\n *\n *   await init({ apiKey, workflowName });\n *   const agent = piAgentHooks(new Agent({ initialState: { systemPrompt, model } }));\n *   await agent.prompt('Hello');\n *\n * Pi Agent's `Agent` exposes a first-class `subscribe(listener)` API and emits\n * AgentEvents for its run lifecycle (message_*, tool_execution_*, turn_*,\n * agent_*). It does NOT emit its own OpenTelemetry spans — so we LISTEN to those\n * events (no monkey-patching) and translate them into neatlogs OTel spans:\n *\n *   AGENT  agent run                    (agent_start → agent_end)\n *     ↳ LLM   assistant message         (each assistant message_end)\n *     ↳ TOOL  tool call                 (tool_execution_start → tool_execution_end)\n *\n * The AGENT span is opened as the active span so the LLM/TOOL children nest under\n * it (and under any user @span / trace() block active when prompt() is called).\n */\n\nimport {\n  trace,\n  context as otelContext,\n  SpanStatusCode,\n  type Span,\n  type Context,\n} from '@opentelemetry/api';\n\nconst TRACER_NAME = 'neatlogs.pi-agent';\nconst PATCH_FLAG = '_neatlogs_patched';\n\n// Minimal structural types for the Pi Agent event surface (we duck-type — no\n// hard dependency on the pi-agent-core package).\ninterface PiUsage {\n  input?: number;\n  output?: number;\n  cacheRead?: number;\n  cacheWrite?: number;\n  totalTokens?: number;\n}\ninterface PiToolCall {\n  type: 'toolCall';\n  id?: string;\n  name?: string;\n  arguments?: Record<string, unknown>;\n}\ninterface PiAssistantMessage {\n  role: 'assistant';\n  content?: Array<{ type: string; text?: string; name?: string; arguments?: unknown }>;\n  model?: string;\n  provider?: string;\n  usage?: PiUsage;\n  stopReason?: string;\n}\ninterface PiAgentEvent {\n  type: string;\n  message?: any;\n  messages?: any[];\n  toolCallId?: string;\n  toolName?: string;\n  args?: unknown;\n  result?: unknown;\n  isError?: boolean;\n}\n\ninterface PerAgentState {\n  agentSpan?: Span;\n  agentCtx?: Context;\n  toolSpans: Map<string, Span>;\n  /** Running conversation (system/user/tool turns) to use as LLM-span input.\n   *  Pi Agent's assistant message_end carries only the response, not the prompt. */\n  inputMessages: Array<{ role: string; content: string }>;\n}\n\n/**\n * Subscribe neatlogs tracing to a Pi Agent instance. Returns the same agent\n * (marked so re-subscribing is a no-op). Idempotent per agent.\n */\nexport function piAgentHooks<T extends object>(agent: T): T {\n  if (!agent || (agent as any)[PATCH_FLAG]) return agent;\n  const a = agent as any;\n  if (typeof a.subscribe !== 'function') return agent; // not a Pi Agent — leave alone\n\n  const state: PerAgentState = { toolSpans: new Map(), inputMessages: [] };\n  const tracer = trace.getTracer(TRACER_NAME);\n\n  a.subscribe((event: PiAgentEvent) => {\n    try {\n      handleEvent(tracer, state, event);\n    } catch {\n      // never let tracing break the agent run\n    }\n  });\n\n  markPatched(a);\n  return agent;\n}\n\nfunction handleEvent(\n  tracer: ReturnType<typeof trace.getTracer>,\n  state: PerAgentState,\n  event: PiAgentEvent,\n): void {\n  switch (event.type) {\n    case 'agent_start': {\n      // Open the AGENT (run) span as the active span so children nest under it.\n      const span = tracer.startSpan(\n        'pi_agent.run',\n        { attributes: { 'neatlogs.span.kind': 'AGENT' } },\n        otelContext.active(),\n      );\n      state.agentSpan = span;\n      state.agentCtx = trace.setSpan(otelContext.active(), span);\n      state.inputMessages = [];\n      break;\n    }\n\n    case 'message_end': {\n      const msg = event.message as any;\n      if (!msg) return;\n      if (msg.role === 'assistant') {\n        // Assistant message = the LLM response. Emit an LLM span using the\n        // accumulated conversation as input, then record the assistant turn too.\n        emitLlmSpan(tracer, state, msg as PiAssistantMessage);\n        const { text } = splitAssistantContent(msg.content);\n        if (text) state.inputMessages.push({ role: 'assistant', content: text });\n      } else {\n        // user / toolResult turns — accumulate as input context for later LLM spans.\n        const role = msg.role === 'toolResult' ? 'tool' : String(msg.role || 'user');\n        const content = messageText(msg);\n        if (content) state.inputMessages.push({ role, content });\n      }\n      break;\n    }\n\n    case 'tool_execution_start': {\n      const parent = state.agentCtx ?? otelContext.active();\n      const span = tracer.startSpan(\n        `pi_agent.tool.${event.toolName ?? 'tool'}`,\n        {\n          attributes: {\n            'neatlogs.span.kind': 'TOOL',\n            ...(event.toolName ? { 'neatlogs.tool.name': String(event.toolName) } : {}),\n            ...(event.args !== undefined\n              ? { 'input.value': safeStringify(event.args).slice(0, 10000) }\n              : {}),\n          },\n        },\n        parent,\n      );\n      if (event.toolCallId) state.toolSpans.set(event.toolCallId, span);\n      break;\n    }\n\n    case 'tool_execution_end': {\n      const span = event.toolCallId ? state.toolSpans.get(event.toolCallId) : undefined;\n      if (!span) return;\n      if (event.result !== undefined) {\n        span.setAttribute('output.value', safeStringify(event.result).slice(0, 10000));\n      }\n      if (event.isError) {\n        span.setStatus({ code: SpanStatusCode.ERROR });\n        span.setAttribute('neatlogs.tool.is_error', true);\n      } else {\n        span.setStatus({ code: SpanStatusCode.OK });\n      }\n      span.end();\n      if (event.toolCallId) state.toolSpans.delete(event.toolCallId);\n      break;\n    }\n\n    case 'agent_end': {\n      // Close any tool spans that never received an end event, then the agent span.\n      for (const ts of state.toolSpans.values()) {\n        try {\n          ts.end();\n        } catch {\n          /* ignore */\n        }\n      }\n      state.toolSpans.clear();\n      if (state.agentSpan) {\n        // Agent input = the first user message of the run; output = final answer.\n        const firstUser = state.inputMessages.find((m) => m.role === 'user');\n        if (firstUser) state.agentSpan.setAttribute('input.value', firstUser.content.slice(0, 10000));\n        const finalText = lastAssistantText(event.messages);\n        if (finalText) state.agentSpan.setAttribute('output.value', finalText.slice(0, 10000));\n        state.agentSpan.setStatus({ code: SpanStatusCode.OK });\n        state.agentSpan.end();\n        state.agentSpan = undefined;\n        state.agentCtx = undefined;\n      }\n      break;\n    }\n\n    default:\n      break;\n  }\n}\n\nfunction emitLlmSpan(\n  tracer: ReturnType<typeof trace.getTracer>,\n  state: PerAgentState,\n  msg: PiAssistantMessage,\n): void {\n  const attrs: Record<string, any> = { 'neatlogs.span.kind': 'LLM' };\n  if (msg.model) attrs['neatlogs.llm.model_name'] = String(msg.model);\n  if (msg.provider) attrs['neatlogs.llm.provider'] = String(msg.provider);\n  if (msg.stopReason) attrs['neatlogs.llm.stop_reason'] = String(msg.stopReason);\n\n  // Input = the conversation accumulated up to this assistant turn (system +\n  // user + prior assistant/tool messages). Pi Agent's message_end doesn't carry\n  // the prompt, so we reconstruct it from the running inputMessages list.\n  const inMsgs = state.inputMessages;\n  if (inMsgs.length) {\n    inMsgs.forEach((m, i) => {\n      attrs[`neatlogs.llm.input_messages.${i}.role`] = m.role;\n      attrs[`neatlogs.llm.input_messages.${i}.content`] = m.content.slice(0, 10000);\n    });\n    attrs['neatlogs.llm.input'] = safeStringify({ messages: inMsgs }).slice(0, 20000);\n    attrs['input.value'] = safeStringify({ messages: inMsgs }).slice(0, 10000);\n  }\n\n  const { text, toolCalls } = splitAssistantContent(msg.content);\n  // Output: text if present, else a readable tool-call summary so the span isn't blank.\n  const outText = text || toolCalls.map((tc) => `${tc.name}(${safeStringify(tc.arguments)})`).join('\\n');\n  if (outText || toolCalls.length) {\n    attrs['neatlogs.llm.output_messages.0.role'] = 'assistant';\n    attrs['neatlogs.llm.output_messages.0.content'] = (outText || '').slice(0, 10000);\n    const outBlob: Record<string, unknown> = { role: 'assistant', content: outText || '' };\n    if (toolCalls.length) {\n      outBlob.tool_calls = toolCalls.map((tc) => ({ name: tc.name, arguments: tc.arguments }));\n      toolCalls.forEach((tc, j) => {\n        if (tc.name) attrs[`neatlogs.llm.tool_calls.${j}.name`] = tc.name;\n        if (tc.arguments !== undefined)\n          attrs[`neatlogs.llm.tool_calls.${j}.arguments`] = safeStringify(tc.arguments);\n        if (tc.id) attrs[`neatlogs.llm.tool_calls.${j}.id`] = String(tc.id);\n      });\n    }\n    attrs['neatlogs.llm.output'] = safeStringify(outBlob).slice(0, 20000);\n    attrs['output.value'] = (outText || '').slice(0, 10000);\n  }\n\n  const usage = msg.usage;\n  if (usage) {\n    if (usage.input != null) attrs['neatlogs.llm.token_count.prompt'] = usage.input;\n    if (usage.output != null) attrs['neatlogs.llm.token_count.completion'] = usage.output;\n    const total = usage.totalTokens ?? ((usage.input ?? 0) + (usage.output ?? 0));\n    if (total) attrs['neatlogs.llm.token_count.total'] = total;\n    if (usage.cacheRead) attrs['neatlogs.llm.token_count.cache_read'] = usage.cacheRead;\n    if (usage.cacheWrite) attrs['neatlogs.llm.token_count.cache_write'] = usage.cacheWrite;\n  }\n\n  const parent = state.agentCtx ?? otelContext.active();\n  const span = tracer.startSpan(\n    `pi_agent.llm.${msg.model || 'model'}`,\n    { attributes: attrs },\n    parent,\n  );\n  span.setStatus({ code: SpanStatusCode.OK });\n  span.end();\n}\n\n// ---------------------------------------------------------------------------\n// Content helpers\n// ---------------------------------------------------------------------------\n\nfunction splitAssistantContent(\n  content: PiAssistantMessage['content'],\n): { text: string; toolCalls: PiToolCall[] } {\n  const texts: string[] = [];\n  const toolCalls: PiToolCall[] = [];\n  if (Array.isArray(content)) {\n    for (const block of content) {\n      if (!block || typeof block !== 'object') continue;\n      if (block.type === 'text' && typeof block.text === 'string') texts.push(block.text);\n      else if (block.type === 'toolCall')\n        toolCalls.push(block as unknown as PiToolCall);\n      // thinking blocks intentionally omitted from the main output text\n    }\n  } else if (typeof content === 'string') {\n    texts.push(content);\n  }\n  return { text: texts.join(''), toolCalls };\n}\n\n/** Flatten any message's content (string or block array) to readable text. */\nfunction messageText(msg: any): string {\n  if (!msg) return '';\n  const c = msg.content;\n  if (typeof c === 'string') return c;\n  if (!Array.isArray(c)) return '';\n  const parts: string[] = [];\n  for (const block of c) {\n    if (typeof block === 'string') parts.push(block);\n    else if (block && typeof block === 'object') {\n      if (typeof block.text === 'string') parts.push(block.text);\n      else if (block.type === 'toolCall') parts.push(`${block.name ?? 'tool'}(${safeStringify(block.arguments)})`);\n    }\n  }\n  return parts.join('');\n}\n\nfunction lastAssistantText(messages: any[] | undefined): string {\n  if (!Array.isArray(messages)) return '';\n  for (let i = messages.length - 1; i >= 0; i--) {\n    const m = messages[i];\n    if (m && m.role === 'assistant') {\n      const { text } = splitAssistantContent(m.content);\n      if (text) return text;\n    }\n  }\n  return '';\n}\n\nfunction markPatched(e: any): void {\n  try {\n    Object.defineProperty(e, PATCH_FLAG, { value: true, enumerable: false, configurable: true });\n  } catch {\n    e[PATCH_FLAG] = true;\n  }\n}\n\nfunction safeStringify(value: unknown): string {\n  if (typeof value === 'string') return value;\n  try {\n    return JSON.stringify(value) ?? '';\n  } catch {\n    return '';\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAyBA,iBAMO;AAEP,IAAM,cAAc;AACpB,IAAM,aAAa;AAiDZ,SAAS,aAA+B,OAAa;AAC1D,MAAI,CAAC,SAAU,MAAc,UAAU,EAAG,QAAO;AACjD,QAAM,IAAI;AACV,MAAI,OAAO,EAAE,cAAc,WAAY,QAAO;AAE9C,QAAM,QAAuB,EAAE,WAAW,oBAAI,IAAI,GAAG,eAAe,CAAC,EAAE;AACvE,QAAM,SAAS,iBAAM,UAAU,WAAW;AAE1C,IAAE,UAAU,CAAC,UAAwB;AACnC,QAAI;AACF,kBAAY,QAAQ,OAAO,KAAK;AAAA,IAClC,QAAQ;AAAA,IAER;AAAA,EACF,CAAC;AAED,cAAY,CAAC;AACb,SAAO;AACT;AAEA,SAAS,YACP,QACA,OACA,OACM;AACN,UAAQ,MAAM,MAAM;AAAA,IAClB,KAAK,eAAe;AAElB,YAAM,OAAO,OAAO;AAAA,QAClB;AAAA,QACA,EAAE,YAAY,EAAE,sBAAsB,QAAQ,EAAE;AAAA,QAChD,WAAAA,QAAY,OAAO;AAAA,MACrB;AACA,YAAM,YAAY;AAClB,YAAM,WAAW,iBAAM,QAAQ,WAAAA,QAAY,OAAO,GAAG,IAAI;AACzD,YAAM,gBAAgB,CAAC;AACvB;AAAA,IACF;AAAA,IAEA,KAAK,eAAe;AAClB,YAAM,MAAM,MAAM;AAClB,UAAI,CAAC,IAAK;AACV,UAAI,IAAI,SAAS,aAAa;AAG5B,oBAAY,QAAQ,OAAO,GAAyB;AACpD,cAAM,EAAE,KAAK,IAAI,sBAAsB,IAAI,OAAO;AAClD,YAAI,KAAM,OAAM,cAAc,KAAK,EAAE,MAAM,aAAa,SAAS,KAAK,CAAC;AAAA,MACzE,OAAO;AAEL,cAAM,OAAO,IAAI,SAAS,eAAe,SAAS,OAAO,IAAI,QAAQ,MAAM;AAC3E,cAAM,UAAU,YAAY,GAAG;AAC/B,YAAI,QAAS,OAAM,cAAc,KAAK,EAAE,MAAM,QAAQ,CAAC;AAAA,MACzD;AACA;AAAA,IACF;AAAA,IAEA,KAAK,wBAAwB;AAC3B,YAAM,SAAS,MAAM,YAAY,WAAAA,QAAY,OAAO;AACpD,YAAM,OAAO,OAAO;AAAA,QAClB,iBAAiB,MAAM,YAAY,MAAM;AAAA,QACzC;AAAA,UACE,YAAY;AAAA,YACV,sBAAsB;AAAA,YACtB,GAAI,MAAM,WAAW,EAAE,sBAAsB,OAAO,MAAM,QAAQ,EAAE,IAAI,CAAC;AAAA,YACzE,GAAI,MAAM,SAAS,SACf,EAAE,eAAe,cAAc,MAAM,IAAI,EAAE,MAAM,GAAG,GAAK,EAAE,IAC3D,CAAC;AAAA,UACP;AAAA,QACF;AAAA,QACA;AAAA,MACF;AACA,UAAI,MAAM,WAAY,OAAM,UAAU,IAAI,MAAM,YAAY,IAAI;AAChE;AAAA,IACF;AAAA,IAEA,KAAK,sBAAsB;AACzB,YAAM,OAAO,MAAM,aAAa,MAAM,UAAU,IAAI,MAAM,UAAU,IAAI;AACxE,UAAI,CAAC,KAAM;AACX,UAAI,MAAM,WAAW,QAAW;AAC9B,aAAK,aAAa,gBAAgB,cAAc,MAAM,MAAM,EAAE,MAAM,GAAG,GAAK,CAAC;AAAA,MAC/E;AACA,UAAI,MAAM,SAAS;AACjB,aAAK,UAAU,EAAE,MAAM,0BAAe,MAAM,CAAC;AAC7C,aAAK,aAAa,0BAA0B,IAAI;AAAA,MAClD,OAAO;AACL,aAAK,UAAU,EAAE,MAAM,0BAAe,GAAG,CAAC;AAAA,MAC5C;AACA,WAAK,IAAI;AACT,UAAI,MAAM,WAAY,OAAM,UAAU,OAAO,MAAM,UAAU;AAC7D;AAAA,IACF;AAAA,IAEA,KAAK,aAAa;AAEhB,iBAAW,MAAM,MAAM,UAAU,OAAO,GAAG;AACzC,YAAI;AACF,aAAG,IAAI;AAAA,QACT,QAAQ;AAAA,QAER;AAAA,MACF;AACA,YAAM,UAAU,MAAM;AACtB,UAAI,MAAM,WAAW;AAEnB,cAAM,YAAY,MAAM,cAAc,KAAK,CAAC,MAAM,EAAE,SAAS,MAAM;AACnE,YAAI,UAAW,OAAM,UAAU,aAAa,eAAe,UAAU,QAAQ,MAAM,GAAG,GAAK,CAAC;AAC5F,cAAM,YAAY,kBAAkB,MAAM,QAAQ;AAClD,YAAI,UAAW,OAAM,UAAU,aAAa,gBAAgB,UAAU,MAAM,GAAG,GAAK,CAAC;AACrF,cAAM,UAAU,UAAU,EAAE,MAAM,0BAAe,GAAG,CAAC;AACrD,cAAM,UAAU,IAAI;AACpB,cAAM,YAAY;AAClB,cAAM,WAAW;AAAA,MACnB;AACA;AAAA,IACF;AAAA,IAEA;AACE;AAAA,EACJ;AACF;AAEA,SAAS,YACP,QACA,OACA,KACM;AACN,QAAM,QAA6B,EAAE,sBAAsB,MAAM;AACjE,MAAI,IAAI,MAAO,OAAM,yBAAyB,IAAI,OAAO,IAAI,KAAK;AAClE,MAAI,IAAI,SAAU,OAAM,uBAAuB,IAAI,OAAO,IAAI,QAAQ;AACtE,MAAI,IAAI,WAAY,OAAM,0BAA0B,IAAI,OAAO,IAAI,UAAU;AAK7E,QAAM,SAAS,MAAM;AACrB,MAAI,OAAO,QAAQ;AACjB,WAAO,QAAQ,CAAC,GAAG,MAAM;AACvB,YAAM,+BAA+B,CAAC,OAAO,IAAI,EAAE;AACnD,YAAM,+BAA+B,CAAC,UAAU,IAAI,EAAE,QAAQ,MAAM,GAAG,GAAK;AAAA,IAC9E,CAAC;AACD,UAAM,oBAAoB,IAAI,cAAc,EAAE,UAAU,OAAO,CAAC,EAAE,MAAM,GAAG,GAAK;AAChF,UAAM,aAAa,IAAI,cAAc,EAAE,UAAU,OAAO,CAAC,EAAE,MAAM,GAAG,GAAK;AAAA,EAC3E;AAEA,QAAM,EAAE,MAAM,UAAU,IAAI,sBAAsB,IAAI,OAAO;AAE7D,QAAM,UAAU,QAAQ,UAAU,IAAI,CAAC,OAAO,GAAG,GAAG,IAAI,IAAI,cAAc,GAAG,SAAS,CAAC,GAAG,EAAE,KAAK,IAAI;AACrG,MAAI,WAAW,UAAU,QAAQ;AAC/B,UAAM,qCAAqC,IAAI;AAC/C,UAAM,wCAAwC,KAAK,WAAW,IAAI,MAAM,GAAG,GAAK;AAChF,UAAM,UAAmC,EAAE,MAAM,aAAa,SAAS,WAAW,GAAG;AACrF,QAAI,UAAU,QAAQ;AACpB,cAAQ,aAAa,UAAU,IAAI,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,WAAW,GAAG,UAAU,EAAE;AACvF,gBAAU,QAAQ,CAAC,IAAI,MAAM;AAC3B,YAAI,GAAG,KAAM,OAAM,2BAA2B,CAAC,OAAO,IAAI,GAAG;AAC7D,YAAI,GAAG,cAAc;AACnB,gBAAM,2BAA2B,CAAC,YAAY,IAAI,cAAc,GAAG,SAAS;AAC9E,YAAI,GAAG,GAAI,OAAM,2BAA2B,CAAC,KAAK,IAAI,OAAO,GAAG,EAAE;AAAA,MACpE,CAAC;AAAA,IACH;AACA,UAAM,qBAAqB,IAAI,cAAc,OAAO,EAAE,MAAM,GAAG,GAAK;AACpE,UAAM,cAAc,KAAK,WAAW,IAAI,MAAM,GAAG,GAAK;AAAA,EACxD;AAEA,QAAM,QAAQ,IAAI;AAClB,MAAI,OAAO;AACT,QAAI,MAAM,SAAS,KAAM,OAAM,iCAAiC,IAAI,MAAM;AAC1E,QAAI,MAAM,UAAU,KAAM,OAAM,qCAAqC,IAAI,MAAM;AAC/E,UAAM,QAAQ,MAAM,gBAAiB,MAAM,SAAS,MAAM,MAAM,UAAU;AAC1E,QAAI,MAAO,OAAM,gCAAgC,IAAI;AACrD,QAAI,MAAM,UAAW,OAAM,qCAAqC,IAAI,MAAM;AAC1E,QAAI,MAAM,WAAY,OAAM,sCAAsC,IAAI,MAAM;AAAA,EAC9E;AAEA,QAAM,SAAS,MAAM,YAAY,WAAAA,QAAY,OAAO;AACpD,QAAM,OAAO,OAAO;AAAA,IAClB,gBAAgB,IAAI,SAAS,OAAO;AAAA,IACpC,EAAE,YAAY,MAAM;AAAA,IACpB;AAAA,EACF;AACA,OAAK,UAAU,EAAE,MAAM,0BAAe,GAAG,CAAC;AAC1C,OAAK,IAAI;AACX;AAMA,SAAS,sBACP,SAC2C;AAC3C,QAAM,QAAkB,CAAC;AACzB,QAAM,YAA0B,CAAC;AACjC,MAAI,MAAM,QAAQ,OAAO,GAAG;AAC1B,eAAW,SAAS,SAAS;AAC3B,UAAI,CAAC,SAAS,OAAO,UAAU,SAAU;AACzC,UAAI,MAAM,SAAS,UAAU,OAAO,MAAM,SAAS,SAAU,OAAM,KAAK,MAAM,IAAI;AAAA,eACzE,MAAM,SAAS;AACtB,kBAAU,KAAK,KAA8B;AAAA,IAEjD;AAAA,EACF,WAAW,OAAO,YAAY,UAAU;AACtC,UAAM,KAAK,OAAO;AAAA,EACpB;AACA,SAAO,EAAE,MAAM,MAAM,KAAK,EAAE,GAAG,UAAU;AAC3C;AAGA,SAAS,YAAY,KAAkB;AACrC,MAAI,CAAC,IAAK,QAAO;AACjB,QAAM,IAAI,IAAI;AACd,MAAI,OAAO,MAAM,SAAU,QAAO;AAClC,MAAI,CAAC,MAAM,QAAQ,CAAC,EAAG,QAAO;AAC9B,QAAM,QAAkB,CAAC;AACzB,aAAW,SAAS,GAAG;AACrB,QAAI,OAAO,UAAU,SAAU,OAAM,KAAK,KAAK;AAAA,aACtC,SAAS,OAAO,UAAU,UAAU;AAC3C,UAAI,OAAO,MAAM,SAAS,SAAU,OAAM,KAAK,MAAM,IAAI;AAAA,eAChD,MAAM,SAAS,WAAY,OAAM,KAAK,GAAG,MAAM,QAAQ,MAAM,IAAI,cAAc,MAAM,SAAS,CAAC,GAAG;AAAA,IAC7G;AAAA,EACF;AACA,SAAO,MAAM,KAAK,EAAE;AACtB;AAEA,SAAS,kBAAkB,UAAqC;AAC9D,MAAI,CAAC,MAAM,QAAQ,QAAQ,EAAG,QAAO;AACrC,WAAS,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;AAC7C,UAAM,IAAI,SAAS,CAAC;AACpB,QAAI,KAAK,EAAE,SAAS,aAAa;AAC/B,YAAM,EAAE,KAAK,IAAI,sBAAsB,EAAE,OAAO;AAChD,UAAI,KAAM,QAAO;AAAA,IACnB;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,YAAY,GAAc;AACjC,MAAI;AACF,WAAO,eAAe,GAAG,YAAY,EAAE,OAAO,MAAM,YAAY,OAAO,cAAc,KAAK,CAAC;AAAA,EAC7F,QAAQ;AACN,MAAE,UAAU,IAAI;AAAA,EAClB;AACF;AAEA,SAAS,cAAc,OAAwB;AAC7C,MAAI,OAAO,UAAU,SAAU,QAAO;AACtC,MAAI;AACF,WAAO,KAAK,UAAU,KAAK,KAAK;AAAA,EAClC,QAAQ;AACN,WAAO;AAAA,EACT;AACF;","names":["otelContext"]}