rd-mini 0.1.0

package/dist/wrappers/ai-sdk.js

@@ -0,0 +1,295 @@
+/**
+ * Vercel AI SDK Wrapper
+ * Wraps AI SDK model providers to auto-capture all calls
+ */
+/**
+ * Wrap a Vercel AI SDK model to auto-trace calls
+ *
+ * Usage:
+ *   const model = raindrop.wrap(openai('gpt-4o'));
+ *   const result = await generateText({ model, prompt: '...' });
+ */
+export function wrapAISDKModel(model, context, defaultOptions) {
+    // Create a proxy that intercepts method calls
+    const handler = {
+        get(target, prop, receiver) {
+            const value = Reflect.get(target, prop, receiver);
+            // Intercept doGenerate (non-streaming)
+            if (prop === 'doGenerate' && typeof value === 'function') {
+                return async function (options) {
+                    const traceId = (defaultOptions?.traceId) || context.generateTraceId();
+                    const startTime = Date.now();
+                    const userId = defaultOptions?.userId || context.getUserId();
+                    if (context.debug) {
+                        console.log('[raindrop] AI SDK generate started:', traceId);
+                    }
+                    try {
+                        const result = await value.call(target, options);
+                        const endTime = Date.now();
+                        // Extract data from result
+                        const text = result.text || '';
+                        const usage = result.usage;
+                        const toolCalls = result.toolCalls;
+                        // Check if we're within an interaction
+                        const interaction = context.getInteractionContext();
+                        if (interaction) {
+                            // Add as span to the interaction
+                            const span = {
+                                spanId: traceId,
+                                parentId: interaction.interactionId,
+                                name: `ai-sdk:${target.modelId}`,
+                                type: 'ai',
+                                startTime,
+                                endTime,
+                                latencyMs: endTime - startTime,
+                                input: options.prompt || options.messages,
+                                output: text,
+                                properties: {
+                                    ...defaultOptions?.properties,
+                                    input_tokens: usage?.promptTokens,
+                                    output_tokens: usage?.completionTokens,
+                                    tool_calls: toolCalls?.map(tc => ({ name: tc.toolName, arguments: tc.args })),
+                                },
+                            };
+                            interaction.spans.push(span);
+                        }
+                        else {
+                            context.sendTrace({
+                                traceId,
+                                provider: 'ai-sdk',
+                                model: target.modelId,
+                                input: options.prompt || options.messages,
+                                output: text,
+                                startTime,
+                                endTime,
+                                latencyMs: endTime - startTime,
+                                tokens: usage ? {
+                                    input: usage.promptTokens,
+                                    output: usage.completionTokens,
+                                    total: (usage.promptTokens || 0) + (usage.completionTokens || 0),
+                                } : undefined,
+                                toolCalls: toolCalls?.map(tc => ({
+                                    id: '',
+                                    name: tc.toolName,
+                                    arguments: tc.args,
+                                })),
+                                userId,
+                                conversationId: defaultOptions?.conversationId,
+                                properties: defaultOptions?.properties,
+                            });
+                        }
+                        // Attach traceId to result
+                        return Object.assign(result, { _traceId: traceId });
+                    }
+                    catch (error) {
+                        const endTime = Date.now();
+                        const interaction = context.getInteractionContext();
+                        if (interaction) {
+                            const span = {
+                                spanId: traceId,
+                                parentId: interaction.interactionId,
+                                name: `ai-sdk:${target.modelId}`,
+                                type: 'ai',
+                                startTime,
+                                endTime,
+                                latencyMs: endTime - startTime,
+                                input: options.prompt || options.messages,
+                                error: error instanceof Error ? error.message : String(error),
+                            };
+                            interaction.spans.push(span);
+                        }
+                        else {
+                            context.sendTrace({
+                                traceId,
+                                provider: 'ai-sdk',
+                                model: target.modelId,
+                                input: options.prompt || options.messages,
+                                startTime,
+                                endTime,
+                                latencyMs: endTime - startTime,
+                                userId,
+                                conversationId: defaultOptions?.conversationId,
+                                properties: defaultOptions?.properties,
+                                error: error instanceof Error ? error.message : String(error),
+                            });
+                        }
+                        throw error;
+                    }
+                };
+            }
+            // Intercept doStream (streaming)
+            if (prop === 'doStream' && typeof value === 'function') {
+                return async function (options) {
+                    const traceId = (defaultOptions?.traceId) || context.generateTraceId();
+                    const startTime = Date.now();
+                    const userId = defaultOptions?.userId || context.getUserId();
+                    const interaction = context.getInteractionContext();
+                    if (context.debug) {
+                        console.log('[raindrop] AI SDK stream started:', traceId);
+                    }
+                    try {
+                        const result = await value.call(target, options);
+                        // Wrap the stream to capture data
+                        const originalStream = result.stream;
+                        const collectedText = [];
+                        const collectedToolCalls = [];
+                        let usage = { promptTokens: 0, completionTokens: 0 };
+                        const wrappedStream = {
+                            [Symbol.asyncIterator]: async function* () {
+                                try {
+                                    for await (const chunk of originalStream) {
+                                        const c = chunk;
+                                        // Collect text
+                                        if (c.type === 'text-delta' && typeof c.textDelta === 'string') {
+                                            collectedText.push(c.textDelta);
+                                        }
+                                        // Collect tool calls
+                                        if (c.type === 'tool-call' && c.toolName) {
+                                            collectedToolCalls.push({
+                                                id: c.toolCallId || '',
+                                                name: c.toolName,
+                                                arguments: c.args,
+                                            });
+                                        }
+                                        // Collect usage
+                                        if (c.type === 'finish' && c.usage) {
+                                            const u = c.usage;
+                                            usage = {
+                                                promptTokens: u.promptTokens || 0,
+                                                completionTokens: u.completionTokens || 0,
+                                            };
+                                        }
+                                        yield chunk;
+                                    }
+                                    // Stream complete - send trace
+                                    const endTime = Date.now();
+                                    const output = collectedText.join('');
+                                    if (interaction) {
+                                        // Add as span to the interaction
+                                        const span = {
+                                            spanId: traceId,
+                                            parentId: interaction.interactionId,
+                                            name: `ai-sdk:${target.modelId}`,
+                                            type: 'ai',
+                                            startTime,
+                                            endTime,
+                                            latencyMs: endTime - startTime,
+                                            input: options.prompt || options.messages,
+                                            output,
+                                            properties: {
+                                                ...defaultOptions?.properties,
+                                                input_tokens: usage.promptTokens,
+                                                output_tokens: usage.completionTokens,
+                                                tool_calls: collectedToolCalls.length > 0 ? collectedToolCalls : undefined,
+                                            },
+                                        };
+                                        interaction.spans.push(span);
+                                    }
+                                    else {
+                                        context.sendTrace({
+                                            traceId,
+                                            provider: 'ai-sdk',
+                                            model: target.modelId,
+                                            input: options.prompt || options.messages,
+                                            output,
+                                            startTime,
+                                            endTime,
+                                            latencyMs: endTime - startTime,
+                                            tokens: {
+                                                input: usage.promptTokens,
+                                                output: usage.completionTokens,
+                                                total: usage.promptTokens + usage.completionTokens,
+                                            },
+                                            toolCalls: collectedToolCalls.length > 0 ? collectedToolCalls : undefined,
+                                            userId,
+                                            conversationId: defaultOptions?.conversationId,
+                                            properties: defaultOptions?.properties,
+                                        });
+                                    }
+                                }
+                                catch (error) {
+                                    const endTime = Date.now();
+                                    if (interaction) {
+                                        const span = {
+                                            spanId: traceId,
+                                            parentId: interaction.interactionId,
+                                            name: `ai-sdk:${target.modelId}`,
+                                            type: 'ai',
+                                            startTime,
+                                            endTime,
+                                            latencyMs: endTime - startTime,
+                                            input: options.prompt || options.messages,
+                                            error: error instanceof Error ? error.message : String(error),
+                                        };
+                                        interaction.spans.push(span);
+                                    }
+                                    else {
+                                        context.sendTrace({
+                                            traceId,
+                                            provider: 'ai-sdk',
+                                            model: target.modelId,
+                                            input: options.prompt || options.messages,
+                                            startTime,
+                                            endTime,
+                                            latencyMs: endTime - startTime,
+                                            userId,
+                                            conversationId: defaultOptions?.conversationId,
+                                            properties: defaultOptions?.properties,
+                                            error: error instanceof Error ? error.message : String(error),
+                                        });
+                                    }
+                                    throw error;
+                                }
+                            },
+                        };
+                        return Object.assign(result, {
+                            stream: wrappedStream,
+                            _traceId: traceId,
+                        });
+                    }
+                    catch (error) {
+                        const endTime = Date.now();
+                        if (interaction) {
+                            const span = {
+                                spanId: traceId,
+                                parentId: interaction.interactionId,
+                                name: `ai-sdk:${target.modelId}`,
+                                type: 'ai',
+                                startTime,
+                                endTime,
+                                latencyMs: endTime - startTime,
+                                input: options.prompt || options.messages,
+                                error: error instanceof Error ? error.message : String(error),
+                            };
+                            interaction.spans.push(span);
+                        }
+                        else {
+                            context.sendTrace({
+                                traceId,
+                                provider: 'ai-sdk',
+                                model: target.modelId,
+                                input: options.prompt || options.messages,
+                                startTime,
+                                endTime,
+                                latencyMs: endTime - startTime,
+                                userId,
+                                conversationId: defaultOptions?.conversationId,
+                                properties: defaultOptions?.properties,
+                                error: error instanceof Error ? error.message : String(error),
+                            });
+                        }
+                        throw error;
+                    }
+                };
+            }
+            return value;
+        },
+    };
+    const proxied = new Proxy(model, handler);
+    // Attach context for potential later use
+    return Object.assign(proxied, {
+        _raindropContext: context,
+        _raindropOptions: defaultOptions,
+    });
+}
+//# sourceMappingURL=ai-sdk.js.map

