Hide the assistant entry when there isn't data2summary agent

CHANGELOG.md

@@ -4,7 +4,6 @@ Inspired from [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 
 ## Unreleased
 
-
 ### Features
 
 - introduces Pipeline to execute asynchronous operations ([#376](https://github.com/opensearch-project/dashboards-assistant/pull/376))
@@ -20,7 +19,7 @@ Inspired from [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 - Use feature flag to disable the rename conversation api ([#401](https://github.com/opensearch-project/dashboards-assistant/pull/410))
 - Disable delete conversation api based on feature flag([#409](https://github.com/opensearch-project/dashboards-assistant/pull/409))
 - Add options to hide "stop generation" and regenerate button based on feature flag ([#394](https://github.com/opensearch-project/dashboards-assistant/pull/394))
-
+- Hide the assistant entry when there isn't data2summary agent ([#417](https://github.com/opensearch-project/dashboards-assistant/pull/417))
 
 ### Bug Fixes
 
@@ -37,6 +36,7 @@ Inspired from [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 ### Maintenance
 
 ### Refactoring
+
 - Add query assistant summary to the assistant dropdown list ([#395](https://github.com/opensearch-project/dashboards-assistant/pull/395))
 - Update dropdown list button label and remove popover title ([#407](https://github.com/opensearch-project/dashboards-assistant/pull/407))
 - Add support for registerMessageParser ([#5](https://github.com/opensearch-project/dashboards-assistant/pull/5))
@@ -45,5 +45,3 @@ Inspired from [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 - Refactor the code to get root agent id by calling the API in ml-commons plugin ([#128](https://github.com/opensearch-project/dashboards-assistant/pull/128))
 - build(deps): bump braces from 3.0.2 to 3.0.3 ([#213](https://github.com/opensearch-project/dashboards-assistant/pull/213))
 - build(deps): bump ws from 8.16.0 to 8.18.0 ([#221](https://github.com/opensearch-project/dashboards-assistant/pull/221))
-
-

common/constants/llm.ts

@@ -47,3 +47,4 @@ export const TEXT2VEGA_INPUT_SIZE_LIMIT = 400;
 export const TEXT2VEGA_RULE_BASED_AGENT_CONFIG_ID = 'os_text2vega';
 export const TEXT2VEGA_WITH_INSTRUCTIONS_AGENT_CONFIG_ID = 'os_text2vega_with_instructions';
 export const TEXT2PPL_AGENT_CONFIG_ID = 'os_query_assist_ppl';
+export const DATA2SUMMARY_AGENT_CONFIG_ID = 'os_data2summary';

public/components/ui_action_context_menu.tsx

@@ -16,15 +16,18 @@ import { i18n } from '@osd/i18n';
 
 import { BehaviorSubject } from 'rxjs';
 import { useObservable } from 'react-use';
+import { HttpSetup } from '../../../../src/core/public';
 import { buildContextMenuForActions } from '../../../../src/plugins/ui_actions/public';
 import { AI_ASSISTANT_QUERY_EDITOR_TRIGGER } from '../ui_triggers';
 import { getUiActions } from '../services';
 import { DataPublicPluginSetup } from '../../../../src/plugins/data/public';
-
+import { AGENT_API, DATA2SUMMARY_AGENT_CONFIG_ID } from '../../common/constants/llm';
 interface Props {
   data: DataPublicPluginSetup;
   isQuerySummaryCollapsed$: BehaviorSubject<boolean>;
   resultSummaryEnabled$: BehaviorSubject<boolean>;
+  isSummaryAgentAvailable$: BehaviorSubject<boolean>;
+  httpSetup: HttpSetup;
   label?: string;
 }
 
@@ -39,6 +42,35 @@ export const ActionContextMenu = (props: Props) => {
   });
   const resultSummaryEnabled = useObservable(props.resultSummaryEnabled$, false);
   const isQuerySummaryCollapsed = useObservable(props.isQuerySummaryCollapsed$, false);
+  const isSummaryAgentAvailable = useObservable(props.isSummaryAgentAvailable$, false);
+
+  async function checkAgentsExist(
+    http: HttpSetup,
+    agentConfigName: string | string[],
+    dataSourceId?: string
+  ) {
+    const response = await http.get(AGENT_API.CONFIG_EXISTS, {
+      query: { agentConfigName, dataSourceId },
+    });
+    return response;
+  }
+
+  useEffect(() => {
+    const fetchSummaryAgent = async () => {
+      try {
+        const summaryAgentStatus = await checkAgentsExist(
+          props.httpSetup,
+          DATA2SUMMARY_AGENT_CONFIG_ID,
+          props.data.query.queryString.getQuery().dataset?.dataSource?.id
+        );
+        props.isSummaryAgentAvailable$.next(!!summaryAgentStatus.exists);
+      } catch (error) {
+        console.error(error);
+      }
+    };
+
+    fetchSummaryAgent();
+  }, [props, props.httpSetup]);
 
   useEffect(() => {
     const subscription = props.data.query.queryString.getUpdates$().subscribe((query) => {
@@ -75,14 +107,13 @@ export const ActionContextMenu = (props: Props) => {
     [actionContext.datasetId, actionContext.datasetType, actionContext.dataSourceId]
   );
 
-  // The action button should be not displayed when there is no action and result summary disabled.
-  if (actionsRef.current.length === 0 && !resultSummaryEnabled) {
+  // The action button should be not displayed when there is no action and result summary disabled or there is no data2Summary agent
+  if (!isSummaryAgentAvailable || (actionsRef.current.length === 0 && !resultSummaryEnabled)) {
     return null;
   }
 
-  // The action button should be disabled when context menu has no item and result summary disabled.
+  // The action button should be disabled when context menu has no item and result summary disabled
   const actionDisabled = (panels.value?.[0]?.items ?? []).length === 0 && !resultSummaryEnabled;
-
   return (
     <EuiPopover
       button={

public/plugin.tsx

@@ -282,7 +282,11 @@ export class AssistantPlugin
       }
     })();
 
-    const { isQuerySummaryCollapsed$, resultSummaryEnabled$ } = setupDeps.queryEnhancements;
+    const {
+      isQuerySummaryCollapsed$,
+      resultSummaryEnabled$,
+      isSummaryAgentAvailable$,
+    } = setupDeps.queryEnhancements;
     setupDeps.data.__enhance({
       editor: {
         queryEditorExtension: {
@@ -292,9 +296,11 @@ export class AssistantPlugin
           getSearchBarButton: () => {
             return (
               <ActionContextMenu
+                httpSetup={core.http}
                 label={this.config.branding.label}
                 isQuerySummaryCollapsed$={isQuerySummaryCollapsed$}
                 resultSummaryEnabled$={resultSummaryEnabled$}
+                isSummaryAgentAvailable$={isSummaryAgentAvailable$}
                 data={setupDeps.data}
               />
             );