diff --git a/src/backend/base/langflow/components/inputs/ChatInput.py b/src/backend/base/langflow/components/inputs/ChatInput.py
index 150d5f2af..28cbc5529 100644
--- a/src/backend/base/langflow/components/inputs/ChatInput.py
+++ b/src/backend/base/langflow/components/inputs/ChatInput.py
@@ -2,6 +2,8 @@ from typing import Optional
 
 from langflow.base.io.chat import ChatComponent
 from langflow.schema.message import Message
+from langflow.field_typing import Text
+from typing import Union
 
 
 class ChatInput(ChatComponent):
@@ -26,7 +28,7 @@ class ChatInput(ChatComponent):
         input_value: Optional[str] = None,
         files: Optional[list[str]] = None,
         session_id: Optional[str] = None,
-    ) -> Message:
+    ) -> Union[Message, Text]:
         return super().build_with_record(
             sender=sender,
             sender_name=sender_name,