@@ -33,17 +33,25 @@ final class GoogleGenerativeAITests: XCTestCase {
3333 let systemInstruction = ModelContent ( role: " system " , parts: [ . text( " Talk like a pirate. " ) ] )
3434
3535 // Permutations without optional arguments.
36- let _ = GenerativeModel ( name: " gemini-1.0-pro " , apiKey: " API_KEY " )
37- let _ = GenerativeModel ( name: " gemini-1.0-pro " , apiKey: " API_KEY " , safetySettings: filters)
38- let _ = GenerativeModel ( name: " gemini-1.0-pro " , apiKey: " API_KEY " , generationConfig: config)
36+ let _ = GenerativeModel ( name: " gemini-1.5-pro-latest " , apiKey: " API_KEY " )
3937 let _ = GenerativeModel (
40- name: " gemini-1.0-pro " ,
38+ name: " gemini-1.5-pro-latest " ,
39+ apiKey: " API_KEY " ,
40+ safetySettings: filters
41+ )
42+ let _ = GenerativeModel (
43+ name: " gemini-1.5-pro-latest " ,
44+ apiKey: " API_KEY " ,
45+ generationConfig: config
46+ )
47+ let _ = GenerativeModel (
48+ name: " gemini-1.5-pro-latest " ,
4149 apiKey: " API_KEY " ,
4250 systemInstruction: systemInstruction
4351 )
4452
4553 // All arguments passed.
46- let genAI = GenerativeModel ( name: " gemini-1.0 -pro " ,
54+ let genAI = GenerativeModel ( name: " gemini-1.5 -pro-latest " ,
4755 apiKey: " API_KEY " ,
4856 generationConfig: config, // Optional
4957 safetySettings: filters, // Optional
0 commit comments