ningzhi-agent-cli 0.0.1

package/dist/agent/colors.js

@@ -0,0 +1,92 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.color = void 0;
+exports.initColors = initColors;
+exports.formatToolLog = formatToolLog;
+// chalk 实例，懒加载，未初始化时为 null
+let _chalk = null;
+// 初始化颜色模块，必须在首次使用前调用一次
+async function initColors() {
+    const mod = await Promise.resolve().then(() => __importStar(require("chalk")));
+    _chalk = mod.default;
+}
+// chalk 未就绪时返回的空实现：任意链式调用都返回原字符串，避免抛错
+function createPlainFallback() {
+    // get 拦截属性访问（如 .red、.bold），返回新的 fallback，使链式调用不会断
+    // apply 拦截函数调用（如 chalk("text")），返回首个参数原样输出
+    const handler = {
+        get: () => createPlainFallback(),
+        apply: (_target, _thisArg, args) => args[0] ?? "",
+    };
+    // 目标函数本身也实现为「返回首个参数」，与 apply 行为保持一致
+    const fn = (...args) => args[0] ?? "";
+    // 用 Proxy 包裹函数，模拟 chalk 的链式 + 调用接口
+    return new Proxy(fn, handler);
+}
+// 获取 chalk 实例，未初始化则降级为空实现
+function ck() {
+    if (!_chalk) {
+        return createPlainFallback();
+    }
+    return _chalk;
+}
+// 统一的颜色输出工具集合
+exports.color = {
+    banner: (text) => ck().cyanBright.bold(text),
+    userPrefix: () => ck().greenBright.bold("You: "),
+    aiPrefix: () => ck().cyanBright.bold("AI: "),
+    goodbye: (text) => ck().yellowBright(text),
+    cancelled: (text) => ck().yellowBright.bold(text),
+    error: (text) => ck().redBright.bold(text),
+    // 工具调用日志相关样式
+    toolTag: () => ck().magentaBright("[Tool]"), // magentaBright 粉色高亮
+    toolName: (name) => ck().yellowBright(name), // yellowBright 黄色高亮
+    toolAction: () => ck().gray("called:"), // gray 灰色
+    toolArg: (arg) => ck().cyan(arg), // cyan 青色
+    gray: (text) => ck().gray(text),
+    tokenInfo: (text) => ck().gray(text),
+    warn: (text) => ck().yellow(text),
+    info: (text) => ck().blueBright(text),
+    good: (text) => ck().greenBright(text),
+};
+// 格式化工具调用日志：形如 [Tool] toolName called: detail
+function formatToolLog(name, detail) {
+    const parts = [exports.color.toolTag(), exports.color.toolName(name), exports.color.toolAction()];
+    if (detail) {
+        parts.push(exports.color.toolArg(detail));
+    }
+    return "\n" + parts.join(" ");
+}

package/dist/agent/commands.d.ts

@@ -0,0 +1,7 @@
+export declare let threadId: string;
+export interface ChatCommand {
+    name: string;
+    description: string;
+    execute(args: string[]): void | Promise<void>;
+}
+export declare const commands: Map<string, ChatCommand>;

package/dist/agent/commands.js

