other template fixes
This commit is contained in:
parent
17c58fd70d
commit
f0ba11c840
2 changed files with 9 additions and 0 deletions
|
|
@ -197,6 +197,7 @@ class FrontendNode(BaseModel):
|
||||||
"examples",
|
"examples",
|
||||||
"code",
|
"code",
|
||||||
"headers",
|
"headers",
|
||||||
|
"description",
|
||||||
}
|
}
|
||||||
|
|
||||||
# Replace dict type with str
|
# Replace dict type with str
|
||||||
|
|
|
||||||
|
|
@ -77,6 +77,12 @@ class PromptTemplateNode(FrontendNode):
|
||||||
def to_dict(self):
|
def to_dict(self):
|
||||||
return super().to_dict()
|
return super().to_dict()
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def format_field(field: TemplateField, name: Optional[str] = None) -> None:
|
||||||
|
FrontendNode.format_field(field, name)
|
||||||
|
if field.name == "examples":
|
||||||
|
field.advanced = False
|
||||||
|
|
||||||
|
|
||||||
class PythonFunctionNode(FrontendNode):
|
class PythonFunctionNode(FrontendNode):
|
||||||
name: str = "PythonFunction"
|
name: str = "PythonFunction"
|
||||||
|
|
@ -486,6 +492,7 @@ class PromptFrontendNode(FrontendNode):
|
||||||
"suffix",
|
"suffix",
|
||||||
"prefix",
|
"prefix",
|
||||||
"examples",
|
"examples",
|
||||||
|
"format_instructions",
|
||||||
]
|
]
|
||||||
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"
|
||||||
|
|
@ -496,6 +503,7 @@ class PromptFrontendNode(FrontendNode):
|
||||||
|
|
||||||
if field.name in PROMPT_FIELDS:
|
if field.name in PROMPT_FIELDS:
|
||||||
field.field_type = "prompt"
|
field.field_type = "prompt"
|
||||||
|
field.advanced = False
|
||||||
|
|
||||||
if (
|
if (
|
||||||
"Union" in field.field_type
|
"Union" in field.field_type
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue