@finos/legend-application-query 13.8.43 → 13.8.45

package/src/__lib__/LegendQueryEvent.ts

@@ -23,6 +23,8 @@ export enum LEGEND_QUERY_APP_EVENT {
   UPDATE_QUERY__SUCCESS = 'query-editor.update-query.success',
   RENAME_QUERY__SUCCESS = 'query-editor.rename.query.success',
   INITIALIZE_QUERY_STATE__SUCCESS = 'query-editor.initialize-query-state.success',
+  LEGENDAI_QUERY_AGENT_CHAT__OPENED = 'query-editor.legendai-query-agent-chat.opened',
+  LEGENDAI_QUERY_AGENT_CHAT__QUERY_LOADED = 'query-editor.legendai-query-agent-chat.query-loaded',
   LEGENDAI_QUERY_SUGGEST__LAUNCH = 'query-editor.legendai-query-suggest.launch',
   LEGENDAI_QUERY_SUGGEST__APPLY = 'query-editor.legendai-query-suggest.apply',
   LEGENDAI_QUERY_SUGGEST__DISCARD = 'query-editor.legendai-query-suggest.discard',

package/src/__lib__/LegendQueryTelemetryHelper.ts

@@ -30,7 +30,7 @@ type Query_TelemetryData = {
     groupId: string;
     artifactId: string;
   };
-};
+} & Record<string, unknown>;
 
 type QueryGraphInitialization_TelemetryData = Query_TelemetryData & {
   graph: GraphInitializationReport;

package/src/application/LegendQueryApplicationConfig.ts

@@ -161,7 +161,8 @@ export interface LegendQueryApplicationConfigurationData
     url: string;
   };
   legendAI?: {
-    url: string;
+    url?: string;
+    agentURL?: string;
   };
 }
 
@@ -178,6 +179,7 @@ export class LegendQueryApplicationConfig extends LegendApplicationConfig {
   readonly marketplaceProductionParallelUrl?: string;
   readonly lakehouseContractUrl?: string;
   readonly legendAIUrl?: string;
+  readonly legendAIAgentUrl?: string;
   readonly studioInstances: LegendStudioApplicationInstanceConfigurationData[] =
     [];
 
@@ -266,13 +268,20 @@ export class LegendQueryApplicationConfig extends LegendApplicationConfig {
       );
     }
 
-    // legend AI
+    // legend AI (AI-assisted features such as query title/description suggester)
     if (input.configData.legendAI?.url) {
       this.legendAIUrl = LegendApplicationConfig.resolveAbsoluteUrl(
         input.configData.legendAI.url,
       );
     }
 
+    // legend AI agent chat
+    if (input.configData.legendAI?.agentURL) {
+      this.legendAIAgentUrl = LegendApplicationConfig.resolveAbsoluteUrl(
+        input.configData.legendAI.agentURL,
+      );
+    }
+
     // options
     this.options = LegendQueryApplicationCoreOptions.create(
       input.configData.extensions?.core ?? {},

package/src/components/QueryEditor.tsx

@@ -43,6 +43,7 @@ import {
   MoonIcon,
   SunIcon,
   SparkleIcon,
+  RobotIcon,
 } from '@finos/legend-art';
 import { observer } from 'mobx-react-lite';
 import { useEffect, useMemo, useRef, useState } from 'react';
@@ -57,7 +58,7 @@ import {
   generateExistingQueryEditorRoute,
 } from '../__lib__/LegendQueryNavigation.js';
 import { ExistingQueryEditorStore } from '../stores/QueryEditorStore.js';
-import { LegendQueryTelemetryHelper } from '../__lib__/LegendQueryTelemetryHelper.js';
+import { LegendQueryAgentChatTelemetryHelper } from '../__lib__/LegendQueryAgentChatTelemetryHelper.js';
 import {
   LEGEND_APPLICATION_COLOR_THEME,
   ReleaseLogManager,
@@ -186,7 +187,7 @@ const CreateQueryDialog = observer(() => {
     if (!aiSuggester || !editorStore.queryBuilderState || !legendAIUrl) {
       return;
     }
-    LegendQueryTelemetryHelper.logEvent_QueryAISuggestLaunched(
+    LegendQueryAgentChatTelemetryHelper.logEvent_QueryAISuggestLaunched(
       applicationStore.telemetryService,
     );
     setIsSuggestingWithAI(true);
@@ -199,7 +200,7 @@ const CreateQueryDialog = observer(() => {
       setAISuggestion(suggestion);
     } catch (error) {
       assertErrorThrown(error);
-      LegendQueryTelemetryHelper.logEvent_QueryAISuggestFailure(
+      LegendQueryAgentChatTelemetryHelper.logEvent_QueryAISuggestFailure(
         applicationStore.telemetryService,
         error.message,
       );
@@ -224,7 +225,7 @@ const CreateQueryDialog = observer(() => {
     if (!aiSuggestion) {
       return;
     }
-    LegendQueryTelemetryHelper.logEvent_QueryAISuggestApplied(
+    LegendQueryAgentChatTelemetryHelper.logEvent_QueryAISuggestApplied(
       applicationStore.telemetryService,
     );
     createQueryState.setQueryName(aiSuggestion.title);
@@ -232,7 +233,7 @@ const CreateQueryDialog = observer(() => {
     setAISuggestion(undefined);
   };
   const discardAISuggestion = (): void => {
-    LegendQueryTelemetryHelper.logEvent_QueryAISuggestDiscarded(
+    LegendQueryAgentChatTelemetryHelper.logEvent_QueryAISuggestDiscarded(
       applicationStore.telemetryService,
     );
     setAISuggestion(undefined);
@@ -519,7 +520,7 @@ const RenameQueryDialog = observer(
         setAISuggestion(suggestion);
       } catch (error) {
         assertErrorThrown(error);
-        LegendQueryTelemetryHelper.logEvent_QueryAISuggestFailure(
+        LegendQueryAgentChatTelemetryHelper.logEvent_QueryAISuggestFailure(
           applicationStore.telemetryService,
           error.message,
         );
@@ -1093,6 +1094,20 @@ export const QueryEditor = observer(() => {
     );
   };
 
+  const toggleAgentChat = (): void => {
+    const qbState = editorStore.queryBuilderState;
+    if (!qbState) {
+      return;
+    }
+    const nextOpened = !qbState.isAgentChatOpened;
+    if (nextOpened) {
+      LegendQueryAgentChatTelemetryHelper.logEvent_QueryAgentChatOpened(
+        applicationStore.telemetryService,
+      );
+    }
+    qbState.setIsAgentChatOpened(nextOpened);
+  };
+
   // open the query version history (same viewer used by the query loader)
   const showQueryHistory = (): void => {
     if (editorStore instanceof ExistingQueryEditorStore) {
@@ -1165,6 +1180,46 @@ export const QueryEditor = observer(() => {
           </div>
         </div>
         <div className="query-editor__header__action__content">
+          {!isLoadingEditor &&
+            !editorStore.queryBuilderState?.config
+              ?.TEMPORARY__disableQueryBuilderAgentChat &&
+            editorStore.queryBuilderState?.canBuildQuery && (
+              <button
+                title={
+                  editorStore.queryBuilderState.isAgentChatOpened
+                    ? 'Close Legend AI Agent Chat'
+                    : 'Open Legend AI Agent Chat'
+                }
+                onClick={toggleAgentChat}
+                className={clsx(
+                  'query-editor__header__action query-editor__header__action__theme-toggler',
+                  {
+                    'query-editor__header__action--toggled':
+                      editorStore.queryBuilderState.isAgentChatOpened,
+                  },
+                )}
+              >
+                <div
+                  className={
+                    applicationStore.layoutService
+                      .TEMPORARY__isLightColorThemeEnabled
+                      ? 'query-editor__header__action__chat__label--light'
+                      : 'query-editor__header__action__chat__label--dark'
+                  }
+                >
+                  Legend AI
+                </div>
+                <RobotIcon
+                  className={clsx(
+                    'query-editor__header__action__chat__icon',
+                    applicationStore.layoutService
+                      .TEMPORARY__isLightColorThemeEnabled
+                      ? 'query-editor__header__action__chat__label--light'
+                      : 'query-editor__header__action__chat__label--dark',
+                  )}
+                />
+              </button>
+            )}
           <button
             title="Toggle light/dark mode"
             onClick={TEMPORARY__toggleLightDarkMode}

package/src/components/data-space/DataSpaceQuerySetup.tsx

@@ -87,6 +87,7 @@ const DataProductQuerySetupSetupPanelContent = observer(
     const onDataSpaceOptionChange = (
       option: DataSpaceOption | DataProductOption | null,
     ): void => {
+      queryBuilderState.queryAgentChatState?.abort();
       const value = option?.value;
       if (value instanceof ResolvedDataSpaceEntityWithOrigin) {
         queryBuilderState.changeHandlers.onDataSpaceChange(value);

package/src/components/data-space/LegendQueryDataSpaceQueryBuilder.tsx

@@ -112,10 +112,12 @@ const LegendQueryDataSpaceQueryBuilderSetupPanelContent = observer(
       }
       const value = option.value;
       if (value instanceof ResolvedDataSpaceEntityWithOrigin) {
+        queryBuilderState.queryAgentChatState?.abort();
         queryBuilderState
           .onDataSpaceChange(value)
           .catch(queryBuilderState.applicationStore.alertUnhandledError);
       } else if (value instanceof DepotEntityWithOrigin) {
+        queryBuilderState.queryAgentChatState?.abort();
         queryBuilderState.onDataProductChange(value);
       }
     };

package/src/index.ts

@@ -21,6 +21,8 @@ export * from './application/LegendQueryPluginManager.js';
 
 export * from './__lib__/LegendQueryEvent.js';
 export * from './__lib__/LegendQueryEventHelper.js';
+export { LegendQueryTelemetryHelper } from './__lib__/LegendQueryTelemetryHelper.js';
+export { LegendQueryAgentChatTelemetryHelper } from './__lib__/LegendQueryAgentChatTelemetryHelper.js';
 export {
   generateExistingQueryEditorRoute,
   generateServiceQueryCreatorRoute,

package/src/stores/QueryEditorStore.ts

@@ -280,6 +280,7 @@ export class QueryCreatorState {
             artifactId: query.artifactId,
             versionId: query.versionId,
           },
+          ...queryBuilderState.getExtraTelemetryMetadata(),
         },
       );
       queryBuilderState.changeDetectionState.initialize(rawLambda);
@@ -1764,6 +1765,7 @@ export class ExistingQueryUpdateState {
             artifactId: query.artifactId,
             versionId: query.versionId,
           },
+          ...queryBuilderState.getExtraTelemetryMetadata(),
         },
       );
       config.onQueryUpdate?.(updatedQuery);
@@ -1800,6 +1802,8 @@ export class ExistingQueryUpdateState {
         `Successfully updated query!`,
       );
 
+      const extraTelemetryMetadata =
+        this.editorStore.queryBuilderState?.getExtraTelemetryMetadata() ?? {};
       LegendQueryTelemetryHelper.logEvent_UpdateQuerySucceeded(
         this.editorStore.applicationStore.telemetryService,
         {
@@ -1810,6 +1814,7 @@ export class ExistingQueryUpdateState {
             artifactId: updatedQuery.artifactId,
             versionId: updatedQuery.versionId,
           },
+          ...extraTelemetryMetadata,
         },
       );
     } catch (error) {

package/tsconfig.json

@@ -54,6 +54,7 @@
     "./src/index.ts",
     "./src/__lib__/DSL_DataSpace_LegendQueryDocumentation.ts",
     "./src/__lib__/DSL_DataSpace_LegendQueryNavigation.ts",
+    "./src/__lib__/LegendQueryAgentChatTelemetryHelper.ts",
     "./src/__lib__/LegendQueryApplicationNavigationContext.ts",
     "./src/__lib__/LegendQueryEvent.ts",
     "./src/__lib__/LegendQueryEventHelper.ts",