Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -152,7 +152,7 @@ def build_graph(provider: str = "groq") -> StateGraph:
|
|
| 152 |
# 6.2) Node: human → stash the GAIA task ID, then append the question
|
| 153 |
def human_node(state: AgentState, id: str, question: str) -> AgentState:
|
| 154 |
# keep the GAIA task id so we can submit it later
|
| 155 |
-
state["task_id"] =
|
| 156 |
state["messages"].append(HumanMessage(content=question))
|
| 157 |
return state
|
| 158 |
# 6.3) Node: assistant → call LLM on current messages
|
|
|
|
| 152 |
# 6.2) Node: human → stash the GAIA task ID, then append the question
|
| 153 |
def human_node(state: AgentState, id: str, question: str) -> AgentState:
|
| 154 |
# keep the GAIA task id so we can submit it later
|
| 155 |
+
state["task_id"] = task_id
|
| 156 |
state["messages"].append(HumanMessage(content=question))
|
| 157 |
return state
|
| 158 |
# 6.3) Node: assistant → call LLM on current messages
|