File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
FirebaseVertexAI/Tests/TestApp/Tests/Integration Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff 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)
You can’t perform that action at this time.
0 commit comments