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

use new graph state

Browse Source
This commit is contained in:
goulustis
2025-11-13 00:23:30 +08:00
parent dd09d92645
commit bb0f2785e6
1 changed files with 2 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lang_agent/graphs/routing.py
View File

@@ -14,6 +14,8 @@ import glob
from lang_agent.config import KeyConfig from lang_agent.config import KeyConfig
from lang_agent.tool_manager import ToolManager, ToolManagerConfig from lang_agent.tool_manager import ToolManager, ToolManagerConfig
from lang_agent.base import GraphBase from lang_agent.base import GraphBase
from lang_agent.graphs.graph_state import State
from lang_agent.graphs.tool_nodes import AnnotatedToolNode, ToolNode
from langchain.chat_models import init_chat_model from langchain.chat_models import init_chat_model
from langchain_core.messages import SystemMessage, HumanMessage, BaseMessage from langchain_core.messages import SystemMessage, HumanMessage, BaseMessage
@@ -60,16 +62,6 @@ class Route(BaseModel):
) )
class State(TypedDict):
inp: Tuple[Dict[str, List[SystemMessage | HumanMessage]],
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
class RoutingGraph(GraphBase): class RoutingGraph(GraphBase):
def __init__(self, config: RoutingConfig): def __init__(self, config: RoutingConfig):
self.config = config self.config = config