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

make variables access robuster (#3406)

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


### Type of change

- [x] Performance Improvement
This commit is contained in:
Kevin Hu 2024-11-14 18:28:41 +08:00 committed by GitHub
parent e4c4fdabbd
commit 30c1f7ee29
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
agent/component/base.py
View File

@ -399,7 +399,7 @@ class ComponentBase(ABC):
self._param.check() self._param.check()
def get_dependent_components(self): def get_dependent_components(self):
cpnts = [para["component_id"] for para in self._param.query if para["component_id"].lower().find("answer") < 0] cpnts = [para["component_id"] for para in self._param.query if para.get("component_id") and para["component_id"].lower().find("answer") < 0]
return cpnts return cpnts
def run(self, history, **kwargs): def run(self, history, **kwargs):

2
agent/component/generate.py
View File

@ -63,7 +63,7 @@ class Generate(ComponentBase):
component_name = "Generate" component_name = "Generate"
def get_dependent_components(self): def get_dependent_components(self):
cpnts = [para["component_id"] for para in self._param.parameters if para["component_id"].lower().find("answer") < 0] cpnts = [para["component_id"] for para in self._param.parameters if para.get("component_id") and para["component_id"].lower().find("answer") < 0]
return cpnts return cpnts
def set_cite(self, retrieval_res, answer): def set_cite(self, retrieval_res, answer):