package/dist/wrappers/ai-sdk.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-sdk.js","sourceRoot":"","sources":["../../src/wrappers/ai-sdk.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAqBH;;;;;;GAMG;AACH,MAAM,UAAU,cAAc,CAC5B,KAAQ,EACR,OAA4B,EAC5B,cAAuC;IAEvC,8CAA8C;IAC9C,MAAM,OAAO,GAAoB;QAC/B,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ;YACxB,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;YAElD,uCAAuC;YACvC,IAAI,IAAI,KAAK,YAAY,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;gBACzD,OAAO,KAAK,WAAmB,OAAgC;oBAC7D,MAAM,OAAO,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;oBACvE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC7B,MAAM,MAAM,GAAG,cAAc,EAAE,MAAM,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;oBAE7D,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;wBAClB,OAAO,CAAC,GAAG,CAAC,qCAAqC,EAAE,OAAO,CAAC,CAAC;oBAC9D,CAAC;oBAED,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAA4B,CAAC;wBAC5E,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;wBAE3B,2BAA2B;wBAC3B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAc,IAAI,EAAE,CAAC;wBACzC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAyE,CAAC;wBAC/F,MAAM,SAAS,GAAG,MAAM,CAAC,SAAmE,CAAC;wBAE7F,uCAAuC;wBACvC,MAAM,WAAW,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC;wBAEpD,IAAI,WAAW,EAAE,CAAC;4BAChB,iCAAiC;4BACjC,MAAM,IAAI,GAAa;gCACrB,MAAM,EAAE,OAAO;gCACf,QAAQ,EAAE,WAAW,CAAC,aAAa;gCACnC,IAAI,EAAE,UAAU,MAAM,CAAC,OAAO,EAAE;gCAChC,IAAI,EAAE,IAAI;gCACV,SAAS;gCACT,OAAO;gCACP,SAAS,EAAE,OAAO,GAAG,SAAS;gCAC9B,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,QAAQ;gCACzC,MAAM,EAAE,IAAI;gCACZ,UAAU,EAAE;oCACV,GAAG,cAAc,EAAE,UAAU;oCAC7B,YAAY,EAAE,KAAK,EAAE,YAAY;oCACjC,aAAa,EAAE,KAAK,EAAE,gBAAgB;oCACtC,UAAU,EAAE,SAAS,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;iCAC9E;6BACF,CAAC;4BACF,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC/B,CAAC;6BAAM,CAAC;4BACN,OAAO,CAAC,SAAS,CAAC;gCAChB,OAAO;gCACP,QAAQ,EAAE,QAAQ;gCAClB,KAAK,EAAE,MAAM,CAAC,OAAO;gCACrB,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,QAAQ;gCACzC,MAAM,EAAE,IAAI;gCACZ,SAAS;gCACT,OAAO;gCACP,SAAS,EAAE,OAAO,GAAG,SAAS;gCAC9B,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;oCACd,KAAK,EAAE,KAAK,CAAC,YAAY;oCACzB,MAAM,EAAE,KAAK,CAAC,gBAAgB;oCAC9B,KAAK,EAAE,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,gBAAgB,IAAI,CAAC,CAAC;iCACjE,CAAC,CAAC,CAAC,SAAS;gCACb,SAAS,EAAE,SAAS,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;oCAC/B,EAAE,EAAE,EAAE;oCACN,IAAI,EAAE,EAAE,CAAC,QAAQ;oCACjB,SAAS,EAAE,EAAE,CAAC,IAAI;iCACnB,CAAC,CAAC;gCACH,MAAM;gCACN,cAAc,EAAE,cAAc,EAAE,cAAc;gCAC9C,UAAU,EAAE,cAAc,EAAE,UAAU;6BACvC,CAAC,CAAC;wBACL,CAAC;wBAED,2BAA2B;wBAC3B,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;oBACtD,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;wBAC3B,MAAM,WAAW,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC;wBAEpD,IAAI,WAAW,EAAE,CAAC;4BAChB,MAAM,IAAI,GAAa;gCACrB,MAAM,EAAE,OAAO;gCACf,QAAQ,EAAE,WAAW,CAAC,aAAa;gCACnC,IAAI,EAAE,UAAU,MAAM,CAAC,OAAO,EAAE;gCAChC,IAAI,EAAE,IAAI;gCACV,SAAS;gCACT,OAAO;gCACP,SAAS,EAAE,OAAO,GAAG,SAAS;gCAC9B,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,QAAQ;gCACzC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;6BAC9D,CAAC;4BACF,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC/B,CAAC;6BAAM,CAAC;4BACN,OAAO,CAAC,SAAS,CAAC;gCAChB,OAAO;gCACP,QAAQ,EAAE,QAAQ;gCAClB,KAAK,EAAE,MAAM,CAAC,OAAO;gCACrB,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,QAAQ;gCACzC,SAAS;gCACT,OAAO;gCACP,SAAS,EAAE,OAAO,GAAG,SAAS;gCAC9B,MAAM;gCACN,cAAc,EAAE,cAAc,EAAE,cAAc;gCAC9C,UAAU,EAAE,cAAc,EAAE,UAAU;gCACtC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;6BAC9D,CAAC,CAAC;wBACL,CAAC;wBAED,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC,CAAC;YACJ,CAAC;YAED,iCAAiC;YACjC,IAAI,IAAI,KAAK,UAAU,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;gBACvD,OAAO,KAAK,WAAmB,OAAgC;oBAC7D,MAAM,OAAO,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;oBACvE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC7B,MAAM,MAAM,GAAG,cAAc,EAAE,MAAM,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;oBAC7D,MAAM,WAAW,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC;oBAEpD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;wBAClB,OAAO,CAAC,GAAG,CAAC,mCAAmC,EAAE,OAAO,CAAC,CAAC;oBAC5D,CAAC;oBAED,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAG9C,CAAC;wBAEF,kCAAkC;wBAClC,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC;wBACrC,MAAM,aAAa,GAAa,EAAE,CAAC;wBACnC,MAAM,kBAAkB,GAA4D,EAAE,CAAC;wBACvF,IAAI,KAAK,GAAG,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC;wBAErD,MAAM,aAAa,GAAG;4BACpB,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,KAAK,SAAS,CAAC;gCACrC,IAAI,CAAC;oCACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;wCACzC,MAAM,CAAC,GAAG,KAAgC,CAAC;wCAE3C,eAAe;wCACf,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,IAAI,OAAO,CAAC,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;4CAC/D,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;wCAClC,CAAC;wCAED,qBAAqB;wCACrB,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;4CACzC,kBAAkB,CAAC,IAAI,CAAC;gDACtB,EAAE,EAAG,CAAC,CAAC,UAAqB,IAAI,EAAE;gDAClC,IAAI,EAAE,CAAC,CAAC,QAAkB;gDAC1B,SAAS,EAAE,CAAC,CAAC,IAAI;6CAClB,CAAC,CAAC;wCACL,CAAC;wCAED,gBAAgB;wCAChB,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC;4CACnC,MAAM,CAAC,GAAG,CAAC,CAAC,KAA6D,CAAC;4CAC1E,KAAK,GAAG;gDACN,YAAY,EAAE,CAAC,CAAC,YAAY,IAAI,CAAC;gDACjC,gBAAgB,EAAE,CAAC,CAAC,gBAAgB,IAAI,CAAC;6CAC1C,CAAC;wCACJ,CAAC;wCAED,MAAM,KAAK,CAAC;oCACd,CAAC;oCAED,+BAA+B;oCAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oCAC3B,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oCAEtC,IAAI,WAAW,EAAE,CAAC;wCAChB,iCAAiC;wCACjC,MAAM,IAAI,GAAa;4CACrB,MAAM,EAAE,OAAO;4CACf,QAAQ,EAAE,WAAW,CAAC,aAAa;4CACnC,IAAI,EAAE,UAAU,MAAM,CAAC,OAAO,EAAE;4CAChC,IAAI,EAAE,IAAI;4CACV,SAAS;4CACT,OAAO;4CACP,SAAS,EAAE,OAAO,GAAG,SAAS;4CAC9B,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,QAAQ;4CACzC,MAAM;4CACN,UAAU,EAAE;gDACV,GAAG,cAAc,EAAE,UAAU;gDAC7B,YAAY,EAAE,KAAK,CAAC,YAAY;gDAChC,aAAa,EAAE,KAAK,CAAC,gBAAgB;gDACrC,UAAU,EAAE,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,SAAS;6CAC3E;yCACF,CAAC;wCACF,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oCAC/B,CAAC;yCAAM,CAAC;wCACN,OAAO,CAAC,SAAS,CAAC;4CAChB,OAAO;4CACP,QAAQ,EAAE,QAAQ;4CAClB,KAAK,EAAE,MAAM,CAAC,OAAO;4CACrB,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,QAAQ;4CACzC,MAAM;4CACN,SAAS;4CACT,OAAO;4CACP,SAAS,EAAE,OAAO,GAAG,SAAS;4CAC9B,MAAM,EAAE;gDACN,KAAK,EAAE,KAAK,CAAC,YAAY;gDACzB,MAAM,EAAE,KAAK,CAAC,gBAAgB;gDAC9B,KAAK,EAAE,KAAK,CAAC,YAAY,GAAG,KAAK,CAAC,gBAAgB;6CACnD;4CACD,SAAS,EAAE,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,SAAS;4CACzE,MAAM;4CACN,cAAc,EAAE,cAAc,EAAE,cAAc;4CAC9C,UAAU,EAAE,cAAc,EAAE,UAAU;yCACvC,CAAC,CAAC;oCACL,CAAC;gCACH,CAAC;gCAAC,OAAO,KAAK,EAAE,CAAC;oCACf,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oCAE3B,IAAI,WAAW,EAAE,CAAC;wCAChB,MAAM,IAAI,GAAa;4CACrB,MAAM,EAAE,OAAO;4CACf,QAAQ,EAAE,WAAW,CAAC,aAAa;4CACnC,IAAI,EAAE,UAAU,MAAM,CAAC,OAAO,EAAE;4CAChC,IAAI,EAAE,IAAI;4CACV,SAAS;4CACT,OAAO;4CACP,SAAS,EAAE,OAAO,GAAG,SAAS;4CAC9B,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,QAAQ;4CACzC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;yCAC9D,CAAC;wCACF,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oCAC/B,CAAC;yCAAM,CAAC;wCACN,OAAO,CAAC,SAAS,CAAC;4CAChB,OAAO;4CACP,QAAQ,EAAE,QAAQ;4CAClB,KAAK,EAAE,MAAM,CAAC,OAAO;4CACrB,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,QAAQ;4CACzC,SAAS;4CACT,OAAO;4CACP,SAAS,EAAE,OAAO,GAAG,SAAS;4CAC9B,MAAM;4CACN,cAAc,EAAE,cAAc,EAAE,cAAc;4CAC9C,UAAU,EAAE,cAAc,EAAE,UAAU;4CACtC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;yCAC9D,CAAC,CAAC;oCACL,CAAC;oCAED,MAAM,KAAK,CAAC;gCACd,CAAC;4BACH,CAAC;yBACF,CAAC;wBAEF,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE;4BAC3B,MAAM,EAAE,aAAa;4BACrB,QAAQ,EAAE,OAAO;yBAClB,CAAC,CAAC;oBACL,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;wBAE3B,IAAI,WAAW,EAAE,CAAC;4BAChB,MAAM,IAAI,GAAa;gCACrB,MAAM,EAAE,OAAO;gCACf,QAAQ,EAAE,WAAW,CAAC,aAAa;gCACnC,IAAI,EAAE,UAAU,MAAM,CAAC,OAAO,EAAE;gCAChC,IAAI,EAAE,IAAI;gCACV,SAAS;gCACT,OAAO;gCACP,SAAS,EAAE,OAAO,GAAG,SAAS;gCAC9B,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,QAAQ;gCACzC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;6BAC9D,CAAC;4BACF,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC/B,CAAC;6BAAM,CAAC;4BACN,OAAO,CAAC,SAAS,CAAC;gCAChB,OAAO;gCACP,QAAQ,EAAE,QAAQ;gCAClB,KAAK,EAAE,MAAM,CAAC,OAAO;gCACrB,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,QAAQ;gCACzC,SAAS;gCACT,OAAO;gCACP,SAAS,EAAE,OAAO,GAAG,SAAS;gCAC9B,MAAM;gCACN,cAAc,EAAE,cAAc,EAAE,cAAc;gCAC9C,UAAU,EAAE,cAAc,EAAE,UAAU;gCACtC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;6BAC9D,CAAC,CAAC;wBACL,CAAC;wBAED,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC;QACf,CAAC;KACF,CAAC;IAEF,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAE1C,yCAAyC;IACzC,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE;QAC5B,gBAAgB,EAAE,OAAO;QACzB,gBAAgB,EAAE,cAAc;KACjC,CAAC,CAAC;AACL,CAAC"}

