系统架构概述AI写作助手采用多智能体协作架构包含以下核心模块核心基础设施层- 向量存储、缓存、流式输出、工作流引擎智能体层- 11个专业智能体协同工作记忆系统层- 7个记忆模块管理长期记忆数据管理层- 项目管理、素材收集、导出UI层- PyQt6桌面应用核心模块详解1. 向量存储系统 (src/core/vector_store.py)基于ChromaDB的向量数据库支持语义检索。fromsrc.core.vector_storeimportget_vector_store,get_rag_system# 获取向量存储实例vector_storeget_vector_store()# 添加文档vector_store.add_documents(characters,[{id:char_001,text:林云修仙少年,metadata:{role:主角}}])# 语义搜索resultsvector_store.search(characters,修仙,n_results5)# RAG系统ragget_rag_system()contextrag.generate_writing_context(chapter_info,involved_characters)2. 缓存系统 (src/core/cache_manager.py)多级缓存架构加速上下文加载。fromsrc.core.cache_managerimportget_chapter_cache,get_context_cache# 章节缓存cacheget_chapter_cache()cache.save_chapter(1,chapter_data)chaptercache.get_chapter(1)# 上下文缓存ctx_cacheget_context_cache()ctx_cache.set_context(ctx_001,context_text)3. 流式输出 (src/core/stream_handler.py)支持LLM流式响应和实时UI更新。fromsrc.core.stream_handlerimportStreamingWriter writerStreamingWriter(llm_client)writer.write_stream(prompt写一个修仙小说开头,on_chunklambdaevent,data:print(data[content]))4. 工作流引擎 (src/core/workflow_engine.py)状态机管理多智能体协作。fromsrc.core.workflow_engineimportget_workflow_engine engineget_workflow_engine()engine.create_novel_workflow()engine.start({idea:废柴逆袭修仙})# 查看状态statusengine.get_status()智能体系统智能体列表智能体文件职责大纲师outline_agent.py生成故事大纲和章节细纲码字工writer_agent.py撰写章节内容督察reviewer_agent.py审核内容一致性运营polisher_agent.py润色和生成标题评论家critic_agent.py批评评分、Prompt自进化复盘智能体review_agent.py定期总结、调整规划读者模拟器review_agent.py模拟读者反馈智能体协作流程用户输入创意 ↓ 大纲师 → 生成故事Bible ↓ 码字工 → 撰写章节初稿 ↓ 督察 → 审核12项检查 ↓ (不通过则返回码字工) 运营 → 润色优化 ↓ 数据持久化 ↓ 复盘智能体 → 定期总结 ↓ 评论家 → Prompt自进化记忆系统记忆模块列表模块文件功能记忆管理器memory_manager.py人物状态、钩子追踪、大事记实体词典entity_dictionary.py物品/地点/技能管理世界观规则world_rules.py规则清单、全局时间轴心理维度psychology.py心理档案、情感案例风格指南style_guide.py文风统一要求分层记忆架构层级内容存储方式更新频率层级0故事BibleJSON极少修改层级1人物档案向量结构化每章更新层级2大事记列表摘要每章追加层级3最近N章缓存滚动更新层级4钩子清单结构化每章更新审核系统12项审核检查人物时空一致性人物性格一致性能力体系一致性时间线一致性世界观规则遵守钩子一致性前文连续性实体名称一致性信息密度智商合理性情感铺垫爽点重复度数据存储存储结构data/ ├── projects/ # 项目数据 │ └── proj_xxx/ │ ├── config.json │ ├── story_bible.json │ ├── characters.json │ ├── chapters.json │ ├── hooks.json │ ├── events.json │ └── memory/ ├── chromadb/ # 向量数据库 ├── cache/ # 缓存 ├── materials/ # 素材库 └── exports/ # 导出文件LLM集成支持的LLM提供商提供商模型成本智谱GLMglm-4-flash免费DeepSeekdeepseek-v4-flash¥1-2/百万tokenOpenAIgpt-4o-mini$0.15/百万tokenLLM调用示例fromsrc.utils.llmimportget_llm_client clientget_llm_client(zhipuai)responseclient.chat(写一个修仙小说开头,system_prompt你是一位网文作者)性能优化缓存策略内存LRU缓存最近访问的章节和上下文文件缓存持久化存储向量检索优化使用ChromaDB本地部署避免网络延迟批量索引减少索引次数流式输出实时返回生成内容提升用户体验进度追踪显示生成状态