cassanof 10 months ago
parent 3dec2edc92
commit 1d70a0026f

@ -5,9 +5,16 @@ from typing import Optional
def parse_code_block(string: str, lang: str) -> Optional[str]:
code_pattern = fr"```{lang}\n(.*?)\n```"
match = re.search(code_pattern, string, re.DOTALL)
if match:
return match.group(1)
else:
generic_code_pattern = r"```\n(.*?)\n```"
match = re.search(generic_code_pattern, string, re.DOTALL)
if match:
return match.group(1)
return parse_first_func(string, lang)

Loading…
Cancel
Save