Fix Workers AI local mode fetcher not returning headers to client worker

cloudflare · Mar 1, 2024 · d5c2e39 · d5c2e39
1 parent 9d1d11c
commit d5c2e39
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/.changeset/yellow-phones-buy.md b/.changeset/yellow-phones-buy.md
@@ -0,0 +1,5 @@
+---
+"wrangler": patch
+---
+
+Fix Workers-AI local mode fetcher not returning headers to client worker
diff --git a/packages/wrangler/src/ai/fetcher.ts b/packages/wrangler/src/ai/fetcher.ts
@@ -16,5 +16,12 @@ export async function AIFetcher(request: Request) {
 		duplex: "half",
 	});
 
-	return new Response(res.body, { status: res.status });
+	const respHeaders: Record<string, string> = {};
+	for (const [name, value] of res.headers) {
+		respHeaders[name] = value;
+	}
+	respHeaders.headers.delete("Host");
+	respHeaders.headers.delete("Content-Length");
+
+	return new Response(res.body, { status: res.status, headers: respHeaders });
 }