rivetkit 0.0.0-codex-docsrs-rivetkit-build.b91ad8d → 0.0.0-codex-rivetkit-quickstart-preview.d4284e1

package/dist/tsup/mod.js

@@ -1,6 +1,6 @@
 import {
   decodeWorkflowHistoryTransport
-} from "./chunk-2GYNUOP4.js";
+} from "./chunk-Q2Z22VKI.js";
 import {
   ACTOR_CONTEXT_INTERNAL_SYMBOL,
   ActorConfigSchema,
@@ -10,7 +10,7 @@ import {
   getRunFunction,
   getRunInspectorConfig,
   getRunMetadata
-} from "./chunk-XIX5DOZN.js";
+} from "./chunk-RQFXU6US.js";
 import {
   ALLOWED_PUBLIC_HEADERS,
   HEADER_CONN_PARAMS,
@@ -20,7 +20,7 @@ import {
   getDatacenters,
   tryParseEndpoint,
   updateRunnerConfig
-} from "./chunk-VQ4EAFUD.js";
+} from "./chunk-67RJEZ3M.js";
 import {
   KEYS,
   makePrefixedKey,
@@ -28,17 +28,17 @@ import {
   queueMetadataKey,
   removePrefixFromKey,
   workflowStoragePrefix
-} from "./chunk-HXUEHHJF.js";
+} from "./chunk-VXO53AYC.js";
 import {
   assertJsonCompatValue,
   decodeCborCompat,
   encodeCborCompat
-} from "./chunk-TZCFZNNB.js";
-import "./chunk-JZ7TWV65.js";
-import "./chunk-7FGVV7DV.js";
+} from "./chunk-4HZZFRAX.js";
+import "./chunk-2BWA26HM.js";
 import {
   LogLevelSchema,
   VERSION,
+  configureBaseLogger,
   deconstructError,
   getEnvUniversal,
   getLogger,
@@ -64,7 +64,7 @@ import {
   parsePortEnv,
   stringifyError,
   toUint8Array
-} from "./chunk-YRZLSAQP.js";
+} from "./chunk-X7EOSG3F.js";
 import {
   INTERNAL_ERROR_CODE,
   RivetError,
@@ -76,7 +76,8 @@ import {
   isRivetErrorLike,
   toRivetError,
   unsupportedFeature
-} from "./chunk-ZZ3WBRPD.js";
+} from "./chunk-FLWADOST.js";
+import "./chunk-MYP3D75W.js";
 
 // src/actor/log.ts
 function loggerWithoutContext() {
@@ -389,6 +390,9 @@ function logger2() {
 // src/serverless/configure.ts
 var DEFAULT_CONFIGURE_TIMEOUT_MS = 6e4;
 var CONFIGURE_RETRY_DELAY_MS = 1e3;
+var DEFAULT_REQUEST_LIFESPAN_SECONDS = 60 * 60;
+var DEFAULT_DRAIN_GRACE_PERIOD_SECONDS = 30 * 60;
+var SHORT_REQUEST_DRAIN_GRACE_PERIOD_SECONDS = 30;
 function sleep(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
@@ -404,6 +408,16 @@ function configureTimeoutMs() {
   }
   return parsed;
 }
+function resolveServerlessDrainGracePeriod(requestLifespan, drainGracePeriod) {
+  if (drainGracePeriod !== void 0) return drainGracePeriod;
+  if (requestLifespan > DEFAULT_DRAIN_GRACE_PERIOD_SECONDS) {
+    return DEFAULT_DRAIN_GRACE_PERIOD_SECONDS;
+  }
+  return Math.max(
+    0,
+    Math.min(SHORT_REQUEST_DRAIN_GRACE_PERIOD_SECONDS, requestLifespan - 1)
+  );
+}
 async function configureServerlessPool(config) {
   logger2().debug({ msg: "configuring serverless pool" });
   const startedAt = Date.now();
@@ -437,12 +451,16 @@ async function configureServerlessPool(config) {
         ...serverlessToken ? { "x-rivet-token": serverlessToken } : {},
         ...customConfig.headers ?? {}
       };
+      const requestLifespan = customConfig.requestLifespan ?? DEFAULT_REQUEST_LIFESPAN_SECONDS;
       const serverlessConfig = {
         serverless: {
           url: customConfig.url,
           headers,
-          request_lifespan: customConfig.requestLifespan ?? 60 * 60,
-          drain_grace_period: customConfig.drainGracePeriod,
+          request_lifespan: requestLifespan,
+          drain_grace_period: resolveServerlessDrainGracePeriod(
+            requestLifespan,
+            customConfig.drainGracePeriod
+          ),
           metadata_poll_interval: customConfig.metadataPollInterval ?? 1e3,
           max_runners: 1e5,
           min_runners: 0,
@@ -1734,7 +1752,7 @@ var NapiCoreRuntime = class {
   }
 };
 async function loadNapiRuntime() {
-  const bindings = await import(["@rivetkit", "rivetkit-napi"].join("/"));
+  const bindings = await import("@rivetkit/rivetkit-napi");
   return {
     bindings,
     runtime: new NapiCoreRuntime(bindings)
@@ -2396,7 +2414,7 @@ var WasmCoreRuntime = class {
   }
 };
 async function loadWasmRuntime(config) {
-  const bindings = (config == null ? void 0 : config.bindings) ?? await import(["@rivetkit", "rivetkit-wasm"].join("/"));
+  const bindings = (config == null ? void 0 : config.bindings) ?? await import("@rivetkit/rivetkit-wasm");
   await bindings.default(config == null ? void 0 : config.initInput);
   return {
     bindings,
@@ -2754,7 +2772,7 @@ function decodeNativeKvValue(value, options) {
   }
 }
 async function loadEngineCli() {
-  return import(["@rivetkit", "engine-cli"].join("/"));
+  return import("@rivetkit/engine-cli");
 }
 function decodeValue(value) {
   if (!value || value.length === 0) {
@@ -5821,7 +5839,11 @@ var Registry = class {
   #shutdownInFlight = null;
   #signalHandlers = {};
   constructor(config, deps) {
+    var _a;
     this.#config = config;
+    if ((_a = config.logging) == null ? void 0 : _a.baseLogger) {
+      configureBaseLogger(config.logging.baseLogger);
+    }
     this.#buildConfiguredRegistry = (deps == null ? void 0 : deps.buildConfiguredRegistry) ?? buildConfiguredRegistry;
     this.routes = {
       health: () => this.#healthRoute(),