@raindrop-ai/ai-sdk 0.0.31 → 0.0.33

package/dist/index.browser.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,12 +484,13 @@ 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;
+    private emitProviderExecutedToolSpans;
     onStepFinish: (event: any) => void;
     onEmbedStart: (event: any) => void;
     onEmbedFinish: (event: any) => void;
@@ -469,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;
@@ -893,6 +929,38 @@ type SelfDiagnosticsOptions = {
      */
     toolName?: string;
 };
+type SelfDiagnosticsToolOptions = Omit<SelfDiagnosticsOptions, "enabled"> & {
+    /** Fixed Raindrop event ID to attach diagnostic signals to. */
+    eventId?: string;
+    /** Dynamic event ID resolver, useful when one tool instance serves many runs. */
+    getEventId?: () => string | undefined;
+    /** Behavior when no event ID can be resolved. Default: "warn" */
+    onMissingEventId?: "warn" | "ignore" | "throw";
+};
+type SelfDiagnosticsToolInput = {
+    category: string;
+    detail: string;
+};
+type SelfDiagnosticsToolResult = {
+    acknowledged: true;
+    category: string;
+} | {
+    acknowledged: false;
+    category: string;
+    reason: "missing_event_id";
+};
+type SelfDiagnosticsTool = {
+    name: string;
+    description: string;
+    /**
+     * AI SDK v4 tool schema. This is the same cross-version schema object as
+     * `inputSchema`; both fields are exposed so the tool works on AI SDK v4-v7.
+     */
+    parameters: any;
+    /** AI SDK v5+ tool schema. */
+    inputSchema: any;
+    execute(input: SelfDiagnosticsToolInput): Promise<SelfDiagnosticsToolResult>;
+};
 type WrapAISDKOptions = {
     context: RaindropAISDKContext | ((info: {
         operation: string;
@@ -977,6 +1045,14 @@ type EventPatch = {
 };
 type RaindropAISDKClient = {
     wrap<T extends object>(aiSDK: T, options?: WrapAISDKOptions): WrappedAISDK<T>;
+    /**
+     * Create a standalone AI SDK self-diagnostics tool.
+     *
+     * The tool automatically uses the active Raindrop event while executing
+     * under `createTelemetryIntegration()` with tracing enabled. Pass `eventId`
+     * or `getEventId` when no active trace context is available.
+     */
+    createSelfDiagnosticsTool(options?: SelfDiagnosticsToolOptions): SelfDiagnosticsTool;
     /**
      * Create a TelemetryIntegration instance for direct registration with AI SDK v7+.
      * Use with `registerTelemetryIntegration()` from the `ai` package.
@@ -1061,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 { type AISDKChatRequestLike, type AISDKChatRequestMessageLike, type AISDKMessage, type AgentCallMetadata, type AgentWithMetadata, type Attachment, type BuildEventPatch, ContextManager, type ContextSpan, type CreateSpanArgs, DEFAULT_REDACT_ATTRIBUTE_KEYS, DEFAULT_SECRET_KEY_NAMES, type EndSpanArgs, type EventBuilder, type EventMetadataOptions, type IdentifyInput, type OtlpAnyValue, type OtlpSpan, type ParentToolContext, REDACTED_PLACEHOLDER, type RaindropAISDKClient, type RaindropAISDKContext, type RaindropAISDKOptions, type RaindropCallMetadata, RaindropTelemetryIntegration, type RaindropTelemetryIntegrationOptions, type SelfDiagnosticsOptions, type SelfDiagnosticsSignalDefinition, type SelfDiagnosticsSignalDefinitions, type StartSpanArgs, type TraceSpan, type TransformSpanHook, type WrapAISDKOptions, type WrappedAI, type WrappedAISDK, _resetParentToolContextStorage, _resetRaindropCallMetadataStorage, _resetWarnedMissingUserId, clearParentToolContext, createRaindropAISDK, currentSpan, defaultTransformSpan, enterParentToolContext, eventMetadata, eventMetadataFromChatRequest, getContextManager, getCurrentParentToolContext, getCurrentRaindropCallMetadata, readRaindropCallMetadataFromArgs, redactJsonAttributeValue, redactSecretsInObject, runWithParentToolContext, runWithRaindropCallMetadata, withCurrent };
+export { type AISDKChatRequestLike, type AISDKChatRequestMessageLike, type AISDKMessage, type AgentCallMetadata, type AgentWithMetadata, type Attachment, type BuildEventPatch, ContextManager, type ContextSpan, type CreateSpanArgs, DEFAULT_REDACT_ATTRIBUTE_KEYS, DEFAULT_SECRET_KEY_NAMES, type EndSpanArgs, type EventBuilder, type EventMetadataOptions, type IdentifyInput, type OtlpAnyValue, type OtlpSpan, type ParentToolContext, REDACTED_PLACEHOLDER, type RaindropAISDKClient, type RaindropAISDKContext, type RaindropAISDKOptions, type RaindropCallMetadata, RaindropTelemetryIntegration, type RaindropTelemetryIntegrationOptions, type RaindropTelemetryOptions, type SelfDiagnosticsOptions, type SelfDiagnosticsSignalDefinition, type SelfDiagnosticsSignalDefinitions, type SelfDiagnosticsTool, type SelfDiagnosticsToolInput, type SelfDiagnosticsToolOptions, type SelfDiagnosticsToolResult, type StartSpanArgs, type TraceSpan, type TransformSpanHook, type WrapAISDKOptions, type WrappedAI, type WrappedAISDK, _resetParentToolContextStorage, _resetRaindropCallMetadataStorage, _resetWarnedMissingUserId, clearParentToolContext, createRaindropAISDK, currentSpan, defaultTransformSpan, enterParentToolContext, eventMetadata, eventMetadataFromChatRequest, getContextManager, getCurrentParentToolContext, getCurrentRaindropCallMetadata, raindrop, readRaindropCallMetadataFromArgs, redactJsonAttributeValue, redactSecretsInObject, runWithParentToolContext, runWithRaindropCallMetadata, withCurrent };