scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix: satisfy Gemini tool history checks

Browse source
This commit is contained in:
Genmin 2026-05-01 12:19:28 -07:00
parent 148fa47b91
commit a9d23cf0f5
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/llms/vertex_ai/gemini/transformation.py
View file

@ -598,7 +598,10 @@ def _gemini_convert_messages_with_history( # noqa: PLR0915
and messages[msg_i]["role"] in tool_call_message_roles
):
tool_result_message = messages[msg_i]
tool_call_id = tool_result_message.get("tool_call_id")
tool_call_id_value = tool_result_message.get("tool_call_id")
tool_call_id = (
tool_call_id_value if isinstance(tool_call_id_value, str) else None
)
message_with_tool_call = (
message_by_tool_call_id.get(
tool_call_id, last_message_with_tool_calls