YILING0013 / AI_NovelGenerator
- четверг, 2 октября 2025 г. в 00:00:06
使用ai生成多章节的长篇小说,自动衔接上下文、伏笔
- 当前没有什么精力维护该项目,本身该项目并无任何收益,以及临近毕业,有很多内容要忙,如果后面有时间的话,再考虑基于更新的技术去重构吧。——2025/9/24
✨ 核心功能 ✨
功能模块 | 关键能力 |
---|---|
🎨 小说设定工坊 | 世界观架构 / 角色设定 / 剧情蓝图 |
📖 智能章节生成 | 多阶段生成保障剧情连贯性 |
🧠 状态追踪系统 | 角色发展轨迹 / 伏笔管理系统 |
🔍 语义检索引擎 | 基于向量的长程上下文一致性维护 |
📚 知识库集成 | 支持本地文档参考 |
✅ 自动审校机制 | 检测剧情矛盾与逻辑冲突 |
🖥 可视化工作台 | 全流程GUI操作,配置/生成/审校一体化 |
一款基于大语言模型的多功能小说生成器,助您高效创作逻辑严谨、设定统一的长篇故事
确保满足以下运行条件:
下载项目
git clone https://github.com/YILING0013/AI_NovelGenerator
安装编译工具(可选)
安装依赖并运行
cd AI_NovelGenerator
pip install -r requirements.txt
python main.py
如果缺失部分依赖,后续手动执行
pip install XXX进行安装即可
novel-generator/
├── main.py # 入口文件, 运行 GUI
├── consistency_checker.py # 一致性检查, 防止剧情冲突
|—— chapter_directory_parser.py # 目录解析
|—— embedding_adapters.py # Embedding 接口封装
|—— llm_adapters.py # LLM 接口封装
├── prompt_definitions.py # 定义 AI 提示词
├── utils.py # 常用工具函数, 文件操作
├── config_manager.py # 管理配置 (API Key, Base URL)
├── config.json # 用户配置文件 (可选)
├── novel_generator/ # 章节生成核心逻辑
├── ui/ # 图形界面
└── vectorstore/ # (可选) 本地向量数据库存储
{
"api_key": "sk-XXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
"base_url": "https://api.openai.com/v1",
"interface_format": "OpenAI",
"model_name": "gpt-4o-mini",
"temperature": 0.7,
"max_tokens": 4096,
"embedding_api_key": "sk-XXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
"embedding_interface_format": "OpenAI",
"embedding_url": "https://api.openai.com/v1",
"embedding_model_name": "text-embedding-ada-002",
"embedding_retrieval_k": 4,
"topic": "星穹铁道主角星穿越到原神提瓦特大陆,拯救提瓦特大陆,并与其中的角色展开爱恨情仇的小说",
"genre": "玄幻",
"num_chapters": 120,
"word_number": 4000,
"filepath": "D:/AI_NovelGenerator/filepath"
}
生成模型配置
api_key
: 大模型服务的API密钥base_url
: API终端地址(本地服务填Ollama等地址)interface_format
: 接口模式model_name
: 主生成模型名称(如gpt-4, claude-3等)temperature
: 创意度参数(0-1,越高越有创造性)max_tokens
: 模型最大回复长度Embedding模型配置
embedding_model_name
: 模型名称(如Ollama的nomic-embed-text)embedding_url
: 服务地址embedding_retrieval_k
:小说参数配置
topic
: 核心故事主题genre
: 作品类型num_chapters
: 总章节数word_number
: 单章目标字数filepath
: 生成文件存储路径python main.py
执行后,GUI 将会启动,你可以在图形界面中进行各项操作。
如果你想在无 Python 环境的机器上使用本工具,可以使用 PyInstaller 进行打包:
pip install pyinstaller
pyinstaller main.spec
打包完成后,会在 dist/
目录下生成可执行文件(如 Windows 下的 main.exe
)。
启动后,先完成基本参数设置:
https://api.openai.com/v1
)gpt-3.5-turbo
、gpt-4o
等)点击「Step1. 生成设定」
Novel_setting.txt
:包含世界观、角色信息、雷点暗线等。Novel_setting.txt
中查看或修改设定内容。点击「Step2. 生成目录」
Novel_setting.txt
内容,为全部章节生成:
Novel_directory.txt
:包括每章标题和简要提示。点击「Step3. 生成章节草稿」
1
)Novel_directory.txt
、以及已定稿章节outline_X.txt
) 及正文 (chapter_X.txt
)点击「Step4. 定稿当前章节」
global_summary.txt
)character_state.txt
)plot_arcs.txt
)chapter_X.txt
中看到定稿后的文本。一致性检查(可选)
重复第 4-6 步 直到所有章节生成并定稿!
向量检索配置提示
- embedding模型需要显示指定接口和模型名称;
- 使用本地Ollama的Embedding时需提前启动Ollama服务:
ollama serve # 启动服务 ollama pull nomic-embed-text # 下载/启用模型- 切换不同Embedding模型后建议清空vectorstore目录
- 云端Embedding需确保对应API权限已开通
该问题大概率由于API未正确响应造成,也许响应了一个html?其它内容,导致出现该报错;
确认接口是否稳定;
在GUI界面中对应输入即可。
如有更多问题或需求,欢迎在项目 Issues 中提出。