Kimi Writing Agent

这是一个由 kimi-k2-thinking 模型驱动的自主智能体，专门用于创作小说、书籍和短篇故事集。

功能特性

• 🤖 自主创作：智能体能够独立规划并执行创意写作任务
• 📚 多种格式：支持创作小说、书籍或短篇故事集
• ⚡ 实时流式传输：实时查看智能体的推理过程及生成内容
• 💾 智能上下文管理：当接近 token 限制时自动压缩上下文
• 🔄 恢复模式：支持从保存的上下文摘要中恢复中断的工作
• 📊 Token 监控：实时追踪 token 使用情况并进行自动优化
• 🛠️ 工具使用：智能体具备创建项目、编写文件及管理工作空间的能力

安装指南

前置条件

我们推荐使用 uv 来进行高效的 Python 包管理：

# 安装 uv (如果尚未安装)
curl -LsSf https://astral.sh/uv/install.sh | sh

设置步骤

1. 安装依赖：

使用 uv (推荐):

uv pip install -r requirements.txt

或者使用 pip:

pip install -r requirements.txt

2. 配置 API Key：

创建一个包含 API Key 的 .env 文件：

# 复制示例文件
cp env.example .env

# 编辑 .env 并填入你的 API Key
# 文件应包含：
MOONSHOT_API_KEY=your-api-key-here

可选： 设置自定义 base URL (默认为 https://api.moonshot.ai/v1)：

# 添加到 .env 文件中：
MOONSHOT_BASE_URL=https://api.moonshot.ai/v1

使用方法

快速开始

通过命令行直接输入提示词：

# 使用 uv (推荐)
uv run kimi-writer.py "Create a collection of 5 sci-fi short stories about AI"

# 或者直接使用 python
python kimi-writer.py "Create a collection of 5 sci-fi short stories about AI"

或者以交互模式运行：

uv run kimi-writer.py
# 或: python kimi-writer.py

然后根据提示输入你的要求。

恢复模式

如果智能体运行中断，或者你想继续之前的工作：

uv run kimi-writer.py --recover output/my_project/.context_summary_20250107_143022.md
# 或: python kimi-writer.py --recover output/my_project/.context_summary_20250107_143022.md

工作原理

智能体工具

智能体可以使用以下三种工具：

1. create_project：创建项目文件夹以组织写作内容
2. write_file：以三种模式编写 Markdown 文件：
   • create：新建文件（若文件已存在则失败）
   • append：向现有文件追加内容
   • overwrite：覆盖整个文件内容
3. compress_context：自动触发以管理上下文大小

智能体循环

1. 智能体接收你的提示词
2. 使用 kimi-k2-thinking 对任务进行推理
3. 决定调用哪些工具并执行
4. 审查结果并持续迭代，直到任务完成
5. 最多支持 300 次迭代，并带有自动上下文压缩功能

上下文管理

• Token 限制：200,000 tokens
• 自动压缩：在达到 180,000 tokens（限制的 90%）时触发
• 备份：每 50 次迭代自动生成一次上下文摘要
• 恢复：所有摘要均以时间戳保存，方便恢复进度

项目结构

kimi-writer/
├── kimi-writer.py        # 主智能体程序
├── tools/
│   ├── __init__.py       # 工具注册
│   ├── writer.py         # 文件写入工具
│   ├── project.py        # 项目管理工具
│   └── compression.py    # 上下文压缩工具
├── utils.py              # 工具函数（token 计数等）
├── requirements.txt      # Python 依赖
├── env.example           # 配置示例
├── .gitignore            # Git 忽略规则
└── README.md             # 本文件

# 使用期间生成：
output/                   # 所有 AI 生成的项目存放于此
├── your_project_name/    # 由智能体创建
│   ├── chapter_01.md     # 由智能体撰写
│   ├── chapter_02.md
│   └── .context_summary_*.md  # 自动保存的上下文摘要
└── another_project/
    └── ...

示例

示例 1：小说

uv run kimi-writer.py "Write a mystery novel set in Victorian London with 10 chapters"

示例 2：短篇故事集

uv run kimi-writer.py "Create 7 interconnected sci-fi short stories exploring the theme of memory"

示例 3：书籍

uv run kimi-writer.py "Write a comprehensive guide to Python programming with 15 chapters"

高级功能

实时流式传输

实时观看智能体的思考与创作过程：

• 🧠 推理流 (Reasoning Stream)：查看智能体规划任务时的思维过程
• 💬 内容流 (Content Stream)：实时观察故事逐字生成
• 🔧 工具调用进度：生成大量内容时提供实时更新（显示字符/单词统计）
• ⚡ 零等待：即时反馈，无需对着空白屏幕发呆

迭代计数器

智能体会显示当前进度：Iteration X/300

Token 监控

实时显示 token 使用情况：Current tokens: 45,234