@raindrop-ai/ai-sdk 0.0.32 → 0.0.33

package/README.md

@@ -115,9 +115,50 @@ raindrop.traces.endSpan(agentTurn);
 await raindrop.flush();
 ```
 
-### AI SDK v7+ native telemetry (opt-in)
+### AI SDK v7 quick start: `registerTelemetry(raindrop())`
 
-On AI SDK v7+, you can use the native `Telemetry` callback interface (formerly `TelemetryIntegration` before beta.111) instead of Proxy wrapping. This avoids Proxy overhead and works with all AI SDK entry points (including `ToolLoopAgent`).
+On AI SDK v7, the simplest setup is the `raindrop()` factory, which returns a
+ready-to-register telemetry integration. The `writeKey` defaults to the
+`RAINDROP_WRITE_KEY` environment variable.
+
+```ts
+import { generateText, registerTelemetry } from "ai";
+import { OpenTelemetry } from "@ai-sdk/otel";
+import { raindrop } from "@raindrop-ai/ai-sdk";
+
+registerTelemetry(new OpenTelemetry(), raindrop());
+
+const result = await generateText({
+  model: anthropic("claude-sonnet-4-5"),
+  prompt: "what is the weather in Tokyo?",
+  telemetry: { functionId: "weather-agent" }, // no `isEnabled` needed in v7
+});
+```
+
+`raindrop()` is self-contained — `@ai-sdk/otel` is optional; you can register
+Raindrop on its own with `registerTelemetry(raindrop())`. Pass `context` and
+`subagentWrapping` to customise behaviour, e.g. `raindrop({ context: { userId: "user_123" } })`.
+
+For short-lived scripts that exit before the background flush timer fires, keep
+the reference and drain it before exiting:
+
+```ts
+const rd = raindrop();
+registerTelemetry(rd);
+// ... run your generations ...
+await rd.flush(); // or rd.shutdown()
+```
+
+Sub-agents (a `generateText`/`streamText` invoked from inside a tool's
+`execute`) nest automatically: the inner generation's spans are parented under
+the tool span so parent and child spans stay correctly nested.
+
+### AI SDK v7+ native telemetry (advanced)
+
+For finer control you can build the integration from a `createRaindropAISDK`
+client instead. The native `Telemetry` callback interface (formerly
+`TelemetryIntegration` before beta.111) avoids Proxy overhead and works with all
+AI SDK entry points (including `ToolLoopAgent`).
 
 ```ts
 // Option A: wrap() with nativeTelemetry flag
@@ -251,7 +292,7 @@ This package is tested against multiple Vercel AI SDK versions:
 | v4.x    | ✅ Supported | Proxy |
 | v5.x    | ✅ Supported | Proxy |
 | v6.x    | ✅ Supported | Proxy |
-| v7.x (beta) | ✅ Supported | Proxy (default) or native `Telemetry` (opt-in, formerly `TelemetryIntegration` pre-beta.111). Verified against beta.116. |
+| v7.x (beta/canary) | ✅ Supported | Proxy (default) or native `Telemetry` (opt-in, formerly `TelemetryIntegration` pre-beta.111). Verified against beta.116 and canary.171. |
 
 ### Version Differences Handled
 

package/dist/{chunk-YDMJ6ABM.mjs → chunk-FIX66Y7M.mjs}

@@ -1781,6 +1781,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;
@@ -2174,8 +2193,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,
@@ -2333,11 +2400,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);
@@ -2411,7 +2479,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) {
@@ -2459,38 +2527,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;
@@ -4575,7 +4652,7 @@ function extractNestedTokens(usage, key) {
 // 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;
@@ -4845,5 +4922,12 @@ 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 });
+}
 
-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 };
+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 };

package/dist/{index-CwK0GdEe.d.mts → index-Dcf4FPZL.d.mts}

@@ -365,6 +365,29 @@ declare class TraceShipper extends TraceShipper$1 {
 }
 
 type Listener<T> = (event: T) => PromiseLike<void> | void;
