Update ChatInput and ChatVertex classes
This commit is contained in:
parent
0b53d36e76
commit
21bff66c1a
2 changed files with 13 additions and 6 deletions
|
|
@ -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 = ""
|
||||||
|
|
|
||||||
|
|
@ -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)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue