From 363daf0b9cb87865be6e8450a9aa26258bdedde3 Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Thu, 13 Nov 2025 00:08:17 +0800
Subject: [PATCH] add state

---
 lang_agent/graphs/routing.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/lang_agent/graphs/routing.py b/lang_agent/graphs/routing.py
index 142b094..c7df924 100644
--- a/lang_agent/graphs/routing.py
+++ b/lang_agent/graphs/routing.py
@@ -1,5 +1,5 @@
 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
 from pydantic import BaseModel, Field
 from loguru import logger
@@ -65,6 +65,8 @@ class State(TypedDict):
                Dict[str, Dict[str, str|int]]]
     messages: List[SystemMessage | HumanMessage]
     decision: str
+    subgraph_states: Dict[str, Any]   # NOTE: Naively assuming subgraphs
+                                      #       won't be so complicated