diff --git a/agent/component/message.py b/agent/component/message.py
index a193dd122..c60d4d307 100644
--- a/agent/component/message.py
+++ b/agent/component/message.py
@@ -40,7 +40,9 @@ class Message(ComponentBase, ABC):
         if kwargs.get("stream"):
             return partial(self.stream_output)
 
-        return Message.be_output(random.choice(self._param.messages))
+        res = Message.be_output(random.choice(self._param.messages))
+        self.set_output(res)
+        return res
 
     def stream_output(self):
         res = None