fix KB search skipping when system message follows user message

This commit is contained in:
CREDO23 2026-04-13 18:03:16 +02:00
parent c72b737a81
commit c9e3b1a08a

View file

@ -774,11 +774,16 @@ class KnowledgeBaseSearchMiddleware(AgentMiddleware): # type: ignore[type-arg]
messages = state.get("messages") or []
if not messages:
return None
last_message = messages[-1]
if not isinstance(last_message, HumanMessage):
last_human = None
for msg in reversed(messages):
if isinstance(msg, HumanMessage):
last_human = msg
break
if last_human is None:
return None
user_text = _extract_text_from_message(last_message).strip()
user_text = _extract_text_from_message(last_human).strip()
if not user_text:
return None