diff --git a/README.md b/README.md index 4ed933d3..b90fa2d0 100644 --- a/README.md +++ b/README.md @@ -16,7 +16,7 @@ For example, with just a few lines of code, you can access Gemini's multimodal c generate text from text-and-image input: ```swift -let model = GenerativeModel(name: "gemini-1.0-pro-vision-latest", apiKey: "YOUR_API_KEY") +let model = GenerativeModel(name: "gemini-1.5-pro-latest", apiKey: "YOUR_API_KEY") let cookieImage = UIImage(...) let prompt = "Do these look store-bought or homemade?" diff --git a/Sources/GoogleAI/GenerativeModel.swift b/Sources/GoogleAI/GenerativeModel.swift index 5e18ab20..4922a8ea 100644 --- a/Sources/GoogleAI/GenerativeModel.swift +++ b/Sources/GoogleAI/GenerativeModel.swift @@ -48,7 +48,7 @@ public final class GenerativeModel { /// Initializes a new remote model with the given parameters. /// /// - Parameters: - /// - name: The name of the model to use, e.g., `"gemini-1.0-pro"`; see + /// - name: The name of the model to use, e.g., `"gemini-1.5-pro-latest"`; see /// [Gemini models](https://ai.google.dev/models/gemini) for a list of supported model names. /// - apiKey: The API key for your project. /// - generationConfig: The content generation parameters your model should use. diff --git a/Tests/GoogleAITests/GoogleAITests.swift b/Tests/GoogleAITests/GoogleAITests.swift index 90b6fa69..12adf379 100644 --- a/Tests/GoogleAITests/GoogleAITests.swift +++ b/Tests/GoogleAITests/GoogleAITests.swift @@ -33,17 +33,25 @@ final class GoogleGenerativeAITests: XCTestCase { let systemInstruction = ModelContent(role: "system", parts: [.text("Talk like a pirate.")]) // Permutations without optional arguments. - let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY") - let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", safetySettings: filters) - let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", generationConfig: config) + let _ = GenerativeModel(name: "gemini-1.5-pro-latest", apiKey: "API_KEY") let _ = GenerativeModel( - name: "gemini-1.0-pro", + name: "gemini-1.5-pro-latest", + apiKey: "API_KEY", + safetySettings: filters + ) + let _ = GenerativeModel( + name: "gemini-1.5-pro-latest", + apiKey: "API_KEY", + generationConfig: config + ) + let _ = GenerativeModel( + name: "gemini-1.5-pro-latest", apiKey: "API_KEY", systemInstruction: systemInstruction ) // All arguments passed. - let genAI = GenerativeModel(name: "gemini-1.0-pro", + let genAI = GenerativeModel(name: "gemini-1.5-pro-latest", apiKey: "API_KEY", generationConfig: config, // Optional safetySettings: filters, // Optional