diff --git a/api/core/prompt/prompt_builder.py b/api/core/prompt/prompt_builder.py index cbe41576f1..164bf6375b 100644 --- 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