| import re |
| import json |
| import ast |
|
|
| def extract_function_signatures(code): |
| function_defs = [] |
| pattern = r'def\s+([a-zA-Z0-9_]+)\s*\(([^)]*)\)' |
| matches = re.finditer(pattern, code) |
| |
| for match in matches: |
| func_name = match.group(1) |
| params = match.group(2) |
| function_defs.append(f"def {func_name}({params})") |
| |
| return function_defs |
|
|
| def clear_code(code): |
| if "```python" in code: |
| code = code.split("```python")[-1].replace("<|im_end|>", "").replace("<|EOT|>", "") |
| if "```" in code: |
| code = code.split("```")[0] |
| return code |
|
|
| def extract_function_calls(code): |
| calls = [] |
| pattern = r'([a-zA-Z0-9_]+)\s*\(([^)]*)\)' |
| matches = re.finditer(pattern, code) |
| |
| for match in matches: |
| func_name = match.group(1) |
| args = match.group(2) |
| calls.append(f"{func_name}({args})") |
| |
| return calls |
|
|
| def clear_json(response): |
| if type(response) is dict: |
| return response |
| elif type(response) is not str: |
| response = str(response) |
| base_response = response |
| try: |
| response = base_response.replace("\n", " ") |
| response = re.search('({.+})', response).group(0) |
| response = re.sub(r"(\w)'(\w|\s)", r"\1\\'\2", response) |
| result = ast.literal_eval(response) |
| return result |
| except (SyntaxError, NameError, AttributeError): |
| pass |
| try: |
| response = base_response.split('assistantfinal')[-1] |
| result = ast.literal_eval(response) |
| return result |
| except: |
| pass |
| try: |
| response = base_response.split('assistantfinal JSON')[-1] |
| result = ast.literal_eval(response) |
| return result |
| except: |
| pass |
| try: |
| response = base_response.split('assistantfinal json')[-1] |
| result = ast.literal_eval(response) |
| return result |
| except: |
| pass |
| try: |
| response = list(re.finditer(r'\{[\s\S]*?\}', base_response))[-1].group(0) |
| result = ast.literal_eval(response) |
| return result |
| except: |
| pass |
| return 'ERR_SYNTAX' |
|
|
