kimi-code-memory-mcp-server 0.1.0 → 0.1.1

package/CHANGELOG.md

@@ -22,5 +22,20 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - CI workflow running tests and lint on pull requests.
 - MCP server versioning rules: SemVer for repository version, runtime version synced from `package.json`, and a release workflow that includes real-instance validation.
 
-[Unreleased]: https://github.com/Zehee/kimi-code-memory-mcp-server/compare/v0.1.0...HEAD
+## [0.1.1] - 2026-06-25
+
+### Added
+- `kimi-memory-setup` command for one-click Kimi Code CLI integration.
+- `assets/user-agents.md` template injected into `~/.kimi-code/AGENTS.md` with a prominent bootstrap rule.
+- Automatic skill installation to `~/.kimi-code/skills/memory-manage`.
+- Automatic `kimi-memory` MCP server entry in `~/.kimi-code/mcp.json`.
+- `--dry-run` and `--undo` options for the setup command.
+
+### Changed
+- `search_context` now returns refined summaries directly when the original `wire.jsonl` is missing.
+- Improved `scripts/session-search-verify.mjs` with CLI/env args, safer parsing, and cleanup.
+- Updated READMEs to reflect npm-published status and setup command.
+
+[Unreleased]: https://github.com/Zehee/kimi-code-memory-mcp-server/compare/v0.1.1...HEAD
+[0.1.1]: https://github.com/Zehee/kimi-code-memory-mcp-server/releases/tag/v0.1.1
 [0.1.0]: https://github.com/Zehee/kimi-code-memory-mcp-server/releases/tag/v0.1.0

package/README.md

