@raindrop-ai/ai-sdk 0.0.32 → 0.0.33

package/dist/index.workers.js

@@ -1024,7 +1024,7 @@ globalThis.RAINDROP_ASYNC_LOCAL_STORAGE = async_hooks.AsyncLocalStorage;
 // package.json
 var package_default = {
   name: "@raindrop-ai/ai-sdk",
-  version: "0.0.32"};
+  version: "0.0.33"};
 
 // src/internal/version.ts
 var libraryName = package_default.name;
@@ -1807,6 +1807,25 @@ var RaindropTelemetryIntegration = class {
      * (the `event.callId` can be the same for parallel sibling tools).
      */
     this.priorParentContexts = /* @__PURE__ */ new Map();
+    // ── lifecycle ─────────────────────────────────────────────────────────────
+    //
+    // The shippers buffer spans/events and flush on a timer, so a short-lived
+    // script (e.g. a single `generateText` then `process.exit`) can exit before
+    // anything ships. Expose `flush`/`shutdown` on the integration itself so the
+    // value returned by `raindrop()` is self-sufficient — callers can keep the
+    // reference they pass to `registerTelemetry` and `await rd.flush()` before
+    // exiting, without also constructing a separate client.
+    /** Flush any buffered events and trace spans to their destinations. */
+    this.flush = async () => {
+      await Promise.all([this.eventShipper.flush(), this.traceShipper.flush()]);
+    };
+    /** Flush and stop the background flush timers. */
+    this.shutdown = async () => {
+      await Promise.all([
+        this.eventShipper.shutdown(),
+        this.traceShipper.shutdown()
+      ]);
+    };
     // ── onStart ─────────────────────────────────────────────────────────────
     this.onStart = (event) => {
       var _a, _b, _c, _d, _e;
@@ -2200,8 +2219,56 @@ var RaindropTelemetryIntegration = class {
       if (state.subagentToolCallSpan) {
         this.traceShipper.endSpan(state.subagentToolCallSpan, { error: actualError });
       }
+      const isEmbed = state.operationId === "ai.embed" || state.operationId === "ai.embedMany";
+      if (!isEmbed) {
+        this.finalizeGenerateEvent(state, state.accumulatedText || void 0, void 0);
+      }
       this.cleanup(event.callId);
     };
+    // ── onAbort ─────────────────────────────────────────────────────────────
+    //
+    // Dispatched by the v7 canary line (post-beta.116) when a `streamText` call
+    // is cancelled via its `AbortSignal`. On abort the SDK fires neither `onEnd`
+    // nor `onError`, so without this every open span would leak and the event
+    // would hang forever in its `isPending` state. We close every open span with
+    // an `ai.response.aborted` marker (no error status — an abort is not a model
+    // failure) and flush the partial event.
+    this.onAbort = (event) => {
+      const callId = event == null ? void 0 : event.callId;
+      if (!callId) return;
+      const state = this.getState(callId);
+      if (!state) return;
+      const abortedAttr = attrString("ai.response.aborted", "true");
+      if (state.stepSpan) {
+        this.traceShipper.endSpan(state.stepSpan, { attributes: [abortedAttr] });
+        state.stepSpan = void 0;
+        state.stepParent = void 0;
+      }
+      for (const embedSpan of state.embedSpans.values()) {
+        this.traceShipper.endSpan(embedSpan, { attributes: [abortedAttr] });
+      }
+      state.embedSpans.clear();
+      for (const toolCallId of state.parentContextToolCallIds) {
+        this.priorParentContexts.delete(toolCallId);
+      }
+      state.parentContextToolCallIds.clear();
+      for (const toolSpan of state.toolSpans.values()) {
+        this.traceShipper.endSpan(toolSpan, { attributes: [abortedAttr] });
+      }
+      state.toolSpans.clear();
+      if (state.rootSpan) {
+        this.traceShipper.endSpan(state.rootSpan, {
+          attributes: [abortedAttr, attrInt("ai.toolCall.count", state.toolCallCount)]
+        });
+      }
+      if (state.subagentToolCallSpan) {
+        this.traceShipper.endSpan(state.subagentToolCallSpan, {
+          attributes: [abortedAttr]
+        });
+      }
+      this.finalizeGenerateEvent(state, state.accumulatedText || void 0, void 0);
+      this.cleanup(callId);
+    };
     // ── executeTool ─────────────────────────────────────────────────────────
     this.executeTool = async ({
       callId,
@@ -2359,11 +2426,12 @@ var RaindropTelemetryIntegration = class {
     const toolSpan = state.toolSpans.get(event.toolCall.toolCallId);
     if (!toolSpan) return;
     state.toolCallCount += 1;
-    if (event.success) {
-      const outputAttrs = state.recordOutputs ? [attrString("ai.toolCall.result", safeJsonWithUint8(event.output))] : [];
+    const outcome = "toolOutput" in event ? event.toolOutput.type === "tool-result" ? { success: true, output: event.toolOutput.output } : { success: false, error: event.toolOutput.error } : event.success ? { success: true, output: event.output } : { success: false, error: event.error };
+    if (outcome.success) {
+      const outputAttrs = state.recordOutputs ? [attrString("ai.toolCall.result", safeJsonWithUint8(outcome.output))] : [];
       this.traceShipper.endSpan(toolSpan, { attributes: outputAttrs });
     } else {
-      this.traceShipper.endSpan(toolSpan, { error: event.error });
+      this.traceShipper.endSpan(toolSpan, { error: outcome.error });
     }
     this.emitLive(state, "tool_result", event.toolCall.toolName);
     state.toolSpans.delete(event.toolCall.toolCallId);
@@ -2437,7 +2505,7 @@ var RaindropTelemetryIntegration = class {
     }
   }
   finishGenerate(event, state) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
+    var _a, _b, _c, _d, _e, _f;
     if (state.rootSpan) {
       const outputAttrs = [];
       if (state.recordOutputs) {
@@ -2485,38 +2553,47 @@ var RaindropTelemetryIntegration = class {
       );
       this.traceShipper.endSpan(state.rootSpan, { attributes: outputAttrs });
     }
+    this.finalizeGenerateEvent(
+      state,
+      (_e = event.text) != null ? _e : state.accumulatedText || void 0,
+      (_f = event.response) == null ? void 0 : _f.modelId
+    );
+  }
+  /**
+   * Patch the Raindrop event for a completed, aborted, or errored text
+   * generation so it leaves the `isPending` state. Shared by `onFinish`/`onEnd`,
+   * `onAbort`, and `onError`.
+   */
+  finalizeGenerateEvent(state, output, model) {
+    var _a, _b, _c, _d, _e, _f, _g, _h;
     const suppressSubagentEvent = state.subagentName !== void 0 && state.subagentToolCallSpan !== void 0;
-    if (this.sendEvents && !suppressSubagentEvent) {
-      const callMeta = this.extractRaindropMetadata(state.metadata);
-      const userId = (_f = callMeta.userId) != null ? _f : (_e = this.defaultContext) == null ? void 0 : _e.userId;
-      if (userId) {
-        const eventName = (_i = (_h = callMeta.eventName) != null ? _h : (_g = this.defaultContext) == null ? void 0 : _g.eventName) != null ? _i : state.operationId;
-        const output = (_j = event.text) != null ? _j : state.accumulatedText || void 0;
-        const input = state.inputText;
-        const model = (_k = event.response) == null ? void 0 : _k.modelId;
-        const properties = {
-          ...(_l = this.defaultContext) == null ? void 0 : _l.properties,
-          ...callMeta.properties
-        };
-        const convoId = (_n = callMeta.convoId) != null ? _n : (_m = this.defaultContext) == null ? void 0 : _m.convoId;
-        void this.eventShipper.patch(state.eventId, {
-          eventName,
-          userId,
-          convoId,
-          input,
-          output,
-          model,
-          properties: Object.keys(properties).length > 0 ? properties : void 0,
-          isPending: false
-        }).catch((err) => {
-          if (this.debug) {
-            console.warn(
-              `[raindrop-ai/ai-sdk] event patch failed: ${err instanceof Error ? err.message : err}`
-            );
-          }
-        });
+    if (!this.sendEvents || suppressSubagentEvent) return;
+    const callMeta = this.extractRaindropMetadata(state.metadata);
+    const userId = (_b = callMeta.userId) != null ? _b : (_a = this.defaultContext) == null ? void 0 : _a.userId;
+    if (!userId) return;
+    const eventName = (_e = (_d = callMeta.eventName) != null ? _d : (_c = this.defaultContext) == null ? void 0 : _c.eventName) != null ? _e : state.operationId;
+    const input = state.inputText;
+    const properties = {
+      ...(_f = this.defaultContext) == null ? void 0 : _f.properties,
+      ...callMeta.properties
+    };
+    const convoId = (_h = callMeta.convoId) != null ? _h : (_g = this.defaultContext) == null ? void 0 : _g.convoId;
+    void this.eventShipper.patch(state.eventId, {
+      eventName,
+      userId,
+      convoId,
+      input,
+      output,
+      model,
+      properties: Object.keys(properties).length > 0 ? properties : void 0,
+      isPending: false
+    }).catch((err) => {
+      if (this.debug) {
+        console.warn(
+          `[raindrop-ai/ai-sdk] event patch failed: ${err instanceof Error ? err.message : err}`
+        );
       }
-    }
+    });
   }
   finishEmbed(event, state) {
     var _a;
@@ -4849,6 +4926,13 @@ function createRaindropAISDK(opts) {
     }
   };
 }
+function raindrop(options = {}) {
+  var _a;
+  const { context, subagentWrapping, writeKey, ...rest } = options;
+  const resolvedWriteKey = writeKey != null ? writeKey : typeof process !== "undefined" ? (_a = process.env) == null ? void 0 : _a.RAINDROP_WRITE_KEY : void 0;
+  const client = createRaindropAISDK({ ...rest, writeKey: resolvedWriteKey });
+  return client.createTelemetryIntegration({ context, subagentWrapping });
+}
 
 // src/index.workers.ts
 if (!globalThis.RAINDROP_ASYNC_LOCAL_STORAGE) {
@@ -4872,6 +4956,7 @@ exports.eventMetadataFromChatRequest = eventMetadataFromChatRequest;
 exports.getContextManager = getContextManager;
 exports.getCurrentParentToolContext = getCurrentParentToolContext;
 exports.getCurrentRaindropCallMetadata = getCurrentRaindropCallMetadata;
+exports.raindrop = raindrop;
 exports.readRaindropCallMetadataFromArgs = readRaindropCallMetadataFromArgs;
 exports.redactJsonAttributeValue = redactJsonAttributeValue;
 exports.redactSecretsInObject = redactSecretsInObject;

package/dist/index.workers.mjs

@@ -1,4 +1,4 @@
-export { DEFAULT_REDACT_ATTRIBUTE_KEYS, DEFAULT_SECRET_KEY_NAMES, REDACTED_PLACEHOLDER, RaindropTelemetryIntegration, _resetParentToolContextStorage, _resetRaindropCallMetadataStorage, _resetWarnedMissingUserId, clearParentToolContext, createRaindropAISDK, currentSpan, defaultTransformSpan, enterParentToolContext, eventMetadata, eventMetadataFromChatRequest, getContextManager, getCurrentParentToolContext, getCurrentRaindropCallMetadata, readRaindropCallMetadataFromArgs, redactJsonAttributeValue, redactSecretsInObject, runWithParentToolContext, runWithRaindropCallMetadata, withCurrent } from './chunk-YDMJ6ABM.mjs';
+export { DEFAULT_REDACT_ATTRIBUTE_KEYS, DEFAULT_SECRET_KEY_NAMES, REDACTED_PLACEHOLDER, RaindropTelemetryIntegration, _resetParentToolContextStorage, _resetRaindropCallMetadataStorage, _resetWarnedMissingUserId, clearParentToolContext, createRaindropAISDK, currentSpan, defaultTransformSpan, enterParentToolContext, eventMetadata, eventMetadataFromChatRequest, getContextManager, getCurrentParentToolContext, getCurrentRaindropCallMetadata, raindrop, readRaindropCallMetadataFromArgs, redactJsonAttributeValue, redactSecretsInObject, runWithParentToolContext, runWithRaindropCallMetadata, withCurrent } from './chunk-FIX66Y7M.mjs';
 import { AsyncLocalStorage } from 'async_hooks';
 
 if (!globalThis.RAINDROP_ASYNC_LOCAL_STORAGE) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@raindrop-ai/ai-sdk",
-  "version": "0.0.32",
+  "version": "0.0.33",
   "description": "Standalone Vercel AI SDK integration for Raindrop (events + OTLP/HTTP JSON traces, no OTEL runtime)",
   "main": "dist/index.node.js",
   "module": "dist/index.node.mjs",
@@ -62,7 +62,7 @@
     "test:v4": "pnpm build && cd tests/v4 && pnpm install --ignore-workspace --lockfile=false && pnpm test",
     "test:v5": "pnpm build && cd tests/v5 && pnpm install --ignore-workspace --lockfile=false && pnpm test",
     "test:v6": "pnpm build && cd tests/v6 && pnpm install --ignore-workspace --lockfile=false && pnpm test",
-    "test:v7": "pnpm build && cd tests/v7 && pnpm install --ignore-workspace --lockfile=false && pnpm test",
+    "test:v7": "pnpm build && cd tests/v7 && npm_config_minimum_release_age=0 pnpm install --ignore-workspace --lockfile=false && pnpm test",
     "test:install": "cd tests/v4 && pnpm install --ignore-workspace --lockfile=false && cd ../v5 && pnpm install --ignore-workspace --lockfile=false && cd ../v6 && pnpm install --ignore-workspace --lockfile=false && cd ../v7 && pnpm install --ignore-workspace --lockfile=false"
   }
 }