From 1a9992f42d685fcd18c3b862fc21ae239100bb0e Mon Sep 17 00:00:00 2001 From: Shin Date: Sat, 7 Feb 2026 02:00:55 +0000 Subject: [PATCH] fix(test): update deprecated gemini embedding model Replace text-embedding-004 with gemini-embedding-001. The old model was deprecated and returns 404: 'models/text-embedding-004 is not found for API version v1beta' --- tests/local_testing/test_embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_embedding.py b/tests/local_testing/test_embedding.py index d46def7f3ae..48d535fe450 100644 --- a/tests/local_testing/test_embedding.py +++ b/tests/local_testing/test_embedding.py @@ -932,12 +932,12 @@ async def test_gemini_embeddings(sync_mode, input): litellm.set_verbose = True if sync_mode: response = litellm.embedding( - model="gemini/text-embedding-004", + model="gemini/gemini-embedding-001", input=input, ) else: response = await litellm.aembedding( - model="gemini/text-embedding-004", + model="gemini/gemini-embedding-001", input=input, ) print(f"response: {response}")