@@ -7,7 +7,7 @@
 
 A local stdio MCP server that gives [Kimi Code CLI](https://github.com/MoonshotAI/kimi-code) cross-session memory.
 
-> **Note:** This package is not yet published to npm. Install and run it from source (see below).
+> **Note:** This package is published to npm as `kimi-code-memory-mcp-server`. You can install it directly or run it from source.
 
 User-facing memories are stored as plain Markdown files on disk. Refined turn summaries use a local SQLite cache, but no vector database, graph database, or external cloud service is required.
 
@@ -40,7 +40,15 @@ See [`docs/ARCHITECTURE.md`](./docs/ARCHITECTURE.md) for the design rationale.
 
 ## Install
 
-Currently the package must be installed from source. Node.js ≥ 18 is required.
+Node.js ≥ 18 is required.
+
+### From npm (recommended)
+
+```bash
+npm install -g kimi-code-memory-mcp-server
+```
+
+### From source
 
 ```bash
 git clone https://github.com/Zehee/kimi-code-memory-mcp-server.git
@@ -49,9 +57,66 @@ npm install
 npm run build
 ```
 
-## Configure Kimi Code CLI
+## Quick Setup (recommended)
+
+After installing from npm, run the setup command to configure Kimi Code CLI automatically:
+
+```bash
+npx kimi-memory-setup
+```
+
+This will:
+
+1. Detect your `~/.kimi-code` directory.
+2. Inject memory protocol rules at the top of `~/.kimi-code/AGENTS.md`.
+3. Install the `memory-manage` skill to `~/.kimi-code/skills/memory-manage`.
+4. Add the `kimi-memory` MCP server entry to `~/.kimi-code/mcp.json`.
+
+Preview changes without writing anything:
+
+```bash
+npx kimi-memory-setup --dry-run
+```
+
+Remove the injected configuration later:
+
+```bash
+npx kimi-memory-setup --undo
+```
+
+## Configure Kimi Code CLI (manual)
+
+If you prefer to configure manually, edit `~/.kimi-code/mcp.json` and add the server.
+
+If you installed from npm with `-g`, use the absolute path to `dist/server.js` inside your global `node_modules`:
+
+```json
+{
+  "mcpServers": {
+    "kimi-memory": {
+      "command": "node",
+      "args": ["/absolute/path/to/global/node_modules/kimi-code-memory-mcp-server/dist/server.js"],
+      "enabled": true
+    }
+  }
+}
+```
+
+Or run it directly via `npx` (no install required):
+
+```json
+{
+  "mcpServers": {
+    "kimi-memory": {
+      "command": "npx",
+      "args": ["-y", "kimi-code-memory-mcp-server"],
+      "enabled": true
+    }
+  }
+}
+```
 
-Edit `~/.kimi-code/mcp.json` and point to the built server entry:
+If you built from source, point to your local `dist/server.js`:
 
 ```json
 {

package/README.zh-CN.md

@@ -7,7 +7,7 @@
 
 一个为 [Kimi Code CLI](https://github.com/MoonshotAI/kimi-code) 提供跨会话记忆的本地 stdio MCP 服务器。
 
-> **注意：** 本包尚未发布到 npm，请按下面的说明从源码安装运行。
+> **注意：** 本包已发布到 npm，包名为 `kimi-code-memory-mcp-server`。你可以直接安装，也可以从源码运行。
 
 面向用户的记忆以普通 Markdown 文件存储在磁盘上。精炼轮次摘要使用本地 SQLite 缓存，但无需向量数据库、图数据库或外部云服务。
 
@@ -40,7 +40,15 @@ Markdown + YAML frontmatter 带来：
 
 ## 安装
 
-目前本包需要从源码安装。需要 Node.js ≥ 18。
+需要 Node.js ≥ 18。
+
+### 从 npm 安装（推荐）
+
+```bash
+npm install -g kimi-code-memory-mcp-server
+```
+
+### 从源码安装
 
 ```bash
 git clone https://github.com/Zehee/kimi-code-memory-mcp-server.git
@@ -49,9 +57,66 @@ npm install
 npm run build
 ```
 
-## 配置 Kimi Code CLI
+## 快速配置（推荐）
+
+从 npm 安装后，运行 setup 命令自动配置 Kimi Code CLI：
+
+```bash
+npx kimi-memory-setup
+```
+
+它会完成：
+
+1. 检测 `~/.kimi-code` 目录。
+2. 在 `~/.kimi-code/AGENTS.md` 顶部注入记忆协议规则。
+3. 将 `memory-manage` Skill 安装到 `~/.kimi-code/skills/memory-manage`。
+4. 在 `~/.kimi-code/mcp.json` 中添加 `kimi-memory` MCP 服务器配置。
+
+预览变更而不写入文件：
+
+```bash
+npx kimi-memory-setup --dry-run
+```
+
+后续如需移除注入的配置：
+
+```bash
+npx kimi-memory-setup --undo
+```
+
+## 配置 Kimi Code CLI（手动）
+
+如果你希望手动配置，编辑 `~/.kimi-code/mcp.json` 并添加服务器。
+
+如果你用 `npm install -g` 安装，使用全局 `node_modules` 中 `dist/server.js` 的绝对路径：
+
+```json
+{
+  "mcpServers": {
+    "kimi-memory": {
+      "command": "node",
+      "args": ["/absolute/path/to/global/node_modules/kimi-code-memory-mcp-server/dist/server.js"],
+      "enabled": true
+    }
+  }
+}
+```
+
+或者直接通过 `npx` 运行（无需安装）：
+
+```json
+{
+  "mcpServers": {
+    "kimi-memory": {
+      "command": "npx",
+      "args": ["-y", "kimi-code-memory-mcp-server"],
+      "enabled": true
+    }
+  }
+}
+```
 
-编辑 `~/.kimi-code/mcp.json`，指向构建后的服务器入口：
+如果你从源码构建，指向本地 `dist/server.js`：
 
 ```json
 {

package/assets/user-agents.md

@@ -0,0 +1,111 @@
+# ⚠️ 强制：每次会话启动必须先执行 `mcp__kimi-memory__bootstrap_workspace()`
+
+**在回答用户任何问题或执行任何操作之前，必须先调用 `mcp__kimi-memory__bootstrap_workspace()`。**
+
+未完成 `bootstrap_workspace()` 前，**严禁**：
+
+- Read / Write / Edit 等文件操作
+- Bash 命令执行
+- 项目代码相关的 MCP 工具调用
+- 任何可能改变项目状态的操作
+
+调用后必须阅读并内化返回内容：
+
+1. `workspace` —— 工作区基本信息
+2. `recentContext` —— 最近对话上下文
+3. `essence` —— `essence/essence.md` 的内容
+4. `memoryIndexTree` —— 记忆目录结构，关注 `[new]` 标记
+5. `notesRefs` —— 是否有待处理的速记
+
+然后向用户汇报当前上下文（任务、状态、下一步、阻塞项）。
+
+---
+
+# Kimi 全局记忆协议
+
+
+## 记忆分类约定
+
+| 类型 | folder | tags | 用途 |
+|------|--------|------|------|
+| 决策 | `memory/decisions/` | `decision` | 架构、接口、流程决策 |
+| 规则 | `memory/rules/` | `rule` | 编码规范、协作红线 |
+| 知识 | `memory/knowledge/` | `knowledge` | 技术栈、业务背景 |
+| 参考 | `memory/reference/` | `reference` | 外部链接、文档引用 |
+| 速记 | `notes/` | `scratch` | 临时便签，不进入长期精要 |
+
+## 什么时候必须 `remember`？
+
+出现以下信号时立即写入 `memory/`：
+
+- 用户说"我们决定…"、"拍板了" → `memory/decisions/`，tag `decision`
+- 用户说"这里必须…"、"红线是…" → `memory/rules/`，tag `rule`
+- 技术选型、接口约定首次出现 → `memory/knowledge/`，tag `knowledge`
+- 定位到 bug 根因和修复方案 → `memory/decisions/` 或 `memory/knowledge/`，tag `fix`
+- 编码风格、目录结构、命名规范 → `memory/rules/`，tag `convention`
+
+## 什么时候不要 `remember`？
+
+- ❌ 临时调试命令或探索性代码
+- ❌ 用户未确认的推测
+- ❌ 一次性报错（没有形成通用结论）
+- ❌ 当前会话的临时上下文（由 `wire.jsonl` 自动捕获）
+- ❌ 已经在 `memory/` 中存在的知识
+
+判断口诀：如果这条信息三天后回头看仍然有价值，才写入 `memory/`。
+
+## 决策守卫
+
+准备执行任何可能改变项目状态的操作之前（写文件、改结构、做选择、引入依赖）：
+
+1. 提取关键词（技术实体 + 动作）
+2. 并行查询：
+   - `search(query, folder="memory")` —— 查决策/规则/知识
+   - `search_context(query)` —— 查历史对话
+3. 命中且一致 → 引用来源后继续
+4. 命中但矛盾 → 向用户报告冲突，请求澄清
+5. 未命中但重要 → 询问用户"这是一个新决策，是否记录？"
+
+## 引用来源的纪律
+
+所有从 `memory/` 召回的关键结论，回复中必须标注来源：
+
+- ✅ "根据 `memory/decisions/xxx`，我们决定…"
+- ❌ "我们之前决定…"（无来源）
+
+## 主题追溯
+
+当用户要求追溯某个主题时：
+
+1. 确认主题范围
+2. 用 `search_context` 打捞相关 turns（调用时会自动按 90 秒时间簇扩展并提炼命中的 turns）
+3. 用 `list_search_views` 查看最近的搜索视图，作为 theme 挂载的候选集
+4. 对搜索命中的 turns 做语义分析，用 `tag_theme` 把 genuinely belongs to the theme 的 turns 逐个挂载到 `themes/<theme>.json`
+5. 如果候选不足，用新的 query 再次 `search_context` 打捞，生成新的 search view
+6. 用 `trace_theme` 输出主题时间线
+
+`search_context` 已内置提炼能力，不需要单独调用 `refine_session_turns`。
+
+## 记忆卫生检查清单
+
+每次调用 `remember` 前：
+
+- [ ] 这条信息三天后还有价值吗？
+- [ ] 是否已有重复记忆？（先 `search`）
+- [ ] `folder` 选对了吗？项目相关不进 `notes/`
+- [ ] `tags` 是否包含最贴切的分类标签？
+- [ ] 正文是否包含决策原因/影响范围/相关文件？
+
+每次做决策前：
+
+- [ ] 是否已 `search` 相关历史决策？
+- [ ] 是否已检查 `essence.md` 中的核心约束？
+- [ ] 如果与历史记忆冲突，是否已向用户澄清？
+
+## 禁止行为
+
+- 禁止用 `folder="notes"` 保存项目相关记忆
+- 禁止写入空 tags
+- 禁止只口头答应而不调用工具
+- 禁止手动修改 `~/.kimi-code/sessions/` 下的 `wire.jsonl`
+- **当用户输入以 `？` / `?` 结尾时，禁止执行任何写入/修改/构建/运行等改变项目状态的操作**：只能回答问题和进行讨论，必须等待用户明确说出「确定」「可以」「开始吧」等指令后才能行动

package/dist/setup-cli.d.ts

@@ -0,0 +1,11 @@
+#!/usr/bin/env node
+/**
+ * CLI entry for kimi-code-memory-mcp-server setup.
+ *
+ * Usage:
+ *   npx kimi-code-memory-mcp-server setup
+ *   npx kimi-code-memory-mcp-server setup --undo
+ *   npx kimi-code-memory-mcp-server setup --dry-run
+ *   npx kimi-code-memory-mcp-server setup --kimi-code-home /path/to/.kimi-code
+ */
+export {};

package/dist/setup-cli.js

@@ -0,0 +1,84 @@
+#!/usr/bin/env node
+/**
+ * CLI entry for kimi-code-memory-mcp-server setup.
+ *
+ * Usage:
+ *   npx kimi-code-memory-mcp-server setup
+ *   npx kimi-code-memory-mcp-server setup --undo
+ *   npx kimi-code-memory-mcp-server setup --dry-run
+ *   npx kimi-code-memory-mcp-server setup --kimi-code-home /path/to/.kimi-code
+ */
+import { runSetup } from './setup.js';
+function parseArgs(argv) {
+    const options = {};
+    for (let i = 0; i < argv.length; i++) {
+        const arg = argv[i];
+        switch (arg) {
+            case '--dry-run':
+                options.dryRun = true;
+                break;
+            case '--force':
+                options.force = true;
+                break;
+            case '--undo':
+                options.undo = true;
+                break;
+            case '--kimi-code-home':
+                options.kimiCodeHome = argv[++i];
+                break;
+            case '--help':
+            case '-h':
+                printHelp();
+                process.exit(0);
+                break;
+            default:
+                if (arg.startsWith('-')) {
+                    console.error(`Unknown option: ${arg}`);
+                    printHelp();
+                    process.exit(1);
+                }
+                break;
+        }
+    }
+    return options;
+}
+function printHelp() {
+    console.log(`kimi-code-memory-mcp-server setup
+
+Integrates the MCP server with Kimi Code CLI by updating:
+  - ~/.kimi-code/AGENTS.md
+  - ~/.kimi-code/skills/memory-manage
+  - ~/.kimi-code/mcp.json
+
+Options:
+  --dry-run            Preview changes without writing files
+  --force              Overwrite existing skill without prompting
+  --undo               Remove injected configuration
+  --kimi-code-home <p> Use a custom Kimi Code home directory
+  --help, -h           Show this help
+`);
+}
+async function main() {
+    const options = parseArgs(process.argv.slice(2));
+    const result = await runSetup(options);
+    console.log(`Kimi Code home: ${result.kimiCodeHome}`);
+    console.log(`AGENTS.md:      ${result.agentsMdPath}`);
+    console.log(`Skill:          ${result.skillPath}`);
+    console.log(`mcp.json:       ${result.mcpJsonPath}`);
+    console.log('');
+    for (const action of result.actions) {
+        console.log(`✓ ${action}`);
+    }
+    if (result.warnings.length > 0) {
+        console.log('');
+        console.log('Warnings:');
+        for (const warning of result.warnings) {
+            console.log(`⚠ ${warning}`);
+        }
+    }
+}
+main().catch((err) => {
+    console.error('Setup failed:', err instanceof Error ? err.message : String(err));
+    process.exit(1);
+});
+//# sourceMappingURL=setup-cli.js.map

package/dist/setup-cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup-cli.js","sourceRoot":"","sources":["../src/setup-cli.ts"],"names":[],"mappings":";AACA;;;;;;;;GAQG;AAEH,OAAO,EAAE,QAAQ,EAAqB,MAAM,YAAY,CAAC;AAEzD,SAAS,SAAS,CAAC,IAAc;IAC/B,MAAM,OAAO,GAAiB,EAAE,CAAC;IACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACrC,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,QAAQ,GAAG,EAAE,CAAC;YACZ,KAAK,WAAW;gBACd,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;gBACtB,MAAM;YACR,KAAK,SAAS;gBACZ,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC;gBACrB,MAAM;YACR,KAAK,QAAQ;gBACX,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;gBACpB,MAAM;YACR,KAAK,kBAAkB;gBACrB,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;gBACjC,MAAM;YACR,KAAK,QAAQ,CAAC;YACd,KAAK,IAAI;gBACP,SAAS,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAChB,MAAM;YACR;gBACE,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBACxB,OAAO,CAAC,KAAK,CAAC,mBAAmB,GAAG,EAAE,CAAC,CAAC;oBACxC,SAAS,EAAE,CAAC;oBACZ,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAClB,CAAC;gBACD,MAAM;QACV,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,SAAS;IAChB,OAAO,CAAC,GAAG,CAAC;;;;;;;;;;;;;CAab,CAAC,CAAC;AACH,CAAC;AAED,KAAK,UAAU,IAAI;IACjB,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACjD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,CAAC;IAEvC,OAAO,CAAC,GAAG,CAAC,mBAAmB,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,CAAC,mBAAmB,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,CAAC,mBAAmB,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,mBAAmB,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;IACrD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAEhB,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACpC,OAAO,CAAC,GAAG,CAAC,KAAK,MAAM,EAAE,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAChB,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACzB,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACtC,OAAO,CAAC,GAAG,CAAC,KAAK,OAAO,EAAE,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;AACH,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACnB,OAAO,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;IACjF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/setup.d.ts

@@ -0,0 +1,33 @@
+/**
+ * Setup / teardown helpers for integrating the MCP server with Kimi Code CLI.
+ *
+ * This module modifies the user's Kimi Code home directory:
+ *   - ~/.kimi-code/AGENTS.md      (injected memory protocol block)
+ *   - ~/.kimi-code/skills/memory-manage
+ *   - ~/.kimi-code/mcp.json       (kimi-memory server entry)
+ *
+ * All modifications are guarded by backup + marked injection blocks so they
+ * can be undone later.
+ */
+export interface SetupResult {
+    kimiCodeHome: string;
+    agentsMdPath: string;
+    skillPath: string;
+    mcpJsonPath: string;
+    actions: string[];
+    warnings: string[];
+}
+export interface SetupOptions {
+    /** Preview changes without writing anything. */
+    dryRun?: boolean;
+    /** Overwrite existing skill / mcp.json entry without prompting. */
+    force?: boolean;
+    /** Remove injected blocks instead of adding them. */
+    undo?: boolean;
+    /** Override the detected Kimi Code home directory. */
+    kimiCodeHome?: string;
+}
+/**
+ * Run setup or teardown.
+ */
+export declare function runSetup(options?: SetupOptions): Promise<SetupResult>;

package/dist/setup.js

@@ -0,0 +1,206 @@
+/**
+ * Setup / teardown helpers for integrating the MCP server with Kimi Code CLI.
+ *
+ * This module modifies the user's Kimi Code home directory:
+ *   - ~/.kimi-code/AGENTS.md      (injected memory protocol block)
+ *   - ~/.kimi-code/skills/memory-manage
+ *   - ~/.kimi-code/mcp.json       (kimi-memory server entry)
+ *
+ * All modifications are guarded by backup + marked injection blocks so they
+ * can be undone later.
+ */
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+const INJECT_START = '<!-- KIMI-MEMORY-INJECTED-START -->';
+const INJECT_END = '<!-- KIMI-MEMORY-INJECTED-END -->';
+function getKimiCodeHome(options) {
+    if (options.kimiCodeHome)
+        return path.resolve(options.kimiCodeHome);
+    const fromEnv = process.env.KIMI_CODE_HOME;
+    if (fromEnv)
+        return path.resolve(fromEnv);
+    return path.join(os.homedir(), '.kimi-code');
+}
+function fileExists(p) {
+    try {
+        return fs.existsSync(p);
+    }
+    catch {
+        return false;
+    }
+}
+function readText(p) {
+    try {
+        return fs.readFileSync(p, 'utf8');
+    }
+    catch {
+        return null;
+    }
+}
+function writeText(p, content, dryRun) {
+    if (dryRun)
+        return;
+    fs.mkdirSync(path.dirname(p), { recursive: true });
+    fs.writeFileSync(p, content, 'utf8');
+}
+function backupFile(p, dryRun) {
+    if (dryRun || !fileExists(p))
+        return;
+    const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+    const backupPath = `${p}.bak.${timestamp}`;
+    fs.copyFileSync(p, backupPath);
+}
+function getPackageRoot() {
+    // __dirname in dist/ points to dist/, package root is one level up.
+    return path.join(path.dirname(fileURLToPath(import.meta.url)), '..');
+}
+import { fileURLToPath } from 'url';
+/**
+ * Inject or replace the marked block in AGENTS.md.
+ */
+function updateAgentsMd(agentsPath, injectedContent, options) {
+    const dryRun = !!options.dryRun;
+    const undo = !!options.undo;
+    const existing = readText(agentsPath) ?? '';
+    const startIdx = existing.indexOf(INJECT_START);
+    const endIdx = existing.indexOf(INJECT_END);
+    let newContent;
+    if (undo) {
+        if (startIdx === -1 || endIdx === -1 || endIdx <= startIdx) {
+            return 'No injected block found; nothing to undo.';
+        }
+        // Remove the block plus surrounding blank lines.
+        const before = existing.slice(0, startIdx).replace(/\n+\s*$/, '');
+        const after = existing.slice(endIdx + INJECT_END.length).replace(/^\s*\n+/, '');
+        newContent = before ? before + '\n\n' + after : after;
+        if (!dryRun)
+            backupFile(agentsPath, dryRun);
+        writeText(agentsPath, newContent, dryRun);
+        return 'Removed injected memory protocol block.';
+    }
+    const block = [
+        INJECT_START,
+        '<!-- 以下内容由 kimi-code-memory-mcp-server 自动生成，请勿手动编辑此区块。 -->',
+        '<!-- 如需更新或移除，请运行 npx kimi-code-memory-mcp-server setup --undo -->',
+        injectedContent.trim(),
+        INJECT_END,
+    ].join('\n');
+    if (startIdx !== -1 && endIdx !== -1 && endIdx > startIdx) {
+        // Replace existing block.
+        newContent = existing.slice(0, startIdx) + block + existing.slice(endIdx + INJECT_END.length);
+        if (!dryRun)
+            backupFile(agentsPath, dryRun);
+        writeText(agentsPath, newContent, dryRun);
+        return 'Updated existing injected memory protocol block.';
+    }
+    // No existing block: prepend to file.
+    const separator = existing.trim() ? '\n\n' : '';
+    newContent = block + separator + existing;
+    if (!dryRun)
+        backupFile(agentsPath, dryRun);
+    writeText(agentsPath, newContent, dryRun);
+    return 'Injected memory protocol block at the top of AGENTS.md.';
+}
+/**
+ * Copy the bundled skill into the user's Kimi Code skills directory.
+ */
+function installSkill(skillTarget, options) {
+    const dryRun = !!options.dryRun;
+    const force = !!options.force;
+    const undo = !!options.undo;
+    if (undo) {
+        if (!fileExists(skillTarget))
+            return 'Skill not installed; nothing to undo.';
+        if (!dryRun)
+            fs.rmSync(skillTarget, { recursive: true, force: true });
+        return 'Removed memory-manage skill.';
+    }
+    if (fileExists(skillTarget) && !force) {
+        return 'Skill already exists; use --force to overwrite.';
+    }
+    const source = path.join(getPackageRoot(), 'skills', 'memory-manage');
+    if (!fileExists(source)) {
+        throw new Error(`Bundled skill not found at ${source}`);
+    }
+    if (!dryRun) {
+        fs.rmSync(skillTarget, { recursive: true, force: true });
+        fs.cpSync(source, skillTarget, { recursive: true });
+    }
+    return force ? 'Replaced memory-manage skill.' : 'Installed memory-manage skill.';
+}
+/**
+ * Add or remove the kimi-memory MCP server entry in mcp.json.
+ */
+function updateMcpJson(mcpPath, options) {
+    const dryRun = !!options.dryRun;
+    const undo = !!options.undo;
+    let config = {};
+    const existingText = readText(mcpPath);
+    if (existingText) {
+        try {
+            config = JSON.parse(existingText);
+            if (typeof config !== 'object' || config === null)
+                config = {};
+        }
+        catch {
+            throw new Error(`Existing ${mcpPath} is not valid JSON.`);
+        }
+    }
+    if (!config.mcpServers || typeof config.mcpServers !== 'object') {
+        config.mcpServers = {};
+    }
+    if (undo) {
+        if (!Object.prototype.hasOwnProperty.call(config.mcpServers, 'kimi-memory')) {
+            return 'kimi-memory MCP server entry not found; nothing to undo.';
+        }
+        delete config.mcpServers['kimi-memory'];
+        if (!dryRun)
+            writeText(mcpPath, JSON.stringify(config, null, 2) + '\n', dryRun);
+        return 'Removed kimi-memory MCP server entry.';
+    }
+    const serverEntry = {
+        command: 'npx',
+        args: ['-y', 'kimi-code-memory-mcp-server'],
+        enabled: true,
+    };
+    const hadEntry = Object.prototype.hasOwnProperty.call(config.mcpServers, 'kimi-memory');
+    config.mcpServers['kimi-memory'] = serverEntry;
+    if (!dryRun)
+        writeText(mcpPath, JSON.stringify(config, null, 2) + '\n', dryRun);
+    return hadEntry ? 'Updated kimi-memory MCP server entry.' : 'Added kimi-memory MCP server entry.';
+}
+/**
+ * Run setup or teardown.
+ */
+export async function runSetup(options = {}) {
+    const kimiCodeHome = getKimiCodeHome(options);
+    const agentsMdPath = path.join(kimiCodeHome, 'AGENTS.md');
+    const skillPath = path.join(kimiCodeHome, 'skills', 'memory-manage');
+    const mcpJsonPath = path.join(kimiCodeHome, 'mcp.json');
+    const actions = [];
+    const warnings = [];
+    if (!fileExists(kimiCodeHome)) {
+        warnings.push(`Kimi Code home not found at ${kimiCodeHome}. Is Kimi Code CLI installed?`);
+    }
+    // AGENTS.md
+    const injectedSource = path.join(getPackageRoot(), 'assets', 'user-agents.md');
+    const injectedContent = readText(injectedSource);
+    if (!injectedContent) {
+        throw new Error(`Bundled user-agents.md template not found at ${injectedSource}`);
+    }
+    actions.push(updateAgentsMd(agentsMdPath, injectedContent, options));
+    // Skill
+    actions.push(installSkill(skillPath, options));
+    // mcp.json
+    actions.push(updateMcpJson(mcpJsonPath, options));
+    return {
+        kimiCodeHome,
+        agentsMdPath,
+        skillPath,
+        mcpJsonPath,
+        actions,
+        warnings,
+    };
+}
+//# sourceMappingURL=setup.js.map

package/dist/setup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup.js","sourceRoot":"","sources":["../src/setup.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,IAAI,CAAC;AAEpB,MAAM,YAAY,GAAG,qCAAqC,CAAC;AAC3D,MAAM,UAAU,GAAG,mCAAmC,CAAC;AAsBvD,SAAS,eAAe,CAAC,OAAqB;IAC5C,IAAI,OAAO,CAAC,YAAY;QAAE,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IACpE,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IAC3C,IAAI,OAAO;QAAE,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC1C,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,YAAY,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,UAAU,CAAC,CAAS;IAC3B,IAAI,CAAC;QACH,OAAO,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IAC1B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,SAAS,QAAQ,CAAC,CAAS;IACzB,IAAI,CAAC;QACH,OAAO,EAAE,CAAC,YAAY,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;IACpC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,CAAS,EAAE,OAAe,EAAE,MAAe;IAC5D,IAAI,MAAM;QAAE,OAAO;IACnB,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;AACvC,CAAC;AAED,SAAS,UAAU,CAAC,CAAS,EAAE,MAAe;IAC5C,IAAI,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;QAAE,OAAO;IACrC,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IACjE,MAAM,UAAU,GAAG,GAAG,CAAC,QAAQ,SAAS,EAAE,CAAC;IAC3C,EAAE,CAAC,YAAY,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,cAAc;IACrB,oEAAoE;IACpE,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;AACvE,CAAC;AAED,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AAEpC;;GAEG;AACH,SAAS,cAAc,CAAC,UAAkB,EAAE,eAAuB,EAAE,OAAqB;IACxF,MAAM,MAAM,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;IAChC,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;IAC5B,MAAM,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;IAE5C,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAChD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAE5C,IAAI,UAAkB,CAAC;IAEvB,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,QAAQ,KAAK,CAAC,CAAC,IAAI,MAAM,KAAK,CAAC,CAAC,IAAI,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC3D,OAAO,2CAA2C,CAAC;QACrD,CAAC;QACD,iDAAiD;QACjD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAClE,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChF,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;QACtD,IAAI,CAAC,MAAM;YAAE,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAC5C,SAAS,CAAC,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAC1C,OAAO,yCAAyC,CAAC;IACnD,CAAC;IAED,MAAM,KAAK,GAAG;QACZ,YAAY;QACZ,4DAA4D;QAC5D,mEAAmE;QACnE,eAAe,CAAC,IAAI,EAAE;QACtB,UAAU;KACX,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEb,IAAI,QAAQ,KAAK,CAAC,CAAC,IAAI,MAAM,KAAK,CAAC,CAAC,IAAI,MAAM,GAAG,QAAQ,EAAE,CAAC;QAC1D,0BAA0B;QAC1B,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,GAAG,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC;QAC9F,IAAI,CAAC,MAAM;YAAE,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAC5C,SAAS,CAAC,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAC1C,OAAO,kDAAkD,CAAC;IAC5D,CAAC;IAED,sCAAsC;IACtC,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;IAChD,UAAU,GAAG,KAAK,GAAG,SAAS,GAAG,QAAQ,CAAC;IAC1C,IAAI,CAAC,MAAM;QAAE,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC5C,SAAS,CAAC,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAC1C,OAAO,yDAAyD,CAAC;AACnE,CAAC;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,WAAmB,EAAE,OAAqB;IAC9D,MAAM,MAAM,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;IAChC,MAAM,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;IAC9B,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;IAE5B,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;YAAE,OAAO,uCAAuC,CAAC;QAC7E,IAAI,CAAC,MAAM;YAAE,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACtE,OAAO,8BAA8B,CAAC;IACxC,CAAC;IAED,IAAI,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACtC,OAAO,iDAAiD,CAAC;IAC3D,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC;IACtE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,8BAA8B,MAAM,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACzD,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC;IACD,OAAO,KAAK,CAAC,CAAC,CAAC,+BAA+B,CAAC,CAAC,CAAC,gCAAgC,CAAC;AACpF,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,OAAe,EAAE,OAAqB;IAC3D,MAAM,MAAM,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;IAChC,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;IAE5B,IAAI,MAAM,GAA6C,EAAE,CAAC;IAC1D,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;IACvC,IAAI,YAAY,EAAE,CAAC;QACjB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAClC,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI;gBAAE,MAAM,GAAG,EAAE,CAAC;QACjE,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,KAAK,CAAC,YAAY,OAAO,qBAAqB,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;QAChE,MAAM,CAAC,UAAU,GAAG,EAAE,CAAC;IACzB,CAAC;IAED,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,EAAE,CAAC;YAC5E,OAAO,0DAA0D,CAAC;QACpE,CAAC;QACD,OAAO,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM;YAAE,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,MAAM,CAAC,CAAC;QAChF,OAAO,uCAAuC,CAAC;IACjD,CAAC;IAED,MAAM,WAAW,GAAG;QAClB,OAAO,EAAE,KAAK;QACd,IAAI,EAAE,CAAC,IAAI,EAAE,6BAA6B,CAAC;QAC3C,OAAO,EAAE,IAAI;KACd,CAAC;IAEF,MAAM,QAAQ,GAAG,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;IACxF,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,GAAG,WAAW,CAAC;IAE/C,IAAI,CAAC,MAAM;QAAE,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,MAAM,CAAC,CAAC;IAChF,OAAO,QAAQ,CAAC,CAAC,CAAC,uCAAuC,CAAC,CAAC,CAAC,qCAAqC,CAAC;AACpG,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,UAAwB,EAAE;IACvD,MAAM,YAAY,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IAC9C,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;IAC1D,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC;IACrE,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;IAExD,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;QAC9B,QAAQ,CAAC,IAAI,CAAC,+BAA+B,YAAY,+BAA+B,CAAC,CAAC;IAC5F,CAAC;IAED,YAAY;IACZ,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,QAAQ,EAAE,gBAAgB,CAAC,CAAC;IAC/E,MAAM,eAAe,GAAG,QAAQ,CAAC,cAAc,CAAC,CAAC;IACjD,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CAAC,gDAAgD,cAAc,EAAE,CAAC,CAAC;IACpF,CAAC;IACD,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC,CAAC;IAErE,QAAQ;IACR,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC;IAE/C,WAAW;IACX,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC;IAElD,OAAO;QACL,YAAY;QACZ,YAAY;QACZ,SAAS;QACT,WAAW;QACX,OAAO;QACP,QAAQ;KACT,CAAC;AACJ,CAAC"}

package/dist/version.d.ts

@@ -6,4 +6,4 @@
  * during protocol initialization and must match the repository (package)
  * version unless explicitly noted otherwise.
  */
-export declare const VERSION = "0.1.0";
+export declare const VERSION = "0.1.1";

package/dist/version.js

@@ -6,5 +6,5 @@
  * during protocol initialization and must match the repository (package)
  * version unless explicitly noted otherwise.
  */
-export const VERSION = '0.1.0';
+export const VERSION = '0.1.1';
 //# sourceMappingURL=version.js.map

package/package.json

@@ -1,18 +1,20 @@
 {
   "name": "kimi-code-memory-mcp-server",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Local stdio MCP server providing cross-session memory for Kimi Code CLI",
   "main": "dist/server.js",
   "types": "dist/server.d.ts",
   "type": "module",
   "bin": {
-    "kimi-code-memory-mcp-server": "dist/server.js"
+    "kimi-code-memory-mcp-server": "dist/server.js",
+    "kimi-memory-setup": "dist/setup-cli.js"
   },
   "files": [
     "dist/",
     "docs/",
     "examples/",
     "skills/",
+    "assets/",
     "AGENTS.md",
     "README.md",
     "README.zh-CN.md",