@@ -0,0 +1,73 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.commands = exports.threadId = void 0;
+const node_crypto_1 = require("node:crypto");
+const cli_table3_1 = __importDefault(require("cli-table3"));
+const colors_1 = require("./colors");
+const db_1 = require("./db");
+const agent_1 = require("./agent");
+exports.threadId = (0, node_crypto_1.randomUUID)();
+exports.commands = new Map();
+// 新建会话
+exports.commands.set("new", {
+    name: "new",
+    description: "Start a new chat session",
+    execute() {
+        exports.threadId = (0, node_crypto_1.randomUUID)();
+        console.log(colors_1.color.goodbye(`\nNew session started ${exports.threadId}.\n`));
+    },
+});
+// 列出最近会话
+exports.commands.set("sessions", {
+    name: "sessions",
+    description: "List recent chat sessions",
+    execute() {
+        const sessions = (0, db_1.listRecentSessions)();
+        if (sessions.length === 0) {
+            console.log("\n暂无聊天记录\n");
+            return;
+        }
+        const table = new cli_table3_1.default({
+            head: ["thread_id", "最后用户输入的问题", "时间"],
+        });
+        for (const s of sessions) {
+            table.push([s.thread_id, s.last_question, s.last_ts]);
+        }
+        console.log("\n" + table.toString() + "\n");
+    },
+});
+// 切换会话
+exports.commands.set("rewind", {
+    name: "rewind",
+    description: "Restore a chat session by thread_id",
+    execute(args) {
+        const id = args[0];
+        if (!id) {
+            console.log(colors_1.color.error("\nError: please provide a thread_id. Usage: /rewind <thread_id>\n"));
+            return;
+        }
+        if (!(0, db_1.threadIdExists)(id)) {
+            console.log(colors_1.color.error(`\nError: thread_id "${id}" not found.\n`));
+            return;
+        }
+        exports.threadId = id;
+        console.log(colors_1.color.goodbye(`\nRestored session ${exports.threadId}.\n`));
+    },
+});
+// 压缩上下文
+exports.commands.set("compact", {
+    name: "compact",
+    description: "Compress context for the current session",
+    async execute() {
+        const result = await (0, agent_1.compressContext)(exports.threadId);
+        if (result.didCompress) {
+            console.log(colors_1.color.goodbye(`\nContext compressed (count: ${result.count}).\n`));
+        }
+        else {
+            console.log("\nNo context to compress.\n");
+        }
+    },
+});

package/dist/agent/config.d.ts

@@ -0,0 +1,38 @@
+export declare const WORKSPACE_DIR: string;
+export declare const CONFIG_PATH: string;
+interface ModelConfig {
+    model: string;
+    apiKey: string;
+    baseURL: string;
+}
+export interface HookConfig {
+    matcher: string;
+    command: string;
+}
+export interface HooksConfig {
+    hooks: {
+        PreToolUse?: HookConfig[];
+        PostToolUse?: HookConfig[];
+        SessionStart?: HookConfig[];
+    };
+}
+export interface McpServerConfig {
+    command?: string;
+    args?: string[];
+    env?: Record<string, string>;
+    url?: string;
+    headers?: Record<string, string>;
+}
+interface NingzhiConfig {
+    model?: ModelConfig;
+    env?: Record<string, string>;
+    hooks?: HooksConfig["hooks"];
+    mcpServers?: Record<string, McpServerConfig>;
+}
+export declare function loadConfig(): NingzhiConfig;
+export declare function getModelConfig(): ModelConfig;
+export declare function getEnv(key: string): string | undefined;
+export declare function getHooksConfig(): HooksConfig;
+export declare function getMCPServerConfig(): Record<string, McpServerConfig>;
+export declare function clearConfigCache(): void;
+export {};

package/dist/agent/config.js

