From 64bce1f83417320bf5f27fb598baaf9b92f55308 Mon Sep 17 00:00:00 2001 From: Zhenyi Qi Date: Wed, 14 Feb 2024 14:49:53 -0800 Subject: [PATCH] fix: [vertexai] make setApiEndpoint effective for LlmUtilityServiceClient. PiperOrigin-RevId: 607113154 --- .../src/main/java/com/google/cloud/vertexai/VertexAI.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/java-vertexai/google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/VertexAI.java b/java-vertexai/google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/VertexAI.java index 9fdf6f2b7877..3cdbed5ed7b0 100644 --- a/java-vertexai/google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/VertexAI.java +++ b/java-vertexai/google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/VertexAI.java @@ -336,7 +336,7 @@ public PredictionServiceClient getPredictionServiceRestClient() throws IOExcepti public LlmUtilityServiceClient getLlmUtilityClient() throws IOException { if (llmUtilityClient == null) { LlmUtilityServiceSettings.Builder settingsBuilder = LlmUtilityServiceSettings.newBuilder(); - settingsBuilder.setEndpoint(String.format("%s-aiplatform.googleapis.com:443", this.location)); + settingsBuilder.setEndpoint(String.format("%s:443", this.apiEndpoint)); if (this.credentialsProvider != null) { settingsBuilder.setCredentialsProvider(this.credentialsProvider); } @@ -370,7 +370,7 @@ public LlmUtilityServiceClient getLlmUtilityRestClient() throws IOException { if (llmUtilityRestClient == null) { LlmUtilityServiceSettings.Builder settingsBuilder = LlmUtilityServiceSettings.newHttpJsonBuilder(); - settingsBuilder.setEndpoint(String.format("%s-aiplatform.googleapis.com:443", this.location)); + settingsBuilder.setEndpoint(String.format("%s:443", this.apiEndpoint)); if (this.credentialsProvider != null) { settingsBuilder.setCredentialsProvider(this.credentialsProvider); }