Skip to content

Commit 7e452a3

Browse files
Removing task type any
1 parent e9a9e2c commit 7e452a3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ public void testDefaultConfigs_Returns_DefaultChatCompletion_V1_WhenTaskTypeIsCo
9292

9393
try (var service = createElasticInferenceService()) {
9494
service.waitForAuthorizationToComplete(TIMEOUT);
95-
assertThat(service.supportedStreamingTasks(), is(EnumSet.of(TaskType.CHAT_COMPLETION, TaskType.ANY)));
95+
assertThat(service.supportedStreamingTasks(), is(EnumSet.of(TaskType.CHAT_COMPLETION)));
9696
assertThat(
9797
service.defaultConfigIds(),
9898
is(
@@ -126,7 +126,7 @@ public void testRemoves_DefaultChatCompletion_V1_WhenAuthorizationReturnsEmpty()
126126

127127
try (var service = createElasticInferenceService()) {
128128
service.waitForAuthorizationToComplete(TIMEOUT);
129-
assertThat(service.supportedStreamingTasks(), is(EnumSet.of(TaskType.CHAT_COMPLETION, TaskType.ANY)));
129+
assertThat(service.supportedStreamingTasks(), is(EnumSet.of(TaskType.CHAT_COMPLETION)));
130130
assertThat(
131131
service.defaultConfigIds(),
132132
is(
@@ -199,7 +199,7 @@ public void testRemoves_DefaultChatCompletion_V1_WhenAuthorizationDoesNotReturnA
199199

200200
try (var service = createElasticInferenceService()) {
201201
service.waitForAuthorizationToComplete(TIMEOUT);
202-
assertThat(service.supportedStreamingTasks(), is(EnumSet.of(TaskType.CHAT_COMPLETION, TaskType.ANY)));
202+
assertThat(service.supportedStreamingTasks(), is(EnumSet.of(TaskType.CHAT_COMPLETION)));
203203
assertThat(
204204
service.defaultConfigIds(),
205205
is(

0 commit comments

Comments
 (0)