package/dist/wrappers/anthropic.d.ts

@@ -0,0 +1,23 @@
+/**
+ * Anthropic SDK Wrapper
+ * Wraps Anthropic client to auto-capture all message creations
+ */
+import type { TraceData, InteractionContext } from '../types.js';
+type AnthropicClient = {
+    messages: {
+        create: (params: unknown, options?: unknown) => Promise<unknown>;
+    };
+};
+export interface AnthropicWrapperContext {
+    generateTraceId: () => string;
+    sendTrace: (trace: TraceData) => void;
+    getUserId: () => string | undefined;
+    getInteractionContext: () => InteractionContext | undefined;
+    debug: boolean;
+}
+/**
+ * Creates a wrapped Anthropic client that auto-traces all calls
+ */
+export declare function wrapAnthropic<T extends AnthropicClient>(client: T, context: AnthropicWrapperContext): T;
+export {};
+//# sourceMappingURL=anthropic.d.ts.map

package/dist/wrappers/anthropic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/wrappers/anthropic.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAuC,kBAAkB,EAAY,MAAM,aAAa,CAAC;AAEhH,KAAK,eAAe,GAAG;IACrB,QAAQ,EAAE;QACR,MAAM,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;KAClE,CAAC;CACH,CAAC;AA+CF,MAAM,WAAW,uBAAuB;IACtC,eAAe,EAAE,MAAM,MAAM,CAAC;IAC9B,SAAS,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,IAAI,CAAC;IACtC,SAAS,EAAE,MAAM,MAAM,GAAG,SAAS,CAAC;IACpC,qBAAqB,EAAE,MAAM,kBAAkB,GAAG,SAAS,CAAC;IAC5D,KAAK,EAAE,OAAO,CAAC;CAChB;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,CAAC,SAAS,eAAe,EACrD,MAAM,EAAE,CAAC,EACT,OAAO,EAAE,uBAAuB,GAC/B,CAAC,CA2IH"}

package/dist/wrappers/anthropic.js

@@ -0,0 +1,251 @@
+/**
+ * Anthropic SDK Wrapper
+ * Wraps Anthropic client to auto-capture all message creations
+ */
+/**
+ * Creates a wrapped Anthropic client that auto-traces all calls
+ */
+export function wrapAnthropic(client, context) {
+    const originalCreate = client.messages.create.bind(client.messages);
+    const wrappedCreate = async (params, options) => {
+        const traceId = options?.raindrop?.traceId || context.generateTraceId();
+        const startTime = Date.now();
+        const userId = options?.raindrop?.userId || context.getUserId();
+        if (context.debug) {
+            console.log('[raindrop] Anthropic request started:', traceId);
+        }
+        // Handle streaming
+        if (params.stream) {
+            const stream = await originalCreate(params, options);
+            return wrapStream(stream, {
+                traceId,
+                startTime,
+                userId,
+                conversationId: options?.raindrop?.conversationId,
+                properties: options?.raindrop?.properties,
+                model: params.model,
+                input: params.messages,
+                context,
+            });
+        }
+        // Handle non-streaming
+        try {
+            const response = await originalCreate(params, options);
+            const endTime = Date.now();
+            // Extract text output
+            const textBlocks = response.content.filter(c => c.type === 'text');
+            const output = textBlocks.map(c => c.text).join('');
+            // Extract tool calls
+            const toolUseBlocks = response.content.filter(c => c.type === 'tool_use');
+            const toolCalls = toolUseBlocks.map(c => ({
+                id: c.id || '',
+                name: c.name || '',
+                arguments: c.input,
+            }));
+            // Check if we're within an interaction
+            const interaction = context.getInteractionContext();
+            if (interaction) {
+                // Add as span to the interaction
+                const span = {
+                    spanId: traceId,
+                    parentId: interaction.interactionId,
+                    name: `anthropic:${params.model}`,
+                    type: 'ai',
+                    startTime,
+                    endTime,
+                    latencyMs: endTime - startTime,
+                    input: params.messages,
+                    output,
+                    properties: {
+                        ...options?.raindrop?.properties,
+                        input_tokens: response.usage.input_tokens,
+                        output_tokens: response.usage.output_tokens,
+                        tool_calls: toolCalls.length > 0 ? toolCalls : undefined,
+                    },
+                };
+                interaction.spans.push(span);
+            }
+            else {
+                // Send trace
+                context.sendTrace({
+                    traceId,
+                    provider: 'anthropic',
+                    model: params.model,
+                    input: params.messages,
+                    output,
+                    startTime,
+                    endTime,
+                    latencyMs: endTime - startTime,
+                    tokens: {
+                        input: response.usage.input_tokens,
+                        output: response.usage.output_tokens,
+                        total: response.usage.input_tokens + response.usage.output_tokens,
+                    },
+                    toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
+                    userId,
+                    conversationId: options?.raindrop?.conversationId,
+                    properties: options?.raindrop?.properties,
+                });
+            }
+            return Object.assign(response, { _traceId: traceId });
+        }
+        catch (error) {
+            const endTime = Date.now();
+            const interaction = context.getInteractionContext();
+            if (interaction) {
+                const span = {
+                    spanId: traceId,
+                    parentId: interaction.interactionId,
+                    name: `anthropic:${params.model}`,
+                    type: 'ai',
+                    startTime,
+                    endTime,
+                    latencyMs: endTime - startTime,
+                    input: params.messages,
+                    error: error instanceof Error ? error.message : String(error),
+                };
+                interaction.spans.push(span);
+            }
+            else {
+                context.sendTrace({
+                    traceId,
+                    provider: 'anthropic',
+                    model: params.model,
+                    input: params.messages,
+                    startTime,
+                    endTime,
+                    latencyMs: endTime - startTime,
+                    userId,
+                    conversationId: options?.raindrop?.conversationId,
+                    properties: options?.raindrop?.properties,
+                    error: error instanceof Error ? error.message : String(error),
+                });
+            }
+            throw error;
+        }
+    };
+    // Create wrapped client
+    const wrappedClient = Object.create(Object.getPrototypeOf(client));
+    Object.assign(wrappedClient, client);
+    wrappedClient.messages = Object.create(client.messages);
+    wrappedClient.messages.create = wrappedCreate;
+    return wrappedClient;
+}
+/**
+ * Wrap a stream to capture events and send trace on completion
+ */
+function wrapStream(stream, options) {
+    const interaction = options.context.getInteractionContext();
+    const collectedText = [];
+    const toolCalls = [];
+    let usage = { input_tokens: 0, output_tokens: 0 };
+    const wrappedStream = {
+        [Symbol.asyncIterator]: async function* () {
+            try {
+                for await (const event of stream) {
+                    // Collect text deltas
+                    if (event.type === 'content_block_delta' && event.delta?.type === 'text_delta') {
+                        collectedText.push(event.delta.text || '');
+                    }
+                    // Collect tool use blocks
+                    if (event.type === 'content_block_start' && event.content_block?.type === 'tool_use') {
+                        toolCalls.push({
+                            id: event.content_block.id || '',
+                            name: event.content_block.name || '',
+                            arguments: event.content_block.input,
+                        });
+                    }
+                    // Collect usage from message_delta
+                    if (event.type === 'message_delta' && event.message?.usage) {
+                        usage = {
+                            input_tokens: event.message.usage.input_tokens || usage.input_tokens,
+                            output_tokens: event.message.usage.output_tokens || usage.output_tokens,
+                        };
+                    }
+                    yield event;
+                }
+                // Stream complete - send trace
+                const endTime = Date.now();
+                const output = collectedText.join('');
+                if (interaction) {
+                    // Add as span to the interaction
+                    const span = {
+                        spanId: options.traceId,
+                        parentId: interaction.interactionId,
+                        name: `anthropic:${options.model}`,
+                        type: 'ai',
+                        startTime: options.startTime,
+                        endTime,
+                        latencyMs: endTime - options.startTime,
+                        input: options.input,
+                        output,
+                        properties: {
+                            ...options.properties,
+                            input_tokens: usage.input_tokens,
+                            output_tokens: usage.output_tokens,
+                            tool_calls: toolCalls.length > 0 ? toolCalls : undefined,
+                        },
+                    };
+                    interaction.spans.push(span);
+                }
+                else {
+                    options.context.sendTrace({
+                        traceId: options.traceId,
+                        provider: 'anthropic',
+                        model: options.model,
+                        input: options.input,
+                        output,
+                        startTime: options.startTime,
+                        endTime,
+                        latencyMs: endTime - options.startTime,
+                        tokens: {
+                            input: usage.input_tokens,
+                            output: usage.output_tokens,
+                            total: usage.input_tokens + usage.output_tokens,
+                        },
+                        toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
+                        userId: options.userId,
+                        conversationId: options.conversationId,
+                        properties: options.properties,
+                    });
+                }
+            }
+            catch (error) {
+                const endTime = Date.now();
+                if (interaction) {
+                    const span = {
+                        spanId: options.traceId,
+                        parentId: interaction.interactionId,
+                        name: `anthropic:${options.model}`,
+                        type: 'ai',
+                        startTime: options.startTime,
+                        endTime,
+                        latencyMs: endTime - options.startTime,
+                        input: options.input,
+                        error: error instanceof Error ? error.message : String(error),
+                    };
+                    interaction.spans.push(span);
+                }
+                else {
+                    options.context.sendTrace({
+                        traceId: options.traceId,
+                        provider: 'anthropic',
+                        model: options.model,
+                        input: options.input,
+                        startTime: options.startTime,
+                        endTime,
+                        latencyMs: endTime - options.startTime,
+                        userId: options.userId,
+                        conversationId: options.conversationId,
+                        properties: options.properties,
+                        error: error instanceof Error ? error.message : String(error),
+                    });
+                }
+                throw error;
+            }
+        },
+        _traceId: options.traceId,
+    };
+    return wrappedStream;
+}
+//# sourceMappingURL=anthropic.js.map

