【Code With SOLO】TRAE SOLO创造AI小说系统

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接口规范。

四、实践过程:

  1. 项目初始化与规划:

    任务拆解:分析项目需求和目标、进行技术选型、设计项目结构

    使用的 TRAE SOLO 能力:

     1、搜索能力:分析当前流行的技术栈,选择适合的后端和前端框架
    ​
      2、文件管理能力:创建项目目录结构和基础文件
    

    关键操作:

     1、分析需求:确定开发一个 AI 小说生成工具,支持从零创作、重写、润色、续写等功能
    ​
     2、技术选型:选择 Python + FastAPI 作为后端,React + TypeScript 作为前端
    ​
     3、设计项目结构:创建清晰的目录结构,分离后端和前端代码
    
  2. 后端开发

    任务拆解:

     1、搭建后端基础架构
    ​
     2、实现核心功能模块
    ​
     3、设计数据库结构
    

    使用的 TRAE SOLO 能力:

     1、代码生成能力:生成后端基础架构和模板代码
    ​
     2、代码编辑能力:实现核心功能和修复问题
    

    关键操作:

     1、创建 FastAPI 应用入口
    ​
     2、实现认证、项目管理、小说生成等 API 模块
    ​
     3、设计数据库表结构,包括用户表、项目表、章节表等
    
  3. 前端开发

    任务拆解:

     1、搭建前端基础架构
    ​
     2、实现核心页面
    ​
     3、实现响应式布局和国际化功能
    

    使用的 TRAE SOLO 能力:

     1、代码生成能力:生成前端基础架构和组件模板
    ​
     2、代码编辑能力:实现页面功能和交互逻辑
    

    关键操作:

     1、创建 React 应用入口和路由配置
    ​
     2、实现登录、注册、创作等页面
    ​
     3、实现响应式布局,支持桌面端和移动端
    ​
     4、集成国际化功能,支持中英文切换
    
  4. 集成测试与部署

    任务拆解:

     1、进行 API 测试
    ​
     2、进行前端功能测试
    ​
     3、进行前后端集成测试
    ​
     4、编写启动脚本和部署文档
    

    使用的 TRAE SOLO 能力:

     1、命令执行能力:运行测试和部署命令
    ​
     2、代码编辑能力:编写启动脚本和部署文档
    

    关键操作:

     1、测试 API 接口功能
    ​
     2、测试前端页面导航和表单提交
    ​
     3、测试完整的创作流程
    ​
     4、编写一键启动脚本,自动处理端口冲突和配置更新
    ​
     5、编写详细的部署文档,包括不同系统的安装步骤
    
  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 的各种能力,我成功完成了这个项目的开发:

  1. 文件管理能力:创建和管理项目文件结构

  2. 代码生成能力:生成基础代码架构和模板

  3. 代码编辑能力:实现核心功能和修复问题

  4. 命令执行能力:运行测试和部署命令

  5. 搜索能力:查找技术资料和解决方案

    项目最终实现了所有需求功能,包括 AI 小说生成、重写、润色、续写等,采用了前后端分离架构,提供了流畅的创作体验。

实话实说,有点没看明白,一次性生成那么多章节不乱吗?阅读效果如何?

不会乱的,加入了向量数据库。

1 个赞

好奇,这是相当于一个agent吗

不是agent