Adding internal input types (elastic#132849) (elastic#133144)

jonathan-buttner · web-flow · commit be37851758fb · 2025-08-19T18:34:52.000+02:00
(cherry picked from commit 1f9b6bb)
diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/elastic/request/ElasticInferenceServiceSparseEmbeddingsRequest.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/elastic/request/ElasticInferenceServiceSparseEmbeddingsRequest.java
@@ -108,10 +108,10 @@ public boolean[] getTruncationInfo() {
     // visible for testing
     static ElasticInferenceServiceUsageContext inputTypeToUsageContext(InputType inputType) {
         switch (inputType) {
-            case SEARCH -> {
+            case SEARCH, INTERNAL_SEARCH -> {
                 return ElasticInferenceServiceUsageContext.SEARCH;
             }
-            case INGEST -> {
+            case INGEST, INTERNAL_INGEST -> {
                 return ElasticInferenceServiceUsageContext.INGEST;
             }
             default -> {

Original file line number	Diff line number	Diff line change
`@@ -108,10 +108,10 @@ public boolean[] getTruncationInfo() {`
`108`	`108`	`// visible for testing`
`109`	`109`	`static ElasticInferenceServiceUsageContext inputTypeToUsageContext(InputType inputType) {`
`110`	`110`	`switch (inputType) {`
`111`		`- case SEARCH -> {`
	`111`	`+ case SEARCH, INTERNAL_SEARCH -> {`
`112`	`112`	`return ElasticInferenceServiceUsageContext.SEARCH;`
`113`	`113`	`}`
`114`		`- case INGEST -> {`
	`114`	`+ case INGEST, INTERNAL_INGEST -> {`
`115`	`115`	`return ElasticInferenceServiceUsageContext.INGEST;`
`116`	`116`	`}`
`117`	`117`	`default -> {`