从零搭建AI编程助手：本地代码生成与自动执行全攻略

SYSTEM_PROMPT = """你是一个专业的 Python 助手。
用户提出编程需求，你只返回可直接运行的 Python 代码，用 ```python ... ``` 包裹。
不要添加任何解释、注释或额外文字。"""

import ollama
import re

def generate_code(requirement: str) -> str:
    response = ollama.chat(
        model="qwen2.5-coder:7b",
        messages=[
            {"role": "system", "content": SYSTEM_PROMPT},
            {"role": "user", "content": requirement}
        ]
    )
    raw = response["message"]["content"]
    match = re.search(r"```python\n(.*?)\n```", raw, re.DOTALL)
    return match.group(1) if match else raw

import subprocess
import tempfile
import os

def safe_run(code: str, timeout: int = 5) -> str:
    # 静态检查危险模块
    dangerous = ["import os", "import shutil", "import socket", "eval", "exec"]
    for word in dangerous:
        if word in code:
            return "[安全检查] 检测到危险操作，已拦截。"

    with tempfile.NamedTemporaryFile(mode="w", suffix=".py", delete=False) as f:
        f.write(code)
        tmp_path = f.name
    try:
        result = subprocess.run(
            ["python", tmp_path],
            capture_output=True,
            text=True,
            timeout=timeout
        )
        return result.stdout if result.stdout else result.stderr
    except subprocess.TimeoutExpired:
        return "[超时] 代码执行超过 %s 秒" % timeout
    finally:
        os.unlink(tmp_path)

def ai_programmer(requirement: str):
    print(f"需求：{requirement}")
    code = generate_code(requirement)
    print(f"生成的代码：\n{code}")
    output = safe_run(code)
    print(f"执行结果：\n{output}")

if __name__ == "__main__":
    ai_programmer("计算斐波那契数列第10项并打印")

需求：计算斐波那契数列第10项并打印
生成的代码：
def fib(n):
    a,b=0,1
    for _ in range(n):
        a,b=b,a+b
    return a
print(fib(10))
执行结果：
55