Disable query assist when AI feature is off (#10851)

Signed-off-by: Owen Wang <[email protected]>

Author: FriedhelmWS

src/plugins/query_enhancements/public/query_assist/utils/create_extension.test.tsx

@@ -89,7 +89,7 @@ describe('CreateExtension', () => {
     summary: { enabled: false, branding: { label: '' } },
   };
 
-  it('should be enabled if at least one language is configured', async () => {
+  it('should be enabled if at least one language is configured and AI features enabled', async () => {
     httpMock.get.mockResolvedValueOnce({ configuredLanguages: ['PPL'] });
     const extension = createQueryAssistExtension(
       coreSetupMock,
@@ -123,6 +123,36 @@ describe('CreateExtension', () => {
     });
   });
 
+  it('should be disabled when AI features are disabled', async () => {
+    const mockCoreSetup = {
+      ...coreSetupMock,
+      getStartServices: jest.fn().mockResolvedValue([
+        {
+          ...coreMock.createStart(),
+          uiSettings: {
+            ...uiSettingsServiceMock.createStartContract(),
+            get: jest.fn().mockReturnValue(false),
+          },
+        },
+        {},
+        {},
+      ]),
+    };
+    httpMock.get.mockResolvedValueOnce({ configuredLanguages: ['PPL'] });
+    const extension = createQueryAssistExtension(
+      mockCoreSetup,
+      dataMock,
+      config,
+      mockIsQuerySummaryCollapsed$,
+      mockIsSummaryAgentAvailable$,
+      mockresultSummaryEnabled$
+    );
+    // Wait for assistantEnabled$ to be updated
+    await new Promise((resolve) => setTimeout(resolve, 0));
+    const isEnabled = await firstValueFrom(extension.isEnabled$(dependencies));
+    expect(isEnabled).toBeFalsy();
+  });
+
   it('creates data structure meta', async () => {
     httpMock.get.mockResolvedValueOnce({ configuredLanguages: ['PPL'] });
     const extension = createQueryAssistExtension(

src/plugins/query_enhancements/public/query_assist/utils/create_extension.tsx

@@ -133,7 +133,9 @@ export const createQueryAssistExtension = (
       }
     },
     isEnabled$: () =>
-      getAvailableLanguages$(http, data).pipe(map((languages) => languages.length > 0)),
+      getAvailableLanguages$(http, data).pipe(
+        map((languages) => languages.length > 0 && assistantEnabled$.value)
+      ),
     getComponent: (dependencies) => {
       // only show the component if user is on a supported language.
       return (