+type ToolExecutionEndEvent = {
+    callId: string;
+    toolCall: {
+        toolCallId: string;
+        toolName: string;
+    };
+    toolOutput: {
+        type: "tool-result";
+        output: unknown;
+    } | {
+        type: "tool-error";
+        error: unknown;
+    };
+} | {
+    callId: string;
+    toolCall: {
+        toolCallId: string;
+        toolName: string;
+    };
+    success: boolean;
+    output?: unknown;
+    error?: unknown;
+};
 interface TelemetryIntegration {
     onStart?: Listener<any>;
     onStepStart?: Listener<any>;
@@ -381,6 +404,7 @@ interface TelemetryIntegration {
     onEmbedEnd?: Listener<any>;
     onFinish?: Listener<any>;
     onEnd?: Listener<any>;
+    onAbort?: Listener<any>;
     onError?: Listener<any>;
     executeTool?: <T>(params: {
         callId: string;
@@ -441,6 +465,10 @@ declare class RaindropTelemetryIntegration implements TelemetryIntegration {
      */
     private readonly priorParentContexts;
     constructor(opts: RaindropTelemetryIntegrationOptions);
+    /** Flush any buffered events and trace spans to their destinations. */
+    flush: () => Promise<void>;
+    /** Flush and stop the background flush timers. */
+    shutdown: () => Promise<void>;
     private getState;
     private cleanup;
     private spanParentRef;
@@ -456,9 +484,9 @@ declare class RaindropTelemetryIntegration implements TelemetryIntegration {
     private toolExecutionStart;
     private toolExecutionEnd;
     onToolExecutionStart: (event: any) => void;
-    onToolExecutionEnd: (event: any) => void;
+    onToolExecutionEnd: (event: ToolExecutionEndEvent) => void;
     onToolCallStart: (event: any) => void;
-    onToolCallFinish: (event: any) => void;
+    onToolCallFinish: (event: ToolExecutionEndEvent) => void;
     onLanguageModelCallStart: (_event: any) => void;
     onLanguageModelCallEnd: (_event: any) => void;
     onChunk: (event: any) => void;
@@ -470,8 +498,15 @@ declare class RaindropTelemetryIntegration implements TelemetryIntegration {
     onFinish: (event: any) => void;
     onEnd: (event: any) => void;
     private finishGenerate;
+    /**
+     * 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`.
+     */
+    private finalizeGenerateEvent;
     private finishEmbed;
     onError: (error: unknown) => void;
+    onAbort: (event: any) => void;
     executeTool: <T>({ callId, toolCallId, execute, }: {
         callId: string;
         toolCallId: string;
@@ -1102,5 +1137,53 @@ type RaindropAISDKClient = {
     shutdown(): Promise<void>;
 };
 declare function createRaindropAISDK(opts: RaindropAISDKOptions): RaindropAISDKClient;
+/**
+ * Options for the {@link raindrop} telemetry-integration factory. A superset of
+ * {@link RaindropAISDKOptions} plus the per-integration `context` defaults and
+ * `subagentWrapping` toggle accepted by `createTelemetryIntegration()`.
+ */
+type RaindropTelemetryOptions = RaindropAISDKOptions & {
+    /** Default Raindrop context (userId/eventName/convoId/properties) applied to every call. */
+    context?: RaindropTelemetryIntegrationOptions["context"];
+    /**
+     * Wrap sub-agent generations (a `generateText`/`streamText` invoked from
+     * inside a tool's `execute`) in a synthetic tool span so parent and child
+     * spans nest correctly. Enabled by default.
+     */
+    subagentWrapping?: boolean;
+};
+/**
+ * One-line factory for the AI SDK v7 `registerTelemetry` DX.
+ *
+ * Returns a {@link RaindropTelemetryIntegration} wired to Raindrop's trace +
+ * event shippers, ready to hand to the AI SDK's `registerTelemetry`. The
+ * `writeKey` defaults to the `RAINDROP_WRITE_KEY` environment variable.
+ *
+ * @example
+ * ```ts
+ * import { generateText, registerTelemetry } from "ai";
+ * import { OpenTelemetry } from "@ai-sdk/otel";
+ * import { raindrop } from "@raindrop-ai/ai-sdk";
+ *
+ * registerTelemetry(new OpenTelemetry(), raindrop());
+ *
+ * await generateText({
+ *   model: anthropic("claude-sonnet-4-5"),
+ *   prompt: "what is the weather in Tokyo?",
+ *   telemetry: { functionId: "weather-agent" },
+ * });
+ * ```
+ *
+ * The returned integration also exposes `flush()` / `shutdown()` for
+ * short-lived scripts that need to drain buffered telemetry before exiting:
+ *
+ * ```ts
+ * const rd = raindrop();
+ * registerTelemetry(rd);
+ * // ...
+ * await rd.flush();
+ * ```
+ */
+declare function raindrop(options?: RaindropTelemetryOptions): RaindropTelemetryIntegration;
 
-export { redactJsonAttributeValue as $, type AISDKChatRequestLike as A, type BuildEventPatch as B, ContextManager as C, DEFAULT_REDACT_ATTRIBUTE_KEYS as D, type EndSpanArgs as E, type WrappedAISDK as F, _resetRaindropCallMetadataStorage as G, _resetWarnedMissingUserId as H, type IdentifyInput as I, clearParentToolContext as J, createRaindropAISDK as K, currentSpan as L, defaultTransformSpan as M, enterParentToolContext as N, type OtlpAnyValue as O, type ParentToolContext as P, eventMetadata as Q, REDACTED_PLACEHOLDER as R, type SelfDiagnosticsOptions as S, type TraceSpan as T, eventMetadataFromChatRequest as U, getContextManager as V, type WrapAISDKOptions as W, getCurrentParentToolContext as X, getCurrentRaindropCallMetadata as Y, readRaindropCallMetadataFromArgs as Z, _resetParentToolContextStorage as _, type AISDKChatRequestMessageLike as a, redactSecretsInObject as a0, runWithParentToolContext as a1, runWithRaindropCallMetadata as a2, withCurrent as a3, type AISDKMessage as b, type AgentCallMetadata as c, type AgentWithMetadata as d, type Attachment as e, type ContextSpan as f, type CreateSpanArgs as g, DEFAULT_SECRET_KEY_NAMES as h, type EventBuilder as i, type EventMetadataOptions as j, type OtlpSpan as k, type RaindropAISDKClient as l, type RaindropAISDKContext as m, type RaindropAISDKOptions as n, type RaindropCallMetadata as o, RaindropTelemetryIntegration as p, type RaindropTelemetryIntegrationOptions as q, type SelfDiagnosticsSignalDefinition as r, type SelfDiagnosticsSignalDefinitions as s, type SelfDiagnosticsTool as t, type SelfDiagnosticsToolInput as u, type SelfDiagnosticsToolOptions as v, type SelfDiagnosticsToolResult as w, type StartSpanArgs as x, type TransformSpanHook as y, type WrappedAI as z };
+export { raindrop as $, type AISDKChatRequestLike as A, type BuildEventPatch as B, ContextManager as C, DEFAULT_REDACT_ATTRIBUTE_KEYS as D, type EndSpanArgs as E, type WrappedAI as F, type WrappedAISDK as G, _resetRaindropCallMetadataStorage as H, type IdentifyInput as I, _resetWarnedMissingUserId as J, clearParentToolContext as K, createRaindropAISDK as L, currentSpan as M, defaultTransformSpan as N, type OtlpAnyValue as O, type ParentToolContext as P, enterParentToolContext as Q, REDACTED_PLACEHOLDER as R, type SelfDiagnosticsOptions as S, type TraceSpan as T, eventMetadata as U, eventMetadataFromChatRequest as V, type WrapAISDKOptions as W, getContextManager as X, getCurrentParentToolContext as Y, getCurrentRaindropCallMetadata as Z, _resetParentToolContextStorage as _, type AISDKChatRequestMessageLike as a, readRaindropCallMetadataFromArgs as a0, redactJsonAttributeValue as a1, redactSecretsInObject as a2, runWithParentToolContext as a3, runWithRaindropCallMetadata as a4, withCurrent as a5, type AISDKMessage as b, type AgentCallMetadata as c, type AgentWithMetadata as d, type Attachment as e, type ContextSpan as f, type CreateSpanArgs as g, DEFAULT_SECRET_KEY_NAMES as h, type EventBuilder as i, type EventMetadataOptions as j, type OtlpSpan as k, type RaindropAISDKClient as l, type RaindropAISDKContext as m, type RaindropAISDKOptions as n, type RaindropCallMetadata as o, RaindropTelemetryIntegration as p, type RaindropTelemetryIntegrationOptions as q, type RaindropTelemetryOptions as r, type SelfDiagnosticsSignalDefinition as s, type SelfDiagnosticsSignalDefinitions as t, type SelfDiagnosticsTool as u, type SelfDiagnosticsToolInput as v, type SelfDiagnosticsToolOptions as w, type SelfDiagnosticsToolResult as x, type StartSpanArgs as y, type TransformSpanHook as z };

package/dist/{index-CwK0GdEe.d.ts → index-Dcf4FPZL.d.ts}

@@ -365,6 +365,29 @@ declare class TraceShipper extends TraceShipper$1 {
 }
 
 type Listener<T> = (event: T) => PromiseLike<void> | void;
+type ToolExecutionEndEvent = {
+    callId: string;
+    toolCall: {
+        toolCallId: string;
+        toolName: string;
+    };
+    toolOutput: {
+        type: "tool-result";
+        output: unknown;
+    } | {
+        type: "tool-error";
+        error: unknown;
+    };
+} | {
+    callId: string;
+    toolCall: {
+        toolCallId: string;
+        toolName: string;
+    };
+    success: boolean;
+    output?: unknown;
+    error?: unknown;
+};
 interface TelemetryIntegration {
     onStart?: Listener<any>;
     onStepStart?: Listener<any>;
@@ -381,6 +404,7 @@ interface TelemetryIntegration {
     onEmbedEnd?: Listener<any>;
     onFinish?: Listener<any>;
     onEnd?: Listener<any>;
+    onAbort?: Listener<any>;
     onError?: Listener<any>;
     executeTool?: <T>(params: {
         callId: string;
@@ -441,6 +465,10 @@ declare class RaindropTelemetryIntegration implements TelemetryIntegration {
      */
     private readonly priorParentContexts;
     constructor(opts: RaindropTelemetryIntegrationOptions);
+    /** Flush any buffered events and trace spans to their destinations. */
+    flush: () => Promise<void>;
+    /** Flush and stop the background flush timers. */
+    shutdown: () => Promise<void>;
     private getState;
     private cleanup;
     private spanParentRef;
@@ -456,9 +484,9 @@ declare class RaindropTelemetryIntegration implements TelemetryIntegration {
     private toolExecutionStart;
     private toolExecutionEnd;
     onToolExecutionStart: (event: any) => void;
-    onToolExecutionEnd: (event: any) => void;
+    onToolExecutionEnd: (event: ToolExecutionEndEvent) => void;
     onToolCallStart: (event: any) => void;
-    onToolCallFinish: (event: any) => void;
+    onToolCallFinish: (event: ToolExecutionEndEvent) => void;
     onLanguageModelCallStart: (_event: any) => void;
     onLanguageModelCallEnd: (_event: any) => void;
     onChunk: (event: any) => void;
@@ -470,8 +498,15 @@ declare class RaindropTelemetryIntegration implements TelemetryIntegration {
     onFinish: (event: any) => void;
     onEnd: (event: any) => void;
     private finishGenerate;
+    /**
+     * 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`.
+     */
+    private finalizeGenerateEvent;
     private finishEmbed;
     onError: (error: unknown) => void;
+    onAbort: (event: any) => void;
     executeTool: <T>({ callId, toolCallId, execute, }: {
         callId: string;
         toolCallId: string;
@@ -1102,5 +1137,53 @@ type RaindropAISDKClient = {
     shutdown(): Promise<void>;
 };
 declare function createRaindropAISDK(opts: RaindropAISDKOptions): RaindropAISDKClient;
+/**
+ * Options for the {@link raindrop} telemetry-integration factory. A superset of
+ * {@link RaindropAISDKOptions} plus the per-integration `context` defaults and
+ * `subagentWrapping` toggle accepted by `createTelemetryIntegration()`.
+ */
+type RaindropTelemetryOptions = RaindropAISDKOptions & {
+    /** Default Raindrop context (userId/eventName/convoId/properties) applied to every call. */
+    context?: RaindropTelemetryIntegrationOptions["context"];
+    /**
+     * Wrap sub-agent generations (a `generateText`/`streamText` invoked from
+     * inside a tool's `execute`) in a synthetic tool span so parent and child
+     * spans nest correctly. Enabled by default.
+     */
+    subagentWrapping?: boolean;
+};
+/**
+ * One-line factory for the AI SDK v7 `registerTelemetry` DX.
+ *
+ * Returns a {@link RaindropTelemetryIntegration} wired to Raindrop's trace +
+ * event shippers, ready to hand to the AI SDK's `registerTelemetry`. The
+ * `writeKey` defaults to the `RAINDROP_WRITE_KEY` environment variable.
+ *
+ * @example
+ * ```ts
+ * import { generateText, registerTelemetry } from "ai";
+ * import { OpenTelemetry } from "@ai-sdk/otel";
+ * import { raindrop } from "@raindrop-ai/ai-sdk";
+ *
+ * registerTelemetry(new OpenTelemetry(), raindrop());
+ *
+ * await generateText({
+ *   model: anthropic("claude-sonnet-4-5"),
+ *   prompt: "what is the weather in Tokyo?",
+ *   telemetry: { functionId: "weather-agent" },
+ * });
+ * ```
+ *
+ * The returned integration also exposes `flush()` / `shutdown()` for
+ * short-lived scripts that need to drain buffered telemetry before exiting:
+ *
+ * ```ts
+ * const rd = raindrop();
+ * registerTelemetry(rd);
+ * // ...
+ * await rd.flush();
+ * ```
+ */
+declare function raindrop(options?: RaindropTelemetryOptions): RaindropTelemetryIntegration;
 
-export { redactJsonAttributeValue as $, type AISDKChatRequestLike as A, type BuildEventPatch as B, ContextManager as C, DEFAULT_REDACT_ATTRIBUTE_KEYS as D, type EndSpanArgs as E, type WrappedAISDK as F, _resetRaindropCallMetadataStorage as G, _resetWarnedMissingUserId as H, type IdentifyInput as I, clearParentToolContext as J, createRaindropAISDK as K, currentSpan as L, defaultTransformSpan as M, enterParentToolContext as N, type OtlpAnyValue as O, type ParentToolContext as P, eventMetadata as Q, REDACTED_PLACEHOLDER as R, type SelfDiagnosticsOptions as S, type TraceSpan as T, eventMetadataFromChatRequest as U, getContextManager as V, type WrapAISDKOptions as W, getCurrentParentToolContext as X, getCurrentRaindropCallMetadata as Y, readRaindropCallMetadataFromArgs as Z, _resetParentToolContextStorage as _, type AISDKChatRequestMessageLike as a, redactSecretsInObject as a0, runWithParentToolContext as a1, runWithRaindropCallMetadata as a2, withCurrent as a3, type AISDKMessage as b, type AgentCallMetadata as c, type AgentWithMetadata as d, type Attachment as e, type ContextSpan as f, type CreateSpanArgs as g, DEFAULT_SECRET_KEY_NAMES as h, type EventBuilder as i, type EventMetadataOptions as j, type OtlpSpan as k, type RaindropAISDKClient as l, type RaindropAISDKContext as m, type RaindropAISDKOptions as n, type RaindropCallMetadata as o, RaindropTelemetryIntegration as p, type RaindropTelemetryIntegrationOptions as q, type SelfDiagnosticsSignalDefinition as r, type SelfDiagnosticsSignalDefinitions as s, type SelfDiagnosticsTool as t, type SelfDiagnosticsToolInput as u, type SelfDiagnosticsToolOptions as v, type SelfDiagnosticsToolResult as w, type StartSpanArgs as x, type TransformSpanHook as y, type WrappedAI as z };
+export { raindrop as $, type AISDKChatRequestLike as A, type BuildEventPatch as B, ContextManager as C, DEFAULT_REDACT_ATTRIBUTE_KEYS as D, type EndSpanArgs as E, type WrappedAI as F, type WrappedAISDK as G, _resetRaindropCallMetadataStorage as H, type IdentifyInput as I, _resetWarnedMissingUserId as J, clearParentToolContext as K, createRaindropAISDK as L, currentSpan as M, defaultTransformSpan as N, type OtlpAnyValue as O, type ParentToolContext as P, enterParentToolContext as Q, REDACTED_PLACEHOLDER as R, type SelfDiagnosticsOptions as S, type TraceSpan as T, eventMetadata as U, eventMetadataFromChatRequest as V, type WrapAISDKOptions as W, getContextManager as X, getCurrentParentToolContext as Y, getCurrentRaindropCallMetadata as Z, _resetParentToolContextStorage as _, type AISDKChatRequestMessageLike as a, readRaindropCallMetadataFromArgs as a0, redactJsonAttributeValue as a1, redactSecretsInObject as a2, runWithParentToolContext as a3, runWithRaindropCallMetadata as a4, withCurrent as a5, type AISDKMessage as b, type AgentCallMetadata as c, type AgentWithMetadata as d, type Attachment as e, type ContextSpan as f, type CreateSpanArgs as g, DEFAULT_SECRET_KEY_NAMES as h, type EventBuilder as i, type EventMetadataOptions as j, type OtlpSpan as k, type RaindropAISDKClient as l, type RaindropAISDKContext as m, type RaindropAISDKOptions as n, type RaindropCallMetadata as o, RaindropTelemetryIntegration as p, type RaindropTelemetryIntegrationOptions as q, type RaindropTelemetryOptions as r, type SelfDiagnosticsSignalDefinition as s, type SelfDiagnosticsSignalDefinitions as t, type SelfDiagnosticsTool as u, type SelfDiagnosticsToolInput as v, type SelfDiagnosticsToolOptions as w, type SelfDiagnosticsToolResult as x, type StartSpanArgs as y, type TransformSpanHook as z };