neatlogs 1.0.5 → 1.0.8

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) } : {}
+          }
+        },
+        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));
+      }
+      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);
+        const finalText = lastAssistantText(event.messages);
+        if (finalText) state.agentSpan.setAttribute("output.value", finalText);
+        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;
+    });
+    attrs["neatlogs.llm.input"] = safeStringify({ messages: inMsgs });
+    attrs["input.value"] = safeStringify({ messages: inMsgs });
+  }
+  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 || "";
+    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);
+    attrs["output.value"] = outText || "";
+  }
+  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) }\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));\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);\n        const finalText = lastAssistantText(event.messages);\n        if (finalText) state.agentSpan.setAttribute('output.value', finalText);\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;\n    });\n    attrs['neatlogs.llm.input'] = safeStringify({ messages: inMsgs });\n    attrs['input.value'] = safeStringify({ messages: inMsgs });\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 || '');\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);\n    attrs['output.value'] = (outText || '');\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,IAC3C,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,CAAC;AAAA,MAC/D;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,OAAO;AAC5E,cAAM,YAAY,kBAAkB,MAAM,QAAQ;AAClD,YAAI,UAAW,OAAM,UAAU,aAAa,gBAAgB,SAAS;AACrE,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;AAAA,IACxD,CAAC;AACD,UAAM,oBAAoB,IAAI,cAAc,EAAE,UAAU,OAAO,CAAC;AAChE,UAAM,aAAa,IAAI,cAAc,EAAE,UAAU,OAAO,CAAC;AAAA,EAC3D;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,IAAK,WAAW;AAC9D,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;AACpD,UAAM,cAAc,IAAK,WAAW;AAAA,EACtC;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"]}

package/dist/pi-agent.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Neatlogs Pi Agent integration.
+ *
+ * Usage:
+ *   import { init } from 'neatlogs';
+ *   import { piAgentHooks } from 'neatlogs/pi-agent';
+ *   import { Agent } from '@mariozechner/pi-agent-core';
+ *
+ *   await init({ apiKey, workflowName });
+ *   const agent = piAgentHooks(new Agent({ initialState: { systemPrompt, model } }));
+ *   await agent.prompt('Hello');
+ *
+ * Pi Agent's `Agent` exposes a first-class `subscribe(listener)` API and emits
+ * AgentEvents for its run lifecycle (message_*, tool_execution_*, turn_*,
+ * agent_*). It does NOT emit its own OpenTelemetry spans — so we LISTEN to those
+ * events (no monkey-patching) and translate them into neatlogs OTel spans:
+ *
+ *   AGENT  agent run                    (agent_start → agent_end)
+ *     ↳ LLM   assistant message         (each assistant message_end)
+ *     ↳ TOOL  tool call                 (tool_execution_start → tool_execution_end)
+ *
+ * The AGENT span is opened as the active span so the LLM/TOOL children nest under
+ * it (and under any user @span / trace() block active when prompt() is called).
+ */
+/**
+ * Subscribe neatlogs tracing to a Pi Agent instance. Returns the same agent
+ * (marked so re-subscribing is a no-op). Idempotent per agent.
+ */
+declare function piAgentHooks<T extends object>(agent: T): T;
+
+export { piAgentHooks };

package/dist/pi-agent.mjs

@@ -0,0 +1,216 @@
+// src/pi-agent.ts
+import {
+  trace,
+  context as otelContext,
+  SpanStatusCode
+} from "@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 = 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" } },
+        otelContext.active()
+      );
+      state.agentSpan = span;
+      state.agentCtx = trace.setSpan(otelContext.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 ?? otelContext.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) } : {}
+          }
+        },
+        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));
+      }
+      if (event.isError) {
+        span.setStatus({ code: SpanStatusCode.ERROR });
+        span.setAttribute("neatlogs.tool.is_error", true);
+      } else {
+        span.setStatus({ code: 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);
+        const finalText = lastAssistantText(event.messages);
+        if (finalText) state.agentSpan.setAttribute("output.value", finalText);
+        state.agentSpan.setStatus({ code: 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;
+    });
+    attrs["neatlogs.llm.input"] = safeStringify({ messages: inMsgs });
+    attrs["input.value"] = safeStringify({ messages: inMsgs });
+  }
+  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 || "";
+    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);
+    attrs["output.value"] = outText || "";
+  }
+  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 ?? otelContext.active();
+  const span = tracer.startSpan(
+    `pi_agent.llm.${msg.model || "model"}`,
+    { attributes: attrs },
+    parent
+  );
+  span.setStatus({ code: 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 "";
+  }
+}
+export {
+  piAgentHooks
+};
+//# sourceMappingURL=pi-agent.mjs.map

package/dist/pi-agent.mjs.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) }\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));\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);\n        const finalText = lastAssistantText(event.messages);\n        if (finalText) state.agentSpan.setAttribute('output.value', finalText);\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;\n    });\n    attrs['neatlogs.llm.input'] = safeStringify({ messages: inMsgs });\n    attrs['input.value'] = safeStringify({ messages: inMsgs });\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 || '');\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);\n    attrs['output.value'] = (outText || '');\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":";AAyBA;AAAA,EACE;AAAA,EACA,WAAW;AAAA,EACX;AAAA,OAGK;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,MAAM,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,YAAY,OAAO;AAAA,MACrB;AACA,YAAM,YAAY;AAClB,YAAM,WAAW,MAAM,QAAQ,YAAY,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,YAAY,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,IAC3C,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,CAAC;AAAA,MAC/D;AACA,UAAI,MAAM,SAAS;AACjB,aAAK,UAAU,EAAE,MAAM,eAAe,MAAM,CAAC;AAC7C,aAAK,aAAa,0BAA0B,IAAI;AAAA,MAClD,OAAO;AACL,aAAK,UAAU,EAAE,MAAM,eAAe,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,OAAO;AAC5E,cAAM,YAAY,kBAAkB,MAAM,QAAQ;AAClD,YAAI,UAAW,OAAM,UAAU,aAAa,gBAAgB,SAAS;AACrE,cAAM,UAAU,UAAU,EAAE,MAAM,eAAe,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;AAAA,IACxD,CAAC;AACD,UAAM,oBAAoB,IAAI,cAAc,EAAE,UAAU,OAAO,CAAC;AAChE,UAAM,aAAa,IAAI,cAAc,EAAE,UAAU,OAAO,CAAC;AAAA,EAC3D;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,IAAK,WAAW;AAC9D,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;AACpD,UAAM,cAAc,IAAK,WAAW;AAAA,EACtC;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,YAAY,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,eAAe,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":[]}