@raindrop-ai/pi-agent 0.0.10 → 0.1.0

package/dist/{chunk-GXL5WZE5.js → chunk-XQJ6G4YA.js}

@@ -1,4 +1,4 @@
-// ../core/dist/chunk-267IXHO5.js
+// ../core/dist/chunk-QTUH3BHZ.js
 function getCrypto() {
   const c = globalThis.crypto;
   return c;
@@ -391,19 +391,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();
@@ -652,6 +685,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 = {
@@ -661,12 +699,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
       },
@@ -824,6 +863,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 {
@@ -948,6 +994,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);
@@ -1003,7 +1050,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({
@@ -1171,7 +1219,7 @@ installTracingSuppressionHook();
 // package.json
 var package_default = {
   name: "@raindrop-ai/pi-agent",
-  version: "0.0.10",
+  version: "0.1.0",
   description: "Raindrop observability for Pi Agent \u2014 automatic tracing via subscriber or pi-coding-agent extension",
   type: "module",
   license: "MIT",

package/dist/extension.cjs

@@ -25,7 +25,7 @@ __export(extension_exports, {
 });
 module.exports = __toCommonJS(extension_exports);
 
-// ../core/dist/chunk-267IXHO5.js
+// ../core/dist/chunk-QTUH3BHZ.js
 function getCrypto() {
   const c = globalThis.crypto;
   return c;
@@ -418,19 +418,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 +712,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 +726,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
       },
@@ -851,6 +890,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 {
@@ -975,6 +1021,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 +1077,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 +1301,7 @@ function resolveLocalWorkshopUrl(fileValue) {
 // package.json
 var package_default = {
   name: "@raindrop-ai/pi-agent",
-  version: "0.0.10",
+  version: "0.1.0",
   description: "Raindrop observability for Pi Agent \u2014 automatic tracing via subscriber or pi-coding-agent extension",
   type: "module",
   license: "MIT",
@@ -1578,7 +1626,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();
@@ -1772,13 +1822,16 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
       await eventShipper.finish((_h = state.currentEventRequestId) != null ? _h : 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

@@ -14,7 +14,7 @@ import {
   resolveLocalDebuggerBaseUrl,
   safeStringify,
   truncate
-} from "./chunk-GXL5WZE5.js";
+} from "./chunk-XQJ6G4YA.js";
 
 // src/internal/config.ts
 import { existsSync, readFileSync } from "fs";
@@ -141,7 +141,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();
@@ -335,13 +337,16 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
       await eventShipper.finish((_h = state.currentEventRequestId) != null ? _h : 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/index.cjs

@@ -24,7 +24,7 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 
-// ../core/dist/chunk-267IXHO5.js
+// ../core/dist/chunk-QTUH3BHZ.js
 function getCrypto() {
   const c = globalThis.crypto;
   return c;
@@ -414,19 +414,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();
@@ -675,6 +708,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 = {
@@ -684,12 +722,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
       },
@@ -847,6 +886,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 {
@@ -971,6 +1017,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);
@@ -1026,7 +1073,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({
@@ -1194,7 +1242,7 @@ installTracingSuppressionHook();
 // package.json
 var package_default = {
   name: "@raindrop-ai/pi-agent",
-  version: "0.0.10",
+  version: "0.1.0",
   description: "Raindrop observability for Pi Agent \u2014 automatic tracing via subscriber or pi-coding-agent extension",
   type: "module",
   license: "MIT",
@@ -1493,14 +1541,24 @@ function getUsername() {
 }
 
 // src/internal/subscriber.ts
+function resolveEventId(source) {
+  if (typeof source !== "function") return randomUUID();
+  try {
+    const id = source();
+    return typeof id === "string" && id.trim() !== "" ? id : randomUUID();
+  } catch (e) {
+    return randomUUID();
+  }
+}
 function createSubscriber(agent, eventShipper, traceShipper, defaultOptions, options, debug) {
-  var _a, _b, _c, _d, _e;
+  var _a, _b, _c, _d, _e, _f;
   const userId = (_a = options.userId) != null ? _a : defaultOptions.userId;
   const convoId = (_b = options.convoId) != null ? _b : defaultOptions.convoId;
   const eventName = (_c = options.eventName) != null ? _c : defaultOptions.eventName;
+  const eventIdSource = (_d = options.eventId) != null ? _d : defaultOptions.eventId;
   const properties = {
-    ...(_d = defaultOptions.properties) != null ? _d : {},
-    ...(_e = options.properties) != null ? _e : {}
+    ...(_e = defaultOptions.properties) != null ? _e : {},
+    ...(_f = options.properties) != null ? _f : {}
   };
   let currentRun;
   function log(msg) {
@@ -1531,7 +1589,7 @@ function createSubscriber(agent, eventShipper, traceShipper, defaultOptions, opt
         }
         currentRun = void 0;
       }
-      const eventId = randomUUID();
+      const eventId = resolveEventId(eventIdSource);
       const rootSpan = traceShipper ? traceShipper.startSpan({
         name: "ai.event",
         eventId,
@@ -1643,6 +1701,13 @@ function createSubscriber(agent, eventShipper, traceShipper, defaultOptions, opt
         attributes: llmAttrs
       });
       const errorForSpan = stopReason === "error" || stopReason === "aborted" ? "errorMessage" in message && typeof message.errorMessage === "string" ? message.errorMessage : `Assistant ${stopReason}` : void 0;
+      if (errorForSpan) {
+        const assistantError = new Error(errorForSpan);
+        assistantError.name = stopReason === "aborted" ? "AbortError" : "PiAgentError";
+        currentRun.error = assistantError;
+      } else {
+        currentRun.error = void 0;
+      }
       traceShipper.endSpan(llmSpan, errorForSpan ? { error: errorForSpan } : void 0);
       const toolCallIds = extractToolCallIds(message);
       for (const tcId of toolCallIds) {
@@ -1787,13 +1852,16 @@ function createSubscriber(agent, eventShipper, traceShipper, defaultOptions, opt
         eventShipper.finish(run.eventId, {
           userId: userId != null ? userId : "anonymous",
           model: run.lastModel || void 0,
-          output: outputText || void 0,
+          output: run.error ? void 0 : outputText || void 0,
+          usage: {
+            promptTokens: run.totalInputTokens > 0 ? run.totalInputTokens : void 0,
+            completionTokens: run.totalOutputTokens > 0 ? run.totalOutputTokens : void 0
+          },
+          error: run.error,
           properties: {
             ...properties,
             sdk_version: libraryVersion,
             ...run.lastProvider ? { "ai.provider": run.lastProvider } : {},
-            ...run.totalInputTokens > 0 ? { "ai.usage.prompt_tokens": run.totalInputTokens } : {},
-            ...run.totalOutputTokens > 0 ? { "ai.usage.completion_tokens": run.totalOutputTokens } : {},
             ...run.totalCacheReadTokens > 0 ? { "ai.usage.cache_read_tokens": run.totalCacheReadTokens } : {}
           }
         }).catch(() => {
@@ -1894,6 +1962,7 @@ function createRaindropPiAgent(opts) {
     userId: opts.userId,
     convoId: opts.convoId,
     eventName: opts.eventName,
+    eventId: opts.eventId,
     properties: opts.properties
   };
   return {

package/dist/{index.d-DBLTz2Nz.d.cts → index.d-DPYV95P7.d.cts}

@@ -45,6 +45,10 @@ type IdentifyInput = {
     userId: string;
     traits?: Record<string, unknown>;
 };
+type EventUsage = {
+    promptTokens?: number;
+    completionTokens?: number;
+};
 type Patch = {
     eventName?: string;
     userId?: string;
@@ -52,6 +56,8 @@ type Patch = {
     input?: string;
     output?: string;
     model?: string;
+    usage?: EventUsage;
+    error?: unknown;
     properties?: Record<string, unknown>;
     featureFlags?: Record<string, string>;
     attachments?: Attachment[];
@@ -150,13 +156,7 @@ declare class EventShipper {
      */
     private requestOpts;
     patch(eventId: string, patch: Patch): Promise<void>;
-    finish(eventId: string, patch: {
-        output?: string;
-        model?: string;
-        properties?: Record<string, unknown>;
-        featureFlags?: Record<string, string>;
-        userId?: string;
-    }): Promise<void>;
+    finish(eventId: string, patch: Pick<Patch, "output" | "model" | "usage" | "error" | "properties" | "featureFlags" | "userId">): Promise<void>;
     flush(): Promise<void>;
     shutdown(): Promise<void>;
     trackSignal(signal: SignalInput): Promise<void>;
@@ -329,6 +329,9 @@ declare class TraceShipper {
             spanIdB64: string;
         };
         eventId: string;
+        userId?: string;
+        convoId?: string;
+        eventName?: string;
         operationId?: string;
         attributes?: Array<OtlpKeyValue | undefined>;
         startTimeUnixNano?: string;
@@ -347,6 +350,9 @@ declare class TraceShipper {
             spanIdB64: string;
         };
         eventId: string;
+        userId?: string;
+        convoId?: string;
+        eventName?: string;
         startTimeUnixNano: string;
         endTimeUnixNano: string;
         attributes?: Array<OtlpKeyValue | undefined>;

package/dist/{index.d-DBLTz2Nz.d.ts → index.d-DPYV95P7.d.ts}

@@ -45,6 +45,10 @@ type IdentifyInput = {
     userId: string;
     traits?: Record<string, unknown>;
 };
+type EventUsage = {
+    promptTokens?: number;
+    completionTokens?: number;
+};
 type Patch = {
     eventName?: string;
     userId?: string;
@@ -52,6 +56,8 @@ type Patch = {
     input?: string;
     output?: string;
     model?: string;
+    usage?: EventUsage;
+    error?: unknown;
     properties?: Record<string, unknown>;
     featureFlags?: Record<string, string>;
     attachments?: Attachment[];
@@ -150,13 +156,7 @@ declare class EventShipper {
      */
     private requestOpts;
     patch(eventId: string, patch: Patch): Promise<void>;
-    finish(eventId: string, patch: {
-        output?: string;
-        model?: string;
-        properties?: Record<string, unknown>;
-        featureFlags?: Record<string, string>;
-        userId?: string;
-    }): Promise<void>;
+    finish(eventId: string, patch: Pick<Patch, "output" | "model" | "usage" | "error" | "properties" | "featureFlags" | "userId">): Promise<void>;
     flush(): Promise<void>;
     shutdown(): Promise<void>;
     trackSignal(signal: SignalInput): Promise<void>;
@@ -329,6 +329,9 @@ declare class TraceShipper {
             spanIdB64: string;
         };
         eventId: string;
+        userId?: string;
+        convoId?: string;
+        eventName?: string;
         operationId?: string;
         attributes?: Array<OtlpKeyValue | undefined>;
         startTimeUnixNano?: string;
@@ -347,6 +350,9 @@ declare class TraceShipper {
             spanIdB64: string;
         };
         eventId: string;
+        userId?: string;
+        convoId?: string;
+        eventName?: string;
         startTimeUnixNano: string;
         endTimeUnixNano: string;
         attributes?: Array<OtlpKeyValue | undefined>;

package/dist/index.d.cts

@@ -1,5 +1,5 @@
 import { Agent } from '@earendil-works/pi-agent-core';
-import { A as Attachment } from './index.d-DBLTz2Nz.cjs';
+import { A as Attachment } from './index.d-DPYV95P7.cjs';
 
 /**
  * Options for the Raindrop Pi Agent client.
@@ -24,6 +24,15 @@ interface RaindropPiAgentOptions {
     convoId?: string;
     /** Default event name (default: "pi_agent_prompt") */
     eventName?: string;
+    /**
+     * Per-run event id source. Invoked once per agent run to mint that run's
+     * event id (reused for the run's Raindrop event and every linked span). A
+     * static string is intentionally not accepted: a long-lived client runs many
+     * agent runs, and a fixed id would collide across them. Absent / throwing /
+     * non-string / empty results fall back to a random UUID, so a caller's id
+     * source never crashes telemetry.
+     */
+    eventId?: () => string;
     /** Default properties attached to every event */
     properties?: Record<string, unknown>;
     /**
@@ -69,6 +78,8 @@ interface PiAgentSubscribeOptions {
     convoId?: string;
     /** Event name override */
     eventName?: string;
+    /** Per-run event id source override (see RaindropPiAgentOptions.eventId) */
+    eventId?: () => string;
     /** Additional properties */
     properties?: Record<string, unknown>;
 }

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 import { Agent } from '@earendil-works/pi-agent-core';
-import { A as Attachment } from './index.d-DBLTz2Nz.js';
+import { A as Attachment } from './index.d-DPYV95P7.js';
 
 /**
  * Options for the Raindrop Pi Agent client.
@@ -24,6 +24,15 @@ interface RaindropPiAgentOptions {
     convoId?: string;
     /** Default event name (default: "pi_agent_prompt") */
     eventName?: string;
+    /**
+     * Per-run event id source. Invoked once per agent run to mint that run's
+     * event id (reused for the run's Raindrop event and every linked span). A
+     * static string is intentionally not accepted: a long-lived client runs many
+     * agent runs, and a fixed id would collide across them. Absent / throwing /
+     * non-string / empty results fall back to a random UUID, so a caller's id
+     * source never crashes telemetry.
+     */
+    eventId?: () => string;
     /** Default properties attached to every event */
     properties?: Record<string, unknown>;
     /**
@@ -69,6 +78,8 @@ interface PiAgentSubscribeOptions {
     convoId?: string;
     /** Event name override */
     eventName?: string;
+    /** Per-run event id source override (see RaindropPiAgentOptions.eventId) */
+    eventId?: () => string;
     /** Additional properties */
     properties?: Record<string, unknown>;
 }

package/dist/index.js

@@ -17,17 +17,27 @@ import {
   resolveLocalDebuggerBaseUrl,
   safeStringify,
   truncate
-} from "./chunk-GXL5WZE5.js";
+} from "./chunk-XQJ6G4YA.js";
 
 // src/internal/subscriber.ts
+function resolveEventId(source) {
+  if (typeof source !== "function") return randomUUID();
+  try {
+    const id = source();
+    return typeof id === "string" && id.trim() !== "" ? id : randomUUID();
+  } catch (e) {
+    return randomUUID();
+  }
+}
 function createSubscriber(agent, eventShipper, traceShipper, defaultOptions, options, debug) {
-  var _a, _b, _c, _d, _e;
+  var _a, _b, _c, _d, _e, _f;
   const userId = (_a = options.userId) != null ? _a : defaultOptions.userId;
   const convoId = (_b = options.convoId) != null ? _b : defaultOptions.convoId;
   const eventName = (_c = options.eventName) != null ? _c : defaultOptions.eventName;
+  const eventIdSource = (_d = options.eventId) != null ? _d : defaultOptions.eventId;
   const properties = {
-    ...(_d = defaultOptions.properties) != null ? _d : {},
-    ...(_e = options.properties) != null ? _e : {}
+    ...(_e = defaultOptions.properties) != null ? _e : {},
+    ...(_f = options.properties) != null ? _f : {}
   };
   let currentRun;
   function log(msg) {
@@ -58,7 +68,7 @@ function createSubscriber(agent, eventShipper, traceShipper, defaultOptions, opt
         }
         currentRun = void 0;
       }
-      const eventId = randomUUID();
+      const eventId = resolveEventId(eventIdSource);
       const rootSpan = traceShipper ? traceShipper.startSpan({
         name: "ai.event",
         eventId,
@@ -170,6 +180,13 @@ function createSubscriber(agent, eventShipper, traceShipper, defaultOptions, opt
         attributes: llmAttrs
       });
       const errorForSpan = stopReason === "error" || stopReason === "aborted" ? "errorMessage" in message && typeof message.errorMessage === "string" ? message.errorMessage : `Assistant ${stopReason}` : void 0;
+      if (errorForSpan) {
+        const assistantError = new Error(errorForSpan);
+        assistantError.name = stopReason === "aborted" ? "AbortError" : "PiAgentError";
+        currentRun.error = assistantError;
+      } else {
+        currentRun.error = void 0;
+      }
       traceShipper.endSpan(llmSpan, errorForSpan ? { error: errorForSpan } : void 0);
       const toolCallIds = extractToolCallIds(message);
       for (const tcId of toolCallIds) {
@@ -314,13 +331,16 @@ function createSubscriber(agent, eventShipper, traceShipper, defaultOptions, opt
         eventShipper.finish(run.eventId, {
           userId: userId != null ? userId : "anonymous",
           model: run.lastModel || void 0,
-          output: outputText || void 0,
+          output: run.error ? void 0 : outputText || void 0,
+          usage: {
+            promptTokens: run.totalInputTokens > 0 ? run.totalInputTokens : void 0,
+            completionTokens: run.totalOutputTokens > 0 ? run.totalOutputTokens : void 0
+          },
+          error: run.error,
           properties: {
             ...properties,
             sdk_version: libraryVersion,
             ...run.lastProvider ? { "ai.provider": run.lastProvider } : {},
-            ...run.totalInputTokens > 0 ? { "ai.usage.prompt_tokens": run.totalInputTokens } : {},
-            ...run.totalOutputTokens > 0 ? { "ai.usage.completion_tokens": run.totalOutputTokens } : {},
             ...run.totalCacheReadTokens > 0 ? { "ai.usage.cache_read_tokens": run.totalCacheReadTokens } : {}
           }
         }).catch(() => {
@@ -421,6 +441,7 @@ function createRaindropPiAgent(opts) {
     userId: opts.userId,
     convoId: opts.convoId,
     eventName: opts.eventName,
+    eventId: opts.eventId,
     properties: opts.properties
   };
   return {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@raindrop-ai/pi-agent",
-  "version": "0.0.10",
+  "version": "0.1.0",
   "description": "Raindrop observability for Pi Agent — automatic tracing via subscriber or pi-coding-agent extension",
   "type": "module",
   "license": "MIT",
@@ -54,7 +54,7 @@
     "tsup": "^8.5.1",
     "typescript": "^5.7.3",
     "vitest": "^2.1.9",
-    "@raindrop-ai/core": "0.1.2"
+    "@raindrop-ai/core": "0.1.3"
   },
   "tsup": {
     "entry": [