raindrop-ai 0.0.74 → 0.0.75-otelv2

package/dist/{chunk-VLA74EMW.mjs → chunk-CCT3KBLT.mjs}

@@ -1,5 +1,5 @@
 // src/tracing/tracer-core.ts
-import { context, trace as trace2 } from "@opentelemetry/api";
+import { context as context2, trace as trace2 } from "@opentelemetry/api";
 import { AnthropicInstrumentation } from "@traceloop/instrumentation-anthropic";
 import { BedrockInstrumentation } from "@traceloop/instrumentation-bedrock";
 import { ChromaDBInstrumentation } from "@traceloop/instrumentation-chromadb";
@@ -16,23 +16,53 @@ import * as traceloop3 from "@traceloop/node-server-sdk";
 import { WeakValueMap } from "weakref";
 
 // src/tracing/liveInteraction.ts
-import { trace } from "@opentelemetry/api";
+import { context, trace } from "@opentelemetry/api";
 import * as traceloop from "@traceloop/node-server-sdk";
-function getPropertiesFromContext(context2) {
+
+// src/tracing/raindropSpanProcessor.ts
+import { createContextKey } from "@opentelemetry/api";
+var RAINDROP_CONTEXT_KEY = createContextKey("raindrop.association_properties");
+var RaindropContextSpanProcessor = class {
+  onStart(span, parentContext) {
+    const properties = parentContext.getValue(RAINDROP_CONTEXT_KEY);
+    if (!properties) {
+      return;
+    }
+    for (const [key, value] of Object.entries(properties)) {
+      if (value !== void 0) {
+        span.setAttribute(`traceloop.association.properties.${key}`, value);
+      }
+    }
+  }
+  onEnd(_span) {
+  }
+  shutdown() {
+    return Promise.resolve();
+  }
+  forceFlush() {
+    return Promise.resolve();
+  }
+};
+function setRaindropContext(ctx, properties) {
+  return ctx.setValue(RAINDROP_CONTEXT_KEY, properties);
+}
+
+// src/tracing/liveInteraction.ts
+function getPropertiesFromContext(context3) {
   const properties = {
-    ...context2.userId && { user_id: context2.userId },
-    ...context2.convoId && { convo_id: context2.convoId },
-    ...context2.eventId && { event_id: context2.eventId },
-    ...context2.properties || {}
+    ...context3.userId && { user_id: context3.userId },
+    ...context3.convoId && { convo_id: context3.convoId },
+    ...context3.eventId && { event_id: context3.eventId },
+    ...context3.properties || {}
   };
-  if (context2.attachments && context2.attachments.length > 0) {
-    properties.attachments = JSON.stringify(context2.attachments);
+  if (context3.attachments && context3.attachments.length > 0) {
+    properties.attachments = JSON.stringify(context3.attachments);
   }
   return properties;
 }
 var LiveInteraction = class {
-  constructor(context2, traceId, analytics) {
-    this.context = context2;
+  constructor(context3, traceId, analytics) {
+    this.context = context3;
     this.analytics = analytics;
     this.tracer = trace.getTracer("traceloop.tracer");
     this.traceId = traceId;
@@ -50,18 +80,21 @@ var LiveInteraction = class {
     if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
       console.log("[raindrop] using withSpan in liveInteraction");
     }
-    return traceloop.withTask(
-      {
-        name: taskName,
-        associationProperties: properties,
-        inputParameters,
-        traceContent: params.traceContent,
-        suppressTracing: params.suppressTracing
-      },
-      fn,
-      thisArg,
-      ...args
-    );
+    const raindropContext = setRaindropContext(context.active(), properties);
+    return context.with(raindropContext, () => {
+      return traceloop.withTask(
+        {
+          name: taskName,
+          associationProperties: properties,
+          inputParameters,
+          traceContent: params.traceContent,
+          suppressTracing: params.suppressTracing
+        },
+        fn,
+        thisArg,
+        ...args
+      );
+    });
   }
   async withTool(params, fn, thisArg, ...args) {
     var _a;
@@ -74,19 +107,22 @@ var LiveInteraction = class {
     if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
       console.log("[raindrop] using withTool in liveInteraction");
     }
-    return traceloop.withTool(
-      {
-        name: toolName,
-        associationProperties: properties,
-        inputParameters: inputParams,
-        version: params.version,
-        traceContent: params.traceContent,
-        suppressTracing: params.suppressTracing
-      },
-      fn,
-      thisArg,
-      ...args
-    );
+    const raindropContext = setRaindropContext(context.active(), properties);
+    return context.with(raindropContext, () => {
+      return traceloop.withTool(
+        {
+          name: toolName,
+          associationProperties: properties,
+          inputParameters: inputParams,
+          version: params.version,
+          traceContent: params.traceContent,
+          suppressTracing: params.suppressTracing
+        },
+        fn,
+        thisArg,
+        ...args
+      );
+    });
   }
   startSpan(params) {
     const { name, properties = {} } = params;
@@ -133,6 +169,24 @@ var LiveInteraction = class {
       input
     });
   }
+  vercelAiSdkMetadata() {
+    const metadata = {};
+    if (this.context.userId) {
+      metadata["traceloop.association.properties.user_id"] = this.context.userId;
+    }
+    if (this.context.convoId) {
+      metadata["traceloop.association.properties.convo_id"] = this.context.convoId;
+    }
+    if (this.context.eventId) {
+      metadata["traceloop.association.properties.event_id"] = this.context.eventId;
+    }
+    if (this.context.properties) {
+      for (const [key, value] of Object.entries(this.context.properties)) {
+        metadata[`traceloop.association.properties.${key}`] = value;
+      }
+    }
+    return metadata;
+  }
   finish(resultEvent) {
     var _a;
     if (this.traceId) {
@@ -249,7 +303,7 @@ var activeInteractions = new WeakValueMap();
 var activeInteractionsByEventId = new WeakValueMap();
 function getCurrentTraceId() {
   var _a;
-  return (_a = trace2.getSpan(context.active())) == null ? void 0 : _a.spanContext().traceId;
+  return (_a = trace2.getSpan(context2.active())) == null ? void 0 : _a.spanContext().traceId;
 }
 var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
   const { logLevel, useExternalOtel, tracingEnabled, ...otherOptions } = options;
@@ -289,6 +343,21 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
         tracingEnabled: tracingEnabled !== false
       });
       traceloopInitialized = true;
+      if (tracingEnabled !== false) {
+        try {
+          const provider = trace2.getTracerProvider();
+          if (provider && typeof provider.addSpanProcessor === "function") {
+            provider.addSpanProcessor(new RaindropContextSpanProcessor());
+            if (isDebug) {
+              console.log("[raindrop] Registered RaindropContextSpanProcessor for Vercel AI SDK support");
+            }
+          }
+        } catch (e) {
+          if (isDebug) {
+            console.log("[raindrop] Could not register context span processor:", e);
+          }
+        }
+      }
     }
   } catch (error) {
     console.warn(
@@ -362,11 +431,28 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
           }
         };
       }
