refactor(t2viz): register the AI actions to query controls in discover (

#327) * refactor(t2viz): register the AI actions to query controls in discover Signed-off-by: Yulong Ruan <[email protected]> * add changelogs Signed-off-by: Yulong Ruan <[email protected]> * remove default of config entry branding.label Signed-off-by: Yulong Ruan <[email protected]> --------- Signed-off-by: Yulong Ruan <[email protected]>
opensearch-project · Sep 26, 2024 · 9930241 · 9930241
1 parent 64fb466
commit 9930241
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 9 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -15,6 +15,7 @@ Inspired from [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 - fix: pass data source id for alert summary/insight([#321](https://github.com/opensearch-project/dashboards-assistant/pull/321))
 - feat: support navigating to discover in alerting popover([#316](https://github.com/opensearch-project/dashboards-assistant/pull/316))
 - fix: incorrect string escaping of vega schema([325](https://github.com/opensearch-project/dashboards-assistant/pull/325))
+- feat: register the AI actions to query controls in discover([#327](https://github.com/opensearch-project/dashboards-assistant/pull/327))
 
 ### 📈 Features/Enhancements
 

diff --git a/common/types/config.ts b/common/types/config.ts
@@ -25,6 +25,9 @@ export const configSchema = schema.object({
   smartAnomalyDetector: schema.object({
     enabled: schema.boolean({ defaultValue: false }),
   }),
+  branding: schema.object({
+    label: schema.string({ defaultValue: '' }),
+  }),
 });
 
 export type ConfigSchema = TypeOf<typeof configSchema>;
diff --git a/public/components/ui_action_context_menu.tsx b/public/components/ui_action_context_menu.tsx
@@ -5,14 +5,18 @@
 
 import React, { useState, useRef } from 'react';
 import useAsync from 'react-use/lib/useAsync';
-import { EuiButtonIcon, EuiContextMenu, EuiPopover } from '@elastic/eui';
+import { EuiButtonEmpty, EuiContextMenu, EuiPopover } from '@elastic/eui';
+import { i18n } from '@osd/i18n';
 
 import { buildContextMenuForActions } from '../../../../src/plugins/ui_actions/public';
 import { AI_ASSISTANT_QUERY_EDITOR_TRIGGER } from '../ui_triggers';
 import { getUiActions } from '../services';
-import assistantTriggerIcon from '../assets/assistant_trigger.svg';
 
-export const ActionContextMenu = () => {
+interface Props {
+  label?: string;
+}
+
+export const ActionContextMenu = (props: Props) => {
   const uiActions = getUiActions();
   const actionsRef = useRef(uiActions.getTriggerActions(AI_ASSISTANT_QUERY_EDITOR_TRIGGER));
   const [open, setOpen] = useState(false);
@@ -38,12 +42,20 @@ export const ActionContextMenu = () => {
   return (
     <EuiPopover
       button={
-        <EuiButtonIcon
+        <EuiButtonEmpty
+          color="text"
           aria-label="AI assistant trigger button"
-          size="s"
-          iconType={assistantTriggerIcon}
+          size="xs"
+          iconType="arrowDown"
           onClick={() => setOpen(!open)}
-        />
+          iconSide="right"
+          flush="both"
+        >
+          {props.label ||
+            i18n.translate('dashboardAssistant.branding.assistantActionButton.label', {
+              defaultMessage: 'AI assistant',
+            })}
+        </EuiButtonEmpty>
       }
       isOpen={open}
       panelPaddingSize="none"

diff --git a/public/plugin.tsx b/public/plugin.tsx
@@ -258,8 +258,8 @@ export class AssistantPlugin
           id: 'assistant-query-actions',
           order: 2000,
           isEnabled$: () => of(true),
-          getComponent: () => {
-            return <ActionContextMenu />;
+          getSearchBarButton: () => {
+            return <ActionContextMenu label={this.config.branding.label} />;
           },
         },
       },

diff --git a/server/index.ts b/server/index.ts
@@ -15,6 +15,7 @@ export const config: PluginConfigDescriptor<ConfigSchema> = {
     text2viz: true,
     alertInsight: true,
     smartAnomalyDetector: true,
+    branding: true,
   },
   schema: configSchema,
 };