Update app.py

app.py

@@ -49,19 +49,19 @@ class BasicAgent:
         # Add nodes
         workflow.add_node("analyze", self._analyze_question)
         workflow.add_node("search", self._use_search)
-        workflow.add_node("final_answer", self._generate_final_answer)
+        workflow.add_node("generate_answer", self._generate_final_answer)
         
         # Define edges
         workflow.add_edge("analyze", "search")
-        workflow.add_edge("analyze", "final_answer")
-        workflow.add_edge("search", "final_answer")
+        workflow.add_edge("analyze", "generate_answer")
+        workflow.add_edge("search", "generate_answer")
         
         # Define conditional edges
         def router(state: AgentState) -> str:
             if state["current_step"] == 'search':
                 return 'search'
             elif state["current_step"] == 'final_answer':
-                return 'final_answer'
+                return 'generate_answer'
             return 'analyze'
         
         workflow.add_conditional_edges(
@@ -69,13 +69,13 @@ class BasicAgent:
             router,
             {
                 "search": "search",
-                "final_answer": "final_answer"
+                "final_answer": "generate_answer"
             }
         )
         
         # Set entry and exit points
         workflow.set_entry_point("analyze")
-        workflow.set_finish_point("final_answer")
+        workflow.set_finish_point("generate_answer")
         
         return workflow.compile()