用ZeusAPI搭建AI写作助手:从零到上线
随着大语言模型能力的快速提升,越来越多的开发者开始考虑将AI写作能力集成到自己的产品中。本文将以一个实际可运行的AI写作助手为例,带你完整走一遍从API接入到功能上线的全流程。
为什么选择API方式而不是直接用现成产品
现成的AI写作工具虽然开箱即用,但定制空间有限——你无法控制提示词逻辑、无法与自己的用户系统集成,也无法根据业务场景调整模型行为。通过API自建,你可以:
- 完全掌控系统提示词,针对特定写作场景精细调优
- 与现有用户账户、权限体系无缝对接
- 按实际用量付费,避免闲置浪费
- 灵活切换底层模型,比如在成本敏感场景用 Claude Haiku,在质量优先场景换 GPT-4o
API服务选型:ZeusAPI的成本优势
直接向 OpenAI 或 Anthropic 申请 API 需要境外信用卡,充值门槛高,汇率损耗也不小。这里推荐使用 ZeusAPI,它聚合了 Claude、GPT-4 等主流模型,最大的亮点是1元人民币等值1美元额度,没有汇率折扣,对国内开发者非常友好。
注册后充值10元即可开始测试,调用方式与 OpenAI 官方接口完全兼容,现有代码几乎不需要改动。
项目结构设计
我们要搭建的写作助手具备以下核心功能:根据用户输入的主题和风格要求生成文章草稿,支持续写和改写,并保留多轮对话上下文。技术栈选择 Python + FastAPI 作为后端,前端用简单的 HTML + JavaScript 快速验证。
项目目录结构如下:
writing-assistant/
├── main.py # FastAPI 主程序
├── prompts.py # 提示词模板管理
├── requirements.txt
└── static/
└── index.html # 前端页面
后端接口实现
安装依赖:
pip install fastapi uvicorn openai
核心代码(main.py):
from fastapi import FastAPI
from fastapi.responses import StreamingResponse
from pydantic import BaseModel
from openai import OpenAI
app = FastAPI()
# 将 base_url 指向 ZeusAPI 端点,api_key 填写你的 ZeusAPI Key
client = OpenAI(
api_key="YOUR_ZEUSAPI_KEY",
base_url="https://api.aitoolsforsellers.com/v1"
)
class WriteRequest(BaseModel):
topic: str
style: str = "专业"
length: str = "800字左右"
history: list = []
def build_system_prompt(style: str, length: str) -> str:
return f"""你是一位资深内容创作者,擅长撰写{style}风格的文章。
请根据用户提供的主题,生成结构清晰、逻辑连贯的文章。
目标字数:{length}。
输出格式:直接给出正文,不需要额外说明。"""
@app.post("/write")
async def write_article(req: WriteRequest):
messages = [{"role": "system", "content": build_system_prompt(req.style, req.length)}]
messages += req.history
messages.append({"role": "user", "content": f"请写一篇关于"{req.topic}"的文章"})
def stream_response():
with client.chat.completions.stream(
model="gpt-4o",
messages=messages,
) as stream:
for text in stream.text_stream:
yield text
return StreamingResponse(stream_response(), media_type="text/plain")
启动服务:
uvicorn main:app --reload --port 8000
前端页面与流式展示
使用 fetch 的流式读取配合 ReadableStream,可以实现打字机效果:
async function generateArticle() {
const topic = document.getElementById('topic').value;
const output = document.getElementById('output');
output.textContent = '';
const response = await fetch('/write', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({ topic, style: '专业', length: '800字左右' })
});
const reader = response.body.getReader();
const decoder = new TextDecoder();
while (true) {
const { done, value } = await reader.read();
if (done) break;
output.textContent += decoder.decode(value);
}
}
实际效果与成本测算
以生成一篇800字文章为例,使用 GPT-4o 模型,输入约300 tokens(系统提示词 + 用户指令),输出约1100 tokens,单次调用成本约为 0.006元人民币。也就是说,充值100元可以支撑约16,000次完整生成请求,对于个人项目或小团队验证阶段来说完全够用。
如果对成本更敏感,可以将模型切换为 Claude Haiku,同等任务的费用还能再降低约70%,且响应速度更快,非常适合高频调用场景。
上线前的几个关键检查点
- API Key 安全:不要将 Key 硬编码在前端或提交到 Git 仓库,使用环境变量管理
- 请求频率限制:为每个用户添加速率限制,防止恶意刷量
- 错误处理:对网络超时、模型过载等异常做好降级提示
- 内容过滤:根据业务需要在系统提示词中加入内容规范约束
立即开始构建
整个项目从零到跑通核心功能,实际编码时间不超过两小时。AI写作助手只是一个起点——同样的架构可以快速扩展为翻译工具、代码注释生成器、客服问答系统等。
如果你想跳过繁琐的海外支付流程,直接用人民币低门槛体验 Claude 和 GPT-4 的完整能力,可以前往 ZeusAPI 官网注册账号,充值10元即可开始调用,1元等值1美元额度,接口完全兼容 OpenAI 标准,现有项目改一行 base_url 就能迁移过来。