fix: [vertexai] make setApiEndpoint effective for LlmUtilityServiceCl…

…ient. (#10385) PiperOrigin-RevId: 607113154 Co-authored-by: Zhenyi Qi <zhenyiqi@google.com>
googleapis · Feb 15, 2024 · b4ec244 · b4ec244
1 parent 651ba65
commit b4ec244
Showing 1 changed file with 2 additions and 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
@@ -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);
       }