From 7f48fadd41107bc3c2d847bca35f36088341ed41 Mon Sep 17 00:00:00 2001
From: John Wang <takatost@users.noreply.github.com>
Date: Mon, 29 May 2023 19:10:31 +0800
Subject: [PATCH] fix: prompt template parantheses select error (#244)

---
 api/core/prompt/prompt_builder.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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