@@ -0,0 +1,119 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CONFIG_PATH = exports.WORKSPACE_DIR = void 0;
+exports.loadConfig = loadConfig;
+exports.getModelConfig = getModelConfig;
+exports.getEnv = getEnv;
+exports.getHooksConfig = getHooksConfig;
+exports.getMCPServerConfig = getMCPServerConfig;
+exports.clearConfigCache = clearConfigCache;
+const fs = __importStar(require("node:fs"));
+const path = __importStar(require("node:path"));
+const os = __importStar(require("node:os"));
+exports.WORKSPACE_DIR = path.join(os.homedir(), ".ningzhiAgentCli");
+exports.CONFIG_PATH = path.join(exports.WORKSPACE_DIR, "ningzhi.json");
+let cachedConfig = null;
+function loadConfig() {
+    if (cachedConfig) {
+        return cachedConfig;
+    }
+    if (!fs.existsSync(exports.CONFIG_PATH)) {
+        throw new Error(`找不到配置文件 ${exports.CONFIG_PATH}\n\n请创建该文件并配置相关信息，例如：\n{\n  "model": {\n    "model": "kimi-k2.6",\n    "apiKey": "your-api-key",\n    "baseURL": "https://api.moonshot.cn/v1"\n  },\n  "env": {\n    "TAVILY_API_KEY": "your-tavily-api-key"\n  }\n}`);
+    }
+    let config;
+    try {
+        const content = fs.readFileSync(exports.CONFIG_PATH, "utf-8");
+        config = JSON.parse(content);
+    }
+    catch (e) {
+        throw new Error(`配置文件 ${exports.CONFIG_PATH} 解析失败：${e instanceof Error ? e.message : String(e)}`);
+    }
+    cachedConfig = config;
+    return config;
+}
+function getModelConfig() {
+    const config = loadConfig();
+    if (!config.model || typeof config.model !== "object") {
+        throw new Error(`配置文件 ${exports.CONFIG_PATH} 中缺少 model 对象，请参考 https://chat.ningzhi2.site/getStart`);
+    }
+    if (!config.model.model) {
+        throw new Error(`配置文件 ${exports.CONFIG_PATH} 中缺少 model.model 字段，请参考 https://chat.ningzhi2.site/getStart`);
+    }
+    if (!config.model.apiKey) {
+        throw new Error(`配置文件 ${exports.CONFIG_PATH} 中缺少 model.apiKey 字段，请参考 https://chat.ningzhi2.site/getStart`);
+    }
+    if (!config.model.baseURL) {
+        throw new Error(`配置文件 ${exports.CONFIG_PATH} 中缺少 model.baseURL 字段，请参考 https://chat.ningzhi2.site/getStart`);
+    }
+    return config.model;
+}
+function getEnv(key) {
+    const config = loadConfig();
+    return config.env?.[key];
+}
+function getHooksConfig() {
+    try {
+        const config = loadConfig();
+        return { hooks: config.hooks || {} };
+    }
+    catch {
+        return { hooks: {} };
+    }
+}
+function getMCPServerConfig() {
+    try {
+        const content = fs.readFileSync(exports.CONFIG_PATH, "utf-8");
+        const parsed = JSON.parse(content);
+        if (!parsed.mcpServers || typeof parsed.mcpServers !== "object") {
+            console.warn('[MCP] ningzhi.json missing "mcpServers" field, skipping');
+            return {};
+        }
+        return parsed.mcpServers;
+    }
+    catch (err) {
+        // `ENOENT` 是 Node.js 中一个标准的系统错误码，全称是 Error NO ENTry （无此条目/文件不存在）
+        // 当调用`readFileSync` 时，如果`ningzhi.json` 文件 不存在 ，
+        // Node.js 会抛出一个带有`err.code === "ENOENT"` 的错误。
+        if (err.code === "ENOENT") {
+            return {};
+        }
+        console.warn(`[MCP] Failed to load ningzhi.json: ${err.message}, skipping`);
+        return {};
+    }
+}
+function clearConfigCache() {
+    cachedConfig = null;
+}

package/dist/agent/context.d.ts

@@ -0,0 +1,17 @@
+import { BaseMessage } from "@langchain/core/messages";
+export declare function getModelContextLimit(): number;
+/**
+ * 压缩消息为简洁的摘要
+ * @param messages 消息数组
+ * @param existingSummary 已存在的摘要
+ * @returns 压缩后的摘要
+ */
+export declare function compressMessages(messages: BaseMessage[], existingSummary: string | null): Promise<string>;
+/**
+ * 计算安全的压缩边界索引，确保保留区域内的 tool_calls 和 ToolMessage 配对完整。
+ * 防止 破坏了消息列表的完整度（ ToolMessage 匹配不到 对应的 AIMessage ），导致报错。
+ *
+ * 策略：从 messages.length - minKeep 开始，如果边界切在了 ToolMessage 或其对应的
+ * AIMessage 之后，就把边界往前移动，直到整个工具调用单元都被保留。
+ */
+export declare function findSafeCompressionIndex(messages: BaseMessage[], minKeep: number): number;

