From 838cdbf51f7d534576b9207f79a095fcd0c7575d Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Tue, 6 Jan 2026 18:12:57 +0800
Subject: [PATCH] clearner human message addition

---
 lang_agent/base.py | 24 +++++++++---------------
 1 file changed, 9 insertions(+), 15 deletions(-)

diff --git a/lang_agent/base.py b/lang_agent/base.py
index 86fcc52..7e33f35 100644
--- 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)