Adding internal input types (#132849)

jonathan-buttner · jonathan-buttner · commit d8358b3e76e0 · 2025-08-19T11:33:18.000-04:00
(cherry picked from commit 1f9b6bb)
diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/external/request/elastic/ElasticInferenceServiceSparseEmbeddingsRequest.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/external/request/elastic/ElasticInferenceServiceSparseEmbeddingsRequest.java
@@ -105,10 +105,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
`@@ -105,10 +105,10 @@ public boolean[] getTruncationInfo() {`
`105`	`105`	`// visible for testing`
`106`	`106`	`static ElasticInferenceServiceUsageContext inputTypeToUsageContext(InputType inputType) {`
`107`	`107`	`switch (inputType) {`
`108`		`- case SEARCH -> {`
	`108`	`+ case SEARCH, INTERNAL_SEARCH -> {`
`109`	`109`	`return ElasticInferenceServiceUsageContext.SEARCH;`
`110`	`110`	`}`
`111`		`- case INGEST -> {`
	`111`	`+ case INGEST, INTERNAL_INGEST -> {`
`112`	`112`	`return ElasticInferenceServiceUsageContext.INGEST;`
`113`	`113`	`}`
`114`	`114`	`default -> {`