From b7a63154c3bd6be34bb2dcc8f3fff05ba7c326ba Mon Sep 17 00:00:00 2001
From: Rodrigo <rodrigosilvanader@gmail.com>
Date: Tue, 11 Jun 2024 15:27:57 -0300
Subject: [PATCH] chore: Update ChatInput.py to include Text import and modify
 return type

---
 src/backend/base/langflow/components/inputs/ChatInput.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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,