@@ -54,7 +54,7 @@ final class IntegrationTests: XCTestCase {
5454
5555 vertex = VertexAI . vertexAI ( )
5656 model = vertex. generativeModel (
57- modelName: " gemini-1.5 -flash " ,
57+ modelName: " gemini-2.0 -flash " ,
5858 generationConfig: generationConfig,
5959 safetySettings: safetySettings,
6060 tools: [ ] ,
@@ -80,7 +80,7 @@ final class IntegrationTests: XCTestCase {
8080 let app = try FirebaseApp . defaultNamedCopy ( name: TestAppCheckProviderFactory . notConfiguredName)
8181 addTeardownBlock { await app. delete ( ) }
8282 let vertex = VertexAI . vertexAI ( app: app)
83- let model = vertex. generativeModel ( modelName: " gemini-1.5 -flash " )
83+ let model = vertex. generativeModel ( modelName: " gemini-2.0 -flash " )
8484 let prompt = " Where is Google headquarters located? Answer with the city name only. "
8585
8686 do {
@@ -183,7 +183,7 @@ final class IntegrationTests: XCTestCase {
183183 parameters: [ " x " : . integer( ) , " y " : . integer( ) ]
184184 )
185185 model = vertex. generativeModel (
186- modelName: " gemini-1.5 -flash " ,
186+ modelName: " gemini-2.0 -flash " ,
187187 tools: [ . functionDeclarations( [ sumDeclaration] ) ] ,
188188 toolConfig: . init( functionCallingConfig: . any( allowedFunctionNames: [ " sum " ] ) )
189189 )
@@ -203,7 +203,7 @@ final class IntegrationTests: XCTestCase {
203203
204204 func testCountTokens_jsonSchema( ) async throws {
205205 model = vertex. generativeModel (
206- modelName: " gemini-1.5 -flash " ,
206+ modelName: " gemini-2.0 -flash " ,
207207 generationConfig: GenerationConfig (
208208 responseMIMEType: " application/json " ,
209209 responseSchema: Schema . object ( properties: [
@@ -226,7 +226,7 @@ final class IntegrationTests: XCTestCase {
226226 let app = try FirebaseApp . defaultNamedCopy ( name: TestAppCheckProviderFactory . notConfiguredName)
227227 addTeardownBlock { await app. delete ( ) }
228228 let vertex = VertexAI . vertexAI ( app: app)
229- let model = vertex. generativeModel ( modelName: " gemini-1.5 -flash " )
229+ let model = vertex. generativeModel ( modelName: " gemini-2.0 -flash " )
230230 let prompt = " Why is the sky blue? "
231231
232232 do {
0 commit comments