feat: check if label suggestion is enabled in hooks (#13331)

app/javascript/dashboard/composables/spec/useCaptain.spec.js

@@ -76,18 +76,6 @@ describe('useCaptain', () => {
     });
   });
 
-  it('gets label suggestions', async () => {
-    TasksAPI.labelSuggestion.mockResolvedValue({
-      data: { message: 'label1, label2' },
-    });
-
-    const { getLabelSuggestions } = useCaptain();
-    const result = await getLabelSuggestions();
-
-    expect(TasksAPI.labelSuggestion).toHaveBeenCalledWith('123');
-    expect(result).toEqual(['label1', 'label2']);
-  });
-
   it('rewrites content', async () => {
     TasksAPI.rewrite.mockResolvedValue({
       data: { message: 'Rewritten content', follow_up_context: { id: 'ctx1' } },
@@ -193,21 +181,4 @@ describe('useCaptain', () => {
     await processEvent('improve', 'content', {});
     expect(TasksAPI.rewrite).toHaveBeenCalled();
   });
-
-  it('returns empty array when no conversation ID for label suggestions', async () => {
-    useMapGetter.mockImplementation(getter => {
-      const mockValues = {
-        'accounts/getUIFlags': { isFetchingLimits: false },
-        getSelectedChat: { id: null },
-        'draftMessages/getReplyEditorMode': 'reply',
-      };
-      return { value: mockValues[getter] };
-    });
-
-    const { getLabelSuggestions } = useCaptain();
-    const result = await getLabelSuggestions();
-
-    expect(result).toEqual([]);
-    expect(TasksAPI.labelSuggestion).not.toHaveBeenCalled();
-  });
 });

app/javascript/dashboard/composables/useCaptain.js

@@ -13,20 +13,6 @@ import { FEATURE_FLAGS } from 'dashboard/featureFlags';
 import { OPEN_AI_EVENTS } from 'dashboard/helper/AnalyticsHelper/events';
 import TasksAPI from 'dashboard/api/captain/tasks';
 
-/**
- * Cleans and normalizes a list of labels.
- * @param {string} labels - A comma-separated string of labels.
- * @returns {string[]} An array of cleaned and unique labels.
- */
-const cleanLabels = labels => {
-  return labels
-    .toLowerCase()
-    .split(',')
-    .filter(label => label.trim())
-    .map(label => label.trim())
-    .filter((label, index, self) => self.indexOf(label) === index);
-};
-
 export function useCaptain() {
   const store = useStore();
   const { t } = useI18n();
@@ -183,24 +169,6 @@ export function useCaptain() {
     }
   };
 
-  /**
-   * Gets label suggestions for the current conversation.
-   * @returns {Promise<string[]>} An array of suggested labels.
-   */
-  const getLabelSuggestions = async () => {
-    if (!conversationId.value) return [];
-
-    try {
-      const result = await TasksAPI.labelSuggestion(conversationId.value);
-      const {
-        data: { message: labels },
-      } = result;
-      return cleanLabels(labels);
-    } catch {
-      return [];
-    }
-  };
-
   /**
    * Sends a follow-up message to refine a previous AI task result.
    * @param {Object} options - The follow-up options.
@@ -264,7 +232,6 @@ export function useCaptain() {
     rewriteContent,
     summarizeConversation,
     getReplySuggestion,
-    getLabelSuggestions,
     followUp,
     processEvent,
 

app/javascript/dashboard/composables/useLabelSuggestions.js

@@ -0,0 +1,80 @@
+import { computed, onMounted } from 'vue';
+import { useMapGetter, useStore } from 'dashboard/composables/store';
+import { useAccount } from 'dashboard/composables/useAccount';
+import { FEATURE_FLAGS } from 'dashboard/featureFlags';
+import TasksAPI from 'dashboard/api/captain/tasks';
+
+/**
+ * Cleans and normalizes a list of labels.
+ * @param {string} labels - A comma-separated string of labels.
+ * @returns {string[]} An array of cleaned and unique labels.
+ */
+const cleanLabels = labels => {
+  return labels
+    .toLowerCase()
+    .split(',')
+    .filter(label => label.trim())
+    .map(label => label.trim())
+    .filter((label, index, self) => self.indexOf(label) === index);
+};
+
+export function useLabelSuggestions() {
+  const store = useStore();
+  const { isCloudFeatureEnabled } = useAccount();
+  const appIntegrations = useMapGetter('integrations/getAppIntegrations');
+  const currentChat = useMapGetter('getSelectedChat');
+  const conversationId = computed(() => currentChat.value?.id);
+
+  const captainTasksEnabled = computed(() => {
+    return isCloudFeatureEnabled(FEATURE_FLAGS.CAPTAIN_TASKS);
+  });
+
+  const aiIntegration = computed(
+    () =>
+      appIntegrations.value.find(
+        integration => integration.id === 'openai' && !!integration.hooks.length
+      )?.hooks[0]
+  );
+
+  const isLabelSuggestionFeatureEnabled = computed(() => {
+    if (aiIntegration.value) {
+      const { settings = {} } = aiIntegration.value || {};
+      return !!settings.label_suggestion;
+    }
+    return false;
+  });
+
+  const fetchIntegrationsIfRequired = async () => {
+    if (!appIntegrations.value.length) {
+      await store.dispatch('integrations/get');
+    }
+  };
+
+  /**
+   * Gets label suggestions for the current conversation.
+   * @returns {Promise<string[]>} An array of suggested labels.
+   */
+  const getLabelSuggestions = async () => {
+    if (!conversationId.value) return [];
+
+    try {
+      const result = await TasksAPI.labelSuggestion(conversationId.value);
+      const {
+        data: { message: labels },
+      } = result;
+      return cleanLabels(labels);
+    } catch {
+      return [];
+    }
+  };
+
+  onMounted(() => {
+    fetchIntegrationsIfRequired();
+  });
+
+  return {
+    captainTasksEnabled,
+    isLabelSuggestionFeatureEnabled,
+    getLabelSuggestions,
+  };
+}