rivetkit 2.3.7 → 2.3.9

package/dist/tsup/test/mod.cjs

@@ -1,13 +1,13 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
-var _chunkDOPH7DT2cjs = require('../chunk-DOPH7DT2.cjs');
+var _chunkAVUMOGUEcjs = require('../chunk-AVUMOGUE.cjs');
 
 
-var _chunk46AE7JOIcjs = require('../chunk-46AE7JOI.cjs');
+var _chunk3IDKUXQTcjs = require('../chunk-3IDKUXQT.cjs');
 require('../chunk-MYWZCDML.cjs');
-require('../chunk-MNZFNIVI.cjs');
 require('../chunk-UEULFXUI.cjs');
-require('../chunk-2SGPZEAJ.cjs');
+require('../chunk-S57DMBTM.cjs');
+require('../chunk-M4R2O5NJ.cjs');
 require('../chunk-4FC7TVS6.cjs');
 
 // src/test/ready.ts
@@ -36,8 +36,8 @@ async function setupTest(c, registry) {
   registry.config.noWelcome = true;
   registry.start();
   await waitForRegistryReady(registry);
-  const client = _chunkDOPH7DT2cjs.createClient.call(void 0, {
-    ..._chunk46AE7JOIcjs.convertRegistryConfigToClientConfig.call(void 0, registry.parseConfig()),
+  const client = _chunkAVUMOGUEcjs.createClient.call(void 0, {
+    ..._chunk3IDKUXQTcjs.convertRegistryConfigToClientConfig.call(void 0, registry.parseConfig()),
     disableMetadataLookup: false
   });
   c.onTestFinished(async () => {

package/dist/tsup/test/mod.d.cts

@@ -1,5 +1,5 @@
 import { TestContext } from 'vitest';
-import { aC as Registry, a2 as Client } from '../config-BobvQpex.cjs';
+import { aC as Registry, a2 as Client } from '../config-CpB-e4I9.cjs';
 import 'zod/v4';
 import '../config-BxOimoGX.cjs';
 import '@rivetkit/rivetkit-wasm';

package/dist/tsup/test/mod.d.ts

@@ -1,5 +1,5 @@
 import { TestContext } from 'vitest';
-import { aC as Registry, a2 as Client } from '../config-BRpVbJCD.js';
+import { aC as Registry, a2 as Client } from '../config-CUymbyM6.js';
 import 'zod/v4';
 import '../config-BxOimoGX.js';
 import '@rivetkit/rivetkit-wasm';

package/dist/tsup/test/mod.js

@@ -1,13 +1,13 @@
 import {
   createClient
-} from "../chunk-CH25U5NO.js";
+} from "../chunk-KVKWTOW7.js";
 import {
   convertRegistryConfigToClientConfig
-} from "../chunk-YU3C5OCU.js";
+} from "../chunk-MQDQ3A72.js";
 import "../chunk-JI6GZ2C2.js";
-import "../chunk-YHPFCYL3.js";
 import "../chunk-7AFZMFPQ.js";
-import "../chunk-OG4OR2CM.js";
+import "../chunk-X25YYVFL.js";
+import "../chunk-UBYPRYJR.js";
 import "../chunk-ZZ3WBRPD.js";
 
 // src/test/ready.ts

package/dist/tsup/utils.cjs

@@ -19,7 +19,7 @@
 
 
 
-var _chunk2SGPZEAJcjs = require('./chunk-2SGPZEAJ.cjs');
+var _chunkM4R2O5NJcjs = require('./chunk-M4R2O5NJ.cjs');
 require('./chunk-4FC7TVS6.cjs');
 
 
@@ -42,5 +42,5 @@ require('./chunk-4FC7TVS6.cjs');
 
 
 
-exports.EXTRA_ERROR_LOG = _chunk2SGPZEAJcjs.EXTRA_ERROR_LOG; exports.SinglePromiseQueue = _chunk2SGPZEAJcjs.SinglePromiseQueue; exports.VERSION = _chunk2SGPZEAJcjs.VERSION; exports.arrayBuffersEqual = _chunk2SGPZEAJcjs.arrayBuffersEqual; exports.assertUnreachable = _chunk2SGPZEAJcjs.assertUnreachable; exports.bufferToArrayBuffer = _chunk2SGPZEAJcjs.bufferToArrayBuffer; exports.combineUrlPath = _chunk2SGPZEAJcjs.combineUrlPath; exports.dbg = _chunk2SGPZEAJcjs.dbg; exports.detectRuntime = _chunk2SGPZEAJcjs.detectRuntime; exports.getEnvUniversal = _chunk2SGPZEAJcjs.getEnvUniversal; exports.httpUserAgent = _chunk2SGPZEAJcjs.httpUserAgent; exports.interval = _chunk2SGPZEAJcjs.interval; exports.joinSignals = _chunk2SGPZEAJcjs.joinSignals; exports.jsonParseCompat = _chunk2SGPZEAJcjs.jsonParseCompat; exports.jsonStringifyCompat = _chunk2SGPZEAJcjs.jsonStringifyCompat; exports.promiseWithResolvers = _chunk2SGPZEAJcjs.promiseWithResolvers; exports.setLongTimeout = _chunk2SGPZEAJcjs.setLongTimeout; exports.sleep = _chunk2SGPZEAJcjs.sleep; exports.stringifyError = _chunk2SGPZEAJcjs.stringifyError; exports.toUint8Array = _chunk2SGPZEAJcjs.toUint8Array;
+exports.EXTRA_ERROR_LOG = _chunkM4R2O5NJcjs.EXTRA_ERROR_LOG; exports.SinglePromiseQueue = _chunkM4R2O5NJcjs.SinglePromiseQueue; exports.VERSION = _chunkM4R2O5NJcjs.VERSION; exports.arrayBuffersEqual = _chunkM4R2O5NJcjs.arrayBuffersEqual; exports.assertUnreachable = _chunkM4R2O5NJcjs.assertUnreachable; exports.bufferToArrayBuffer = _chunkM4R2O5NJcjs.bufferToArrayBuffer; exports.combineUrlPath = _chunkM4R2O5NJcjs.combineUrlPath; exports.dbg = _chunkM4R2O5NJcjs.dbg; exports.detectRuntime = _chunkM4R2O5NJcjs.detectRuntime; exports.getEnvUniversal = _chunkM4R2O5NJcjs.getEnvUniversal; exports.httpUserAgent = _chunkM4R2O5NJcjs.httpUserAgent; exports.interval = _chunkM4R2O5NJcjs.interval; exports.joinSignals = _chunkM4R2O5NJcjs.joinSignals; exports.jsonParseCompat = _chunkM4R2O5NJcjs.jsonParseCompat; exports.jsonStringifyCompat = _chunkM4R2O5NJcjs.jsonStringifyCompat; exports.promiseWithResolvers = _chunkM4R2O5NJcjs.promiseWithResolvers; exports.setLongTimeout = _chunkM4R2O5NJcjs.setLongTimeout; exports.sleep = _chunkM4R2O5NJcjs.sleep; exports.stringifyError = _chunkM4R2O5NJcjs.stringifyError; exports.toUint8Array = _chunkM4R2O5NJcjs.toUint8Array;
 //# sourceMappingURL=utils.cjs.map

package/dist/tsup/utils.js

@@ -19,7 +19,7 @@ import {
   sleep,
   stringifyError,
   toUint8Array
-} from "./chunk-OG4OR2CM.js";
+} from "./chunk-UBYPRYJR.js";
 import "./chunk-ZZ3WBRPD.js";
 export {
   EXTRA_ERROR_LOG,

package/dist/tsup/workflow/mod.cjs

@@ -1,7 +1,7 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } async function _asyncNullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return await rhsFn(); } } var _class; var _class2;
 
