Spaces:

enzostvs
/

deepsite

Running

enzostvs HF Staff commited on 26 days ago

Commit

6b93566

1 Parent(s): da77bf0

fix think

Files changed (2) hide show

server.js CHANGED Viewed

@@ -227,6 +227,9 @@ app.post("/api/ask-ai", async (req, res) => {
     });
   }
   const selectedModel = MODELS.find(
     (m) => m.value === model || m.label === model
   );
@@ -314,8 +317,9 @@ app.post("/api/ask-ai", async (req, res) => {
               },
             ]
           : []),
-        ...(html
-          ? [
               {
                 role: "assistant",
                 content: `The current code is: ${html}.`,

     });
   }
+  const isFollowUp =
+    previousPrompt && previousPrompt.length > 0 && html && html.length > 0;
   const selectedModel = MODELS.find(
     (m) => m.value === model || m.label === model
   );
               },
             ]
           : []),
+        ...(isFollowUp
+          ? // TODO try to do the git diff thing to only generate the changes and not the whole html
+            [
               {
                 role: "assistant",
                 content: `The current code is: ${html}.`,

src/components/ask-ai/ask-ai.tsx CHANGED Viewed

@@ -60,6 +60,7 @@ function AskAI({
     setIsThinking(true);
     let contentResponse = "";
     let lastRenderTime = 0;
     try {
       onNewPrompt(prompt);
@@ -121,10 +122,10 @@ function AskAI({
           }
           const chunk = decoder.decode(value, { stream: true });
-          contentResponse += chunk;
           if (selectedModel?.isThinker) {
-            const thinkMatch = contentResponse.match(/<think>[\s\S]*/)?.[0];
-            if (thinkMatch && !contentResponse?.includes("</think>")) {
               if ((contentThink?.length ?? 0) < 3) {
                 setOpenThink(true);
               }
@@ -134,6 +135,8 @@ function AskAI({
             }
           }
           const newHtml = contentResponse.match(/<!DOCTYPE html>[\s\S]*/)?.[0];
           if (newHtml) {
             setIsThinking(false);

     setIsThinking(true);
     let contentResponse = "";
+    let thinkResponse = "";
     let lastRenderTime = 0;
     try {
       onNewPrompt(prompt);
           }
           const chunk = decoder.decode(value, { stream: true });
+          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);
               }
             }
           }
+          contentResponse += chunk;
           const newHtml = contentResponse.match(/<!DOCTYPE html>[\s\S]*/)?.[0];
           if (newHtml) {
             setIsThinking(false);