diff --git a/agent/component/base.py b/agent/component/base.py index 103325016bd..0ae965a8a22 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 ee405b3dcc6..8ebaad5077c 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