Skip to content

Commit e8e3577

Browse files
Fixing gp llm v2 name
1 parent 6656baf commit e8e3577

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
import static org.elasticsearch.xpack.inference.InferenceBaseRestTest.getAllModels;
2020
import static org.elasticsearch.xpack.inference.InferenceBaseRestTest.getModels;
2121
import static org.elasticsearch.xpack.inference.services.elastic.response.AuthorizationResponseEntityTests.ELSER_V2_ENDPOINT_ID;
22-
import static org.elasticsearch.xpack.inference.services.elastic.response.AuthorizationResponseEntityTests.GP_LLM_V1_CHAT_COMPLETION_ENDPOINT_ID;
22+
import static org.elasticsearch.xpack.inference.services.elastic.response.AuthorizationResponseEntityTests.GP_LLM_V2_CHAT_COMPLETION_ENDPOINT_ID;
2323
import static org.elasticsearch.xpack.inference.services.elastic.response.AuthorizationResponseEntityTests.JINA_EMBED_V3_ENDPOINT_ID;
2424
import static org.elasticsearch.xpack.inference.services.elastic.response.AuthorizationResponseEntityTests.RAINBOW_SPRINKLES_ENDPOINT_ID;
2525
import static org.elasticsearch.xpack.inference.services.elastic.response.AuthorizationResponseEntityTests.RERANK_V1_ENDPOINT_ID;
@@ -55,7 +55,7 @@ public void testGetDefaultEndpoints() throws IOException {
5555
}
5656

5757
assertInferenceIdTaskType(allModels, RAINBOW_SPRINKLES_ENDPOINT_ID, TaskType.CHAT_COMPLETION);
58-
assertInferenceIdTaskType(allModels, GP_LLM_V1_CHAT_COMPLETION_ENDPOINT_ID, TaskType.CHAT_COMPLETION);
58+
assertInferenceIdTaskType(allModels, GP_LLM_V2_CHAT_COMPLETION_ENDPOINT_ID, TaskType.CHAT_COMPLETION);
5959
assertInferenceIdTaskType(allModels, ELSER_V2_ENDPOINT_ID, TaskType.SPARSE_EMBEDDING);
6060
assertInferenceIdTaskType(allModels, JINA_EMBED_V3_ENDPOINT_ID, TaskType.TEXT_EMBEDDING);
6161
assertInferenceIdTaskType(allModels, RERANK_V1_ENDPOINT_ID, TaskType.RERANK);

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/elastic/response/AuthorizationResponseEntityTests.java

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -50,8 +50,8 @@ public class AuthorizationResponseEntityTests extends AbstractBWCWireSerializati
5050
public static final String EIS_CHAT_PATH = "chat";
5151

5252
// gp-llm-v2
53-
public static final String GP_LLM_V1_CHAT_COMPLETION_ENDPOINT_ID = ".gp-llm-v2-chat_completion";
54-
public static final String GP_LLM_V1_MODEL_NAME = "gp-llm-v2";
53+
public static final String GP_LLM_V2_CHAT_COMPLETION_ENDPOINT_ID = ".gp-llm-v2-chat_completion";
54+
public static final String GP_LLM_V2_MODEL_NAME = "gp-llm-v2";
5555

5656
// elser-2
5757
public static final String ELSER_V2_ENDPOINT_ID = ".elser-2-elastic";
@@ -354,10 +354,10 @@ private static AuthorizationResponseEntity.AuthorizedEndpoint createRainbowSprin
354354

355355
private static ElasticInferenceServiceModel createGpLlmV2ExpectedEndpoint(String url) {
356356
return new ElasticInferenceServiceCompletionModel(
357-
GP_LLM_V1_CHAT_COMPLETION_ENDPOINT_ID,
357+
GP_LLM_V2_CHAT_COMPLETION_ENDPOINT_ID,
358358
TaskType.CHAT_COMPLETION,
359359
ElasticInferenceService.NAME,
360-
new ElasticInferenceServiceCompletionServiceSettings(GP_LLM_V1_MODEL_NAME),
360+
new ElasticInferenceServiceCompletionServiceSettings(GP_LLM_V2_MODEL_NAME),
361361
EmptyTaskSettings.INSTANCE,
362362
EmptySecretSettings.INSTANCE,
363363
new ElasticInferenceServiceComponents(url)
@@ -366,8 +366,8 @@ private static ElasticInferenceServiceModel createGpLlmV2ExpectedEndpoint(String
366366

367367
private static AuthorizationResponseEntity.AuthorizedEndpoint createGpLlmV2AuthorizedEndpoint() {
368368
return new AuthorizationResponseEntity.AuthorizedEndpoint(
369-
GP_LLM_V1_CHAT_COMPLETION_ENDPOINT_ID,
370-
GP_LLM_V1_MODEL_NAME,
369+
GP_LLM_V2_CHAT_COMPLETION_ENDPOINT_ID,
370+
GP_LLM_V2_MODEL_NAME,
371371
createTaskTypeObject(EIS_CHAT_PATH, "chat_completion"),
372372
"ga",
373373
List.of("multilingual"),

0 commit comments

Comments
 (0)