fix bug, agent invoke can not get params from begin (#4390)

### What problem does this PR solve? fix bug, agent invoke can not get params from begin ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) Co-authored-by: wangrui <[email protected]>
infiniflow · Jan 7, 2025 · 01a122d · 01a122d
1 parent 8ec392a
commit 01a122d
Showing 1 changed file with 16 additions and 6 deletions.
diff --git a/agent/component/invoke.py b/agent/component/invoke.py
@@ -50,12 +50,22 @@ def _run(self, history, **kwargs):
         args = {}
         for para in self._param.variables:
             if para.get("component_id"):
-                cpn = self._canvas.get_component(para["component_id"])["obj"]
-                if cpn.component_name.lower() == "answer":
-                    args[para["key"]] = self._canvas.get_history(1)[0]["content"]
-                    continue
-                _, out = cpn.output(allow_partial=False)
-                args[para["key"]] = "\n".join(out["content"])
+                if '@' in para["component_id"]:
+                    component = para["component_id"].split('@')[0]
+                    field = para["component_id"].split('@')[1]
+                    cpn = self._canvas.get_component(component)["obj"]
+                    for param in cpn._param.query:
+                        if param["key"] == field:
+                            if "value" in param:
+                                args[para["key"]] = param["value"]
+                else:
+                    cpn = self._canvas.get_component(para["component_id"])["obj"]
+                    if cpn.component_name.lower() == "answer":
+                        args[para["key"]] = self._canvas.get_history(1)[0]["content"]
+                        continue
+                    _, out = cpn.output(allow_partial=False)
+                    if not out.empty:
+                        args[para["key"]] = "\n".join(out["content"])
             else:
                 args[para["key"]] = para["value"]