Spaces:
Sleeping
Sleeping
chore: add returned type in assistant(state: AgentState)
Browse files
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"""
|