diff --git a/packages/ai/test/context-overflow.test.ts b/packages/ai/test/context-overflow.test.ts index 59a3fb6be..e66d21913 100644 --- a/packages/ai/test/context-overflow.test.ts +++ b/packages/ai/test/context-overflow.test.ts @@ -203,8 +203,8 @@ describe("Context overflow error handling", () => { // ============================================================================= describe.skipIf(!process.env.GEMINI_API_KEY)("Google", () => { - it("gemini-2.0-flash - should detect overflow via isContextOverflow", async () => { - const model = getModel("google", "gemini-2.0-flash"); + it("gemini-2.5-flash - should detect overflow via isContextOverflow", async () => { + const model = getModel("google", "gemini-2.5-flash"); const result = await testContextOverflow(model, process.env.GEMINI_API_KEY!); logResult(result); diff --git a/packages/ai/test/total-tokens.test.ts b/packages/ai/test/total-tokens.test.ts index ae4255e10..fabff7f69 100644 --- a/packages/ai/test/total-tokens.test.ts +++ b/packages/ai/test/total-tokens.test.ts @@ -222,10 +222,10 @@ describe("totalTokens field", () => { describe.skipIf(!process.env.GEMINI_API_KEY)("Google", () => { it( - "gemini-2.0-flash - should return totalTokens equal to sum of components", + "gemini-2.5-flash - should return totalTokens equal to sum of components", { retry: 3, timeout: 60000 }, async () => { - const llm = getModel("google", "gemini-2.0-flash"); + const llm = getModel("google", "gemini-2.5-flash"); console.log(`\nGoogle / ${llm.id}:`); const { first, second } = await testTotalTokensWithCache(llm);