Skip to content

Commit 8326319

Browse files
committed
Remove "no text in response" and add comment for image size
1 parent 0ca33d4 commit 8326319

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

FirebaseVertexAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -138,10 +138,7 @@ struct GenerateContentIntegrationTests {
138138
generationConfig: generationConfig,
139139
safetySettings: safetySettings
140140
)
141-
let prompt = """
142-
Generate an image of a cute cartoon kitten playing with a ball of yarn. Do not respond with any
143-
text.
144-
"""
141+
let prompt = "Generate an image of a cute cartoon kitten playing with a ball of yarn."
145142

146143
let response = try await model.generateContent(prompt)
147144

@@ -152,6 +149,7 @@ struct GenerateContentIntegrationTests {
152149
#expect(inlineDataPart.data.count > 0)
153150
#if canImport(UIKit)
154151
let uiImage = try #require(UIImage(data: inlineDataPart.data))
152+
// Gemini 2.0 Flash Experimental returns 1024x1024 sized images.
155153
#expect(uiImage.size.width == 1024.0)
156154
#expect(uiImage.size.height == 1024.0)
157155
#endif // canImport(UIKit)

0 commit comments

Comments
 (0)