Fix tests

Author: davidkyle

x-pack/plugin/inference/qa/test-service-plugin/src/main/java/org/elasticsearch/xpack/inference/mock/TestRerankingServiceExtension.java

@@ -49,6 +49,8 @@
 
 public class TestRerankingServiceExtension implements InferenceServiceExtension {
 
+    public static final int RERANK_WINDOW_SIZE = 333;
+
     @Override
     public List<Factory> getInferenceServiceFactories() {
         return List.of(TestInferenceService::new);
@@ -203,7 +205,7 @@ protected ServiceSettings getServiceSettingsFromMap(Map<String, Object> serviceS
 
         @Override
         public int rerankerWindowSize(String modelId) {
-            return 333;
+            return RERANK_WINDOW_SIZE;
         }
 
         public static class Configuration {

x-pack/plugin/inference/src/internalClusterTest/java/org/elasticsearch/xpack/inference/integration/RerankWindowSizeIT.java

@@ -15,6 +15,7 @@
 import org.elasticsearch.xpack.inference.LocalStateInferencePlugin;
 import org.elasticsearch.xpack.inference.Utils;
 import org.elasticsearch.xpack.inference.mock.TestInferenceServicePlugin;
+import org.elasticsearch.xpack.inference.mock.TestRerankingServiceExtension;
 import org.elasticsearch.xpack.inference.registry.ModelRegistry;
 import org.junit.Before;
 
@@ -41,7 +42,7 @@ protected Collection<Class<? extends Plugin>> nodePlugins() {
     public void testRerankWindowSizeAction() {
         var response = client().execute(GetRerankerWindowSizeAction.INSTANCE, new GetRerankerWindowSizeAction.Request("rerank-endpoint"))
             .actionGet();
-        assertEquals(333, response.getWindowSize());
+        assertEquals(TestRerankingServiceExtension.RERANK_WINDOW_SIZE, response.getWindowSize());
     }
 
     public void testActionNotAReranker() {

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/ai21/Ai21ServiceTests.java

@@ -19,6 +19,7 @@
 import org.elasticsearch.common.xcontent.XContentHelper;
 import org.elasticsearch.core.TimeValue;
 import org.elasticsearch.inference.EmptyTaskSettings;
+import org.elasticsearch.inference.InferenceService;
 import org.elasticsearch.inference.InferenceServiceConfiguration;
 import org.elasticsearch.inference.InferenceServiceResults;
 import org.elasticsearch.inference.InputType;
@@ -562,4 +563,8 @@ private Map<String, Object> getRequestConfigMap(Map<String, Object> serviceSetti
         return new HashMap<>(Map.of(ModelConfigurations.SERVICE_SETTINGS, builtServiceSettings));
     }
 
+    @Override
+    public InferenceService createInferenceService() {
+        return createService();
+    }
 }