From 674d342761d7f03df5c58854f9c9309fb5cf40f2 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Fri, 27 Sep 2024 20:20:36 +0800 Subject: [PATCH] refine get_input (#2630) ### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- agent/component/base.py | 4 ++-- agent/component/concentrator.py | 7 +------ 2 files changed, 3 insertions(+), 8 deletions(-) diff --git a/agent/component/base.py b/agent/component/base.py index 103325016b..0ae965a8a2 100644 --- a/agent/component/base.py +++ b/agent/component/base.py @@ -444,7 +444,7 @@ def get_input(self): if DEBUG: print(self.component_name, reversed_cpnts[::-1]) for u in reversed_cpnts[::-1]: - if self.get_component_name(u) in ["switch"]: continue + if self.get_component_name(u) in ["switch", "concentrator"]: continue if self.component_name.lower() == "generate" and self.get_component_name(u) == "retrieval": o = self._canvas.get_component(u)["obj"].output(allow_partial=False)[1] if o is not None: @@ -472,7 +472,7 @@ def get_input(self): if "content" in df: df = df.drop_duplicates(subset=['content']).reset_index(drop=True) return df - return pd.DataFrame() + return pd.DataFrame(self._canvas.get_history(3)[-1:]) def get_stream_input(self): reversed_cpnts = [] diff --git a/agent/component/concentrator.py b/agent/component/concentrator.py index ee405b3dcc..8ebaad5077 100644 --- a/agent/component/concentrator.py +++ b/agent/component/concentrator.py @@ -33,9 +33,4 @@ class Concentrator(ComponentBase, ABC): component_name = "Concentrator" def _run(self, history, **kwargs): - input = self.get_input() - ans = " - ".join(input["content"]) if "content" in input else "" - if not ans: - return Concentrator.be_output("") - - return Concentrator.be_output(input) + return Concentrator.be_output("") \ No newline at end of file