rivetkit 2.3.0-rc.7 → 2.3.0-rc.9

package/dist/browser/client.d.ts

@@ -929,7 +929,6 @@ type Client<A extends Registry<any>> = ClientRaw & {
 };
 
 declare const ACTOR_CONTEXT_INTERNAL_SYMBOL: unique symbol;
-declare const RAW_STATE_SYMBOL: unique symbol;
 interface ActorLogger {
     level: any;
     fatal: any;
@@ -1049,8 +1048,6 @@ interface Conn<TState = unknown, TConnParams = unknown, TConnState = unknown, TV
 }
 interface ActorContext<TState, TConnParams, TConnState, TVars, TInput, TDatabase extends AnyDatabaseProvider, TEvents extends EventSchemaConfig = Record<never, never>, TQueues extends QueueSchemaConfig = Record<never, never>> {
     [ACTOR_CONTEXT_INTERNAL_SYMBOL]?: unknown;
-    /** Returns the raw unwrapped state without the write-through proxy. */
-    [RAW_STATE_SYMBOL](): TState;
     state: TState;
     vars: TVars;
     readonly kv: ActorKv;
@@ -1409,11 +1406,7 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
     }>>, z$2.ZodTransform<"auto" | "native" | "wasm", "auto" | "native" | "wasm" | undefined>>;
     wasm: z$2.ZodDefault<z$2.ZodOptional<z$2.ZodObject<{
         bindings: z$2.ZodOptional<z$2.ZodCustom<typeof _rivetkit_rivetkit_wasm, typeof _rivetkit_rivetkit_wasm>>;
-        initInput: z$2.ZodOptional<z$2.ZodCustom<{
-            module_or_path: _rivetkit_rivetkit_wasm.InitInput | Promise<_rivetkit_rivetkit_wasm.InitInput>;
-        } | _rivetkit_rivetkit_wasm.InitInput | Promise<_rivetkit_rivetkit_wasm.InitInput> | undefined, {
-            module_or_path: _rivetkit_rivetkit_wasm.InitInput | Promise<_rivetkit_rivetkit_wasm.InitInput>;
-        } | _rivetkit_rivetkit_wasm.InitInput | Promise<_rivetkit_rivetkit_wasm.InitInput> | undefined>>;
+        initInput: z$2.ZodOptional<z$2.ZodCustom<any, any>>;
     }, z$2.core.$strip>>>;
     sqlite: z$2.ZodPipe<z$2.ZodOptional<z$2.ZodUnion<readonly [z$2.ZodEnum<{
         local: "local";
@@ -1504,9 +1497,7 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
     runtime: "auto" | "native" | "wasm";
     wasm: {
         bindings?: typeof _rivetkit_rivetkit_wasm | undefined;
-        initInput?: {
-            module_or_path: _rivetkit_rivetkit_wasm.InitInput | Promise<_rivetkit_rivetkit_wasm.InitInput>;
-        } | _rivetkit_rivetkit_wasm.InitInput | Promise<_rivetkit_rivetkit_wasm.InitInput> | undefined;
+        initInput?: any;
     };
     noWelcome: boolean;
     logging: {
@@ -1551,9 +1542,7 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
     runtime: "auto" | "native" | "wasm";
     wasm: {
         bindings?: typeof _rivetkit_rivetkit_wasm | undefined;
-        initInput?: {
-            module_or_path: _rivetkit_rivetkit_wasm.InitInput | Promise<_rivetkit_rivetkit_wasm.InitInput>;
-        } | _rivetkit_rivetkit_wasm.InitInput | Promise<_rivetkit_rivetkit_wasm.InitInput> | undefined;
+        initInput?: any;
     };
     sqlite: {
         backend: "local" | "remote";
@@ -1609,12 +1598,14 @@ type FetchHandler = (request: Request, ...args: any) => Response | Promise<Respo
 interface ServerlessHandler {
     fetch: FetchHandler;
 }
-interface RegistryDiagnostics {
-    mode: string;
-    envoyActiveActorCount?: number | null;
+interface RegistryRoutes {
+    health(): Promise<Response>;
+    metadata(): Promise<Response>;
+    prometheusMetrics(request?: Request): Promise<Response>;
 }
 declare class Registry<A extends RegistryActors> {
     #private;
+    readonly routes: RegistryRoutes;
     get config(): RegistryConfigInput<A>;
     parseConfig(): RegistryConfig;
     constructor(config: RegistryConfigInput<A>);
@@ -1638,26 +1629,6 @@ declare class Registry<A extends RegistryActors> {
      * ```
      */
     serve(): ServerlessHandler;
-    /**
-     * Starts an HTTP server that dispatches every request through the
-     * serverless handler. Uses `crossPlatformServe` to pick the right
-     * runtime (Node, Bun, Deno).
-     *
-     * @param opts.port      Port to listen on. Defaults to 3000.
-     * @param opts.publicDir If set, serves static files from this directory
-     *                       before falling through to the registry handler.
-     *
-     * @example
-     * ```ts
-     * await registry.listen();
-     * await registry.listen({ port: 8080, publicDir: "./public" });
-     * ```
-     */
-    listen(opts?: {
-        port?: number;
-        publicDir?: string;
-    }): Promise<void>;
-    diagnostics(): Promise<RegistryDiagnostics>;
     startEnvoy(): void;
     /**
      * Starts the actor envoy for standalone server deployments.

package/dist/browser/client.js

@@ -200,7 +200,7 @@ function noopNext() {
 // package.json
 var package_default = {
   name: "rivetkit",
-  version: "2.3.0-rc.7",
+  version: "2.3.0-rc.9",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [
@@ -368,14 +368,13 @@ 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:*",
@@ -393,10 +392,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",
@@ -472,7 +471,6 @@ 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) {
@@ -807,10 +805,6 @@ 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") {
@@ -845,15 +839,9 @@ 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()];
   }
@@ -863,32 +851,8 @@ 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]];
     }
@@ -901,10 +865,7 @@ function encodeJsonCompatValue(input) {
     }
     return encoded;
   }
-  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`
-  );
+  return input;
 }
 function reviveJsonCompatValue(input, options = {}) {
   if (typeof input === "bigint") {
@@ -913,16 +874,6 @@ 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) {
@@ -937,12 +888,6 @@ 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),
@@ -4750,6 +4695,7 @@ 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;
@@ -4801,6 +4747,9 @@ var ActorConnRaw = class {
     this.#encoding = encoding;
     this.#actorResolutionState = actorResolutionState;
     this.#gatewayOptions = resolveActorGatewayOptions(gatewayOptions);
+    if ("getForId" in actorResolutionState) {
+      this.#actorId = actorResolutionState.getForId.actorId;
+    }
     this.#readyPromise = promiseWithResolvers(
       (reason) => logger2().warn({
         msg: "unhandled ready promise rejection",
@@ -4818,9 +4767,7 @@ var ActorConnRaw = class {
         );
       }
     });
-    this.#keepNodeAliveInterval = setInterval(() => {
-    }, 6e4);
-    this.#keepNodeAliveInterval.unref?.();
+    this.#keepNodeAliveInterval = setInterval(() => 6e4);
   }
   #clearResolvedActorIdentity() {
     this.#actorId = void 0;
@@ -5628,8 +5575,31 @@ var ActorConnRaw = class {
               }
             }
           );
-          this.#websocket.send(messageSerialized);
           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);
           logger2().trace({
             msg: "sent websocket message",
             len: serializedLength