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

build graph

Browse Source
This commit is contained in:
goulustis
2025-11-13 00:21:38 +08:00
parent 363daf0b9c
commit 84457e2d78
1 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
lang_agent/graphs/tool_nodes.py
View File

@@ -21,6 +21,7 @@ from langchain.agents import create_agent
from langchain.chat_models import init_chat_model from langchain.chat_models import init_chat_model
from langgraph.checkpoint.memory import MemorySaver from langgraph.checkpoint.memory import MemorySaver
from langgraph.graph import StateGraph, START, END
@dataclass @dataclass
@@ -87,6 +88,8 @@ class ChattyToolNode:
self.tool_done = False self.tool_done = False
self.populate_modules() self.populate_modules()
self.build_graph()
def populate_modules(self): def populate_modules(self):
self.chatty_llm = init_chat_model(model=self.config.llm_name, self.chatty_llm = init_chat_model(model=self.config.llm_name,
@@ -141,6 +144,22 @@ class ChattyToolNode:
return {"messages": state["messages"] + chat_msgs + tool_msgs} return {"messages": state["messages"] + chat_msgs + tool_msgs}
def build_graph(self):
builder = StateGraph(State)
builder.add_node("chatty_tool_call", self._tool_node_call)
builder.add_node("chatty_chat_call", self._chat_node_call)
builder.add_node("chatty_handoff_node", self._handoff_node)
builder.add_edge(START, "chatty_tool_call")
builder.add_edge(START, "chatty_chat_call")
builder.add_edge("chatty_chat_call", "chatty_handoff_node")
builder.add_edge("chatty_node_call", "chatty_handoff_node")
builder.add_edge("chatty_handoff_node", END)
self.workflow = builder.compile()
tool_node_dict = { tool_node_dict = {
"tool_node" : ToolNodeConfig(), "tool_node" : ToolNodeConfig(),
"chatty_tool_node" : ChattyToolNodeConfig() "chatty_tool_node" : ChattyToolNodeConfig()