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

add human message in template

Browse Source
This commit is contained in:
goulustis
2026-01-06 18:00:42 +08:00
parent 1112d57335
commit 9136fa3b59
1 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lang_agent/base.py
View File

@@ -139,18 +139,28 @@ class GraphBase(ABC):
def _agent_call_template(self, system_prompt:str, def _agent_call_template(self, system_prompt:str,
model:CompiledStateGraph, model:CompiledStateGraph,
state:State): state:State,
human_msg:str = None):
if state.get("messages") is not None: if state.get("messages") is not None:
inp = state["messages"], state["inp"][1] inp = state["messages"], state["inp"][1]
else: else:
inp = state["inp"] inp = state["inp"]
if human_msg is None:
inp = {"messages":[ inp = {"messages":[
SystemMessage( SystemMessage(
system_prompt system_prompt
), ),
*state["inp"][0]["messages"][1:] *state["inp"][0]["messages"][1:]
]}, state["inp"][1] ]}, state["inp"][1]
else:
inp = {"messages":[
SystemMessage(
system_prompt
),
*state["inp"][0]["messages"][1:],
HumanMessage(human_msg)
]}, state["inp"][1]
out = model.invoke(*inp) out = model.invoke(*inp)