scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 69

fix test_vertexai_multimodal_embedding_base64image_in_input

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-27 20:17:08 -07:00
parent 9fb1ee2294
commit 39b5d8f383
1 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
litellm/tests/test_amazing_vertex_completion.py
View file

@ -2050,12 +2050,19 @@ async def test_vertexai_multimodal_embedding_image_in_input():
@pytest.mark.asyncio
async def test_vertexai_multimodal_embedding_base64image_in_input():
import base64
import requests
load_vertex_ai_credentials()
mock_response = AsyncMock()
image_path = "../proxy/cached_logo.jpg"
# Getting the base64 string
base64_image = encode_image(image_path)
url = "https://dummyimage.com/100/100/fff&text=Test+image"
response = requests.get(url)
file_data = response.content
encoded_file = base64.b64encode(file_data).decode("utf-8")
base64_image = f"data:image/png;base64,{encoded_file}"
def return_val():
return {