-var _chunkTQHMPWL6cjs = require('../chunk-TQHMPWL6.cjs');
-require('../chunk-UIZZCJ5W.cjs');
+var _chunkKPP2YRS5cjs = require('../chunk-KPP2YRS5.cjs');
+require('../chunk-GZ5DYSIR.cjs');
 
 
 
@@ -12,7 +12,7 @@ var _chunkHNHLBM3Ycjs = require('../chunk-HNHLBM3Y.cjs');
 var _chunkMYWZCDMLcjs = require('../chunk-MYWZCDML.cjs');
 
 
-var _chunk2SGPZEAJcjs = require('../chunk-2SGPZEAJ.cjs');
+var _chunkM4R2O5NJcjs = require('../chunk-M4R2O5NJ.cjs');
 
 
 
@@ -716,7 +716,7 @@ function workflow(fn, options = {}) {
   function getWorkflowInspector(actorId) {
     let workflowInspector = workflowInspectors.get(actorId);
     if (!workflowInspector) {
-      workflowInspector = _chunkTQHMPWL6cjs.createWorkflowInspectorAdapter.call(void 0, );
+      workflowInspector = _chunkKPP2YRS5cjs.createWorkflowInspectorAdapter.call(void 0, );
       workflowInspectors.set(actorId, workflowInspector);
     }
     return workflowInspector;
@@ -784,13 +784,13 @@ function workflow(fn, options = {}) {
       if (shouldRethrowWorkflowError(error)) {
         runCtx.log.error({
           msg: "workflow run failed",
-          error: _chunk2SGPZEAJcjs.stringifyError.call(void 0, error)
+          error: _chunkM4R2O5NJcjs.stringifyError.call(void 0, error)
         });
         throw error;
       }
       runCtx.log.warn({
         msg: "workflow failed and will sleep until woken",
-        error: _chunk2SGPZEAJcjs.stringifyError.call(void 0, error)
+        error: _chunkM4R2O5NJcjs.stringifyError.call(void 0, error)
       });
     } finally {
       runCtx.abortSignal.removeEventListener("abort", onAbort);

package/dist/tsup/workflow/mod.d.cts

@@ -1,9 +1,9 @@
 import { WorkflowErrorEvent } from '@rivetkit/workflow-engine';
 export { Loop, TryBlockCatchKind, TryBlockConfig, TryBlockFailure, TryBlockResult, TryStepCatchKind, TryStepConfig, TryStepFailure, TryStepResult, WorkflowError, WorkflowErrorEvent } from '@rivetkit/workflow-engine';
-import { E as EventSchemaConfig, Q as QueueSchemaConfig, h as RunContext } from '../config-BobvQpex.cjs';
+import { E as EventSchemaConfig, Q as QueueSchemaConfig, h as RunContext } from '../config-CpB-e4I9.cjs';
 import { A as AnyDatabaseProvider } from '../config-BxOimoGX.cjs';
-import { d as WorkflowContext } from '../context-YEBNJrws.cjs';
-export { e as WorkflowBranchConfig, W as WorkflowBranchContextOf, a as WorkflowContextOf, f as WorkflowLoopConfig, b as WorkflowLoopContextOf, g as WorkflowStepConfig, h as WorkflowStepContext, c as WorkflowStepContextOf, i as WorkflowTryConfig, j as WorkflowTryStepConfig } from '../context-YEBNJrws.cjs';
+import { d as WorkflowContext } from '../context-c-6G4mjk.cjs';
+export { e as WorkflowBranchConfig, W as WorkflowBranchContextOf, a as WorkflowContextOf, f as WorkflowLoopConfig, b as WorkflowLoopContextOf, g as WorkflowStepConfig, h as WorkflowStepContext, c as WorkflowStepContextOf, i as WorkflowTryConfig, j as WorkflowTryStepConfig } from '../context-c-6G4mjk.cjs';
 import 'zod/v4';
 import '@rivetkit/rivetkit-wasm';
 import 'zod';

package/dist/tsup/workflow/mod.d.ts

@@ -1,9 +1,9 @@
 import { WorkflowErrorEvent } from '@rivetkit/workflow-engine';
 export { Loop, TryBlockCatchKind, TryBlockConfig, TryBlockFailure, TryBlockResult, TryStepCatchKind, TryStepConfig, TryStepFailure, TryStepResult, WorkflowError, WorkflowErrorEvent } from '@rivetkit/workflow-engine';
-import { E as EventSchemaConfig, Q as QueueSchemaConfig, h as RunContext } from '../config-BRpVbJCD.js';
+import { E as EventSchemaConfig, Q as QueueSchemaConfig, h as RunContext } from '../config-CUymbyM6.js';
 import { A as AnyDatabaseProvider } from '../config-BxOimoGX.js';
-import { d as WorkflowContext } from '../context-Cj-DFyiy.js';
-export { e as WorkflowBranchConfig, W as WorkflowBranchContextOf, a as WorkflowContextOf, f as WorkflowLoopConfig, b as WorkflowLoopContextOf, g as WorkflowStepConfig, h as WorkflowStepContext, c as WorkflowStepContextOf, i as WorkflowTryConfig, j as WorkflowTryStepConfig } from '../context-Cj-DFyiy.js';
+import { d as WorkflowContext } from '../context-YBUBTwTS.js';
+export { e as WorkflowBranchConfig, W as WorkflowBranchContextOf, a as WorkflowContextOf, f as WorkflowLoopConfig, b as WorkflowLoopContextOf, g as WorkflowStepConfig, h as WorkflowStepContext, c as WorkflowStepContextOf, i as WorkflowTryConfig, j as WorkflowTryStepConfig } from '../context-YBUBTwTS.js';
 import 'zod/v4';
 import '@rivetkit/rivetkit-wasm';
 import 'zod';

package/dist/tsup/workflow/mod.js

@@ -1,7 +1,7 @@
 import {
   createWorkflowInspectorAdapter
-} from "../chunk-7YQJXWBW.js";
-import "../chunk-6D2WRHJV.js";
+} from "../chunk-BQU4XTKM.js";
+import "../chunk-CQ23UFQ4.js";
 import {
   ACTOR_CONTEXT_INTERNAL_SYMBOL,
   RUN_FUNCTION_CONFIG_SYMBOL
@@ -12,7 +12,7 @@ import {
 } from "../chunk-JI6GZ2C2.js";
 import {
   stringifyError
-} from "../chunk-OG4OR2CM.js";
+} from "../chunk-UBYPRYJR.js";
 import {
   RivetError,
   isActorAbortedError

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "rivetkit",
-	"version": "2.3.7",
+	"version": "2.3.9",
 	"description": "Lightweight libraries for building stateful actors on edge platforms",
 	"license": "Apache-2.0",
 	"keywords": [
@@ -201,14 +201,14 @@
 		"@hono/zod-openapi": "^1.1.5",
 		"@rivet-dev/agent-os-core": "^0.1.1",
 		"@rivetkit/bare-ts": "^0.6.2",
-		"@rivetkit/engine-cli": "2.3.7",
-		"@rivetkit/engine-envoy-protocol": "2.3.7",
+		"@rivetkit/engine-cli": "2.3.9",
+		"@rivetkit/engine-envoy-protocol": "2.3.9",
 		"@rivetkit/on-change": "6.0.1",
-		"@rivetkit/rivetkit-napi": "2.3.7",
-		"@rivetkit/rivetkit-wasm": "2.3.7",
-		"@rivetkit/traces": "2.3.7",
-		"@rivetkit/virtual-websocket": "2.3.7",
-		"@rivetkit/workflow-engine": "2.3.7",
+		"@rivetkit/rivetkit-napi": "2.3.9",
+		"@rivetkit/rivetkit-wasm": "2.3.9",
+		"@rivetkit/traces": "2.3.9",
+		"@rivetkit/virtual-websocket": "2.3.9",
+		"@rivetkit/workflow-engine": "2.3.9",
 		"cbor-x": "^1.6.0",
 		"drizzle-orm": "^0.44.2",
 		"hono": "^4.7.0",

package/src/client/actor-handle.ts

@@ -4,6 +4,7 @@ import {
 	HEADER_CONN_PARAMS,
 	HEADER_ENCODING,
 } from "@/common/actor-router-consts";
+import { isRequestLike } from "@/common/fetch-like";
 import type * as protocol from "@/common/client-protocol";
 import {
 	CURRENT_VERSION as CLIENT_PROTOCOL_CURRENT_VERSION,
@@ -640,6 +641,7 @@ export class ActorHandleRaw {
 	) {
 		const maxAttempts = this.#getDynamicQueryMaxAttempts();
 		let useQueryTarget = false;
+		const retryableRequest = isRequestLike(input) ? input.clone() : undefined;
 		const { skipReadyWait, ...requestInit } = init ?? {};
 		const gatewayOptions = resolveActorGatewayOptions(
 			this.#gatewayOptions,
@@ -660,7 +662,7 @@ export class ActorHandleRaw {
 					this.#driver,
 					target,
 					this.#params,
-					input,
+					retryableRequest ? retryableRequest.clone() : input,
 					requestInit,
 					gatewayOptions,
 				);

package/src/client/raw-utils.ts

@@ -1,8 +1,10 @@
 import {
 	HEADER_CONN_PARAMS,
+	HEADER_ORIGINAL_REQUEST_URL,
 	PATH_WEBSOCKET_PREFIX,
 } from "@/common/actor-router-consts";
 import { deconstructError } from "@/common/utils";
+import { isRequestLike, isUrlLike } from "@/common/fetch-like";
 import type {
 	EngineControlClient,
 	GatewayRequestOptions,
@@ -24,16 +26,19 @@ export async function rawHttpFetch(
 ): Promise<Response> {
 	// Extract path and merge init options
 	let path: string;
+	let originalUrl: string | undefined;
 	let mergedInit: RequestInit = init || {};
 
 	if (typeof input === "string") {
 		path = input;
-	} else if (input instanceof URL) {
+	} else if (isUrlLike(input)) {
 		path = input.pathname + input.search;
-	} else if (input instanceof Request) {
+		originalUrl = input.toString();
+	} else if (isRequestLike(input)) {
 		// Extract path from Request URL
 		const url = new URL(input.url);
 		path = url.pathname + url.search;
+		originalUrl = url.toString();
 		// Merge Request properties with init
 		const requestHeaders = new Headers(input.headers);
 		const initHeaders = new Headers(init?.headers || {});
@@ -85,6 +90,10 @@ export async function rawHttpFetch(
 
 		// Forward conn params if provided
 		const proxyRequestHeaders = new Headers(mergedInit.headers);
+		proxyRequestHeaders.delete(HEADER_ORIGINAL_REQUEST_URL);
+		if (originalUrl) {
+			proxyRequestHeaders.set(HEADER_ORIGINAL_REQUEST_URL, originalUrl);
+		}
 		if (params) {
 			proxyRequestHeaders.set(HEADER_CONN_PARAMS, JSON.stringify(params));
 		}

package/src/common/actor-router-consts.ts

@@ -12,6 +12,7 @@ export const HEADER_ENCODING = "x-rivet-encoding";
 
 // IMPORTANT: Params must be in headers or in an E2EE part of the request (i.e. NOT the URL or query string) in order to ensure that tokens can be securely passed in params.
 export const HEADER_CONN_PARAMS = "x-rivet-conn-params";
+export const HEADER_ORIGINAL_REQUEST_URL = "x-rivet-internal-original-request-url";
 
 export const HEADER_ACTOR_ID = "x-rivet-actor";
 export const HEADER_ACTOR_GENERATION = "x-rivet-actor-generation";

package/src/common/encoding.ts

@@ -484,7 +484,10 @@ export function jsonStringifyCompat(
 				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)];

package/src/common/fetch-like.ts

@@ -0,0 +1,49 @@
+type HeadersLike = {
+	entries(): IterableIterator<[string, string]>;
+};
+
+export type RequestLike = Request & {
+	readonly headers: HeadersLike & Request["headers"];
+};
+
+export type ResponseLike = Response & {
+	readonly headers: HeadersLike & Response["headers"];
+};
+
+export function isUrlLike(value: unknown): value is URL {
+	return (
+		typeof value === "object" &&
+		value !== null &&
+		typeof (value as URL).href === "string" &&
+		typeof (value as URL).pathname === "string" &&
+		typeof (value as URL).search === "string"
+	);
+}
+
+export function isRequestLike(value: unknown): value is RequestLike {
+	return (
+		typeof value === "object" &&
+		value !== null &&
+		typeof (value as Request).url === "string" &&
+		typeof (value as Request).method === "string" &&
+		isHeadersLike((value as Request).headers)
+	);
+}
+
+export function isResponseLike(value: unknown): value is ResponseLike {
+	return (
+		typeof value === "object" &&
+		value !== null &&
+		typeof (value as Response).status === "number" &&
+		isHeadersLike((value as Response).headers) &&
+		typeof (value as Response).arrayBuffer === "function"
+	);
+}
+
+function isHeadersLike(value: unknown): value is HeadersLike {
+	return (
+		typeof value === "object" &&
+		value !== null &&
+		typeof (value as HeadersLike).entries === "function"
+	);
+}

package/src/engine-client/actor-http-client.ts

@@ -17,8 +17,7 @@ export async function sendHttpRequestToGateway(
 	actorRequest: Request,
 	options: HttpGatewayRequestOptions = {},
 ): Promise<Response> {
-	// Handle body properly based on method and presence
-	let bodyToSend: ArrayBuffer | null = null;
+	let bodyToSend: ReadableStream<Uint8Array> | null = null;
 	const guardHeaders = buildGuardHeaders(runConfig, actorRequest, options);
 
 	if (actorRequest.method !== "GET" && actorRequest.method !== "HEAD") {
@@ -26,35 +25,20 @@ export async function sendHttpRequestToGateway(
 			throw new Error("Request body has already been consumed");
 		}
 
-		// TODO: This buffers the entire request in memory every time. We
-		// need to properly implement streaming bodies.
-		const reqBody = await actorRequest.arrayBuffer();
-
-		if (reqBody.byteLength !== 0) {
-			bodyToSend = reqBody;
-
-			// If this is a streaming request, we need to convert the headers
-			// for the basic array buffer.
+		if (actorRequest.body) {
+			bodyToSend = actorRequest.body;
 			guardHeaders.delete("transfer-encoding");
 			guardHeaders.delete("content-length");
 		}
 	}
 
-	const guardRequest = new Request(gatewayUrl, {
+	return fetch(gatewayUrl, {
 		method: actorRequest.method,
 		headers: guardHeaders,
 		body: bodyToSend,
 		signal: actorRequest.signal,
-	});
-
-	return mutableResponse(await fetch(guardRequest));
-}
-
-function mutableResponse(fetchRes: Response): Response {
-	// We cannot return the raw response from `fetch` since the response type is not mutable.
-	//
-	// In order for middleware to be able to mutate the response, we need to build a new Response object that is mutable.
-	return new Response(fetchRes.body, fetchRes);
+		...(bodyToSend ? { duplex: "half" } : {}),
+	} as RequestInit);
 }
 
 function buildGuardHeaders(