Merge pull request #104 from replicate/prevent-response-caching-in-ne…

…xt.js Prevent response caching in Next.js
replicate · May 28, 2024 · e840c5c · e840c5c
2 parents 421025d + d5cde8f
commit e840c5c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/app/api/predictions/[id]/route.js b/app/api/predictions/[id]/route.js
@@ -5,6 +5,12 @@ const replicate = new Replicate({
   auth: process.env.REPLICATE_API_TOKEN,
 });
 
+// Prevent Next.js / Vercel from caching responses
+// See https://github.com/replicate/replicate-javascript/issues/136#issuecomment-1728053102
+replicate.fetch = (url, options) => {
+  return fetch(url, { cache: "no-store", ...options });
+};
+
 export async function GET(request, {params}) {
   const { id } = params;
   const prediction = await replicate.predictions.get(id);

diff --git a/app/api/predictions/route.js b/app/api/predictions/route.js
@@ -5,6 +5,12 @@ const replicate = new Replicate({
   auth: process.env.REPLICATE_API_TOKEN,
 });
 
+// Prevent Next.js / Vercel from caching responses
+// See https://github.com/replicate/replicate-javascript/issues/136#issuecomment-1728053102
+replicate.fetch = (url, options) => {
+  return fetch(url, { ...options, cache: "no-store" });
+};
+
 // In production and preview deployments (on Vercel), the VERCEL_URL environment variable is set.
 // In development (on your local machine), the NGROK_HOST environment variable is set.
 const WEBHOOK_HOST = process.env.VERCEL_URL