rivetkit 2.0.7-rc.1 → 2.0.8

package/dist/tsup/{chunk-3Y45CIF4.cjs → chunk-CKSA7NOS.cjs}

@@ -4,7 +4,7 @@
 
 
 
-var _chunkG4ABMAQYcjs = require('./chunk-G4ABMAQY.cjs');
+var _chunkQGUQB3NCcjs = require('./chunk-QGUQB3NC.cjs');
 
 
 
@@ -14,10 +14,10 @@ var _chunkG4ABMAQYcjs = require('./chunk-G4ABMAQY.cjs');
 
 
 
-var _chunk5ZOHIKWGcjs = require('./chunk-5ZOHIKWG.cjs');
+var _chunkKDNB2BQXcjs = require('./chunk-KDNB2BQX.cjs');
 
 
-var _chunkB3TLRM4Qcjs = require('./chunk-B3TLRM4Q.cjs');
+var _chunkL5MHM6JJcjs = require('./chunk-L5MHM6JJ.cjs');
 
 
 
@@ -28,7 +28,7 @@ var _chunkB3TLRM4Qcjs = require('./chunk-B3TLRM4Q.cjs');
 
 
 
-var _chunkDFS77KAAcjs = require('./chunk-DFS77KAA.cjs');
+var _chunk3WRAGTDCcjs = require('./chunk-3WRAGTDC.cjs');
 
 
 
@@ -54,12 +54,12 @@ var _chunkDFS77KAAcjs = require('./chunk-DFS77KAA.cjs');
 
 
 
-var _chunkSAZCNSVYcjs = require('./chunk-SAZCNSVY.cjs');
+var _chunkFGOZELKNcjs = require('./chunk-FGOZELKN.cjs');
 
 
 
 
-var _chunk3MBP4WNCcjs = require('./chunk-3MBP4WNC.cjs');
+var _chunkTPJNKVFBcjs = require('./chunk-TPJNKVFB.cjs');
 
 
 
@@ -69,7 +69,7 @@ var _chunk3MBP4WNCcjs = require('./chunk-3MBP4WNC.cjs');
 
 
 
-var _chunk6EUWRXLTcjs = require('./chunk-6EUWRXLT.cjs');
+var _chunkDQVVH5ZKcjs = require('./chunk-DQVVH5ZK.cjs');
 
 
 
@@ -151,7 +151,7 @@ var _streaming = require('hono/streaming');
 
 // src/manager/log.ts
 function logger() {
-  return _chunk3MBP4WNCcjs.getLogger.call(void 0, "actor-manager");
+  return _chunkTPJNKVFBcjs.getLogger.call(void 0, "actor-manager");
 }
 
 // src/manager/hono-websocket-adapter.ts
