rivetkit 2.3.19 → 2.3.20

package/dist/tsup/agent-os/index.cjs

@@ -33,6 +33,26 @@ function flattenActionInputSchemas(actions, schemas) {
   }
   return flattened;
 }
+function flattenActionTraceSamplers(actions, rates = {}) {
+  const handlers = flattenActionHandlers(actions);
+  const flattened = Object.fromEntries(flattenSampleRates(rates, []));
+  for (const name of Object.keys(flattened)) {
+    if (!Object.hasOwn(handlers, name)) {
+      throw new TypeError(
+        `tracing.actions.${name} does not name an action`
+      );
+    }
+  }
+  return flattened;
+}
+function flattenSampleRates(rates, path) {
+  return Object.entries(rates).flatMap(
+    ([segment, rate]) => {
+      const childPath = [...path, segment];
+      return typeof rate === "number" ? [[childPath.join("."), rate]] : flattenSampleRates(rate, childPath);
+    }
+  );
+}
 function collectActionEntries(actions) {
   const entries = [];
   const names = /* @__PURE__ */ new Set();
@@ -228,6 +248,10 @@ var InstanceActorOptionsSchema = InstanceActorOptionsBaseSchema.prefault(() => (
 var ActorOptionsSchema = GlobalActorOptionsBaseSchema.extend(
   InstanceActorOptionsBaseSchema.shape
 ).strict().prefault(() => ({}));
+var zSampleRate = _v4.z.number().min(0).max(1);
+var zActionSampleRates = _v4.z.lazy(
+  () => _v4.z.record(_v4.z.string(), _v4.z.union([zSampleRate, zActionSampleRates]))
+);
 var ActorConfigSchema = _v4.z.object({
   onCreate: zFunction().optional(),
   onDestroy: zFunction().optional(),
@@ -244,6 +268,10 @@ var ActorConfigSchema = _v4.z.object({
   onWebSocket: zFunction().optional(),
   actions: zActionTree.default(() => ({})),
   actionInputSchemas: _v4.z.record(_v4.z.string(), _v4.z.any()).optional(),
+  tracing: _v4.z.object({
+    sampler: zSampleRate.optional(),
+    actions: zActionSampleRates.optional()
+  }).strict().optional(),
   connParamsSchema: _v4.z.any().optional(),
   events: _v4.z.record(_v4.z.string(), _v4.z.any()).optional(),
   queues: _v4.z.record(_v4.z.string(), _v4.z.any()).optional(),
@@ -391,6 +419,9 @@ var DocActorConfigSchema = _v4.z.object({
   actionInputSchemas: _v4.z.record(_v4.z.string(), _v4.z.unknown()).optional().describe(
     "Optional schemas for validating action argument tuples in native runtimes. May mirror nested action groups or use dot-separated low-level action names."
   ),
+  tracing: _v4.z.unknown().optional().describe(
+    "Trace sampling for this actor: `sampler` is the share of its invocations that are recorded, from 0 to 1, with optional per-action rates under `actions`. A rate lowers what is recorded and never raises it. Defaults to the process sampler from OTEL_TRACES_SAMPLER."
+  ),
   connParamsSchema: _v4.z.unknown().optional().describe(
     "Optional schema for validating connection params in native runtimes."
   ),
@@ -420,7 +451,7 @@ function isRivetErrorCode(error, group, code) {
 // 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: [
@@ -889,12 +920,14 @@ var ActorDefinition = class {
   }
 };
 function actor(input) {
+  var _a;
   warnDeprecatedOptionKeys(
     input == null ? void 0 : input.options
   );
   const config = ActorConfigSchema.parse(input);
   flattenActionHandlers(config.actions);
   flattenActionInputSchemas(config.actions, config.actionInputSchemas);
+  flattenActionTraceSamplers(config.actions, (_a = config.tracing) == null ? void 0 : _a.actions);
   return new ActorDefinition(config);
 }