-      return traceloop3.createSpanProcessor({
+      const traceloopProcessor = traceloop3.createSpanProcessor({
         baseUrl: apiUrl,
         apiKey: writeKey,
         ...processorOptions != null ? processorOptions : {}
       });
+      const contextProcessor = new RaindropContextSpanProcessor();
+      return {
+        onStart(span, parentContext) {
+          contextProcessor.onStart(span, parentContext);
+          traceloopProcessor.onStart(span, parentContext);
+        },
+        onEnd(span) {
+          contextProcessor.onEnd(span);
+          traceloopProcessor.onEnd(span);
+        },
+        async forceFlush() {
+          await traceloopProcessor.forceFlush();
+        },
+        async shutdown() {
+          await traceloopProcessor.shutdown();
+        }
+      };
     },
     begin(traceContext) {
       if (!traceContext.eventId) {

package/dist/index.d.mts

@@ -380,6 +380,37 @@ type Interaction = {
      * interaction.end("I can help you with various tasks. Here are some examples...");
      */
     finish(resultEvent: FinishInteractionOptions): void;
+    /**
+     * Returns metadata for Vercel AI SDK's experimental_telemetry option.
+     *
+     * Use this when making Vercel AI SDK calls outside of withSpan() to ensure
+     * the interaction's properties (userId, convoId, eventId) are attached to the spans.
+     *
+     * When using withSpan(), this is handled automatically via context propagation.
+     * This method is a fallback for cases where you can't use withSpan().
+     *
+     * @returns Metadata object with traceloop association properties
+     *
+     * @example
+     * ```typescript
+     * const interaction = raindrop.begin({ userId: "user-123", convoId: "convo-456" });
+     *
+     * // Option 1: Use withSpan (recommended - automatic context propagation)
+     * await interaction.withSpan("my-task", async () => {
+     *   await generateText({ experimental_telemetry: { isEnabled: true }, ... });
+     * });
+     *
+     * // Option 2: Use vercelAiSdkMetadata (fallback when withSpan isn't possible)
+     * await generateText({
+     *   experimental_telemetry: {
+     *     isEnabled: true,
+     *     metadata: interaction.vercelAiSdkMetadata(),
+     *   },
+     *   ...
+     * });
+     * ```
+     */
+    vercelAiSdkMetadata(): Record<string, string>;
 };
 type Tracer = {
     /**

package/dist/index.d.ts

@@ -380,6 +380,37 @@ type Interaction = {
      * interaction.end("I can help you with various tasks. Here are some examples...");
      */
     finish(resultEvent: FinishInteractionOptions): void;
+    /**
+     * Returns metadata for Vercel AI SDK's experimental_telemetry option.
+     *
+     * Use this when making Vercel AI SDK calls outside of withSpan() to ensure
+     * the interaction's properties (userId, convoId, eventId) are attached to the spans.
+     *
+     * When using withSpan(), this is handled automatically via context propagation.
+     * This method is a fallback for cases where you can't use withSpan().
+     *
+     * @returns Metadata object with traceloop association properties
+     *
+     * @example
+     * ```typescript
+     * const interaction = raindrop.begin({ userId: "user-123", convoId: "convo-456" });
+     *
+     * // Option 1: Use withSpan (recommended - automatic context propagation)
+     * await interaction.withSpan("my-task", async () => {
+     *   await generateText({ experimental_telemetry: { isEnabled: true }, ... });
+     * });
+     *
+     * // Option 2: Use vercelAiSdkMetadata (fallback when withSpan isn't possible)
+     * await generateText({
+     *   experimental_telemetry: {
+     *     isEnabled: true,
+     *     metadata: interaction.vercelAiSdkMetadata(),
+     *   },
+     *   ...
+     * });
+     * ```
+     */
+    vercelAiSdkMetadata(): Record<string, string>;
 };
 type Tracer = {
     /**

package/dist/index.js

@@ -138,7 +138,7 @@ var CategorizationRequestSchema = import_zod.z.object({
 // package.json
 var package_default = {
   name: "raindrop-ai",
-  version: "0.0.74",
+  version: "0.0.75",
   main: "dist/index.js",
   module: "dist/index.mjs",
   types: "dist/index.d.ts",
@@ -316,7 +316,7 @@ function redactPII(event) {
 }
 
 // src/tracing/tracer-core.ts
-var import_api2 = require("@opentelemetry/api");
+var import_api3 = require("@opentelemetry/api");
 var import_instrumentation_anthropic = require("@traceloop/instrumentation-anthropic");
 var import_instrumentation_bedrock = require("@traceloop/instrumentation-bedrock");
 var import_instrumentation_chromadb = require("@traceloop/instrumentation-chromadb");
@@ -330,25 +330,55 @@ var traceloop3 = __toESM(require("@traceloop/node-server-sdk"));
 var import_weakref = require("weakref");
 
 // src/tracing/liveInteraction.ts
-var import_api = require("@opentelemetry/api");
+var import_api2 = require("@opentelemetry/api");
 var traceloop = __toESM(require("@traceloop/node-server-sdk"));
-function getPropertiesFromContext(context2) {
+
+// src/tracing/raindropSpanProcessor.ts
+var import_api = require("@opentelemetry/api");
+var RAINDROP_CONTEXT_KEY = (0, import_api.createContextKey)("raindrop.association_properties");
+var RaindropContextSpanProcessor = class {
+  onStart(span, parentContext) {
+    const properties = parentContext.getValue(RAINDROP_CONTEXT_KEY);
+    if (!properties) {
+      return;
+    }
+    for (const [key, value] of Object.entries(properties)) {
+      if (value !== void 0) {
+        span.setAttribute(`traceloop.association.properties.${key}`, value);
+      }
+    }
+  }
+  onEnd(_span) {
+  }
+  shutdown() {
+    return Promise.resolve();
+  }
+  forceFlush() {
+    return Promise.resolve();
+  }
+};
+function setRaindropContext(ctx, properties) {
+  return ctx.setValue(RAINDROP_CONTEXT_KEY, properties);
+}
+
+// src/tracing/liveInteraction.ts
+function getPropertiesFromContext(context3) {
   const properties = {
-    ...context2.userId && { user_id: context2.userId },
-    ...context2.convoId && { convo_id: context2.convoId },
-    ...context2.eventId && { event_id: context2.eventId },
-    ...context2.properties || {}
+    ...context3.userId && { user_id: context3.userId },
+    ...context3.convoId && { convo_id: context3.convoId },
+    ...context3.eventId && { event_id: context3.eventId },
+    ...context3.properties || {}
   };
-  if (context2.attachments && context2.attachments.length > 0) {
-    properties.attachments = JSON.stringify(context2.attachments);
+  if (context3.attachments && context3.attachments.length > 0) {
+    properties.attachments = JSON.stringify(context3.attachments);
   }
   return properties;
 }
 var LiveInteraction = class {
-  constructor(context2, traceId, analytics) {
-    this.context = context2;
+  constructor(context3, traceId, analytics) {
+    this.context = context3;
     this.analytics = analytics;
-    this.tracer = import_api.trace.getTracer("traceloop.tracer");
+    this.tracer = import_api2.trace.getTracer("traceloop.tracer");
     this.traceId = traceId;
   }
   async withSpan(params, fn, thisArg, ...args) {
@@ -364,18 +394,21 @@ var LiveInteraction = class {
     if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
       console.log("[raindrop] using withSpan in liveInteraction");
     }
-    return traceloop.withTask(
-      {
-        name: taskName,
-        associationProperties: properties,
-        inputParameters,
-        traceContent: params.traceContent,
-        suppressTracing: params.suppressTracing
-      },
-      fn,
-      thisArg,
-      ...args
-    );
+    const raindropContext = setRaindropContext(import_api2.context.active(), properties);
+    return import_api2.context.with(raindropContext, () => {
+      return traceloop.withTask(
+        {
+          name: taskName,
+          associationProperties: properties,
+          inputParameters,
+          traceContent: params.traceContent,
+          suppressTracing: params.suppressTracing
+        },
+        fn,
+        thisArg,
+        ...args
+      );
+    });
   }
   async withTool(params, fn, thisArg, ...args) {
     var _a;
@@ -388,19 +421,22 @@ var LiveInteraction = class {
     if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
       console.log("[raindrop] using withTool in liveInteraction");
     }
-    return traceloop.withTool(
-      {
-        name: toolName,
-        associationProperties: properties,
-        inputParameters: inputParams,
-        version: params.version,
-        traceContent: params.traceContent,
-        suppressTracing: params.suppressTracing
-      },
-      fn,
-      thisArg,
-      ...args
-    );
+    const raindropContext = setRaindropContext(import_api2.context.active(), properties);
+    return import_api2.context.with(raindropContext, () => {
+      return traceloop.withTool(
+        {
+          name: toolName,
+          associationProperties: properties,
+          inputParameters: inputParams,
+          version: params.version,
+          traceContent: params.traceContent,
+          suppressTracing: params.suppressTracing
+        },
+        fn,
+        thisArg,
+        ...args
+      );
+    });
   }
   startSpan(params) {
     const { name, properties = {} } = params;
@@ -447,6 +483,24 @@ var LiveInteraction = class {
       input
     });
   }
+  vercelAiSdkMetadata() {
+    const metadata = {};
+    if (this.context.userId) {
+      metadata["traceloop.association.properties.user_id"] = this.context.userId;
+    }
+    if (this.context.convoId) {
+      metadata["traceloop.association.properties.convo_id"] = this.context.convoId;
+    }
+    if (this.context.eventId) {
+      metadata["traceloop.association.properties.event_id"] = this.context.eventId;
+    }
+    if (this.context.properties) {
+      for (const [key, value] of Object.entries(this.context.properties)) {
+        metadata[`traceloop.association.properties.${key}`] = value;
+      }
+    }
+    return metadata;
+  }
   finish(resultEvent) {
     var _a;
     if (this.traceId) {
@@ -563,7 +617,7 @@ var activeInteractions = new import_weakref.WeakValueMap();
 var activeInteractionsByEventId = new import_weakref.WeakValueMap();
 function getCurrentTraceId() {
   var _a;
-  return (_a = import_api2.trace.getSpan(import_api2.context.active())) == null ? void 0 : _a.spanContext().traceId;
+  return (_a = import_api3.trace.getSpan(import_api3.context.active())) == null ? void 0 : _a.spanContext().traceId;
 }
 var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
   const { logLevel, useExternalOtel, tracingEnabled, ...otherOptions } = options;
@@ -603,6 +657,21 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
         tracingEnabled: tracingEnabled !== false
       });
       traceloopInitialized = true;
+      if (tracingEnabled !== false) {
+        try {
+          const provider = import_api3.trace.getTracerProvider();
+          if (provider && typeof provider.addSpanProcessor === "function") {
+            provider.addSpanProcessor(new RaindropContextSpanProcessor());
+            if (isDebug) {
+              console.log("[raindrop] Registered RaindropContextSpanProcessor for Vercel AI SDK support");
+            }
+          }
+        } catch (e) {
+          if (isDebug) {
+            console.log("[raindrop] Could not register context span processor:", e);
+          }
+        }
+      }
     }
   } catch (error) {
     console.warn(
@@ -676,11 +745,28 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
           }
         };
       }
-      return traceloop3.createSpanProcessor({
+      const traceloopProcessor = traceloop3.createSpanProcessor({
         baseUrl: apiUrl,
         apiKey: writeKey,
         ...processorOptions != null ? processorOptions : {}
       });
+      const contextProcessor = new RaindropContextSpanProcessor();
+      return {
+        onStart(span, parentContext) {
+          contextProcessor.onStart(span, parentContext);
+          traceloopProcessor.onStart(span, parentContext);
+        },
+        onEnd(span) {
+          contextProcessor.onEnd(span);
+          traceloopProcessor.onEnd(span);
+        },
+        async forceFlush() {
+          await traceloopProcessor.forceFlush();
+        },
+        async shutdown() {
+          await traceloopProcessor.shutdown();
+        }
+      };
     },
     begin(traceContext) {
       if (!traceContext.eventId) {
@@ -1167,7 +1253,7 @@ var Raindrop = class {
   }
   getContext() {
     const runtime = detectRuntime();
-    const context2 = {
+    const context3 = {
       library: {
         name: package_default.name,
         version: package_default.version
@@ -1177,9 +1263,9 @@ var Raindrop = class {
       }
     };
     if (runtime === "node") {
-      context2.metadata.nodeVersion = process.version;
+      context3.metadata.nodeVersion = process.version;
     }
-    return context2;
+    return context3;
   }
   formatZodError(error) {
     return error.issues.map((issue) => `(path: ${issue.path.join(".")}): ${issue.message}`).join(", ");

package/dist/index.mjs

@@ -1,6 +1,6 @@
 import {
   tracing
-} from "./chunk-VLA74EMW.mjs";
+} from "./chunk-CCT3KBLT.mjs";
 
 // ../schemas/src/ingest/index.ts
 import { z } from "zod";
@@ -104,7 +104,7 @@ var CategorizationRequestSchema = z.object({
 // package.json
 var package_default = {
   name: "raindrop-ai",
-  version: "0.0.74",
+  version: "0.0.75",
   main: "dist/index.js",
   module: "dist/index.mjs",
   types: "dist/index.d.ts",

package/dist/tracing/index.js

@@ -36,7 +36,7 @@ __export(tracing_exports, {
 module.exports = __toCommonJS(tracing_exports);
 
 // src/tracing/tracer-core.ts
-var import_api2 = require("@opentelemetry/api");
+var import_api3 = require("@opentelemetry/api");
 var import_instrumentation_anthropic = require("@traceloop/instrumentation-anthropic");
 var import_instrumentation_bedrock = require("@traceloop/instrumentation-bedrock");
 var import_instrumentation_chromadb = require("@traceloop/instrumentation-chromadb");
@@ -50,25 +50,55 @@ var traceloop3 = __toESM(require("@traceloop/node-server-sdk"));
 var import_weakref = require("weakref");
 
 // src/tracing/liveInteraction.ts
-var import_api = require("@opentelemetry/api");
+var import_api2 = require("@opentelemetry/api");
 var traceloop = __toESM(require("@traceloop/node-server-sdk"));
-function getPropertiesFromContext(context2) {
+
+// src/tracing/raindropSpanProcessor.ts
+var import_api = require("@opentelemetry/api");
+var RAINDROP_CONTEXT_KEY = (0, import_api.createContextKey)("raindrop.association_properties");
+var RaindropContextSpanProcessor = class {
+  onStart(span, parentContext) {
+    const properties = parentContext.getValue(RAINDROP_CONTEXT_KEY);
+    if (!properties) {
+      return;
+    }
+    for (const [key, value] of Object.entries(properties)) {
+      if (value !== void 0) {
+        span.setAttribute(`traceloop.association.properties.${key}`, value);
+      }
+    }
+  }
+  onEnd(_span) {
+  }
+  shutdown() {
+    return Promise.resolve();
+  }
+  forceFlush() {
+    return Promise.resolve();
+  }
+};
+function setRaindropContext(ctx, properties) {
+  return ctx.setValue(RAINDROP_CONTEXT_KEY, properties);
+}
+
+// src/tracing/liveInteraction.ts
+function getPropertiesFromContext(context3) {
   const properties = {
-    ...context2.userId && { user_id: context2.userId },
-    ...context2.convoId && { convo_id: context2.convoId },
-    ...context2.eventId && { event_id: context2.eventId },
-    ...context2.properties || {}
+    ...context3.userId && { user_id: context3.userId },
+    ...context3.convoId && { convo_id: context3.convoId },
+    ...context3.eventId && { event_id: context3.eventId },
+    ...context3.properties || {}
   };
-  if (context2.attachments && context2.attachments.length > 0) {
-    properties.attachments = JSON.stringify(context2.attachments);
+  if (context3.attachments && context3.attachments.length > 0) {
+    properties.attachments = JSON.stringify(context3.attachments);
   }
   return properties;
 }
 var LiveInteraction = class {
-  constructor(context2, traceId, analytics) {
-    this.context = context2;
+  constructor(context3, traceId, analytics) {
+    this.context = context3;
     this.analytics = analytics;
-    this.tracer = import_api.trace.getTracer("traceloop.tracer");
+    this.tracer = import_api2.trace.getTracer("traceloop.tracer");
     this.traceId = traceId;
   }
   async withSpan(params, fn, thisArg, ...args) {
@@ -84,18 +114,21 @@ var LiveInteraction = class {
     if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
       console.log("[raindrop] using withSpan in liveInteraction");
     }
-    return traceloop.withTask(
-      {
-        name: taskName,
-        associationProperties: properties,
-        inputParameters,
-        traceContent: params.traceContent,
-        suppressTracing: params.suppressTracing
-      },
-      fn,
-      thisArg,
-      ...args
-    );
+    const raindropContext = setRaindropContext(import_api2.context.active(), properties);
+    return import_api2.context.with(raindropContext, () => {
+      return traceloop.withTask(
+        {
+          name: taskName,
+          associationProperties: properties,
+          inputParameters,
+          traceContent: params.traceContent,
+          suppressTracing: params.suppressTracing
+        },
+        fn,
+        thisArg,
+        ...args
+      );
+    });
   }
   async withTool(params, fn, thisArg, ...args) {
     var _a;
@@ -108,19 +141,22 @@ var LiveInteraction = class {
     if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
       console.log("[raindrop] using withTool in liveInteraction");
     }
-    return traceloop.withTool(
-      {
-        name: toolName,
-        associationProperties: properties,
-        inputParameters: inputParams,
-        version: params.version,
-        traceContent: params.traceContent,
-        suppressTracing: params.suppressTracing
-      },
-      fn,
-      thisArg,
-      ...args
-    );
+    const raindropContext = setRaindropContext(import_api2.context.active(), properties);
+    return import_api2.context.with(raindropContext, () => {
+      return traceloop.withTool(
+        {
+          name: toolName,
+          associationProperties: properties,
+          inputParameters: inputParams,
+          version: params.version,
+          traceContent: params.traceContent,
+          suppressTracing: params.suppressTracing
+        },
+        fn,
+        thisArg,
+        ...args
+      );
+    });
   }
   startSpan(params) {
     const { name, properties = {} } = params;
@@ -167,6 +203,24 @@ var LiveInteraction = class {
       input
     });
   }
+  vercelAiSdkMetadata() {
+    const metadata = {};
+    if (this.context.userId) {
+      metadata["traceloop.association.properties.user_id"] = this.context.userId;
+    }
+    if (this.context.convoId) {
+      metadata["traceloop.association.properties.convo_id"] = this.context.convoId;
+    }
+    if (this.context.eventId) {
+      metadata["traceloop.association.properties.event_id"] = this.context.eventId;
+    }
+    if (this.context.properties) {
+      for (const [key, value] of Object.entries(this.context.properties)) {
+        metadata[`traceloop.association.properties.${key}`] = value;
+      }
+    }
+    return metadata;
+  }
   finish(resultEvent) {
     var _a;
     if (this.traceId) {
@@ -283,7 +337,7 @@ var activeInteractions = new import_weakref.WeakValueMap();
 var activeInteractionsByEventId = new import_weakref.WeakValueMap();
 function getCurrentTraceId() {
   var _a;
-  return (_a = import_api2.trace.getSpan(import_api2.context.active())) == null ? void 0 : _a.spanContext().traceId;
+  return (_a = import_api3.trace.getSpan(import_api3.context.active())) == null ? void 0 : _a.spanContext().traceId;
 }
 var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
   const { logLevel, useExternalOtel, tracingEnabled, ...otherOptions } = options;
@@ -323,6 +377,21 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
         tracingEnabled: tracingEnabled !== false
       });
       traceloopInitialized = true;
+      if (tracingEnabled !== false) {
+        try {
+          const provider = import_api3.trace.getTracerProvider();
+          if (provider && typeof provider.addSpanProcessor === "function") {
+            provider.addSpanProcessor(new RaindropContextSpanProcessor());
+            if (isDebug) {
+              console.log("[raindrop] Registered RaindropContextSpanProcessor for Vercel AI SDK support");
+            }
+          }
+        } catch (e) {
+          if (isDebug) {
+            console.log("[raindrop] Could not register context span processor:", e);
+          }
+        }
+      }
     }
   } catch (error) {
     console.warn(
@@ -396,11 +465,28 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
           }
         };
       }
-      return traceloop3.createSpanProcessor({
+      const traceloopProcessor = traceloop3.createSpanProcessor({
         baseUrl: apiUrl,
         apiKey: writeKey,
         ...processorOptions != null ? processorOptions : {}
       });
+      const contextProcessor = new RaindropContextSpanProcessor();
+      return {
+        onStart(span, parentContext) {
+          contextProcessor.onStart(span, parentContext);
+          traceloopProcessor.onStart(span, parentContext);
+        },
+        onEnd(span) {
+          contextProcessor.onEnd(span);
+          traceloopProcessor.onEnd(span);
+        },
+        async forceFlush() {
+          await traceloopProcessor.forceFlush();
+        },
+        async shutdown() {
+          await traceloopProcessor.shutdown();
+        }
+      };
     },
     begin(traceContext) {
       if (!traceContext.eventId) {

package/dist/tracing/index.mjs

@@ -1,6 +1,6 @@
 import {
   tracing
-} from "../chunk-VLA74EMW.mjs";
+} from "../chunk-CCT3KBLT.mjs";
 
 // src/tracing/index.ts
 function initTracing() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "raindrop-ai",
-  "version": "0.0.74",
+  "version": "0.0.75-otelv2",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
@@ -46,7 +46,6 @@
     "@dawn-analytics/redact-pii": "0.1.2",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/resources": "^2.0.1",
-    "@traceloop/node-server-sdk": "0.19.0-otel-v1",
     "weakref": "^0.2.1",
     "zod": "^3.23.8"
   },
@@ -85,5 +84,8 @@
       "resolve": true
     },
     "clean": true
+  },
+  "optionalDependencies": {
+    "@traceloop/node-server-sdk": "0.22.2"
   }
 }