Skip to content

Commit 4162383

Browse files
committed
Update vertex golden files
1 parent 89e515d commit 4162383

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

firebase-vertexai/src/test/java/com/google/firebase/vertexai/VertexAIStreamingSnapshotTests.kt

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,10 +62,9 @@ internal class VertexAIStreamingSnapshotTests {
6262
withTimeout(testTimeout) {
6363
val responseList = responses.toList()
6464
responseList.isEmpty() shouldBe false
65-
responseList.forEach {
66-
it.candidates.first().finishReason shouldBe FinishReason.STOP
67-
it.candidates.first().content.parts.isEmpty() shouldBe false
68-
it.candidates.first().safetyRatings.isEmpty() shouldBe false
65+
responseList.last().candidates.first().apply {
66+
finishReason shouldBe FinishReason.STOP
67+
content.parts.isEmpty() shouldBe false
6968
}
7069
}
7170
}

firebase-vertexai/update_responses.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
# This script replaces mock response files for Vertex AI unit tests with a fresh
1818
# clone of the shared repository of Vertex AI test data.
1919

20-
RESPONSES_VERSION='v7.*' # The major version of mock responses to use
20+
RESPONSES_VERSION='v13.*' # The major version of mock responses to use
2121
REPO_NAME="vertexai-sdk-test-data"
2222
REPO_LINK="https://github.com/FirebaseExtended/$REPO_NAME.git"
2323

0 commit comments

Comments
 (0)