NovelFlow

基于 AI Agent 的小说创作平台。

技术栈

层级	技术
语言	Go 1.26
Web 框架	Gin
AI Agent	CloudWeGo Eino(v0.9.0-alpha.x+)
语言模型	Claude
数据库	MySQL (GORM) + MongoDB
缓存	Redis
认证	JWT (Access/Refresh Token)
配置	Viper + YAML + 环境变量
日志	slog + 按日切割文件

项目结构

novelflow/
├── agents/                  # AI Agent 引擎
│   ├── agent.go            # Agent 核心（基于 Eino ADK deep 模式）
│   ├── mainagent.go        # 小说创作主 Agent
│   ├── outline.go          # 大纲子代理（生成/修改章纲）
│   ├── writer.go           # 写作子代理（按大纲撰写章节）
│   ├── reviewer.go         # 审查子代理（质量评分）
│   ├── middleware.go        # 工具调用安全中间件
│   ├── session.go          # MongoDB 会话管理
│   ├── message.go          # 消息模型
│   ├── model.go            # LLM 模型初始化
│   ├── prompt.go           # 小说创作系统提示词
│   ├── tools.go            # 工具注册
│   ├── tools_file.go       # 文件读写工具
│   ├── tools_db.go         # 数据库查询工具
│   └── utils.go            # 流式处理工具
├── backend/
│   ├── cmd/server/main.go  # 服务入口
│   └── internal/
│       ├── route.go         # 路由配置
│       ├── handler/         # HTTP 处理器
│       ├── service/         # 业务逻辑层
│       ├── middleware/      # 认证与 CORS 中间件
│       ├── response/        # 统一响应结构
│       └── servicecontext/  # 服务上下文（依赖管理）
├── cache/                   # Redis 客户端（JWT 黑名单）
├── config/                  # Viper 配置加载（支持环境变量覆盖）
├── database/
│   ├── mysql/              # 用户模型、会话关联、GORM 仓库（单例连接）
│   ├── mongodb/            # MongoDB 连接（Agent 会话存储）
│   └── task/               # 异步生成任务状态管理
├── frontend/               # 前端（待开发）
└── cmd/                    # 其他命令入口（待开发）

功能模块

用户认证

• 注册 / 登录 / 登出 / 令牌刷新
• JWT Access Token + Refresh Token 双令牌机制
• Redis 令牌黑名单（登出即时失效）

异步小说生成

• POST /generate 异步启动生成任务，立即返回 session_id
• 后台 goroutine 调用 AI Agent 引擎执行小说创作
• 任务状态持久化到 MongoDB（pending → running → completed / failed）
• GET /generate/:session_id 轮询查询生成状态
• 请求体字段：

字段	类型	必填	说明
genre	string	是	小说类型（修仙、玄幻、都市等）
concept	string	是	核心概念/故事简介
protagonist	string	否	主角设定
world_setting	string	否	世界观设定
chapter_count	int	否	预期章节数
style	string	否	风格要求
requirements	string	否	其他特殊要求

各字段按模板合成为完整 prompt 后发给 AI Agent 执行。

AI Agent

• 基于 CloudWeGo Eino deep 模式的 Agent 运行器
• 流式输出（支持思考过程展示）
• 工具调用安全中间件（错误不影响对话流程）
• 未知工具处理器（模型可自动修正）
• MongoDB 持久化会话管理
• 自动重试（限流 429 重试）
• 会话与用户关联（MongoDB session + MySQL user_sessions 映射）

API 路由

方法	路径	说明	认证
POST	/auth/register	用户注册	否
POST	/auth/login	用户登录	否
POST	/auth/refresh	刷新令牌	否
POST	/auth/logout	用户登出	是
GET	/users/:id	获取用户	是
PUT	/users/:id	更新用户	是
DELETE	/users/:id	删除用户	是
POST	/generate	异步启动小说生成	是
GET	/generate/:session_id	查询生成任务状态	是
GET	/health	健康检查	否

快速开始

前置条件

• Go 1.26+
• MySQL 8.0+
• Redis 6.0+
• MongoDB 6.0+

配置

cp config/config.example.yaml config/config.yaml
# 编辑 config.yaml 填入实际的数据库和 API 密钥

运行

go run backend/cmd/server/main.go

健康检查

curl http://localhost:8080/health
# {"status":"ok"}