@ia-ccun/code-agent-cli 0.0.18 → 0.0.19

package/config/agent/extensions/context-monitor.ts

@@ -0,0 +1,274 @@
+/**
+ * Context Monitor Extension — 自动检测上下文大小并在接近阈值时警告
+ *
+ * 功能：
+ * 1. 自动检测每轮对话后的上下文使用情况
+ * 2. 在超过阈值时自动发出不同级别的警告
+ * 3. 提供清理建议和上下文统计
+ *
+ * 阈值级别：
+ * - 70%   : info 级别提示
+ * - 80%   : warning 级别警告
+ * - 90%   : error 级别严重警告
+ * - 100%  : 阻止继续并强制提示压缩
+ */
+
+import type { ExtensionAPI, ExtensionContext, ToolResultEvent } from "@mariozechner/pi-coding-agent";
+
+interface ContextThresholds {
+	info: number;    // 70%
+	warning: number; // 80%
+	error: number;   // 90%
+	critical: number; // 100%
+}
+
+const DEFAULT_THRESHOLDS: ContextThresholds = {
+	info: 70,
+	warning: 80,
+	error: 90,
+	critical: 100,
+};
+
+interface ContextStats {
+	contextWindow: number;
+	usedTokens: number;
+	remainingTokens: number;
+	percent: number;
+	systemTokens: number;
+	toolsTokens: number;
+	conversationTokens: number;
+}
+
+function estimateTokens(text: string): number {
+	return Math.max(0, Math.ceil(text.length / 4));
+}
+
+function formatTokens(n: number): string {
+	if (n < 1000) return n.toString();
+	if (n < 1000000) return `${(n / 1000).toFixed(1)}k`;
+	return `${(n / 1000000).toFixed(1)}M`;
+}
+
+function getContextLevel(percent: number): "normal" | "info" | "warning" | "error" | "critical" {
+	if (percent >= DEFAULT_THRESHOLDS.critical) return "critical";
+	if (percent >= DEFAULT_THRESHOLDS.error) return "error";
+	if (percent >= DEFAULT_THRESHOLDS.warning) return "warning";
+	if (percent >= DEFAULT_THRESHOLDS.info) return "info";
+	return "normal";
+}
+
+function getContextMessage(stats: ContextStats): { level: "info" | "warning" | "error"; title: string; body: string } {
+	const level = getContextLevel(stats.percent);
+	const contextBar = renderContextBar(stats.percent);
+
+	let title: string;
+	let body: string;
+
+	switch (level) {
+		case "critical":
+			title = "🚨 上下文已满!";
+			body = `上下文窗口已完全使用! (${contextBar})\n剩余: ${formatTokens(stats.remainingTokens)} tokens\n\n⚠️ 请立即使用 /clear 清理会话或开启新会话!`;
+			break;
+		case "error":
+			title = "⚠️ 上下文即将耗尽";
+			body = `上下文使用: ${contextBar}\n已用: ${formatTokens(stats.usedTokens)} / ${formatTokens(stats.contextWindow)}\n剩余: ${formatTokens(stats.remainingTokens)} tokens\n\n💡 建议: 使用 /clear 清理会话`;
+			break;
+		case "warning":
+			title = "📊 上下文使用较高";
+			body = `上下文使用: ${contextBar}\n已用: ${formatTokens(stats.usedTokens)} / ${formatTokens(stats.contextWindow)}\n剩余: ${formatTokens(stats.remainingTokens)} tokens\n\n💡 提示: 可以考虑压缩会话`;
+			break;
+		case "info":
+			title = "ℹ️ 上下文使用提醒";
+			body = `上下文使用: ${contextBar}\n已用: ${formatTokens(stats.usedTokens)} / ${formatTokens(stats.contextWindow)}\n\n💡 提示: 上下文已使用过半，注意控制对话长度`;
+			break;
+		default:
+			title = "✅ 上下文状态正常";
+			body = `上下文使用: ${contextBar}\n已用: ${formatTokens(stats.usedTokens)} / ${formatTokens(stats.contextWindow)}\n剩余: ${formatTokens(stats.remainingTokens)} tokens`;
+	}
+
+	return { level, title, body };
+}
+
+function renderContextBar(percent: number, width: number = 20): string {
+	const filled = Math.min(width, Math.round((percent / 100) * width));
+	const empty = width - filled;
+
+	const filledChar = "█";
+	const emptyChar = "░";
+
+	return filledChar.repeat(filled) + emptyChar.repeat(empty) + ` ${percent.toFixed(1)}%`;
+}
+
+export default function contextMonitorExtension(pi: ExtensionAPI) {
+	// 配置：是否启用自动警告
+	let autoWarnEnabled = true;
+	let lastWarnedLevel: string = "normal";
+	let warnCooldown = 0; // 防止连续警告
+
+	/**
+	 * 获取当前上下文统计
+	 */
+	function getContextStats(ctx: ExtensionContext): ContextStats | null {
+		const usage = ctx.getContextUsage();
+		if (!usage) return null;
+
+		const contextWindow = usage.contextWindow || ctx.model?.contextWindow || 0;
+		if (contextWindow === 0) return null;
+
+		const usedTokens = usage.tokens || 0;
+		const remainingTokens = Math.max(0, contextWindow - usedTokens);
+		const percent = (usedTokens / contextWindow) * 100;
+
+		// 估算各部分 token
+		const systemPrompt = ctx.getSystemPrompt();
+		const systemTokens = systemPrompt ? estimateTokens(systemPrompt) : 0;
+		const toolsTokens = Math.round(
+			pi.getActiveTools().reduce((sum, name) => {
+				const info = pi.getAllTools().find((t) => t.name === name);
+				return sum + estimateTokens(`${name}\n${info?.description || ""}`);
+			}, 0) * 1.5
+		);
+		const conversationTokens = Math.max(0, usedTokens - systemTokens - toolsTokens);
+
+		return {
+			contextWindow,
+			usedTokens,
+			remainingTokens,
+			percent,
+			systemTokens,
+			toolsTokens,
+			conversationTokens,
+		};
+	}
+
+	/**
+	 * 处理上下文警告
+	 */
+	async function handleContextWarning(stats: ContextStats, ctx: ExtensionContext) {
+		if (!autoWarnEnabled) return;
+
+		const level = getContextLevel(stats.percent);
+		const message = getContextMessage(stats);
+
+		// 避免重复警告（同级别）
+		if (level === lastWarnedLevel && warnCooldown > 0) {
+			warnCooldown--;
+			return;
+		}
+
+		// 只在有新级别时警告
+		const levels = ["normal", "info", "warning", "error", "critical"];
+		const currentIdx = levels.indexOf(level);
+		const lastIdx = levels.indexOf(lastWarnedLevel);
+
+		if (currentIdx <= lastIdx && level !== "critical") return;
+
+		lastWarnedLevel = level;
+		warnCooldown = 3; // 设置冷却
+
+		// 发送通知
+		ctx.ui.notify(message.title, message.level as any);
+	}
+
+	// 监听轮次结束事件，检查上下文使用
+	pi.on("turn_end", async (_event, ctx: ExtensionContext) => {
+		const stats = getContextStats(ctx);
+		if (!stats) return;
+
+		await handleContextWarning(stats, ctx);
+	});
+
+	// 监听会话开始，初始化
+	pi.on("session_start", async (_event, ctx: ExtensionContext) => {
+		lastWarnedLevel = "normal";
+		warnCooldown = 0;
+	});
+
+	/**
+	 * /context-stats 命令 - 显示详细上下文统计
+	 */
+	pi.registerCommand("context-stats", {
+		description: "显示当前上下文详细统计",
+		usage: "/context-stats",
+		handler: async (_args, ctx: ExtensionContext) => {
+			const stats = getContextStats(ctx);
+
+			if (!stats) {
+				ctx.ui.notify("无法获取上下文统计", "warning");
+				return;
+			}
+
+			const message = getContextMessage(stats);
+			const lines = message.body.split("\n");
+
+			// 构建详细统计文本
+			const modelName = ctx.model?.name || "unknown";
+			const modelMaxTokens = ctx.model?.maxTokens || 0;
+			const contextWindow = stats.contextWindow;
+
+			// 根据使用情况给出建议
+			let suggestion = "";
+			if (stats.percent >= 90) {
+				suggestion = `⚠️ 上下文使用过高，建议清理会话\n💡 模型 ${modelName} 最大窗口 ${formatTokens(contextWindow)}`;
+			} else if (stats.percent >= 70) {
+				suggestion = `💡 上下文已过半，注意控制对话长度`;
+			}
+
+			const detailLines = [
+				`${"─".repeat(40)}`,
+				`📊 上下文详细统计`,
+				`${"─".repeat(40)}`,
+				``,
+				`模型:     ${modelName}`,
+				`总窗口:   ${formatTokens(contextWindow)} tokens`,
+				`最大输出: ${formatTokens(modelMaxTokens)} tokens`,
+				`已使用:   ${formatTokens(stats.usedTokens)} tokens (${stats.percent.toFixed(1)}%)`,
+				`剩余:     ${formatTokens(stats.remainingTokens)} tokens`,
+				``,
+				`📝 分布估算:`,
+				`  系统:   ${formatTokens(stats.systemTokens)} tokens`,
+				`  工具:   ${formatTokens(stats.toolsTokens)} tokens`,
+				`  对话:   ${formatTokens(stats.conversationTokens)} tokens`,
+				``,
+				suggestion,
+				``,
+				`${"─".repeat(40)}`,
+			];
+
+			// 显示给用户
+			const fullText = detailLines.join("\n");
+			pi.sendMessage(
+				{
+					customType: "context-stats",
+					content: fullText,
+					display: true,
+				},
+				{ triggerTurn: false }
+			);
+
+			// 如果超过阈值也显示通知
+			if (stats.percent >= 80) {
+				ctx.ui.notify(message.title, message.level as any);
+			}
+		},
+	});
+
+	/**
+	 * /context-warn 命令 - 切换自动警告
+	 */
+	pi.registerCommand("context-warn", {
+		description: "开启/关闭上下文警告",
+		usage: "/context-warn [on|off]",
+		handler: async (args, ctx: ExtensionContext) => {
+			if (!args || args === "on") {
+				autoWarnEnabled = true;
+				ctx.ui.notify("✅ 上下文警告已开启", "info");
+			} else if (args === "off") {
+				autoWarnEnabled = false;
+				ctx.ui.notify("🔕 上下文警告已关闭", "info");
+			} else {
+				ctx.ui.notify(`当前状态: ${autoWarnEnabled ? "开启" : "关闭"}`, "info");
+			}
+		},
+	});
+}

package/config/agent/models.json

@@ -4,18 +4,17 @@
       "name": "OpenAI",
       "baseUrl": "https://ai-llm.xxx.com/v1",
       "api": "openai-completions",
-      "apiKey": "YOUR_API_KEY_HERE",
+      "apiKey": "你自己的APIKEY",
       "headers": {
-        "User-Agent": "aicode-cli",
-        "sourceIp": "!hostname -I 2>/dev/null | awk '{print $1}' || ipconfig getifaddr en0 2>/dev/null || echo ''"
+        "User-Agent": "${VERSION}"
       },
       "models": [
         {
           "id": "GLM-4.7-W8A8",
           "name": "GLM4.7",
-          "contextWindow": 200000,
-          "maxTokens": 100000,
-          "reasoning": false,
+          "contextWindow": 99000,
+          "maxTokens": 99000,
+          "reasoning": true,
           "input": ["text"],
           "cost": {
             "input": 5,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ia-ccun/code-agent-cli",
-  "version": "0.0.18",
+  "version": "0.0.19",
   "description": "AI Coding Agent CLI - 基于OpenClaw🦞底层Agent原理实现的的编码智能体（供学习使用）。",
   "type": "module",
   "main": "./dist/index.js",
@@ -35,10 +35,10 @@
     "prepare": "husky"
   },
   "dependencies": {
-    "@mariozechner/pi-agent-core": "0.60.0",
-    "@mariozechner/pi-ai": "0.60.0",
-    "@mariozechner/pi-coding-agent": "0.60.0",
-    "@mariozechner/pi-tui": "0.60.0"
+    "@mariozechner/pi-agent-core": "0.62.0",
+    "@mariozechner/pi-ai": "0.62.0",
+    "@mariozechner/pi-coding-agent": "0.62.0",
+    "@mariozechner/pi-tui": "0.62.0"
   },
   "overrides": {
     "glob": "^11.0.0",