从 0 到 1 搭建 AI 自动写作助手，Python + API 完整教程

你是否也想过：能不能一键让 AI 帮我写周报、写文案、写邮件？今天我就带你从 0 开始，用 Python 搭建一个属于自己的 AI 写作助手。

全程只需 50 行代码，复制就能用。

一、准备工作

首先安装依赖：

pip install openai

然后准备一个 API Key（随便哪个平台的都行，只要是 OpenAI 兼容接口）。

二、最简单的调用（10 行代码）

from openai import OpenAI

client = OpenAI(
    api_key="sk-your-key",
    base_url="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 生成的文案。就这么简单。

三、加入流式输出（打字机效果）

流式输出让 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 一个字一个字地输出诗歌，效果很酷。

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

下面是一个完整的交互式程序，支持多轮对话、切换模型：

import os
from openai import OpenAI

API_KEY = "sk-your-key"
BASE_URL = "https://api-aiapi.cn/v1"

client = OpenAI(api_key=API_KEY, base_url=BASE_URL)

MODELS = {
    "1": "deepseek-v4-flash",
    "2": "gpt-4o",
    "3": "claude-sonnet-4",
    "4": "qwen-plus"
}

print("=== AI 写作助手 ===")
print("可选模型：")
for k, v in MODELS.items():
    print(f"  {k}. {v}")

model_key = input("\n选择模型 (1-4，默认1): ").strip() or "1"
model = MODELS.get(model_key, "deepseek-v4-flash")

system_prompt = input("设定角色（直接回车=通用助手）: ").strip()
messages = [{"role": "system", "content": system_prompt or "你是一个有用的助手。"}]

print("\n开始对话（输入 quit 退出）\n")

while True:
    user_input = input("你: ")
    if user_input.lower() == "quit":
        break
    
    messages.append({"role": "user", "content": user_input})
    
    stream = client.chat.completions.create(
        model=model, messages=messages, stream=True
    )
    
    print("AI: ", end="", flush=True)
    reply = ""
    for chunk in stream:
        content = chunk.choices[0].delta.content
        if content:
            print(content, end="", flush=True)
            reply += content
    print("\n")
    
    messages.append({"role": "assistant", "content": reply})

运行之后你就可以：

选择不同模型（DeepSeek/GPT/Claude/通义千问）
设定角色（"你是一个文案专家""你是一个翻译官"）
多轮对话，AI 能记住上下文
流式输出，体验丝滑

五、进阶：一键切换模型对比效果

你可以写个脚本，用同一个问题让不同模型回答，对比效果：

def ask_model(model, prompt):
    client = OpenAI(api_key=API_KEY, base_url=BASE_URL)
    response = client.chat.completions.create(
        model=model,
        messages=[{"role": "user", "content": prompt}]
    )
    return response.choices[0].message.content

prompt = "用一句话解释什么是 API"

models = ["deepseek-v4-flash", "gpt-4o", "claude-sonnet-4", "qwen-plus"]
for m in models:
    result = ask_model(m, prompt)
    print(f"\n【{m}】\n{result}\n{'-'*30}")

六、让 AI 写作更好用的小技巧

1. 写好 System Prompt

这是最关键的一步。好的角色设定直接决定输出质量：

# 差的 prompt
"写一篇产品介绍"

# 好的 prompt
"你是一个资深市场营销专家，擅长用简洁有力的语言打动客户。请为我们的 AI API 平台写一篇产品介绍，突出'一个 Key 调所有模型'的核心卖点，语气要专业但不枯燥。"

2. 控制输出格式

"请用 Markdown 格式输出，包含标题、列表和代码块。"

3. 控制输出长度

"控制在 300 字以内，分三段，每段一个小标题。"

4. 多轮迭代优化

"第一版挺好，但语气再活泼一点，多加点例子。"

七、总结

从 10 行代码到完整的交互式写作助手，你已经掌握了 AI 应用开发的核心技能：

✅ OpenAI 库的基本调用
✅ 流式输出实现打字机效果
✅ 多轮对话上下文管理
✅ 多模型切换和对比
✅ Prompt Engineering 基础技巧

有了这些基础，你可以进一步开发：自动周报生成器、小红书文案助手、翻译工具、代码审查机器人……一切皆有可能。

一个 Key，调用全球 AI 模型。

👉 https://api-aiapi.cn —— 注册即用，¥10 起充

* 本文由慧朗AI（AI·API）团队原创，代码均为实测可运行。