rivetkit 2.3.12-rc.3 → 2.3.12

package/dist/browser/client.d.ts

@@ -1679,6 +1679,7 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
     engineHost: z$2.ZodDefault<z$2.ZodOptional<z$2.ZodString>>;
     enginePort: z$2.ZodDefault<z$2.ZodOptional<z$2.ZodNumber>>;
     engineVersion: z$2.ZodDefault<z$2.ZodOptional<z$2.ZodString>>;
+    startServices: z$2.ZodOptional<z$2.ZodBoolean>;
     configurePool: z$2.ZodOptional<z$2.ZodOptional<z$2.ZodObject<{
         name: z$2.ZodOptional<z$2.ZodString>;
         url: z$2.ZodString;
@@ -1710,6 +1711,7 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
         backend: "local" | "remote";
     } | undefined;
     endpoint: string | undefined;
+    startServices: boolean;
     namespace: string;
     token: string | undefined;
     publicEndpoint: string | undefined;
@@ -1821,6 +1823,7 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
     };
     staticDir?: string | undefined;
     httpHost?: string | undefined;
+    startServices?: boolean | undefined;
     configurePool?: {
         url: string;
         name?: string | undefined;
@@ -1861,6 +1864,12 @@ interface RuntimeHttpRequest {
     headers?: Record<string, string>;
     body?: RuntimeBytes;
 }
+interface RuntimeHttpResponseBodyStream {
+    cancelled(): Promise<void>;
+    write(chunk: Uint8Array): Promise<void>;
+    end(): Promise<void>;
+    error(message: string): Promise<void>;
+}
 interface RuntimeStateDeltaPayload {
     state?: RuntimeBytes;
     connHibernation: Array<{
@@ -2068,6 +2077,8 @@ interface RuntimeServeConfig {
     namespace: string;
     poolName: string;
     engineBinaryPath?: string;
+    startServices?: boolean;
+    servicesBinaryPath?: string;
     engineHost?: string;
     enginePort?: number;
     handleInspectorHttpInRuntime?: boolean;
@@ -2092,12 +2103,16 @@ interface RuntimeServerlessRequest {
     headers: Record<string, string>;
     body: RuntimeBytes;
 }
+interface RuntimeApplicationRequest extends RuntimeServerlessRequest {
+    cancelToken?: CancellationTokenHandle;
+}
 interface RuntimeApplicationResponse {
     status: number;
     headers: Record<string, string>;
-    body: RuntimeBytes;
+    body?: RuntimeBytes;
+    stream?: boolean;
 }
-type RuntimeApplicationFetch = (request: RuntimeServerlessRequest) => Promise<RuntimeApplicationResponse>;
+type RuntimeApplicationFetch = (request: RuntimeApplicationRequest, responseBodyStream?: RuntimeHttpResponseBodyStream) => Promise<RuntimeApplicationResponse>;
 interface RuntimeApplicationListenerConfig extends RuntimeListenerConfig {
     application: RuntimeApplicationFetch;
 }

package/dist/browser/client.js

@@ -201,7 +201,7 @@ function noopNext() {
 // package.json
 var package_default = {
   name: "rivetkit",
-  version: "2.3.12-rc.3",
+  version: "2.3.12",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [
@@ -411,6 +411,7 @@ var package_default = {
   dependencies: {
     "@hono/zod-openapi": "^1.1.5",
     "@rivet-dev/agent-os-core": "^0.1.1",
+    "@rivet-dev/services": "^0.1.3",
     "@rivetkit/bare-ts": "^0.6.2",
     "@rivetkit/engine-cli": "workspace:*",
     "@rivetkit/engine-envoy-protocol": "workspace:*",
@@ -1129,6 +1130,7 @@ var PATH_WEBSOCKET_BASE = "/websocket";
 var PATH_WEBSOCKET_PREFIX = "/websocket/";
 var HEADER_ENCODING = "x-rivet-encoding";
 var HEADER_CONN_PARAMS = "x-rivet-conn-params";
+var HEADER_ORIGINAL_REQUEST_URL = "x-rivet-internal-original-request-url";
 var HEADER_RIVET_TOKEN = "x-rivet-token";
 var HEADER_RIVET_TARGET = "x-rivet-target";
 var HEADER_RIVET_ACTOR = "x-rivet-actor";
@@ -1155,23 +1157,19 @@ async function sendHttpRequestToGateway(runConfig, gatewayUrl, actorRequest, opt
     if (actorRequest.bodyUsed) {
       throw new Error("Request body has already been consumed");
     }
-    const reqBody = await actorRequest.arrayBuffer();
-    if (reqBody.byteLength !== 0) {
-      bodyToSend = reqBody;
+    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
+    signal: actorRequest.signal,
+    ...bodyToSend ? { duplex: "half" } : {}
   });
-  return mutableResponse(await fetch(guardRequest));
-}
-function mutableResponse(fetchRes) {
-  return new Response(fetchRes.body, fetchRes);
 }
 function buildGuardHeaders(runConfig, actorRequest, options) {
   const headers = new Headers();
@@ -5883,6 +5881,17 @@ var ActorConnRaw = class {
   }
 };
 
+// src/common/fetch-like.ts
+function isUrlLike(value) {
+  return typeof value === "object" && value !== null && typeof value.href === "string" && typeof value.pathname === "string" && typeof value.search === "string";
+}
+function isRequestLike(value) {
+  return typeof value === "object" && value !== null && typeof value.url === "string" && typeof value.method === "string" && isHeadersLike(value.headers);
+}
+function isHeadersLike(value) {
+  return typeof value === "object" && value !== null && typeof value.entries === "function";
+}
+
 // src/common/database/shared.ts
 var MIGRATION_TRANSACTION_TIMEOUT_MS = 5 * 6e4;
 var AsyncMutex = class {
@@ -5911,28 +5920,20 @@ var AsyncMutex = class {
   }
 };
 
-// src/common/fetch-like.ts
-function isUrlLike(value) {
-  return typeof value === "object" && value !== null && typeof value.href === "string" && typeof value.pathname === "string" && typeof value.search === "string";
-}
-function isRequestLike(value) {
-  return typeof value === "object" && value !== null && typeof value.url === "string" && typeof value.method === "string" && isHeadersLike(value.headers);
-}
-function isHeadersLike(value) {
-  return typeof value === "object" && value !== null && typeof value.entries === "function";
-}
-
 // src/client/raw-utils.ts
 async function rawHttpFetch(driver, target, params, input, init, options = {}) {
   let path;
+  let originalUrl;
   let mergedInit = init || {};
   if (typeof input === "string") {
     path = input;
   } else if (isUrlLike(input)) {
     path = input.pathname + input.search;
+    originalUrl = input.toString();
   } else if (isRequestLike(input)) {
     const url = new URL(input.url);
     path = url.pathname + url.search;
+    originalUrl = url.toString();
     const requestHeaders = new Headers(input.headers);
     const initHeaders = new Headers(init?.headers || {});
     const mergedHeaders = new Headers(requestHeaders);
@@ -5974,6 +5975,10 @@ async function rawHttpFetch(driver, target, params, input, init, options = {}) {
     const normalizedPath = path.startsWith("/") ? path.slice(1) : path;
     const url = new URL(`http://actor/request/${normalizedPath}`);
     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));
     }
@@ -6103,7 +6108,9 @@ var ActorHandleRaw = class {
   async #sendQueueMessage(name, body, options) {
     return await this.#queueSendMutex.run(async () => {
       const maxAttempts = this.#getDynamicQueryMaxAttempts();
-      let useQueryTarget = isDynamicActorQuery(this.#actorResolutionState);
+      let useQueryTarget = isDynamicActorQuery(
+        this.#actorResolutionState
+      );
       for (let attempt = 0; attempt < maxAttempts; attempt++) {
         let actorId;
         try {
@@ -6452,9 +6459,10 @@ var ActorHandleRaw = class {
     return this.#fetchWithResolvedActor(input, init);
   }
   async #fetchWithResolvedActor(input, init) {
-    const maxAttempts = this.#getDynamicQueryMaxAttempts();
-    let useQueryTarget = isDynamicActorQuery(this.#actorResolutionState);
     const { skipReadyWait, ...requestInit } = init ?? {};
+    const hasBody = requestInit.body !== void 0 ? requestInit.body !== null : isRequestLike(input) && input.body !== null;
+    const maxAttempts = hasBody ? 1 : this.#getDynamicQueryMaxAttempts();
+    let useQueryTarget = isDynamicActorQuery(this.#actorResolutionState);
     const gatewayOptions = resolveActorGatewayOptions(
       this.#gatewayOptions,
       {