AC-Angelo93 commited on
Commit
72fbc8e
·
verified ·
1 Parent(s): afe3bcf

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
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"] = 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