Update app.py

app.py

@@ -4,14 +4,14 @@ import requests
 import inspect
 import pandas as pd
 import ast
-from typing import TypedDict, List, Dict, Any, Optional
+from typing import TypedDict, List, Dict, Any, Optional, Annotated
 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 
+from typing_extensions import TypedDict, Annotated
 print("trial")
 # (Keep Constants as is)
 # --- Constants ---
@@ -19,16 +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: str
-    current_step: str
-    tool_output: str
-    final_answer: str
-    history: List[Dict[str, str]]
-    needs_more_info: bool
-    search_query: str
+    question: Annotated[str, "input"]
+    current_step: Annotated[str, "input"]
+    tool_output: Annotated[str, "input"]
+    final_answer: Annotated[str, "input"]
+    history: Annotated[List[Dict[str, str]], "input"]
+    needs_more_info: Annotated[bool, "input"]
+    search_query: Annotated[str, "input"]
 
 class BasicAgent:
     def __init__(self):