cryogenic22 commited on
Commit
5924432
·
verified ·
1 Parent(s): 4c919b6

Update agents/sql_generator.py

Browse files
Files changed (1) hide show
  1. agents/sql_generator.py +2 -2
agents/sql_generator.py CHANGED
@@ -95,10 +95,10 @@ def sql_generator_agent(anthropic_client: Anthropic, state: AgentState) -> Agent
95
  sql_queries = _extract_sql_queries(agent_response)
96
 
97
  new_state["sql_queries"] = sql_queries
98
- # Append next agent
99
  new_state["current_agent"].append("executor_agent")
100
  else:
101
- # Append current agent if SQL generation is not complete
102
  new_state["current_agent"].append("sql_generator_agent")
103
 
104
  return new_state
 
95
  sql_queries = _extract_sql_queries(agent_response)
96
 
97
  new_state["sql_queries"] = sql_queries
98
+ # Use .append() to add next agent to the list
99
  new_state["current_agent"].append("executor_agent")
100
  else:
101
+ # Use .append() to add current agent if SQL generation is not complete
102
  new_state["current_agent"].append("sql_generator_agent")
103
 
104
  return new_state