From 64fb056ba906bbf6512c5a4ce10543243ef7f25b Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Tue, 4 Apr 2023 21:45:45 -0300 Subject: [PATCH] fix: removed chat agents for now --- src/backend/langflow/template/nodes.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/template/nodes.py b/src/backend/langflow/template/nodes.py index dae282163..acfc5c2d2 100644 --- a/src/backend/langflow/template/nodes.py +++ b/src/backend/langflow/template/nodes.py @@ -7,6 +7,12 @@ from langflow.template.base import FrontendNode, Template, TemplateField from langflow.template.constants import DEFAULT_PROMPT, HUMAN_PROMPT, SYSTEM_PROMPT from langflow.utils.constants import DEFAULT_PYTHON_FUNCTION +NON_CHAT_AGENTS = { + agent_type: agent_class + for agent_type, agent_class in loading.AGENT_TO_CLASS.items() + if "chat" not in agent_type.value +} + class BasePromptFrontendNode(FrontendNode): name: str @@ -176,8 +182,8 @@ class InitializeAgentNode(FrontendNode): is_list=True, show=True, multiline=False, - options=list(loading.AGENT_TO_CLASS.keys()), - value=list(loading.AGENT_TO_CLASS.keys())[0], + options=list(NON_CHAT_AGENTS.keys()), + value=list(NON_CHAT_AGENTS.keys())[0], name="agent", ), TemplateField( @@ -247,7 +253,7 @@ class PromptFrontendNode(FrontendNode): def format_field(field: TemplateField, name: Optional[str] = None) -> None: # if field.field_type == "StringPromptTemplate" # change it to str - if field.field_type == "StringPromptTemplate" and "Message" in name: + if field.field_type == "StringPromptTemplate" and "Message" in str(name): field.field_type = "str" field.multiline = True field.value = HUMAN_PROMPT if "Human" in field.name else SYSTEM_PROMPT