Update app.py

app.py

@@ -687,6 +687,17 @@ class SuperSmartAgent:
         
             return None  # Added missing return statement
 
+        class AgentState(TypedDict, total=False):
+            question: str
+            is_reversed: bool
+            is_python: bool
+            is_riddle: bool
+            is_wiki: bool  # Added for Wikipedia suitability check
+            needs_reasoning: bool  # Added for reasoning check
+            response: str
+            use_tool: str  # Keep this if it's being used elsewhere
+    
+        builder = StateGraph(AgentState)
         
         # Entry point remains the same
         builder.set_entry_point("check_reversed")