Update app.py

app.py

@@ -9,8 +9,8 @@ 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 langgraph.graph import State
+# from typing import Annotated
 from typing_extensions import TypedDict 
 print("trial")
 # (Keep Constants as is)
@@ -19,12 +19,14 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 HF_TOKEN = os.getenv("HF_TOKEN")  # Make sure to set this environment variable
 
 
+from typing import TypedDict, List, Dict
+
 class AgentState(TypedDict):
-    question: Annotated[str, State.Immutable]
+    question: str
     current_step: str
     tool_output: str
     final_answer: str
-    history: Annotated[List[Dict[str, str]], State.MergingList]
+    history: List[Dict[str, str]]
     needs_more_info: bool
     search_query: str