prompt_engineering.py

# prompt_engineering.py

import openai

openai.api_key = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"

def interpret_command(command: str, history: str = "") -> str:
    """
    使用 GPT 模型解析用户命令，返回修正后的 prompt
    """
    system_prompt = "你是一个角色外观设计师，请根据用户的微调需求优化提示词。保持风格不变，仅调整细节，如头发颜色、服装、表情等。"

    messages = [
        {"role": "system", "content": system_prompt},
        {"role": "user", "content": f"原始风格：{history}\n用户指令：{command}"}
    ]

    response = openai.ChatCompletion.create(
        model="gpt-4",
        messages=messages
    )
    return response["choices"][0]["message"]["content"]