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

remove system prompt correctly to use local node

Browse Source
This commit is contained in:
goulustis
2026-01-30 11:18:01 +08:00
parent 781b3f450e
commit 94dcc95881
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lang_agent/graphs/routing.py
View File

@@ -161,7 +161,7 @@ class RoutingGraph(GraphBase):
SystemMessage(
self.prompt_dict["chat_prompt"]
),
*state["inp"][0]["messages"][1:]
*self._get_inp_msgs(state)
]}, state["inp"][1]

8
lang_agent/graphs/tool_nodes.py
View File

@@ -50,7 +50,7 @@ class ToolNode(ToolNodeBase):
SystemMessage(
self.sys_prompt
),
*state["inp"][0]["messages"][1:]
*self._get_inp_msgs(state)
]}, state["inp"][1]
out = self.tool_agent.invoke(*inp)
@@ -62,7 +62,7 @@ class ToolNode(ToolNodeBase):
SystemMessage(
self.sys_prompt
),
*state["inp"][0]["messages"][1:]
*self._get_inp_msgs(state)
]}, state["inp"][1]
out = await self.tool_agent.ainvoke(*inp)
@@ -160,7 +160,7 @@ class ChattyToolNode(ToolNodeBase):
# SystemMessage(
# self.tool_sys_prompt
# ),
# *state["inp"][0]["messages"][1:]
# *self._get_inp_msgs(state)
# ]}, state["inp"][1]
out = self.tool_agent.invoke(state)
@@ -176,7 +176,7 @@ class ChattyToolNode(ToolNodeBase):
SystemMessage(
f"回复的最开始应该是{self.chat_key}\n"+self.chatty_sys_prompt
),
*state["inp"][0]["messages"][1:]
*self._get_inp_msgs(state)
]}, state["inp"][1]
outs.extend(self.chatty_agent.invoke(*inp)["messages"])