AIWriting – AI 小说生成系统,自动批量生成章节
助手工具
AIWriting – AI 小说生成系统,自动批量生成章节
🔥 热度:18

API部署网页

DeepSeek 导航网:AI 小说生成系统 AIWriting 基于 LangChain 的 AI 小说生成工具,支持多种大语言模型,具备状态管理和 Web 界面,支持自动批量生成章节,设定更新,记忆上下文等。

功能特性

  • 🤖 支持多种大语言模型(DeepSeek、OpenAI、Claude、Gemini 等)
  • 📚 章节状态管理和世界设定保存
  • 🌐 Web 界面,便于交互式创作
  • 📝 多小说项目隔离管理
  • 🔄 多版本生成和比较

安装使用

1. 安装依赖

pip install -r requirements.txt

2. 配置 API 密钥

编辑 .env 文件:

# 至少配置一个模型的 API 密钥
DEEPSEEK_API_KEY=your_deepseek_api_key
OPENAI_API_KEY=your_openai_api_key
ANTHROPIC_API_KEY=your_anthropic_api_key
GOOGLE_API_KEY=your_google_api_key
DSF_API_MODEL="第三方 api 模型"
DSF_API_KEY="第三方 api 秘钥"
DSF_API_URL="第三方 api 链接"

3. 启动方式

Web 界面(推荐):

python start_web.py

访问 http://127.0.0.1:5001 推荐

命令行使用: 使用麻烦,不推荐

from main import NovelGenerator

generator = NovelGenerator()
content = generator.generate_chapter(
    chapter_outline="第一章:开始的故事",
    model_name="deepseek_chat",
    novel_id="my_novel"
)

数据存储格式

章节状态文件

位置: data/{novel_id}_chapter_{章节号}_state.json

{
  "chapter_index": 1,
  "protagonist": {
    "name": "主角姓名",
    "age": 18,
    "level": "练气一层",
    "status": "健康",
    "personality": "坚韧不拔",
    "abilities": ["基础剑法"],
    "goal": "成为强者"
  },
  "inventory": [
    {
      "item_name": "铁剑",
      "description": "普通的铁制长剑"
    }
  ],
  "relationships": [
    {
      "name": "师父",
      "relation": "师徒",
      "status": "友好"
    }
  ],
  "current_plot_summary": "主角开始修炼之路"
}

世界设定文件

位置: data/{novel_id}_world_bible_{版本号}.json

{
  "world_name": "修仙世界",
  "setting": "古代修仙背景",
  "power_system": "练气->筑基->金丹->元婴",
  "locations": ["青云宗", "天剑峰"],
  "important_items": ["九转玄功", "天剑"],
  "key_npcs": ["掌门", "师兄"]
}

章节内容文件

位置: xiaoshuo/{novel_id}_chapter_{章节号}.txt

纯文本格式,存储生成的章节内容。

API 接口

核心方法

# 生成章节
generator.generate_chapter(
    chapter_outline="章节大纲",
    model_name="deepseek_chat",  # 模型选择
    novel_id="项目 ID",           # 小说项目 ID
    use_state=True,              # 是否使用状态
    update_state=True            # 是否更新状态
)

# 交互调用(命令行使用)
generator.chat(
    user_input="用户输入",
    session_id="会话 ID",
    model_name="deepseek_chat"
)

# 状态更新
new_state = generator.update_state(
    chapter_content="章节内容",
    current_state=current_state,
    novel_id="项目 ID"
)

Web API 端点

  • GET / - Web 界面
  • POST /generate - 生成章节
  • GET /novels - 获取小说列表
  • GET /novel/{novel_id}/info - 获取小说信息

支持的模型

  • DeepSeek: deepseek_chat, deepseek_reasoner
  • OpenAI: openai_gpt4, openai_gpt35
  • Anthropic: anthropic_claude
  • Google: google_gemini
  • 其他: dsf (第三方接口)

下载地址

相关推荐