Update local AI fetcher to forward method and pathname to upstream (#…

…7315)

* Update local AI fetcher to forward method and pathname to upstream

* Add unit test to cover changes

* Lint files

* Rename x-forward-for to x-forward header name

.changeset/stupid-moons-juggle.md

@@ -0,0 +1,5 @@
+---
+"wrangler": minor
+---
+
+Update local AI fetcher to forward method and url to upstream

packages/wrangler/src/__tests__/ai.local.test.ts

@@ -0,0 +1,69 @@
+import { Request } from "miniflare";
+import { HttpResponse } from "msw";
+import { AIFetcher } from "../ai/fetcher";
+import * as internal from "../cfetch/internal";
+import * as user from "../user";
+import type { RequestInit } from "undici";
+
+describe("ai", () => {
+	describe("fetcher", () => {
+		afterEach(() => {
+			vi.restoreAllMocks();
+		});
+
+		describe("local", () => {
+			it("should send x-forwarded header", async () => {
+				vi.spyOn(user, "getAccountId").mockImplementation(async () => "123");
+				vi.spyOn(internal, "performApiFetch").mockImplementation(
+					async (resource: string, init: RequestInit = {}) => {
+						const headers = new Headers(init.headers);
+						return HttpResponse.json({
+							xForwarded: headers.get("X-Forwarded"),
+							method: init.method,
+						});
+					}
+				);
+
+				const url = "http://internal.ai/ai/test/path?version=123";
+				const resp = await AIFetcher(
+					new Request(url, {
+						method: "PATCH",
+						headers: {
+							"x-example": "test",
+						},
+					})
+				);
+
+				expect(await resp.json()).toEqual({
+					xForwarded: url,
+					method: "PATCH",
+				});
+			});
+
+			it("account id should be set", async () => {
+				vi.spyOn(user, "getAccountId").mockImplementation(async () => "123");
+				vi.spyOn(internal, "performApiFetch").mockImplementation(
+					async (resource: string) => {
+						return HttpResponse.json({
+							resource: resource,
+						});
+					}
+				);
+
+				const url = "http://internal.ai/ai/test/path?version=123";
+				const resp = await AIFetcher(
+					new Request(url, {
+						method: "PATCH",
+						headers: {
+							"x-example": "test",
+						},
+					})
+				);
+
+				expect(await resp.json()).toEqual({
+					resource: "/accounts/123/ai/run/proxy",
+				});
+			});
+		});
+	});
+});

packages/wrangler/src/ai/fetcher.ts

@@ -16,12 +16,14 @@ export default function (env) {
 export async function AIFetcher(request: Request): Promise<Response> {
 	const accountId = await getAccountId();
 
-	request.headers.delete("Host");
-	request.headers.delete("Content-Length");
+	const reqHeaders = new Headers(request.headers);
+	reqHeaders.delete("Host");
+	reqHeaders.delete("Content-Length");
+	reqHeaders.set("X-Forwarded", request.url);
 
 	const res = await performApiFetch(`/accounts/${accountId}/ai/run/proxy`, {
-		method: "POST",
-		headers: Object.fromEntries(request.headers.entries()),
+		method: request.method,
+		headers: Object.fromEntries(reqHeaders.entries()),
 		body: request.body,
 		duplex: "half",
 	});