@raindrop-ai/ai-sdk 0.0.19 → 0.0.20

package/dist/index.browser.d.mts

@@ -1,7 +1,188 @@
-type IdentifyInput = {
+type OtlpAnyValue = {
+    stringValue?: string;
+    intValue?: string;
+    doubleValue?: number;
+    boolValue?: boolean;
+    arrayValue?: {
+        values: OtlpAnyValue[];
+    };
+};
+type OtlpKeyValue = {
+    key: string;
+    value: OtlpAnyValue;
+};
+declare const SpanStatusCode: {
+    readonly UNSET: 0;
+    readonly OK: 1;
+    readonly ERROR: 2;
+};
+type OtlpSpanStatus = {
+    code: (typeof SpanStatusCode)[keyof typeof SpanStatusCode] | number;
+    message?: string;
+};
+type OtlpSpan = {
+    traceId: string;
+    spanId: string;
+    parentSpanId?: string;
+    name: string;
+    startTimeUnixNano: string;
+    endTimeUnixNano: string;
+    attributes?: OtlpKeyValue[];
+    status?: OtlpSpanStatus;
+};
+type SpanIds = {
+    traceIdB64: string;
+    spanIdB64: string;
+    parentSpanIdB64?: string;
+};
+
+type Attachment$1 = {
+    type: string;
+    role: string;
+    name?: string;
+    value: string;
+};
+type IdentifyInput$1 = {
     userId: string;
     traits?: Record<string, unknown>;
 };
+type Patch = {
+    eventName?: string;
+    userId?: string;
+    convoId?: string;
+    input?: string;
+    output?: string;
+    model?: string;
+    properties?: Record<string, unknown>;
+    attachments?: Attachment$1[];
+    isPending?: boolean;
+    timestamp?: string;
+};
+type SignalInput = {
+    eventId: string;
+    name: string;
+    type?: "default" | "feedback" | "edit" | "standard" | "agent" | "agent_internal";
+    sentiment?: "POSITIVE" | "NEGATIVE";
+    timestamp?: string;
+    properties?: Record<string, unknown>;
+    attachmentId?: string;
+    comment?: string;
+    after?: string;
+};
+type EventShipperOptions = {
+    writeKey?: string;
+    endpoint?: string;
+    enabled?: boolean;
+    debug: boolean;
+    partialFlushMs?: number;
+    sdkName?: string;
+    libraryName?: string;
+    libraryVersion?: string;
+    defaultEventName?: string;
+};
+declare class EventShipper$1 {
+    private baseUrl;
+    private writeKey?;
+    private enabled;
+    private debug;
+    private partialFlushMs;
+    private sdkName;
+    private prefix;
+    private defaultEventName;
+    private context;
+    private buffers;
+    private sticky;
+    private timers;
+    private inFlight;
+    constructor(opts: EventShipperOptions);
+    isDebugEnabled(): boolean;
+    private authHeaders;
+    patch(eventId: string, patch: Patch): Promise<void>;
+    finish(eventId: string, patch: {
+        output?: string;
+        model?: string;
+        properties?: Record<string, unknown>;
+        userId?: string;
+    }): Promise<void>;
+    flush(): Promise<void>;
+    shutdown(): Promise<void>;
+    trackSignal(signal: SignalInput): Promise<void>;
+    identify(users: IdentifyInput$1 | IdentifyInput$1[]): Promise<void>;
+    private flushOne;
+}
+
+type InternalSpan = {
+    ids: SpanIds;
+    name: string;
+    startTimeUnixNano: string;
+    endTimeUnixNano?: string;
+    attributes: Array<OtlpKeyValue | undefined>;
+};
+type TraceShipperOptions = {
+    writeKey?: string;
+    endpoint?: string;
+    enabled?: boolean;
+    debug: boolean;
+    debugSpans?: boolean;
+    flushIntervalMs?: number;
+    maxBatchSize?: number;
+    maxQueueSize?: number;
+    sdkName?: string;
+    serviceName?: string;
+    serviceVersion?: string;
+};
+declare class TraceShipper$1 {
+    private baseUrl;
+    private writeKey?;
+    private enabled;
+    private debug;
+    private debugSpans;
+    private sdkName;
+    private prefix;
+    private serviceName;
+    private serviceVersion;
+    private flushIntervalMs;
+    private maxBatchSize;
+    private maxQueueSize;
+    private queue;
+    private timer;
+    private inFlight;
+    constructor(opts: TraceShipperOptions);
+    isDebugEnabled(): boolean;
+    private authHeaders;
+    startSpan(args: {
+        name: string;
+        parent?: {
+            traceIdB64: string;
+            spanIdB64: string;
+        };
+        eventId: string;
+        operationId?: string;
+        attributes?: Array<OtlpKeyValue | undefined>;
+        startTimeUnixNano?: string;
+    }): InternalSpan;
+    endSpan(span: InternalSpan, extra?: {
+        attributes?: InternalSpan["attributes"];
+        error?: unknown;
+        status?: OtlpSpanStatus;
+        endTimeUnixNano?: string;
+    }): void;
+    createSpan(args: {
+        name: string;
+        parent?: {
+            traceIdB64: string;
+            spanIdB64: string;
+        };
+        eventId: string;
+        startTimeUnixNano: string;
+        endTimeUnixNano: string;
+        attributes?: Array<OtlpKeyValue | undefined>;
+        status?: OtlpSpanStatus;
+    }): void;
+    enqueue(span: OtlpSpan): void;
+    flush(): Promise<void>;
+    shutdown(): Promise<void>;
+}
 
 type ParentSpanContext = {
     traceIdB64: string;
@@ -33,6 +214,98 @@ declare function getContextManager(): ContextManager;
 declare function currentSpan(): ContextSpan;
 declare function withCurrent<R>(span: ContextSpan, callback: () => R): R;
 
+declare class EventShipper extends EventShipper$1 {
+    constructor(opts: ConstructorParameters<typeof EventShipper$1>[0]);
+}
+
+declare class TraceShipper extends TraceShipper$1 {
+    constructor(opts: ConstructorParameters<typeof TraceShipper$1>[0]);
+}
+
+/**
+ * Raindrop TelemetryIntegration for AI SDK v7+
+ *
+ * Implements the AI SDK's TelemetryIntegration interface to capture traces and
+ * events natively, replacing the Proxy-based wrapping used for v4-v6.
+ *
+ * Modeled after the upstream OpenTelemetryIntegration but uses Raindrop's
+ * TraceShipper (OTLP/HTTP) + EventShipper instead of the OTel Tracer API.
+ */
+
+type Listener<T> = (event: T) => PromiseLike<void> | void;
+interface TelemetryIntegration {
+    onStart?: Listener<any>;
+    onStepStart?: Listener<any>;
+    onToolCallStart?: Listener<any>;
+    onToolCallFinish?: Listener<any>;
+    onChunk?: Listener<any>;
+    onStepFinish?: Listener<any>;
+    onEmbedStart?: Listener<any>;
+    onEmbedFinish?: Listener<any>;
+    onFinish?: Listener<any>;
+    onError?: Listener<any>;
+    executeTool?: <T>(params: {
+        callId: string;
+        toolCallId: string;
+        execute: () => PromiseLike<T>;
+    }) => PromiseLike<T>;
+}
+type RaindropTelemetryIntegrationOptions = {
+    traceShipper: TraceShipper;
+    eventShipper: EventShipper;
+    sendTraces?: boolean;
+    sendEvents?: boolean;
+    debug?: boolean;
+    context?: {
+        userId?: string;
+        eventId?: string;
+        eventName?: string;
+        convoId?: string;
+        properties?: Record<string, unknown>;
+    };
+};
+declare class RaindropTelemetryIntegration implements TelemetryIntegration {
+    private readonly traceShipper;
+    private readonly eventShipper;
+    private readonly sendTraces;
+    private readonly sendEvents;
+    private readonly debug;
+    private readonly defaultContext;
+    private readonly callStates;
+    constructor(opts: RaindropTelemetryIntegrationOptions);
+    private getState;
+    private cleanup;
+    private spanParentRef;
+    private extractRaindropMetadata;
+    /**
+     * Extract the user-facing input text from an onStart event.
+     * Mirrors the logic in the v4-v6 Proxy path (lastUserMessageTextFromArgs / extractInputFromArgs).
+     */
+    private extractInputText;
+    onStart: (event: any) => void;
+    onStepStart: (event: any) => void;
+    onToolCallStart: (event: any) => void;
+    onToolCallFinish: (event: any) => void;
+    onChunk: (event: any) => void;
+    onStepFinish: (event: any) => void;
+    onEmbedStart: (event: any) => void;
+    onEmbedFinish: (event: any) => void;
+    onFinish: (event: any) => void;
+    private finishGenerate;
+    private finishEmbed;
+    onError: (error: unknown) => void;
+    executeTool: <T>({ callId, toolCallId, execute, }: {
+        callId: string;
+        toolCallId: string;
+        execute: () => PromiseLike<T>;
+    }) => Promise<T>;
+}
+
+type IdentifyInput = {
+    userId: string;
+    traits?: Record<string, unknown>;
+};
+
 declare function _resetWarnedMissingUserId(): void;
 
 /**
@@ -264,6 +537,16 @@ type WrapAISDKOptions = {
         events?: boolean;
         traces?: boolean;
     };
+    /**
+     * Use the AI SDK's native TelemetryIntegration instead of Proxy-based wrapping.
+     * Requires AI SDK v7+. Throws on older versions.
+     *
+     * When enabled, `wrap()` registers a `RaindropTelemetryIntegration` globally
+     * and returns the original module instead of a Proxy.
+     *
+     * Default: false (uses Proxy wrapping for backward compatibility).
+     */
+    nativeTelemetry?: boolean;
 };
 type EventPatch = {
     eventName?: string;
@@ -279,6 +562,11 @@ type EventPatch = {
 };
 type RaindropAISDKClient = {
     wrap<T extends object>(aiSDK: T, options?: WrapAISDKOptions): WrappedAISDK<T>;
+    /**
+     * Create a TelemetryIntegration instance for direct registration with AI SDK v7+.
+     * Use with `registerTelemetryIntegration()` from the `ai` package.
+     */
+    createTelemetryIntegration(context?: RaindropTelemetryIntegrationOptions["context"]): RaindropTelemetryIntegration;
     events: {
         patch(eventId: string, patch: EventPatch): Promise<void>;
         addAttachments(eventId: string, attachments: Attachment[]): Promise<void>;
@@ -310,4 +598,4 @@ type RaindropAISDKClient = {
 };
 declare function createRaindropAISDK(opts: RaindropAISDKOptions): RaindropAISDKClient;
 
-export { type AISDKChatRequestLike, type AISDKChatRequestMessageLike, type AISDKMessage, type AgentCallMetadata, type AgentWithMetadata, type Attachment, type BuildEventPatch, ContextManager, type ContextSpan, type EventBuilder, type EventMetadataOptions, type IdentifyInput, type RaindropAISDKClient, type RaindropAISDKContext, type RaindropAISDKOptions, type SelfDiagnosticsOptions, type SelfDiagnosticsSignalDefinition, type SelfDiagnosticsSignalDefinitions, type WrapAISDKOptions, type WrappedAI, type WrappedAISDK, _resetWarnedMissingUserId, createRaindropAISDK, currentSpan, eventMetadata, eventMetadataFromChatRequest, getContextManager, withCurrent };
+export { type AISDKChatRequestLike, type AISDKChatRequestMessageLike, type AISDKMessage, type AgentCallMetadata, type AgentWithMetadata, type Attachment, type BuildEventPatch, ContextManager, type ContextSpan, type EventBuilder, type EventMetadataOptions, type IdentifyInput, type RaindropAISDKClient, type RaindropAISDKContext, type RaindropAISDKOptions, RaindropTelemetryIntegration, type RaindropTelemetryIntegrationOptions, type SelfDiagnosticsOptions, type SelfDiagnosticsSignalDefinition, type SelfDiagnosticsSignalDefinitions, type WrapAISDKOptions, type WrappedAI, type WrappedAISDK, _resetWarnedMissingUserId, createRaindropAISDK, currentSpan, eventMetadata, eventMetadataFromChatRequest, getContextManager, withCurrent };

package/dist/index.browser.d.ts

@@ -1,7 +1,188 @@
-type IdentifyInput = {
+type OtlpAnyValue = {
+    stringValue?: string;
+    intValue?: string;
+    doubleValue?: number;
+    boolValue?: boolean;
+    arrayValue?: {
+        values: OtlpAnyValue[];
+    };
+};
+type OtlpKeyValue = {
+    key: string;
+    value: OtlpAnyValue;
+};
+declare const SpanStatusCode: {
+    readonly UNSET: 0;
+    readonly OK: 1;
+    readonly ERROR: 2;
+};
+type OtlpSpanStatus = {
+    code: (typeof SpanStatusCode)[keyof typeof SpanStatusCode] | number;
+    message?: string;
+};
+type OtlpSpan = {
+    traceId: string;
+    spanId: string;
+    parentSpanId?: string;
+    name: string;
+    startTimeUnixNano: string;
+    endTimeUnixNano: string;
+    attributes?: OtlpKeyValue[];
+    status?: OtlpSpanStatus;
+};
+type SpanIds = {
+    traceIdB64: string;
+    spanIdB64: string;
+    parentSpanIdB64?: string;
+};
+
+type Attachment$1 = {
+    type: string;
+    role: string;
+    name?: string;
+    value: string;
+};
+type IdentifyInput$1 = {
     userId: string;
     traits?: Record<string, unknown>;
 };
+type Patch = {
+    eventName?: string;
+    userId?: string;
+    convoId?: string;
+    input?: string;
+    output?: string;
+    model?: string;
+    properties?: Record<string, unknown>;
+    attachments?: Attachment$1[];
+    isPending?: boolean;
+    timestamp?: string;
+};
+type SignalInput = {
+    eventId: string;
+    name: string;
+    type?: "default" | "feedback" | "edit" | "standard" | "agent" | "agent_internal";
+    sentiment?: "POSITIVE" | "NEGATIVE";
+    timestamp?: string;
+    properties?: Record<string, unknown>;
+    attachmentId?: string;
+    comment?: string;
+    after?: string;
+};
+type EventShipperOptions = {
+    writeKey?: string;
+    endpoint?: string;
+    enabled?: boolean;
+    debug: boolean;
+    partialFlushMs?: number;
+    sdkName?: string;
+    libraryName?: string;
+    libraryVersion?: string;
+    defaultEventName?: string;
+};
+declare class EventShipper$1 {
+    private baseUrl;
+    private writeKey?;
+    private enabled;
+    private debug;
+    private partialFlushMs;
+    private sdkName;
+    private prefix;
+    private defaultEventName;
+    private context;
+    private buffers;
+    private sticky;
+    private timers;
+    private inFlight;
+    constructor(opts: EventShipperOptions);
+    isDebugEnabled(): boolean;
+    private authHeaders;
+    patch(eventId: string, patch: Patch): Promise<void>;
+    finish(eventId: string, patch: {
+        output?: string;
+        model?: string;
+        properties?: Record<string, unknown>;
+        userId?: string;
+    }): Promise<void>;
+    flush(): Promise<void>;
+    shutdown(): Promise<void>;
+    trackSignal(signal: SignalInput): Promise<void>;
+    identify(users: IdentifyInput$1 | IdentifyInput$1[]): Promise<void>;
+    private flushOne;
+}
+
+type InternalSpan = {
+    ids: SpanIds;
+    name: string;
+    startTimeUnixNano: string;
+    endTimeUnixNano?: string;
+    attributes: Array<OtlpKeyValue | undefined>;
+};
+type TraceShipperOptions = {
+    writeKey?: string;
+    endpoint?: string;
+    enabled?: boolean;
+    debug: boolean;
+    debugSpans?: boolean;
+    flushIntervalMs?: number;
+    maxBatchSize?: number;
+    maxQueueSize?: number;
+    sdkName?: string;
+    serviceName?: string;
+    serviceVersion?: string;
+};
+declare class TraceShipper$1 {
+    private baseUrl;
+    private writeKey?;
+    private enabled;
+    private debug;
+    private debugSpans;
+    private sdkName;
+    private prefix;
+    private serviceName;
+    private serviceVersion;
+    private flushIntervalMs;
+    private maxBatchSize;
+    private maxQueueSize;
+    private queue;
+    private timer;
+    private inFlight;
+    constructor(opts: TraceShipperOptions);
+    isDebugEnabled(): boolean;
+    private authHeaders;
+    startSpan(args: {
+        name: string;
+        parent?: {
+            traceIdB64: string;
+            spanIdB64: string;
+        };
+        eventId: string;
+        operationId?: string;
+        attributes?: Array<OtlpKeyValue | undefined>;
+        startTimeUnixNano?: string;
+    }): InternalSpan;
+    endSpan(span: InternalSpan, extra?: {
+        attributes?: InternalSpan["attributes"];
+        error?: unknown;
+        status?: OtlpSpanStatus;
+        endTimeUnixNano?: string;
+    }): void;
+    createSpan(args: {
+        name: string;
+        parent?: {
+            traceIdB64: string;
+            spanIdB64: string;
+        };
+        eventId: string;
+        startTimeUnixNano: string;
+        endTimeUnixNano: string;
+        attributes?: Array<OtlpKeyValue | undefined>;
+        status?: OtlpSpanStatus;
+    }): void;
+    enqueue(span: OtlpSpan): void;
+    flush(): Promise<void>;
+    shutdown(): Promise<void>;
+}
 
 type ParentSpanContext = {
     traceIdB64: string;
@@ -33,6 +214,98 @@ declare function getContextManager(): ContextManager;
 declare function currentSpan(): ContextSpan;
 declare function withCurrent<R>(span: ContextSpan, callback: () => R): R;
 
+declare class EventShipper extends EventShipper$1 {
+    constructor(opts: ConstructorParameters<typeof EventShipper$1>[0]);
+}
+
+declare class TraceShipper extends TraceShipper$1 {
+    constructor(opts: ConstructorParameters<typeof TraceShipper$1>[0]);
+}
+
+/**
+ * Raindrop TelemetryIntegration for AI SDK v7+
+ *
+ * Implements the AI SDK's TelemetryIntegration interface to capture traces and
+ * events natively, replacing the Proxy-based wrapping used for v4-v6.
+ *
+ * Modeled after the upstream OpenTelemetryIntegration but uses Raindrop's
+ * TraceShipper (OTLP/HTTP) + EventShipper instead of the OTel Tracer API.
+ */
+
+type Listener<T> = (event: T) => PromiseLike<void> | void;
+interface TelemetryIntegration {
+    onStart?: Listener<any>;
+    onStepStart?: Listener<any>;
+    onToolCallStart?: Listener<any>;
+    onToolCallFinish?: Listener<any>;
+    onChunk?: Listener<any>;
+    onStepFinish?: Listener<any>;
+    onEmbedStart?: Listener<any>;
+    onEmbedFinish?: Listener<any>;
+    onFinish?: Listener<any>;
+    onError?: Listener<any>;
+    executeTool?: <T>(params: {
+        callId: string;
+        toolCallId: string;
+        execute: () => PromiseLike<T>;
+    }) => PromiseLike<T>;
+}
+type RaindropTelemetryIntegrationOptions = {
+    traceShipper: TraceShipper;
+    eventShipper: EventShipper;
+    sendTraces?: boolean;
+    sendEvents?: boolean;
+    debug?: boolean;
+    context?: {
+        userId?: string;
+        eventId?: string;
+        eventName?: string;
+        convoId?: string;
+        properties?: Record<string, unknown>;
+    };
+};
+declare class RaindropTelemetryIntegration implements TelemetryIntegration {
+    private readonly traceShipper;
+    private readonly eventShipper;
+    private readonly sendTraces;
+    private readonly sendEvents;
+    private readonly debug;
+    private readonly defaultContext;
+    private readonly callStates;
+    constructor(opts: RaindropTelemetryIntegrationOptions);
+    private getState;
+    private cleanup;
+    private spanParentRef;
+    private extractRaindropMetadata;
+    /**
+     * Extract the user-facing input text from an onStart event.
+     * Mirrors the logic in the v4-v6 Proxy path (lastUserMessageTextFromArgs / extractInputFromArgs).
+     */
+    private extractInputText;
+    onStart: (event: any) => void;
+    onStepStart: (event: any) => void;
+    onToolCallStart: (event: any) => void;
+    onToolCallFinish: (event: any) => void;
+    onChunk: (event: any) => void;
+    onStepFinish: (event: any) => void;
+    onEmbedStart: (event: any) => void;
+    onEmbedFinish: (event: any) => void;
+    onFinish: (event: any) => void;
+    private finishGenerate;
+    private finishEmbed;
+    onError: (error: unknown) => void;
+    executeTool: <T>({ callId, toolCallId, execute, }: {
+        callId: string;
+        toolCallId: string;
+        execute: () => PromiseLike<T>;
+    }) => Promise<T>;
+}
+
+type IdentifyInput = {
+    userId: string;
+    traits?: Record<string, unknown>;
+};
+
 declare function _resetWarnedMissingUserId(): void;
 
 /**
@@ -264,6 +537,16 @@ type WrapAISDKOptions = {
         events?: boolean;
         traces?: boolean;
     };
+    /**
+     * Use the AI SDK's native TelemetryIntegration instead of Proxy-based wrapping.
+     * Requires AI SDK v7+. Throws on older versions.
+     *
+     * When enabled, `wrap()` registers a `RaindropTelemetryIntegration` globally
+     * and returns the original module instead of a Proxy.
+     *
+     * Default: false (uses Proxy wrapping for backward compatibility).
+     */
+    nativeTelemetry?: boolean;
 };
 type EventPatch = {
     eventName?: string;
@@ -279,6 +562,11 @@ type EventPatch = {
 };
 type RaindropAISDKClient = {
     wrap<T extends object>(aiSDK: T, options?: WrapAISDKOptions): WrappedAISDK<T>;
+    /**
+     * Create a TelemetryIntegration instance for direct registration with AI SDK v7+.
+     * Use with `registerTelemetryIntegration()` from the `ai` package.
+     */
+    createTelemetryIntegration(context?: RaindropTelemetryIntegrationOptions["context"]): RaindropTelemetryIntegration;
     events: {
         patch(eventId: string, patch: EventPatch): Promise<void>;
         addAttachments(eventId: string, attachments: Attachment[]): Promise<void>;
@@ -310,4 +598,4 @@ type RaindropAISDKClient = {
 };
 declare function createRaindropAISDK(opts: RaindropAISDKOptions): RaindropAISDKClient;
 
-export { type AISDKChatRequestLike, type AISDKChatRequestMessageLike, type AISDKMessage, type AgentCallMetadata, type AgentWithMetadata, type Attachment, type BuildEventPatch, ContextManager, type ContextSpan, type EventBuilder, type EventMetadataOptions, type IdentifyInput, type RaindropAISDKClient, type RaindropAISDKContext, type RaindropAISDKOptions, type SelfDiagnosticsOptions, type SelfDiagnosticsSignalDefinition, type SelfDiagnosticsSignalDefinitions, type WrapAISDKOptions, type WrappedAI, type WrappedAISDK, _resetWarnedMissingUserId, createRaindropAISDK, currentSpan, eventMetadata, eventMetadataFromChatRequest, getContextManager, withCurrent };
+export { type AISDKChatRequestLike, type AISDKChatRequestMessageLike, type AISDKMessage, type AgentCallMetadata, type AgentWithMetadata, type Attachment, type BuildEventPatch, ContextManager, type ContextSpan, type EventBuilder, type EventMetadataOptions, type IdentifyInput, type RaindropAISDKClient, type RaindropAISDKContext, type RaindropAISDKOptions, RaindropTelemetryIntegration, type RaindropTelemetryIntegrationOptions, type SelfDiagnosticsOptions, type SelfDiagnosticsSignalDefinition, type SelfDiagnosticsSignalDefinitions, type WrapAISDKOptions, type WrappedAI, type WrappedAISDK, _resetWarnedMissingUserId, createRaindropAISDK, currentSpan, eventMetadata, eventMetadataFromChatRequest, getContextManager, withCurrent };