| import subprocess | |
| import tempfile | |
| # ------------------------- | |
| # ์ฝ๋ ์คํ๊ธฐ | |
| # ------------------------- | |
| def run_code(code): | |
| try: | |
| with tempfile.NamedTemporaryFile(suffix=".py", delete=False) as f: | |
| f.write(code.encode("utf-8")) | |
| path = f.name | |
| result = subprocess.run( | |
| ["python", path], | |
| capture_output=True, | |
| timeout=3 | |
| ) | |
| return result.stdout.decode(), result.stderr.decode() | |
| except Exception as e: | |
| return "", str(e) | |
| # ------------------------- | |
| # ์ฝ๋ ์์ ๋ฃจํ | |
| # ------------------------- | |
| def fix_loop(model_generate, prompt, max_iter=3): | |
| code = model_generate(prompt) | |
| for i in range(max_iter): | |
| out, err = run_code(code) | |
| if err == "": | |
| return code # ์ฑ๊ณต | |
| fix_prompt = f""" | |
| ๋ค์ ์ฝ๋ ์ค๋ฅ๋ฅผ ์์ ํด์ฃผ์ญ์์ค. | |
| ์ฝ๋: | |
| {code} | |
| ์๋ฌ: | |
| {err} | |
| ํ๊ตญ์ด ์กด๋๋ง๋ก ์์ ๋ ์ฝ๋๋ง ์ถ๋ ฅํ์ญ์์ค. | |
| """ | |
| code = model_generate(fix_prompt) | |
| return code | |