package/dist/wrappers/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/wrappers/anthropic.ts"],"names":[],"mappings":"AAAA;;;GAGG;AA+DH;;GAEG;AACH,MAAM,UAAU,aAAa,CAC3B,MAAS,EACT,OAAgC;IAEhC,MAAM,cAAc,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAEpE,MAAM,aAAa,GAAG,KAAK,EACzB,MAAqB,EACrB,OAA+C,EACkC,EAAE;QACnF,MAAM,OAAO,GAAG,OAAO,EAAE,QAAQ,EAAE,OAAO,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;QACxE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,OAAO,EAAE,QAAQ,EAAE,MAAM,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QAEhE,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,OAAO,CAAC,GAAG,CAAC,uCAAuC,EAAE,OAAO,CAAC,CAAC;QAChE,CAAC;QAED,mBAAmB;QACnB,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,MAAM,EAAE,OAAO,CAA+B,CAAC;YACnF,OAAO,UAAU,CAAC,MAAM,EAAE;gBACxB,OAAO;gBACP,SAAS;gBACT,MAAM;gBACN,cAAc,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc;gBACjD,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,UAAU;gBACzC,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,KAAK,EAAE,MAAM,CAAC,QAAQ;gBACtB,OAAO;aACR,CAAC,CAAC;QACL,CAAC;QAED,uBAAuB;QACvB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,MAAM,EAAE,OAAO,CAAoB,CAAC;YAC1E,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE3B,sBAAsB;YACtB,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;YACnE,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAEpD,qBAAqB;YACrB,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YAC1E,MAAM,SAAS,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACxC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE;gBACd,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE;gBAClB,SAAS,EAAE,CAAC,CAAC,KAAK;aACnB,CAAC,CAAC,CAAC;YAEJ,uCAAuC;YACvC,MAAM,WAAW,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC;YAEpD,IAAI,WAAW,EAAE,CAAC;gBAChB,iCAAiC;gBACjC,MAAM,IAAI,GAAa;oBACrB,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE,WAAW,CAAC,aAAa;oBACnC,IAAI,EAAE,aAAa,MAAM,CAAC,KAAK,EAAE;oBACjC,IAAI,EAAE,IAAI;oBACV,SAAS;oBACT,OAAO;oBACP,SAAS,EAAE,OAAO,GAAG,SAAS;oBAC9B,KAAK,EAAE,MAAM,CAAC,QAAQ;oBACtB,MAAM;oBACN,UAAU,EAAE;wBACV,GAAG,OAAO,EAAE,QAAQ,EAAE,UAAU;wBAChC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;wBACzC,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;wBAC3C,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;qBACzD;iBACF,CAAC;gBACF,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACN,aAAa;gBACb,OAAO,CAAC,SAAS,CAAC;oBAChB,OAAO;oBACP,QAAQ,EAAE,WAAW;oBACrB,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,KAAK,EAAE,MAAM,CAAC,QAAQ;oBACtB,MAAM;oBACN,SAAS;oBACT,OAAO;oBACP,SAAS,EAAE,OAAO,GAAG,SAAS;oBAC9B,MAAM,EAAE;wBACN,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;wBAClC,MAAM,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;wBACpC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa;qBAClE;oBACD,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;oBACvD,MAAM;oBACN,cAAc,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc;oBACjD,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,UAAU;iBAC1C,CAAC,CAAC;YACL,CAAC;YAED,OAAO,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,WAAW,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC;YAEpD,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,IAAI,GAAa;oBACrB,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE,WAAW,CAAC,aAAa;oBACnC,IAAI,EAAE,aAAa,MAAM,CAAC,KAAK,EAAE;oBACjC,IAAI,EAAE,IAAI;oBACV,SAAS;oBACT,OAAO;oBACP,SAAS,EAAE,OAAO,GAAG,SAAS;oBAC9B,KAAK,EAAE,MAAM,CAAC,QAAQ;oBACtB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC;gBACF,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,SAAS,CAAC;oBAChB,OAAO;oBACP,QAAQ,EAAE,WAAW;oBACrB,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,KAAK,EAAE,MAAM,CAAC,QAAQ;oBACtB,SAAS;oBACT,OAAO;oBACP,SAAS,EAAE,OAAO,GAAG,SAAS;oBAC9B,MAAM;oBACN,cAAc,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc;oBACjD,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,UAAU;oBACzC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC,CAAC;YACL,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CAAC;IAEF,wBAAwB;IACxB,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;IACnE,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IAErC,aAAa,CAAC,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACxD,aAAa,CAAC,QAAQ,CAAC,MAAM,GAAG,aAAa,CAAC;IAE9C,OAAO,aAAkB,CAAC;AAC5B,CAAC;AAED;;GAEG;AACH,SAAS,UAAU,CACjB,MAAkC,EAClC,OASC;IAED,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAC5D,MAAM,aAAa,GAAa,EAAE,CAAC;IACnC,MAAM,SAAS,GAA4D,EAAE,CAAC;IAC9E,IAAI,KAAK,GAAG,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC;IAElD,MAAM,aAAa,GAAG;QACpB,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,KAAK,SAAS,CAAC;YACrC,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBACjC,sBAAsB;oBACtB,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;wBAC/E,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;oBAC7C,CAAC;oBAED,0BAA0B;oBAC1B,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,IAAI,KAAK,CAAC,aAAa,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;wBACrF,SAAS,CAAC,IAAI,CAAC;4BACb,EAAE,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE;4BAChC,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,IAAI,IAAI,EAAE;4BACpC,SAAS,EAAE,KAAK,CAAC,aAAa,CAAC,KAAK;yBACrC,CAAC,CAAC;oBACL,CAAC;oBAED,mCAAmC;oBACnC,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,IAAI,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;wBAC3D,KAAK,GAAG;4BACN,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,YAAY;4BACpE,aAAa,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,IAAI,KAAK,CAAC,aAAa;yBACxE,CAAC;oBACJ,CAAC;oBAED,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,+BAA+B;gBAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC3B,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAEtC,IAAI,WAAW,EAAE,CAAC;oBAChB,iCAAiC;oBACjC,MAAM,IAAI,GAAa;wBACrB,MAAM,EAAE,OAAO,CAAC,OAAO;wBACvB,QAAQ,EAAE,WAAW,CAAC,aAAa;wBACnC,IAAI,EAAE,aAAa,OAAO,CAAC,KAAK,EAAE;wBAClC,IAAI,EAAE,IAAI;wBACV,SAAS,EAAE,OAAO,CAAC,SAAS;wBAC5B,OAAO;wBACP,SAAS,EAAE,OAAO,GAAG,OAAO,CAAC,SAAS;wBACtC,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,MAAM;wBACN,UAAU,EAAE;4BACV,GAAG,OAAO,CAAC,UAAU;4BACrB,YAAY,EAAE,KAAK,CAAC,YAAY;4BAChC,aAAa,EAAE,KAAK,CAAC,aAAa;4BAClC,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;yBACzD;qBACF,CAAC;oBACF,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC/B,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC;wBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;wBACxB,QAAQ,EAAE,WAAW;wBACrB,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,MAAM;wBACN,SAAS,EAAE,OAAO,CAAC,SAAS;wBAC5B,OAAO;wBACP,SAAS,EAAE,OAAO,GAAG,OAAO,CAAC,SAAS;wBACtC,MAAM,EAAE;4BACN,KAAK,EAAE,KAAK,CAAC,YAAY;4BACzB,MAAM,EAAE,KAAK,CAAC,aAAa;4BAC3B,KAAK,EAAE,KAAK,CAAC,YAAY,GAAG,KAAK,CAAC,aAAa;yBAChD;wBACD,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;wBACvD,MAAM,EAAE,OAAO,CAAC,MAAM;wBACtB,cAAc,EAAE,OAAO,CAAC,cAAc;wBACtC,UAAU,EAAE,OAAO,CAAC,UAAU;qBAC/B,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAE3B,IAAI,WAAW,EAAE,CAAC;oBAChB,MAAM,IAAI,GAAa;wBACrB,MAAM,EAAE,OAAO,CAAC,OAAO;wBACvB,QAAQ,EAAE,WAAW,CAAC,aAAa;wBACnC,IAAI,EAAE,aAAa,OAAO,CAAC,KAAK,EAAE;wBAClC,IAAI,EAAE,IAAI;wBACV,SAAS,EAAE,OAAO,CAAC,SAAS;wBAC5B,OAAO;wBACP,SAAS,EAAE,OAAO,GAAG,OAAO,CAAC,SAAS;wBACtC,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D,CAAC;oBACF,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC/B,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC;wBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;wBACxB,QAAQ,EAAE,WAAW;wBACrB,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,SAAS,EAAE,OAAO,CAAC,SAAS;wBAC5B,OAAO;wBACP,SAAS,EAAE,OAAO,GAAG,OAAO,CAAC,SAAS;wBACtC,MAAM,EAAE,OAAO,CAAC,MAAM;wBACtB,cAAc,EAAE,OAAO,CAAC,cAAc;wBACtC,UAAU,EAAE,OAAO,CAAC,UAAU;wBAC9B,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QACD,QAAQ,EAAE,OAAO,CAAC,OAAO;KAC1B,CAAC;IAEF,OAAO,aAAwD,CAAC;AAClE,CAAC"}

