Skip to content

Commit 9c113c6

Browse files
author
David Motsonashvili
committed
re-added truncated test
1 parent 35fa173 commit 9c113c6

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

firebase-ai/src/test/java/com/google/firebase/ai/VertexAIUnarySnapshotTests.kt

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -591,6 +591,15 @@ internal class VertexAIUnarySnapshotTests {
591591
}
592592
}
593593

594+
@Test
595+
fun `generateImages should contain safety data`() =
596+
goldenVertexUnaryFile("unary-success-generate-images-safety_info.json") {
597+
withTimeout(testTimeout) {
598+
val response = imagenModel.generateImages("prompt")
599+
// There is no public API, but if it parses then success
600+
}
601+
}
602+
594603
@Test
595604
fun `google search grounding metadata is parsed correctly`() =
596605
goldenVertexUnaryFile("unary-success-google-search-grounding.json") {

0 commit comments

Comments
 (0)