pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make Categorize see more chat hisotry. (#4538)

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

#4521

### Type of change
- [x] Performance Improvement
This commit is contained in:
Kevin Hu
2025-01-20 11:57:56 +08:00
committed by GitHub
parent 2962284c79
commit 367babda2f
2 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
agent/component/base.py
View File

@ -482,11 +482,12 @@ class ComponentBase(ABC):
continue
if q["component_id"].lower().find("answer") == 0:
for r, c in self._canvas.history[::-1]:
if r == "user":
self._param.inputs.append({"content": c, "component_id": q["component_id"]})
outs.append(pd.DataFrame([{"content": c}]))
break
txt = []
for r, c in self._canvas.history[::-1][:self._param.message_history_window_size]:
txt.append(f"{r.upper()}: {c}")
txt = "\n".join(txt)
self._param.inputs.append({"content": txt, "component_id": q["component_id"]})
outs.append(pd.DataFrame([{"content": txt}]))
continue
outs.append(self._canvas.get_component(q["component_id"])["obj"].output(allow_partial=False)[1])