AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-10 21:39:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Set Output In Category Component (#8010)

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

https://github.com/infiniflow/ragflow/issues/8006
The category should work well, but the category's downstream seems to be
unable to get the upstream output.
Add the category's output as an attribute.
However, in base.py, there is logic
` if self.component_name.lower().find("switch") < 0 and
self.get_component_name(u) in ["relevant", "categorize"]:
                continue`
If goto this cases will not tried to get output from Category (but I do
not have full context about this if logic).



### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Stephen Hu 2025-06-03 15:40:16 +08:00 committed by GitHub
parent 4cd0df0567
commit f56f7a5f94
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
agent/component/categorize.py
View File

@ -99,9 +99,13 @@ class Categorize(Generate, ABC):
# If a category is found, return the category with the highest count. # If a category is found, return the category with the highest count.
if any(category_counts.values()): if any(category_counts.values()):
max_category = max(category_counts.items(), key=lambda x: x[1]) max_category = max(category_counts.items(), key=lambda x: x[1])
return Categorize.be_output(self._param.category_description[max_category[0]]["to"]) res = Categorize.be_output(self._param.category_description[max_category[0]]["to"])
self.set_output(res)
return res
return Categorize.be_output(list(self._param.category_description.items())[-1][1]["to"]) res = Categorize.be_output(list(self._param.category_description.items())[-1][1]["to"])
self.set_output(res)
return res
def debug(self, **kwargs): def debug(self, **kwargs):
df = self._run([], **kwargs) df = self._run([], **kwargs)