[ML] Trained models list: disables 'View training data' action if data frame analytics job no longer exists

x-pack/plugins/ml/common/types/trained_models.ts

@@ -98,6 +98,7 @@ export type TrainedModelConfigResponse = estypes.MlTrainedModelConfig & {
    * Associated pipelines. Extends response from the ES endpoint.
    */
   pipelines?: Record<string, PipelineDefinition> | null;
+  origin_job_exists?: boolean;
 
   metadata?: {
     analytics_config: DataFrameAnalyticsConfig;

x-pack/plugins/ml/public/application/model_management/model_actions.tsx

@@ -7,9 +7,10 @@
 
 import { Action } from '@elastic/eui/src/components/basic_table/action_types';
 import { i18n } from '@kbn/i18n';
+import { FormattedMessage } from '@kbn/i18n-react';
 import { isPopulatedObject } from '@kbn/ml-is-populated-object';
-import { EuiToolTip } from '@elastic/eui';
-import React, { useCallback, useMemo, useEffect, useState } from 'react';
+import { EuiIcon, EuiLink, EuiText, EuiToolTip } from '@elastic/eui';
+import React, { FC, useCallback, useMemo, useEffect, useState } from 'react';
 import {
   BUILT_IN_MODEL_TAG,
   DEPLOYMENT_STATE,
@@ -33,6 +34,80 @@ import { isTestable, isDfaTrainedModel } from './test_models';
 import { ModelItem } from './models_list';
 import { usePermissionCheck } from '../capabilities/check_capabilities';
 
+const ViewTrainingDataAction: FC<{
+  item: ModelItem;
+}> = ({ item }) => {
+  const urlLocator = useMlLocator()!;
+  const {
+    services: {
+      application: { navigateToUrl },
+    },
+  } = useMlKibana();
+
+  const handleClick = async () => {
+    if (item.metadata?.analytics_config === undefined) return;
+
+    const analysisType = getAnalysisType(
+      item.metadata?.analytics_config.analysis
+    ) as DataFrameAnalysisConfigType;
+
+    const url = await urlLocator.getUrl({
+      page: ML_PAGES.DATA_FRAME_ANALYTICS_EXPLORATION,
+      pageState: {
+        jobId: item.metadata?.analytics_config.id as string,
+        analysisType,
+        ...(analysisType === 'classification' || analysisType === 'regression'
+          ? {
+              queryText: `${item.metadata?.analytics_config.dest.results_field}.is_training : true`,
+            }
+          : {}),
+      },
+    });
+
+    await navigateToUrl(url);
+  };
+
+  const buttonContent = (
+    <>
+      <EuiIcon type={'visTable'} css={{ marginRight: '8px' }} />
+      {i18n.translate('xpack.ml.trainedModels.modelsList.viewTrainingDataActionLabel', {
+        defaultMessage: 'View training data',
+      })}
+    </>
+  );
+
+  const isEnabled = item.origin_job_exists;
+  const button = isEnabled ? (
+    <EuiLink onClick={handleClick} color={'text'}>
+      <EuiText size="s" color={'text'}>
+        {buttonContent}
+      </EuiText>
+    </EuiLink>
+  ) : (
+    <EuiText size="s" color={'subdued'} css={{ fontWeight: 300 }}>
+      {buttonContent}
+    </EuiText>
+  );
+
+  if (isEnabled) {
+    return button;
+  } else {
+    return (
+      <EuiToolTip
+        position="top"
+        content={
+          <FormattedMessage
+            id="xpack.ml.trainedModels.modelsList.viewTrainingDataActionTooltip"
+            defaultMessage="Related data frame analytics job was not found."
+          />
+        }
+      >
+        {button}
+      </EuiToolTip>
+    );
+  }
+};
+
 export function useModelActions({
   onDfaTestAction,
   onTestAction,
@@ -54,7 +129,6 @@ export function useModelActions({
 }): Array<Action<ModelItem>> {
   const {
     services: {
-      application: { navigateToUrl },
       overlays,
       theme,
       i18n: i18nStart,
@@ -130,40 +204,11 @@ export function useModelActions({
   return useMemo(
     () => [
       {
-        name: i18n.translate('xpack.ml.trainedModels.modelsList.viewTrainingDataActionLabel', {
-          defaultMessage: 'View training data',
-        }),
-        description: i18n.translate(
-          'xpack.ml.trainedModels.modelsList.viewTrainingDataActionLabel',
-          {
-            defaultMessage: 'View training data',
-          }
-        ),
-        icon: 'visTable',
-        type: 'icon',
-        available: (item) => !!item.metadata?.analytics_config?.id,
-        onClick: async (item) => {
-          if (item.metadata?.analytics_config === undefined) return;
-
-          const analysisType = getAnalysisType(
-            item.metadata?.analytics_config.analysis
-          ) as DataFrameAnalysisConfigType;
-
-          const url = await urlLocator.getUrl({
-            page: ML_PAGES.DATA_FRAME_ANALYTICS_EXPLORATION,
-            pageState: {
-              jobId: item.metadata?.analytics_config.id as string,
-              analysisType,
-              ...(analysisType === 'classification' || analysisType === 'regression'
-                ? {
-                    queryText: `${item.metadata?.analytics_config.dest.results_field}.is_training : true`,
-                  }
-                : {}),
-            },
-          });
-
-          await navigateToUrl(url);
+        render: (item) => {
+          return <ViewTrainingDataAction item={item} />;
         },
+        available: (item) => !!item.metadata?.analytics_config?.id,
+        enabled: (item) => item.origin_job_exists === true,
         isPrimary: true,
       },
       {
@@ -606,7 +651,6 @@ export function useModelActions({
       isLoading,
       modelAndDeploymentIds,
       navigateToPath,
-      navigateToUrl,
       onDfaTestAction,
       onLoading,
       onModelDeployRequest,

x-pack/plugins/ml/public/application/model_management/models_list.tsx

@@ -79,6 +79,7 @@ export type ModelItem = TrainedModelConfigResponse & {
   type?: string[];
   stats?: Stats & { deployment_stats: TrainedModelDeploymentStatsResponse[] };
   pipelines?: ModelPipelines['pipelines'] | null;
+  origin_job_exists?: boolean;
   deployment_ids: string[];
   putModelConfig?: object;
   state: ModelState;

x-pack/plugins/ml/server/routes/trained_models.ts

@@ -190,11 +190,41 @@ export function trainedModelsRoutes(
             // we don't need to fill kibana's log with these messages.
             mlLog.debug(e);
           }
+          const enabledFeatures = getEnabledFeatures();
+          try {
+            if (enabledFeatures.dfa) {
+              const jobIds = result.map((model) => {
+                let id = model.metadata?.analytics_config?.id;
+                if (id) {
+                  id = `${id}*`;
+                }
+                return id;
+              });
+              const filteredJobIds = jobIds.filter((id) => id !== undefined);
 
-          const body = filterForEnabledFeatureModels(result, getEnabledFeatures());
+              const { data_frame_analytics: jobs } = await mlClient.getDataFrameAnalytics({
+                id: filteredJobIds.join(','),
+                allow_no_match: true,
+              });
+
+              jobs.forEach(({ id }) => {
+                const model = result.find(
+                  (modelWithJob) => id === modelWithJob.metadata?.analytics_config?.id
+                );
+
+                if (model) {
+                  model.origin_job_exists = true;
+                }
+              });
+            }
+          } catch (e) {
+            // Swallow error to prevent blocking trained models result
+          }
+
+          const filteredModels = filterForEnabledFeatureModels(result, enabledFeatures);
 
           return response.ok({
-            body,
+            body: filteredModels,
           });
         } catch (e) {
           return response.customError(wrapError(e));