Enable Legend AI button for DataSpaceQuerySetupState also. (#3485)

finos · Aug 29, 2024 · 03ec0a3 · 03ec0a3
1 parent 9cef2f8
commit 03ec0a3
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/.changeset/thin-trees-do.md b/.changeset/thin-trees-do.md
@@ -0,0 +1,5 @@
+---
+'@finos/legend-application-query': patch
+---
+
+Adding conditions for Legend AI button and exporting DataSpaceQuerySetupState.
diff --git a/packages/legend-application-query/src/components/QueryEditor.tsx b/packages/legend-application-query/src/components/QueryEditor.tsx
@@ -57,6 +57,7 @@ import {
 } from '../__lib__/LegendQueryNavigation.js';
 import { ExistingQueryEditorStore } from '../stores/QueryEditorStore.js';
 import { LegendQueryTelemetryHelper } from '../__lib__/LegendQueryTelemetryHelper.js';
+import { DataSpaceQuerySetupState } from './../stores/data-space/DataSpaceQuerySetupState.js';
 import {
  LEGEND_APPLICATION_COLOR_THEME,
  ReleaseLogManager,
@@ -687,8 +688,10 @@ export const QueryEditor = observer(() => {
  {!isLoadingEditor &&
  !editorStore.queryBuilderState?.config
  ?.TEMPORARY__disableQueryBuilderChat &&
- editorStore.queryBuilderState instanceof
- DataSpaceQueryBuilderState &&
+ (editorStore.queryBuilderState instanceof
+ DataSpaceQueryBuilderState ||
+ editorStore.queryBuilderState instanceof
+ DataSpaceQuerySetupState) &&
  editorStore.queryBuilderState.canBuildQuery && (
  <button
  title="Open Query Chat."

diff --git a/packages/legend-application-query/src/index.ts b/packages/legend-application-query/src/index.ts
@@ -45,6 +45,8 @@ export {
  createViewProjectHandler,
  createViewSDLCProjectHandler,
 } from './stores/data-space/DataSpaceQueryBuilderHelper.js';
+export { DataSpaceQuerySetupState } from './stores/data-space/DataSpaceQuerySetupState.js';
+
 // components
 export {
  QueryEditorStoreContext,