Update app.py

app.py

@@ -654,31 +654,32 @@ 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()
 
 
 
@@ -686,7 +687,7 @@ safe_workflow.add_edge("tool_executor", "safety")
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        self.safe_app = safe_workflow.compile()
+        # self.safe_app = safe_workflow.compile()
 
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
@@ -701,7 +702,7 @@ class BasicAgent:
             state = {
                 "messages": question,
             }
-            response = self.safe_app.invoke(state)
+            # response = self.safe_app.invoke(state)
             
             # agent_answer = response["output"]
             agent_answer = "Hello World!"