updated NLP_CONFIG_KEYS to use common constant as source to match ser…

…ver code
elastic · Oct 26, 2022 · e0b5bdb · e0b5bdb
1 parent 55df7af
commit e0b5bdb
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/...rch/public/applications/enterprise_search_content/components/shared/ml_inference/utils.ts b/...rch/public/applications/enterprise_search_content/components/shared/ml_inference/utils.ts
@@ -8,14 +8,9 @@
 import { i18n } from '@kbn/i18n';
 import { TrainedModelConfigResponse } from '@kbn/ml-plugin/common/types/trained_models';
 
-export const NLP_CONFIG_KEYS = [
-  'fill_mask',
-  'ner',
-  'text_classification',
-  'text_embedding',
-  'question_answering',
-  'zero_shot_classification',
-];
+import { SUPPORTED_PYTORCH_TASKS } from '../../../../../../common/ml_inference_pipeline';
+
+export const NLP_CONFIG_KEYS = Object.values(SUPPORTED_PYTORCH_TASKS);
 export const RECOMMENDED_FIELDS = ['body', 'body_content', 'title'];
 
 export const NLP_DISPLAY_TITLES: Record<string, string | undefined> = {