Fixing tests

Author: jonathan-buttner

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/custom/CustomServiceSettingsTests.java

@@ -158,7 +158,8 @@ public void testFromMap() {
                     requestContentString,
                     responseParser,
                     new RateLimitSettings(10_000),
-                    11
+                    11,
+                    InputTypeTranslator.EMPTY_TRANSLATOR
                 )
             )
         );
@@ -587,7 +588,8 @@ public void testXContent() throws IOException {
                 },
                 "rate_limit": {
                     "requests_per_minute": 10000
-                }
+                },
+                "batch_size": 10
             }
             """);
 
@@ -603,6 +605,7 @@ public void testXContent_WithInputTypeTranslationValues() throws IOException {
             "string",
             new TextEmbeddingResponseParser("$.result.embeddings[*].embedding"),
             null,
+            null,
             new InputTypeTranslator(Map.of(InputType.SEARCH, "do_search", InputType.INGEST, "do_ingest"), "a_default")
         );
 
@@ -648,7 +651,8 @@ public void testXContent_BatchSize11() throws IOException {
             "string",
             new TextEmbeddingResponseParser("$.result.embeddings[*].embedding"),
             null,
-            11
+            11,
+            InputTypeTranslator.EMPTY_TRANSLATOR
         );
 
         XContentBuilder builder = XContentFactory.contentBuilder(XContentType.JSON);
@@ -667,6 +671,10 @@ public void testXContent_BatchSize11() throws IOException {
                         "text_embeddings": "$.result.embeddings[*].embedding"
                     }
                 },
+                "input_type": {
+                    "translation": {},
+                    "default": ""
+                },
                 "rate_limit": {
                     "requests_per_minute": 10000
                 },

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/custom/CustomServiceTests.java

@@ -276,7 +276,8 @@ private static CustomModel createInternalEmbeddingModel(
                 "{\"input\":${input}}",
                 parser,
                 new RateLimitSettings(10_000),
-                batchSize
+                batchSize,
+                InputTypeTranslator.EMPTY_TRANSLATOR
             ),
             new CustomTaskSettings(Map.of("key", "test_value")),
             new CustomSecretSettings(Map.of("test_key", new SecureString("test_value".toCharArray()))),

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/custom/request/CustomRequestTests.java

@@ -68,6 +68,7 @@ public void testCreateRequest() throws IOException {
             requestContentString,
             new TextEmbeddingResponseParser("$.result.embeddings"),
             new RateLimitSettings(10_000),
+            null,
             new InputTypeTranslator(Map.of(InputType.INGEST, "value"), "default")
         );
 
@@ -130,6 +131,7 @@ public void testCreateRequest_QueryParametersAreEscaped_AndEncoded() throws IOEx
             requestContentString,
             new TextEmbeddingResponseParser("$.result.embeddings"),
             new RateLimitSettings(10_000),
+            null,
             new InputTypeTranslator(Map.of(InputType.INGEST, "value"), "default")
         );