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

fix: add prompt as field_type for validation

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-05 21:53:32 -03:00
commit a344ef2986
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -254,11 +254,12 @@ class PromptFrontendNode(FrontendNode):
# if field.field_type == "StringPromptTemplate" # if field.field_type == "StringPromptTemplate"
# change it to str # change it to str
if field.field_type == "StringPromptTemplate" and "Message" in str(name): if field.field_type == "StringPromptTemplate" and "Message" in str(name):
field.field_type = "str" 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
field.field_type = "prompt"
if ( if (
"Union" in field.field_type "Union" in field.field_type