raindrop-ai 0.1.1 → 0.1.2

package/dist/tracing/index.js

@@ -50,7 +50,9 @@ var import_instrumentation_vertexai = require("@traceloop/instrumentation-vertex
 var traceloop3 = __toESM(require("@traceloop/node-server-sdk"));
 var import_weakref = require("weakref");
 
-// ../core/dist/chunk-OLYXZCOK.js
+// ../core/dist/chunk-U5HUTMR5.js
+var DEFAULT_REQUEST_TIMEOUT_MS = 3e4;
+var MAX_RETRY_DELAY_MS = 3e4;
 function wait(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
@@ -58,6 +60,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");
@@ -74,12 +116,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]";
@@ -114,7 +156,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, {
@@ -123,7 +167,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(() => "");
@@ -140,6 +185,97 @@ async function postJson(url, body, headers, opts) {
     opts
   );
 }
+var DEFAULT_MAX_TEXT_FIELD_CHARS = 1e6;
+var TRUNCATION_MARKER = "...[truncated by raindrop]";
+var BOUNDED_CLONE_MAX_DEPTH = 12;
+var BOUNDED_CLONE_SLACK = 256;
+var currentDefaultMaxTextFieldChars = DEFAULT_MAX_TEXT_FIELD_CHARS;
+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);
+}
+function boundedClone(obj, charBudget) {
+  let budget = charBudget;
+  const seen = /* @__PURE__ */ new WeakSet();
+  const walk = (node, depth) => {
+    if (budget <= 0) return TRUNCATION_MARKER;
+    if (typeof node === "string") {
+      if (node.length > budget) {
+        const taken = node.slice(0, Math.max(0, budget)) + TRUNCATION_MARKER;
+        budget = 0;
+        return taken;
+      }
+      budget -= Math.max(node.length, 1);
+      return node;
+    }
+    if (node === null || typeof node === "number" || typeof node === "boolean") {
+      budget -= 8;
+      return node;
+    }
+    if (typeof node !== "object") {
+      budget -= 8;
+      return node;
+    }
+    if (seen.has(node)) return "[CIRCULAR]";
+    if (depth >= BOUNDED_CLONE_MAX_DEPTH) {
+      budget -= 16;
+      return `<max depth>`;
+    }
+    seen.add(node);
+    if (Array.isArray(node)) {
+      const out2 = [];
+      for (const v of node) {
+        if (budget <= 0) {
+          out2.push(TRUNCATION_MARKER);
+          break;
+        }
+        out2.push(walk(v, depth + 1));
+      }
+      return out2;
+    }
+    if (typeof node.toJSON === "function") {
+      budget -= 16;
+      return node;
+    }
+    const out = {};
+    for (const k in node) {
+      if (!Object.prototype.hasOwnProperty.call(node, k)) continue;
+      if (budget <= 0) {
+        out["..."] = TRUNCATION_MARKER;
+        break;
+      }
+      const key = walk(k, depth + 1);
+      out[key] = walk(node[k], depth + 1);
+    }
+    return out;
+  };
+  return walk(obj, 0);
+}
+function stringifyBounded(value, limit) {
+  const max = limit != null ? limit : currentDefaultMaxTextFieldChars;
+  if (typeof value === "string") {
+    return capText(value, max);
+  }
+  const pruned = boundedClone(value, max + TRUNCATION_MARKER.length + BOUNDED_CLONE_SLACK);
+  let json;
+  try {
+    json = JSON.stringify(pruned);
+  } catch (e) {
+    json = void 0;
+  }
+  if (json === void 0) {
+    return capText(String(value), max);
+  }
+  return json.length <= max ? json : truncateToLimit(json, max);
+}
 var LOCAL_DEBUGGER_ENV_VAR = "RAINDROP_LOCAL_DEBUGGER";
 var WORKSHOP_ENV_VAR = "RAINDROP_WORKSHOP";
 var DEFAULT_LOCAL_WORKSHOP_URL = "http://localhost:5899/v1/";
@@ -236,6 +372,13 @@ function sendLocalDebuggerLiveEvent(event, options = {}) {
   ).catch(() => {
   });
 }
+var SHUTDOWN_DEADLINE_MS = 1e4;
+var POST_SHUTDOWN_TIMEOUT_MS = 5e3;
+var DEFAULT_REDACT_ATTRIBUTE_KEYS = [
+  "ai.request.providerOptions",
+  "ai.response.providerMetadata"
+];
+var REDACT_JSON_ATTRIBUTE_KEYS = new Set(DEFAULT_REDACT_ATTRIBUTE_KEYS);
 
 // ../core/dist/index.node.js
 var import_async_hooks = require("async_hooks");
