From a9d23cf0f52e3d7c3c8b7606c9993048bf331096 Mon Sep 17 00:00:00 2001 From: Genmin Date: Fri, 1 May 2026 12:19:28 -0700 Subject: [PATCH] fix: satisfy Gemini tool history checks --- litellm/llms/vertex_ai/gemini/transformation.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/llms/vertex_ai/gemini/transformation.py b/litellm/llms/vertex_ai/gemini/transformation.py index afdadb46730..3a4230890e0 100644 --- a/litellm/llms/vertex_ai/gemini/transformation.py +++ b/litellm/llms/vertex_ai/gemini/transformation.py @@ -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