Update app.py

app.py

@@ -4,8 +4,8 @@ import requests
 import inspect
 import pandas as pd
 import ast
-from typing import TypedDict, List, Dict, Any, Optional, Annotated
-from langgraph.graph import Graph, StateGraph
+from typing import List, Dict, Any, Optional, Annotated
+from langgraph.graph import Graph, StateGraph, State
 from langgraph.prebuilt import ToolNode
 from tools import simple_search
 from huggingface_hub import InferenceClient
@@ -19,12 +19,12 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 HF_TOKEN = os.getenv("HF_TOKEN")  # Make sure to set this environment variable
 
 
-class AgentState(TypedDict):
+class AgentState(State):
     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"]
+    history: Annotated[List[Dict[str, str]], "accumulate"]
     needs_more_info: Annotated[bool, "input"]
     search_query: Annotated[str, "input"]