fix: prompt template parantheses select error (#244)

This commit is contained in:
John Wang
2023-05-29 19:10:31 +08:00
committed by GitHub
parent af2138e8b8
commit 7f48fadd41

View File

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