naman1102 commited on
Commit
c99f0eb
·
1 Parent(s): a979fc9

Update state.py

Browse files
Files changed (1) hide show
  1. state.py +4 -5
state.py CHANGED
@@ -1,15 +1,14 @@
1
- from typing import TypedDict, Annotated
 
2
  from langgraph.graph.message import add_messages
 
3
  class AgentState(TypedDict, total=False):
4
  messages: Annotated[list, add_messages]
5
- # Fields that the agent node can set to request a tool
6
  web_search_query: str
7
  ocr_path: str
8
  excel_path: str
9
  excel_sheet_name: str
10
- # Fields to hold the tool outputs
11
  web_search_result: str
12
  ocr_result: str
13
  excel_result: str
14
- # A “final_answer field that the last agent node will fill
15
- final_answer: str# (Keep Constants as is)
 
1
+ from typing_extensions import TypedDict
2
+ from typing import Annotated
3
  from langgraph.graph.message import add_messages
4
+
5
  class AgentState(TypedDict, total=False):
6
  messages: Annotated[list, add_messages]
 
7
  web_search_query: str
8
  ocr_path: str
9
  excel_path: str
10
  excel_sheet_name: str
 
11
  web_search_result: str
12
  ocr_result: str
13
  excel_result: str
14
+ final_answer: str