graph test

app.py

@@ -182,12 +182,12 @@ class BasicAgent:
 
         if state["is_final_answer"]:
             return "FINAL_ANSWER"
-        if state["attempt"] > 3:
+        elif state["attempt"] > 3:
             return "EXCEEDED_ATTEMPTS"
-
-        if state["search_request"]:
+        elif state["search_request"]:
             return "WEB_SEARCH"
-
+        elif state["wiki_request"]:
+            return "WIKI_SEARCH"
         return "FINAL_ANSWER"
 
     def invoke_model(self, state: AnswerState) -> Dict[str, Any]: