tan-en-yao commited on
Commit
9485d99
·
1 Parent(s): cf661d0

fix: include triage agent in agent_states

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -268,8 +268,9 @@ def process_report(message: str, image, history: list, session_logs: list, sessi
268
  # Track reasoning trace for autonomous reasoning display
269
  current_reasoning = reasoning_state or "*Initializing autonomous reasoning...*"
270
 
271
- # Track agent states for Gradio component updates
272
  agent_states = {
 
273
  "research": {"status": "pending", "tools": [], "result": None, "start_time": None, "duration": None},
274
  "report": {"status": "pending", "tools": [], "result": None, "start_time": None, "duration": None},
275
  }
@@ -308,7 +309,8 @@ def process_report(message: str, image, history: list, session_logs: list, sessi
308
  elapsed = time.time() - processing_start
309
  total_tools = sum(len(s["tools"]) for s in agent_states.values())
310
  completed = sum(1 for s in agent_states.values() if s["status"] == "done")
311
- return f"**Processing** | ⏱️ {elapsed:.1f}s | Agents: {completed}/2 | Tools: {total_tools}"
 
312
 
313
  def build_outputs():
314
  """Build all output values for yield."""
 
268
  # Track reasoning trace for autonomous reasoning display
269
  current_reasoning = reasoning_state or "*Initializing autonomous reasoning...*"
270
 
271
+ # Track agent states for Gradio component updates (all 3 agents)
272
  agent_states = {
273
+ "triage": {"status": "pending", "tools": [], "result": None, "start_time": None, "duration": None},
274
  "research": {"status": "pending", "tools": [], "result": None, "start_time": None, "duration": None},
275
  "report": {"status": "pending", "tools": [], "result": None, "start_time": None, "duration": None},
276
  }
 
309
  elapsed = time.time() - processing_start
310
  total_tools = sum(len(s["tools"]) for s in agent_states.values())
311
  completed = sum(1 for s in agent_states.values() if s["status"] == "done")
312
+ total_agents = len(agent_states)
313
+ return f"**Processing** | ⏱️ {elapsed:.1f}s | Agents: {completed}/{total_agents} | Tools: {total_tools}"
314
 
315
  def build_outputs():
316
  """Build all output values for yield."""