Browse Source

make variables access robuster (#3406)

### What problem does this PR solve?


### Type of change

- [x] Performance Improvement
tags/v0.14.0
Kevin Hu 11 months ago
parent
commit
30c1f7ee29
No account linked to committer's email address
2 changed files with 2 additions and 2 deletions
  1. 1
    1
      agent/component/base.py
  2. 1
    1
      agent/component/generate.py

+ 1
- 1
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):

+ 1
- 1
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):

Loading…
Cancel
Save