mirror of
https://github.com/VinciGit00/Scrapegraph-ai.git
synced 2026-06-28 21:01:55 +08:00
15 lines
301 B
Python
15 lines
301 B
Python
"""
|
|
This utility function extracts the code from a given string.
|
|
"""
|
|
import re
|
|
|
|
def extract_code(code: str) -> str:
|
|
"""
|
|
Module for extracting code
|
|
"""
|
|
pattern = r'```(?:python)?\n(.*?)```'
|
|
|
|
match = re.search(pattern, code, re.DOTALL)
|
|
|
|
return match.group(1) if match else code
|