From 30c1f7ee29e0b55f177030de829a3592bf06d89b Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Thu, 14 Nov 2024 18:28:41 +0800 Subject: [PATCH] make variables access robuster (#3406) ### What problem does this PR solve? ### Type of change - [x] Performance Improvement --- agent/component/base.py | 2 +- agent/component/generate.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/agent/component/base.py b/agent/component/base.py index 2eee4a25f..9af0b546e 100644 --- a/agent/component/base.py +++ b/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): diff --git a/agent/component/generate.py b/agent/component/generate.py index ee58fb87e..bdcb90bcd 100644 --- a/agent/component/generate.py +++ b/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):