Skip to content

Commit 0cd4ef6

Browse files
authored
Switch to gemini-1.5-pro-latest in docs and tests (#138)
1 parent 8e3787f commit 0cd4ef6

File tree

3 files changed

+15
-7
lines changed

3 files changed

+15
-7
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ For example, with just a few lines of code, you can access Gemini's multimodal c
1616
generate text from text-and-image input:
1717

1818
```swift
19-
let model = GenerativeModel(name: "gemini-1.0-pro-vision-latest", apiKey: "YOUR_API_KEY")
19+
let model = GenerativeModel(name: "gemini-1.5-pro-latest", apiKey: "YOUR_API_KEY")
2020
let cookieImage = UIImage(...)
2121
let prompt = "Do these look store-bought or homemade?"
2222

Sources/GoogleAI/GenerativeModel.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ public final class GenerativeModel {
4848
/// Initializes a new remote model with the given parameters.
4949
///
5050
/// - Parameters:
51-
/// - name: The name of the model to use, e.g., `"gemini-1.0-pro"`; see
51+
/// - name: The name of the model to use, e.g., `"gemini-1.5-pro-latest"`; see
5252
/// [Gemini models](https://ai.google.dev/models/gemini) for a list of supported model names.
5353
/// - apiKey: The API key for your project.
5454
/// - generationConfig: The content generation parameters your model should use.

Tests/GoogleAITests/GoogleAITests.swift

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)