neatlogs 1.0.5 → 1.0.6

package/dist/index.cjs

@@ -54,6 +54,7 @@ __export(index_exports, {
   listPrompts: () => listPrompts,
   log: () => log,
   openaiAgentsProcessor: () => openaiAgentsProcessor,
+  piAgentHooks: () => piAgentHooks,
   registerCrewaiTask: () => registerCrewaiTask,
   removeTag: () => removeTag,
   saveAsVersion: () => saveAsVersion,
@@ -5862,7 +5863,7 @@ async function removeTag(name, tag) {
 }
 
 // src/version.ts
-var __version__ = "1.0.0";
+var __version__ = "1.0.6";
 
 // src/init.ts
 var logger13 = getLogger();
@@ -7391,20 +7392,143 @@ function safeStringify5(value) {
 }
 
 // src/strands.ts
+var PATCH_FLAG = "_neatlogs_patched";
 function strandsHooks(agent) {
+  installEventHookFromAgent(agent);
   if (agent && typeof agent === "object") {
     try {
-      Object.defineProperty(agent, "_neatlogs_patched", {
+      Object.defineProperty(agent, PATCH_FLAG, {
         value: true,
         enumerable: false,
         configurable: true
       });
     } catch {
-      agent._neatlogs_patched = true;
+      agent[PATCH_FLAG] = true;
     }
   }
   return agent;
 }
+function installEventHookFromAgent(agent) {
+  const tracer = agent?._tracer;
+  const proto = tracer ? Object.getPrototypeOf(tracer) : void 0;
+  if (!proto || typeof proto._addEvent !== "function") return;
+  if (proto._addEvent[PATCH_FLAG]) return;
+  const orig = proto._addEvent;
+  function patchedAddEvent(span2, eventName, eventAttributes) {
+    const result = orig.call(this, span2, eventName, eventAttributes);
+    try {
+      enrichSpanFromEvent(span2, eventName, eventAttributes || {});
+    } catch {
+    }
+    return result;
+  }
+  patchedAddEvent[PATCH_FLAG] = true;
+  proto._addEvent = patchedAddEvent;
+}
+function enrichSpanFromEvent(span2, eventName, attrs) {
+  if (!span2 || typeof span2.setAttribute !== "function") return;
+  if (typeof span2.isRecording === "function" && !span2.isRecording()) return;
+  const isTool = readSpanOp(span2) === "execute_tool";
+  if (eventName.startsWith("gen_ai.") && eventName.endsWith(".message")) {
+    const role = eventName.slice("gen_ai.".length, -".message".length);
+    const content = flattenStrandsContent(attrs.content);
+    if (content) {
+      if (isTool) {
+        span2.setAttribute("input.value", content);
+        span2.setAttribute("neatlogs.tool.input", content);
+      } else {
+        appendInputMessage(span2, role, content);
+      }
+    }
+    return;
+  }
+  if (eventName === "gen_ai.choice") {
+    const out = flattenStrandsContent(attrs.message);
+    if (out) setOutput(span2, isTool, out);
+    return;
+  }
+  if (eventName === "gen_ai.client.inference.operation.details") {
+    const out = flattenStrandsContent(attrs["gen_ai.output.messages"]);
+    if (out) setOutput(span2, isTool, out);
+  }
+}
+function setOutput(span2, isTool, out) {
+  span2.setAttribute("output.value", out);
+  if (isTool) {
+    span2.setAttribute("neatlogs.tool.output", out);
+  } else {
+    span2.setAttribute("neatlogs.llm.output_messages.0.role", "assistant");
+    span2.setAttribute("neatlogs.llm.output_messages.0.content", out);
+    span2.setAttribute("neatlogs.llm.output", safeStringify6({ role: "assistant", content: out }));
+  }
+}
+function appendInputMessage(span2, role, content) {
+  const idx = span2.__neatlogs_in_idx ?? 0;
+  span2.setAttribute(`neatlogs.llm.input_messages.${idx}.role`, role);
+  span2.setAttribute(`neatlogs.llm.input_messages.${idx}.content`, content);
+  span2.__neatlogs_in_idx = idx + 1;
+  const existing = span2.__neatlogs_in_msgs ?? [];
+  existing.push({ role, content });
+  span2.__neatlogs_in_msgs = existing;
+  const blob = safeStringify6({ messages: existing });
+  span2.setAttribute("input.value", blob);
+  span2.setAttribute("neatlogs.llm.input", blob);
+}
+function readSpanOp(span2) {
+  try {
+    const a = span2.attributes;
+    if (!a) return "";
+    const v = typeof a.get === "function" ? a.get("gen_ai.operation.name") : a["gen_ai.operation.name"];
+    return typeof v === "string" ? v.toLowerCase() : "";
+  } catch {
+    return "";
+  }
+}
+function flattenStrandsContent(content) {
+  if (content == null) return "";
+  let val = content;
+  if (typeof val === "string") {
+    const s = val.trim();
+    if (!(s.startsWith("[") || s.startsWith("{"))) return val;
+    try {
+      val = JSON.parse(s);
+    } catch {
+      return val;
+    }
+  }
+  return flattenBlocks(val);
+}
+function flattenBlocks(val) {
+  const items = Array.isArray(val) ? val : [val];
+  const out = [];
+  for (const item of items) {
+    if (typeof item === "string") {
+      out.push(item);
+      continue;
+    }
+    if (!item || typeof item !== "object") {
+      if (item != null) out.push(String(item));
+      continue;
+    }
+    const o = item;
+    if (typeof o.text === "string") out.push(o.text);
+    else if (typeof o.content === "string") out.push(o.content);
+    else if (o.toolUse) out.push(`${o.toolUse.name ?? "tool"}(${safeStringify6(o.toolUse.input ?? {})})`);
+    else if (o.toolResult) out.push(flattenBlocks(o.toolResult.content ?? o.toolResult));
+    else if (Array.isArray(o.parts)) out.push(flattenBlocks(o.parts));
+    else if (Array.isArray(o.content)) out.push(flattenBlocks(o.content));
+    else out.push(safeStringify6(o));
+  }
+  return out.filter(Boolean).join("\n");
+}
+function safeStringify6(value) {
+  if (typeof value === "string") return value;
+  try {
+    return JSON.stringify(value) ?? "";
+  } catch {
+    return "";
+  }
+}
 
 // src/openai-agents.ts
 var import_api12 = require("@opentelemetry/api");
@@ -7476,7 +7600,7 @@ var NeatlogsTraceProcessor = class {
           const role = typeof msg === "object" ? msg.role ?? "" : "";
           const content = typeof msg === "object" ? msg.content ?? "" : String(msg);
           if (role) attrs[`neatlogs.llm.input_messages.${i}.role`] = role;
-          if (content) attrs[`neatlogs.llm.input_messages.${i}.content`] = (typeof content === "string" ? content : safeStringify6(content)).slice(0, 1e4);
+          if (content) attrs[`neatlogs.llm.input_messages.${i}.content`] = (typeof content === "string" ? content : safeStringify7(content)).slice(0, 1e4);
         }
       }
       otelSpan = tracer.startSpan("openai_agents.generation", { attributes: attrs }, parentCtx);
@@ -7488,7 +7612,7 @@ var NeatlogsTraceProcessor = class {
       };
       const toolInput = data?.input ?? data?.arguments;
       if (toolInput !== void 0) {
-        attrs["input.value"] = (typeof toolInput === "string" ? toolInput : safeStringify6(toolInput)).slice(0, 1e4);
+        attrs["input.value"] = (typeof toolInput === "string" ? toolInput : safeStringify7(toolInput)).slice(0, 1e4);
       }
       otelSpan = tracer.startSpan(`openai_agents.tool.${toolName}`, { attributes: attrs }, parentCtx);
     } else if (spanType === "handoff") {
@@ -7525,13 +7649,13 @@ var NeatlogsTraceProcessor = class {
         for (let i = 0; i < toolCalls.length; i++) {
           const tc = toolCalls[i];
           otelSpan.setAttribute(`neatlogs.llm.tool_calls.${i}.name`, tc.name ?? "");
-          otelSpan.setAttribute(`neatlogs.llm.tool_calls.${i}.arguments`, typeof tc.arguments === "string" ? tc.arguments : safeStringify6(tc.arguments ?? {}));
+          otelSpan.setAttribute(`neatlogs.llm.tool_calls.${i}.arguments`, typeof tc.arguments === "string" ? tc.arguments : safeStringify7(tc.arguments ?? {}));
           if (tc.callId ?? tc.id) otelSpan.setAttribute(`neatlogs.llm.tool_calls.${i}.id`, tc.callId ?? tc.id);
         }
       } else if (outputItems?.content) {
         otelSpan.setAttribute("neatlogs.llm.output_messages.0.role", "assistant");
         const c = outputItems.content;
-        otelSpan.setAttribute("neatlogs.llm.output_messages.0.content", (typeof c === "string" ? c : safeStringify6(c)).slice(0, 1e4));
+        otelSpan.setAttribute("neatlogs.llm.output_messages.0.content", (typeof c === "string" ? c : safeStringify7(c)).slice(0, 1e4));
       }
       const usage = data?.usage ?? resp?.usage;
       if (usage) {
@@ -7548,12 +7672,12 @@ var NeatlogsTraceProcessor = class {
     } else if (spanType === "function" || spanType === "tool" || spanType === "tool_call") {
       const output = data?.output ?? data?.result;
       if (output != null) {
-        otelSpan.setAttribute("output.value", (typeof output === "string" ? output : safeStringify6(output)).slice(0, 1e4));
+        otelSpan.setAttribute("output.value", (typeof output === "string" ? output : safeStringify7(output)).slice(0, 1e4));
       }
     } else if (spanType === "agent" || spanType === "agent_run") {
       const output = data?.output;
       if (output != null) {
-        otelSpan.setAttribute("output.value", (typeof output === "string" ? output : safeStringify6(output)).slice(0, 1e4));
+        otelSpan.setAttribute("output.value", (typeof output === "string" ? output : safeStringify7(output)).slice(0, 1e4));
       }
     }
     const error = data?.error ?? span2?.error;
@@ -7585,7 +7709,7 @@ var NeatlogsTraceProcessor = class {
   forceFlush() {
   }
 };
-function safeStringify6(value) {
+function safeStringify7(value) {
   try {
     return typeof value === "string" ? value : JSON.stringify(value);
   } catch {
@@ -7596,9 +7720,9 @@ function safeStringify6(value) {
 // src/mastra-wrap.ts
 var import_api13 = require("@opentelemetry/api");
 var TRACER_NAME7 = "neatlogs.mastra";
-var PATCH_FLAG = "_neatlogs_patched";
+var PATCH_FLAG2 = "_neatlogs_patched";
 function wrapMastra(entity) {
-  if (!entity || entity[PATCH_FLAG]) return entity;
+  if (!entity || entity[PATCH_FLAG2]) return entity;
   const e = entity;
   const className = entity.constructor?.name ?? "";
   if (isRootMastra(e)) {
@@ -7711,7 +7835,7 @@ function patchModelInPlace(model) {
       if (provider) attrs["neatlogs.llm.provider"] = provider;
       if (isStream) attrs["neatlogs.llm.is_streaming"] = true;
       const promptInput = callOpts?.prompt ?? callOpts?.messages;
-      if (promptInput !== void 0) attrs["input.value"] = safeStringify7(promptInput).slice(0, 1e4);
+      if (promptInput !== void 0) attrs["input.value"] = safeStringify8(promptInput).slice(0, 1e4);
       captureInvocationParams(attrs, callOpts);
       return withActiveSpan(`mastra.llm.${modelId || "model"}.${fn}`, attrs, async (span2) => {
         const result = await orig(callOpts);
@@ -7745,12 +7869,12 @@ function patchToolExecute(tool, key) {
     const attrs = {
       "neatlogs.span.kind": "TOOL",
       "neatlogs.tool.name": toolName,
-      "input.value": safeStringify7(params).slice(0, 1e4)
+      "input.value": safeStringify8(params).slice(0, 1e4)
     };
     if (tool.description) attrs["neatlogs.tool.description"] = String(tool.description).slice(0, 2e3);
     return withActiveSpan(`mastra.tool.${toolName}`, attrs, async (span2) => {
       const result = await orig(params, options);
-      span2.setAttribute("output.value", safeStringify7(result).slice(0, 1e4));
+      span2.setAttribute("output.value", safeStringify8(result).slice(0, 1e4));
       return result;
     });
   };
@@ -7776,13 +7900,13 @@ function patchRunMethod(run, method, workflowName) {
       "neatlogs.workflow.name": workflowName
     };
     if (startOpts?.inputData !== void 0) {
-      attrs["input.value"] = safeStringify7(startOpts.inputData).slice(0, 1e4);
+      attrs["input.value"] = safeStringify8(startOpts.inputData).slice(0, 1e4);
     }
     return withActiveSpan(`mastra.workflow.${workflowName}`, attrs, async (span2) => {
       const result = await orig(startOpts);
-      if (result?.status) span2.setAttribute("neatlogs.metadata", safeStringify7({ status: result.status }));
+      if (result?.status) span2.setAttribute("neatlogs.metadata", safeStringify8({ status: result.status }));
       if (result?.result !== void 0) {
-        span2.setAttribute("output.value", safeStringify7(result.result).slice(0, 1e4));
+        span2.setAttribute("output.value", safeStringify8(result.result).slice(0, 1e4));
       }
       return result;
     });
@@ -7803,14 +7927,14 @@ function patchVectorOp(vector, op, kind) {
       "neatlogs.span.kind": kind,
       "neatlogs.db.system": dbName,
       "neatlogs.db.operation": op,
-      "input.value": safeStringify7(params).slice(0, 1e4)
+      "input.value": safeStringify8(params).slice(0, 1e4)
     };
     const indexName = params?.indexName;
     if (indexName) attrs["neatlogs.vectordb.index_name"] = String(indexName);
     if (kind === "RETRIEVER" && params?.topK != null) attrs["neatlogs.retriever.top_k"] = params.topK;
     return withActiveSpan(`mastra.vector.${op}`, attrs, async (span2) => {
       const result = await orig(params);
-      span2.setAttribute("output.value", safeStringify7(result).slice(0, 1e4));
+      span2.setAttribute("output.value", safeStringify8(result).slice(0, 1e4));
       return result;
     });
   };
@@ -7824,11 +7948,11 @@ function patchMemory(memory) {
       const attrs = {
         "neatlogs.span.kind": "CHAIN",
         "neatlogs.db.operation": op,
-        "input.value": safeStringify7(args?.[0]).slice(0, 1e4)
+        "input.value": safeStringify8(args?.[0]).slice(0, 1e4)
       };
       return withActiveSpan(`mastra.memory.${op}`, attrs, async (span2) => {
         const result = await orig(...args);
-        span2.setAttribute("output.value", safeStringify7(result).slice(0, 8e3));
+        span2.setAttribute("output.value", safeStringify8(result).slice(0, 8e3));
         return result;
       });
     };
@@ -8011,7 +8135,7 @@ function finalizeModelResult(span2, result) {
       const tc = toolCalls[i];
       setToolCall(span2, i, tc.toolName, tc.input ?? tc.args, tc.toolCallId);
     }
-    span2.setAttribute("output.value", safeStringify7(result.content).slice(0, 1e4));
+    span2.setAttribute("output.value", safeStringify8(result.content).slice(0, 1e4));
   }
   if (result.usage) recordUsage(span2, result.usage);
   span2.setAttribute("neatlogs.llm.stop_reason", normalizeFinishReason(result.finishReason));
@@ -8019,7 +8143,7 @@ function finalizeModelResult(span2, result) {
 function normalizeFinishReason(fr) {
   if (fr == null) return "";
   if (typeof fr === "string") return fr;
-  return String(fr.unified ?? fr.reason ?? fr.type ?? safeStringify7(fr));
+  return String(fr.unified ?? fr.reason ?? fr.type ?? safeStringify8(fr));
 }
 function tokenValue(v) {
   if (v == null) return void 0;
@@ -8048,22 +8172,22 @@ function captureInvocationParams(attrs, callOpts) {
     }
   }
   if (Array.isArray(callOpts.stopSequences) && callOpts.stopSequences.length) {
-    attrs["neatlogs.llm.stop_sequences"] = safeStringify7(callOpts.stopSequences);
+    attrs["neatlogs.llm.stop_sequences"] = safeStringify8(callOpts.stopSequences);
     invocation.stopSequences = callOpts.stopSequences;
   }
   if (Array.isArray(callOpts.tools)) {
     for (let i = 0; i < callOpts.tools.length; i++) {
-      attrs[`neatlogs.llm.tools.${i}`] = safeStringify7(callOpts.tools[i]).slice(0, 4e3);
+      attrs[`neatlogs.llm.tools.${i}`] = safeStringify8(callOpts.tools[i]).slice(0, 4e3);
     }
     invocation.toolChoice = callOpts.toolChoice;
   }
   if (Object.keys(invocation).length) {
-    attrs["neatlogs.llm.invocation_parameters"] = safeStringify7(invocation).slice(0, 4e3);
+    attrs["neatlogs.llm.invocation_parameters"] = safeStringify8(invocation).slice(0, 4e3);
   }
 }
 function setToolCall(span2, i, name, args, id) {
   span2.setAttribute(`neatlogs.llm.tool_calls.${i}.name`, name ?? "");
-  span2.setAttribute(`neatlogs.llm.tool_calls.${i}.arguments`, safeStringify7(args ?? {}));
+  span2.setAttribute(`neatlogs.llm.tool_calls.${i}.arguments`, safeStringify8(args ?? {}));
   if (id) span2.setAttribute(`neatlogs.llm.tool_calls.${i}.id`, id);
 }
 function recordUsage(span2, usage) {
@@ -8097,9 +8221,9 @@ function withActiveSpan(name, attrs, fn) {
 }
 function markPatched(e) {
   try {
-    Object.defineProperty(e, PATCH_FLAG, { value: true, enumerable: false, configurable: true });
+    Object.defineProperty(e, PATCH_FLAG2, { value: true, enumerable: false, configurable: true });
   } catch {
-    e[PATCH_FLAG] = true;
+    e[PATCH_FLAG2] = true;
   }
 }
 function extractModelId(model) {
@@ -8108,9 +8232,9 @@ function extractModelId(model) {
   return model.modelId ?? model.name ?? "";
 }
 function toInputValue(input) {
-  return (typeof input === "string" ? input : safeStringify7(input)).slice(0, 1e4);
+  return (typeof input === "string" ? input : safeStringify8(input)).slice(0, 1e4);
 }
-function safeStringify7(value) {
+function safeStringify8(value) {
   if (typeof value === "string") return value;
   try {
     return JSON.stringify(value) ?? "";
@@ -8127,6 +8251,215 @@ function recordError3(span2, err) {
   }
 }
 
+// src/pi-agent.ts
+var import_api14 = require("@opentelemetry/api");
+var TRACER_NAME8 = "neatlogs.pi-agent";
+var PATCH_FLAG3 = "_neatlogs_patched";
+function piAgentHooks(agent) {
+  if (!agent || agent[PATCH_FLAG3]) return agent;
+  const a = agent;
+  if (typeof a.subscribe !== "function") return agent;
+  const state = { toolSpans: /* @__PURE__ */ new Map(), inputMessages: [] };
+  const tracer = import_api14.trace.getTracer(TRACER_NAME8);
+  a.subscribe((event) => {
+    try {
+      handleEvent(tracer, state, event);
+    } catch {
+    }
+  });
+  markPatched2(a);
+  return agent;
+}
+function handleEvent(tracer, state, event) {
+  switch (event.type) {
+    case "agent_start": {
+      const span2 = tracer.startSpan(
+        "pi_agent.run",
+        { attributes: { "neatlogs.span.kind": "AGENT" } },
+        import_api14.context.active()
+      );
+      state.agentSpan = span2;
+      state.agentCtx = import_api14.trace.setSpan(import_api14.context.active(), span2);
+      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_api14.context.active();
+      const span2 = tracer.startSpan(
+        `pi_agent.tool.${event.toolName ?? "tool"}`,
+        {
+          attributes: {
+            "neatlogs.span.kind": "TOOL",
+            ...event.toolName ? { "neatlogs.tool.name": String(event.toolName) } : {},
+            ...event.args !== void 0 ? { "input.value": safeStringify9(event.args).slice(0, 1e4) } : {}
+          }
+        },
+        parent
+      );
+      if (event.toolCallId) state.toolSpans.set(event.toolCallId, span2);
+      break;
+    }
+    case "tool_execution_end": {
+      const span2 = event.toolCallId ? state.toolSpans.get(event.toolCallId) : void 0;
+      if (!span2) return;
+      if (event.result !== void 0) {
+        span2.setAttribute("output.value", safeStringify9(event.result).slice(0, 1e4));
+      }
+      if (event.isError) {
+        span2.setStatus({ code: import_api14.SpanStatusCode.ERROR });
+        span2.setAttribute("neatlogs.tool.is_error", true);
+      } else {
+        span2.setStatus({ code: import_api14.SpanStatusCode.OK });
+      }
+      span2.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_api14.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"] = safeStringify9({ messages: inMsgs }).slice(0, 2e4);
+    attrs["input.value"] = safeStringify9({ messages: inMsgs }).slice(0, 1e4);
+  }
+  const { text, toolCalls } = splitAssistantContent(msg.content);
+  const outText = text || toolCalls.map((tc) => `${tc.name}(${safeStringify9(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`] = safeStringify9(tc.arguments);
+        if (tc.id) attrs[`neatlogs.llm.tool_calls.${j}.id`] = String(tc.id);
+      });
+    }
+    attrs["neatlogs.llm.output"] = safeStringify9(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_api14.context.active();
+  const span2 = tracer.startSpan(
+    `pi_agent.llm.${msg.model || "model"}`,
+    { attributes: attrs },
+    parent
+  );
+  span2.setStatus({ code: import_api14.SpanStatusCode.OK });
+  span2.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"}(${safeStringify9(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 markPatched2(e) {
+  try {
+    Object.defineProperty(e, PATCH_FLAG3, { value: true, enumerable: false, configurable: true });
+  } catch {
+    e[PATCH_FLAG3] = true;
+  }
+}
+function safeStringify9(value) {
+  if (typeof value === "string") return value;
+  try {
+    return JSON.stringify(value) ?? "";
+  } catch {
+    return "";
+  }
+}
+
 // src/core/llm-binder.ts
 var logger15 = getLogger();
 function bindTemplates(llm, systemTpl, userTpl, compiledVars) {
@@ -8202,6 +8535,7 @@ function bindTemplates(llm, systemTpl, userTpl, compiledVars) {
   listPrompts,
   log,
   openaiAgentsProcessor,
+  piAgentHooks,
   registerCrewaiTask,
   removeTag,
   saveAsVersion,