Skip to content

Commit d6042e7

Browse files
committed
Correcting QA tests
1 parent 850ab5d commit d6042e7

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

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

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ public class InferenceGetServicesIT extends BaseMockEISAuthServerTest {
2525
@SuppressWarnings("unchecked")
2626
public void testGetServicesWithoutTaskType() throws IOException {
2727
List<Object> services = getAllServices();
28-
assertThat(services.size(), equalTo(19));
28+
assertThat(services.size(), equalTo(20));
2929

3030
String[] providers = new String[services.size()];
3131
for (int i = 0; i < services.size(); i++) {
@@ -53,6 +53,7 @@ public void testGetServicesWithoutTaskType() throws IOException {
5353
"test_reranking_service",
5454
"test_service",
5555
"text_embedding_test_service",
56+
"voyageai",
5657
"watsonxai"
5758
).toArray(),
5859
providers
@@ -62,7 +63,7 @@ public void testGetServicesWithoutTaskType() throws IOException {
6263
@SuppressWarnings("unchecked")
6364
public void testGetServicesWithTextEmbeddingTaskType() throws IOException {
6465
List<Object> services = getServices(TaskType.TEXT_EMBEDDING);
65-
assertThat(services.size(), equalTo(14));
66+
assertThat(services.size(), equalTo(15));
6667

6768
String[] providers = new String[services.size()];
6869
for (int i = 0; i < services.size(); i++) {
@@ -85,6 +86,7 @@ public void testGetServicesWithTextEmbeddingTaskType() throws IOException {
8586
"mistral",
8687
"openai",
8788
"text_embedding_test_service",
89+
"voyageai",
8890
"watsonxai"
8991
).toArray(),
9092
providers
@@ -94,7 +96,7 @@ public void testGetServicesWithTextEmbeddingTaskType() throws IOException {
9496
@SuppressWarnings("unchecked")
9597
public void testGetServicesWithRerankTaskType() throws IOException {
9698
List<Object> services = getServices(TaskType.RERANK);
97-
assertThat(services.size(), equalTo(6));
99+
assertThat(services.size(), equalTo(7));
98100

99101
String[] providers = new String[services.size()];
100102
for (int i = 0; i < services.size(); i++) {
@@ -103,7 +105,15 @@ public void testGetServicesWithRerankTaskType() throws IOException {
103105
}
104106

105107
assertArrayEquals(
106-
List.of("alibabacloud-ai-search", "cohere", "elasticsearch", "googlevertexai", "jinaai", "test_reranking_service").toArray(),
108+
List.of(
109+
"alibabacloud-ai-search",
110+
"cohere",
111+
"elasticsearch",
112+
"googlevertexai",
113+
"jinaai",
114+
"test_reranking_service",
115+
"voyageai"
116+
).toArray(),
107117
providers
108118
);
109119
}

0 commit comments

Comments
 (0)