jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add Text import to ChatInput.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-26 21:56:52 -03:00
commit cb999b26bb
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/components/io/ChatInput.py
View file

@ -1,6 +1,7 @@
from typing import Optional from typing import Optional, Union
from langflow import CustomComponent from langflow import CustomComponent
from langflow.field_typing import Text
from langflow.schema import Record from langflow.schema import Record
@ -37,7 +38,7 @@ class ChatInput(CustomComponent):
message: Optional[str] = None, message: Optional[str] = None,
session_id: Optional[str] = None, session_id: Optional[str] = None,
return_record: Optional[bool] = False, return_record: Optional[bool] = False,
) -> Record: ) -> Union[Text, Record]:
if return_record: if return_record:
if isinstance(message, Record): if isinstance(message, Record):
# Update the data of the record # Update the data of the record