clearner human message addition
This commit is contained in:
@@ -146,21 +146,15 @@ class GraphBase(ABC):
|
|||||||
else:
|
else:
|
||||||
inp = state["inp"]
|
inp = state["inp"]
|
||||||
|
|
||||||
if human_msg is None:
|
messages = [
|
||||||
inp = {"messages":[
|
SystemMessage(system_prompt),
|
||||||
SystemMessage(
|
|
||||||
system_prompt
|
|
||||||
),
|
|
||||||
*state["inp"][0]["messages"][1:]
|
*state["inp"][0]["messages"][1:]
|
||||||
]}, state["inp"][1]
|
]
|
||||||
else:
|
|
||||||
inp = {"messages":[
|
if human_msg is not None:
|
||||||
SystemMessage(
|
messages.append(HumanMessage(human_msg))
|
||||||
system_prompt
|
|
||||||
),
|
inp = ({"messages": messages}, state["inp"][1])
|
||||||
*state["inp"][0]["messages"][1:],
|
|
||||||
HumanMessage(human_msg)
|
|
||||||
]}, state["inp"][1]
|
|
||||||
|
|
||||||
|
|
||||||
out = model.invoke(*inp)
|
out = model.invoke(*inp)
|
||||||
|
|||||||
Reference in New Issue
Block a user