package/dist/agent/context.js

@@ -0,0 +1,146 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getModelContextLimit = getModelContextLimit;
+exports.compressMessages = compressMessages;
+exports.findSafeCompressionIndex = findSafeCompressionIndex;
+const messages_1 = require("@langchain/core/messages");
+const model_1 = require("./model");
+const config_1 = require("./config");
+const MODEL_CONTEXT_LIMITS = {
+    // Moonshot / Kimi
+    "moonshot-v1-8k": 8192,
+    "moonshot-v1-32k": 32768,
+    "moonshot-v1-128k": 131072,
+    "kimi-k2.6": 256000,
+    // DeepSeek
+    "deepseek-chat": 1048576,
+    "deepseek-reasoner": 1048576,
+    "deepseek-v3": 65536,
+    "deepseek-v3.1": 131072,
+    "deepseek-v4": 1048576,
+    "deepseek-r1": 65536,
+    // MiniMax
+    "minimax-text-01": 1048576,
+    "minimax-m1": 1048576,
+    "minimax-m3": 1048576,
+    // GLM (智谱)
+    "glm-4": 131072,
+    "glm-4-plus": 131072,
+    "glm-4-air": 131072,
+    "glm-4-flash": 131072,
+    "glm-4v": 131072,
+    // Qwen (Alibaba)
+    "qwen-max": 32768,
+    "qwen-plus": 131072,
+    "qwen-plus-2025-07-28": 1048576,
+    "qwen-turbo": 1048576,
+    "qwen-long": 10485760,
+    // Xiaomi MiMo
+    "mimo-7b": 32768,
+    "mimo-v2.5": 1048576,
+};
+function getModelContextLimit() {
+    const modelName = (0, config_1.getModelConfig)().model || "";
+    return MODEL_CONTEXT_LIMITS[modelName.toLowerCase()] || 128000;
+}
+/**
+ * 格式化消息为压缩格式
+ * @param messages 消息数组
+ * @returns 压缩后的消息字符串
+ */
+function formatMessagesForCompression(messages) {
+    return messages
+        .map((msg) => {
+        const role = msg.type;
+        // 格式化消息内容为字符串
+        let content = typeof msg.content === "string"
+            ? msg.content
+            : JSON.stringify(msg.content);
+        // 如果是 AI 消息且有工具调用，添加工具调用信息
+        if (role === "ai" && msg.tool_calls?.length) {
+            content += `\n[Tool Calls: ${JSON.stringify(msg.tool_calls)}]`;
+        }
+        // 格式化消息为压缩格式,  [消息角色]：消息内容
+        return `[${role}] ${content}`;
+    })
+        .join("\n\n");
+}
+/**
+ * 压缩消息为简洁的摘要
+ * @param messages 消息数组
+ * @param existingSummary 已存在的摘要
+ * @returns 压缩后的摘要
+ */
+async function compressMessages(messages, existingSummary) {
+    // 如果消息数组为空，直接返回已存在的摘要
+    if (messages.length === 0) {
+        return existingSummary || "";
+    }
+    // 格式化消息为压缩格式
+    const text = formatMessagesForCompression(messages);
+    // 构建压缩提示
+    const prompt = `请将以下对话内容压缩为一段简洁的摘要，保留关键信息、决策和结论：\n\n${text}`;
+    // 调用大模型压缩摘要
+    const model = (0, model_1.createModel)({ streaming: false });
+    // 处理大模型响应
+    // 如果响应内容是字符串，直接返回；如果是对象，转换为字符串
+    const response = await model.invoke([new messages_1.HumanMessage(prompt)]);
+    const newSummary = typeof response.content === "string"
+        ? response.content
+        : JSON.stringify(response.content);
+    // 如果已存在摘要，添加到新摘要中
+    if (existingSummary) {
+        return `${existingSummary}\n\n[后续对话摘要]\n${newSummary}`;
+    }
+    // 如果已不存在摘要，直接返回新摘要
+    return newSummary;
+}
+/**
+ * 计算安全的压缩边界索引，确保保留区域内的 tool_calls 和 ToolMessage 配对完整。
+ * 防止 破坏了消息列表的完整度（ ToolMessage 匹配不到 对应的 AIMessage ），导致报错。
+ *
+ * 策略：从 messages.length - minKeep 开始，如果边界切在了 ToolMessage 或其对应的
+ * AIMessage 之后，就把边界往前移动，直到整个工具调用单元都被保留。
+ */
+function findSafeCompressionIndex(messages, minKeep) {
+    // 初始化压缩边界索引为 messages.length - minKeep
+    let index = Math.max(0, messages.length - minKeep);
+    // 收集当前保留区域内所有 ToolMessage 的 tool_call_id
+    const toolCallIdsInKeepRegion = new Set();
+    // 从保留区域开始，向后遍历，收集所有 ToolMessage 的 tool_call_id
+    for (let i = index; i < messages.length; i++) {
+        if (messages[i].type === "tool") {
+            toolCallIdsInKeepRegion.add(messages[i].tool_call_id);
+        }
+    }
+    // 往前移动 index，直到所有 toolCallIdsInKeepRegion 都能在保留区域内找到对应的 AIMessage
+    while (index > 0) {
+        // 获取前一条消息
+        const prevMsg = messages[index - 1];
+        // 【1】如果前一条是 ToolMessage，它也会被纳入保留区，所以加入待匹配集合
+        if (prevMsg.type === "tool") {
+            toolCallIdsInKeepRegion.add(prevMsg.tool_call_id);
+            index--;
+            continue;
+        }
+        // 【2】如果前一条是 AIMessage 且有 tool_calls，检查它 是否匹配 保留区内的 ToolMessage
+        if (messages_1.AIMessage.isInstance(prevMsg) && prevMsg.tool_calls?.length) {
+            const hasMatchingCall = prevMsg.tool_calls.some((call) => call.id && toolCallIdsInKeepRegion.has(call.id));
+            // 如果前一条 AIMessage 匹配了保留区内的 ToolMessage，就保留它
+            if (hasMatchingCall) {
+                // 这个 AIMessage 必须保留，index 前移
+                index--;
+                // 这个 AIMessage 可能还有其他 tool_calls，它们对应的 ToolMessage 也必须在保留区
+                for (const call of prevMsg.tool_calls) {
+                    if (call.id)
+                        toolCallIdsInKeepRegion.add(call.id);
+                }
+                continue;
+            }
+        }
+        // 【3】如果前一条不是 ToolMessage，并且，前一条不是 AIMessage 或者其 tool_calls 为空/匹配不上，就直接跳出循环
+        break;
+    }
+    // 返回最终的压缩边界索引 《===  压缩边界的 结束索引 前移  《===  向前嗅探匹配的 ToolMessage 和 AIMessage 配对完整
+    return index;
+}

package/dist/agent/db.d.ts

@@ -0,0 +1,25 @@
+export declare const DB_PATH: string;
+export interface SessionRow {
+    thread_id: string;
+    last_question: string;
+    last_ts: string;
+}
+export interface MemorySearchResult {
+    id: number;
+    type: string;
+    content: string;
+    keywords: string | null;
+    importance: number;
+    session_id: string | null;
+    created_at: string;
+    updated_at: string;
+    final_score: number;
+}
+export declare function searchMemories(query: string[], limit?: number): MemorySearchResult[];
+export declare function threadIdExists(threadId: string): boolean;
+export declare function initDb(): void;
+/**
+ * 列出最近 10 个会话（线程）
+ * @returns 会话列表
+ */
+export declare function listRecentSessions(): SessionRow[];