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

Add 'as_record' parameter to ChatOutput class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-19 19:00:51 -03:00
commit f417378779
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/components/io/ChatOutput.py
View file

@ -1,4 +1,4 @@
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.field_typing import Text
@ -28,6 +28,10 @@ class ChatOutput(CustomComponent):
"info": "Session ID of the chat history.", "info": "Session ID of the chat history.",
"input_types": ["Text"], "input_types": ["Text"],
}, },
"as_record": {
"display_name": "As Record",
"info": "If true, the message will be returned as a Record.",
},
} }
def build( def build(
@ -37,7 +41,7 @@ class ChatOutput(CustomComponent):
session_id: Optional[str] = None, session_id: Optional[str] = None,
message: Optional[str] = None, message: Optional[str] = None,
as_record: Optional[bool] = False, as_record: Optional[bool] = False,
) -> Text: ) -> Union[Text, Record]:
self.status = message self.status = message
if as_record: if as_record:
if isinstance(message, Record): if isinstance(message, Record):