Update app.py

app.py

@@ -655,31 +655,31 @@ def tool_executor(state: AgentState):
     return state
 
 
-# safe_workflow = StateGraph(AgentState)
-# # safe_workflow = StateGraph(dict)
-
-# safe_workflow.add_node("planner", planner_node)
-# safe_workflow.add_node("tool_executor", tool_executor)
-# safe_workflow.add_node("safety", safety_node)
-# # safe_workflow.add_node("judge", Judge)
-
-# # safe_workflow.set_entry_point("planner")
-
-# safe_workflow.add_edge(START, "planner")
-# safe_workflow.add_edge("planner", "tool_executor")
-# safe_workflow.add_edge("tool_executor", "safety")
-# # safe_workflow.add_edge("safety", "judge")
-# # safe_workflow.add_conditional_edges(
-# #     "safety",
-# #     route,
-# #     {
-# #         "allow": "tool_executor",
-# #         "block": END,
-# #     },
-# # )
-# # safe_workflow.add_edge("tool_executor", END)
-
-# # safe_app = safe_workflow.compile()
+safe_workflow = StateGraph(AgentState)
+# safe_workflow = StateGraph(dict)
+
+safe_workflow.add_node("planner", planner_node)
+safe_workflow.add_node("tool_executor", tool_executor)
+safe_workflow.add_node("safety", safety_node)
+# safe_workflow.add_node("judge", Judge)
+
+# safe_workflow.set_entry_point("planner")
+
+safe_workflow.add_edge(START, "planner")
+safe_workflow.add_edge("planner", "tool_executor")
+safe_workflow.add_edge("tool_executor", "safety")
+# safe_workflow.add_edge("safety", "judge")
+# safe_workflow.add_conditional_edges(
+#     "safety",
+#     route,
+#     {
+#         "allow": "tool_executor",
+#         "block": END,
+#     },
+# )
+# safe_workflow.add_edge("tool_executor", END)
+
+# safe_app = safe_workflow.compile()