raindrop-ai 0.0.77 → 0.0.78

package/dist/{chunk-7PZNI6M6.mjs → chunk-B5UZWYFP.mjs}

@@ -1,5 +1,5 @@
 // src/tracing/tracer-core.ts
-import { context as context2, trace as trace2 } from "@opentelemetry/api";
+import { context, trace as trace3 } from "@opentelemetry/api";
 import { AnthropicInstrumentation } from "@traceloop/instrumentation-anthropic";
 import { BedrockInstrumentation } from "@traceloop/instrumentation-bedrock";
 import { ChromaDBInstrumentation } from "@traceloop/instrumentation-chromadb";
@@ -16,23 +16,54 @@ import * as traceloop3 from "@traceloop/node-server-sdk";
 import { WeakValueMap } from "weakref";
 
 // src/tracing/liveInteraction.ts
-import { trace } from "@opentelemetry/api";
+import { SpanStatusCode, trace } from "@opentelemetry/api";
 import * as traceloop from "@traceloop/node-server-sdk";
-function getPropertiesFromContext(context3) {
+function getPropertiesFromContext(context2) {
   const properties = {
-    ...context3.userId && { user_id: context3.userId },
-    ...context3.convoId && { convo_id: context3.convoId },
-    ...context3.eventId && { event_id: context3.eventId },
-    ...context3.properties || {}
+    ...context2.userId && { user_id: context2.userId },
+    ...context2.convoId && { convo_id: context2.convoId },
+    ...context2.eventId && { event_id: context2.eventId },
+    ...context2.properties || {}
   };
-  if (context3.attachments && context3.attachments.length > 0) {
-    properties.attachments = JSON.stringify(context3.attachments);
+  if (context2.attachments && context2.attachments.length > 0) {
+    properties.attachments = JSON.stringify(context2.attachments);
   }
   return properties;
 }
+function setAssociationProperties(span, properties) {
+  Object.entries(properties).forEach(([key, value]) => {
+    span.setAttribute("traceloop.association.properties." + key, String(value));
+  });
+}
+var LiveToolSpan = class {
+  constructor(span) {
+    this.hasError = false;
+    this.span = span;
+  }
+  setInput(input) {
+    const inputStr = typeof input === "string" ? input : JSON.stringify(input);
+    this.span.setAttribute("traceloop.entity.input", inputStr);
+  }
+  setOutput(output) {
+    const outputStr = typeof output === "string" ? output : JSON.stringify(output);
+    this.span.setAttribute("traceloop.entity.output", outputStr);
+  }
+  setError(error) {
+    this.hasError = true;
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    this.span.setStatus({ code: SpanStatusCode.ERROR, message: errorMessage });
+    this.span.recordException(error instanceof Error ? error : new Error(errorMessage));
+  }
+  end() {
+    if (!this.hasError) {
+      this.span.setStatus({ code: SpanStatusCode.OK });
+    }
+    this.span.end();
+  }
+};
 var LiveInteraction = class {
-  constructor(context3, traceId, analytics) {
-    this.context = context3;
+  constructor(context2, traceId, analytics) {
+    this.context = context2;
     this.analytics = analytics;
     this.tracer = trace.getTracer("traceloop.tracer");
     this.traceId = traceId;
@@ -91,13 +122,29 @@ var LiveInteraction = class {
   startSpan(params) {
     const { name, properties = {} } = params;
     const span = this.tracer.startSpan(name);
-    span.setAttributes(getPropertiesFromContext(this.context));
+    const contextProperties = getPropertiesFromContext(this.context);
+    setAssociationProperties(span, contextProperties);
     span.setAttribute("traceloop.span.kind", "task");
-    Object.entries(properties).forEach(([key, value]) => {
-      span.setAttribute("traceloop.association.properties." + key, String(value));
-    });
+    setAssociationProperties(span, properties);
     return span;
   }
+  startToolSpan(params) {
+    var _a;
+    const { name, properties = {}, inputParameters } = params;
+    const span = this.tracer.startSpan(name);
+    const contextProperties = getPropertiesFromContext(this.context);
+    setAssociationProperties(span, contextProperties);
+    span.setAttribute("traceloop.span.kind", "tool");
+    setAssociationProperties(span, properties);
+    if (inputParameters !== void 0) {
+      const inputStr = typeof inputParameters === "string" ? inputParameters : JSON.stringify(inputParameters);
+      span.setAttribute("traceloop.entity.input", inputStr);
+    }
+    if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
+      console.log(`[raindrop] startToolSpan: started tool span "${name}"`);
+    }
+    return new LiveToolSpan(span);
+  }
   setProperties(properties) {
     var _a;
     this.context.properties = { ...this.context.properties, ...properties };
@@ -162,12 +209,54 @@ var LiveInteraction = class {
       isPending: false
     });
   }
+  trackTool(params) {
+    var _a;
+    const { name, input, output, durationMs, startTime, error, properties = {} } = params;
+    const startTimeMs = startTime instanceof Date ? startTime.getTime() : typeof startTime === "number" ? startTime : Date.now() - (durationMs != null ? durationMs : 0);
+    const endTimeMs = startTimeMs + (durationMs != null ? durationMs : 0);
+    const span = this.tracer.startSpan(name, {
+      startTime: startTimeMs
+    });
+    const contextProperties = getPropertiesFromContext(this.context);
+    setAssociationProperties(span, contextProperties);
+    span.setAttribute("traceloop.span.kind", "tool");
+    setAssociationProperties(span, properties);
+    if (input !== void 0) {
+      const inputStr = typeof input === "string" ? input : JSON.stringify(input);
+      span.setAttribute("traceloop.entity.input", inputStr);
+    }
+    if (output !== void 0) {
+      const outputStr = typeof output === "string" ? output : JSON.stringify(output);
+      span.setAttribute("traceloop.entity.output", outputStr);
+    }
+    if (durationMs !== void 0) {
+      span.setAttribute("traceloop.entity.duration_ms", durationMs);
+    }
+    if (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      span.setStatus({ code: SpanStatusCode.ERROR, message: errorMessage });
+      span.recordException(error instanceof Error ? error : new Error(errorMessage));
+    } else {
+      span.setStatus({ code: SpanStatusCode.OK });
+    }
+    span.end(endTimeMs);
+    if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
+      console.log(`[raindrop] trackTool: logged tool span "${name}"`, {
+        input,
+        output,
+        durationMs,
+        error: error ? String(error) : void 0
+      });
+    }
+  }
 };
 
 // src/tracing/liveTracer.ts
+import { SpanStatusCode as SpanStatusCode2, trace as trace2 } from "@opentelemetry/api";
 import * as traceloop2 from "@traceloop/node-server-sdk";
 var LiveTracer = class {
   constructor(globalProperties) {
+    this.tracer = trace2.getTracer("traceloop.tracer");
     this.globalProperties = globalProperties || {};
   }
   async withSpan(params, fn, thisArg, ...args) {
@@ -192,6 +281,40 @@ var LiveTracer = class {
       ...args
     );
   }
+  trackTool(params) {
+    const { name, input, output, durationMs, startTime, error, properties = {} } = params;
+    const startTimeMs = startTime instanceof Date ? startTime.getTime() : typeof startTime === "number" ? startTime : Date.now() - (durationMs != null ? durationMs : 0);
+    const endTimeMs = startTimeMs + (durationMs != null ? durationMs : 0);
+    const span = this.tracer.startSpan(name, {
+      startTime: startTimeMs
+    });
+    Object.entries(this.globalProperties).forEach(([key, value]) => {
+      span.setAttribute("traceloop.association.properties." + key, String(value));
+    });
+    Object.entries(properties).forEach(([key, value]) => {
+      span.setAttribute("traceloop.association.properties." + key, String(value));
+    });
+    span.setAttribute("traceloop.span.kind", "tool");
+    if (input !== void 0) {
+      const inputStr = typeof input === "string" ? input : JSON.stringify(input);
+      span.setAttribute("traceloop.entity.input", inputStr);
+    }
+    if (output !== void 0) {
+      const outputStr = typeof output === "string" ? output : JSON.stringify(output);
+      span.setAttribute("traceloop.entity.output", outputStr);
+    }
+    if (durationMs !== void 0) {
+      span.setAttribute("traceloop.entity.duration_ms", durationMs);
+    }
+    if (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      span.setStatus({ code: SpanStatusCode2.ERROR, message: errorMessage });
+      span.recordException(error instanceof Error ? error : new Error(errorMessage));
+    } else {
+      span.setStatus({ code: SpanStatusCode2.OK });
+    }
+    span.end(endTimeMs);
+  }
 };
 
 // src/tracing/tracer-core.ts
@@ -267,7 +390,7 @@ var activeInteractions = new WeakValueMap();
 var activeInteractionsByEventId = new WeakValueMap();
 function getCurrentTraceId() {
   var _a;
-  return (_a = trace2.getSpan(context2.active())) == null ? void 0 : _a.spanContext().traceId;
+  return (_a = trace3.getSpan(context.active())) == null ? void 0 : _a.spanContext().traceId;
 }
 var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
   const { logLevel, useExternalOtel, tracingEnabled, disableBatching, ...otherOptions } = options;

package/dist/index.d.mts

@@ -197,6 +197,69 @@ interface ToolParams {
     traceContent?: boolean;
     suppressTracing?: boolean;
 }
+/**
+ * Parameters for directly logging a tool span without wrapping a function.
+ *
+ * Use this when you want to record a tool invocation that has already completed,
+ * rather than wrapping the tool call with withTool().
+ *
+ * @property name - Name of the tool for identification in traces
+ * @property input - The input provided to the tool (will be JSON stringified if object)
+ * @property output - The output returned by the tool (will be JSON stringified if object)
+ * @property durationMs - Duration of the tool execution in milliseconds
+ * @property startTime - Optional start time of the tool execution
+ * @property error - Optional error if the tool failed
+ * @property properties - Optional key-value pairs for additional metadata
+ */
+interface TrackToolParams {
+    name: string;
+    input?: unknown;
+    output?: unknown;
+    durationMs?: number;
+    startTime?: Date | number;
+    error?: Error | string;
+    properties?: Record<string, string>;
+}
+/**
+ * A wrapper around an OpenTelemetry span for tool invocations.
+ *
+ * Provides a clean API for setting tool input, output, and error status
+ * without needing to know the internal traceloop attribute names.
+ *
+ * @example
+ * ```typescript
+ * const toolSpan = interaction.startToolSpan({ name: "web_search" });
+ * try {
+ *   const result = await performSearch(query);
+ *   toolSpan.setOutput(result);
+ * } catch (error) {
+ *   toolSpan.setError(error);
+ * } finally {
+ *   toolSpan.end();
+ * }
+ * ```
+ */
+interface ToolSpan {
+    /**
+     * Sets the input for this tool span.
+     * @param input The input value (will be JSON stringified if object)
+     */
+    setInput(input: unknown): void;
+    /**
+     * Sets the output for this tool span.
+     * @param output The output value (will be JSON stringified if object)
+     */
+    setOutput(output: unknown): void;
+    /**
+     * Marks this tool span as failed with an error.
+     * @param error The error that occurred
+     */
+    setError(error: Error | string): void;
+    /**
+     * Ends this tool span. Must be called when the tool execution is complete.
+     */
+    end(): void;
+}
 /**
  * Represents an active tracing interaction for tracking and analyzing AI interactions.
  *
@@ -293,6 +356,33 @@ type Interaction = {
      * @returns A span that can be used to record a task
      */
     startSpan(params: SpanParams): Span;
+    /**
+     * Creates a tool span that can be used to manually record a tool invocation.
+     *
+     * Similar to startSpan but sets the span kind to "tool" instead of "task".
+     * Use this when you want manual control over the span lifecycle for tool calls.
+     * Returns a ToolSpan wrapper with convenient methods for setting input, output, and errors.
+     *
+     * @param params ToolParams object with tool configuration
+     * @returns A ToolSpan wrapper for recording the tool execution
+     *
+     * @example
+     * ```typescript
+     * const interaction = raindrop.begin({ userId: "user-123", event: "agent_run" });
+     *
+     * // Start a tool span with manual control
+     * const toolSpan = interaction.startToolSpan({ name: "web_search" });
+     * try {
+     *   const result = await performSearch(query);
+     *   toolSpan.setOutput(result);
+     * } catch (error) {
+     *   toolSpan.setError(error);
+     * } finally {
+     *   toolSpan.end();
+     * }
+     * ```
+     */
+    startToolSpan(params: ToolParams): ToolSpan;
     /**
      * Sets multiple properties on the interaction context.
      *
@@ -411,6 +501,41 @@ type Interaction = {
      * ```
      */
     vercelAiSdkMetadata(): Record<string, string>;
+    /**
+     * Logs a tool span directly without wrapping a function.
+     *
+     * Use this when you want to record a tool invocation that has already completed,
+     * rather than wrapping the tool call with withTool(). This is useful for:
+     * - Recording tool calls from external systems
+     * - Logging tools that were executed outside of the tracing context
+     * - Retroactively adding tool spans with known input/output/duration
+     *
+     * @param params TrackToolParams object with tool execution details
+     *
+     * @example
+     * ```typescript
+     * const interaction = raindrop.begin({ userId: "user-123", event: "agent_run" });
+     *
+     * // Record a tool that was already executed
+     * interaction.trackTool({
+     *   name: "web_search",
+     *   input: { query: "weather in NYC" },
+     *   output: { results: ["Sunny, 72°F"] },
+     *   durationMs: 150,
+     * });
+     *
+     * // Record a failed tool
+     * interaction.trackTool({
+     *   name: "database_query",
+     *   input: { sql: "SELECT * FROM users" },
+     *   error: new Error("Connection timeout"),
+     *   durationMs: 5000,
+     * });
+     *
+     * interaction.finish({ output: "Here's the weather..." });
+     * ```
+     */
+    trackTool(params: TrackToolParams): void;
 };
 type Tracer = {
     /**
@@ -420,6 +545,29 @@ type Tracer = {
      * @returns A span that can be used to record a task
      */
     withSpan<T>(params: SpanParams, fn: (...args: any[]) => Promise<T> | T, thisArg?: any, ...args: any[]): Promise<T>;
+    /**
+     * Logs a tool span directly without wrapping a function.
+     *
+     * Use this when you want to record a tool invocation that has already completed,
+     * rather than wrapping the tool call. This is useful for batch jobs or
+     * non-interactive use-cases where you only care about tracing and token usage.
+     *
+     * @param params TrackToolParams object with tool execution details
+     *
+     * @example
+     * ```typescript
+     * const tracer = raindrop.tracer({ job_id: "batch-123" });
+     *
+     * // Record a tool that was already executed
+     * tracer.trackTool({
+     *   name: "web_search",
+     *   input: { query: "weather in NYC" },
+     *   output: { results: ["Sunny, 72°F"] },
+     *   durationMs: 150,
+     * });
+     * ```
+     */
+    trackTool(params: TrackToolParams): void;
 };
 
 interface AnalyticsConfig {
@@ -632,4 +780,4 @@ declare class Raindrop {
     close(): Promise<void>;
 }
 
-export { type AiTrackEvent, type Attachment, type AttachmentType, type BeginInteractionOptions, type FinishInteractionOptions, type IdentifyEvent, type Interaction, MAX_INGEST_SIZE_BYTES, type PartialAiTrackEvent, Raindrop, type RaindropSpanProcessor, type SignalEvent, type SpanParams, type ToolParams, type TraceContext, type Tracer, type WorkflowParams, Raindrop as default };
+export { type AiTrackEvent, type Attachment, type AttachmentType, type BeginInteractionOptions, type FinishInteractionOptions, type IdentifyEvent, type Interaction, MAX_INGEST_SIZE_BYTES, type PartialAiTrackEvent, Raindrop, type RaindropSpanProcessor, type SignalEvent, type SpanParams, type ToolParams, type ToolSpan, type TraceContext, type Tracer, type TrackToolParams, type WorkflowParams, Raindrop as default };

package/dist/index.d.ts

@@ -197,6 +197,69 @@ interface ToolParams {
     traceContent?: boolean;
     suppressTracing?: boolean;
 }
+/**
+ * Parameters for directly logging a tool span without wrapping a function.
+ *
+ * Use this when you want to record a tool invocation that has already completed,
+ * rather than wrapping the tool call with withTool().
+ *
+ * @property name - Name of the tool for identification in traces
+ * @property input - The input provided to the tool (will be JSON stringified if object)
+ * @property output - The output returned by the tool (will be JSON stringified if object)
+ * @property durationMs - Duration of the tool execution in milliseconds
+ * @property startTime - Optional start time of the tool execution
+ * @property error - Optional error if the tool failed
+ * @property properties - Optional key-value pairs for additional metadata
+ */
+interface TrackToolParams {
+    name: string;
+    input?: unknown;
+    output?: unknown;
+    durationMs?: number;
+    startTime?: Date | number;
+    error?: Error | string;
+    properties?: Record<string, string>;
+}
+/**
+ * A wrapper around an OpenTelemetry span for tool invocations.
+ *
+ * Provides a clean API for setting tool input, output, and error status
+ * without needing to know the internal traceloop attribute names.
+ *
+ * @example
+ * ```typescript
+ * const toolSpan = interaction.startToolSpan({ name: "web_search" });
+ * try {
+ *   const result = await performSearch(query);
+ *   toolSpan.setOutput(result);
+ * } catch (error) {
+ *   toolSpan.setError(error);
+ * } finally {
+ *   toolSpan.end();
+ * }
+ * ```
+ */
+interface ToolSpan {
+    /**
+     * Sets the input for this tool span.
+     * @param input The input value (will be JSON stringified if object)
+     */
+    setInput(input: unknown): void;
+    /**
+     * Sets the output for this tool span.
+     * @param output The output value (will be JSON stringified if object)
+     */
+    setOutput(output: unknown): void;
+    /**
+     * Marks this tool span as failed with an error.
+     * @param error The error that occurred
+     */
+    setError(error: Error | string): void;
+    /**
+     * Ends this tool span. Must be called when the tool execution is complete.
+     */
+    end(): void;
+}
 /**
  * Represents an active tracing interaction for tracking and analyzing AI interactions.
  *
@@ -293,6 +356,33 @@ type Interaction = {
      * @returns A span that can be used to record a task
      */
     startSpan(params: SpanParams): Span;
+    /**
+     * Creates a tool span that can be used to manually record a tool invocation.
+     *
+     * Similar to startSpan but sets the span kind to "tool" instead of "task".
+     * Use this when you want manual control over the span lifecycle for tool calls.
+     * Returns a ToolSpan wrapper with convenient methods for setting input, output, and errors.
+     *
+     * @param params ToolParams object with tool configuration
+     * @returns A ToolSpan wrapper for recording the tool execution
+     *
+     * @example
+     * ```typescript
+     * const interaction = raindrop.begin({ userId: "user-123", event: "agent_run" });
+     *
+     * // Start a tool span with manual control
+     * const toolSpan = interaction.startToolSpan({ name: "web_search" });
+     * try {
+     *   const result = await performSearch(query);
+     *   toolSpan.setOutput(result);
+     * } catch (error) {
+     *   toolSpan.setError(error);
+     * } finally {
+     *   toolSpan.end();
+     * }
+     * ```
+     */
+    startToolSpan(params: ToolParams): ToolSpan;
     /**
      * Sets multiple properties on the interaction context.
      *
@@ -411,6 +501,41 @@ type Interaction = {
      * ```
      */
     vercelAiSdkMetadata(): Record<string, string>;
+    /**
+     * Logs a tool span directly without wrapping a function.
+     *
+     * Use this when you want to record a tool invocation that has already completed,
+     * rather than wrapping the tool call with withTool(). This is useful for:
+     * - Recording tool calls from external systems
+     * - Logging tools that were executed outside of the tracing context
+     * - Retroactively adding tool spans with known input/output/duration
+     *
+     * @param params TrackToolParams object with tool execution details
+     *
+     * @example
+     * ```typescript
+     * const interaction = raindrop.begin({ userId: "user-123", event: "agent_run" });
+     *
+     * // Record a tool that was already executed
+     * interaction.trackTool({
+     *   name: "web_search",
+     *   input: { query: "weather in NYC" },
+     *   output: { results: ["Sunny, 72°F"] },
+     *   durationMs: 150,
+     * });
+     *
+     * // Record a failed tool
+     * interaction.trackTool({
+     *   name: "database_query",
+     *   input: { sql: "SELECT * FROM users" },
+     *   error: new Error("Connection timeout"),
+     *   durationMs: 5000,
+     * });
+     *
+     * interaction.finish({ output: "Here's the weather..." });
+     * ```
+     */
+    trackTool(params: TrackToolParams): void;
 };
 type Tracer = {
     /**
@@ -420,6 +545,29 @@ type Tracer = {
      * @returns A span that can be used to record a task
      */
     withSpan<T>(params: SpanParams, fn: (...args: any[]) => Promise<T> | T, thisArg?: any, ...args: any[]): Promise<T>;
+    /**
+     * Logs a tool span directly without wrapping a function.
+     *
+     * Use this when you want to record a tool invocation that has already completed,
+     * rather than wrapping the tool call. This is useful for batch jobs or
+     * non-interactive use-cases where you only care about tracing and token usage.
+     *
+     * @param params TrackToolParams object with tool execution details
+     *
+     * @example
+     * ```typescript
+     * const tracer = raindrop.tracer({ job_id: "batch-123" });
+     *
+     * // Record a tool that was already executed
+     * tracer.trackTool({
+     *   name: "web_search",
+     *   input: { query: "weather in NYC" },
+     *   output: { results: ["Sunny, 72°F"] },
+     *   durationMs: 150,
+     * });
+     * ```
+     */
+    trackTool(params: TrackToolParams): void;
 };
 
 interface AnalyticsConfig {
@@ -632,4 +780,4 @@ declare class Raindrop {
     close(): Promise<void>;
 }
 
-export { type AiTrackEvent, type Attachment, type AttachmentType, type BeginInteractionOptions, type FinishInteractionOptions, type IdentifyEvent, type Interaction, MAX_INGEST_SIZE_BYTES, type PartialAiTrackEvent, Raindrop, type RaindropSpanProcessor, type SignalEvent, type SpanParams, type ToolParams, type TraceContext, type Tracer, type WorkflowParams, Raindrop as default };
+export { type AiTrackEvent, type Attachment, type AttachmentType, type BeginInteractionOptions, type FinishInteractionOptions, type IdentifyEvent, type Interaction, MAX_INGEST_SIZE_BYTES, type PartialAiTrackEvent, Raindrop, type RaindropSpanProcessor, type SignalEvent, type SpanParams, type ToolParams, type ToolSpan, type TraceContext, type Tracer, type TrackToolParams, type WorkflowParams, Raindrop as default };

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.77",
+  version: "0.0.78",
   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");
@@ -332,21 +332,52 @@ var import_weakref = require("weakref");
 // src/tracing/liveInteraction.ts
 var import_api = require("@opentelemetry/api");
 var traceloop = __toESM(require("@traceloop/node-server-sdk"));
-function getPropertiesFromContext(context3) {
+function getPropertiesFromContext(context2) {
   const properties = {
-    ...context3.userId && { user_id: context3.userId },
-    ...context3.convoId && { convo_id: context3.convoId },
-    ...context3.eventId && { event_id: context3.eventId },
-    ...context3.properties || {}
+    ...context2.userId && { user_id: context2.userId },
+    ...context2.convoId && { convo_id: context2.convoId },
+    ...context2.eventId && { event_id: context2.eventId },
+    ...context2.properties || {}
   };
-  if (context3.attachments && context3.attachments.length > 0) {
-    properties.attachments = JSON.stringify(context3.attachments);
+  if (context2.attachments && context2.attachments.length > 0) {
+    properties.attachments = JSON.stringify(context2.attachments);
   }
   return properties;
 }
+function setAssociationProperties(span, properties) {
+  Object.entries(properties).forEach(([key, value]) => {
+    span.setAttribute("traceloop.association.properties." + key, String(value));
+  });
+}
+var LiveToolSpan = class {
+  constructor(span) {
+    this.hasError = false;
+    this.span = span;
+  }
+  setInput(input) {
+    const inputStr = typeof input === "string" ? input : JSON.stringify(input);
+    this.span.setAttribute("traceloop.entity.input", inputStr);
+  }
+  setOutput(output) {
+    const outputStr = typeof output === "string" ? output : JSON.stringify(output);
+    this.span.setAttribute("traceloop.entity.output", outputStr);
+  }
+  setError(error) {
+    this.hasError = true;
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    this.span.setStatus({ code: import_api.SpanStatusCode.ERROR, message: errorMessage });
+    this.span.recordException(error instanceof Error ? error : new Error(errorMessage));
+  }
+  end() {
+    if (!this.hasError) {
+      this.span.setStatus({ code: import_api.SpanStatusCode.OK });
+    }
+    this.span.end();
+  }
+};
 var LiveInteraction = class {
-  constructor(context3, traceId, analytics) {
-    this.context = context3;
+  constructor(context2, traceId, analytics) {
+    this.context = context2;
     this.analytics = analytics;
     this.tracer = import_api.trace.getTracer("traceloop.tracer");
     this.traceId = traceId;
@@ -405,13 +436,29 @@ var LiveInteraction = class {
   startSpan(params) {
     const { name, properties = {} } = params;
     const span = this.tracer.startSpan(name);
-    span.setAttributes(getPropertiesFromContext(this.context));
+    const contextProperties = getPropertiesFromContext(this.context);
+    setAssociationProperties(span, contextProperties);
     span.setAttribute("traceloop.span.kind", "task");
-    Object.entries(properties).forEach(([key, value]) => {
-      span.setAttribute("traceloop.association.properties." + key, String(value));
-    });
+    setAssociationProperties(span, properties);
     return span;
   }
+  startToolSpan(params) {
+    var _a;
+    const { name, properties = {}, inputParameters } = params;
+    const span = this.tracer.startSpan(name);
+    const contextProperties = getPropertiesFromContext(this.context);
+    setAssociationProperties(span, contextProperties);
+    span.setAttribute("traceloop.span.kind", "tool");
+    setAssociationProperties(span, properties);
+    if (inputParameters !== void 0) {
+      const inputStr = typeof inputParameters === "string" ? inputParameters : JSON.stringify(inputParameters);
+      span.setAttribute("traceloop.entity.input", inputStr);
+    }
+    if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
+      console.log(`[raindrop] startToolSpan: started tool span "${name}"`);
+    }
+    return new LiveToolSpan(span);
+  }
   setProperties(properties) {
     var _a;
     this.context.properties = { ...this.context.properties, ...properties };
@@ -476,12 +523,54 @@ var LiveInteraction = class {
       isPending: false
     });
   }
+  trackTool(params) {
+    var _a;
+    const { name, input, output, durationMs, startTime, error, properties = {} } = params;
+    const startTimeMs = startTime instanceof Date ? startTime.getTime() : typeof startTime === "number" ? startTime : Date.now() - (durationMs != null ? durationMs : 0);
+    const endTimeMs = startTimeMs + (durationMs != null ? durationMs : 0);
+    const span = this.tracer.startSpan(name, {
+      startTime: startTimeMs
+    });
+    const contextProperties = getPropertiesFromContext(this.context);
+    setAssociationProperties(span, contextProperties);
+    span.setAttribute("traceloop.span.kind", "tool");
+    setAssociationProperties(span, properties);
+    if (input !== void 0) {
+      const inputStr = typeof input === "string" ? input : JSON.stringify(input);
+      span.setAttribute("traceloop.entity.input", inputStr);
+    }
+    if (output !== void 0) {
+      const outputStr = typeof output === "string" ? output : JSON.stringify(output);
+      span.setAttribute("traceloop.entity.output", outputStr);
+    }
+    if (durationMs !== void 0) {
+      span.setAttribute("traceloop.entity.duration_ms", durationMs);
+    }
+    if (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      span.setStatus({ code: import_api.SpanStatusCode.ERROR, message: errorMessage });
+      span.recordException(error instanceof Error ? error : new Error(errorMessage));
+    } else {
+      span.setStatus({ code: import_api.SpanStatusCode.OK });
+    }
+    span.end(endTimeMs);
+    if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
+      console.log(`[raindrop] trackTool: logged tool span "${name}"`, {
+        input,
+        output,
+        durationMs,
+        error: error ? String(error) : void 0
+      });
+    }
+  }
 };
 
 // src/tracing/liveTracer.ts
+var import_api2 = require("@opentelemetry/api");
 var traceloop2 = __toESM(require("@traceloop/node-server-sdk"));
 var LiveTracer = class {
   constructor(globalProperties) {
+    this.tracer = import_api2.trace.getTracer("traceloop.tracer");
     this.globalProperties = globalProperties || {};
   }
   async withSpan(params, fn, thisArg, ...args) {
@@ -506,6 +595,40 @@ var LiveTracer = class {
       ...args
     );
   }
+  trackTool(params) {
+    const { name, input, output, durationMs, startTime, error, properties = {} } = params;
+    const startTimeMs = startTime instanceof Date ? startTime.getTime() : typeof startTime === "number" ? startTime : Date.now() - (durationMs != null ? durationMs : 0);
+    const endTimeMs = startTimeMs + (durationMs != null ? durationMs : 0);
+    const span = this.tracer.startSpan(name, {
+      startTime: startTimeMs
+    });
+    Object.entries(this.globalProperties).forEach(([key, value]) => {
+      span.setAttribute("traceloop.association.properties." + key, String(value));
+    });
+    Object.entries(properties).forEach(([key, value]) => {
+      span.setAttribute("traceloop.association.properties." + key, String(value));
+    });
+    span.setAttribute("traceloop.span.kind", "tool");
+    if (input !== void 0) {
+      const inputStr = typeof input === "string" ? input : JSON.stringify(input);
+      span.setAttribute("traceloop.entity.input", inputStr);
+    }
+    if (output !== void 0) {
+      const outputStr = typeof output === "string" ? output : JSON.stringify(output);
+      span.setAttribute("traceloop.entity.output", outputStr);
+    }
+    if (durationMs !== void 0) {
+      span.setAttribute("traceloop.entity.duration_ms", durationMs);
+    }
+    if (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      span.setStatus({ code: import_api2.SpanStatusCode.ERROR, message: errorMessage });
+      span.recordException(error instanceof Error ? error : new Error(errorMessage));
+    } else {
+      span.setStatus({ code: import_api2.SpanStatusCode.OK });
+    }
+    span.end(endTimeMs);
+  }
 };
 
 // src/tracing/tracer-core.ts
@@ -581,7 +704,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, disableBatching, ...otherOptions } = options;
@@ -1190,7 +1313,7 @@ var Raindrop = class {
   }
   getContext() {
     const runtime = detectRuntime();
-    const context3 = {
+    const context2 = {
       library: {
         name: package_default.name,
         version: package_default.version
@@ -1200,9 +1323,9 @@ var Raindrop = class {
       }
     };
     if (runtime === "node") {
-      context3.metadata.nodeVersion = process.version;
+      context2.metadata.nodeVersion = process.version;
     }
-    return context3;
+    return context2;
   }
   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-7PZNI6M6.mjs";
+} from "./chunk-B5UZWYFP.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.77",
+  version: "0.0.78",
   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");
@@ -52,21 +52,52 @@ var import_weakref = require("weakref");
 // src/tracing/liveInteraction.ts
 var import_api = require("@opentelemetry/api");
 var traceloop = __toESM(require("@traceloop/node-server-sdk"));
-function getPropertiesFromContext(context3) {
+function getPropertiesFromContext(context2) {
   const properties = {
-    ...context3.userId && { user_id: context3.userId },
-    ...context3.convoId && { convo_id: context3.convoId },
-    ...context3.eventId && { event_id: context3.eventId },
-    ...context3.properties || {}
+    ...context2.userId && { user_id: context2.userId },
+    ...context2.convoId && { convo_id: context2.convoId },
+    ...context2.eventId && { event_id: context2.eventId },
+    ...context2.properties || {}
   };
-  if (context3.attachments && context3.attachments.length > 0) {
-    properties.attachments = JSON.stringify(context3.attachments);
+  if (context2.attachments && context2.attachments.length > 0) {
+    properties.attachments = JSON.stringify(context2.attachments);
   }
   return properties;
 }
+function setAssociationProperties(span, properties) {
+  Object.entries(properties).forEach(([key, value]) => {
+    span.setAttribute("traceloop.association.properties." + key, String(value));
+  });
+}
+var LiveToolSpan = class {
+  constructor(span) {
+    this.hasError = false;
+    this.span = span;
+  }
+  setInput(input) {
+    const inputStr = typeof input === "string" ? input : JSON.stringify(input);
+    this.span.setAttribute("traceloop.entity.input", inputStr);
+  }
+  setOutput(output) {
+    const outputStr = typeof output === "string" ? output : JSON.stringify(output);
+    this.span.setAttribute("traceloop.entity.output", outputStr);
+  }
+  setError(error) {
+    this.hasError = true;
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    this.span.setStatus({ code: import_api.SpanStatusCode.ERROR, message: errorMessage });
+    this.span.recordException(error instanceof Error ? error : new Error(errorMessage));
+  }
+  end() {
+    if (!this.hasError) {
+      this.span.setStatus({ code: import_api.SpanStatusCode.OK });
+    }
+    this.span.end();
+  }
+};
 var LiveInteraction = class {
-  constructor(context3, traceId, analytics) {
-    this.context = context3;
+  constructor(context2, traceId, analytics) {
+    this.context = context2;
     this.analytics = analytics;
     this.tracer = import_api.trace.getTracer("traceloop.tracer");
     this.traceId = traceId;
@@ -125,13 +156,29 @@ var LiveInteraction = class {
   startSpan(params) {
     const { name, properties = {} } = params;
     const span = this.tracer.startSpan(name);
-    span.setAttributes(getPropertiesFromContext(this.context));
+    const contextProperties = getPropertiesFromContext(this.context);
+    setAssociationProperties(span, contextProperties);
     span.setAttribute("traceloop.span.kind", "task");
-    Object.entries(properties).forEach(([key, value]) => {
-      span.setAttribute("traceloop.association.properties." + key, String(value));
-    });
+    setAssociationProperties(span, properties);
     return span;
   }
+  startToolSpan(params) {
+    var _a;
+    const { name, properties = {}, inputParameters } = params;
+    const span = this.tracer.startSpan(name);
+    const contextProperties = getPropertiesFromContext(this.context);
+    setAssociationProperties(span, contextProperties);
+    span.setAttribute("traceloop.span.kind", "tool");
+    setAssociationProperties(span, properties);
+    if (inputParameters !== void 0) {
+      const inputStr = typeof inputParameters === "string" ? inputParameters : JSON.stringify(inputParameters);
+      span.setAttribute("traceloop.entity.input", inputStr);
+    }
+    if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
+      console.log(`[raindrop] startToolSpan: started tool span "${name}"`);
+    }
+    return new LiveToolSpan(span);
+  }
   setProperties(properties) {
     var _a;
     this.context.properties = { ...this.context.properties, ...properties };
@@ -196,12 +243,54 @@ var LiveInteraction = class {
       isPending: false
     });
   }
+  trackTool(params) {
+    var _a;
+    const { name, input, output, durationMs, startTime, error, properties = {} } = params;
+    const startTimeMs = startTime instanceof Date ? startTime.getTime() : typeof startTime === "number" ? startTime : Date.now() - (durationMs != null ? durationMs : 0);
+    const endTimeMs = startTimeMs + (durationMs != null ? durationMs : 0);
+    const span = this.tracer.startSpan(name, {
+      startTime: startTimeMs
+    });
+    const contextProperties = getPropertiesFromContext(this.context);
+    setAssociationProperties(span, contextProperties);
+    span.setAttribute("traceloop.span.kind", "tool");
+    setAssociationProperties(span, properties);
+    if (input !== void 0) {
+      const inputStr = typeof input === "string" ? input : JSON.stringify(input);
+      span.setAttribute("traceloop.entity.input", inputStr);
+    }
+    if (output !== void 0) {
+      const outputStr = typeof output === "string" ? output : JSON.stringify(output);
+      span.setAttribute("traceloop.entity.output", outputStr);
+    }
+    if (durationMs !== void 0) {
+      span.setAttribute("traceloop.entity.duration_ms", durationMs);
+    }
+    if (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      span.setStatus({ code: import_api.SpanStatusCode.ERROR, message: errorMessage });
+      span.recordException(error instanceof Error ? error : new Error(errorMessage));
+    } else {
+      span.setStatus({ code: import_api.SpanStatusCode.OK });
+    }
+    span.end(endTimeMs);
+    if ((_a = this.analytics) == null ? void 0 : _a.debugLogs) {
+      console.log(`[raindrop] trackTool: logged tool span "${name}"`, {
+        input,
+        output,
+        durationMs,
+        error: error ? String(error) : void 0
+      });
+    }
+  }
 };
 
 // src/tracing/liveTracer.ts
+var import_api2 = require("@opentelemetry/api");
 var traceloop2 = __toESM(require("@traceloop/node-server-sdk"));
 var LiveTracer = class {
   constructor(globalProperties) {
+    this.tracer = import_api2.trace.getTracer("traceloop.tracer");
     this.globalProperties = globalProperties || {};
   }
   async withSpan(params, fn, thisArg, ...args) {
@@ -226,6 +315,40 @@ var LiveTracer = class {
       ...args
     );
   }
+  trackTool(params) {
+    const { name, input, output, durationMs, startTime, error, properties = {} } = params;
+    const startTimeMs = startTime instanceof Date ? startTime.getTime() : typeof startTime === "number" ? startTime : Date.now() - (durationMs != null ? durationMs : 0);
+    const endTimeMs = startTimeMs + (durationMs != null ? durationMs : 0);
+    const span = this.tracer.startSpan(name, {
+      startTime: startTimeMs
+    });
+    Object.entries(this.globalProperties).forEach(([key, value]) => {
+      span.setAttribute("traceloop.association.properties." + key, String(value));
+    });
+    Object.entries(properties).forEach(([key, value]) => {
+      span.setAttribute("traceloop.association.properties." + key, String(value));
+    });
+    span.setAttribute("traceloop.span.kind", "tool");
+    if (input !== void 0) {
+      const inputStr = typeof input === "string" ? input : JSON.stringify(input);
+      span.setAttribute("traceloop.entity.input", inputStr);
+    }
+    if (output !== void 0) {
+      const outputStr = typeof output === "string" ? output : JSON.stringify(output);
+      span.setAttribute("traceloop.entity.output", outputStr);
+    }
+    if (durationMs !== void 0) {
+      span.setAttribute("traceloop.entity.duration_ms", durationMs);
+    }
+    if (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      span.setStatus({ code: import_api2.SpanStatusCode.ERROR, message: errorMessage });
+      span.recordException(error instanceof Error ? error : new Error(errorMessage));
+    } else {
+      span.setStatus({ code: import_api2.SpanStatusCode.OK });
+    }
+    span.end(endTimeMs);
+  }
 };
 
 // src/tracing/tracer-core.ts
@@ -301,7 +424,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, disableBatching, ...otherOptions } = options;

package/dist/tracing/index.mjs

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "raindrop-ai",
-  "version": "0.0.77",
+  "version": "0.0.78",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",