scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

fix test test_vertexai_multimodal_embedding

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-21 15:25:13 -07:00
parent 0e9efb3669
commit 2c7f5a0d27
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_amazing_vertex_completion.py
View file

@ -1828,10 +1828,8 @@ def test_vertexai_embedding():
@pytest.mark.asyncio()
async def test_vertexai_multimodal_embedding():
image_path = "../proxy/cached_logo.jpg"
# Getting the base64 string
base64_image = encode_image(image_path)
print("base 64 img ", base64_image)
load_vertex_ai_credentials()
try:
litellm.set_verbose = True
response = await litellm.aembedding(