File tree Expand file tree Collapse file tree 2 files changed +4
-5
lines changed
src/test/java/com/google/firebase/vertexai Expand file tree Collapse file tree 2 files changed +4
-5
lines changed Original file line number Diff line number Diff line change @@ -62,10 +62,9 @@ internal class VertexAIStreamingSnapshotTests {
62
62
withTimeout(testTimeout) {
63
63
val responseList = responses.toList()
64
64
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
69
68
}
70
69
}
71
70
}
Original file line number Diff line number Diff line change 17
17
# This script replaces mock response files for Vertex AI unit tests with a fresh
18
18
# clone of the shared repository of Vertex AI test data.
19
19
20
- RESPONSES_VERSION=' v7 .*' # The major version of mock responses to use
20
+ RESPONSES_VERSION=' v13 .*' # The major version of mock responses to use
21
21
REPO_NAME=" vertexai-sdk-test-data"
22
22
REPO_LINK=" https://github.com/FirebaseExtended/$REPO_NAME .git"
23
23
You can’t perform that action at this time.
0 commit comments