Merge pull request cloudflare#133 from cloudflare/fix/r2-screenshots

AshishKumar4 · web-flow · commit e27c47fa9943 · 2025-10-03T01:47:55.000-04:00
fix: use CUSTOM_DOMAIN env var instead of hostname for r2 screenshot urls
diff --git a/worker/agents/core/simpleGeneratorAgent.ts b/worker/agents/core/simpleGeneratorAgent.ts
@@ -203,7 +203,7 @@ export class SimpleCodeGeneratorAgent extends Agent<Env, CodeGenState> {
         // Build a public URL served via Worker route
         const fileName = key.split('/').pop() as string;
         const protocol = getProtocolForHost(this.state.hostname);
-        const base = this.state.hostname ? `${protocol}://${this.state.hostname}` : '';
+        const base = `${protocol}://${this.env.CUSTOM_DOMAIN}`;
         const agentId = this.state.inferenceContext.agentId;
         const url = `${base}/api/screenshots/${encodeURIComponent(agentId)}/${encodeURIComponent(fileName)}`;
         return url;
