wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-19 12:24:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: handle empty agent tuple in streaming workflow (#427)

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

Fixes #425
This commit is contained in:
Affan Shaikhsurab
2025-07-16 06:29:11 +05:30
committed by GitHub
parent b155e1eca6
commit b04225b7c8
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/server/app.py
View File

@@ -153,9 +153,13 @@ async def _astream_workflow_generator(
message_chunk, message_metadata = cast( message_chunk, message_metadata = cast(
tuple[BaseMessage, dict[str, any]], event_data 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] = { event_stream_message: dict[str, any] = {
"thread_id": thread_id, "thread_id": thread_id,
"agent": agent[0].split(":")[0], "agent": agent_name,
"id": message_chunk.id, "id": message_chunk.id,
"role": "assistant", "role": "assistant",
"content": message_chunk.content, "content": message_chunk.content,