rivetkit 2.3.0-rc.10 → 2.3.0-rc.11

package/dist/browser/client.js

@@ -200,7 +200,7 @@ function noopNext() {
 // package.json
 var package_default = {
   name: "rivetkit",
-  version: "2.3.0-rc.10",
+  version: "2.3.0-rc.11",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [
@@ -272,6 +272,16 @@ var package_default = {
         default: "./dist/tsup/db/drizzle.cjs"
       }
     },
+    "./dynamic": {
+      import: {
+        types: "./dist/tsup/dynamic/mod.d.ts",
+        default: "./dist/tsup/dynamic/mod.js"
+      },
+      require: {
+        types: "./dist/tsup/dynamic/mod.d.cts",
+        default: "./dist/tsup/dynamic/mod.cjs"
+      }
+    },
     "./client": {
       import: {
         browser: {
@@ -351,7 +361,7 @@ var package_default = {
     "./dist/tsup/chunk-*.cjs"
   ],
   scripts: {
-    build: "tsup src/mod.ts src/client/mod.ts src/common/log.ts src/common/websocket.ts src/actor/errors.ts src/utils.ts src/workflow/mod.ts src/test/mod.ts src/inspector/mod.ts src/db/mod.ts src/db/drizzle.ts && tsup src/agent-os/index.ts --no-clean --out-dir dist/tsup/agent-os",
+    build: "tsup src/mod.ts src/client/mod.ts src/common/log.ts src/common/websocket.ts src/actor/errors.ts src/utils.ts src/workflow/mod.ts src/test/mod.ts src/inspector/mod.ts src/db/mod.ts src/db/drizzle.ts src/dynamic/mod.ts && tsup src/agent-os/index.ts --no-clean --out-dir dist/tsup/agent-os",
     "build:browser": "tsup --config tsup.browser.config.ts",
     "check-types": "tsc --noEmit",
     lint: "biome check . && pnpm run check:test-skips && pnpm run check:wait-for-comments",
@@ -368,13 +378,14 @@ var package_default = {
     "actor-config-schema-gen": "tsx scripts/actor-config-schema-gen.ts"
   },
   dependencies: {
-    "@rivet-dev/agent-os-core": "^0.1.1",
     "@hono/node-server": "^1.18.2",
     "@hono/node-ws": "^1.1.1",
     "@hono/zod-openapi": "^1.1.5",
+    "@rivet-dev/agent-os-core": "^0.1.1",
     "@rivetkit/bare-ts": "^0.6.2",
     "@rivetkit/engine-cli": "workspace:*",
     "@rivetkit/engine-envoy-protocol": "workspace:*",
+    "@rivetkit/on-change": "6.0.1",
     "@rivetkit/rivetkit-napi": "workspace:*",
     "@rivetkit/rivetkit-wasm": "workspace:*",
     "@rivetkit/traces": "workspace:*",
@@ -392,10 +403,10 @@ var package_default = {
     zod: "^4.1.0"
   },
   devDependencies: {
+    "@biomejs/biome": "^2.3",
     "@copilotkit/llmock": "^1.6.0",
     "@rivet-dev/agent-os-common": "*",
     "@rivet-dev/agent-os-pi": "^0.1.1",
-    "@biomejs/biome": "^2.3",
     "@standard-schema/spec": "^1.0.0",
     "@types/invariant": "^2",
     "@types/node": "^22.13.1",
@@ -471,6 +482,7 @@ function configureDefaultLogger(logLevel) {
       messageKey: "msg",
       // Do not include pid/hostname in output
       base: {},
+      errorKey: "error",
       // Keep the numeric level so the logfmt sink can match Pino's levels.
       formatters: {
         level(_label, number) {
@@ -762,7 +774,7 @@ function deserializeActorKey(keyString) {
     }
   }
   if (escaping) {
-    parts.push(currentPart + "\\");
+    parts.push(`${currentPart}\\`);
   } else if (isEmptyStringMarker) {
     parts.push("");
   } else if (currentPart !== "" || parts.length > 0) {
@@ -805,6 +817,10 @@ var JSON_COMPAT_BIGINT = "$BigInt";
 var JSON_COMPAT_ARRAY_BUFFER = "$ArrayBuffer";
 var JSON_COMPAT_UINT8_ARRAY = "$Uint8Array";
 var JSON_COMPAT_UNDEFINED = "$Undefined";
+var JSON_COMPAT_SET = "$Set";
+function isTypedArray(value) {
+  return value instanceof Uint8ClampedArray || value instanceof Uint16Array || value instanceof Uint32Array || value instanceof BigUint64Array || value instanceof Int8Array || value instanceof Int16Array || value instanceof Int32Array || value instanceof BigInt64Array || value instanceof Float32Array || value instanceof Float64Array;
+}
 var EncodingSchema = z2.enum(["json", "cbor", "bare"]);
 async function inputDataToBuffer(data) {
   if (typeof data === "string") {
@@ -839,9 +855,15 @@ function isPlainObject(value) {
   return proto === Object.prototype || proto === null;
 }
 function encodeJsonCompatValue(input) {
+  if (input === null) {
+    return input;
+  }
   if (input === void 0) {
     return [JSON_COMPAT_UNDEFINED, 0];
   }
+  if (typeof input === "string" || typeof input === "number" || typeof input === "boolean") {
+    return input;
+  }
   if (typeof input === "bigint") {
     return [JSON_COMPAT_BIGINT, input.toString()];
   }
@@ -851,10 +873,34 @@ function encodeJsonCompatValue(input) {
   if (input instanceof Uint8Array) {
     return [JSON_COMPAT_UINT8_ARRAY, base64EncodeUint8Array(input)];
   }
+  if (isTypedArray(input)) {
+    return input;
+  }
+  if (input instanceof Date || input instanceof RegExp || input instanceof Error) {
+    return input;
+  }
+  if (input instanceof Set) {
+    const encoded = [...input.values()].map(
+      (v) => encodeJsonCompatValue(v)
+    );
+    return [JSON_COMPAT_SET, encoded];
+  }
+  if (input instanceof Map) {
+    const encoded = /* @__PURE__ */ new Map();
+    for (const [k, v] of input.entries()) {
+      encoded.set(
+        encodeJsonCompatValue(k),
+        encodeJsonCompatValue(v)
+      );
+    }
+    return encoded;
+  }
   if (Array.isArray(input)) {
-    const encoded = input.map((value) => encodeJsonCompatValue(value));
+    const encoded = input.map(
+      (value) => encodeJsonCompatValue(value)
+    );
     if (encoded.length === 2 && typeof encoded[0] === "string" && encoded[0].startsWith("$")) {
-      return ["$" + encoded[0], encoded[1]];
+      return [`$${encoded[0]}`, encoded[1]];
     }
     return encoded;
   }
@@ -865,7 +911,8 @@ function encodeJsonCompatValue(input) {
     }
     return encoded;
   }
-  return input;
+  const typeName = typeof input === "object" && input !== null ? input.constructor?.name ?? typeof input : typeof input;
+  throw new TypeError(`Value of type "${typeName}" is not CBOR serializable`);
 }
 function reviveJsonCompatValue(input, options = {}) {
   if (typeof input === "bigint") {
@@ -874,6 +921,16 @@ function reviveJsonCompatValue(input, options = {}) {
     }
     return input;
   }
+  if (input instanceof Map) {
+    const revived = /* @__PURE__ */ new Map();
+    for (const [k, v] of input.entries()) {
+      revived.set(
+        reviveJsonCompatValue(k, options),
+        reviveJsonCompatValue(v, options)
+      );
+    }
+    return revived;
+  }
   if (Array.isArray(input)) {
     if (input.length === 2 && typeof input[0] === "string" && input[0].startsWith("$")) {
       if (input[0] === JSON_COMPAT_BIGINT) {
@@ -888,6 +945,12 @@ function reviveJsonCompatValue(input, options = {}) {
       if (input[0] === JSON_COMPAT_UNDEFINED) {
         return void 0;
       }
+      if (input[0] === JSON_COMPAT_SET) {
+        const items = input[1].map(
+          (v) => reviveJsonCompatValue(v, options)
+        );
+        return new Set(items);
+      }
       if (input[0].startsWith("$$")) {
         return [
           input[0].substring(1),
@@ -929,19 +992,13 @@ function jsonStringifyCompat(input) {
       return [JSON_COMPAT_BIGINT, value.toString()];
     }
     if (value instanceof ArrayBuffer) {
-      return [
-        JSON_COMPAT_ARRAY_BUFFER,
-        base64EncodeArrayBuffer(value)
-      ];
+      return [JSON_COMPAT_ARRAY_BUFFER, base64EncodeArrayBuffer(value)];
     }
     if (value instanceof Uint8Array) {
-      return [
-        JSON_COMPAT_UINT8_ARRAY,
-        base64EncodeUint8Array(value)
-      ];
+      return [JSON_COMPAT_UINT8_ARRAY, base64EncodeUint8Array(value)];
     }
     if (Array.isArray(value) && value.length === 2 && typeof value[0] === "string" && value[0].startsWith("$")) {
-      return ["$" + value[0], value[1]];
+      return [`$${value[0]}`, value[1]];
     }
     return value;
   });
@@ -3905,11 +3962,11 @@ async function lookupMetadataCached(config) {
 }
 
 // src/engine-client/ws-proxy.ts
-async function createWebSocketProxy(c, targetUrl, protocols) {
+async function createWebSocketProxy(_c, targetUrl, protocols) {
   const WebSocket2 = await importWebSocket();
   const state = {};
   return {
-    onOpen: async (event, clientWs) => {
+    onOpen: async (_event, clientWs) => {
       logger3().debug({ msg: "client websocket connected", targetUrl });
       if (clientWs.readyState !== 1) {
         logger3().warn({
@@ -3947,23 +4004,23 @@ async function createWebSocketProxy(c, targetUrl, protocols) {
           reject(error);
         });
       });
-      state.targetWs.addEventListener("message", (event2) => {
-        if (typeof event2.data === "string" || event2.data instanceof ArrayBuffer) {
-          clientWs.send(event2.data);
-        } else if (event2.data instanceof Blob) {
-          event2.data.arrayBuffer().then((buffer) => {
+      state.targetWs.addEventListener("message", (event) => {
+        if (typeof event.data === "string" || event.data instanceof ArrayBuffer) {
+          clientWs.send(event.data);
+        } else if (event.data instanceof Blob) {
+          event.data.arrayBuffer().then((buffer) => {
             clientWs.send(buffer);
           });
         }
       });
-      state.targetWs.addEventListener("close", (event2) => {
+      state.targetWs.addEventListener("close", (event) => {
         logger3().debug({
           msg: "target websocket closed",
           targetUrl,
-          code: event2.code,
-          reason: event2.reason
+          code: event.code,
+          reason: event.reason
         });
-        closeWebSocketIfOpen(clientWs, event2.code, event2.reason);
+        closeWebSocketIfOpen(clientWs, event.code, event.reason);
       });
       state.targetWs.addEventListener("error", (error) => {
         logger3().error({
@@ -4006,7 +4063,7 @@ async function createWebSocketProxy(c, targetUrl, protocols) {
         );
       }
     },
-    onClose: (event, clientWs) => {
+    onClose: (event, _clientWs) => {
       logger3().debug({
         msg: "client websocket closed",
         targetUrl,
@@ -4023,7 +4080,7 @@ async function createWebSocketProxy(c, targetUrl, protocols) {
         }
       }
     },
-    onError: (event, clientWs) => {
+    onError: (event, _clientWs) => {
       logger3().error({ msg: "client websocket error", targetUrl, event });
       if (state.targetWs) {
         if (state.targetWs.readyState === WebSocket2.OPEN) {
@@ -4139,7 +4196,9 @@ var RemoteEngineControlClient = class {
       name,
       key: serializeActorKey(key),
       runner_name_selector: this.#config.poolName,
-      input: actorInput ? uint8ArrayToBase64(encodeCborCompat(actorInput)) : void 0,
+      input: actorInput ? uint8ArrayToBase64(
+        encodeCborCompat(actorInput)
+      ) : void 0,
       crash_policy: crashPolicy ?? "sleep"
     });
     logger3().info({
@@ -4530,7 +4589,7 @@ function classifyActorError(error) {
   return void 0;
 }
 function isRetryableGuardGatewayHttpError(code) {
-  return code === "service_unavailable" || code === "actor_stopped_while_waiting" || code === "tunnel_request_aborted" || code === "tunnel_message_timeout" || code === "tunnel_response_closed" || code === "gateway_response_start_timeout";
+  return code === "service_unavailable" || code === "actor_wake_retries_exceeded" || code === "actor_stopped_while_waiting" || code === "tunnel_request_aborted" || code === "tunnel_message_timeout" || code === "tunnel_response_closed" || code === "gateway_response_start_timeout";
 }
 function classifyTransportError(error) {
   if (error.message.includes("database accessed after actor stopped")) {
@@ -4619,10 +4678,7 @@ async function retryOnLifecycleBoundary(run, opts) {
       if (attempt === maxAttempts - 1) {
         break;
       }
-      const delayMs = Math.min(
-        initialDelayMs * 2 ** attempt,
-        maxDelayMs
-      );
+      const delayMs = Math.min(initialDelayMs * 2 ** attempt, maxDelayMs);
       await waitWithSignal(delayMs, opts?.signal);
     }
   }
@@ -4661,7 +4717,9 @@ function createQueueSender(senderOptions) {
       }),
       requestToBare: (value) => ({
         name: value.name ?? name,
-        body: bufferToArrayBuffer(encodeCborCompat(value.body)),
+        body: bufferToArrayBuffer(
+          encodeCborCompat(value.body)
+        ),
         wait: value.wait ?? false,
         timeout: value.timeout !== void 0 ? BigInt(value.timeout) : null
       }),
@@ -4695,7 +4753,6 @@ function createQueueSender(senderOptions) {
 }
 
 // src/client/actor-conn.ts
-var DEFAULT_MAX_INCOMING_MESSAGE_SIZE = 65536;
 var CONNECT_SYMBOL = /* @__PURE__ */ Symbol("connect");
 var ActorConnRaw = class {
   #disposed = false;
@@ -5174,12 +5231,10 @@ var ActorConnRaw = class {
           );
           return;
         }
-        let errorToThrow = new RivetError(
-          group,
-          code,
-          message,
-          { metadata, actor }
-        );
+        let errorToThrow = new RivetError(group, code, message, {
+          metadata,
+          actor
+        });
         if (isSchedulingError(group, code) && this.#actorId) {
           const schedulingError = await checkForSchedulingError(
             group,
@@ -5575,31 +5630,8 @@ var ActorConnRaw = class {
               }
             }
           );
-          const serializedLength = messageLength(messageSerialized);
-          if (serializedLength > DEFAULT_MAX_INCOMING_MESSAGE_SIZE && message.body.tag === "ActionRequest") {
-            const actionId = Number(message.body.val.id);
-            const inFlight = this.#takeActionInFlight(actionId);
-            const error = new RivetError(
-              "message",
-              "incoming_too_long",
-              "Incoming message too long",
-              {
-                maxSize: DEFAULT_MAX_INCOMING_MESSAGE_SIZE,
-                actualSize: serializedLength
-              }
-            );
-            logger2().warn({
-              msg: "rejecting oversized connection action request",
-              actionId,
-              actionName: inFlight.name,
-              actualSize: serializedLength,
-              maxSize: DEFAULT_MAX_INCOMING_MESSAGE_SIZE
-            });
-            inFlight.reject(error);
-            this.#dispatchActorError(error);
-            return;
-          }
           this.#websocket.send(messageSerialized);
+          const serializedLength = messageLength(messageSerialized);
           logger2().trace({
             msg: "sent websocket message",
             len: serializedLength
@@ -6023,10 +6055,7 @@ var ActorHandleRaw = class {
             }
           }).send(name, body, options);
         } catch (err) {
-          const { group, code, message, metadata, actor } = deconstructError(
-            err,
-            true
-          );
+          const { group, code, message, metadata, actor } = deconstructError(err, true);
           if (this.#shouldRetryQueueDispatchOverload(
             group,
             code,
@@ -6059,7 +6088,10 @@ var ActorHandleRaw = class {
             await this.#waitForRetryWindow();
             continue;
           }
-          const invalidated = this.#invalidateResolvedActorId(group, code);
+          const invalidated = this.#invalidateResolvedActorId(
+            group,
+            code
+          );
           if (invalidated && attempt < maxAttempts - 1) {
             useQueryTarget = code === "starting" || code === "stopping" || code.startsWith("destroyed_");
             if (useQueryTarget) {
@@ -6067,7 +6099,10 @@ var ActorHandleRaw = class {
             }
             continue;
           }
-          throw new RivetError(group, code, message, { metadata, actor });
+          throw new RivetError(group, code, message, {
+            metadata,
+            actor
+          });
         }
       }
       throw new Error("unreachable queue retry state");
@@ -6147,7 +6182,9 @@ var ActorHandleRaw = class {
             args
           }),
           requestToBare: (args) => ({
-            args: bufferToArrayBuffer(encodeCborCompat(args))
+            args: bufferToArrayBuffer(
+              encodeCborCompat(args)
+            )
           }),
           responseFromJson: (json) => json.output,
           responseFromBare: (bare5) => decodeCborCompat(new Uint8Array(bare5.output))
@@ -6157,10 +6194,7 @@ var ActorHandleRaw = class {
         }
         return output;
       } catch (err) {
-        const { group, code, message, metadata, actor } = deconstructError(
-          err,
-          true
-        );
+        const { group, code, message, metadata, actor } = deconstructError(err, true);
         if (await this.#shouldRetrySchedulingError(
           group,
           code,
@@ -6191,11 +6225,16 @@ var ActorHandleRaw = class {
             { metadata, actor }
           );
         }
-        const invalidated = this.#invalidateResolvedActorId(group, code);
+        const invalidated = this.#invalidateResolvedActorId(
+          group,
+          code
+        );
         if (invalidated && attempt < maxAttempts - 1) {
           if (group === "actor" && (code === "starting" || code === "stopping")) {
             useQueryTarget = true;
-            await new Promise((resolve) => setTimeout(resolve, 100));
+            await new Promise(
+              (resolve) => setTimeout(resolve, 100)
+            );
           }
           continue;
         }
@@ -6372,10 +6411,7 @@ var ActorHandleRaw = class {
         }
         return response;
       } catch (err) {
-        const { group, code, message, metadata, actor } = deconstructError(
-          err,
-          true
-        );
+        const { group, code, message, metadata, actor } = deconstructError(err, true);
         if (await this.#shouldRetrySchedulingError(
           group,
           code,
@@ -6398,7 +6434,10 @@ var ActorHandleRaw = class {
           await this.#waitForRetryWindow();
           continue;
         }
-        const invalidated = this.#invalidateResolvedActorId(group, code);
+        const invalidated = this.#invalidateResolvedActorId(
+          group,
+          code
+        );
         if (invalidated && attempt < maxAttempts - 1) {
           useQueryTarget = code === "starting" || code === "stopping" || code.startsWith("destroyed_");
           if (useQueryTarget) {
@@ -6522,7 +6561,9 @@ var ActorHandleRaw = class {
     if ("directId" in target) {
       return target.directId;
     }
-    throw new Error("dynamic actor resolution did not produce a direct actor id");
+    throw new Error(
+      "dynamic actor resolution did not produce a direct actor id"
+    );
   }
   /**
    * Returns the raw URL for routing traffic to the actor.