temporary fix

app/api/ask-ai/route.ts

@@ -192,7 +192,6 @@ export async function POST(request: NextRequest) {
             encoder.encode(
               JSON.stringify({
                 ok: false,
-                openSelectProvider: true,
                 message:
                   error.message ||
                   "An error occurred while processing your request.",
@@ -201,14 +200,7 @@ export async function POST(request: NextRequest) {
           );
         }
       } finally {
-        try {
-          // Only close if writer exists and is not already closed
-          if (writer && !writer.closed) {
-            await writer.close();
-          }
-        } catch {
-          // Ignore errors on close
-        }
+        await writer?.close();
       }
     })();
 

components/editor/ask-ai/index.tsx

@@ -145,21 +145,6 @@ export function AskAI({
           signal: abortController.signal,
         });
         if (request && request.body) {
-          // if (!request.ok) {
-          //   const res = await request.json();
-          //   if (res.openLogin) {
-          //     setOpen(true);
-          //   } else if (res.openSelectProvider) {
-          //     setOpenProvider(true);
-          //     setProviderError(res.message);
-          //   } else if (res.openProModal) {
-          //     setOpenProModal(true);
-          //   } else {
-          //     toast.error(res.message);
-          //   }
-          //   setisAiWorking(false);
-          //   return;
-          // }
           const reader = request.body.getReader();
           const decoder = new TextDecoder("utf-8");
           const selectedModel = MODELS.find(
@@ -169,6 +154,26 @@ export function AskAI({
           const read = async () => {
             const { done, value } = await reader.read();
             if (done) {
+              const isJson =
+                contentResponse.trim().startsWith("{") &&
+                contentResponse.trim().endsWith("}");
+              const jsonResponse = isJson ? JSON.parse(contentResponse) : null;
+              if (jsonResponse && !jsonResponse.ok) {
+                if (jsonResponse.openLogin) {
+                  setOpen(true);
+                } else if (jsonResponse.openSelectProvider) {
+                  setOpenProvider(true);
+                  setProviderError(jsonResponse.message);
+                } else if (jsonResponse.openProModal) {
+                  setOpenProModal(true);
+                } else {
+                  toast.error(jsonResponse.message);
+                }
+                setisAiWorking(false);
+                return;
+              }
+
+              console.log("AI response complete", contentResponse);
               toast.success("AI responded successfully");
               setPreviousPrompt(prompt);
               setPrompt("");
@@ -190,74 +195,55 @@ export function AskAI({
             }
 
             const chunk = decoder.decode(value, { stream: true });
-
-            const isJson =
-              chunk.trim().startsWith("{") && chunk.trim().endsWith("}");
-            const res = isJson ? JSON.parse(chunk) : null;
-            if (res && !res.ok) {
-              if (res.openLogin) {
-                setOpen(true);
-              } else if (res.openSelectProvider) {
-                setOpenProvider(true);
-                setProviderError(res.message);
-              } else if (res.openProModal) {
-                setOpenProModal(true);
-              } else {
-                toast.error(res.message);
-              }
-              setisAiWorking(false);
-              return;
-            } else {
-              thinkResponse += chunk;
-              if (selectedModel?.isThinker) {
-                const thinkMatch = thinkResponse.match(/<think>[\s\S]*/)?.[0];
-                if (thinkMatch && !thinkResponse?.includes("</think>")) {
-                  if ((contentThink?.length ?? 0) < 3) {
-                    setOpenThink(true);
-                  }
-                  setThink(thinkMatch.replace("<think>", "").trim());
-                  contentThink += chunk;
-                  return read();
+            thinkResponse += chunk;
+            if (selectedModel?.isThinker) {
+              const thinkMatch = thinkResponse.match(/<think>[\s\S]*/)?.[0];
+              if (thinkMatch && !thinkResponse?.includes("</think>")) {
+                if ((contentThink?.length ?? 0) < 3) {
+                  setOpenThink(true);
                 }
+                setThink(thinkMatch.replace("<think>", "").trim());
+                contentThink += chunk;
+                return read();
               }
+            }
 
-              contentResponse += chunk;
+            contentResponse += chunk;
 
-              const newHtml = contentResponse.match(
-                /<!DOCTYPE html>[\s\S]*/
-              )?.[0];
-              if (newHtml) {
-                setIsThinking(false);
-                let partialDoc = newHtml;
-                if (
-                  partialDoc.includes("<head>") &&
-                  !partialDoc.includes("</head>")
-                ) {
-                  partialDoc += "\n</head>";
-                }
-                if (
-                  partialDoc.includes("<body") &&
-                  !partialDoc.includes("</body>")
-                ) {
-                  partialDoc += "\n</body>";
-                }
-                if (!partialDoc.includes("</html>")) {
-                  partialDoc += "\n</html>";
-                }
+            const newHtml = contentResponse.match(
+              /<!DOCTYPE html>[\s\S]*/
+            )?.[0];
+            if (newHtml) {
+              setIsThinking(false);
+              let partialDoc = newHtml;
+              if (
+                partialDoc.includes("<head>") &&
+                !partialDoc.includes("</head>")
+              ) {
+                partialDoc += "\n</head>";
+              }
+              if (
+                partialDoc.includes("<body") &&
+                !partialDoc.includes("</body>")
+              ) {
+                partialDoc += "\n</body>";
+              }
+              if (!partialDoc.includes("</html>")) {
+                partialDoc += "\n</html>";
+              }
 
-                // Throttle the re-renders to avoid flashing/flicker
-                const now = Date.now();
-                if (now - lastRenderTime > 300) {
-                  setHtml(partialDoc);
-                  lastRenderTime = now;
-                }
+              // Throttle the re-renders to avoid flashing/flicker
+              const now = Date.now();
+              if (now - lastRenderTime > 300) {
+                setHtml(partialDoc);
+                lastRenderTime = now;
+              }
 
-                if (partialDoc.length > 200) {
-                  onScrollToBottom();
-                }
+              if (partialDoc.length > 200) {
+                onScrollToBottom();
               }
-              read();
             }
+            read();
           };
 
           read();