gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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()
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
def run(self, history, **kwargs):

2
agent/component/generate.py
View file

@ -63,7 +63,7 @@ class Generate(ComponentBase):
component_name = "Generate"
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
def set_cite(self, retrieval_res, answer):