Update app.py

app.py

@@ -13,7 +13,7 @@ import os
 st.title("Checkpoints and Breakpoints")
 st.caption("Demonstrating workflow execution with checkpoints and tool invocation.")
 
-# Fetch API Keys from Hugging Face Secrets
+# Fetch API Keys
 openai_api_key = os.getenv("OPENAI_API_KEY")
 tavily_api_key = os.getenv("TAVILY_API_KEY")
 
@@ -33,6 +33,8 @@ if openai_api_key and tavily_api_key:
 
     # Agent Function
     def Agent(state: State):
+        print("Agent received state:", state)
+        # Force tool invocation
         response = llm_with_tools.invoke(state["messages"])
         print("Agent Response:", response)
         return {"messages": [response]}
@@ -46,7 +48,12 @@ if openai_api_key and tavily_api_key:
 
     graph.add_node("Agent", Agent)
     graph.add_node("tools", tool_node)
-    graph.add_conditional_edges("Agent", tools_condition)
+
+    # Force tools_condition always True for testing purposes
+    def always_true(state):
+        return True
+
+    graph.add_conditional_edges("Agent", always_true, {"True": "tools"})
     graph.add_edge("tools", "Agent")
     graph.set_entry_point("Agent")