clearner human message addition

2026-01-06 18:12:57 +08:00
parent 9136fa3b59
commit 838cdbf51f
1 changed files with 9 additions and 15 deletions
--- a/lang_agent/base.py
+++ b/lang_agent/base.py
@@ -146,21 +146,15 @@ class GraphBase(ABC):
        else:
            inp = state["inp"]
        
-        if human_msg is None:
-            inp = {"messages":[
-                        SystemMessage(
-                            system_prompt
-                        ),
-                        *state["inp"][0]["messages"][1:]
-                    ]}, state["inp"][1]
-        else:
-            inp = {"messages":[
-                        SystemMessage(
-                            system_prompt
-                        ),
-                        *state["inp"][0]["messages"][1:],
-                        HumanMessage(human_msg)
-                    ]}, state["inp"][1]
+        messages = [
+            SystemMessage(system_prompt),
+            *state["inp"][0]["messages"][1:]
+        ]
+
+        if human_msg is not None:
+            messages.append(HumanMessage(human_msg))
+
+        inp = ({"messages": messages}, state["inp"][1])


        out = model.invoke(*inp)