@raindrop-ai/pi-agent 0.0.10 → 0.1.1

package/dist/extension.cjs

@@ -25,7 +25,7 @@ __export(extension_exports, {
 });
 module.exports = __toCommonJS(extension_exports);
 
-// ../core/dist/chunk-267IXHO5.js
+// ../core/dist/chunk-YUIXRQWN.js
 function getCrypto() {
   const c = globalThis.crypto;
   return c;
@@ -83,6 +83,20 @@ function base64Encode(bytes) {
 function generateId() {
   return randomUUID();
 }
+var HANDOFF_ATTRIBUTE_SUFFIXES = {
+  mode: "raindrop.handoff.mode",
+  childEventId: "raindrop.handoff.childEventId",
+  parentEventId: "raindrop.handoff.parentEventId",
+  parentSpanId: "raindrop.handoff.parentSpanId",
+  name: "raindrop.handoff.name",
+  terminal: "raindrop.handoff.terminal",
+  agentRole: "raindrop.agent.role"
+};
+var HANDOFF_TERMINAL_PROPERTY_KEY = HANDOFF_ATTRIBUTE_SUFFIXES.terminal;
+var TRACEPARENT_HEADER = "traceparent";
+var BAGGAGE_HEADER = "baggage";
+var HANDOFF_HEADER = "x-raindrop-handoff";
+var UNRESOLVED_CARRIER_WARNING = `[raindrop] resume: headers were supplied but no hand-off carrier was found on them (searched ${HANDOFF_HEADER}, ${TRACEPARENT_HEADER}, ${BAGGAGE_HEADER}). This run reports as an UNLINKED event, and the launcher that dispatched it will stay on "queued" forever because nothing ever references its child. Passing headers means a parent handed off, so no carrier on them is almost certainly a defect: a gateway or proxy stripping ${HANDOFF_HEADER}, a middleware overwriting ${BAGGAGE_HEADER}, a request forwarded without its headers, or a hand-built carrier whose field names do not match. Send every header the launcher's dispatch returned. Logged once per process; pass no headers for a directly-invoked sub-agent to silence it.`;
 function runWithTracingSuppressed(fn) {
   const hook = globalThis.RAINDROP_SUPPRESS_TRACING;
   if (typeof hook !== "function") return fn();
@@ -418,19 +432,52 @@ function projectIdHeaders(projectId) {
 var SHUTDOWN_DEADLINE_MS = 1e4;
 var POST_SHUTDOWN_TIMEOUT_MS = 5e3;
 function mergePatches(target, source) {
-  var _a, _b, _c, _d, _e, _f;
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i;
   const out = { ...target, ...source };
   if (target.properties || source.properties) {
     out.properties = { ...(_a = target.properties) != null ? _a : {}, ...(_b = source.properties) != null ? _b : {} };
   }
+  if (target.usage || source.usage) {
+    out.usage = { ...(_c = target.usage) != null ? _c : {} };
+    if (((_d = source.usage) == null ? void 0 : _d.promptTokens) !== void 0) {
+      out.usage.promptTokens = source.usage.promptTokens;
+    }
+    if (((_e = source.usage) == null ? void 0 : _e.completionTokens) !== void 0) {
+      out.usage.completionTokens = source.usage.completionTokens;
+    }
+  }
+  if (source.error === void 0) {
+    out.error = target.error;
+  }
   if (target.featureFlags || source.featureFlags) {
-    out.featureFlags = { ...(_c = target.featureFlags) != null ? _c : {}, ...(_d = source.featureFlags) != null ? _d : {} };
+    out.featureFlags = { ...(_f = target.featureFlags) != null ? _f : {}, ...(_g = source.featureFlags) != null ? _g : {} };
   }
   if (target.attachments || source.attachments) {
-    out.attachments = [...(_e = target.attachments) != null ? _e : [], ...(_f = source.attachments) != null ? _f : []];
+    out.attachments = [...(_h = target.attachments) != null ? _h : [], ...(_i = source.attachments) != null ? _i : []];
   }
   return out;
 }
+function isErrorLike(value) {
+  return value !== null && typeof value === "object";
+}
+function eventTelemetryProperties(patch) {
+  var _a, _b;
+  const properties = {};
+  if (((_a = patch.usage) == null ? void 0 : _a.promptTokens) !== void 0 && Number.isFinite(patch.usage.promptTokens)) {
+    properties["ai.usage.prompt_tokens"] = patch.usage.promptTokens;
+  }
+  if (((_b = patch.usage) == null ? void 0 : _b.completionTokens) !== void 0 && Number.isFinite(patch.usage.completionTokens)) {
+    properties["ai.usage.completion_tokens"] = patch.usage.completionTokens;
+  }
+  if (patch.error !== void 0 && patch.error !== null) {
+    const constructorName = patch.error instanceof Error && patch.error.constructor !== Error ? patch.error.constructor.name : void 0;
+    const errorType = patch.error instanceof Error ? patch.error.name && patch.error.name !== "Error" ? patch.error.name : constructorName || patch.error.name || "Error" : isErrorLike(patch.error) && typeof patch.error.name === "string" && patch.error.name.length > 0 ? patch.error.name : "Error";
+    const errorMessage = patch.error instanceof Error ? patch.error.message : isErrorLike(patch.error) && typeof patch.error.message === "string" ? patch.error.message : String(patch.error);
+    properties["error.type"] = errorType;
+    properties["error.message"] = errorMessage;
+  }
+  return properties;
+}
 var EventShipper = class {
   constructor(opts) {
     this.buffers = /* @__PURE__ */ new Map();
@@ -679,6 +726,11 @@ var EventShipper = class {
       return;
     }
     const { wizardSession, ...restProperties } = (_e = accumulated.properties) != null ? _e : {};
+    let telemetryProperties = {};
+    try {
+      telemetryProperties = eventTelemetryProperties(accumulated);
+    } catch (e) {
+    }
     const convoId = (_f = accumulated.convoId) != null ? _f : sticky.convoId;
     const isPending = (_h = (_g = accumulated.isPending) != null ? _g : sticky.isPending) != null ? _h : true;
     const payload = {
@@ -688,12 +740,13 @@ var EventShipper = class {
       timestamp: (_i = accumulated.timestamp) != null ? _i : (/* @__PURE__ */ new Date()).toISOString(),
       ai_data: {
         input: accumulated.input,
-        output: accumulated.output,
+        output: accumulated.error !== void 0 && accumulated.error !== null ? void 0 : accumulated.output,
         model: accumulated.model,
         convo_id: convoId
       },
       properties: {
         ...restProperties,
+        ...telemetryProperties,
         ...wizardSession ? { "raindrop.wizardSession": wizardSession } : {},
         $context: this.context
       },
@@ -762,6 +815,194 @@ var EventShipper = class {
     }
   }
 };
+var MODEL_USAGE_ATTRIBUTES = {
+  providerName: "gen_ai.provider.name",
+  requestModel: "gen_ai.request.model",
+  responseModel: "gen_ai.response.model",
+  inputTokens: "gen_ai.usage.input_tokens",
+  outputTokens: "gen_ai.usage.output_tokens",
+  reasoningTokens: "gen_ai.usage.reasoning_tokens",
+  cacheReadInputTokens: "gen_ai.usage.cache_read_input_tokens",
+  cacheWriteInputTokens: "gen_ai.usage.cache_write_input_tokens",
+  nonCachedInputTokens: "raindrop.usage.input_tokens.non_cached",
+  nonReasoningOutputTokens: "raindrop.usage.output_tokens.non_reasoning",
+  totalOutputTokens: "raindrop.usage.output_tokens.total",
+  reasoningOutputTokens: "raindrop.usage.output_tokens.reasoning",
+  cacheReadTokens: "raindrop.usage.input_tokens.cache_read",
+  cacheWriteTokens: "raindrop.usage.input_tokens.cache_write"
+};
+var MODEL_PROVIDER_NAMES = [
+  ["google.vertex", "google-vertex"],
+  ["vertex.anthropic", "google-vertex"],
+  ["amazon-bedrock", "amazon-bedrock"],
+  ["bedrock", "amazon-bedrock"],
+  ["anthropic", "anthropic"],
+  ["openai", "openai"],
+  ["google", "google"],
+  ["azure", "azure"],
+  ["openrouter", "openrouter"]
+];
+function canonicalModelProvider(provider) {
+  var _a;
+  const normalizedProvider = provider.trim().toLowerCase();
+  if (normalizedProvider === "gateway") return "vercel";
+  const match = MODEL_PROVIDER_NAMES.find(
+    ([family]) => normalizedProvider === family || normalizedProvider.startsWith(`${family}.`)
+  );
+  return (_a = match == null ? void 0 : match[1]) != null ? _a : provider;
+}
+function tokenCount(value) {
+  return typeof value === "number" && Number.isSafeInteger(value) && value >= 0 ? value : void 0;
+}
+function exclusiveTokens({
+  explicit,
+  total,
+  parts,
+  semantics
+}) {
+  if (explicit !== void 0) return explicit;
+  if (total === void 0) return void 0;
+  if (semantics === "exclusive") return total;
+  const observedParts = parts.filter((part) => part !== void 0 && part > 0);
+  if (semantics === void 0 && observedParts.length > 0) return void 0;
+  const exclusive = total - observedParts.reduce((sum, part) => sum + part, 0);
+  return exclusive >= 0 ? exclusive : void 0;
+}
+function buildModelUsageAttributes(usage) {
+  const inputTokens = tokenCount(usage.inputTokens);
+  const outputTokens = tokenCount(usage.outputTokens);
+  const explicitNonCachedInputTokens = tokenCount(usage.nonCachedInputTokens);
+  const explicitNonReasoningOutputTokens = tokenCount(usage.nonReasoningOutputTokens);
+  const reasoningTokens = tokenCount(usage.reasoningTokens);
+  const cacheReadInputTokens = tokenCount(usage.cacheReadInputTokens);
+  const cacheWriteInputTokens = tokenCount(usage.cacheWriteInputTokens);
+  const nonCachedInputTokens = exclusiveTokens({
+    explicit: explicitNonCachedInputTokens,
+    total: inputTokens,
+    parts: [cacheReadInputTokens, cacheWriteInputTokens],
+    semantics: usage.inputTokenSemantics
+  });
+  const nonReasoningOutputTokens = exclusiveTokens({
+    explicit: explicitNonReasoningOutputTokens,
+    total: outputTokens,
+    parts: [reasoningTokens],
+    semantics: usage.outputTokenSemantics
+  });
+  const exclusivePoolAttributes = nonCachedInputTokens !== void 0 && nonReasoningOutputTokens !== void 0 ? [
+    // backend consumes this complete raindrop.* set as one atomic billing-pool contract.
+    attrInt(MODEL_USAGE_ATTRIBUTES.nonCachedInputTokens, nonCachedInputTokens),
+    attrInt(MODEL_USAGE_ATTRIBUTES.nonReasoningOutputTokens, nonReasoningOutputTokens),
+    attrInt(MODEL_USAGE_ATTRIBUTES.reasoningOutputTokens, reasoningTokens),
+    attrInt(MODEL_USAGE_ATTRIBUTES.cacheReadTokens, cacheReadInputTokens),
+    attrInt(MODEL_USAGE_ATTRIBUTES.cacheWriteTokens, cacheWriteInputTokens)
+  ] : [];
+  return [
+    attrInt(MODEL_USAGE_ATTRIBUTES.inputTokens, inputTokens),
+    attrInt(MODEL_USAGE_ATTRIBUTES.outputTokens, outputTokens),
+    attrInt(MODEL_USAGE_ATTRIBUTES.reasoningTokens, reasoningTokens),
+    attrInt(MODEL_USAGE_ATTRIBUTES.cacheReadInputTokens, cacheReadInputTokens),
+    attrInt(MODEL_USAGE_ATTRIBUTES.cacheWriteInputTokens, cacheWriteInputTokens),
+    ...exclusivePoolAttributes
+  ].filter((attribute) => attribute !== void 0);
+}
+var STRING_ALIASES = {
+  [MODEL_USAGE_ATTRIBUTES.requestModel]: ["ai.model.id", "ai.model"]
+};
+var NUMBER_ALIASES = {
+  [MODEL_USAGE_ATTRIBUTES.inputTokens]: [
+    "gen_ai.usage.prompt_tokens",
+    "ai.usage.prompt_tokens",
+    "ai.usage.promptTokens",
+    "ai.usage.input_tokens",
+    "ai.usage.inputTokens"
+  ],
+  [MODEL_USAGE_ATTRIBUTES.outputTokens]: [
+    "gen_ai.usage.completion_tokens",
+    "ai.usage.completion_tokens",
+    "ai.usage.completionTokens",
+    "ai.usage.output_tokens",
+    "ai.usage.outputTokens"
+  ],
+  [MODEL_USAGE_ATTRIBUTES.reasoningTokens]: [
+    "gen_ai.usage.reasoning.output_tokens",
+    "ai.usage.reasoningTokens",
+    "ai.usage.thoughts_tokens"
+  ],
+  [MODEL_USAGE_ATTRIBUTES.cacheReadInputTokens]: [
+    "gen_ai.usage.cache_read_tokens",
+    "gen_ai.usage.cache_read.input_tokens",
+    "ai.usage.cachedInputTokens",
+    "ai.usage.cached_tokens",
+    "ai.usage.cache_read_tokens",
+    "ai.usage.cache_read_input_tokens",
+    "ai.usage.cacheReadInputTokens"
+  ],
+  [MODEL_USAGE_ATTRIBUTES.cacheWriteInputTokens]: [
+    "gen_ai.usage.cache_creation_input_tokens",
+    "gen_ai.usage.cache_creation.input_tokens",
+    "ai.usage.cacheWriteInputTokens",
+    "ai.usage.cache_creation_input_tokens",
+    "ai.usage.cacheCreationInputTokens"
+  ]
+};
+function hasAttribute(attributes, key) {
+  return attributes.some((attribute) => attribute.key === key);
+}
+function findStringAttribute(attributes, keys) {
+  return attributes.find(
+    (attribute) => keys.includes(attribute.key) && typeof attribute.value.stringValue === "string" && attribute.value.stringValue.length > 0
+  );
+}
+function findNumberAttribute(attributes, keys) {
+  return attributes.find((attribute) => {
+    if (!keys.includes(attribute.key)) return false;
+    if (typeof attribute.value.doubleValue === "number" && Number.isSafeInteger(attribute.value.doubleValue) && attribute.value.doubleValue >= 0) {
+      return true;
+    }
+    const intValue = attribute.value.intValue;
+    return integerValue(intValue) !== void 0;
+  });
+}
+function integerValue(value) {
+  if (typeof value === "number") {
+    return Number.isSafeInteger(value) && value >= 0 ? value : void 0;
+  }
+  if (typeof value !== "string") return void 0;
+  const parsed = Number(value);
+  return Number.isSafeInteger(parsed) && parsed >= 0 ? parsed : void 0;
+}
+function appendAlias(attributes, target, source) {
+  if (!source || hasAttribute(attributes, target)) return attributes;
+  return [...attributes, { key: target, value: { ...source.value } }];
+}
+function appendProviderAlias(attributes) {
+  if (hasAttribute(attributes, MODEL_USAGE_ATTRIBUTES.providerName)) return attributes;
+  const source = findStringAttribute(attributes, ["gen_ai.system", "ai.model.provider"]);
+  if (!(source == null ? void 0 : source.value.stringValue)) return attributes;
+  const attribute = attrString(
+    MODEL_USAGE_ATTRIBUTES.providerName,
+    canonicalModelProvider(source.value.stringValue)
+  );
+  return attribute ? [...attributes, attribute] : attributes;
+}
+function normalizeModelUsageSpan(span) {
+  const original = span.attributes;
+  if (!original || original.length === 0) return span;
+  const responseModel = findStringAttribute(original, [
+    MODEL_USAGE_ATTRIBUTES.responseModel,
+    "ai.response.model"
+  ]);
+  if (!responseModel) return span;
+  let attributes = appendAlias(original, MODEL_USAGE_ATTRIBUTES.responseModel, responseModel);
+  attributes = appendProviderAlias(attributes);
+  for (const [target, aliases] of Object.entries(STRING_ALIASES)) {
+    attributes = appendAlias(attributes, target, findStringAttribute(attributes, aliases));
+  }
+  for (const [target, aliases] of Object.entries(NUMBER_ALIASES)) {
+    attributes = appendAlias(attributes, target, findNumberAttribute(attributes, aliases));
+  }
+  return attributes === original ? span : { ...span, attributes };
+}
 var DEFAULT_SECRET_KEY_NAMES = [
   "apikey",
   "apisecret",
@@ -851,6 +1092,13 @@ function redactJsonAttributeValue(key, value) {
   if (scrubbedJson === json) return void 0;
   return { stringValue: scrubbedJson };
 }
+function identityAttrs(args) {
+  return [
+    attrString("ai.telemetry.metadata.raindrop.ai.userId", args.userId),
+    attrString("ai.telemetry.metadata.raindrop.convoId", args.convoId),
+    attrString("ai.telemetry.metadata.raindrop.eventName", args.eventName)
+  ];
+}
 function applyOtelSpanAttributeLimit(limit) {
   var _a, _b;
   try {
@@ -955,6 +1203,10 @@ var TraceShipper = class {
         return null;
       }
     }
+    try {
+      current = normalizeModelUsageSpan(current);
+    } catch (e) {
+    }
     if (!this.disableDefaultRedaction) {
       current = defaultTransformSpan(current);
     }
@@ -975,6 +1227,7 @@ var TraceShipper = class {
     const started = (_a = args.startTimeUnixNano) != null ? _a : nowUnixNanoString();
     const attrs = [
       attrString("ai.telemetry.metadata.raindrop.eventId", args.eventId),
+      ...identityAttrs(args),
       attrString("ai.operationId", args.operationId)
     ];
     if ((_b = args.attributes) == null ? void 0 : _b.length) attrs.push(...args.attributes);
@@ -1030,7 +1283,8 @@ var TraceShipper = class {
     var _a;
     const ids = createSpanIds(args.parent);
     const attrs = [
-      attrString("ai.telemetry.metadata.raindrop.eventId", args.eventId)
+      attrString("ai.telemetry.metadata.raindrop.eventId", args.eventId),
+      ...identityAttrs(args)
     ];
     if ((_a = args.attributes) == null ? void 0 : _a.length) attrs.push(...args.attributes);
     const otlp = buildOtlpSpan({
@@ -1253,7 +1507,7 @@ function resolveLocalWorkshopUrl(fileValue) {
 // package.json
 var package_default = {
   name: "@raindrop-ai/pi-agent",
-  version: "0.0.10",
+  version: "0.1.1",
   description: "Raindrop observability for Pi Agent \u2014 automatic tracing via subscriber or pi-coding-agent extension",
   type: "module",
   license: "MIT",
@@ -1379,6 +1633,18 @@ var TraceShipper2 = class extends TraceShipper {
 
 // src/internal/helpers.ts
 var import_node_os2 = require("os");
+function modelSpendSpanAttributes(input) {
+  const providerAttr = input.provider ? attrString("gen_ai.provider.name", canonicalModelProvider(input.provider)) : void 0;
+  const usageAttrs = input.usage ? buildModelUsageAttributes({
+    inputTokens: input.usage.input,
+    outputTokens: input.usage.output,
+    cacheReadInputTokens: input.usage.cacheRead,
+    cacheWriteInputTokens: input.usage.cacheWrite,
+    reasoningTokens: 0,
+    inputTokenSemantics: "exclusive"
+  }) : [];
+  return providerAttr ? [providerAttr, ...usageAttrs] : usageAttrs;
+}
 function extractAssistantText(message) {
   if (!("role" in message) || message.role !== "assistant") return void 0;
   if (!Array.isArray(message.content)) return void 0;
@@ -1578,7 +1844,9 @@ function getAssistantError(message) {
   var _a;
   if (message.role !== "assistant") return void 0;
   if (message.stopReason !== "error" && message.stopReason !== "aborted") return void 0;
-  return (_a = message.errorMessage) != null ? _a : `Assistant ${message.stopReason}`;
+  const error = new Error((_a = message.errorMessage) != null ? _a : `Assistant ${message.stopReason}`);
+  error.name = message.stopReason === "aborted" ? "AbortError" : "PiAgentError";
+  return error;
 }
 function getState(stateRef, ctx) {
   const sessionId = ctx.sessionManager.getSessionId();
@@ -1705,7 +1973,7 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
     }
   });
   pi.on("message_end", async (event, ctx) => {
-    var _a, _b, _c, _d, _e, _f, _g, _h;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
     try {
       if (((_a = event.message) == null ? void 0 : _a.role) !== "assistant") return;
       const state = getState(stateRef, ctx);
@@ -1719,19 +1987,26 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
       const outputText = capText2(getAssistantText(message));
       const inputTokens = (_d = message.usage) == null ? void 0 : _d.input;
       const outputTokens = (_e = message.usage) == null ? void 0 : _e.output;
-      const cacheReadTokens = (_f = message.usage) == null ? void 0 : _f.cacheRead;
+      const cacheReadTokens = typeof ((_f = message.usage) == null ? void 0 : _f.cacheRead) === "number" ? message.usage.cacheRead : 0;
+      const cacheWriteTokens = typeof ((_g = message.usage) == null ? void 0 : _g.cacheWrite) === "number" ? message.usage.cacheWrite : 0;
       if (typeof inputTokens === "number") state.totalInputTokens += inputTokens;
       if (typeof outputTokens === "number") state.totalOutputTokens += outputTokens;
-      if (typeof cacheReadTokens === "number" && cacheReadTokens > 0) state.totalCacheReadTokens += cacheReadTokens;
-      const turnParent = (_g = state.currentTurnSpan) != null ? _g : state.currentRootSpan;
+      if (cacheReadTokens > 0) state.totalCacheReadTokens += cacheReadTokens;
+      const turnParent = (_h = state.currentTurnSpan) != null ? _h : state.currentRootSpan;
       const llmAttrs = [
         attrString("ai.operationId", "generateText"),
         provider ? attrString("gen_ai.system", provider) : void 0,
         modelId ? attrString("gen_ai.request.model", modelId) : void 0,
         modelId ? attrString("gen_ai.response.model", modelId) : void 0,
-        typeof inputTokens === "number" ? attrInt("gen_ai.usage.input_tokens", inputTokens) : void 0,
-        typeof outputTokens === "number" ? attrInt("gen_ai.usage.output_tokens", outputTokens) : void 0,
-        typeof cacheReadTokens === "number" && cacheReadTokens > 0 ? attrInt("gen_ai.usage.cache_read_tokens", cacheReadTokens) : void 0,
+        ...modelSpendSpanAttributes({
+          provider: provider || void 0,
+          usage: typeof inputTokens === "number" && typeof outputTokens === "number" ? {
+            input: inputTokens,
+            output: outputTokens,
+            cacheRead: cacheReadTokens,
+            cacheWrite: cacheWriteTokens
+          } : void 0
+        }),
         outputText ? attrString("ai.response.text", truncate(outputText)) : void 0,
         state.currentInput ? attrString("ai.prompt", truncate(state.currentInput)) : void 0,
         message.stopReason ? attrString("ai.stop_reason", message.stopReason) : void 0,
@@ -1769,16 +2044,19 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
         ...errorForSpan ? { error: errorForSpan } : {}
       });
       state.currentRootSpan = void 0;
-      await eventShipper.finish((_h = state.currentEventRequestId) != null ? _h : state.currentEventId, {
+      await eventShipper.finish((_i = state.currentEventRequestId) != null ? _i : state.currentEventId, {
         userId: getUserId(state, config.eventMetadata),
         ...modelId ? { model: modelId } : {},
-        ...outputText.trim() ? { output: outputText } : {},
+        ...!errorForSpan && outputText.trim() ? { output: outputText } : {},
+        usage: {
+          promptTokens: state.totalInputTokens > 0 ? state.totalInputTokens : void 0,
+          completionTokens: state.totalOutputTokens > 0 ? state.totalOutputTokens : void 0
+        },
+        error: errorForSpan,
         properties: {
           sdk_version: libraryVersion,
           stop_reason: message.stopReason,
           ...provider ? { "ai.provider": provider } : {},
-          ...state.totalInputTokens > 0 ? { "ai.usage.prompt_tokens": state.totalInputTokens } : {},
-          ...state.totalOutputTokens > 0 ? { "ai.usage.completion_tokens": state.totalOutputTokens } : {},
           ...state.totalCacheReadTokens > 0 ? { "ai.usage.cache_read_tokens": state.totalCacheReadTokens } : {}
         }
       });

package/dist/extension.d.cts

@@ -1,5 +1,5 @@
 import { ExtensionAPI } from '@earendil-works/pi-coding-agent';
-import { E as EventShipper$1, T as TraceShipper$1, O as OtlpSpan } from './index.d-DBLTz2Nz.cjs';
+import { E as EventShipper$1, T as TraceShipper$1, O as OtlpSpan } from './index.d-DPYV95P7.cjs';
 
 interface EventMetadata {
     userId?: string;

package/dist/extension.d.ts

@@ -1,5 +1,5 @@
 import { ExtensionAPI } from '@earendil-works/pi-coding-agent';
-import { E as EventShipper$1, T as TraceShipper$1, O as OtlpSpan } from './index.d-DBLTz2Nz.js';
+import { E as EventShipper$1, T as TraceShipper$1, O as OtlpSpan } from './index.d-DPYV95P7.js';
 
 interface EventMetadata {
     userId?: string;

package/dist/extension.js

@@ -10,11 +10,12 @@ import {
   getHostname,
   getUsername,
   libraryVersion,
+  modelSpendSpanAttributes,
   nowUnixNanoString,
   resolveLocalDebuggerBaseUrl,
   safeStringify,
   truncate
-} from "./chunk-GXL5WZE5.js";
+} from "./chunk-GTKXT2J5.js";
 
 // src/internal/config.ts
 import { existsSync, readFileSync } from "fs";
@@ -141,7 +142,9 @@ function getAssistantError(message) {
   var _a;
   if (message.role !== "assistant") return void 0;
   if (message.stopReason !== "error" && message.stopReason !== "aborted") return void 0;
-  return (_a = message.errorMessage) != null ? _a : `Assistant ${message.stopReason}`;
+  const error = new Error((_a = message.errorMessage) != null ? _a : `Assistant ${message.stopReason}`);
+  error.name = message.stopReason === "aborted" ? "AbortError" : "PiAgentError";
+  return error;
 }
 function getState(stateRef, ctx) {
   const sessionId = ctx.sessionManager.getSessionId();
@@ -268,7 +271,7 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
     }
   });
   pi.on("message_end", async (event, ctx) => {
-    var _a, _b, _c, _d, _e, _f, _g, _h;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
     try {
       if (((_a = event.message) == null ? void 0 : _a.role) !== "assistant") return;
       const state = getState(stateRef, ctx);
@@ -282,19 +285,26 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
       const outputText = capText(getAssistantText(message));
       const inputTokens = (_d = message.usage) == null ? void 0 : _d.input;
       const outputTokens = (_e = message.usage) == null ? void 0 : _e.output;
-      const cacheReadTokens = (_f = message.usage) == null ? void 0 : _f.cacheRead;
+      const cacheReadTokens = typeof ((_f = message.usage) == null ? void 0 : _f.cacheRead) === "number" ? message.usage.cacheRead : 0;
+      const cacheWriteTokens = typeof ((_g = message.usage) == null ? void 0 : _g.cacheWrite) === "number" ? message.usage.cacheWrite : 0;
       if (typeof inputTokens === "number") state.totalInputTokens += inputTokens;
       if (typeof outputTokens === "number") state.totalOutputTokens += outputTokens;
-      if (typeof cacheReadTokens === "number" && cacheReadTokens > 0) state.totalCacheReadTokens += cacheReadTokens;
-      const turnParent = (_g = state.currentTurnSpan) != null ? _g : state.currentRootSpan;
+      if (cacheReadTokens > 0) state.totalCacheReadTokens += cacheReadTokens;
+      const turnParent = (_h = state.currentTurnSpan) != null ? _h : state.currentRootSpan;
       const llmAttrs = [
         attrString("ai.operationId", "generateText"),
         provider ? attrString("gen_ai.system", provider) : void 0,
         modelId ? attrString("gen_ai.request.model", modelId) : void 0,
         modelId ? attrString("gen_ai.response.model", modelId) : void 0,
-        typeof inputTokens === "number" ? attrInt("gen_ai.usage.input_tokens", inputTokens) : void 0,
-        typeof outputTokens === "number" ? attrInt("gen_ai.usage.output_tokens", outputTokens) : void 0,
-        typeof cacheReadTokens === "number" && cacheReadTokens > 0 ? attrInt("gen_ai.usage.cache_read_tokens", cacheReadTokens) : void 0,
+        ...modelSpendSpanAttributes({
+          provider: provider || void 0,
+          usage: typeof inputTokens === "number" && typeof outputTokens === "number" ? {
+            input: inputTokens,
+            output: outputTokens,
+            cacheRead: cacheReadTokens,
+            cacheWrite: cacheWriteTokens
+          } : void 0
+        }),
         outputText ? attrString("ai.response.text", truncate(outputText)) : void 0,
         state.currentInput ? attrString("ai.prompt", truncate(state.currentInput)) : void 0,
         message.stopReason ? attrString("ai.stop_reason", message.stopReason) : void 0,
@@ -332,16 +342,19 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
         ...errorForSpan ? { error: errorForSpan } : {}
       });
       state.currentRootSpan = void 0;
-      await eventShipper.finish((_h = state.currentEventRequestId) != null ? _h : state.currentEventId, {
+      await eventShipper.finish((_i = state.currentEventRequestId) != null ? _i : state.currentEventId, {
         userId: getUserId(state, config.eventMetadata),
         ...modelId ? { model: modelId } : {},
-        ...outputText.trim() ? { output: outputText } : {},
+        ...!errorForSpan && outputText.trim() ? { output: outputText } : {},
+        usage: {
+          promptTokens: state.totalInputTokens > 0 ? state.totalInputTokens : void 0,
+          completionTokens: state.totalOutputTokens > 0 ? state.totalOutputTokens : void 0
+        },
+        error: errorForSpan,
         properties: {
           sdk_version: libraryVersion,
           stop_reason: message.stopReason,
           ...provider ? { "ai.provider": provider } : {},
-          ...state.totalInputTokens > 0 ? { "ai.usage.prompt_tokens": state.totalInputTokens } : {},
-          ...state.totalOutputTokens > 0 ? { "ai.usage.completion_tokens": state.totalOutputTokens } : {},
           ...state.totalCacheReadTokens > 0 ? { "ai.usage.cache_read_tokens": state.totalCacheReadTokens } : {}
         }
       });