@raindrop-ai/pi-agent 0.0.3 → 0.0.5

package/dist/extension.cjs

@@ -25,7 +25,7 @@ __export(extension_exports, {
 });
 module.exports = __toCommonJS(extension_exports);
 
-// ../core/dist/chunk-OLYXZCOK.js
+// ../core/dist/chunk-U5HUTMR5.js
 function getCrypto() {
   const c = globalThis.crypto;
   return c;
@@ -72,6 +72,8 @@ function base64Encode(bytes) {
 function generateId() {
   return base64Encode(randomBytes(12)).replace(/[+/=]/g, "").slice(0, 16);
 }
+var DEFAULT_REQUEST_TIMEOUT_MS = 3e4;
+var MAX_RETRY_DELAY_MS = 3e4;
 function wait(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
@@ -79,6 +81,46 @@ function formatEndpoint(endpoint) {
   if (!endpoint) return void 0;
   return endpoint.endsWith("/") ? endpoint : `${endpoint}/`;
 }
+function redactUrlForLog(url) {
+  try {
+    const parsed = new URL(url);
+    parsed.username = "";
+    parsed.password = "";
+    parsed.search = "";
+    parsed.hash = "";
+    return parsed.toString();
+  } catch (e) {
+    return "<unparseable-url>";
+  }
+}
+var RATE_LIMITED_LOG_INTERVAL_MS = 3e4;
+var rateLimitedLogLast = /* @__PURE__ */ new Map();
+function rateLimitedLog(key, log) {
+  const now = Date.now();
+  const last = rateLimitedLogLast.get(key);
+  if (last !== void 0 && now - last < RATE_LIMITED_LOG_INTERVAL_MS) {
+    return false;
+  }
+  rateLimitedLogLast.set(key, now);
+  log();
+  return true;
+}
+async function raceWithTimeout(promise, timeoutMs) {
+  let timer;
+  const settledInTime = await Promise.race([
+    promise.then(
+      () => true,
+      () => true
+    ),
+    new Promise((resolve) => {
+      var _a;
+      timer = setTimeout(() => resolve(false), Math.max(0, timeoutMs));
+      (_a = timer.unref) == null ? void 0 : _a.call(timer);
+    })
+  ]);
+  if (timer) clearTimeout(timer);
+  return settledInTime;
+}
 function parseRetryAfter(headers) {
   var _a;
   const value = (_a = headers.get("Retry-After")) != null ? _a : headers.get("retry-after");
@@ -95,12 +137,12 @@ function parseRetryAfter(headers) {
 function getRetryDelayMs(attemptNumber, previousError) {
   if (previousError && typeof previousError === "object" && previousError !== null && "retryAfterMs" in previousError) {
     const v = previousError.retryAfterMs;
-    if (typeof v === "number") return Math.max(0, v);
+    if (typeof v === "number") return Math.min(Math.max(0, v), MAX_RETRY_DELAY_MS);
   }
   if (attemptNumber <= 1) return 0;
   const base = 500;
   const factor = Math.pow(2, attemptNumber - 2);
-  return base * factor;
+  return Math.min(base * factor, MAX_RETRY_DELAY_MS);
 }
 async function withRetry(operation, opName, opts) {
   const prefix = opts.sdkName ? `[raindrop-ai/${opts.sdkName}]` : "[raindrop-ai/core]";
@@ -135,7 +177,9 @@ async function withRetry(operation, opName, opts) {
   throw lastError instanceof Error ? lastError : new Error(String(lastError));
 }
 async function postJson(url, body, headers, opts) {
-  const opName = `POST ${url}`;
+  var _a;
+  const opName = `POST ${redactUrlForLog(url)}`;
+  const timeoutMs = (_a = opts.timeoutMs) != null ? _a : DEFAULT_REQUEST_TIMEOUT_MS;
   await withRetry(
     async () => {
       const resp = await fetch(url, {
@@ -144,7 +188,8 @@ async function postJson(url, body, headers, opts) {
           "Content-Type": "application/json",
           ...headers
         },
-        body: JSON.stringify(body)
+        body: JSON.stringify(body),
+        signal: AbortSignal.timeout(timeoutMs)
       });
       if (!resp.ok) {
         const text = await resp.text().catch(() => "");
@@ -161,6 +206,27 @@ async function postJson(url, body, headers, opts) {
     opts
   );
 }
+var DEFAULT_MAX_TEXT_FIELD_CHARS = 1e6;
+var TRUNCATION_MARKER = "...[truncated by raindrop]";
+var currentDefaultMaxTextFieldChars = DEFAULT_MAX_TEXT_FIELD_CHARS;
+function resolveMaxTextFieldChars(value) {
+  if (typeof value === "number" && Number.isFinite(value) && value > 0) {
+    return Math.floor(value);
+  }
+  return currentDefaultMaxTextFieldChars;
+}
+function truncateToLimit(text, limit) {
+  if (limit > TRUNCATION_MARKER.length) {
+    return text.slice(0, limit - TRUNCATION_MARKER.length) + TRUNCATION_MARKER;
+  }
+  return text.slice(0, Math.max(0, limit));
+}
+function capText(value, limit) {
+  if (typeof value !== "string") return value;
+  const max = limit != null ? limit : currentDefaultMaxTextFieldChars;
+  if (value.length <= max) return value;
+  return truncateToLimit(value, max);
+}
 var SpanStatusCode = {
   UNSET: 0,
   OK: 1,
@@ -303,6 +369,8 @@ function mirrorPartialEventToLocalDebugger(event, options = {}) {
   }).catch(() => {
   });
 }
+var SHUTDOWN_DEADLINE_MS = 1e4;
+var POST_SHUTDOWN_TIMEOUT_MS = 5e3;
 function mergePatches(target, source) {
   var _a, _b, _c, _d;
   const out = { ...target, ...source };
@@ -320,6 +388,7 @@ var EventShipper = class {
     this.sticky = /* @__PURE__ */ new Map();
     this.timers = /* @__PURE__ */ new Map();
     this.inFlight = /* @__PURE__ */ new Set();
+    this.hasShutdown = false;
     var _a, _b, _c, _d, _e, _f, _g, _h;
     this.writeKey = (_a = opts.writeKey) == null ? void 0 : _a.trim();
     this.baseUrl = (_b = formatEndpoint(opts.endpoint)) != null ? _b : "https://api.raindrop.ai/v1/";
@@ -329,6 +398,7 @@ var EventShipper = class {
     this.sdkName = (_d = opts.sdkName) != null ? _d : "core";
     this.prefix = `[raindrop-ai/${this.sdkName}]`;
     this.defaultEventName = (_e = opts.defaultEventName) != null ? _e : "ai_generation";
+    this.maxTextFieldCharsOpt = opts.maxTextFieldChars;
     this.localDebuggerUrl = (_f = resolveLocalDebuggerBaseUrl(opts.localDebuggerUrl)) != null ? _f : void 0;
     if (this.debug && this.localDebuggerUrl) {
       console.log(`${this.prefix} Local debugger mirroring: ${this.localDebuggerUrl}`);
@@ -351,10 +421,52 @@ var EventShipper = class {
   authHeaders() {
     return this.writeKey ? { Authorization: `Bearer ${this.writeKey}` } : {};
   }
+  /**
+   * Build the retry/timeout options for one POST, honoring the shutdown
+   * deadline. Returns `null` when the shutdown drain window is exhausted —
+   * the caller must drop the payload (with a rate-limited warning) instead
+   * of issuing a request that could outlive process exit.
+   *
+   * Checked fresh on EVERY send, so a shutdown that begins while the flush
+   * path is mid-drain takes effect immediately: no further retries, and the
+   * per-attempt timeout is clamped to the remaining window. After
+   * `shutdown()` returns (deadline cleared, `hasShutdown` still set),
+   * sends — late callers, or flush work the deadline abandoned mid-drain —
+   * run as a single short attempt rather than regaining the full retry
+   * schedule.
+   */
+  requestOpts() {
+    if (this.shutdownDeadlineAt !== void 0) {
+      const remainingMs = this.shutdownDeadlineAt - Date.now();
+      if (remainingMs <= 0) return null;
+      return {
+        maxAttempts: 1,
+        debug: this.debug,
+        sdkName: this.sdkName,
+        timeoutMs: Math.min(DEFAULT_REQUEST_TIMEOUT_MS, remainingMs)
+      };
+    }
+    if (this.hasShutdown) {
+      return {
+        maxAttempts: 1,
+        debug: this.debug,
+        sdkName: this.sdkName,
+        timeoutMs: POST_SHUTDOWN_TIMEOUT_MS
+      };
+    }
+    return { maxAttempts: 3, debug: this.debug, sdkName: this.sdkName };
+  }
   async patch(eventId, patch) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
     if (!this.enabled) return;
     if (!eventId || !eventId.trim()) return;
+    const maxChars = resolveMaxTextFieldChars(this.maxTextFieldCharsOpt);
+    if (typeof patch.input === "string" && patch.input.length > maxChars) {
+      patch = { ...patch, input: capText(patch.input, maxChars) };
+    }
+    if (typeof patch.output === "string" && patch.output.length > maxChars) {
+      patch = { ...patch, output: capText(patch.output, maxChars) };
+    }
     if (this.debug) {
       console.log(`${this.prefix} queue patch`, {
         eventId,
@@ -401,9 +513,16 @@ var EventShipper = class {
     })));
   }
   async shutdown() {
-    for (const t of this.timers.values()) clearTimeout(t);
-    this.timers.clear();
-    await this.flush();
+    this.hasShutdown = true;
+    this.shutdownDeadlineAt = Date.now() + SHUTDOWN_DEADLINE_MS;
+    try {
+      for (const t of this.timers.values()) clearTimeout(t);
+      this.timers.clear();
+      const settled = await raceWithTimeout(this.flush(), SHUTDOWN_DEADLINE_MS);
+      if (!settled) this.warnShutdownDrop("in-flight request(s) at shutdown");
+    } finally {
+      this.shutdownDeadlineAt = void 0;
+    }
   }
   async trackSignal(signal) {
     var _a, _b;
@@ -425,15 +544,19 @@ var EventShipper = class {
     ];
     if (!this.writeKey) return;
     const url = `${this.baseUrl}signals/track`;
+    const opts = this.requestOpts();
+    if (!opts) {
+      this.warnShutdownDrop("signal");
+      return;
+    }
     try {
-      await postJson(url, body, this.authHeaders(), {
-        maxAttempts: 3,
-        debug: this.debug,
-        sdkName: this.sdkName
-      });
+      await postJson(url, body, this.authHeaders(), opts);
     } catch (err) {
       const msg = err instanceof Error ? err.message : String(err);
-      console.warn(`${this.prefix} failed to send signal (dropping): ${msg}`);
+      rateLimitedLog(
+        `${this.prefix}.send_signal_failed`,
+        () => console.warn(`${this.prefix} failed to send signal (dropping): ${msg}`)
+      );
     }
   }
   async identify(users) {
@@ -457,17 +580,27 @@ var EventShipper = class {
     if (!this.writeKey) return;
     if (body.length === 0) return;
     const url = `${this.baseUrl}users/identify`;
+    const opts = this.requestOpts();
+    if (!opts) {
+      this.warnShutdownDrop("identify");
+      return;
+    }
     try {
-      await postJson(url, body, this.authHeaders(), {
-        maxAttempts: 3,
-        debug: this.debug,
-        sdkName: this.sdkName
-      });
+      await postJson(url, body, this.authHeaders(), opts);
     } catch (err) {
       const msg = err instanceof Error ? err.message : String(err);
-      console.warn(`${this.prefix} failed to send identify (dropping): ${msg}`);
+      rateLimitedLog(
+        `${this.prefix}.send_identify_failed`,
+        () => console.warn(`${this.prefix} failed to send identify (dropping): ${msg}`)
+      );
     }
   }
+  warnShutdownDrop(what) {
+    rateLimitedLog(
+      `${this.prefix}.shutdown_deadline`,
+      () => console.warn(`${this.prefix} shutdown flush deadline exceeded; dropping ${what}`)
+    );
+  }
   async flushOne(eventId) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
     if (!this.enabled) return;
@@ -543,11 +676,13 @@ var EventShipper = class {
       if (!isPending) this.sticky.delete(eventId);
       return;
     }
-    const p = postJson(url, payload, this.authHeaders(), {
-      maxAttempts: 3,
-      debug: this.debug,
-      sdkName: this.sdkName
-    });
+    const opts = this.requestOpts();
+    if (!opts) {
+      this.warnShutdownDrop(`track_partial ${eventId}`);
+      if (!isPending) this.sticky.delete(eventId);
+      return;
+    }
+    const p = postJson(url, payload, this.authHeaders(), opts);
     this.inFlight.add(p);
     try {
       try {
@@ -557,7 +692,10 @@ var EventShipper = class {
         }
       } catch (err) {
         const msg = err instanceof Error ? err.message : String(err);
-        console.warn(`${this.prefix} failed to send track_partial (dropping): ${msg}`);
+        rateLimitedLog(
+          `${this.prefix}.send_track_partial_failed`,
+          () => console.warn(`${this.prefix} failed to send track_partial (dropping): ${msg}`)
+        );
       }
     } finally {
       this.inFlight.delete(p);
@@ -567,10 +705,113 @@ var EventShipper = class {
     }
   }
 };
+var DEFAULT_SECRET_KEY_NAMES = [
+  "apikey",
+  "apisecret",
+  "apitoken",
+  "secretaccesskey",
+  "sessiontoken",
+  "privatekey",
+  "privatekeyid",
+  "clientsecret",
+  "accesstoken",
+  "refreshtoken",
+  "oauthtoken",
+  "bearertoken",
+  "authorization",
+  "password",
+  "passphrase"
+];
+var REDACTED_PLACEHOLDER = "[REDACTED]";
+function normalizeKeyName(name) {
+  return name.toLowerCase().replace(/[-_.]/g, "");
+}
+function redactSecretsInObject(value, options) {
+  var _a, _b;
+  const normalizedSecretSet = buildSecretSet((_a = options == null ? void 0 : options.secretKeyNames) != null ? _a : DEFAULT_SECRET_KEY_NAMES);
+  const placeholder = (_b = options == null ? void 0 : options.placeholder) != null ? _b : REDACTED_PLACEHOLDER;
+  const seen = /* @__PURE__ */ new WeakSet();
+  const walk = (node) => {
+    if (node === null || typeof node !== "object") return node;
+    if (seen.has(node)) return "[CIRCULAR]";
+    seen.add(node);
+    if (Array.isArray(node)) {
+      return node.map((item) => walk(item));
+    }
+    const out = {};
+    for (const [k, v] of Object.entries(node)) {
+      if (normalizedSecretSet.has(normalizeKeyName(k))) {
+        out[k] = placeholder;
+      } else {
+        out[k] = walk(v);
+      }
+    }
+    return out;
+  };
+  return walk(value);
+}
+function buildSecretSet(names) {
+  const set = /* @__PURE__ */ new Set();
+  for (const name of names) set.add(normalizeKeyName(name));
+  return set;
+}
+var DEFAULT_REDACT_ATTRIBUTE_KEYS = [
+  "ai.request.providerOptions",
+  "ai.response.providerMetadata"
+];
+function defaultTransformSpan(span) {
+  const attrs = span.attributes;
+  if (!attrs || attrs.length === 0) return span;
+  let nextAttrs;
+  for (let i = 0; i < attrs.length; i++) {
+    const attr = attrs[i];
+    const redacted = redactJsonAttributeValue(attr.key, attr.value);
+    if (redacted === void 0) continue;
+    if (!nextAttrs) nextAttrs = attrs.slice();
+    nextAttrs[i] = { key: attr.key, value: redacted };
+  }
+  if (!nextAttrs) return span;
+  return { ...span, attributes: nextAttrs };
+}
+var REDACT_JSON_ATTRIBUTE_KEYS = new Set(DEFAULT_REDACT_ATTRIBUTE_KEYS);
+function redactJsonAttributeValue(key, value) {
+  if (!REDACT_JSON_ATTRIBUTE_KEYS.has(key)) return void 0;
+  const json = value.stringValue;
+  if (typeof json !== "string" || json.length === 0) return void 0;
+  let parsed;
+  try {
+    parsed = JSON.parse(json);
+  } catch (e) {
+    return void 0;
+  }
+  const scrubbed = redactSecretsInObject(parsed);
+  let scrubbedJson;
+  try {
+    scrubbedJson = JSON.stringify(scrubbed);
+  } catch (e) {
+    return void 0;
+  }
+  if (scrubbedJson === json) return void 0;
+  return { stringValue: scrubbedJson };
+}
+function applyOtelSpanAttributeLimit(limit) {
+  var _a, _b;
+  try {
+    const raw = (_b = (_a = globalThis == null ? void 0 : globalThis.process) == null ? void 0 : _a.env) == null ? void 0 : _b.OTEL_SPAN_ATTRIBUTE_VALUE_LENGTH_LIMIT;
+    if (!raw) return limit;
+    const parsed = Number.parseInt(raw, 10);
+    if (Number.isFinite(parsed) && parsed > 0) {
+      return Math.min(limit, parsed);
+    }
+  } catch (e) {
+  }
+  return limit;
+}
 var TraceShipper = class {
   constructor(opts) {
     this.queue = [];
     this.inFlight = /* @__PURE__ */ new Set();
+    this.hasShutdown = false;
     var _a, _b, _c, _d, _e, _f, _g, _h, _i;
     this.writeKey = (_a = opts.writeKey) == null ? void 0 : _a.trim();
     this.baseUrl = (_b = formatEndpoint(opts.endpoint)) != null ? _b : "https://api.raindrop.ai/v1/";
@@ -588,6 +829,75 @@ var TraceShipper = class {
     if (this.debug && this.localDebuggerUrl) {
       console.log(`${this.prefix} Local debugger mirroring: ${this.localDebuggerUrl}`);
     }
+    this.transformSpanHook = opts.transformSpan;
+    this.disableDefaultRedaction = opts.disableDefaultRedaction === true;
+    this.maxTextFieldCharsOpt = opts.maxTextFieldChars;
+  }
+  /**
+   * Cap every string attribute value on the span. O(#attributes) length
+   * checks; only oversized values pay a slice. Runs AFTER the redaction
+   * pipeline so the default secret-scrub still sees parseable JSON in
+   * `ai.request.providerOptions` / `ai.response.providerMetadata` (capping
+   * first could cut a JSON blob mid-way, fail the parse, and ship secrets
+   * in the surviving prefix).
+   *
+   * A stricter `OTEL_SPAN_ATTRIBUTE_VALUE_LENGTH_LIMIT` env var is honored
+   * for span content, matching the Python SDK and the OTel SDK convention.
+   */
+  capSpanAttributes(span) {
+    var _a;
+    const maxChars = applyOtelSpanAttributeLimit(
+      resolveMaxTextFieldChars(this.maxTextFieldCharsOpt)
+    );
+    const attrs = span.attributes;
+    if (!attrs || attrs.length === 0) return span;
+    let nextAttrs;
+    for (let i = 0; i < attrs.length; i++) {
+      const attr = attrs[i];
+      const value = (_a = attr.value) == null ? void 0 : _a.stringValue;
+      if (typeof value !== "string" || value.length <= maxChars) continue;
+      if (!nextAttrs) nextAttrs = attrs.slice();
+      nextAttrs[i] = {
+        key: attr.key,
+        value: { ...attr.value, stringValue: capText(value, maxChars) }
+      };
+    }
+    if (!nextAttrs) return span;
+    return { ...span, attributes: nextAttrs };
+  }
+  /**
+   * Apply the user `transformSpan` hook (if any) followed by the default
+   * redactor (unless disabled). Returns either the (possibly new) span to
+   * ship, or `null` to drop the span entirely.
+   *
+   * Ordering: user hook runs first so callers can rewrite the span freely
+   * (rename attrs, add new ones, scrub things the default doesn't know
+   * about). The default redactor then runs on whatever the user produced,
+   * acting as the always-on floor for documented BYOK secrets. If the user
+   * sets `disableDefaultRedaction: true`, the floor is skipped.
+   *
+   * Fail-closed: if the user hook throws, the span is dropped — a buggy
+   * hook can never accidentally ship raw, un-redacted spans.
+   */
+  redactSpan(span) {
+    let current = span;
+    if (this.transformSpanHook) {
+      try {
+        const result = this.transformSpanHook(current);
+        if (result === null) return null;
+        if (result !== void 0) current = result;
+      } catch (err) {
+        if (this.debug) {
+          const msg = err instanceof Error ? err.message : String(err);
+          console.warn(`${this.prefix} transformSpan hook threw: ${msg}`);
+        }
+        return null;
+      }
+    }
+    if (!this.disableDefaultRedaction) {
+      current = defaultTransformSpan(current);
+    }
+    return this.capSpanAttributes(current);
   }
   isDebugEnabled() {
     return this.debug;
@@ -605,8 +915,8 @@ var TraceShipper = class {
     ];
     if ((_b = args.attributes) == null ? void 0 : _b.length) attrs.push(...args.attributes);
     const span = { ids, name: args.name, startTimeUnixNano: started, attributes: attrs };
-    if (this.localDebuggerUrl) {
-      const openSpan = buildOtlpSpan({
+    this.mirrorToLocalDebugger(
+      buildOtlpSpan({
         ids: span.ids,
         name: span.name,
         startTimeUnixNano: span.startTimeUnixNano,
@@ -614,16 +924,21 @@ var TraceShipper = class {
         // placeholder — will be updated on endSpan
         attributes: span.attributes,
         status: { code: SpanStatusCode.UNSET }
-      });
-      const body = buildExportTraceServiceRequest([openSpan], this.serviceName, this.serviceVersion);
-      mirrorTraceExportToLocalDebugger(body, {
-        baseUrl: this.localDebuggerUrl,
-        debug: false,
-        sdkName: this.sdkName
-      });
-    }
+      })
+    );
     return span;
   }
+  mirrorToLocalDebugger(span) {
+    if (!this.localDebuggerUrl) return;
+    const redacted = this.redactSpan(span);
+    if (redacted === null) return;
+    const body = buildExportTraceServiceRequest([redacted], this.serviceName, this.serviceVersion);
+    mirrorTraceExportToLocalDebugger(body, {
+      baseUrl: this.localDebuggerUrl,
+      debug: false,
+      sdkName: this.sdkName
+    });
+  }
   endSpan(span, extra) {
     var _a, _b;
     if (span.endTimeUnixNano) return;
@@ -645,14 +960,7 @@ var TraceShipper = class {
       status
     });
     this.enqueue(otlp);
-    if (this.localDebuggerUrl) {
-      const body = buildExportTraceServiceRequest([otlp], this.serviceName, this.serviceVersion);
-      mirrorTraceExportToLocalDebugger(body, {
-        baseUrl: this.localDebuggerUrl,
-        debug: false,
-        sdkName: this.sdkName
-      });
-    }
+    this.mirrorToLocalDebugger(otlp);
   }
   createSpan(args) {
     var _a;
@@ -670,14 +978,7 @@ var TraceShipper = class {
       status: args.status
     });
     this.enqueue(otlp);
-    if (this.localDebuggerUrl) {
-      const body = buildExportTraceServiceRequest([otlp], this.serviceName, this.serviceVersion);
-      mirrorTraceExportToLocalDebugger(body, {
-        baseUrl: this.localDebuggerUrl,
-        debug: false,
-        sdkName: this.sdkName
-      });
-    }
+    this.mirrorToLocalDebugger(otlp);
   }
   enqueue(span) {
     if (!this.enabled) return;
@@ -689,10 +990,12 @@ var TraceShipper = class {
         )}`
       );
     }
+    const redacted = this.redactSpan(span);
+    if (redacted === null) return;
     if (this.queue.length >= this.maxQueueSize) {
       this.queue.shift();
     }
-    this.queue.push(span);
+    this.queue.push(redacted);
     if (this.queue.length >= this.maxBatchSize) {
       void this.flush().catch(() => {
       });
@@ -715,6 +1018,16 @@ var TraceShipper = class {
     while (this.queue.length > 0) {
       const batch = this.queue.splice(0, this.maxBatchSize);
       if (!this.writeKey) continue;
+      const opts = this.requestOpts();
+      if (!opts) {
+        rateLimitedLog(
+          `${this.prefix}.shutdown_deadline`,
+          () => console.warn(
+            `${this.prefix} shutdown flush deadline exceeded; dropping ${batch.length} spans`
+          )
+        );
+        continue;
+      }
       const body = buildExportTraceServiceRequest(batch, this.serviceName, this.serviceVersion);
       const url = `${this.baseUrl}traces`;
       if (this.debug) {
@@ -723,11 +1036,7 @@ var TraceShipper = class {
           endpoint: url
         });
       }
-      const p = postJson(url, body, this.authHeaders(), {
-        maxAttempts: 3,
-        debug: this.debug,
-        sdkName: this.sdkName
-      });
+      const p = postJson(url, body, this.authHeaders(), opts);
       this.inFlight.add(p);
       try {
         try {
@@ -735,21 +1044,61 @@ var TraceShipper = class {
           if (this.debug) console.log(`${this.prefix} sent ${batch.length} spans`);
         } catch (err) {
           const msg = err instanceof Error ? err.message : String(err);
-          console.warn(`${this.prefix} failed to send ${batch.length} spans: ${msg}`);
+          rateLimitedLog(
+            `${this.prefix}.send_spans_failed`,
+            () => console.warn(`${this.prefix} failed to send ${batch.length} spans: ${msg}`)
+          );
         }
       } finally {
         this.inFlight.delete(p);
       }
     }
   }
+  /** See EventShipper.requestOpts — same shutdown-budget semantics. */
+  requestOpts() {
+    if (this.shutdownDeadlineAt !== void 0) {
+      const remainingMs = this.shutdownDeadlineAt - Date.now();
+      if (remainingMs <= 0) return null;
+      return {
+        maxAttempts: 1,
+        debug: this.debug,
+        sdkName: this.sdkName,
+        timeoutMs: Math.min(DEFAULT_REQUEST_TIMEOUT_MS, remainingMs)
+      };
+    }
+    if (this.hasShutdown) {
+      return {
+        maxAttempts: 1,
+        debug: this.debug,
+        sdkName: this.sdkName,
+        timeoutMs: POST_SHUTDOWN_TIMEOUT_MS
+      };
+    }
+    return { maxAttempts: 3, debug: this.debug, sdkName: this.sdkName };
+  }
   async shutdown() {
-    if (this.timer) {
-      clearTimeout(this.timer);
-      this.timer = void 0;
+    this.hasShutdown = true;
+    this.shutdownDeadlineAt = Date.now() + SHUTDOWN_DEADLINE_MS;
+    try {
+      if (this.timer) {
+        clearTimeout(this.timer);
+        this.timer = void 0;
+      }
+      const drain = async () => {
+        await this.flush();
+        await Promise.all([...this.inFlight].map((p) => p.catch(() => {
+        })));
+      };
+      const settled = await raceWithTimeout(drain(), SHUTDOWN_DEADLINE_MS);
+      if (!settled) {
+        rateLimitedLog(
+          `${this.prefix}.shutdown_deadline`,
+          () => console.warn(`${this.prefix} shutdown flush deadline exceeded; abandoning in-flight spans`)
+        );
+      }
+    } finally {
+      this.shutdownDeadlineAt = void 0;
     }
-    await this.flush();
-    await Promise.all([...this.inFlight].map((p) => p.catch(() => {
-    })));
   }
 };
 
@@ -814,7 +1163,7 @@ function resolveLocalWorkshopUrl(fileValue) {
 // package.json
 var package_default = {
   name: "@raindrop-ai/pi-agent",
-  version: "0.0.3",
+  version: "0.0.5",
   description: "Raindrop observability for Pi Agent \u2014 automatic tracing via subscriber or pi-coding-agent extension",
   type: "module",
   license: "MIT",
@@ -864,18 +1213,18 @@ var package_default = {
     "test:watch": "vitest"
   },
   peerDependencies: {
-    "@mariozechner/pi-agent-core": ">=0.60.0",
-    "@mariozechner/pi-coding-agent": ">=0.65.2"
+    "@earendil-works/pi-agent-core": ">=0.74.0",
+    "@earendil-works/pi-coding-agent": ">=0.74.0"
   },
   peerDependenciesMeta: {
-    "@mariozechner/pi-coding-agent": {
+    "@earendil-works/pi-coding-agent": {
       optional: true
     }
   },
   devDependencies: {
     "@raindrop-ai/core": "workspace:*",
-    "@mariozechner/pi-agent-core": "^0.66.0",
-    "@mariozechner/pi-coding-agent": "^0.66.0",
+    "@earendil-works/pi-agent-core": "^0.78.0",
+    "@earendil-works/pi-coding-agent": "^0.78.0",
     "@types/node": "^20.11.17",
     msw: "^2.12.7",
     tsup: "^8.5.1",
@@ -969,19 +1318,82 @@ function formatToolSpanName(toolName, args) {
   }
   return toolName;
 }
-function safeStringify(value) {
+var TRUNCATION_MARKER2 = "...[truncated by raindrop]";
+var MAX_TEXT_FIELD_CHARS = 1e6;
+var MAX_ATTR_LENGTH = 32768;
+var MAX_BOUNDED_DEPTH = 12;
+function capText2(value, limit = MAX_TEXT_FIELD_CHARS) {
+  if (value.length <= limit) return value;
+  if (limit > TRUNCATION_MARKER2.length) {
+    return value.slice(0, limit - TRUNCATION_MARKER2.length) + TRUNCATION_MARKER2;
+  }
+  return value.slice(0, limit);
+}
+function boundedClone2(value, budget, depth) {
+  if (budget.remaining <= 0) return TRUNCATION_MARKER2;
+  if (typeof value === "string") {
+    if (value.length > budget.remaining) {
+      const taken = value.slice(0, Math.max(0, budget.remaining)) + TRUNCATION_MARKER2;
+      budget.remaining = 0;
+      return taken;
+    }
+    budget.remaining -= Math.max(value.length, 1);
+    return value;
+  }
+  if (value === null || typeof value === "number" || typeof value === "boolean") {
+    budget.remaining -= 8;
+    return value;
+  }
+  if (typeof value !== "object") {
+    budget.remaining -= 8;
+    return value;
+  }
+  if (depth >= MAX_BOUNDED_DEPTH) {
+    budget.remaining -= 16;
+    return `<max depth: ${TRUNCATION_MARKER2}>`;
+  }
+  const withToJson = value;
+  if (typeof withToJson.toJSON === "function") {
+    try {
+      return boundedClone2(withToJson.toJSON(), budget, depth + 1);
+    } catch (e) {
+    }
+  }
+  if (Array.isArray(value)) {
+    const out2 = [];
+    for (const item of value) {
+      if (budget.remaining <= 0) {
+        out2.push(TRUNCATION_MARKER2);
+        break;
+      }
+      out2.push(boundedClone2(item, budget, depth + 1));
+    }
+    return out2;
+  }
+  const out = {};
+  for (const key of Object.keys(value)) {
+    if (budget.remaining <= 0) {
+      out["..."] = TRUNCATION_MARKER2;
+      break;
+    }
+    budget.remaining -= Math.max(key.length, 1);
+    out[key] = boundedClone2(value[key], budget, depth + 1);
+  }
+  return out;
+}
+function safeStringify(value, limit = MAX_ATTR_LENGTH) {
   if (value === void 0 || value === null) return void 0;
   try {
-    return JSON.stringify(value);
+    const pruned = boundedClone2(value, { remaining: limit + TRUNCATION_MARKER2.length + 256 }, 0);
+    return JSON.stringify(pruned);
   } catch (e) {
     try {
-      return String(value);
+      return capText2(String(value), limit);
     } catch (e2) {
       return "[unserializable]";
     }
   }
 }
-var MAX_ATTR_LENGTH = 32768;
 function truncate(value) {
   if (value === void 0) return void 0;
   if (value.length <= MAX_ATTR_LENGTH) return value;
@@ -1020,6 +1432,15 @@ function safeParsArgs(argsStr) {
   }
 }
 var MAX_SYSTEM_PROMPT_LENGTH = 32768;
+var ERROR_LOG_INTERVAL_MS = 3e4;
+var lastErrorLogAt = /* @__PURE__ */ new Map();
+function rateLimitedErrorLog(key, message) {
+  const now = Date.now();
+  const last = lastErrorLogAt.get(key);
+  if (last !== void 0 && now - last < ERROR_LOG_INTERVAL_MS) return;
+  lastErrorLogAt.set(key, now);
+  console.log(message);
+}
 function createSessionState(sessionId) {
   return {
     sessionId,
@@ -1084,7 +1505,8 @@ function getState(stateRef, ctx) {
 function registerTracing(pi, config, eventShipper, traceShipper) {
   const stateRef = {};
   function logError(hook, err) {
-    console.log(
+    rateLimitedErrorLog(
+      hook,
       `[raindrop-ai/pi-agent] [error] Error in ${hook}: ${err instanceof Error ? err.message : String(err)}`
     );
   }
@@ -1137,7 +1559,7 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
       }
       state.toolSpanStarts.clear();
       state.toolCallToLlmParent.clear();
-      state.currentInput = event.prompt;
+      state.currentInput = capText2(event.prompt);
       state.turnNumber = 0;
       state.currentSystemPrompt = config.captureSystemPrompt ? truncateSystemPrompt(event.systemPrompt) : void 0;
       state.currentEventRequestId = generateId();
@@ -1158,7 +1580,7 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
         userId: getUserId(state, config.eventMetadata),
         convoId: state.sessionId,
         eventName: getEventName(config),
-        input: event.prompt,
+        input: state.currentInput,
         ...attachments.length > 0 ? { attachments } : {},
         properties: getBaseProperties(config, ctx)
       });
@@ -1209,7 +1631,7 @@ function registerTracing(pi, config, eventShipper, traceShipper) {
       const modelId = (_c = message.model) != null ? _c : "";
       const modelName = provider && modelId ? `${provider}/${modelId}` : modelId || "llm";
       const errorForSpan = getAssistantError(message);
-      const outputText = getAssistantText(message);
+      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;