你是否也想过：能不能一键让 AI 帮我写周报、写文案、写邮件？今天我就带你从 0 开始，用 Python 搭建一个属于自己的 AI 写作助手。全程只需 50 行代码，复制就能用。
一、准备工作
安装依赖：pip install openai。准备一个 API Key，随便哪个平台都行，只要是 OpenAI 兼容接口。
二、最简单的调用（10 行代码）
from openai import OpenAI client = OpenAI(apikey="sk-your-key", baseurl="https://api-aiapi.cn/v1") response = client.chat.completions.create(model="deepseek-v4-flash", messages=[{"role": "system", "content": "你是一个专业的文案助手。"}, {"role": "user", "content": "帮我写一段产品介绍，200字以内。"}]) print(response.choices[0].message.content)
运行即可看到 AI 生成的文案。就这么简单。
三、流式输出（打字机效果）
stream = client.chat.completions.create(model="deepseek-v4-flash", messages=[{"role": "user", "content": "写一首关于夏天的诗"}], stream=True) for chunk in stream:     content
= chunk.choices[0].delta.content
    if content:
        print(content, end="", flush=True)

运行后你会看到 AI 一个字一个字地输出诗歌，效果很酷。

四、完整版：交互式写作助手

from openai import OpenAI
API_KEY = "sk-your-key"
BASE_URL = "https://api-aiapi.cn/v1"
client = OpenAI(apikey=APIKEY, baseurl=BASEURL)
models = {"1": "deepseek-v4-flash", "2": "gpt-4o", "3": "claude-sonnet-4", "4": "qwen-plus"}
print("可选模型：", ", ".join(f"{k}.{v}" for k,v in models.items()))
model = models.get(input("选择模型: ").strip() or "1", "deepseek-v4-flash")
role = input("设定角色: ").strip() or "你是一个有用的助手。"
messages = [{"role": "system", "content": role}]
while True:
    user = input("\n你: ")
    if user.lower() == "quit": break
    messages.append({"role": "user", "content": user})
    print("AI: ", end="", flush=True)
    reply = ""
    for chunk
in client.chat.completions.create(model=model, messages=messages, stream=True):         c = chunk.choices[0].delta.content         if c: print(c, end="", flush=True); reply += c     messages.append({"role": "assistant", "content": reply})     print()
运行之后你可以：选择不同模型（DeepSeek/GPT/Claude/通义千问）、设定角色、多轮对话、流式输出。
五、进阶技巧
写好 System Prompt 是最关键的一步。差的 prompt："写一篇产品介绍"。好的 prompt："你是一个资深市场营销专家，擅长用简洁有力的语言打动客户。请为我们的 AI API 平台写一篇产品介绍，突出'一个 Key 调所有模型'的核心卖点，语气要专业但不枯燥。"
还可以控制输出格式（"请用 Markdown 格式输出"）、控制长度（"300字以内"）、多轮迭代优化（"语气再活泼一点"）。
六、总结
从 10 行代码到完整的交互式写作助手，你已经掌握了 AI 应用开发的核心技能。有了这些基础，你可以进一步开发：自动周报生成器、小红书文案助手、翻译工具、代码审查机器人……一切皆有可能。
一个 Key，调用全球 AI 模型。
本文由 AI·API.cn 团队原创，代码均为实测可运行。