基米写作代理

一款由 kimi-k2-thinking 模型驱动的自主代理，用于创作小说、书籍和短篇故事集。

功能

• 🤖 自主写作: 代理可独立规划并执行创意写作任务
• 📚 多种格式: 可创建小说、书籍或短篇故事集
• ⚡ 实时流式输出: 随着生成过程实时展示代理的推理与写作内容
• 💾 智能上下文管理: 在接近令牌限制时自动压缩上下文
• 🔄 恢复模式: 可从保存的上下文摘要中恢复中断的工作
• 📊 令牌监控: 实时跟踪令牌使用情况，并自动优化
• 🛠️ 工具使用: 代理可以创建项目、编写文件并管理其工作空间

安装

先决条件

我们推荐使用 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 密钥：

创建一个包含您的 API 密钥的 .env 文件：

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

# 编辑 .env 并添加您的 API 密钥
# 文件内容应为：
MOONSHOT_API_KEY=your-api-key-here

可选: 设置自定义基础 URL（默认为 https://api.moonshot.ai/v1）：

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

使用方法

新建项目

通过内联提示运行：

# 推荐使用 uv
uv run kimi-writer.py "创作5篇关于人工智能的科幻短篇故事集"

# 或直接使用 python
python kimi-writer.py "创作5篇关于人工智能的科幻短篇故事集"

或者以交互模式运行：

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 次迭代，并自动压缩上下文

上下文管理

• 令牌限制: 200,000 个令牌
• 自动压缩: 当达到 180,000 个令牌时触发（占限制的 90%）
• 备份: 每 50 次迭代自动保存一次上下文摘要
• 恢复: 所有摘要均按时间戳保存，以便后续恢复

项目结构

kimi-writer/
├── kimi-writer.py        # 主代理程序
├── tools/
│   ├── __init__.py       # 工具注册表
│   ├── writer.py         # 文件写入工具
│   ├── project.py        # 项目管理工具
│   └── compression.py    # 上下文压缩工具
├── utils.py              # 工具函数（令牌计数等）
├── 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 "撰写一部以维多利亚时代伦敦为背景的10章悬疑小说"

示例 2: 短篇故事集

uv run kimi-writer.py "创作7篇相互关联的科幻短篇故事，探讨记忆的主题"

示例 3: 书籍

uv run kimi-writer.py "编写一本包含15章的Python编程综合指南"

高级功能

实时流式输出

实时观看代理的思考与写作过程：

• 🧠 推理流: 观看代理在规划任务时的思维过程
• 💬 内容流: 逐字逐句见证故事的创作
• 🔧 工具调用进度: 在生成大量内容时提供实时更新（显示字符/单词数）
• ⚡ 无需等待: 即时反馈——不再需要盯着空白屏幕发呆

迭代计数器

代理会显示其进度：第 X 次迭代/300次

令牌监控

实时显示令牌使用情况：当前令牌数：45,234/200,000（22.6%）

优雅中断

按下 Ctrl+C 即可中断。代理会保存当前上下文以便后续恢复。

获得最佳效果的技巧

1. 明确具体: 清晰的提示能带来更好的结果

   • 好的：创作一部以现代东京为背景的5章爱情小说
   • 不太好的：写点有趣的东西

2. 放手让其工作: 代理会自主规划并完成整个任务

3. 轻松恢复: 如果被中断，只需使用 --recover 标志配合最新的上下文摘要即可恢复

4. 检查进度: 生成的文件会实时出现在项目文件夹中

故障排除

提示“未设置 MOONSHOT_API_KEY 环境变量”

请确保在项目根目录下创建了一个包含您的 API 密钥的 .env 文件：

MOONSHOT_API_KEY=your-actual-api-key-here

出现“401 Unauthorized”或认证错误

• 请确认 .env 文件中的 API 密钥是否正确
• 确保使用正确的基础 URL：https://api.moonshot.ai/v1
• 您可以从 https://platform.moonshot.cn/ 获取 API 密钥

“创建项目文件夹时出错”

请检查当前目录的写入权限

代理似乎卡住了

代理最多可运行 300 次迭代。对于非常复杂的任务，这是正常现象。请查看项目文件夹以了解进展情况。

令牌限制问题

代理会在 180,000 个令牌时自动压缩上下文。如果您看到压缩提示，则说明系统正在正常工作。

技术细节

• 模型: kimi-k2-thinking
• 温度: 1.0（针对此模型优化）
• 每次调用的最大令牌数: 65,536（64K）
• 上下文窗口: 200,000 个令牌
• 最大迭代次数: 300
• 压缩阈值: 180,000 个令牌

您可以根据需要进行自定义。

许可证

MIT 许可证，附带署名要求——详情请参阅 LICENSE 文件。

商业使用：如果您将本软件用于商业产品中，必须明确注明作者为 Pietro Schirano (@Doriandarko)。

API 使用：本项目使用了 Moonshot AI 的 API。请参阅 Moonshot AI 的服务条款以获取 API 使用的相关指南。

致谢

• 创建者：Pietro Schirano (@Doriandarko)
• 技术支持：Moonshot AI 的 kimi-k2-thinking 模型
• 仓库地址：https://github.com/Doriandarko/kimi-writer

点星历史

Kimi Writer 快速上手指南

Kimi Writer 是一个基于 kimi-k2-thinking 模型的自主写作智能体，能够独立规划并执行小说、书籍或短篇故事集的创作任务。

环境准备

• 操作系统：Linux / macOS / Windows (需安装 Python 环境)
• Python 版本：建议 Python 3.8+
• 依赖管理工具：推荐使用 uv 以获得更快的包安装速度（国内网络环境下若下载缓慢，可配置国内镜像或使用 pip 替代）

安装步骤

1. 安装依赖管理工具 uv（推荐）

curl -LsSf https://astral.sh/uv/install.sh | sh

注：如无法访问上述地址，可直接使用系统自带的 pip 进行后续操作。

2. 安装项目依赖

克隆项目后进入目录，执行以下命令安装依赖：

使用 uv（推荐）：

uv pip install -r requirements.txt

或使用 pip：

pip install -r requirements.txt

3. 配置 API 密钥

在项目根目录下创建 .env 文件并填入你的 Moonshot AI API Key：

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

# 编辑 .env 文件，填入真实的 API Key
# 文件内容应包含：
MOONSHOT_API_KEY=your-api-key-here

获取 API Key 请访问：https://platform.moonshot.cn/

基本使用

方式一：直接通过命令行提示词启动

运行以下命令，智能体将自动开始创作：

# 使用 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

运行后，生成的内容将实时保存在 output/ 目录下的项目文件夹中，你可以随时查看进度。