Преглед на файлове

fix: prompt template parantheses select error (#244)

John Wang преди 1 година
родител
ревизия
7f48fadd41
променени са 1 файла, в които са добавени 2 реда и са изтрити 2 реда
  1. 2 2
      api/core/prompt/prompt_builder.py

+ 2 - 2
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