scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

fix(test): update deprecated gemini embedding model

Browse source 
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'
This commit is contained in:
Shin 2026-02-07 02:00:55 +00:00
parent 9b1ccc0608
commit 1a9992f42d
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_embedding.py
View file

@ -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}")