rivetkit 2.3.19 → 2.3.20

package/dist/browser/client.d.ts

@@ -339,6 +339,29 @@ declare class UserError extends RivetError {
     constructor(message: string, options?: UserErrorOptions);
 }
 
+/** Sample rates keyed by action name, nested like the actor's action groups. */
+interface ActionSampleRates {
+    readonly [name: string]: number | ActionSampleRates;
+}
+/**
+ * {@link ActionSampleRates} limited to the actions of one actor, so a rate for
+ * an action that does not exist is a compile error.
+ */
+type ActionTraceSamplers<TActions> = {
+    readonly [K in keyof TActions]?: TActions[K] extends (...args: never[]) => unknown ? number : ActionTraceSamplers<TActions[K]>;
+};
+/** Trace sampling for one actor definition. */
+interface ActorTracingOptions<TActions> {
+    /**
+     * Share of this actor's invocations that are recorded, from 0 to 1. A rate
+     * lowers what is recorded and never raises it, so an invocation inside a
+     * trace the caller did not record is not recorded at any rate.
+     */
+    readonly sampler?: number;
+    /** Rates that replace `sampler` for single actions. */
+    readonly actions?: ActionTraceSamplers<TActions>;
+}
+
 type UpgradeWebSocket = (createEvents: (c: Context) => any) => Handler;
 type GetUpgradeWebSocket = () => UpgradeWebSocket;
 
@@ -638,12 +661,19 @@ interface RuntimeSqlBatchStatement {
 interface RuntimeSqlRunResult {
     changes: number;
 }
+/** Sample rates for the traces an actor starts, in the shape core takes. */
+interface RuntimeActorTracingConfig {
+    sampler?: number;
+    /** Keyed by flattened, dot-separated action name. */
+    actions?: Record<string, number>;
+}
 interface RuntimeActorConfig {
     name?: string;
     icon?: string;
     hasDatabase?: boolean;
     remoteSqlite?: boolean;
     sqliteProfiling?: SqliteProfilingOptions;
+    tracing?: RuntimeActorTracingConfig;
     enableActorRuntimeSocket?: boolean;
     hasState?: boolean;
     canHibernateWebsocket?: boolean;
@@ -1874,6 +1904,10 @@ declare const ActorConfigSchema: z.ZodObject<{
     onWebSocket: z.ZodOptional<z.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>;
     actions: z.ZodDefault<z.ZodCustom<Record<string, unknown>, Record<string, unknown>>>;
     actionInputSchemas: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    tracing: z.ZodOptional<z.ZodObject<{
+        sampler: z.ZodOptional<z.ZodNumber>;
+        actions: z.ZodOptional<z.ZodType<ActionSampleRates, unknown, z.core.$ZodTypeInternals<ActionSampleRates, unknown>>>;
+    }, z.core.$strict>>;
     connParamsSchema: z.ZodOptional<z.ZodAny>;
     events: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
     queues: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
@@ -2114,6 +2148,11 @@ interface BaseActorConfig<TState, TConnParams, TConnState, TVars, TInput, TDatab
      * May mirror nested action groups or use dot-separated low-level action names.
      */
     actionInputSchemas?: ActionInputSchemas;
+    /**
+     * How much of this actor's work is recorded in traces. Defaults to the
+     * process sampler from `OTEL_TRACES_SAMPLER`.
+     */
+    tracing?: ActorTracingOptions<TActions>;
     /**
      * Optional schema for validating connection params in native runtimes.
      */

package/dist/browser/client.js

@@ -220,7 +220,7 @@ function noopNext() {
 // package.json
 var package_default = {
   name: "rivetkit",
-  version: "2.3.19",
+  version: "2.3.20",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [