From b04225b7c83e5635f91f17d691b34d6244a90b1e Mon Sep 17 00:00:00 2001
From: Affan Shaikhsurab <51104750+AffanShaikhsurab@users.noreply.github.com>
Date: Wed, 16 Jul 2025 06:29:11 +0530
Subject: [PATCH] fix: handle empty agent tuple in streaming workflow (#427)

Prevents IndexError when agent[0] is accessed on empty tuple,
resolving display issues with Gemini 2.0 Flash model.

Fixes #425
---
 src/server/app.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/server/app.py b/src/server/app.py
index 3135149..bfcddd2 100644
--- a/src/server/app.py
+++ b/src/server/app.py
@@ -153,9 +153,13 @@ async def _astream_workflow_generator(
         message_chunk, message_metadata = cast(
             tuple[BaseMessage, dict[str, any]], event_data
         )
+        # Handle empty agent tuple gracefully
+        agent_name = "unknown"
+        if agent and len(agent) > 0:
+            agent_name = agent[0].split(":")[0] if ":" in agent[0] else agent[0]
         event_stream_message: dict[str, any] = {
             "thread_id": thread_id,
-            "agent": agent[0].split(":")[0],
+            "agent": agent_name,
             "id": message_chunk.id,
             "role": "assistant",
             "content": message_chunk.content,