Add prefix_string config option for ML models (#2363)

The prefix_strings option was added to support the E5 model in elastic/elasticsearch#102089
elastic · Jan 2, 2024 · 05c39a9 · 05c39a9
1 parent 0270837
commit 05c39a9
Show file tree

Hide file tree

Showing 4 changed files with 102 additions and 19 deletions.
diff --git a/output/openapi/elasticsearch-serverless-openapi.json b/output/openapi/elasticsearch-serverless-openapi.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/TrainedModel.ts b/specification/ml/_types/TrainedModel.ts
@@ -196,6 +196,7 @@ export class TrainedModelConfig {
   metadata?: TrainedModelConfigMetadata
   model_size_bytes?: ByteSize
   location?: TrainedModelLocation
+  prefix_strings?: TrainedModelPrefixStrings
 }
 
 export class TrainedModelConfigInput {
@@ -423,3 +424,14 @@ export class TrainedModelLocation {
 export class TrainedModelLocationIndex {
   name: IndexName
 }
+
+export class TrainedModelPrefixStrings {
+  /**
+   * String prepended to input at ingest
+   */
+  ingest: string
+  /**
+   * String prepended to input at search
+   */
+  search: string
+}