diff --git a/agent/component/generate.py b/agent/component/generate.py index 3b05747e1..ea12f9040 100644 --- a/agent/component/generate.py +++ b/agent/component/generate.py @@ -198,6 +198,7 @@ class Generate(ComponentBase): if len(msg) < 2: msg.append({"role": "user", "content": "Output: "}) ans = chat_mdl.chat(msg[0]["content"], msg[1:], self._param.gen_conf()) + ans = re.sub(r".*", "", ans, flags=re.DOTALL) if self._param.cite and "content_ltks" in retrieval_res.columns and "vector" in retrieval_res.columns: res = self.set_cite(retrieval_res, ans)