package/dist/wrappers/openai.d.ts

@@ -0,0 +1,29 @@
+/**
+ * OpenAI SDK Wrapper
+ * Wraps OpenAI client to auto-capture chat completions AND the new Responses API
+ */
+import type { TraceData, InteractionContext } from '../types.js';
+type OpenAIClient = {
+    chat?: {
+        completions: {
+            create: (params: unknown, options?: unknown) => Promise<unknown>;
+        };
+    };
+    responses?: {
+        create: (params: unknown, options?: unknown) => Promise<unknown>;
+    };
+};
+export interface OpenAIWrapperContext {
+    generateTraceId: () => string;
+    sendTrace: (trace: TraceData) => void;
+    getUserId: () => string | undefined;
+    getInteractionContext: () => InteractionContext | undefined;
+    debug: boolean;
+}
+/**
+ * Creates a wrapped OpenAI client that auto-traces all calls
+ * Supports both chat.completions.create AND responses.create (new API)
+ */
+export declare function wrapOpenAI<T extends OpenAIClient>(client: T, context: OpenAIWrapperContext): T;
+export {};
+//# sourceMappingURL=openai.d.ts.map

package/dist/wrappers/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/wrappers/openai.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAuC,kBAAkB,EAAY,MAAM,aAAa,CAAC;AAGhH,KAAK,YAAY,GAAG;IAClB,IAAI,CAAC,EAAE;QACL,WAAW,EAAE;YACX,MAAM,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;SAClE,CAAC;KACH,CAAC;IACF,SAAS,CAAC,EAAE;QACV,MAAM,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;KAClE,CAAC;CACH,CAAC;AA0EF,MAAM,WAAW,oBAAoB;IACnC,eAAe,EAAE,MAAM,MAAM,CAAC;IAC9B,SAAS,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,IAAI,CAAC;IACtC,SAAS,EAAE,MAAM,MAAM,GAAG,SAAS,CAAC;IACpC,qBAAqB,EAAE,MAAM,kBAAkB,GAAG,SAAS,CAAC;IAC5D,KAAK,EAAE,OAAO,CAAC;CAChB;AAED;;;GAGG;AACH,wBAAgB,UAAU,CAAC,CAAC,SAAS,YAAY,EAC/C,MAAM,EAAE,CAAC,EACT,OAAO,EAAE,oBAAoB,GAC5B,CAAC,CAqBH"}