added back message to only return code for 405b

Nutlope · Nutlope · commit ffe22adea646 · 2024-08-03T13:25:00.000-07:00
diff --git a/app/api/generateCode/route.ts b/app/api/generateCode/route.ts
@@ -32,15 +32,13 @@ export async function POST(req: Request) {
         role: "system",
         content: systemPrompt,
       },
-      ...(model === "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo"
-        ? messages
-        : messages.map((message: any) => {
-            if (message.role === "user") {
-              message.content +=
-                "\nPlease ONLY return code, NO backticks or language names.";
-            }
-            return message;
-          })),
+      ...messages.map((message: any) => {
+        if (message.role === "user") {
+          message.content +=
+            "\nPlease ONLY return code, NO backticks or language names.";
+        }
+        return message;
+      }),
     ],
     stream: true,
   };
