Skip to content

Commit 4dacdf3

Browse files
[Firebase AI] Upgrade Gemini 1.x models in integration tests (#14956)
Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>
1 parent 4494bac commit 4dacdf3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

FirebaseAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ final class IntegrationTests: XCTestCase {
6767
func testCountTokens_text() async throws {
6868
let prompt = "Why is the sky blue?"
6969
model = vertex.generativeModel(
70-
modelName: "gemini-1.5-pro",
70+
modelName: ModelNames.gemini2Flash,
7171
generationConfig: generationConfig,
7272
safetySettings: [
7373
SafetySetting(harmCategory: .harassment, threshold: .blockLowAndAbove, method: .severity),

0 commit comments

Comments
 (0)