Fix Workers AI local mode fetcher not returning headers to client wor…

…ker (#5138) * Fix Workers AI local mode fetcher not returning headers to client worker * Trigger Build * Update .changeset/yellow-phones-buy.md Co-authored-by: MrBBot <[email protected]> * Update packages/wrangler/src/ai/fetcher.ts Co-authored-by: MrBBot <[email protected]> * Add missing Headers import * Lint changes --------- Co-authored-by: MrBBot <[email protected]>
cloudflare · Mar 4, 2024 · 3dd9089 · 3dd9089
1 parent b745f7b
commit 3dd9089
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/.changeset/yellow-phones-buy.md b/.changeset/yellow-phones-buy.md
@@ -0,0 +1,5 @@
+---
+"wrangler": patch
+---
+
+fix: ensure Workers-AI local mode fetcher returns headers to client worker
diff --git a/packages/wrangler/src/ai/fetcher.ts b/packages/wrangler/src/ai/fetcher.ts
@@ -1,4 +1,4 @@
-import { Response } from "miniflare";
+import { Headers, Response } from "miniflare";
 import { performApiFetch } from "../cfetch/internal";
 import { getAccountId } from "../user";
 import type { Request } from "miniflare";
@@ -16,5 +16,9 @@ export async function AIFetcher(request: Request) {
 		duplex: "half",
 	});
 
-	return new Response(res.body, { status: res.status });
+	const respHeaders = new Headers(res.headers);
+	respHeaders.delete("Host");
+	respHeaders.delete("Content-Length");
+
+	return new Response(res.body, { status: res.status, headers: respHeaders });
 }