Skip to content

Commit 1d6b589

Browse files
committed
Add completion_test_service to InferenceGetServicesIT
1 parent c70b0a1 commit 1d6b589

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceGetServicesIT.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ public class InferenceGetServicesIT extends BaseMockEISAuthServerTest {
2525

2626
public void testGetServicesWithoutTaskType() throws IOException {
2727
List<Object> services = getAllServices();
28-
assertThat(services.size(), equalTo(22));
28+
assertThat(services.size(), equalTo(23));
2929

3030
var providers = providers(services);
3131

@@ -49,6 +49,7 @@ public void testGetServicesWithoutTaskType() throws IOException {
4949
"mistral",
5050
"openai",
5151
"streaming_completion_test_service",
52+
"completion_test_service",
5253
"test_reranking_service",
5354
"test_service",
5455
"text_embedding_test_service",
@@ -123,7 +124,7 @@ public void testGetServicesWithRerankTaskType() throws IOException {
123124

124125
public void testGetServicesWithCompletionTaskType() throws IOException {
125126
List<Object> services = getServices(TaskType.COMPLETION);
126-
assertThat(services.size(), equalTo(11));
127+
assertThat(services.size(), equalTo(12));
127128

128129
var providers = providers(services);
129130

@@ -141,6 +142,7 @@ public void testGetServicesWithCompletionTaskType() throws IOException {
141142
"googleaistudio",
142143
"openai",
143144
"streaming_completion_test_service",
145+
"completion_test_service",
144146
"hugging_face"
145147
).toArray()
146148
)

0 commit comments

Comments
 (0)