Skip to content

Commit

Permalink
chore: [vertexai] Update one integration test to use tuned model (goo…
Browse files Browse the repository at this point in the history
…gleapis#10871)

PiperOrigin-RevId: 647405766

Co-authored-by: Zhenyi Qi <zhenyiqi@google.com>
  • Loading branch information
copybara-service[bot] and Zhenyi Qi committed Jun 27, 2024
1 parent 549ee9a commit 0bb1622
Showing 1 changed file with 6 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down

0 comments on commit 0bb1622

Please sign in to comment.