@@ -265,12 +408,12 @@ function parseRetryAfter2(headers) {
 function getRetryDelayMs2(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 withRetry2(operation, opName, opts) {
   let lastError = void 0;
@@ -302,7 +445,9 @@ async function withRetry2(operation, opName, opts) {
   throw lastError instanceof Error ? lastError : new Error(String(lastError));
 }
 async function postJson2(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 withRetry2(
     async () => {
       const resp = await fetch(url, {
@@ -311,7 +456,8 @@ async function postJson2(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(() => "");
@@ -392,7 +538,7 @@ function base64ToHex(base64) {
 // package.json
 var package_default = {
   name: "raindrop-ai",
-  version: "0.1.1",
+  version: "0.1.2",
   main: "dist/index.js",
   module: "dist/index.mjs",
   types: "dist/index.d.ts",
@@ -578,16 +724,19 @@ function getActiveTraceContext() {
 
 // src/tracing/direct/shipper.ts
 function serializeAssociationValue(value) {
-  if (typeof value === "string") {
-    return value;
-  }
-  const jsonValue = JSON.stringify(value);
-  return jsonValue === void 0 ? String(value) : jsonValue;
+  return stringifyBounded(value);
 }
 var DirectTraceShipper = class {
   constructor(opts) {
     this.queue = [];
     this.inFlight = /* @__PURE__ */ new Set();
+    /**
+     * Set once `shutdown()` begins and never cleared. Sends issued after the
+     * drain window (stragglers, or flush work the deadline abandoned
+     * mid-drain) run as a single short attempt instead of regaining the full
+     * retry schedule.
+     */
+    this.hasShutdown = false;
     var _a, _b, _c, _d, _e, _f;
     this.writeKey = opts.writeKey;
     this.baseUrl = (_a = formatEndpoint2(opts.endpoint)) != null ? _a : "https://api.raindrop.ai/v1/";
@@ -712,14 +861,19 @@ var DirectTraceShipper = class {
         if (this.debug) console.log(`[raindrop] direct: dropped ${batch.length} cloud spans (local-only)`);
         continue;
       }
+      const opts = this.requestOpts();
+      if (!opts) {
+        rateLimitedLog(
+          "js-sdk.direct.shutdown_deadline",
+          () => console.warn(
+            `[raindrop] direct: shutdown flush deadline exceeded; dropping ${batch.length} spans`
+          )
+        );
+        continue;
+      }
       const body = buildExportTraceServiceRequest2(batch);
       const url = `${this.baseUrl}traces`;
-      const p = postJson2(
-        url,
-        body,
-        { Authorization: `Bearer ${this.writeKey}` },
-        { maxAttempts: 3, debug: this.debug }
-      );
+      const p = postJson2(url, body, { Authorization: `Bearer ${this.writeKey}` }, opts);
       this.inFlight.add(p);
       try {
         try {
@@ -736,14 +890,48 @@ var DirectTraceShipper = class {
       }
     }
   }
+  /**
+   * Retry/timeout options for one batch POST, honoring the shutdown budget:
+   * outside shutdown, the normal 3-attempt schedule; during shutdown, a
+   * single attempt clamped to the remaining window; `null` once the window
+   * is exhausted (caller drops the batch). After shutdown() returns,
+   * stragglers run as a single short attempt.
+   */
+  requestOpts() {
+    if (this.shutdownDeadlineAt !== void 0) {
+      const remainingMs = this.shutdownDeadlineAt - Date.now();
+      if (remainingMs <= 0) return null;
+      return {
+        maxAttempts: 1,
+        debug: this.debug,
+        timeoutMs: Math.min(DEFAULT_REQUEST_TIMEOUT_MS, remainingMs)
+      };
+    }
+    if (this.hasShutdown) {
+      return { maxAttempts: 1, debug: this.debug, timeoutMs: POST_SHUTDOWN_TIMEOUT_MS };
+    }
+    return { maxAttempts: 3, debug: this.debug };
+  }
   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 && this.debug) {
+        console.warn("[raindrop] direct: shutdown deadline exceeded; abandoning in-flight spans");
+      }
+    } finally {
+      this.shutdownDeadlineAt = void 0;
     }
-    await this.flush();
-    await Promise.all([...this.inFlight].map((p) => p.catch(() => {
-    })));
   }
 };
 
@@ -759,7 +947,7 @@ function getPropertiesFromContext(context5) {
     ...context5.properties || {}
   };
   if (context5.attachments && context5.attachments.length > 0) {
-    properties.attachments = JSON.stringify(context5.attachments);
+    properties.attachments = stringifyBounded(context5.attachments);
   }
   return properties;
 }
@@ -777,11 +965,7 @@ function serializeAssociationProperties(properties) {
   );
 }
 function serializeSpanValue(value) {
-  if (typeof value === "string") {
-    return value;
-  }
-  const jsonValue = JSON.stringify(value);
-  return jsonValue === void 0 ? String(value) : jsonValue;
+  return stringifyBounded(value);
 }
 function getLocalDebuggerMetadata(context5) {
   return {
@@ -1060,7 +1244,7 @@ var LiveInteraction = class {
         error = err instanceof Error ? err.message : String(err);
         const endTimeMs2 = Date.now();
         const durationMs2 = endTimeMs2 - startTimeMs;
-        const inputStr2 = params.inputParameters ? JSON.stringify(params.inputParameters) : void 0;
+        const inputStr2 = params.inputParameters ? stringifyBounded(params.inputParameters) : void 0;
         this.directShipper.sendToolSpan({
           name: toolName,
           spanId: spanIdB64,
@@ -1089,8 +1273,8 @@ var LiveInteraction = class {
       }
       const endTimeMs = Date.now();
       const durationMs = endTimeMs - startTimeMs;
-      const inputStr = params.inputParameters ? JSON.stringify(params.inputParameters) : void 0;
-      const outputStr = result !== void 0 ? typeof result === "string" ? result : JSON.stringify(result) : void 0;
+      const inputStr = params.inputParameters ? stringifyBounded(params.inputParameters) : void 0;
+      const outputStr = result !== void 0 ? stringifyBounded(result) : void 0;
       this.directShipper.sendToolSpan({
         name: toolName,
         spanId: spanIdB64,
@@ -1327,8 +1511,8 @@ var LiveInteraction = class {
       const traceIdB64 = this.ensureTraceId(activeContext.traceIdB64);
       const { parentSpanIdB64 } = activeContext;
       const spanIdB64 = base64Encode2(randomBytes2(8));
-      const inputStr = input !== void 0 ? typeof input === "string" ? input : JSON.stringify(input) : void 0;
-      const outputStr = output !== void 0 ? typeof output === "string" ? output : JSON.stringify(output) : void 0;
+      const inputStr = input !== void 0 ? stringifyBounded(input) : void 0;
+      const outputStr = output !== void 0 ? stringifyBounded(output) : void 0;
       this.directShipper.sendToolSpan({
         name,
         spanId: spanIdB64,
@@ -1375,12 +1559,10 @@ var LiveInteraction = class {
     span.setAttribute("traceloop.span.kind", "tool");
     setAssociationProperties(span, properties);
     if (input !== void 0) {
-      const inputStr = typeof input === "string" ? input : JSON.stringify(input);
-      span.setAttribute("traceloop.entity.input", inputStr);
+      span.setAttribute("traceloop.entity.input", stringifyBounded(input));
     }
     if (output !== void 0) {
-      const outputStr = typeof output === "string" ? output : JSON.stringify(output);
-      span.setAttribute("traceloop.entity.output", outputStr);
+      span.setAttribute("traceloop.entity.output", stringifyBounded(output));
     }
     if (durationMs !== void 0) {
       span.setAttribute("traceloop.entity.duration_ms", durationMs);
@@ -1468,8 +1650,8 @@ var LiveTracer = class {
     if (this.directShipper) {
       const { traceIdB64, parentSpanIdB64 } = getActiveTraceContext();
       const spanIdB64 = base64Encode2(randomBytes2(8));
-      const inputStr = input !== void 0 ? typeof input === "string" ? input : JSON.stringify(input) : void 0;
-      const outputStr = output !== void 0 ? typeof output === "string" ? output : JSON.stringify(output) : void 0;
+      const inputStr = input !== void 0 ? stringifyBounded(input) : void 0;
+      const outputStr = output !== void 0 ? stringifyBounded(output) : void 0;
       this.directShipper.sendToolSpan({
         name,
         spanId: spanIdB64,
@@ -1513,12 +1695,10 @@ var LiveTracer = class {
     });
     span.setAttribute("traceloop.span.kind", "tool");
     if (input !== void 0) {
-      const inputStr = typeof input === "string" ? input : JSON.stringify(input);
-      span.setAttribute("traceloop.entity.input", inputStr);
+      span.setAttribute("traceloop.entity.input", stringifyBounded(input));
     }
     if (output !== void 0) {
-      const outputStr = typeof output === "string" ? output : JSON.stringify(output);
-      span.setAttribute("traceloop.entity.output", outputStr);
+      span.setAttribute("traceloop.entity.output", stringifyBounded(output));
     }
     if (durationMs !== void 0) {
       span.setAttribute("traceloop.entity.duration_ms", durationMs);
@@ -1950,6 +2130,7 @@ function getCurrentTraceId() {
   return (_a = import_api4.trace.getSpan(import_api4.context.active())) == null ? void 0 : _a.spanContext().traceId;
 }
 var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
+  process.env.TRACELOOP_TELEMETRY = "false";
   const {
     logLevel,
     useExternalOtel,

package/dist/tracing/index.mjs

@@ -1,6 +1,6 @@
 import {
   tracing
-} from "../chunk-DP5ED5DR.mjs";
+} from "../chunk-UICWZEYM.mjs";
 import "../chunk-UJCSKKID.mjs";
 
 // src/tracing/index.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "raindrop-ai",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
@@ -46,8 +46,8 @@
     "typescript": "^5.3.3",
     "vite": "^7.3.2",
     "vitest": "^4.0.10",
+    "@raindrop-ai/core": "0.0.3",
     "@raindrop-ai/redact-pii": "0.1.2",
-    "@raindrop-ai/core": "0.0.1",
     "@raindrop-ai/schemas": "0.0.1"
   },
   "dependencies": {