asf

agent.py

@@ -217,7 +217,7 @@ def build_graph(provider: str = "huggingface"):
     builder.add_edge("tools", "assistant")
 
     # Compile graph
-    return builder.compile(recursion_limit=50)
+    return builder.compile()
 
 # test
 if __name__ == "__main__":
@@ -228,7 +228,9 @@ if __name__ == "__main__":
     messages = [HumanMessage(content=question)]
 
     print(messages)
-    messages = graph.invoke({"messages": messages})
+    config = {"recursion_limit": 50}
+
+    messages = graph.invoke({"messages": messages}, config=config)
     for m in messages["messages"]:
         m.pretty_print()