fix: DIA-1629: Enforce template fields in refined prompt (#292)

Co-authored-by: nik <[email protected]> Co-authored-by: pakelley <[email protected]>
HumanSignal · Jan 15, 2025 · 7549f4f · 7549f4f
1 parent b21e909
commit 7549f4f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 611 deletions.
diff --git a/adala/utils/parse.py b/adala/utils/parse.py
@@ -111,6 +111,9 @@ class TemplateChunks(TypedDict):
     start: int
     end: int
     type: str
+
+
+match_fields_regex = re.compile(r"(?<!\{)\{([a-zA-Z0-9_]+)\}(?!})")
 
 
 def parse_template(string, include_texts=True) -> List[TemplateChunks]:
@@ -135,7 +138,7 @@ def parse_template(string, include_texts=True) -> List[TemplateChunks]:
     chunks: List[TemplateChunks] = []
     last_index = 0
 
-    for match in re.finditer(r"(?<!\{)\{(.*?)\}(?!})", string):
+    for match in match_fields_regex.finditer(string):
         # for match in re.finditer(r'\{(.*?)\}', string):
         # Text before field
         if last_index < match.start() and include_texts: