[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

@@ -130,18 +130,19 @@ export function useModelActions({
   return useMemo(
     () => [
       {
-        name: i18n.translate('xpack.ml.trainedModels.modelsList.viewTrainingDataActionLabel', {
+        name: i18n.translate('xpack.ml.trainedModels.modelsList.viewTrainingDataNameActionLabel', {
           defaultMessage: 'View training data',
         }),
         description: i18n.translate(
           'xpack.ml.trainedModels.modelsList.viewTrainingDataActionLabel',
           {
-            defaultMessage: 'View training data',
+            defaultMessage: 'Training data can be viewed when data frame analytics job exists.',
           }
         ),
         icon: 'visTable',
         type: 'icon',
         available: (item) => !!item.metadata?.analytics_config?.id,
+        enabled: (item) => item.origin_job_exists === true,
         onClick: async (item) => {
           if (item.metadata?.analytics_config === undefined) return;
 
@@ -164,7 +165,6 @@ export function useModelActions({
 
           await navigateToUrl(url);
         },
-        isPrimary: true,
       },
       {
         name: i18n.translate('xpack.ml.inference.modelsList.analyticsMapActionLabel', {

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

@@ -29,20 +29,19 @@ import {
   createIngestPipelineSchema,
   modelDownloadsQuery,
 } from './schemas/inference_schema';
-import type {
+import {
   PipelineDefinition,
-  TrainedModelConfigResponse,
+  type TrainedModelConfigResponse,
 } from '../../common/types/trained_models';
 import { mlLog } from '../lib/log';
 import { forceQuerySchema } from './schemas/anomaly_detectors_schema';
 import { modelsProvider } from '../models/model_management';
 
 export const DEFAULT_TRAINED_MODELS_PAGE_SIZE = 10000;
 
-export function filterForEnabledFeatureModels(
-  models: TrainedModelConfigResponse[] | estypes.MlTrainedModelConfig[],
-  enabledFeatures: MlFeatures
-) {
+export function filterForEnabledFeatureModels<
+  T extends TrainedModelConfigResponse | estypes.MlTrainedModelConfig
+>(models: T[], enabledFeatures: MlFeatures) {
   let filteredModels = models;
   if (enabledFeatures.nlp === false) {
     filteredModels = filteredModels.filter((m) => m.model_type === 'tree_ensemble');
@@ -191,10 +190,38 @@ export function trainedModelsRoutes(
             mlLog.debug(e);
           }
 
-          const body = filterForEnabledFeatureModels(result, getEnabledFeatures());
+          const filteredModels = filterForEnabledFeatureModels(result, getEnabledFeatures());
+
+          try {
+            const jobIds = filteredModels
+              .map((model) => {
+                const id = model.metadata?.analytics_config?.id;
+                if (id) {
+                  return `${id}*`;
+                }
+              })
+              .filter((id) => id !== undefined);
+
+            if (jobIds.length) {
+              const { data_frame_analytics: jobs } = await mlClient.getDataFrameAnalytics({
+                id: jobIds.join(','),
+                allow_no_match: true,
+              });
+
+              filteredModels.forEach((model) => {
+                const dfaId = model?.metadata?.analytics_config?.id;
+                if (dfaId !== undefined) {
+                  // if this is a dfa model, set origin_job_exists
+                  model.origin_job_exists = jobs.find((job) => job.id === dfaId) !== undefined;
+                }
+              });
+            }
+          } catch (e) {
+            // Swallow error to prevent blocking trained models result
+          }
 
           return response.ok({
-            body,
+            body: filteredModels,
           });
         } catch (e) {
           return response.customError(wrapError(e));