fix

YousefED · YousefED · commit a72d49172fb0 · 2025-02-05T10:01:09.000+01:00
diff --git a/packages/xl-ai/src/components/BlockNoteAIContext.tsx b/packages/xl-ai/src/components/BlockNoteAIContext.tsx
@@ -99,7 +99,10 @@ export function BlockNoteAIContextProvider(
           }
           ret = await llm.markdown.call(editor, { model, ...options });
         }
-        setAIResponseStatus("done");
+        // if the menu has been closed already, don't set it to done
+        if (aiMenuBlockID !== undefined) {
+          setAIResponseStatus("done");
+        }
         return ret;
       } catch (e) {
         setAIResponseStatus("initial");
@@ -108,7 +111,7 @@ export function BlockNoteAIContextProvider(
         console.error(e);
       }
     },
-    [model, dataFormat, stream, editor]
+    [model, dataFormat, stream, editor, aiMenuBlockID]
   );
 
   // TODO: Revisit - this pattern might be a bit iffy

Original file line number	Diff line number	Diff line change
`@@ -99,7 +99,10 @@ export function BlockNoteAIContextProvider(`
`99`	`99`	`}`
`100`	`100`	`ret = await llm.markdown.call(editor, { model, ...options });`
`101`	`101`	`}`
`102`		`- setAIResponseStatus("done");`
	`102`	`+ // if the menu has been closed already, don't set it to done`
	`103`	`+ if (aiMenuBlockID !== undefined) {`
	`104`	`+ setAIResponseStatus("done");`
	`105`	`+ }`
`103`	`106`	`return ret;`
`104`	`107`	`} catch (e) {`
`105`	`108`	`setAIResponseStatus("initial");`
`@@ -108,7 +111,7 @@ export function BlockNoteAIContextProvider(`
`108`	`111`	`console.error(e);`
`109`	`112`	`}`
`110`	`113`	`},`
`111`		`- [model, dataFormat, stream, editor]`
	`114`	`+ [model, dataFormat, stream, editor, aiMenuBlockID]`
`112`	`115`	`);`
`113`	`116`
`114`	`117`	`// TODO: Revisit - this pattern might be a bit iffy`