From d6e261286fc339c3ae26a1f66d8a12c4d87b178e Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Wed, 5 Apr 2023 22:05:57 -0300
Subject: [PATCH] fix: added default prompt fields

---
 src/backend/langflow/template/nodes.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/backend/langflow/template/nodes.py b/src/backend/langflow/template/nodes.py
index 04329b75a..c76c56b14 100644
--- a/src/backend/langflow/template/nodes.py
+++ b/src/backend/langflow/template/nodes.py
@@ -253,12 +253,20 @@ class PromptFrontendNode(FrontendNode):
     def format_field(field: TemplateField, name: Optional[str] = None) -> None:
         # if field.field_type  == "StringPromptTemplate"
         # change it to str
+        PROMPT_FIELDS = [
+            "template",
+            "suffix",
+            "prefix",
+            "examples",
+        ]
         if field.field_type == "StringPromptTemplate" and "Message" in str(name):
             field.field_type = "prompt"
             field.multiline = True
             field.value = HUMAN_PROMPT if "Human" in field.name else SYSTEM_PROMPT
         if field.name == "template" and field.value == "":
             field.value = DEFAULT_PROMPT
+
+        if field.name in PROMPT_FIELDS:
             field.field_type = "prompt"
 
         if (