Преглед изворни кода

make variables access robuster (#3406)

### What problem does this PR solve?


### Type of change

- [x] Performance Improvement
tags/v0.14.0
Kevin Hu пре 11 месеци
родитељ
комит
30c1f7ee29
No account linked to committer's email address
2 измењених фајлова са 2 додато и 2 уклоњено
  1. 1
    1
      agent/component/base.py
  2. 1
    1
      agent/component/generate.py

+ 1
- 1
agent/component/base.py Прегледај датотеку

@@ -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 Прегледај датотеку

@@ -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…
Откажи
Сачувај