diff --git a/agent/component/rewrite.py b/agent/component/rewrite.py index f6bf0de69..06a1f6b23 100644 --- a/agent/component/rewrite.py +++ b/agent/component/rewrite.py @@ -29,7 +29,6 @@ class RewriteQuestionParam(GenerateParam): super().__init__() self.temperature = 0.9 self.prompt = "" - self.loop = 1 def check(self): super().check() @@ -85,13 +84,6 @@ class RewriteQuestion(Generate, ABC): component_name = "RewriteQuestion" def _run(self, history, **kwargs): - if not hasattr(self, "_loop"): - setattr(self, "_loop", 0) - if self._loop >= self._param.loop: - self._loop = 0 - raise Exception("Sorry! Nothing relevant found.") - self._loop += 1 - hist = self._canvas.get_history(4) conv = [] for m in hist: