phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(vertex_ai.py): check if message length > 0 before merging

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-19 18:47:08 -07:00
parent df1a2ca5c2
commit 4c5f077bd3
2 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/vertex_ai.py
View file

@ -337,7 +337,7 @@ def _gemini_convert_messages_with_history(messages: list) -> List[ContentType]:
_parts: List[PartType] = []
for element in messages[msg_i]["content"]:
if isinstance(element, dict):
if element["type"] == "text":
if element["type"] == "text" and len(element["text"]) > 0:
_part = PartType(text=element["text"])
_parts.append(_part)
elif element["type"] == "image_url":
@ -345,7 +345,10 @@ def _gemini_convert_messages_with_history(messages: list) -> List[ContentType]:
_part = _process_gemini_image(image_url=image_url)
_parts.append(_part) # type: ignore
user_content.extend(_parts)
else:
elif (
isinstance(messages[msg_i]["content"], str)
and len(messages[msg_i]["content"]) > 0
):
_part = PartType(text=messages[msg_i]["content"])
user_content.append(_part)