feat: disable query assistant when dataset type is not supported

Signed-off-by: SuZhou-Joe <[email protected]>

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

@@ -230,4 +230,22 @@ describe('CreateExtension', () => {
 
     expect(screen.getByText('QueryAssistSummary')).toBeInTheDocument();
   });
+
+  it('should return disabled when dataset is not supported', async () => {
+    httpMock.get.mockResolvedValueOnce({ configuredLanguages: ['PPL'] });
+    const extension = createQueryAssistExtension(coreSetupMock, dataMock, config);
+    const isEnabled = await firstValueFrom(
+      extension.isEnabled$({
+        ...dependencies,
+        query: {
+          ...mockQueryWithIndexPattern,
+          dataset: {
+            ...mockQueryWithIndexPattern.dataset,
+            type: 'S3',
+          },
+        },
+      })
+    );
+    expect(isEnabled).toBeFalsy();
+  });
 });

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

@@ -88,6 +88,8 @@ const getAvailableLanguages$ = (http: HttpSetup, data: DataPublicPluginSetup) =>
     })
   );
 
+const queryAssistantSupportedDatasetTypes = ['INDEXES', 'INDEX_PATTERN'];
+
 export const createQueryAssistExtension = (
   core: CoreSetup,
   data: DataPublicPluginSetup,
@@ -115,8 +117,13 @@ export const createQueryAssistExtension = (
         };
       }
     },
-    isEnabled$: () =>
-      getAvailableLanguages$(http, data).pipe(map((languages) => languages.length > 0)),
+    isEnabled$: (dependencies) => {
+      const query = dependencies.query;
+      if (!queryAssistantSupportedDatasetTypes.includes(query.dataset?.type || '')) {
+        return new BehaviorSubject(false);
+      }
+      return getAvailableLanguages$(http, data).pipe(map((languages) => languages.length > 0));
+    },
     getComponent: (dependencies) => {
       // only show the component if user is on a supported language.
       return (