fix(editor): conditionally execute menu action after LLM call

unit-mesh · Nov 1, 2024 · 45172d9 · 45172d9
1 parent 13583ff
commit 45172d9
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/web/core/lib/editor/menu/menu-bubble.tsx b/web/core/lib/editor/menu/menu-bubble.tsx
@@ -78,13 +78,17 @@ export const MenuBubble = ({ editor, customActions } : {
                   setIsOpen(false);
                   setLoading(true);
 
-                  const text = await editor.commands?.callLlm(menu)
-                  setLoading(false);
+                  if (!menu.action) {
+                    const text = await editor.commands?.callLlm(menu)
+                    setLoading(false);
+
+                    const newComment = newAdvice(text || '');
+                    editor.commands?.setAdvice(newComment.id);
+                    editor.commands?.setAdviceCommand(newComment);
+                  } else {
+                    menu.action?.(editor);
+                  }
 
-                  const newComment = newAdvice(text || '');
-                  editor.commands?.setAdvice(newComment.id);
-                  editor.commands?.setAdviceCommand(newComment);
-                  menu.action?.(editor);
                   editor.view?.focus();
                 }}
               >