diff --git a/x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceGetServicesIT.java b/x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceGetServicesIT.java index 9729f26c4334c..21e933292a0ed 100644 --- a/x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceGetServicesIT.java +++ b/x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceGetServicesIT.java @@ -34,6 +34,7 @@ public void testGetServicesWithoutTaskType() throws IOException { allProviders(), containsInAnyOrder( List.of( + "ai21", "alibabacloud-ai-search", "amazonbedrock", "anthropic", @@ -47,6 +48,7 @@ public void testGetServicesWithoutTaskType() throws IOException { "googlevertexai", "hugging_face", "jinaai", + "llama", "mistral", "openai", "streaming_completion_test_service", @@ -91,6 +93,7 @@ public void testGetServicesWithTextEmbeddingTaskType() throws IOException { "googlevertexai", "hugging_face", "jinaai", + "llama", "mistral", "openai", "text_embedding_test_service", @@ -131,6 +134,8 @@ public void testGetServicesWithCompletionTaskType() throws IOException { providersFor(TaskType.COMPLETION), containsInAnyOrder( List.of( + "ai21", + "llama", "alibabacloud-ai-search", "amazonbedrock", "anthropic", @@ -157,6 +162,8 @@ public void testGetServicesWithChatCompletionTaskType() throws IOException { providersFor(TaskType.CHAT_COMPLETION), containsInAnyOrder( List.of( + "ai21", + "llama", "deepseek", "elastic", "openai", diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/ai21/Ai21Service.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/ai21/Ai21Service.java index 381774863049d..364a19b3d5f40 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/ai21/Ai21Service.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/ai21/Ai21Service.java @@ -227,12 +227,6 @@ public TransportVersion getMinimalSupportedVersion() { return TransportVersions.ML_INFERENCE_AI21_COMPLETION_ADDED; } - @Override - public boolean hideFromConfigurationApi() { - // The AI21 service is very configurable so we're going to hide it from being exposed in the service API. - return true; - } - @Override public Set supportedStreamingTasks() { return EnumSet.of(TaskType.COMPLETION, TaskType.CHAT_COMPLETION); diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/llama/LlamaService.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/llama/LlamaService.java index bd6b3c91fc9e9..8a84283f2747e 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/llama/LlamaService.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/llama/LlamaService.java @@ -366,12 +366,6 @@ public TransportVersion getMinimalSupportedVersion() { return TransportVersions.ML_INFERENCE_LLAMA_ADDED; } - @Override - public boolean hideFromConfigurationApi() { - // The Llama service is very configurable so we're going to hide it from being exposed in the service API. - return true; - } - /** * Configuration class for the Llama inference service. * It provides the settings and configurations required for the service.