diff --git a/web/app/api/(newai)/folders/v2/route.ts b/web/app/api/(newai)/folders/v2/route.ts
index 4b764438..f5d9aa63 100644
--- a/web/app/api/(newai)/folders/v2/route.ts
+++ b/web/app/api/(newai)/folders/v2/route.ts
@@ -14,12 +14,16 @@ export async function POST(request: NextRequest) {
     const response = await generateObject({
       model,
       schema: z.object({
-        suggestedFolders: z.array(z.object({
-          score: z.number().min(0).max(100),
-          isNewFolder: z.boolean(),
-          folder: z.string(),
-          reason: z.string(),
-        })).min(3)
+        suggestedFolders: z
+          .array(
+            z.object({
+              score: z.number().min(0).max(100),
+              isNewFolder: z.boolean(),
+              folder: z.string(),
+              reason: z.string(),
+            })
+          )
+          .max(5),
       }),
       system: `Given the content and (if useful) the file name: "${fileName}", suggest relevant folders from the following list: ${folders.join(
         ", if none of the folders are relevant, suggest new folders"
@@ -35,7 +39,9 @@ export async function POST(request: NextRequest) {
     console.log("incrementing token usage folders", userId, tokens);
     await incrementAndLogTokenUsage(userId, tokens);
     return NextResponse.json({
-      folders: response.object.suggestedFolders.sort((a, b) => b.score - a.score),
+      folders: response.object.suggestedFolders.sort(
+        (a, b) => b.score - a.score
+      ),
     });
   } catch (error) {
     if (error) {