Scott Cogan commited on
Commit
d6b1315
·
1 Parent(s): f39a6cf
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -195,7 +195,6 @@ class BasicAgent:
195
 
196
  # Create the graph
197
  self.workflow = StateGraph(AgentState)
198
- self.workflow.set_config({"recursion_limit": 50})
199
 
200
  # Add nodes
201
  self.workflow.add_node("agent", self.call_model)
@@ -208,8 +207,8 @@ class BasicAgent:
208
  # Set entry point
209
  self.workflow.set_entry_point("agent")
210
 
211
- # Compile the graph
212
- self.app = self.workflow.compile()
213
 
214
  logger.info("BasicAgent initialized.")
215
 
 
195
 
196
  # Create the graph
197
  self.workflow = StateGraph(AgentState)
 
198
 
199
  # Add nodes
200
  self.workflow.add_node("agent", self.call_model)
 
207
  # Set entry point
208
  self.workflow.set_entry_point("agent")
209
 
210
+ # Compile the graph with recursion limit
211
+ self.app = self.workflow.compile(recursion_limit=50)
212
 
213
  logger.info("BasicAgent initialized.")
214