Merge pull request #6256 from G4brym/add-ai-gateway-flag-for-rpc

jclee · web-flow · commit f9bb39e02a79 · 2026-03-06T12:08:43.000-08:00
Add flag to route ai gateway methods via rpc
diff --git a/src/cloudflare/internal/ai-api.ts b/src/cloudflare/internal/ai-api.ts
@@ -16,6 +16,7 @@ type AiSearchService = object;
 interface Fetcher {
   fetch: typeof fetch;
   aiSearch: () => AiSearchService;
+  gateway: (gatewayId: string) => AiGateway;
 }
 
 interface AiError {
@@ -418,7 +419,10 @@ export class Ai {
     return service.transform(files, options);
   }
 
-  gateway(gatewayId: string): AiGateway {
+  gateway(gatewayId: string, options?: { beta?: boolean }): AiGateway {
+    if (options?.beta === true) {
+      return this.#fetcher.gateway(gatewayId);
+    }
     return new AiGateway(this.#fetcher, gatewayId);
   }
 

Original file line number	Diff line number	Diff line change
`@@ -16,6 +16,7 @@ type AiSearchService = object;`
`16`	`16`	`interface Fetcher {`
`17`	`17`	`fetch: typeof fetch;`
`18`	`18`	`aiSearch: () => AiSearchService;`
	`19`	`+ gateway: (gatewayId: string) => AiGateway;`
`19`	`20`	`}`
`20`	`21`
`21`	`22`	`interface AiError {`
`@@ -418,7 +419,10 @@ export class Ai {`
`418`	`419`	`return service.transform(files, options);`
`419`	`420`	`}`
`420`	`421`
`421`		`- gateway(gatewayId: string): AiGateway {`
	`422`	`+ gateway(gatewayId: string, options?: { beta?: boolean }): AiGateway {`
	`423`	`+ if (options?.beta === true) {`
	`424`	`+ return this.#fetcher.gateway(gatewayId);`
	`425`	`+ }`
`422`	`426`	`return new AiGateway(this.#fetcher, gatewayId);`
`423`	`427`	`}`
`424`	`428`