@@ -439,14 +439,14 @@ async function handleWebSocketConnect(req, runConfig, actorDriver, actorId, enco
     promise: handlersPromise,
     resolve: handlersResolve,
     reject: handlersReject
-  } = _chunk6EUWRXLTcjs.promiseWithResolvers.call(void 0, );
+  } = _chunkDQVVH5ZKcjs.promiseWithResolvers.call(void 0, );
   let actor2;
   try {
     actor2 = await actorDriver.loadActor(actorId);
   } catch (error) {
     return {
       onOpen: (_evt, ws) => {
-        const { code } = _chunk6EUWRXLTcjs.deconstructError.call(void 0, 
+        const { code } = _chunkDQVVH5ZKcjs.deconstructError.call(void 0, 
           error,
           actor2.rLog,
           {
@@ -465,8 +465,8 @@ async function handleWebSocketConnect(req, runConfig, actorDriver, actorId, enco
       }
     };
   }
-  const closePromise = _chunk6EUWRXLTcjs.promiseWithResolvers.call(void 0, );
-  const socketId = _chunk5ZOHIKWGcjs.generateConnSocketId.call(void 0, );
+  const closePromise = _chunkDQVVH5ZKcjs.promiseWithResolvers.call(void 0, );
+  const socketId = _chunkKDNB2BQXcjs.generateConnSocketId.call(void 0, );
   return {
     onOpen: (_evt, ws) => {
       actor2.rLog.debug("actor websocket open");
@@ -497,7 +497,7 @@ async function handleWebSocketConnect(req, runConfig, actorDriver, actorId, enco
           handlersResolve({ conn, actor: actor2, connId: conn.id });
         } catch (error) {
           handlersReject(error);
-          const { code } = _chunk6EUWRXLTcjs.deconstructError.call(void 0, 
+          const { code } = _chunkDQVVH5ZKcjs.deconstructError.call(void 0, 
             error,
             actor2.rLog,
             {
@@ -513,12 +513,12 @@ async function handleWebSocketConnect(req, runConfig, actorDriver, actorId, enco
       handlersPromise.then(({ conn, actor: actor3 }) => {
         actor3.rLog.debug({ msg: "received message" });
         const value = evt.data.valueOf();
-        _chunkDFS77KAAcjs.parseMessage.call(void 0, value, {
+        _chunk3WRAGTDCcjs.parseMessage.call(void 0, value, {
           encoding,
           maxIncomingMessageSize: runConfig.maxIncomingMessageSize
         }).then((message) => {
           actor3.processMessage(message, conn).catch((error) => {
-            const { code } = _chunk6EUWRXLTcjs.deconstructError.call(void 0, 
+            const { code } = _chunkDQVVH5ZKcjs.deconstructError.call(void 0, 
               error,
               actor3.rLog,
               {
@@ -529,7 +529,7 @@ async function handleWebSocketConnect(req, runConfig, actorDriver, actorId, enco
             ws.close(1011, code);
           });
         }).catch((error) => {
-          const { code } = _chunk6EUWRXLTcjs.deconstructError.call(void 0, 
+          const { code } = _chunkDQVVH5ZKcjs.deconstructError.call(void 0, 
             error,
             actor3.rLog,
             {
@@ -540,7 +540,7 @@ async function handleWebSocketConnect(req, runConfig, actorDriver, actorId, enco
           ws.close(1011, code);
         });
       }).catch((error) => {
-        const { code } = _chunk6EUWRXLTcjs.deconstructError.call(void 0, 
+        const { code } = _chunkDQVVH5ZKcjs.deconstructError.call(void 0, 
           error,
           actor2.rLog,
           {
@@ -571,9 +571,10 @@ async function handleWebSocketConnect(req, runConfig, actorDriver, actorId, enco
       }
       ws.close(1e3, "hack_force_close");
       handlersPromise.then(({ conn, actor: actor3 }) => {
-        actor3.__connDisconnected(conn, event.wasClean, socketId);
+        const wasClean = event.wasClean || event.code === 1e3;
+        actor3.__connDisconnected(conn, wasClean, socketId);
       }).catch((error) => {
-        _chunk6EUWRXLTcjs.deconstructError.call(void 0, 
+        _chunkDQVVH5ZKcjs.deconstructError.call(void 0, 
           error,
           actor2.rLog,
           { wsEvent: "close" },
@@ -585,7 +586,7 @@ async function handleWebSocketConnect(req, runConfig, actorDriver, actorId, enco
       try {
         actor2.rLog.warn({ msg: "websocket error" });
       } catch (error) {
-        _chunk6EUWRXLTcjs.deconstructError.call(void 0, 
+        _chunkDQVVH5ZKcjs.deconstructError.call(void 0, 
           error,
           actor2.rLog,
           { wsEvent: "error" },
@@ -599,9 +600,9 @@ async function handleSseConnect(c, _runConfig, actorDriver, actorId) {
   c.header("Content-Encoding", "Identity");
   const encoding = getRequestEncoding(c.req);
   const parameters = getRequestConnParams(c.req);
-  const socketId = _chunk5ZOHIKWGcjs.generateConnSocketId.call(void 0, );
-  const connId = c.req.header(_chunkSAZCNSVYcjs.HEADER_CONN_ID);
-  const connToken = c.req.header(_chunkSAZCNSVYcjs.HEADER_CONN_TOKEN);
+  const socketId = _chunkKDNB2BQXcjs.generateConnSocketId.call(void 0, );
+  const connId = c.req.header(_chunkFGOZELKNcjs.HEADER_CONN_ID);
+  const connToken = c.req.header(_chunkFGOZELKNcjs.HEADER_CONN_TOKEN);
   return _streaming.streamSSE.call(void 0, c, async (stream) => {
     let actor2;
     let conn;
@@ -626,12 +627,12 @@ async function handleSseConnect(c, _runConfig, actorDriver, actorId) {
         connId,
         connToken
       );
-      const abortResolver = _chunk6EUWRXLTcjs.promiseWithResolvers.call(void 0, );
+      const abortResolver = _chunkDQVVH5ZKcjs.promiseWithResolvers.call(void 0, );
       stream.onAbort(() => {
       });
       c.req.raw.signal.addEventListener("abort", async () => {
         _invariant2.default.call(void 0, actor2, "actor should exist");
-        const rLog = _nullishCoalesce(actor2.rLog, () => ( _chunkSAZCNSVYcjs.loggerWithoutContext.call(void 0, )));
+        const rLog = _nullishCoalesce(actor2.rLog, () => ( _chunkFGOZELKNcjs.loggerWithoutContext.call(void 0, )));
         try {
           rLog.debug("sse stream aborted");
           if (conn) {
@@ -656,7 +657,7 @@ async function handleSseConnect(c, _runConfig, actorDriver, actorId) {
         await stream.sleep(SSE_PING_INTERVAL);
       }
     } catch (error) {
-      _chunkSAZCNSVYcjs.loggerWithoutContext.call(void 0, ).error({ msg: "error in sse connection", error });
+      _chunkFGOZELKNcjs.loggerWithoutContext.call(void 0, ).error({ msg: "error in sse connection", error });
       if (conn && actor2 !== void 0) {
         actor2.__connDisconnected(conn, false, socketId);
       }
@@ -668,13 +669,13 @@ async function handleAction(c, _runConfig, actorDriver, actionName, actorId) {
   const encoding = getRequestEncoding(c.req);
   const parameters = getRequestConnParams(c.req);
   const arrayBuffer = await c.req.arrayBuffer();
-  const request = _chunkSAZCNSVYcjs.deserializeWithEncoding.call(void 0, 
+  const request = _chunkFGOZELKNcjs.deserializeWithEncoding.call(void 0, 
     encoding,
     new Uint8Array(arrayBuffer),
-    _chunkDFS77KAAcjs.HTTP_ACTION_REQUEST_VERSIONED
+    _chunk3WRAGTDCcjs.HTTP_ACTION_REQUEST_VERSIONED
   );
   const actionArgs = cbor.decode(new Uint8Array(request.args));
-  const socketId = _chunk5ZOHIKWGcjs.generateConnSocketId.call(void 0, );
+  const socketId = _chunkKDNB2BQXcjs.generateConnSocketId.call(void 0, );
   let actor2;
   let conn;
   let output;
@@ -689,7 +690,7 @@ async function handleAction(c, _runConfig, actorDriver, actionName, actorId) {
       parameters,
       c.req.raw
     );
-    const ctx = new (0, _chunkDFS77KAAcjs.ActionContext)(actor2.actorContext, conn);
+    const ctx = new (0, _chunk3WRAGTDCcjs.ActionContext)(actor2.actorContext, conn);
     output = await actor2.executeAction(ctx, actionName, actionArgs);
   } finally {
     if (conn) {
@@ -697,24 +698,24 @@ async function handleAction(c, _runConfig, actorDriver, actionName, actorId) {
     }
   }
   const responseData = {
-    output: _chunk6EUWRXLTcjs.bufferToArrayBuffer.call(void 0, cbor.encode(output))
+    output: _chunkDQVVH5ZKcjs.bufferToArrayBuffer.call(void 0, cbor.encode(output))
   };
-  const serialized = _chunkSAZCNSVYcjs.serializeWithEncoding.call(void 0, 
+  const serialized = _chunkFGOZELKNcjs.serializeWithEncoding.call(void 0, 
     encoding,
     responseData,
-    _chunkDFS77KAAcjs.HTTP_ACTION_RESPONSE_VERSIONED
+    _chunk3WRAGTDCcjs.HTTP_ACTION_RESPONSE_VERSIONED
   );
   return c.body(serialized, 200, {
-    "Content-Type": _chunkSAZCNSVYcjs.contentTypeForEncoding.call(void 0, encoding)
+    "Content-Type": _chunkFGOZELKNcjs.contentTypeForEncoding.call(void 0, encoding)
   });
 }
 async function handleConnectionMessage(c, _runConfig, actorDriver, connId, connToken, actorId) {
   const encoding = getRequestEncoding(c.req);
   const arrayBuffer = await c.req.arrayBuffer();
-  const message = _chunkSAZCNSVYcjs.deserializeWithEncoding.call(void 0, 
+  const message = _chunkFGOZELKNcjs.deserializeWithEncoding.call(void 0, 
     encoding,
     new Uint8Array(arrayBuffer),
-    _chunkDFS77KAAcjs.TO_SERVER_VERSIONED
+    _chunk3WRAGTDCcjs.TO_SERVER_VERSIONED
   );
   const actor2 = await actorDriver.loadActor(actorId);
   const conn = actor2.conns.get(connId);
@@ -795,11 +796,11 @@ async function handleRawWebSocketHandler(req, path3, actorDriver, actorId) {
   };
 }
 function getRequestEncoding(req) {
-  const encodingParam = req.header(_chunkSAZCNSVYcjs.HEADER_ENCODING);
+  const encodingParam = req.header(_chunkFGOZELKNcjs.HEADER_ENCODING);
   if (!encodingParam) {
     throw new (0, _chunk5QGQK44Lcjs.InvalidEncoding)("undefined");
   }
-  const result = _chunkSAZCNSVYcjs.EncodingSchema.safeParse(encodingParam);
+  const result = _chunkFGOZELKNcjs.EncodingSchema.safeParse(encodingParam);
   if (!result.success) {
     throw new (0, _chunk5QGQK44Lcjs.InvalidEncoding)(encodingParam);
   }
@@ -809,7 +810,7 @@ function getRequestExposeInternalError(_req) {
   return false;
 }
 function getRequestConnParams(req) {
-  const paramsParam = req.header(_chunkSAZCNSVYcjs.HEADER_CONN_PARAMS);
+  const paramsParam = req.header(_chunkFGOZELKNcjs.HEADER_CONN_PARAMS);
   if (!paramsParam) {
     return null;
   }
@@ -817,7 +818,7 @@ function getRequestConnParams(req) {
     return JSON.parse(paramsParam);
   } catch (err) {
     throw new (0, _chunk5QGQK44Lcjs.InvalidParams)(
-      `Invalid params JSON: ${_chunk6EUWRXLTcjs.stringifyError.call(void 0, err)}`
+      `Invalid params JSON: ${_chunkDQVVH5ZKcjs.stringifyError.call(void 0, err)}`
     );
   }
 }
@@ -825,7 +826,7 @@ function getRequestConnParams(req) {
 // src/common/router.ts
 
 function logger2() {
-  return _chunk3MBP4WNCcjs.getLogger.call(void 0, "router");
+  return _chunkTPJNKVFBcjs.getLogger.call(void 0, "router");
 }
 function loggerMiddleware(logger7) {
   return async (c, next) => {
@@ -851,7 +852,7 @@ function handleRouteNotFound(c) {
 }
 function handleRouteError(error, c) {
   const exposeInternalError = getRequestExposeInternalError(c.req.raw);
-  const { statusCode, group, code, message, metadata } = _chunk6EUWRXLTcjs.deconstructError.call(void 0, 
+  const { statusCode, group, code, message, metadata } = _chunkDQVVH5ZKcjs.deconstructError.call(void 0, 
     error,
     logger2(),
     {
@@ -866,24 +867,24 @@ function handleRouteError(error, c) {
   } catch (_) {
     encoding = "json";
   }
-  const output = _chunkSAZCNSVYcjs.serializeWithEncoding.call(void 0, 
+  const output = _chunkFGOZELKNcjs.serializeWithEncoding.call(void 0, 
     encoding,
     {
       group,
       code,
       message,
       // TODO: Cannot serialize non-binary meta since it requires ArrayBuffer atm
-      metadata: _chunkSAZCNSVYcjs.encodingIsBinary.call(void 0, encoding) ? _chunk6EUWRXLTcjs.bufferToArrayBuffer.call(void 0, cbor2.encode(metadata)) : null
+      metadata: _chunkFGOZELKNcjs.encodingIsBinary.call(void 0, encoding) ? _chunkDQVVH5ZKcjs.bufferToArrayBuffer.call(void 0, cbor2.encode(metadata)) : null
     },
-    _chunkDFS77KAAcjs.HTTP_RESPONSE_ERROR_VERSIONED
+    _chunk3WRAGTDCcjs.HTTP_RESPONSE_ERROR_VERSIONED
   );
   return c.body(output, { status: statusCode });
 }
 
 // src/actor/router.ts
-function createActorRouter(runConfig, actorDriver) {
+function createActorRouter(runConfig, actorDriver, isTest) {
   const router = new (0, _hono.Hono)({ strict: false });
-  router.use("*", loggerMiddleware(_chunkSAZCNSVYcjs.loggerWithoutContext.call(void 0, )));
+  router.use("*", loggerMiddleware(_chunkFGOZELKNcjs.loggerWithoutContext.call(void 0, )));
   router.get("/", (c) => {
     return c.text(
       "This is an RivetKit actor.\n\nLearn more at https://rivetkit.org"
@@ -892,27 +893,29 @@ function createActorRouter(runConfig, actorDriver) {
   router.get("/health", (c) => {
     return c.text("ok");
   });
-  router.post("/.test/force-disconnect", async (c) => {
-    const connId = c.req.query("conn");
-    if (!connId) {
-      return c.text("Missing conn query parameter", 400);
-    }
-    const actor2 = await actorDriver.loadActor(c.env.actorId);
-    const conn = actor2.__getConnForId(connId);
-    if (!conn) {
-      return c.text(`Connection not found: ${connId}`, 404);
-    }
-    const driverState = conn.__driverState;
-    if (driverState && 0 /* WEBSOCKET */ in driverState) {
-      const ws = driverState[0 /* WEBSOCKET */].websocket;
-      ws.raw.terminate();
-    } else if (driverState && 1 /* SSE */ in driverState) {
-      const stream = driverState[1 /* SSE */].stream;
-      stream.abort();
-    }
-    return c.json({ success: true });
-  });
-  router.get(_chunkSAZCNSVYcjs.PATH_CONNECT_WEBSOCKET, async (c) => {
+  if (isTest) {
+    router.post("/.test/force-disconnect", async (c) => {
+      const connId = c.req.query("conn");
+      if (!connId) {
+        return c.text("Missing conn query parameter", 400);
+      }
+      const actor2 = await actorDriver.loadActor(c.env.actorId);
+      const conn = actor2.__getConnForId(connId);
+      if (!conn) {
+        return c.text(`Connection not found: ${connId}`, 404);
+      }
+      const driverState = conn.__driverState;
+      if (driverState && 0 /* WEBSOCKET */ in driverState) {
+        const ws = driverState[0 /* WEBSOCKET */].websocket;
+        ws.raw.terminate();
+      } else if (driverState && 1 /* SSE */ in driverState) {
+        const stream = driverState[1 /* SSE */].stream;
+        stream.abort();
+      }
+      return c.json({ success: true });
+    });
+  }
+  router.get(_chunkFGOZELKNcjs.PATH_CONNECT_WEBSOCKET, async (c) => {
     var _a;
     const upgradeWebSocket = (_a = runConfig.getUpgradeWebSocket) == null ? void 0 : _a.call(runConfig);
     if (upgradeWebSocket) {
@@ -925,20 +928,20 @@ function createActorRouter(runConfig, actorDriver) {
         if (protocols) {
           const protocolList = protocols.split(",").map((p) => p.trim());
           for (const protocol of protocolList) {
-            if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_ENCODING)) {
-              encodingRaw = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_ENCODING.length);
-            } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_PARAMS)) {
+            if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_ENCODING)) {
+              encodingRaw = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_ENCODING.length);
+            } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_PARAMS)) {
               connParamsRaw = decodeURIComponent(
-                protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_PARAMS.length)
+                protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_PARAMS.length)
               );
-            } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_ID)) {
-              connIdRaw = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_ID.length);
-            } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_TOKEN)) {
-              connTokenRaw = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_TOKEN.length);
+            } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_ID)) {
+              connIdRaw = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_ID.length);
+            } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_TOKEN)) {
+              connTokenRaw = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_TOKEN.length);
             }
           }
         }
-        const encoding = _chunkSAZCNSVYcjs.EncodingSchema.parse(encodingRaw);
+        const encoding = _chunkFGOZELKNcjs.EncodingSchema.parse(encodingRaw);
         const connParams = connParamsRaw ? JSON.parse(connParamsRaw) : void 0;
         return await handleWebSocketConnect(
           c2.req.raw,
@@ -950,7 +953,7 @@ function createActorRouter(runConfig, actorDriver) {
           connIdRaw,
           connTokenRaw
         );
-      })(c, _chunk6EUWRXLTcjs.noopNext.call(void 0, ));
+      })(c, _chunkDQVVH5ZKcjs.noopNext.call(void 0, ));
     } else {
       return c.text(
         "WebSockets are not enabled for this driver. Use SSE instead.",
@@ -966,8 +969,8 @@ function createActorRouter(runConfig, actorDriver) {
     return handleAction(c, runConfig, actorDriver, actionName, c.env.actorId);
   });
   router.post("/connections/message", async (c) => {
-    const connId = c.req.header(_chunkSAZCNSVYcjs.HEADER_CONN_ID);
-    const connToken = c.req.header(_chunkSAZCNSVYcjs.HEADER_CONN_TOKEN);
+    const connId = c.req.header(_chunkFGOZELKNcjs.HEADER_CONN_ID);
+    const connToken = c.req.header(_chunkFGOZELKNcjs.HEADER_CONN_TOKEN);
     if (!connId || !connToken) {
       throw new Error("Missing required parameters");
     }
@@ -981,8 +984,8 @@ function createActorRouter(runConfig, actorDriver) {
     );
   });
   router.post("/connections/close", async (c) => {
-    const connId = c.req.header(_chunkSAZCNSVYcjs.HEADER_CONN_ID);
-    const connToken = c.req.header(_chunkSAZCNSVYcjs.HEADER_CONN_TOKEN);
+    const connId = c.req.header(_chunkFGOZELKNcjs.HEADER_CONN_ID);
+    const connToken = c.req.header(_chunkFGOZELKNcjs.HEADER_CONN_TOKEN);
     if (!connId || !connToken) {
       throw new Error("Missing required parameters");
     }
@@ -1006,7 +1009,7 @@ function createActorRouter(runConfig, actorDriver) {
       body: c.req.raw.body,
       duplex: "half"
     });
-    _chunkSAZCNSVYcjs.loggerWithoutContext.call(void 0, ).debug({
+    _chunkFGOZELKNcjs.loggerWithoutContext.call(void 0, ).debug({
       msg: "rewriting http url",
       from: c.req.url,
       to: correctedRequest.url
@@ -1017,14 +1020,14 @@ function createActorRouter(runConfig, actorDriver) {
     }
     return response;
   });
-  router.get(`${_chunkSAZCNSVYcjs.PATH_RAW_WEBSOCKET_PREFIX}*`, async (c) => {
+  router.get(`${_chunkFGOZELKNcjs.PATH_RAW_WEBSOCKET_PREFIX}*`, async (c) => {
     var _a;
     const upgradeWebSocket = (_a = runConfig.getUpgradeWebSocket) == null ? void 0 : _a.call(runConfig);
     if (upgradeWebSocket) {
       return upgradeWebSocket(async (c2) => {
         const url = new URL(c2.req.url);
         const pathWithQuery = c2.req.path + url.search;
-        _chunkSAZCNSVYcjs.loggerWithoutContext.call(void 0, ).debug({
+        _chunkFGOZELKNcjs.loggerWithoutContext.call(void 0, ).debug({
           msg: "actor router raw websocket",
           path: c2.req.path,
           url: c2.req.url,
@@ -1037,7 +1040,7 @@ function createActorRouter(runConfig, actorDriver) {
           actorDriver,
           c2.env.actorId
         );
-      })(c, _chunk6EUWRXLTcjs.noopNext.call(void 0, ));
+      })(c, _chunkDQVVH5ZKcjs.noopNext.call(void 0, ));
     } else {
       return c.text(
         "WebSockets are not enabled for this driver. Use SSE instead.",
@@ -1045,19 +1048,19 @@ function createActorRouter(runConfig, actorDriver) {
       );
     }
   });
-  if (_chunkG4ABMAQYcjs.isInspectorEnabled.call(void 0, runConfig, "actor")) {
+  if (_chunkQGUQB3NCcjs.isInspectorEnabled.call(void 0, runConfig, "actor")) {
     router.route(
       "/inspect",
       new (0, _hono.Hono)().use(
         _cors.cors.call(void 0, runConfig.inspector.cors),
-        _chunkG4ABMAQYcjs.secureInspector.call(void 0, runConfig),
+        _chunkQGUQB3NCcjs.secureInspector.call(void 0, runConfig),
         async (c, next) => {
           const inspector = (await actorDriver.loadActor(c.env.actorId)).inspector;
           _invariant2.default.call(void 0, inspector, "inspector not supported on this platform");
           c.set("inspector", inspector);
           return next();
         }
-      ).route("/", _chunk5ZOHIKWGcjs.createActorInspectorRouter.call(void 0, ))
+      ).route("/", _chunkKDNB2BQXcjs.createActorInspectorRouter.call(void 0, ))
     );
   }
   router.notFound(handleRouteNotFound);
@@ -1068,13 +1071,13 @@ function createActorRouter(runConfig, actorDriver) {
 // src/actor/mod.ts
 function actor(input) {
   const config2 = ActorConfigSchema.parse(input);
-  return new (0, _chunk5ZOHIKWGcjs.ActorDefinition)(config2);
+  return new (0, _chunkKDNB2BQXcjs.ActorDefinition)(config2);
 }
 
 // src/common/inline-websocket-adapter2.ts
 var _ws2 = require('hono/ws');
 function logger3() {
-  return _chunk3MBP4WNCcjs.getLogger.call(void 0, "fake-event-source2");
+  return _chunkTPJNKVFBcjs.getLogger.call(void 0, "fake-event-source2");
 }
 var InlineWebSocketAdapter2 = (_class2 = class {
   // WebSocket readyState values
@@ -1422,7 +1425,7 @@ var KEYS = {
 
 // src/drivers/engine/log.ts
 function logger4() {
-  return _chunk3MBP4WNCcjs.getLogger.call(void 0, "driver-engine");
+  return _chunkTPJNKVFBcjs.getLogger.call(void 0, "driver-engine");
 }
 
 // src/drivers/engine/actor-driver.ts
@@ -1437,6 +1440,7 @@ var EngineActorDriver = class {
   #actorRouter;
   #version = 1;
   // Version for the runner protocol
+  #alarmTimeout;
   #runnerStarted = Promise.withResolvers();
   #runnerStopped = Promise.withResolvers();
   constructor(registryConfig, runConfig, managerDriver, inlineClient, config2) {
@@ -1445,15 +1449,19 @@ var EngineActorDriver = class {
     this.#managerDriver = managerDriver;
     this.#inlineClient = inlineClient;
     this.#config = config2;
-    this.#actorRouter = createActorRouter(runConfig, this);
+    this.#actorRouter = createActorRouter(
+      runConfig,
+      this,
+      registryConfig.test.enabled
+    );
     let hasDisconnected = false;
     const runnerConfig = {
       version: this.#version,
       endpoint: config2.endpoint,
-      token: config2.token,
+      token: _nullishCoalesce(runConfig.token, () => ( config2.token)),
       pegboardEndpoint: config2.pegboardEndpoint,
       namespace: config2.namespace,
-      totalSlots: config2.totalSlots,
+      totalSlots: _nullishCoalesce(runConfig.totalSlots, () => ( config2.totalSlots)),
       runnerName: config2.runnerName,
       runnerKey: config2.runnerKey,
       metadata: {
@@ -1496,7 +1504,7 @@ var EngineActorDriver = class {
       websocket: this.#runnerWebSocket.bind(this),
       onActorStart: this.#runnerOnActorStart.bind(this),
       onActorStop: this.#runnerOnActorStop.bind(this),
-      logger: _chunk3MBP4WNCcjs.getLogger.call(void 0, "engine-runner")
+      logger: _chunkTPJNKVFBcjs.getLogger.call(void 0, "engine-runner")
     };
     this.#runner = new (0, _enginerunner.Runner)(runnerConfig);
     this.#runner.start();
@@ -1541,6 +1549,16 @@ var EngineActorDriver = class {
     await this.#runner.kvPut(actorId, [[KEYS.PERSIST_DATA, data]]);
   }
   async setAlarm(actor2, timestamp) {
+    if (this.#alarmTimeout) {
+      this.#alarmTimeout.abort();
+      this.#alarmTimeout = void 0;
+    }
+    const delay = Math.max(0, timestamp - Date.now());
+    this.#alarmTimeout = _chunkDQVVH5ZKcjs.setLongTimeout.call(void 0, () => {
+      actor2._onAlarm();
+      this.#alarmTimeout = void 0;
+    }, delay);
+    this.#runner.setAlarm(actor2.id, timestamp);
   }
   async getDatabase(_actorId) {
     return void 0;
@@ -1562,15 +1580,15 @@ var EngineActorDriver = class {
     let handler = this.#actors.get(actorId);
     if (!handler) {
       handler = {
-        actorStartPromise: _chunk6EUWRXLTcjs.promiseWithResolvers.call(void 0, ),
-        persistedData: _chunkDFS77KAAcjs.serializeEmptyPersistData.call(void 0, input)
+        actorStartPromise: _chunkDQVVH5ZKcjs.promiseWithResolvers.call(void 0, ),
+        persistedData: _chunk3WRAGTDCcjs.serializeEmptyPersistData.call(void 0, input)
       };
       this.#actors.set(actorId, handler);
     }
     const name = config2.name;
     _invariant2.default.call(void 0, config2.key, "actor should have a key");
-    const key = _chunk5ZOHIKWGcjs.deserializeActorKey.call(void 0, config2.key);
-    const definition = _chunk5ZOHIKWGcjs.lookupInRegistry.call(void 0, 
+    const key = _chunkKDNB2BQXcjs.deserializeActorKey.call(void 0, config2.key);
+    const definition = _chunkKDNB2BQXcjs.lookupInRegistry.call(void 0, 
       this.#registryConfig,
       config2.name
       // TODO: Remove cast
@@ -1619,19 +1637,19 @@ var EngineActorDriver = class {
     if (protocols) {
       const protocolList = protocols.split(",").map((p) => p.trim());
       for (const protocol of protocolList) {
-        if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_ENCODING)) {
-          encodingRaw = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_ENCODING.length);
-        } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_PARAMS)) {
+        if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_ENCODING)) {
+          encodingRaw = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_ENCODING.length);
+        } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_PARAMS)) {
           connParamsRaw = decodeURIComponent(
-            protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_PARAMS.length)
+            protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_PARAMS.length)
           );
         }
       }
     }
-    const encoding = _chunkSAZCNSVYcjs.EncodingSchema.parse(encodingRaw);
+    const encoding = _chunkFGOZELKNcjs.EncodingSchema.parse(encodingRaw);
     const connParams = connParamsRaw ? JSON.parse(connParamsRaw) : void 0;
     let wsHandlerPromise;
-    if (url.pathname === _chunkSAZCNSVYcjs.PATH_CONNECT_WEBSOCKET) {
+    if (url.pathname === _chunkFGOZELKNcjs.PATH_CONNECT_WEBSOCKET) {
       wsHandlerPromise = handleWebSocketConnect(
         request,
         this.#runConfig,
@@ -1643,7 +1661,7 @@ var EngineActorDriver = class {
         void 0,
         void 0
       );
-    } else if (url.pathname.startsWith(_chunkSAZCNSVYcjs.PATH_RAW_WEBSOCKET_PREFIX)) {
+    } else if (url.pathname.startsWith(_chunkFGOZELKNcjs.PATH_RAW_WEBSOCKET_PREFIX)) {
       wsHandlerPromise = handleRawWebSocketHandler(
         request,
         url.pathname + url.search,
@@ -1700,9 +1718,9 @@ var EngineActorDriver = class {
   async serverlessHandleStart(c) {
     await this.#runnerStarted.promise;
     return _streaming.streamSSE.call(void 0, c, async (stream) => {
-      const runnerId = this.#runner.runnerId;
-      _invariant2.default.call(void 0, runnerId, "runnerId not set");
-      stream.writeSSE({ data: runnerId });
+      const payload = this.#runner.getServerlessInitPacket();
+      _invariant2.default.call(void 0, payload, "runnerId not set");
+      stream.writeSSE({ data: payload });
       return this.#runnerStopped.promise;
     });
   }
@@ -1713,15 +1731,15 @@ var EngineActorDriver = class {
 var ConfigSchema = _zod.z.object({
   app: _zod.z.custom().optional(),
   endpoint: _zod.z.string().default(
-    () => _nullishCoalesce(_chunk6EUWRXLTcjs.getEnvUniversal.call(void 0, "RIVET_ENGINE"), () => ( "http://localhost:6420"))
+    () => _nullishCoalesce(_chunkDQVVH5ZKcjs.getEnvUniversal.call(void 0, "RIVET_ENGINE"), () => ( "http://localhost:6420"))
   ),
-  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunk6EUWRXLTcjs.getEnvUniversal.call(void 0, "RIVET_TOKEN")))),
+  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkDQVVH5ZKcjs.getEnvUniversal.call(void 0, "RIVET_TOKEN")))),
   pegboardEndpoint: _zod.z.string().optional(),
-  namespace: _zod.z.string().default(() => _nullishCoalesce(_chunk6EUWRXLTcjs.getEnvUniversal.call(void 0, "RIVET_NAMESPACE"), () => ( "default"))),
-  runnerName: _zod.z.string().default(() => _nullishCoalesce(_chunk6EUWRXLTcjs.getEnvUniversal.call(void 0, "RIVET_RUNNER"), () => ( "rivetkit"))),
+  namespace: _zod.z.string().default(() => _nullishCoalesce(_chunkDQVVH5ZKcjs.getEnvUniversal.call(void 0, "RIVET_NAMESPACE"), () => ( "default"))),
+  runnerName: _zod.z.string().default(() => _nullishCoalesce(_chunkDQVVH5ZKcjs.getEnvUniversal.call(void 0, "RIVET_RUNNER"), () => ( "rivetkit"))),
   // TODO: Automatically attempt to determine key by common env vars (e.g. k8s pod name)
   runnerKey: _zod.z.string().default(
-    () => _nullishCoalesce(_chunk6EUWRXLTcjs.getEnvUniversal.call(void 0, "RIVET_RUNNER_KEY"), () => ( crypto.randomUUID()))
+    () => _nullishCoalesce(_chunkDQVVH5ZKcjs.getEnvUniversal.call(void 0, "RIVET_RUNNER_KEY"), () => ( crypto.randomUUID()))
   ),
   totalSlots: _zod.z.number().default(1e5)
 }).default({});
@@ -1732,7 +1750,7 @@ function createEngineDriver(inputConfig) {
   return {
     name: "engine",
     manager: (_registryConfig, runConfig) => {
-      return new (0, _chunk5ZOHIKWGcjs.RemoteManagerDriver)(runConfig);
+      return new (0, _chunkKDNB2BQXcjs.RemoteManagerDriver)(runConfig);
     },
     actor: (registryConfig, runConfig, managerDriver, inlineClient) => {
       return new EngineActorDriver(
@@ -1787,7 +1805,7 @@ var FileSystemActorDriver = class {
     const state = await this.#state.loadActorStateOrError(actorId);
     await this.#state.writeActor(actorId, {
       ...state,
-      persistedData: _chunk6EUWRXLTcjs.bufferToArrayBuffer.call(void 0, data)
+      persistedData: _chunkDQVVH5ZKcjs.bufferToArrayBuffer.call(void 0, data)
     });
   }
   async setAlarm(actor2, timestamp) {
@@ -1890,13 +1908,13 @@ function decodeActorAlarm(bytes) {
 // src/schemas/file-system-driver/versioned.ts
 var CURRENT_VERSION = 1;
 var migrations = /* @__PURE__ */ new Map();
-var ACTOR_STATE_VERSIONED = _chunkDFS77KAAcjs.createVersionedDataHandler.call(void 0, {
+var ACTOR_STATE_VERSIONED = _chunk3WRAGTDCcjs.createVersionedDataHandler.call(void 0, {
   currentVersion: CURRENT_VERSION,
   migrations,
   serializeVersion: (data) => encodeActorState(data),
   deserializeVersion: (bytes) => decodeActorState(bytes)
 });
-var ACTOR_ALARM_VERSIONED = _chunkDFS77KAAcjs.createVersionedDataHandler.call(void 0, {
+var ACTOR_ALARM_VERSIONED = _chunk3WRAGTDCcjs.createVersionedDataHandler.call(void 0, {
   currentVersion: CURRENT_VERSION,
   migrations,
   serializeVersion: (data) => encodeActorAlarm(data),
@@ -1905,7 +1923,7 @@ var ACTOR_ALARM_VERSIONED = _chunkDFS77KAAcjs.createVersionedDataHandler.call(vo
 
 // src/drivers/file-system/log.ts
 function logger5() {
-  return _chunk3MBP4WNCcjs.getLogger.call(void 0, "driver-fs");
+  return _chunkTPJNKVFBcjs.getLogger.call(void 0, "driver-fs");
 }
 
 // src/drivers/file-system/utils.ts
@@ -2075,7 +2093,7 @@ var FileSystemGlobalState = class {
       name,
       key,
       createdAt: BigInt(Date.now()),
-      persistedData: _chunk6EUWRXLTcjs.bufferToArrayBuffer.call(void 0, _chunkDFS77KAAcjs.serializeEmptyPersistData.call(void 0, input))
+      persistedData: _chunkDQVVH5ZKcjs.bufferToArrayBuffer.call(void 0, _chunk3WRAGTDCcjs.serializeEmptyPersistData.call(void 0, input))
     };
     await this.writeActor(actorId, entry.state);
     return entry;
@@ -2123,7 +2141,7 @@ var FileSystemGlobalState = class {
         name,
         key,
         createdAt: BigInt(Date.now()),
-        persistedData: _chunk6EUWRXLTcjs.bufferToArrayBuffer.call(void 0, _chunkDFS77KAAcjs.serializeEmptyPersistData.call(void 0, input))
+        persistedData: _chunkDQVVH5ZKcjs.bufferToArrayBuffer.call(void 0, _chunk3WRAGTDCcjs.serializeEmptyPersistData.call(void 0, input))
       };
       await this.writeActor(actorId, entry.state);
     }
@@ -2247,9 +2265,9 @@ var FileSystemGlobalState = class {
     if (entry.actor) {
       return entry.actor;
     }
-    entry.startPromise = _chunk6EUWRXLTcjs.promiseWithResolvers.call(void 0, );
+    entry.startPromise = _chunkDQVVH5ZKcjs.promiseWithResolvers.call(void 0, );
     try {
-      const definition = _chunk5ZOHIKWGcjs.lookupInRegistry.call(void 0, registryConfig, entry.state.name);
+      const definition = _chunkKDNB2BQXcjs.lookupInRegistry.call(void 0, registryConfig, entry.state.name);
       entry.actor = definition.instantiate();
       await entry.actor.start(
         actorDriver,
@@ -2309,7 +2327,7 @@ var FileSystemGlobalState = class {
           logger5().error({
             msg: "failed to read alarm file",
             file,
-            error: _chunk6EUWRXLTcjs.stringifyError.call(void 0, err)
+            error: _chunkDQVVH5ZKcjs.stringifyError.call(void 0, err)
           });
         }
       }
@@ -2336,7 +2354,7 @@ var FileSystemGlobalState = class {
     (_a = entry.alarmTimeout) == null ? void 0 : _a.abort();
     entry.alarmTimestamp = timestamp;
     const delay = Math.max(0, timestamp - Date.now());
-    entry.alarmTimeout = _chunk6EUWRXLTcjs.setLongTimeout.call(void 0, async () => {
+    entry.alarmTimeout = _chunkDQVVH5ZKcjs.setLongTimeout.call(void 0, async () => {
       entry.alarmTimestamp = void 0;
       if (this.#persist) {
         try {
@@ -2346,7 +2364,7 @@ var FileSystemGlobalState = class {
             logger5().debug({
               msg: "failed to remove alarm file",
               actorId,
-              error: _chunk6EUWRXLTcjs.stringifyError.call(void 0, err)
+              error: _chunkDQVVH5ZKcjs.stringifyError.call(void 0, err)
             });
           }
         }
@@ -2372,7 +2390,7 @@ var FileSystemGlobalState = class {
         logger5().error({
           msg: "failed to handle alarm",
           actorId,
-          error: _chunk6EUWRXLTcjs.stringifyError.call(void 0, err)
+          error: _chunkDQVVH5ZKcjs.stringifyError.call(void 0, err)
         });
       }
     }, delay);
@@ -2382,7 +2400,7 @@ var FileSystemGlobalState = class {
     if (fsSync2.existsSync(tokenPath)) {
       return fsSync2.readFileSync(tokenPath, "utf-8");
     }
-    const newToken = _chunkSAZCNSVYcjs.generateRandomString.call(void 0, );
+    const newToken = _chunkFGOZELKNcjs.generateRandomString.call(void 0, );
     fsSync2.writeFileSync(tokenPath, newToken);
     return newToken;
   }
@@ -2444,10 +2462,10 @@ function createManagerInspectorRouter() {
       });
       return c.json(actors, 200);
     } catch (error) {
-      _chunkG4ABMAQYcjs.inspectorLogger.call(void 0, ).error({ msg: "Failed to fetch actors", error });
+      _chunkQGUQB3NCcjs.inspectorLogger.call(void 0, ).error({ msg: "Failed to fetch actors", error });
       return c.json("Failed to fetch actors", 500);
     }
-  }).post("/actors", _standardvalidator.sValidator.call(void 0, "json", _chunkB3TLRM4Qcjs.CreateActorSchema), async (c) => {
+  }).post("/actors", _standardvalidator.sValidator.call(void 0, "json", _chunkL5MHM6JJcjs.CreateActorSchema), async (c) => {
     const actor2 = await c.var.inspector.accessors.createActor(
       c.req.valid("json")
     );
@@ -2473,7 +2491,7 @@ var ManagerInspector = class {
   
   constructor(accessors) {
     this.accessors = accessors();
-    _chunkG4ABMAQYcjs.inspectorLogger.call(void 0, ).debug({ msg: "Manager Inspector enabled and ready" });
+    _chunkQGUQB3NCcjs.inspectorLogger.call(void 0, ).debug({ msg: "Manager Inspector enabled and ready" });
   }
 };
 
@@ -2548,14 +2566,18 @@ var FileSystemManagerDriver = class {
         };
       });
     }
-    const inlineClient = _chunk5ZOHIKWGcjs.createClientWithDriver.call(void 0, this);
+    const inlineClient = _chunkKDNB2BQXcjs.createClientWithDriver.call(void 0, this);
     this.#actorDriver = this.#driverConfig.actor(
       registryConfig,
       runConfig,
       this,
       inlineClient
     );
-    this.#actorRouter = createActorRouter(this.#runConfig, this.#actorDriver);
+    this.#actorRouter = createActorRouter(
+      this.#runConfig,
+      this.#actorDriver,
+      registryConfig.test.enabled
+    );
   }
   async sendRequest(actorId, actorRequest) {
     return await this.#actorRouter.fetch(actorRequest, {
@@ -2565,7 +2587,7 @@ var FileSystemManagerDriver = class {
   async openWebSocket(path3, actorId, encoding, params, connId, connToken) {
     const pathOnly = path3.split("?")[0];
     const normalizedPath = pathOnly.startsWith("/") ? pathOnly : `/${pathOnly}`;
-    if (normalizedPath === _chunkSAZCNSVYcjs.PATH_CONNECT_WEBSOCKET) {
+    if (normalizedPath === _chunkFGOZELKNcjs.PATH_CONNECT_WEBSOCKET) {
       const wsHandler = await handleWebSocketConnect(
         void 0,
         this.#runConfig,
@@ -2577,7 +2599,7 @@ var FileSystemManagerDriver = class {
         connToken
       );
       return new InlineWebSocketAdapter2(wsHandler);
-    } else if (normalizedPath.startsWith(_chunkSAZCNSVYcjs.PATH_RAW_WEBSOCKET_PREFIX) || normalizedPath === "/raw/websocket") {
+    } else if (normalizedPath.startsWith(_chunkFGOZELKNcjs.PATH_RAW_WEBSOCKET_PREFIX) || normalizedPath === "/raw/websocket") {
       const wsHandler = await handleRawWebSocketHandler(
         void 0,
         path3,
@@ -2600,7 +2622,7 @@ var FileSystemManagerDriver = class {
     _invariant2.default.call(void 0, upgradeWebSocket, "missing getUpgradeWebSocket");
     const pathOnly = path3.split("?")[0];
     const normalizedPath = pathOnly.startsWith("/") ? pathOnly : `/${pathOnly}`;
-    if (normalizedPath === _chunkSAZCNSVYcjs.PATH_CONNECT_WEBSOCKET) {
+    if (normalizedPath === _chunkFGOZELKNcjs.PATH_CONNECT_WEBSOCKET) {
       const wsHandler = await handleWebSocketConnect(
         c.req.raw,
         this.#runConfig,
@@ -2611,15 +2633,15 @@ var FileSystemManagerDriver = class {
         connId,
         connToken
       );
-      return upgradeWebSocket(() => wsHandler)(c, _chunk6EUWRXLTcjs.noopNext.call(void 0, ));
-    } else if (normalizedPath.startsWith(_chunkSAZCNSVYcjs.PATH_RAW_WEBSOCKET_PREFIX) || normalizedPath === "/raw/websocket") {
+      return upgradeWebSocket(() => wsHandler)(c, _chunkDQVVH5ZKcjs.noopNext.call(void 0, ));
+    } else if (normalizedPath.startsWith(_chunkFGOZELKNcjs.PATH_RAW_WEBSOCKET_PREFIX) || normalizedPath === "/raw/websocket") {
       const wsHandler = await handleRawWebSocketHandler(
         c.req.raw,
         path3,
         this.#actorDriver,
         actorId
       );
-      return upgradeWebSocket(() => wsHandler)(c, _chunk6EUWRXLTcjs.noopNext.call(void 0, ));
+      return upgradeWebSocket(() => wsHandler)(c, _chunkDQVVH5ZKcjs.noopNext.call(void 0, ));
     } else {
       throw new Error(`Unreachable path: ${path3}`);
     }
@@ -2742,7 +2764,7 @@ function chooseDefaultDriver(runConfig) {
     return runConfig.driver;
   }
   if (runConfig.endpoint) {
-    _chunkSAZCNSVYcjs.loggerWithoutContext.call(void 0, ).debug({
+    _chunkFGOZELKNcjs.loggerWithoutContext.call(void 0, ).debug({
       msg: "using rivet engine driver",
       endpoint: runConfig.endpoint
     });
@@ -2751,7 +2773,7 @@ function chooseDefaultDriver(runConfig) {
       token: runConfig.token
     });
   }
-  _chunkSAZCNSVYcjs.loggerWithoutContext.call(void 0, ).debug({ msg: "using default file system driver" });
+  _chunkFGOZELKNcjs.loggerWithoutContext.call(void 0, ).debug({ msg: "using default file system driver" });
   return createFileSystemOrMemoryDriver(true);
 }
 
@@ -2838,20 +2860,20 @@ async function handleWebSocketGateway(runConfig, managerDriver, c) {
   if (protocols) {
     const protocolList = protocols.split(",").map((p) => p.trim());
     for (const protocol of protocolList) {
-      if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_TARGET)) {
-        target = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_TARGET.length);
-      } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_ACTOR)) {
-        actorId = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_ACTOR.length);
-      } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_ENCODING)) {
-        encodingRaw = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_ENCODING.length);
-      } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_PARAMS)) {
+      if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_TARGET)) {
+        target = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_TARGET.length);
+      } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_ACTOR)) {
+        actorId = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_ACTOR.length);
+      } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_ENCODING)) {
+        encodingRaw = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_ENCODING.length);
+      } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_PARAMS)) {
         connParamsRaw = decodeURIComponent(
-          protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_PARAMS.length)
+          protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_PARAMS.length)
         );
-      } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_ID)) {
-        connIdRaw = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_ID.length);
-      } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_TOKEN)) {
-        connTokenRaw = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_TOKEN.length);
+      } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_ID)) {
+        connIdRaw = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_ID.length);
+      } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_TOKEN)) {
+        connTokenRaw = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_TOKEN.length);
       }
     }
   }
@@ -2882,8 +2904,8 @@ async function handleWebSocketGateway(runConfig, managerDriver, c) {
   );
 }
 async function handleHttpGateway(managerDriver, c, next) {
-  const target = c.req.header(_chunkSAZCNSVYcjs.HEADER_RIVET_TARGET);
-  const actorId = c.req.header(_chunkSAZCNSVYcjs.HEADER_RIVET_ACTOR);
+  const target = c.req.header(_chunkFGOZELKNcjs.HEADER_RIVET_TARGET);
+  const actorId = c.req.header(_chunkFGOZELKNcjs.HEADER_RIVET_ACTOR);
   if (target !== "actor") {
     return next();
   }
@@ -2897,8 +2919,8 @@ async function handleHttpGateway(managerDriver, c, next) {
     method: c.req.method
   });
   const proxyHeaders = new Headers(c.req.raw.headers);
-  proxyHeaders.delete(_chunkSAZCNSVYcjs.HEADER_RIVET_TARGET);
-  proxyHeaders.delete(_chunkSAZCNSVYcjs.HEADER_RIVET_ACTOR);
+  proxyHeaders.delete(_chunkFGOZELKNcjs.HEADER_RIVET_TARGET);
+  proxyHeaders.delete(_chunkFGOZELKNcjs.HEADER_RIVET_ACTOR);
   const url = new URL(c.req.url);
   const proxyUrl = new URL(`http://actor${url.pathname}${url.search}`);
   const proxyRequest = new Request(proxyUrl, {
@@ -2915,7 +2937,7 @@ async function createTestWebSocketProxy(clientWsPromise) {
     promise: serverWsPromise,
     resolve: serverWsResolve,
     reject: serverWsReject
-  } = _chunk6EUWRXLTcjs.promiseWithResolvers.call(void 0, );
+  } = _chunkDQVVH5ZKcjs.promiseWithResolvers.call(void 0, );
   try {
     logger().debug({ msg: "awaiting client websocket promise" });
     const ws = await clientWsPromise;
@@ -3122,7 +3144,12 @@ function addServerlessRoutes(serverlessActorDriverBuilder, router, cors2) {
     );
   });
   router.get("/start", cors2, async (c) => {
-    const actorDriver = serverlessActorDriverBuilder();
+    const token = c.req.header("x-rivet-token");
+    let totalSlots = parseInt(
+      c.req.header("x-rivetkit-total-slots")
+    );
+    if (isNaN(totalSlots)) totalSlots = void 0;
+    const actorDriver = serverlessActorDriverBuilder(token, totalSlots);
     _invariant2.default.call(void 0, 
       actorDriver.serverlessHandleStart,
       "missing serverlessHandleStart on ActorDriver"
@@ -3293,7 +3320,7 @@ function addManagerRoutes(registryConfig, runConfig, managerDriver, router, cors
         const output = await managerDriver[method](...args);
         response = { ok: output };
       } catch (rawErr) {
-        const err = _chunk6EUWRXLTcjs.deconstructError.call(void 0, rawErr, logger(), {}, true);
+        const err = _chunkDQVVH5ZKcjs.deconstructError.call(void 0, rawErr, logger(), {}, true);
         response = { err };
       }
       return c.body(cbor4.encode(response));
@@ -3313,29 +3340,29 @@ function addManagerRoutes(registryConfig, runConfig, managerDriver, router, cors
         let connId;
         let connToken;
         for (const protocol of protocols) {
-          if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_ACTOR)) {
-            actorId = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_ACTOR.length);
-          } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_ENCODING)) {
+          if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_ACTOR)) {
+            actorId = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_ACTOR.length);
+          } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_ENCODING)) {
             encoding = protocol.substring(
-              _chunkSAZCNSVYcjs.WS_PROTOCOL_ENCODING.length
+              _chunkFGOZELKNcjs.WS_PROTOCOL_ENCODING.length
             );
-          } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_TRANSPORT)) {
+          } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_TRANSPORT)) {
             transport = protocol.substring(
-              _chunkSAZCNSVYcjs.WS_PROTOCOL_TRANSPORT.length
+              _chunkFGOZELKNcjs.WS_PROTOCOL_TRANSPORT.length
             );
-          } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_PATH)) {
+          } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_PATH)) {
             path3 = decodeURIComponent(
-              protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_PATH.length)
+              protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_PATH.length)
             );
-          } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_PARAMS)) {
+          } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_PARAMS)) {
             const paramsRaw = decodeURIComponent(
-              protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_PARAMS.length)
+              protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_PARAMS.length)
             );
             params = JSON.parse(paramsRaw);
-          } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_ID)) {
-            connId = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_ID.length);
-          } else if (protocol.startsWith(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_TOKEN)) {
-            connToken = protocol.substring(_chunkSAZCNSVYcjs.WS_PROTOCOL_CONN_TOKEN.length);
+          } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_ID)) {
+            connId = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_ID.length);
+          } else if (protocol.startsWith(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_TOKEN)) {
+            connToken = protocol.substring(_chunkFGOZELKNcjs.WS_PROTOCOL_CONN_TOKEN.length);
           }
         }
         logger().debug({
@@ -3355,10 +3382,10 @@ function addManagerRoutes(registryConfig, runConfig, managerDriver, router, cors
           connToken
         );
         return await createTestWebSocketProxy(clientWsPromise);
-      })(c, _chunk6EUWRXLTcjs.noopNext.call(void 0, ));
+      })(c, _chunkDQVVH5ZKcjs.noopNext.call(void 0, ));
     });
     router.all(".test/inline-driver/send-request/*", async (c) => {
-      const actorId = c.req.header(_chunkSAZCNSVYcjs.HEADER_ACTOR_ID);
+      const actorId = c.req.header(_chunkFGOZELKNcjs.HEADER_ACTOR_ID);
       if (!actorId) {
         return c.text("Missing required headers", 400);
       }
@@ -3385,9 +3412,9 @@ function addManagerRoutes(registryConfig, runConfig, managerDriver, router, cors
       } catch (error) {
         logger().error({
           msg: "error in test inline raw http",
-          error: _chunk6EUWRXLTcjs.stringifyError.call(void 0, error)
+          error: _chunkDQVVH5ZKcjs.stringifyError.call(void 0, error)
         });
-        const err = _chunk6EUWRXLTcjs.deconstructError.call(void 0, error, logger(), {}, true);
+        const err = _chunkDQVVH5ZKcjs.deconstructError.call(void 0, error, logger(), {}, true);
         return c.json(
           {
             error: {
@@ -3429,7 +3456,7 @@ function addManagerRoutes(registryConfig, runConfig, managerDriver, router, cors
       } catch (error) {
         logger().error({
           msg: "error forcing disconnect",
-          error: _chunk6EUWRXLTcjs.stringifyError.call(void 0, error)
+          error: _chunkDQVVH5ZKcjs.stringifyError.call(void 0, error)
         });
         return c.text(`Error: ${error}`, 500);
       }
@@ -3443,13 +3470,13 @@ function addManagerRoutes(registryConfig, runConfig, managerDriver, router, cors
     registryConfig,
     router
   );
-  if (_chunkG4ABMAQYcjs.isInspectorEnabled.call(void 0, runConfig, "manager")) {
+  if (_chunkQGUQB3NCcjs.isInspectorEnabled.call(void 0, runConfig, "manager")) {
     if (!managerDriver.inspector) {
       throw new (0, _chunk5QGQK44Lcjs.Unsupported)("inspector");
     }
     router.route(
       "/inspect",
-      new (0, _hono.Hono)().use(_cors.cors.call(void 0, runConfig.inspector.cors)).use(_chunkG4ABMAQYcjs.secureInspector.call(void 0, runConfig)).use((c, next) => {
+      new (0, _hono.Hono)().use(_cors.cors.call(void 0, runConfig.inspector.cors)).use(_chunkQGUQB3NCcjs.secureInspector.call(void 0, runConfig)).use((c, next) => {
         c.set("inspector", managerDriver.inspector);
         return next();
       }).route("/", createManagerInspectorRouter())
@@ -3460,7 +3487,7 @@ function createApiActor(actor2) {
   return {
     actor_id: actor2.actorId,
     name: actor2.name,
-    key: _chunk5ZOHIKWGcjs.serializeActorKey.call(void 0, actor2.key),
+    key: _chunkKDNB2BQXcjs.serializeActorKey.call(void 0, actor2.key),
     namespace_id: "default",
     // Assert default namespace
     runner_name_selector: "rivetkit",
@@ -3494,7 +3521,7 @@ var RegistryConfigSchema = _zod.z.object({
 
 // src/registry/log.ts
 function logger6() {
-  return _chunk3MBP4WNCcjs.getLogger.call(void 0, "registry");
+  return _chunkTPJNKVFBcjs.getLogger.call(void 0, "registry");
 }
 
 // src/registry/serve.ts
@@ -3554,11 +3581,11 @@ var Registry = class {
    */
   start(inputConfig) {
     var _a, _b, _c;
-    const config2 = _chunkDFS77KAAcjs.RunConfigSchema.parse(inputConfig);
+    const config2 = _chunk3WRAGTDCcjs.RunConfigSchema.parse(inputConfig);
     if ((_a = config2.logging) == null ? void 0 : _a.baseLogger) {
-      _chunk3MBP4WNCcjs.configureBaseLogger.call(void 0, config2.logging.baseLogger);
+      _chunkTPJNKVFBcjs.configureBaseLogger.call(void 0, config2.logging.baseLogger);
     } else {
-      _chunk3MBP4WNCcjs.configureDefaultLogger.call(void 0, (_b = config2.logging) == null ? void 0 : _b.level);
+      _chunkTPJNKVFBcjs.configureDefaultLogger.call(void 0, (_b = config2.logging) == null ? void 0 : _b.level);
     }
     const driver = chooseDefaultDriver(config2);
     if (driver.name === "engine") {
@@ -3576,8 +3603,8 @@ var Registry = class {
       config2.getUpgradeWebSocket = () => upgradeWebSocket;
     }
     const managerDriver = driver.manager(this.#config, config2);
-    _chunkG4ABMAQYcjs.configureInspectorAccessToken.call(void 0, config2, managerDriver);
-    const client = _chunk5ZOHIKWGcjs.createClientWithDriver.call(void 0, managerDriver, config2);
+    _chunkQGUQB3NCcjs.configureInspectorAccessToken.call(void 0, config2, managerDriver);
+    const client = _chunkKDNB2BQXcjs.createClientWithDriver.call(void 0, managerDriver, config2);
     const driverLog = _nullishCoalesce(((_c = managerDriver.extraStartupLog) == null ? void 0 : _c.call(managerDriver)), () => ( {}));
     logger6().info({
       msg: "rivetkit ready",
@@ -3585,20 +3612,20 @@ var Registry = class {
       definitions: Object.keys(this.#config.use).length,
       ...driverLog
     });
-    if (_chunkG4ABMAQYcjs.isInspectorEnabled.call(void 0, config2, "manager") && managerDriver.inspector) {
-      logger6().info({ msg: "inspector ready", url: _chunkG4ABMAQYcjs.getInspectorUrl.call(void 0, config2) });
+    if (_chunkQGUQB3NCcjs.isInspectorEnabled.call(void 0, config2, "manager") && managerDriver.inspector) {
+      logger6().info({ msg: "inspector ready", url: _chunkQGUQB3NCcjs.getInspectorUrl.call(void 0, config2) });
     }
     if (!config2.noWelcome) {
       const displayInfo = managerDriver.displayInformation();
       console.log();
-      console.log(`  RivetKit ${_chunk6EUWRXLTcjs.package_default.version} (${displayInfo.name})`);
+      console.log(`  RivetKit ${_chunkDQVVH5ZKcjs.package_default.version} (${displayInfo.name})`);
       console.log(`  - Endpoint:     http://127.0.0.1:6420`);
       for (const [k, v] of Object.entries(displayInfo.properties)) {
         const padding = " ".repeat(Math.max(0, 13 - k.length));
         console.log(`  - ${k}:${padding}${v}`);
       }
-      if (_chunkG4ABMAQYcjs.isInspectorEnabled.call(void 0, config2, "manager") && managerDriver.inspector) {
-        console.log(`  - Inspector:     ${_chunkG4ABMAQYcjs.getInspectorUrl.call(void 0, config2)}`);
+      if (_chunkQGUQB3NCcjs.isInspectorEnabled.call(void 0, config2, "manager") && managerDriver.inspector) {
+        console.log(`  - Inspector:     ${_chunkQGUQB3NCcjs.getInspectorUrl.call(void 0, config2)}`);
       }
       console.log();
     }
@@ -3629,11 +3656,11 @@ var Registry = class {
   }
   startServerless(inputConfig) {
     var _a, _b, _c, _d, _e;
-    const config2 = _chunkDFS77KAAcjs.RunConfigSchema.parse(inputConfig);
+    const config2 = _chunk3WRAGTDCcjs.RunConfigSchema.parse(inputConfig);
     if ((_a = config2.logging) == null ? void 0 : _a.baseLogger) {
-      _chunk3MBP4WNCcjs.configureBaseLogger.call(void 0, config2.logging.baseLogger);
+      _chunkTPJNKVFBcjs.configureBaseLogger.call(void 0, config2.logging.baseLogger);
     } else {
-      _chunk3MBP4WNCcjs.configureDefaultLogger.call(void 0, (_b = config2.logging) == null ? void 0 : _b.level);
+      _chunkTPJNKVFBcjs.configureDefaultLogger.call(void 0, (_b = config2.logging) == null ? void 0 : _b.level);
     }
     const driver = chooseDefaultDriver(config2);
     if (driver.name === "engine") {
@@ -3652,7 +3679,7 @@ var Registry = class {
       config2.getUpgradeWebSocket = () => upgradeWebSocket;
     }
     const managerDriver = driver.manager(this.#config, config2);
-    const client = _chunk5ZOHIKWGcjs.createClientWithDriver.call(void 0, managerDriver, config2);
+    const client = _chunkKDNB2BQXcjs.createClientWithDriver.call(void 0, managerDriver, config2);
     const driverLog = _nullishCoalesce(((_c = managerDriver.extraStartupLog) == null ? void 0 : _c.call(managerDriver)), () => ( {}));
     logger6().info({
       msg: "rivetkit ready",
@@ -3661,23 +3688,25 @@ var Registry = class {
       ...driverLog
     });
     if (((_d = config2.inspector) == null ? void 0 : _d.enabled) && managerDriver.inspector) {
-      logger6().info({ msg: "inspector ready", url: _chunkG4ABMAQYcjs.getInspectorUrl.call(void 0, config2) });
+      logger6().info({ msg: "inspector ready", url: _chunkQGUQB3NCcjs.getInspectorUrl.call(void 0, config2) });
     }
     if (!config2.noWelcome) {
       const displayInfo = managerDriver.displayInformation();
       console.log();
-      console.log(`  RivetKit ${_chunk6EUWRXLTcjs.package_default.version} (${displayInfo.name})`);
+      console.log(`  RivetKit ${_chunkDQVVH5ZKcjs.package_default.version} (${displayInfo.name})`);
       console.log(`  - Endpoint:     http://127.0.0.1:6420`);
       for (const [k, v] of Object.entries(displayInfo.properties)) {
         const padding = " ".repeat(Math.max(0, 13 - k.length));
         console.log(`  - ${k}:${padding}${v}`);
       }
       if (((_e = config2.inspector) == null ? void 0 : _e.enabled) && managerDriver.inspector) {
-        console.log(`  - Inspector:    ${_chunkG4ABMAQYcjs.getInspectorUrl.call(void 0, config2)}`);
+        console.log(`  - Inspector:    ${_chunkQGUQB3NCcjs.getInspectorUrl.call(void 0, config2)}`);
       }
       console.log();
     }
-    let serverlessActorDriverBuilder = () => {
+    let serverlessActorDriverBuilder = (token, totalSlots) => {
+      if (token) config2.token = token;
+      if (totalSlots) config2.totalSlots = totalSlots;
       return driver.actor(this.#config, config2, managerDriver, client);
     };
     if (!config2.disableActorDriver) {
@@ -3723,4 +3752,4 @@ function setup(input) {
 
 exports.handleWebSocketConnect = handleWebSocketConnect; exports.handleRawWebSocketHandler = handleRawWebSocketHandler; exports.createActorRouter = createActorRouter; exports.actor = actor; exports.InlineWebSocketAdapter2 = InlineWebSocketAdapter2; exports.createEngineDriver = createEngineDriver; exports.createFileSystemOrMemoryDriver = createFileSystemOrMemoryDriver; exports.createFileSystemDriver = createFileSystemDriver; exports.createMemoryDriver = createMemoryDriver; exports.createManagerRouter = createManagerRouter; exports.RegistryConfigSchema = RegistryConfigSchema; exports.Registry = Registry; exports.setup = setup;
 //! These configs configs hold anything that's not platform-specific about running actors.
-//# sourceMappingURL=chunk-3Y45CIF4.cjs.map
+//# sourceMappingURL=chunk-CKSA7NOS.cjs.map