fix: prompt template parantheses select error

3 years ago · f0cd7c59ff
parent af2138e8b8
commit f0cd7c59ff
1 changed files with 2 additions and 2 deletions
--- a/api/core/prompt/prompt_builder.py
+++ b/api/core/prompt/prompt_builder.py
@ -32,6 +32,6 @@ class PromptBuilder:

    @classmethod
    def process_template(cls, template: str):
-        processed_template = re.sub(r'\{(.+?)\}', r'\1', template)
-        processed_template = re.sub(r'\{\{(.+?)\}\}', r'{\1}', processed_template)
+        processed_template = re.sub(r'\{([a-zA-Z_]\w+?)\}', r'\1', template)
+        processed_template = re.sub(r'\{\{([a-zA-Z_]\w+?)\}\}', r'{\1}', processed_template)
        return processed_template