update pattern

microsoft · Apr 25, 2024 · 1c88c1d · 1c88c1d
1 parent 803a125
commit 1c88c1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/promptflow-tools/promptflow/tools/common.py b/src/promptflow-tools/promptflow/tools/common.py
@@ -241,7 +241,7 @@ def try_parse_tool_call_id_and_content(role_prompt):
 def try_parse_tool_calls(role_prompt):
     # customer can add ## in front of tool_calls for markdown highlight.
     # and we still support tool_calls without ## prefix for backward compatibility.
-    pattern = r"\n*#{0,2}\s*tool_calls:\n*\s*(\[.*?\])"
+    pattern = r"\n*#{0,2}\s*tool_calls\s*:\s*\n+\s*(\[.*?\])"
     match = re.search(pattern, role_prompt, re.DOTALL)
     if match:
         try: