AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-07-27 16:12:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove redundant param of rewrite component. (#4422)

Browse Source 
### What problem does this PR solve?

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2025-01-09 11:54:31 +08:00 committed by GitHub
parent b6c71c1e01
commit bc681e2ee9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
agent/component/rewrite.py
View File

@ -29,7 +29,6 @@ class RewriteQuestionParam(GenerateParam):
super().__init__() super().__init__()
self.temperature = 0.9 self.temperature = 0.9
self.prompt = "" self.prompt = ""
self.loop = 1
def check(self): def check(self):
super().check() super().check()
@ -85,13 +84,6 @@ class RewriteQuestion(Generate, ABC):
component_name = "RewriteQuestion" component_name = "RewriteQuestion"
def _run(self, history, **kwargs): 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) hist = self._canvas.get_history(4)
conv = [] conv = []
for m in hist: for m in hist: