fix

Author: YousefED

packages/xl-ai/src/components/BlockNoteAIContext.tsx

@@ -99,10 +99,14 @@ export function BlockNoteAIContextProvider(
           }
           ret = await llm.markdown.call(editor, { model, ...options });
         }
-        // if the menu has been closed already, don't set it to done
-        if (aiMenuBlockID !== undefined) {
-          setAIResponseStatus("done");
-        }
+        setAIResponseStatus((old) => {
+          // if the menu has been closed already, it's probably set to "initial" and not "generating" anymore,
+          // in that case, don't set it to "done"
+          if (old === "generating") {
+            return "done";
+          }
+          return old;
+        });
         return ret;
       } catch (e) {
         setAIResponseStatus("initial");
@@ -111,7 +115,7 @@ export function BlockNoteAIContextProvider(
         console.error(e);
       }
     },
-    [model, dataFormat, stream, editor, aiMenuBlockID]
+    [model, dataFormat, stream, editor]
   );
 
   // TODO: Revisit - this pattern might be a bit iffy