chore: [vertexai] Update one integration test to use tuned model (#10871

) PiperOrigin-RevId: 647405766 Co-authored-by: Zhenyi Qi <zhenyiqi@google.com>
googleapis · Jun 27, 2024 · 0bb1622 · 0bb1622
1 parent 549ee9a
commit 0bb1622
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/...vertexai/src/test/java/com/google/cloud/vertexai/it/ITGenerativeModelIntegrationTest.java b/...vertexai/src/test/java/com/google/cloud/vertexai/it/ITGenerativeModelIntegrationTest.java
@@ -129,9 +129,13 @@ private static void assertNonEmptyAndLogTextContentOfResponseStream(
   }
 
   @Test
-  public void generateContent_restTransport_nonEmptyCandidateList() throws IOException {
+  public void generateContent_restTransportWithTunedModel_nonEmptyCandidateList()
+      throws IOException {
     try (VertexAI vertexAiViaRest = new VertexAI.Builder().setTransport(Transport.REST).build()) {
-      GenerativeModel textModelWithRest = new GenerativeModel(MODEL_NAME_TEXT, vertexAiViaRest);
+      GenerativeModel textModelWithRest =
+          new GenerativeModel(
+              "projects/964831358985/locations/us-central1/endpoints/7226683110069370880",
+              vertexAiViaRest);
       GenerateContentResponse response = textModelWithRest.generateContent(TEXT);
 
       assertNonEmptyAndLogResponse(name.getMethodName(), TEXT, response);