Scott Cogan commited on
Commit ·
fa1a73e
1
Parent(s): 1f48484
latest requirements
Browse files
app.py
CHANGED
|
@@ -202,12 +202,12 @@ class BasicAgent:
|
|
| 202 |
self.builder = StateGraph(state_schema=AgentState)
|
| 203 |
|
| 204 |
# Define nodes: these do the work
|
| 205 |
-
self.builder.add_node("
|
| 206 |
self.builder.add_node("assistant", self.assistant)
|
| 207 |
self.builder.add_node("tools", self.tools_node)
|
| 208 |
|
| 209 |
# Define edges: these determine how the control flow moves
|
| 210 |
-
self.builder.add_edge("
|
| 211 |
self.builder.add_edge("assistant", "tools")
|
| 212 |
self.builder.add_edge("tools", "assistant")
|
| 213 |
self.react_graph = self.builder.compile()
|
|
@@ -230,7 +230,7 @@ class BasicAgent:
|
|
| 230 |
fixed_answer = "This is a default answer."
|
| 231 |
|
| 232 |
await asyncio.sleep(60)
|
| 233 |
-
messages = self.react_graph.invoke({"messages": [f'Task id: {task_id}\n {question}']}, node="
|
| 234 |
return messages["messages"][-1].content if messages["messages"] else fixed_answer
|
| 235 |
|
| 236 |
|
|
|
|
| 202 |
self.builder = StateGraph(state_schema=AgentState)
|
| 203 |
|
| 204 |
# Define nodes: these do the work
|
| 205 |
+
self.builder.add_node("START", lambda state: state)
|
| 206 |
self.builder.add_node("assistant", self.assistant)
|
| 207 |
self.builder.add_node("tools", self.tools_node)
|
| 208 |
|
| 209 |
# Define edges: these determine how the control flow moves
|
| 210 |
+
self.builder.add_edge("START", "assistant")
|
| 211 |
self.builder.add_edge("assistant", "tools")
|
| 212 |
self.builder.add_edge("tools", "assistant")
|
| 213 |
self.react_graph = self.builder.compile()
|
|
|
|
| 230 |
fixed_answer = "This is a default answer."
|
| 231 |
|
| 232 |
await asyncio.sleep(60)
|
| 233 |
+
messages = self.react_graph.invoke({"messages": [f'Task id: {task_id}\n {question}']}, node="START")
|
| 234 |
return messages["messages"][-1].content if messages["messages"] else fixed_answer
|
| 235 |
|
| 236 |
|