Fix infer_trained_model API (#2786)

It can return list of lists of integers and floats.
elastic · Aug 21, 2024 · 0fc0098 · 0fc0098
1 parent 0f7334a
commit 0fc0098
Show file tree

Hide file tree

Showing 6 changed files with 76 additions and 6 deletions.
diff --git a/output/openapi/elasticsearch-openapi.json b/output/openapi/elasticsearch-openapi.json
diff --git a/output/openapi/elasticsearch-serverless-openapi.json b/output/openapi/elasticsearch-serverless-openapi.json
diff --git a/output/schema/schema-serverless.json b/output/schema/schema-serverless.json
diff --git a/output/schema/schema.json b/output/schema/schema.json
diff --git a/output/typescript/types.ts b/output/typescript/types.ts
diff --git a/specification/ml/_types/inference.ts b/specification/ml/_types/inference.ts
@@ -454,7 +454,13 @@ export class TrainedModelInferenceFeatureImportance {
   classes?: TrainedModelInferenceClassImportance[]
 }
 
-export type PredictedValue = string | double | boolean | integer
+export type PredictedValue =
+  | string
+  | double
+  | double[]
+  | boolean
+  | integer
+  | integer[]
 
 export class InferenceResponseResult {
   /**