Spaces:
Runtime error
Runtime error
Update agents/sql_generator.py
Browse files- 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 |
-
#
|
| 99 |
new_state["current_agent"].append("executor_agent")
|
| 100 |
else:
|
| 101 |
-
#
|
| 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
|