naman1102's picture
Update state.py
c99f0eb
raw
history blame
386 Bytes
from typing_extensions import TypedDict
from typing import Annotated
from langgraph.graph.message import add_messages
class AgentState(TypedDict, total=False):
messages: Annotated[list, add_messages]
web_search_query: str
ocr_path: str
excel_path: str
excel_sheet_name: str
web_search_result: str
ocr_result: str
excel_result: str
final_answer: str