fmarky commited on
Commit
238f818
·
1 Parent(s): 9138f15

chore: add returned type in assistant(state: AgentState)

Browse files
Files changed (1) hide show
  1. agents/assistant_agent.py +1 -1
agents/assistant_agent.py CHANGED
@@ -148,7 +148,7 @@ def build_langgraph(
148
  # [4] Assistant Node
149
  # ---
150
 
151
- def assistant(state: AgentState):
152
  file = state["input_file"]
153
  sys_msg = SystemMessage(
154
  content=f"""
 
148
  # [4] Assistant Node
149
  # ---
150
 
151
+ def assistant(state: AgentState) -> AgentState:
152
  file = state["input_file"]
153
  sys_msg = SystemMessage(
154
  content=f"""