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

fix: added default prompt fields

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-05 22:05:57 -03:00
commit d6e261286f
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/template/nodes.py
View file

@ -253,12 +253,20 @@ class PromptFrontendNode(FrontendNode):
def format_field(field: TemplateField, name: Optional[str] = None) -> None: def format_field(field: TemplateField, name: Optional[str] = None) -> None:
# if field.field_type == "StringPromptTemplate" # if field.field_type == "StringPromptTemplate"
# change it to str # change it to str
PROMPT_FIELDS = [
"template",
"suffix",
"prefix",
"examples",
]
if field.field_type == "StringPromptTemplate" and "Message" in str(name): if field.field_type == "StringPromptTemplate" and "Message" in str(name):
field.field_type = "prompt" field.field_type = "prompt"
field.multiline = True field.multiline = True
field.value = HUMAN_PROMPT if "Human" in field.name else SYSTEM_PROMPT field.value = HUMAN_PROMPT if "Human" in field.name else SYSTEM_PROMPT
if field.name == "template" and field.value == "": if field.name == "template" and field.value == "":
field.value = DEFAULT_PROMPT field.value = DEFAULT_PROMPT
if field.name in PROMPT_FIELDS:
field.field_type = "prompt" field.field_type = "prompt"
if ( if (