| import re | |
| # def extract_sql(response): | |
| # match = re.search(r"```sql\s+(.*?)\s+```", response, re.DOTALL | re.IGNORECASE) | |
| # return match.group(1) if match else None | |
| def extract_sql(response): | |
| matches = re.findall(r"```sql\s+(.*?)\s+```", response, re.DOTALL | re.IGNORECASE) | |
| if matches: | |
| return matches[0].strip() | |
| return None | |
| def is_sql(response): | |
| return bool(re.search(r"```sql\s+.*?```", response, re.DOTALL | re.IGNORECASE)) | |