From 94dcc9588155e912f6dc008b4feee5899f4f9b8e Mon Sep 17 00:00:00 2001 From: goulustis Date: Fri, 30 Jan 2026 11:18:01 +0800 Subject: [PATCH] remove system prompt correctly to use local node --- lang_agent/graphs/routing.py | 2 +- lang_agent/graphs/tool_nodes.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/lang_agent/graphs/routing.py b/lang_agent/graphs/routing.py index f70d865..93ffd4b 100644 --- a/lang_agent/graphs/routing.py +++ b/lang_agent/graphs/routing.py @@ -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] diff --git a/lang_agent/graphs/tool_nodes.py b/lang_agent/graphs/tool_nodes.py index 3923f2d..8092231 100644 --- a/lang_agent/graphs/tool_nodes.py +++ b/lang_agent/graphs/tool_nodes.py @@ -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"])