From cb999b26bb63b304120425f601be2ada07f08f38 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 26 Feb 2024 21:56:52 -0300 Subject: [PATCH] Add Text import to ChatInput.py --- src/backend/langflow/components/io/ChatInput.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/components/io/ChatInput.py b/src/backend/langflow/components/io/ChatInput.py index 4d8fc509c..6d96a6b96 100644 --- a/src/backend/langflow/components/io/ChatInput.py +++ b/src/backend/langflow/components/io/ChatInput.py @@ -1,6 +1,7 @@ -from typing import Optional +from typing import Optional, Union from langflow import CustomComponent +from langflow.field_typing import Text from langflow.schema import Record @@ -37,7 +38,7 @@ class ChatInput(CustomComponent): message: Optional[str] = None, session_id: Optional[str] = None, return_record: Optional[bool] = False, - ) -> Record: + ) -> Union[Text, Record]: if return_record: if isinstance(message, Record): # Update the data of the record