Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add state

Browse Source
This commit is contained in:
goulustis
2025-11-13 00:08:17 +08:00
parent f2ce3069f0
commit 363daf0b9c
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lang_agent/graphs/routing.py
View File

@@ -1,5 +1,5 @@
from dataclasses import dataclass, field, is_dataclass from dataclasses import dataclass, field, is_dataclass
from typing import Type, TypedDict, Literal, Dict, List, Tuple from typing import Type, TypedDict, Literal, Dict, List, Tuple, Any
import tyro import tyro
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from loguru import logger from loguru import logger
@@ -65,6 +65,8 @@ class State(TypedDict):
Dict[str, Dict[str, str|int]]] Dict[str, Dict[str, str|int]]]
messages: List[SystemMessage | HumanMessage] messages: List[SystemMessage | HumanMessage]
decision: str decision: str
subgraph_states: Dict[str, Any] # NOTE: Naively assuming subgraphs
# won't be so complicated