diff --git a/agent/component/retrieval.py b/agent/component/retrieval.py index 859d65478..218dae969 100644 --- a/agent/component/retrieval.py +++ b/agent/component/retrieval.py @@ -96,6 +96,7 @@ class Retrieval(ComponentBase, ABC): rerank_mdl = LLMBundle(kbs[0].tenant_id, LLMType.RERANK, self._param.rerank_id) if kbs: + query = re.sub(r"^user[::\s]*", "", query, flags=re.IGNORECASE) kbinfos = settings.retrievaler.retrieval( query, embd_mdl,