Update ChatInput and ChatVertex classes

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-08 18:32:08 -03:00
commit 21bff66c1a
2 changed files with 13 additions and 6 deletions

View file

@ -8,13 +8,19 @@ class ChatInput(CustomComponent):
display_name = "Chat Input" display_name = "Chat Input"
description = "Used to get user input from the chat." description = "Used to get user input from the chat."
field_config = { def build_config(self):
"code": { return {
"show": False, "message": {"input_types": ["Text"]},
"sender": {"options": ["Machine", "User"], "display_name": "Sender Type"},
"sender_name": {"display_name": "Sender Name"},
} }
}
def build(self, message: Optional[str] = None) -> Text: def build(
self,
sender: Optional[str] = "User",
sender_name: Optional[str] = "You",
message: Optional[str] = None,
) -> Text:
self.repr_value = message self.repr_value = message
if not message: if not message:
message = "" message = ""

View file

@ -2,6 +2,7 @@ import ast
from typing import Callable, Dict, List, Optional, Union from typing import Callable, Dict, List, Optional, Union
from langchain_core.messages import AIMessage from langchain_core.messages import AIMessage
from langflow.graph.utils import UnbuiltObject, flatten_list from langflow.graph.utils import UnbuiltObject, flatten_list
from langflow.graph.vertex.base import StatefulVertex, StatelessVertex from langflow.graph.vertex.base import StatefulVertex, StatelessVertex
from langflow.interface.utils import extract_input_variables_from_prompt from langflow.interface.utils import extract_input_variables_from_prompt
@ -321,7 +322,7 @@ class ChatVertex(StatelessVertex):
async def _run(self, *args, **kwargs): async def _run(self, *args, **kwargs):
if self.is_interface_component: if self.is_interface_component:
if self.vertex_type == "ChatOutput": if self.vertex_type in ["ChatOutput", "ChatInput"]:
artifacts = None artifacts = None
sender = self.params.get("sender", None) sender = self.params.get("sender", None)
sender_name = self.params.get("sender_name", None) sender_name = self.params.get("sender_name", None)