From 4e5f92f01bb99696b1992850a33d1ae799b7b426 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Tue, 24 Dec 2024 17:58:11 +0800
Subject: [PATCH] Fix interface as input variable for component. (#4212)

### What problem does this PR solve?

#4108

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 agent/component/base.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/agent/component/base.py b/agent/component/base.py
index 663ac2744..592515a89 100644
--- a/agent/component/base.py
+++ b/agent/component/base.py
@@ -477,6 +477,13 @@ class ComponentBase(ABC):
                             assert False, f"Can't find parameter '{key}' for {cpn_id}"
                         continue
 
+                    if q["component_id"].lower().find("answer") == 0:
+                        for r, c in self._canvas.history[::-1]:
+                            if r == "user":
+                                self._param.inputs.append(pd.DataFrame([{"content": c, "component_id": q["component_id"]}]))
+                                break
+                        continue
+
                     outs.append(self._canvas.get_component(q["component_id"])["obj"].output(allow_partial=False)[1])
                     self._param.inputs.append({"component_id": q["component_id"],
                                                "content": "\n".join(