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

chatty tool state

Browse Source
This commit is contained in:
goulustis
2025-11-21 14:47:41 +08:00
parent 9022540b79
commit 57fdd420db
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lang_agent/graphs/graph_states.py
View File

@@ -1,6 +1,6 @@
from typing import Type, TypedDict, Literal, Dict, List, Tuple, Any from typing import Type, TypedDict, Literal, Dict, List, Tuple, Any
from langchain_core.messages import SystemMessage, HumanMessage from langchain_core.messages import SystemMessage, HumanMessage, AIMessage
class State(TypedDict): class State(TypedDict):
inp: Tuple[Dict[str, List[SystemMessage | HumanMessage]], inp: Tuple[Dict[str, List[SystemMessage | HumanMessage]],
@@ -9,3 +9,10 @@ class State(TypedDict):
decision: str decision: str
subgraph_states: Dict[str, Any] # NOTE: Naively assuming subgraphs subgraph_states: Dict[str, Any] # NOTE: Naively assuming subgraphs
# won't be so complicated # won't be so complicated
class ChattyToolState(TypedDict):
inp: Tuple[Dict[str, List[SystemMessage | HumanMessage]],
Dict[str, Dict[str, str|int]]]
tool_messages: List[SystemMessage | HumanMessage | AIMessage]
chatty_messages: List[SystemMessage | HumanMessage | AIMessage]