From 222f15d8a1611931a5317af296d2507e7b96659e Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 12 Dec 2024 17:45:34 -0300 Subject: [PATCH] fix: Ensure safe assignment to tool_blocks_map in agent_message handling (#5239) fix: ensure tool_blocks_map assignment is safe by checking content availability in agent_message --- src/backend/base/langflow/base/agents/events.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/base/agents/events.py b/src/backend/base/langflow/base/agents/events.py index d7bfde54b..26198ff4a 100644 --- a/src/backend/base/langflow/base/agents/events.py +++ b/src/backend/base/langflow/base/agents/events.py @@ -149,7 +149,8 @@ async def handle_on_tool_start( agent_message.content_blocks[0].contents.append(tool_content) agent_message = await send_message_method(message=agent_message) - tool_blocks_map[tool_key] = agent_message.content_blocks[0].contents[-1] + if agent_message.content_blocks and agent_message.content_blocks[0].contents: + tool_blocks_map[tool_key] = agent_message.content_blocks[0].contents[-1] return agent_message, new_start_time