From 100c60017fe48ba3d4434baab30ae3c3cd451bc8 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Tue, 24 Sep 2024 11:31:42 +0800
Subject: [PATCH] fix component rewrite bug (#2549)

### What problem does this PR solve?

#2545

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 agent/component/rewrite.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/agent/component/rewrite.py b/agent/component/rewrite.py
index 84eb73e00..31c8fde3b 100644
--- a/agent/component/rewrite.py
+++ b/agent/component/rewrite.py
@@ -65,6 +65,8 @@ class RewriteQuestion(Generate, ABC):
         chat_mdl = LLMBundle(self._canvas.get_tenant_id(), LLMType.CHAT, self._param.llm_id)
         ans = chat_mdl.chat(self._param.get_prompt(), [{"role": "user", "content": q}],
                             self._param.gen_conf())
+        self._canvas.history.pop()
+        self._canvas.history.append(("user", ans))
 
         print(ans, ":::::::::::::::::::::::::::::::::")
         return RewriteQuestion.be_output(ans)