400 lines
11 KiB
Markdown
400 lines
11 KiB
Markdown
# AI创作系统设计文档
|
||
|
||
## 一、功能矩阵
|
||
|
||
| 功能 | 入口 | 输入 | 输出 | 配额消耗 |
|
||
|-----|------|------|------|---------|
|
||
| AI改写结局 | 结局页 | 原结局+用户指令 | 新结局文本 | 1次 |
|
||
| AI改写节点 | 章节选择页 | 原节点+用户指令 | 新节点+选项 | 1次 |
|
||
| AI续写 | 故事播放页 | 当前节点+用户指令 | 后续2-3个节点 | 2次 |
|
||
| AI创作大纲 | 创作中心 | 题材/关键词 | 标题+简介+大纲 | 1次 |
|
||
| AI完整创作 | 创作中心 | 大纲确认 | 完整故事节点树 | 5次 |
|
||
| AI润色 | 编辑器 | 原文本 | 优化后文本 | 1次 |
|
||
|
||
---
|
||
|
||
## 二、核心流程
|
||
|
||
```
|
||
┌─────────────────────────────────────────────────────────────────┐
|
||
│ AI创作完整流程 │
|
||
└─────────────────────────────────────────────────────────────────┘
|
||
|
||
用户触发 ──► 配额检查 ──► 构建Prompt ──► 调用AI ──► 解析响应 ──► 存储记录
|
||
│ │ │ │
|
||
│ ▼ ▼ ▼
|
||
│ 配额不足？ [ai_generations] 解析失败？
|
||
│ │ 记录调用 │
|
||
│ ▼ ▼
|
||
│ 引导充值/看广告 重试/降级
|
||
│
|
||
▼
|
||
展示结果 ◄── 用户操作 ──► 采纳？ ──► 写入故事表 ──► 进入审核流程
|
||
│
|
||
▼
|
||
放弃/编辑
|
||
```
|
||
|
||
---
|
||
|
||
## 三、Prompt模板设计
|
||
|
||
### 3.1 改写结局
|
||
```
|
||
[系统提示]
|
||
你是一个互动故事创作专家。根据用户的改写指令，重新创作故事结局。
|
||
要求：
|
||
- 保持原故事的世界观和人物性格
|
||
- 结局要有张力和情感冲击
|
||
- 字数控制在200-400字
|
||
- 输出格式：纯文本
|
||
|
||
[用户提示]
|
||
故事标题：{title}
|
||
故事分类：{category}
|
||
原结局名称：{ending_name}
|
||
原结局内容：{ending_content}
|
||
---
|
||
用户改写指令：{user_prompt}
|
||
---
|
||
请创作新的结局：
|
||
```
|
||
|
||
### 3.2 续写剧情
|
||
```
|
||
[系统提示]
|
||
你是一个互动故事创作专家。根据当前剧情，续写后续发展。
|
||
要求：
|
||
- 提供2-3个剧情走向选项
|
||
- 每个选项后续写1个节点内容
|
||
- 保持悬念和代入感
|
||
- 输出JSON格式
|
||
|
||
[用户提示]
|
||
故事标题：{title}
|
||
当前剧情：{current_content}
|
||
已做选择：{choices_history}
|
||
---
|
||
用户期望：{user_prompt}
|
||
---
|
||
请续写剧情，输出格式：
|
||
{
|
||
"choices": [
|
||
{"text": "选项1文本", "content": "选择后的剧情内容", "speaker": "角色名"},
|
||
{"text": "选项2文本", "content": "选择后的剧情内容", "speaker": "角色名"}
|
||
]
|
||
}
|
||
```
|
||
|
||
### 3.3 完整创作
|
||
```
|
||
[系统提示]
|
||
你是一个互动故事创作专家。根据用户提供的关键词，创作一个完整的互动故事。
|
||
要求：
|
||
- 故事有3-5个关键分支点
|
||
- 至少2个不同结局（好结局/坏结局）
|
||
- 每个节点100-200字
|
||
- 输出完整的节点树JSON
|
||
|
||
[用户提示]
|
||
题材：{genre}
|
||
关键词：{keywords}
|
||
主角设定：{protagonist}
|
||
核心冲突：{conflict}
|
||
---
|
||
请创作完整故事，输出格式：
|
||
{
|
||
"title": "故事标题",
|
||
"description": "故事简介",
|
||
"nodes": {
|
||
"start": {"content": "开头内容", "speaker": "", "choices": [...]},
|
||
"node_1": {...},
|
||
"ending_good": {"content": "好结局", "is_ending": true, "ending_type": "good"},
|
||
"ending_bad": {"content": "坏结局", "is_ending": true, "ending_type": "bad"}
|
||
}
|
||
}
|
||
```
|
||
|
||
---
|
||
|
||
## 四、API设计
|
||
|
||
### 4.1 AI改写结局
|
||
```
|
||
POST /api/ai/rewrite-ending
|
||
Request:
|
||
{
|
||
"story_id": 123,
|
||
"ending_name": "双向奔赴",
|
||
"ending_content": "原结局内容...",
|
||
"prompt": "让主角逆袭"
|
||
}
|
||
|
||
Response:
|
||
{
|
||
"code": 0,
|
||
"data": {
|
||
"generation_id": 456,
|
||
"content": "新结局内容...",
|
||
"ending_name": "双向奔赴（改写版）",
|
||
"tokens_used": 580,
|
||
"quota_remaining": 4
|
||
}
|
||
}
|
||
```
|
||
|
||
### 4.2 AI续写
|
||
```
|
||
POST /api/ai/continue
|
||
Request:
|
||
{
|
||
"story_id": 123,
|
||
"current_node_key": "node_5",
|
||
"choices_history": ["选项A", "选项B"],
|
||
"prompt": "希望有反转"
|
||
}
|
||
|
||
Response:
|
||
{
|
||
"code": 0,
|
||
"data": {
|
||
"generation_id": 457,
|
||
"choices": [
|
||
{"text": "追上去", "content": "你快步追上...", "next_key": "ai_node_1"},
|
||
{"text": "放手离开", "content": "你转身离去...", "next_key": "ai_node_2"}
|
||
],
|
||
"nodes": {
|
||
"ai_node_1": {"content": "...", "choices": [...]},
|
||
"ai_node_2": {"content": "...", "is_ending": true}
|
||
}
|
||
}
|
||
}
|
||
```
|
||
|
||
### 4.3 AI完整创作
|
||
```
|
||
POST /api/ai/create
|
||
Request:
|
||
{
|
||
"genre": "都市言情",
|
||
"keywords": "霸总,契约婚姻,追妻火葬场",
|
||
"protagonist": "独立女性设计师",
|
||
"conflict": "假结婚变真爱"
|
||
}
|
||
|
||
Response:
|
||
{
|
||
"code": 0,
|
||
"data": {
|
||
"generation_id": 458,
|
||
"draft_story_id": 789,
|
||
"title": "契约总裁的心动法则",
|
||
"description": "...",
|
||
"node_count": 12,
|
||
"ending_count": 3
|
||
}
|
||
}
|
||
```
|
||
|
||
### 4.4 配额查询
|
||
```
|
||
GET /api/user/ai-quota
|
||
|
||
Response:
|
||
{
|
||
"code": 0,
|
||
"data": {
|
||
"daily_free_remaining": 3,
|
||
"purchased_remaining": 10,
|
||
"vip_bonus": 5,
|
||
"total_available": 18,
|
||
"reset_time": "2026-03-04 00:00:00"
|
||
}
|
||
}
|
||
```
|
||
|
||
---
|
||
|
||
## 五、配额与计费
|
||
|
||
### 5.1 配额规则
|
||
| 用户类型 | 每日免费 | 购买包 | 说明 |
|
||
|---------|---------|-------|------|
|
||
| 普通用户 | 3次 | 10次/6元 | 看广告+1次 |
|
||
| 月卡VIP | 10次 | 同上 | 月费18元 |
|
||
| 年卡VIP | 20次 | 同上 | 年费168元 |
|
||
|
||
### 5.2 消耗逻辑
|
||
```javascript
|
||
// 扣费优先级：每日免费 > 赠送 > 购买 > VIP额外
|
||
async function consumeQuota(userId, amount = 1) {
|
||
const quota = await UserAIQuota.findByPk(userId);
|
||
|
||
// 检查是否需要重置每日配额
|
||
if (quota.daily_reset_date !== today) {
|
||
quota.daily_free_used = 0;
|
||
quota.daily_reset_date = today;
|
||
}
|
||
|
||
// 计算可用配额
|
||
const dailyFreeRemain = quota.daily_free_total - quota.daily_free_used;
|
||
const available = dailyFreeRemain + quota.gift_quota + quota.purchased_quota + quota.vip_daily_bonus;
|
||
|
||
if (available < amount) {
|
||
throw new Error('QUOTA_INSUFFICIENT');
|
||
}
|
||
|
||
// 按优先级扣除
|
||
let toConsume = amount;
|
||
if (dailyFreeRemain > 0) {
|
||
const use = Math.min(toConsume, dailyFreeRemain);
|
||
quota.daily_free_used += use;
|
||
toConsume -= use;
|
||
}
|
||
// ... 依次扣除其他配额
|
||
|
||
await quota.save();
|
||
}
|
||
```
|
||
|
||
---
|
||
|
||
## 六、审核流程
|
||
|
||
```
|
||
用户发布 ──► 机器审核 ──► 通过？──► 直接上架
|
||
│ │
|
||
▼ ▼
|
||
疑似违规 人工审核 ──► 通过/拒绝
|
||
│
|
||
▼
|
||
进入人工队列
|
||
```
|
||
|
||
### 6.1 机器审核维度
|
||
- 敏感词检测（sensitive_words表）
|
||
- 内容安全API（腾讯云/阿里云）
|
||
- AI生成内容标记检测
|
||
|
||
### 6.2 审核状态流转
|
||
```
|
||
草稿(0) ──► 提交审核 ──► 审核中(1) ──► 已发布(2)
|
||
│
|
||
▼
|
||
已拒绝(4) ──► 修改后重新提交
|
||
```
|
||
|
||
---
|
||
|
||
## 七、数据统计
|
||
|
||
### 7.1 核心指标
|
||
- AI调用成功率
|
||
- 平均响应时间
|
||
- 用户采纳率
|
||
- 生成内容发布率
|
||
- Token消耗/成本
|
||
|
||
### 7.2 日报表聚合
|
||
```sql
|
||
-- 每日定时任务聚合到 ai_daily_stats
|
||
INSERT INTO ai_daily_stats (stat_date, gen_type, model_name, call_count, ...)
|
||
SELECT
|
||
DATE(created_at),
|
||
gen_type,
|
||
model_name,
|
||
COUNT(*),
|
||
SUM(CASE WHEN status=1 THEN 1 ELSE 0 END),
|
||
SUM(input_tokens),
|
||
SUM(output_tokens),
|
||
AVG(latency_ms)
|
||
FROM ai_generations
|
||
WHERE DATE(created_at) = CURDATE() - INTERVAL 1 DAY
|
||
GROUP BY DATE(created_at), gen_type, model_name;
|
||
```
|
||
|
||
---
|
||
|
||
## 八、技术实现要点
|
||
|
||
### 8.1 服务端架构
|
||
```
|
||
┌─────────────────────────────────────────────┐
|
||
│ API Gateway │
|
||
└──────────────────────┬──────────────────────┘
|
||
│
|
||
┌──────────────┼──────────────┐
|
||
▼ ▼ ▼
|
||
┌─────────┐ ┌─────────┐ ┌─────────┐
|
||
│ 故事服务 │ │ AI服务 │ │ 用户服务 │
|
||
└─────────┘ └────┬────┘ └─────────┘
|
||
│
|
||
┌────────────┼────────────┐
|
||
▼ ▼ ▼
|
||
┌─────────┐ ┌─────────┐ ┌─────────┐
|
||
│ OpenAI │ │ Claude │ │ 本地模型 │
|
||
└─────────┘ └─────────┘ └─────────┘
|
||
```
|
||
|
||
### 8.2 AI服务封装
|
||
```javascript
|
||
// services/ai.js
|
||
class AIService {
|
||
constructor() {
|
||
this.providers = {
|
||
openai: new OpenAIProvider(),
|
||
claude: new ClaudeProvider(),
|
||
local: new LocalProvider()
|
||
};
|
||
}
|
||
|
||
async generate(options) {
|
||
const { type, provider = 'openai', ...params } = options;
|
||
const template = await this.getPromptTemplate(type);
|
||
const prompt = this.buildPrompt(template, params);
|
||
|
||
const startTime = Date.now();
|
||
try {
|
||
const result = await this.providers[provider].chat(prompt);
|
||
return {
|
||
success: true,
|
||
content: result.content,
|
||
tokens: result.usage,
|
||
latency: Date.now() - startTime
|
||
};
|
||
} catch (error) {
|
||
return { success: false, error: error.message };
|
||
}
|
||
}
|
||
}
|
||
```
|
||
|
||
### 8.3 前端交互优化
|
||
- 流式输出：使用SSE实时展示生成过程
|
||
- 骨架屏：生成中显示打字动画
|
||
- 失败重试：自动重试2次，超时30秒
|
||
- 结果缓存：相同输入5分钟内复用
|
||
|
||
---
|
||
|
||
## 九、后续迭代
|
||
|
||
### Phase 1（当前）
|
||
- [x] AI改写结局（后端接口已完成，模拟实现）
|
||
- [x] 后端基础框架（Python + FastAPI）
|
||
- [x] 故事/用户数据模型
|
||
- [x] 游玩记录接口
|
||
- [ ] 配额系统接入
|
||
- [ ] 基础审核流程
|
||
|
||
### Phase 2
|
||
- [ ] AI续写功能
|
||
- [ ] AI创作大纲
|
||
- [ ] 创作中心入口
|
||
|
||
### Phase 3
|
||
- [ ] AI完整创作
|
||
- [ ] UGC发布流程
|
||
- [ ] 创作者认证
|
||
- [ ] 收益分成
|