Spaces:
Sleeping
Sleeping
Update state.py
Browse files
state.py
CHANGED
@@ -1,15 +1,14 @@
|
|
1 |
-
from
|
|
|
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 |
-
|
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
|
|