Set up the config -> form -> schema -> template transformations for i…

…nput_map Signed-off-by: Tyler Ohlsen <[email protected]>
opensearch-project · Nov 20, 2024 · 9b8fff8 · 9b8fff8
1 parent c2f05e4
commit 9b8fff8
Show file tree

Hide file tree

Showing 8 changed files with 357 additions and 55 deletions.
diff --git a/common/constants.ts b/common/constants.ts
@@ -506,3 +506,9 @@ export enum SOURCE_OPTIONS {
   UPLOAD = 'upload',
   EXISTING_INDEX = 'existing_index',
 }
+export enum TRANSFORM_TYPE {
+  STRING = 'string',
+  FIELD = 'field',
+  EXPRESSION = 'expression',
+  TEMPLATE = 'template',
+}
diff --git a/common/interfaces.ts b/common/interfaces.ts
@@ -10,6 +10,7 @@ import {
   COMPONENT_CLASS,
   PROCESSOR_TYPE,
   PROMPT_FIELD,
+  TRANSFORM_TYPE,
   WORKFLOW_TYPE,
 } from './constants';
 
@@ -35,9 +36,21 @@ export type ConfigFieldType =
   | 'map'
   | 'mapArray'
   | 'boolean'
-  | 'number';
+  | 'number'
+  | 'transform'
+  | 'transformArray';
 
-export type ConfigFieldValue = string | {};
+export type ConfigFieldTransformValue = {
+  transformType: TRANSFORM_TYPE;
+  value: string;
+};
+export type ConfigFieldTransformArrayValue = ConfigFieldTransformValue[];
+
+export type ConfigFieldValue =
+  | string
+  | ConfigFieldTransformValue
+  | ConfigFieldTransformArrayValue
+  | {};
 
 export interface IConfigField {
   type: ConfigFieldType;
@@ -100,6 +113,15 @@ export type MapFormValue = MapEntry[];
 
 export type MapArrayFormValue = MapFormValue[];
 
+export type InputMapEntry = {
+  key: string;
+  value: ConfigFieldTransformValue;
+};
+
+export type InputMapFormValue = InputMapEntry[];
+
+export type InputMapArrayFormValue = InputMapFormValue[];
+
 export type WorkflowFormValues = {
   ingest: FormikValues;
   search: FormikValues;

diff --git a/public/configs/ml_processor.ts b/public/configs/ml_processor.ts
@@ -22,7 +22,7 @@ export abstract class MLProcessor extends Processor {
       },
       {
         id: 'input_map',
-        type: 'mapArray',
+        type: 'transformArray',
       },
       {
         id: 'output_map',

diff --git a/...kflow_detail/workflow_inputs/processor_inputs/ml_processor_inputs/ml_processor_inputs.tsx b/...kflow_detail/workflow_inputs/processor_inputs/ml_processor_inputs/ml_processor_inputs.tsx
@@ -33,6 +33,12 @@ import {
   MapFormValue,
 } from '../../../../../../common';
 import { ModelField } from '../../input_fields';
+import {
+  TRANSFORM_TYPE,
+  InputMapEntry,
+  InputMapFormValue,
+  InputMapArrayFormValue,
+} from '../../../../../../common';
 import {
   ConfigurePromptModal,
   InputTransformModal,
@@ -139,10 +145,13 @@ export function MLProcessorInputs(props: MLProcessorInputsProps) {
       parseModelInputs(newModelInterface).map((modelInput) => {
         return {
           key: modelInput.label,
-          value: '',
-        } as MapEntry;
-      }) as MapFormValue,
-    ] as MapArrayFormValue;
+          value: {
+            transformType: TRANSFORM_TYPE.FIELD,
+            value: '',
+          },
+        } as InputMapEntry;
+      }) as InputMapFormValue,
+    ] as InputMapArrayFormValue;
     const modelOutputsAsForm = [
       parseModelOutputs(newModelInterface).map((modelOutput) => {
         return {