Browse Source

Adding internal input types (#132849) (#133142)

(cherry picked from commit 1f9b6bb938a1bbe261e035a0dfab1e1ddb8661e0)
Jonathan Buttner 1 month ago
parent
commit
a78a92a46e

+ 2 - 2
x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/elastic/request/ElasticInferenceServiceSparseEmbeddingsRequest.java

@@ -108,10 +108,10 @@ public class ElasticInferenceServiceSparseEmbeddingsRequest extends ElasticInfer
     // visible for testing
     // visible for testing
     static ElasticInferenceServiceUsageContext inputTypeToUsageContext(InputType inputType) {
     static ElasticInferenceServiceUsageContext inputTypeToUsageContext(InputType inputType) {
         switch (inputType) {
         switch (inputType) {
-            case SEARCH -> {
+            case SEARCH, INTERNAL_SEARCH -> {
                 return ElasticInferenceServiceUsageContext.SEARCH;
                 return ElasticInferenceServiceUsageContext.SEARCH;
             }
             }
-            case INGEST -> {
+            case INGEST, INTERNAL_INGEST -> {
                 return ElasticInferenceServiceUsageContext.INGEST;
                 return ElasticInferenceServiceUsageContext.INGEST;
             }
             }
             default -> {
             default -> {