Update app.py

app.py

@@ -10,9 +10,10 @@ from langgraph.graph import Graph, StateGraph
 from langgraph.prebuilt import ToolNode
 from tools import simple_search
 from huggingface_hub import InferenceClient
-# from langgraph.graph import State
-# from typing import Annotated
-from typing_extensions import TypedDict, Annotated
+from typing_extensions import TypedDict
+
+def override(_, new): return new
+
 print("trial")
 # (Keep Constants as is)
 # --- Constants ---
@@ -21,13 +22,13 @@ HF_TOKEN = os.getenv("HF_TOKEN")  # Make sure to set this environment variable
 
 
 class AgentState(TypedDict):
-    question: Annotated[str, operator.or_]
-    current_step: Annotated[str, operator.or_]
-    tool_output: Annotated[str, "input"]
-    final_answer: Annotated[str, "input"]
+    question: Annotated[str, override]
+    current_step: Annotated[str, override]
+    tool_output: Annotated[str, override]
+    final_answer: Annotated[str, override]
     history: Annotated[List[Dict[str, str]], operator.add]
-    needs_more_info: Annotated[bool, "input"]
-    search_query: Annotated[str, operator.or_]
+    needs_more_info: Annotated[bool, override]
+    search_query: Annotated[str, override]
 
 class BasicAgent:
     def __init__(self):