Scott Cogan
commited on
Commit
·
7f23189
1
Parent(s):
a3dfb95
latest requirements
Browse files
app.py
CHANGED
|
@@ -201,7 +201,7 @@ class BasicAgent:
|
|
| 201 |
# Graph
|
| 202 |
self.builder = StateGraph(state_schema=AgentState)
|
| 203 |
# Define nodes: these do the work
|
| 204 |
-
self.builder.add_node("START", lambda state: state)
|
| 205 |
self.builder.add_node("assistant", self.assistant)
|
| 206 |
self.builder.add_node("tools", self.tools_node)
|
| 207 |
# Define edges: these determine how the control flow moves
|
|
|
|
| 201 |
# Graph
|
| 202 |
self.builder = StateGraph(state_schema=AgentState)
|
| 203 |
# Define nodes: these do the work
|
| 204 |
+
self.builder.add_node("START", lambda state: AgentState(**state), input=AgentState)
|
| 205 |
self.builder.add_node("assistant", self.assistant)
|
| 206 |
self.builder.add_node("tools", self.tools_node)
|
| 207 |
# Define edges: these determine how the control flow moves
|