improve completion plugin menu items

jupyterlab · Jan 19, 2024 · 9a4d43f · 9a4d43f
1 parent aad63dd
commit 9a4d43f
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/packages/jupyter-ai/src/completions/plugin.ts b/packages/jupyter-ai/src/completions/plugin.ts
@@ -133,7 +133,7 @@ export const completionPlugin: JupyterFrontEndPlugin<void> = {
           enabled: !jaiIcpSettings.enabled
         });
       },
-      label: 'Enable Jupyternaut Completions',
+      label: 'Enable completions by Jupyternaut',
       isToggled: () => {
         return provider.isEnabled();
       }
@@ -159,12 +159,16 @@ export const completionPlugin: JupyterFrontEndPlugin<void> = {
       label: () => {
         const language = findCurrentLanguage();
         return language
-          ? `Enable Completions in ${displayName(language)}`
-          : 'Enable Completions for <language>';
+          ? `Disable completions for ${displayName(language)} files`
+          : 'Disable completions for <language> files';
       },
       isToggled: () => {
         const language = findCurrentLanguage();
-        return !!language && provider.isLanguageEnabled(language.name);
+        return !!language && !provider.isLanguageEnabled(language.name);
+      },
+      isVisible: () => {
+        const language = findCurrentLanguage();
+        return !!language;
       },
       isEnabled: () => {
         const language = findCurrentLanguage();