diff --git a/apps/builder/app/builder/features/ai/ai-fetch-result.ts b/apps/builder/app/builder/features/ai/ai-fetch-result.ts
index dcf48f2943..1e101ef76f 100644
--- a/apps/builder/app/builder/features/ai/ai-fetch-result.ts
+++ b/apps/builder/app/builder/features/ai/ai-fetch-result.ts
@@ -30,7 +30,7 @@ import {
 } from "~/shared/nano-states";
 import { applyOperations, patchTextInstance } from "./apply-operations";
 import { restAi } from "~/shared/router-utils";
-import { RequestParamsSchema } from "~/routes/rest.ai";
+import { RequestParamsSchema } from "~/routes/rest.ai._index";
 import untruncateJson from "untruncate-json";
 import { traverseTemplate } from "@webstudio-is/jsx-utils";
 
diff --git a/apps/builder/app/builder/features/ai/commands-bar.tsx b/apps/builder/app/builder/features/ai/commands-bar.tsx
index 1d892515dd..271f794e86 100644
--- a/apps/builder/app/builder/features/ai/commands-bar.tsx
+++ b/apps/builder/app/builder/features/ai/commands-bar.tsx
@@ -34,7 +34,7 @@ import {
 import { applyOperations, patchTextInstance } from "./apply-operations";
 import { useStore } from "@nanostores/react";
 import { restAi } from "~/shared/router-utils";
-import { RequestParamsSchema } from "~/routes/rest.ai";
+import { RequestParamsSchema } from "~/routes/rest.ai._index";
 import untruncateJson from "untruncate-json";
 import { traverseTemplate } from "@webstudio-is/jsx-utils";
 
diff --git a/apps/builder/app/routes/rest.ai.ts b/apps/builder/app/routes/rest.ai._index.ts
similarity index 99%
rename from apps/builder/app/routes/rest.ai.ts
rename to apps/builder/app/routes/rest.ai._index.ts
index e922fdf9d0..9f95cacba6 100644
--- a/apps/builder/app/routes/rest.ai.ts
+++ b/apps/builder/app/routes/rest.ai._index.ts
@@ -36,7 +36,9 @@ export const RequestParamsSchema = z.object({
 });
 
 // Override Vercel's default serverless functions timeout.
-export const maxDuration = 180; // seconds
+export const config = {
+  maxDuration: 180, // seconds
+};
 
 export const action = async ({ request }: ActionArgs) => {
   // @todo Reinstate isFeatureEnabled('ai')