@@ -54,7 +54,7 @@ final class IntegrationTests: XCTestCase {
54
54
55
55
vertex = VertexAI . vertexAI ( )
56
56
model = vertex. generativeModel (
57
- modelName: " gemini-1.5 -flash " ,
57
+ modelName: " gemini-2.0 -flash " ,
58
58
generationConfig: generationConfig,
59
59
safetySettings: safetySettings,
60
60
tools: [ ] ,
@@ -80,7 +80,7 @@ final class IntegrationTests: XCTestCase {
80
80
let app = try FirebaseApp . defaultNamedCopy ( name: TestAppCheckProviderFactory . notConfiguredName)
81
81
addTeardownBlock { await app. delete ( ) }
82
82
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 " )
84
84
let prompt = " Where is Google headquarters located? Answer with the city name only. "
85
85
86
86
do {
@@ -183,7 +183,7 @@ final class IntegrationTests: XCTestCase {
183
183
parameters: [ " x " : . integer( ) , " y " : . integer( ) ]
184
184
)
185
185
model = vertex. generativeModel (
186
- modelName: " gemini-1.5 -flash " ,
186
+ modelName: " gemini-2.0 -flash " ,
187
187
tools: [ . functionDeclarations( [ sumDeclaration] ) ] ,
188
188
toolConfig: . init( functionCallingConfig: . any( allowedFunctionNames: [ " sum " ] ) )
189
189
)
@@ -203,7 +203,7 @@ final class IntegrationTests: XCTestCase {
203
203
204
204
func testCountTokens_jsonSchema( ) async throws {
205
205
model = vertex. generativeModel (
206
- modelName: " gemini-1.5 -flash " ,
206
+ modelName: " gemini-2.0 -flash " ,
207
207
generationConfig: GenerationConfig (
208
208
responseMIMEType: " application/json " ,
209
209
responseSchema: Schema . object ( properties: [
@@ -226,7 +226,7 @@ final class IntegrationTests: XCTestCase {
226
226
let app = try FirebaseApp . defaultNamedCopy ( name: TestAppCheckProviderFactory . notConfiguredName)
227
227
addTeardownBlock { await app. delete ( ) }
228
228
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 " )
230
230
let prompt = " Why is the sky blue? "
231
231
232
232
do {
0 commit comments