TRAE SOLO创造AI小说系统
一、摘要:
构建一款基于 TRAE SOLO 的 AI 小说创作系统,支持用户从零生成小说(短篇/长篇)、一键切换16种风格进行重写,并提供多种智能润色能力(如精简、扩展、文艺化、情感增强)及自动续写功能。后续将通过性能优化、功能升级、用户体验改进及部署优化,持续提升产品整体表现。
二、背景:
小编曾是一名前端开发工程师,如今已逐渐淡出一线。平日里喜欢阅读小说、观看短剧,也因此萌生了开发一款 AI 小说系统的想法。
过去,小编也曾怀揣作家梦,但受限于词汇表达与想象力,始终未能真正迈出那一步。在不断尝试与反思中,小编逐渐意识到,AI 并不是取代创作者的工具,而是辅助创作的有力帮手——创作的主导权始终掌握在人手中,而 AI 可以帮助构建小说框架、拓展思路,并对后续内容进行优化与完善。
基于这一认知,小编开始尝试借助 TRAE SOLO 进行开发,最终打造出了这款 AI 小说系统。
三、技术栈:
前端使用React 18 + TypeScript + Ant Design + Vite + Zustand,后端使用FastAPI + Python + SQLAlchemy,国际化i18next,采用前后端分离,定义API接口规范。
四、实践过程:
-
项目初始化与规划:
任务拆解:分析项目需求和目标、进行技术选型、设计项目结构
使用的 TRAE SOLO 能力:
1、搜索能力:分析当前流行的技术栈,选择适合的后端和前端框架 2、文件管理能力:创建项目目录结构和基础文件关键操作:
1、分析需求:确定开发一个 AI 小说生成工具,支持从零创作、重写、润色、续写等功能 2、技术选型:选择 Python + FastAPI 作为后端,React + TypeScript 作为前端 3、设计项目结构:创建清晰的目录结构,分离后端和前端代码 -
后端开发
任务拆解:
1、搭建后端基础架构 2、实现核心功能模块 3、设计数据库结构使用的 TRAE SOLO 能力:
1、代码生成能力:生成后端基础架构和模板代码 2、代码编辑能力:实现核心功能和修复问题关键操作:
1、创建 FastAPI 应用入口 2、实现认证、项目管理、小说生成等 API 模块 3、设计数据库表结构,包括用户表、项目表、章节表等 -
前端开发
任务拆解:
1、搭建前端基础架构 2、实现核心页面 3、实现响应式布局和国际化功能使用的 TRAE SOLO 能力:
1、代码生成能力:生成前端基础架构和组件模板 2、代码编辑能力:实现页面功能和交互逻辑关键操作:
1、创建 React 应用入口和路由配置 2、实现登录、注册、创作等页面 3、实现响应式布局,支持桌面端和移动端 4、集成国际化功能,支持中英文切换 -
集成测试与部署
任务拆解:
1、进行 API 测试 2、进行前端功能测试 3、进行前后端集成测试 4、编写启动脚本和部署文档使用的 TRAE SOLO 能力:
1、命令执行能力:运行测试和部署命令 2、代码编辑能力:编写启动脚本和部署文档关键操作:
1、测试 API 接口功能 2、测试前端页面导航和表单提交 3、测试完整的创作流程 4、编写一键启动脚本,自动处理端口冲突和配置更新 5、编写详细的部署文档,包括不同系统的安装步骤 -
遇到的问题与解决方案
端口占用问题:
1、问题:启动时提示端口被占用 2、解决方案:在启动脚本中添加自动清理旧进程的功能前端配置问题:
1、问题:前端无法连接后端 2、解决方案:在启动脚本中添加自动更新前端配置文件的功能依赖安装问题:
1、问题:在不同系统上安装依赖失败 2、解决方案:编写详细的安装指南,包括不同系统的安装步骤和常见问题解决方案数据库迁移问题:
1、问题:数据库表结构变更时的迁移问题 2、解决方案:使用 SQLAlchemy 的自动迁移功能,确保数据库表结构与模型同步
五、成功展示:
下面这张图是用这个项目生成的小说,这本小说大概是一百四十万字。我在项目中加入了向量数据库,它能够使小说保持上下文连贯性,生成新章节时自动检索相关前文、追踪角色状态和情节线索,确保故事逻辑一致、本地部署ChromaDB,无需付费,数据安全可控、提升AI生成内容的相关性和准确性,优化创作体验。
该项目支持PC端和移动端
小说重写、小说润色、从零创作这三个模块非常相似,我就展示从零创作模块,该模块分专门生成短篇小说(短故事)和专门生成长篇小说两个部分,这个两个部分最大的区别,是长篇小说可以设置章节大纲,长篇小说的大纲可以根据前面填写好的小说标题、小说类型、任务设定、世界观设定、剧情构思、章节数量自动生成,也可以自己填写。
最开始项目没有加入向量数据库,是通过上图所示的上下文模式保持上下文连贯。大概能够保持十万字左右,因为效果不理想就加入了向量数据库。
通过项目管理可以查看以往生成好的小说内容,这是个新账号,之前的小说就没有显示在里面。
系统设置里面分API设置和其它参数。API设置中支持的后端类型有Ollama、OpenAI、Claude、其它类型。点击测试按钮可以查看当前API能不能走通。其它参数中的temperature是控制生成内容的随机性,值越高越随机;Top P是控制生成时的词汇选择范围;Top K是限制每步生成时考虑的词汇数量。
上面提到过,这个项目支持注册账号使用,用户之间是无法查看对方的数据的。
六、总结:
通过使用 TRAE SOLO 的各种能力,我成功完成了这个项目的开发:
-
文件管理能力:创建和管理项目文件结构
-
代码生成能力:生成基础代码架构和模板
-
代码编辑能力:实现核心功能和修复问题
-
命令执行能力:运行测试和部署命令
-
搜索能力:查找技术资料和解决方案
项目最终实现了所有需求功能,包括 AI 小说生成、重写、润色、续写等,采用了前后端分离架构,提供了流畅的创作体验。







