@nickyzj2023/ai 1.1.2 → 1.2.0

package/README.md

@@ -46,10 +46,15 @@ for await (const e of runAgent(model, messages, tools)) {
 ### 在终端里使用
 
 ```bash
-# 配置deepseek环境变量、启动对话
-SET APIKEY=sk-xxx && ai
+# 首次使用：交互式配置BASE_URL / APIKEY / MODEL / ...
+ai setup
+
+# 启动对话
+ai
 ```
 
+配置保存在 `~/.@nickyzj2023/ai/config.json`，任意目录下执行`ai`都能读取到
+
 ## License
 
 ISC

package/dist/cli.mjs

@@ -1,7 +1,83 @@
-import { i as defineModel, n as get_weather_default, o as runAgent, r as get_time_default } from "./src-DzLCCWBV.mjs";
-import { existsSync } from "node:fs";
-import { loadEnvFile } from "node:process";
+import { i as defineModel, n as get_weather_default, o as runAgent, r as get_time_default } from "./src-BJFCp7ur.mjs";
+import { extractErrorMessage } from "@nickyzj2023/utils";
 import readline from "node:readline";
+import { mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { dirname, join } from "node:path";
+//#region src/utils/config.ts
+/**
+* 获取全局配置文件路径（Windows/macOS/Linux通用）
+* @returns （Windows下）C:/Users/Administrator/.@nickyzj2023/ai/config.json
+*/
+const getConfigPath = () => {
+	return join(homedir(), ".@nickyzj2023", "ai", "config.json");
+};
+/**
+* 读取配置文件
+* @returns 正常返回Config，配置不存在或损坏时返回null
+*/
+const loadConfig = () => {
+	try {
+		const config = JSON.parse(readFileSync(getConfigPath(), "utf8"));
+		if (typeof config.baseUrl !== "string" || typeof config.apiKey !== "string" || typeof config.model !== "string") return null;
+		return config;
+	} catch {
+		return null;
+	}
+};
+/**
+* 将配置写回全局文件（自动创建目录）
+* @remarks 写入失败会抛异常
+*/
+const saveConfig = (config) => {
+	const configPath = getConfigPath();
+	mkdirSync(dirname(configPath), { recursive: true });
+	writeFileSync(configPath, `${JSON.stringify(config, null, 2)}\n`, "utf8");
+};
+//#endregion
+//#region src/interfaces/setup.ts
+/**
+* 临时创建readline接口，问答结束就关闭
+* @param question 提问
+* @returns 用户输入的回答
+*/
+const ask = (question) => {
+	return new Promise((resolve) => {
+		const rl = readline.createInterface({
+			input: process.stdin,
+			output: process.stdout
+		});
+		rl.question(question, (answer) => {
+			rl.close();
+			resolve(answer);
+		});
+	});
+};
+/**
+* setup入口：依次询问BASE_URL / MODEL / APIKEY，确认后写入全局配置
+*/
+async function runSetup() {
+	const config = loadConfig();
+	if (config) {
+		console.log(`当前配置：BASE_URL ${config.baseUrl}，APIKEY ${config.apiKey}，MODEL ${config.model}`);
+		console.log("直接回车可沿用当前值。\n");
+	}
+	const baseUrl = await ask(`BASE_URL [当前为${config?.baseUrl}]: `) || config?.baseUrl;
+	const apiKey = await ask(`APIKEY [当前为${config?.apiKey}]: `) || config?.apiKey;
+	const model = await ask(`MODEL [当前为${config?.model}]: `) || config?.model;
+	try {
+		saveConfig({
+			baseUrl,
+			apiKey,
+			model
+		});
+		console.log(`配置已保存到${getConfigPath()}，直接运行ai即可开始对话`);
+	} catch (e) {
+		console.error(`保存配置失败：${extractErrorMessage(e)}`);
+		process.exitCode = 1;
+	}
+}
+//#endregion
 //#region src/interfaces/tui.ts
 var TUI = class {
 	rl = null;
@@ -94,39 +170,67 @@ var TUI = class {
 };
 //#endregion
 //#region src/cli.ts
-if (existsSync(".env")) loadEnvFile(".env");
-const model = defineModel({
-	baseUrl: process.env.BASE_URL ?? "https://api.deepseek.com/v1",
-	apiKey: process.env.APIKEY,
-	model: process.env.MODEL ?? "deepseek-v4-flash"
-});
-if (!model.apiKey) {
-	console.error("请先在环境变量或.env文件中填入APIKEY（目前仅支持DeepSeek）");
-	process.exit(1);
-}
-const messages = [];
-const tools = [get_weather_default, get_time_default];
-const tui = new TUI(async (input) => {
-	messages.push({
-		role: "user",
-		content: input
-	});
-	for await (const e of runAgent(model, messages, tools)) switch (e.type) {
-		case "reasoning_delta":
-			tui.printReasoning(e.delta);
-			break;
-		case "content_delta":
-			tui.printContent(e.delta);
-			break;
-		case "tool_call":
-			tui.printToolCall(e.name, e.args);
-			break;
-		case "tool_result":
-			tui.printToolResult(e.name, e.result);
-			break;
-		case "done": tui.printFinish(e.finishReason, e.usage);
+/** 启动交互对话：配置来自全局配置文件，环境变量可临时覆盖 */
+const startChat = () => {
+	const config = loadConfig();
+	if (!config) {
+		console.error("请先运行 `ai setup` 配置一个模型");
+		process.exit(1);
 	}
-});
-tui.start();
+	const model = defineModel(config);
+	const messages = [];
+	const tools = [get_weather_default, get_time_default];
+	const tui = new TUI(async (input) => {
+		messages.push({
+			role: "user",
+			content: input
+		});
+		for await (const e of runAgent(model, messages, tools)) switch (e.type) {
+			case "reasoning_delta":
+				tui.printReasoning(e.delta);
+				break;
+			case "content_delta":
+				tui.printContent(e.delta);
+				break;
+			case "tool_call":
+				tui.printToolCall(e.name, e.args);
+				break;
+			case "tool_result":
+				tui.printToolResult(e.name, e.result);
+				break;
+			case "error":
+				tui.printContent(e.message);
+				break;
+			case "done": tui.printFinish(e.finishReason, e.usage);
+		}
+	});
+	tui.start();
+};
+/** 打印命令用法 */
+const printHelp = () => {
+	console.log(`用法: ai [命令]
+
+	命令:
+  --help    显示帮助
+  setup     交互式配置模型APIKEY / BASE_URL / MODEL（保存到 ~/.@nickyzj2023/ai/config.json）
+
+不带命令则启动对话`);
+};
+const command = process.argv[2];
+switch (command) {
+	case void 0:
+		startChat();
+		break;
+	case "setup":
+		await runSetup();
+		break;
+	case "--help":
+	case "-h":
+		printHelp();
+		break;
+	default:
+		console.error(`未知命令：${command}（可以运行ai --help查看用法）`);
+		process.exit(1);
+}
 //#endregion
 export {};

package/dist/index.d.mts

@@ -150,8 +150,8 @@ declare const _default$1: ToolDefinition;
 //#endregion
 //#region src/utils/compact/types.d.ts
 declare namespace Compact {
-  type ReplacerOfToolResultContent = (content: Message["content"]) => Message["content"];
-  type ReplacerOfMediaContent = (content: Message["content"]) => Message["content"];
+  type ReplacerOfToolResultContent = (content: Message["content"], options?: Record<string, any>) => Promise<Message["content"]> | Message["content"];
+  type ReplacerOfMediaContent = (content: Message["content"], options?: Record<string, any>) => Promise<Message["content"]> | Message["content"];
   type SummarizeOptions = {
     /** 用什么模型总结 */
     model: Model;
@@ -180,8 +180,16 @@ declare namespace Compact {
 }
 //#endregion
 //#region src/utils/compact/strategy.d.ts
-declare const softDeleteToolResults: (messages: Message[], replacer: Compact.ReplacerOfToolResultContent) => Message[];
-declare const softDeleteOldMediaMessages: (messages: Message[], replacer: Compact.ReplacerOfMediaContent) => Message[];
+declare const softDeleteToolResults: (messages: Message[], options: {
+  replacer?: Compact.ReplacerOfToolResultContent;
+  mark: string;
+  model?: Model;
+}) => Promise<Message[]>;
+declare const softDeleteOldMediaMessages: (messages: Message[], options: {
+  replacer?: Compact.ReplacerOfMediaContent;
+  mark: string;
+  model?: Model;
+}) => Promise<Message[]>;
 /**
  * 清理软删除后残留的占位消息（信息在软删除时已丢失，此时删除不损失任何额外信息）
  * @param messages 可压缩区消息数组，直接原地删除
@@ -229,6 +237,10 @@ declare const compact: ((messages: Message[], model: Model, options?: {
    * @default (content) => "（已被丢弃）"
    */
   replacerOfMediaContent?: Compact.ReplacerOfMediaContent;
+  /**
+   * 用于标记哪些消息被软删除了
+   */
+  softDeletedMessageMark?: string;
   /**
    * 上下文>总上下文*ratio时清理软删除残留的占位消息
    * @default 0.7

package/dist/index.mjs

@@ -1,2 +1,2 @@
-import { a as defineTool, i as defineModel, n as get_weather_default, o as runAgent, r as get_time_default, t as compact } from "./src-DzLCCWBV.mjs";
+import { a as defineTool, i as defineModel, n as get_weather_default, o as runAgent, r as get_time_default, t as compact } from "./src-BJFCp7ur.mjs";
 export { compact, defineModel, defineTool, get_time_default as getTime, get_weather_default as getWeather, runAgent };

package/dist/{src-DzLCCWBV.mjs → src-BJFCp7ur.mjs}

@@ -130,7 +130,7 @@ async function* runAgent(model, messages, tools) {
 			if (!tool) result = `不存在工具“${name}”`;
 			else {
 				const [error, response] = await to(tool.execute(JSON.parse(args)));
-				result = error ? `工具“${name}”执行出错：${error.message}` : String(response);
+				result = error ? `工具“${name}”执行出错：${error.message}` : JSON.stringify(response);
 			}
 			messages.push({
 				role: "tool",
@@ -212,27 +212,12 @@ const estimateTokens = (messages) => {
 };
 //#endregion
 //#region src/tools/get-time.ts
-const weekdayNames = {
-	Sunday: "星期日",
-	Monday: "星期一",
-	Tuesday: "星期二",
-	Wednesday: "星期三",
-	Thursday: "星期四",
-	Friday: "星期五",
-	Saturday: "星期六"
-};
 var get_time_default = defineTool("get_time", "查询指定时区的当前时间", { timezone: {
 	type: "string",
 	description: "完整的IANA时区名称，如Europe/Amsterdam。用户未明确提及时，传入对方语言对应的时区，例如对方使用中文时可传入Asia/Shanghai。",
 	required: true
 } }, async ({ timezone }) => {
-	const data = await fetcher("https://timeapi.io/api", { params: { timezone } }).get("/time/current/zone");
-	const pad = (n) => String(n).padStart(2, "0");
-	return [
-		`时区：${data.timeZone}${data.dstActive ? "（夏令时已生效）" : ""}`,
-		`日期：${data.year}-${pad(data.month)}-${pad(data.day)}（${weekdayNames[data.dayOfWeek] ?? data.dayOfWeek}）`,
-		`时间：${pad(data.hour)}:${pad(data.minute)}:${pad(data.seconds)}`
-	].join("\n");
+	return fetcher("https://timeapi.io/api", { params: { timezone } }).get("/time/current/zone");
 });
 //#endregion
 //#region src/tools/get-weather.ts
@@ -241,16 +226,7 @@ var get_weather_default = defineTool("get_weather", "查询指定城市的天气
 	description: "城市名，如shanghai、tokyo",
 	required: true
 } }, async ({ city }) => {
-	const data = await fetcher("https://wttr.in", { params: { format: "j1" } }).get(`/${city}`);
-	const nearestArea = data.nearest_area[0];
-	const currentCondition = data.current_condition[0];
-	return [
-		`位置：${nearestArea.country[0].value}-${nearestArea.region[0].value}-${nearestArea.areaName[0].value}`,
-		`当前温度: ${currentCondition.temp_C}°C`,
-		`体感温度: ${currentCondition.FeelsLikeC}°C`,
-		`天气状况: ${currentCondition.weatherDesc[0].value}`,
-		`湿度: ${currentCondition.humidity}%`
-	].join("\n");
+	return fetcher("https://wttr.in", { params: { format: "j1" } }).get(`/${city}`);
 });
 //#endregion
 //#region src/utils/compact/helper.ts
@@ -278,16 +254,52 @@ const findToolGroupRange = (messages, toolIndex) => {
 };
 //#endregion
 //#region src/utils/compact/strategy.ts
-const softDeleteToolResults = (messages, replacer) => {
+/** 默认的软删除多模态消息策略：让大模型精简消息内容 */
+const defaultReplacerOfToolResultContent = async (content, options) => {
+	const { mark, model } = options ?? {};
+	const messages = [{
+		role: "user",
+		content
+	}, {
+		role: "user",
+		content: "请用一两句话对上条消息做个“省流”"
+	}];
+	let simplifiedContent = mark;
+	for await (const e of runAgent(model, messages, [])) if (e.type === "content_delta") simplifiedContent += e.delta;
+	return simplifiedContent;
+};
+const softDeleteToolResults = async (messages, options) => {
+	const { replacer = defaultReplacerOfToolResultContent, mark, model } = options ?? {};
+	if (!replacer && !model) return [];
 	const softDeleted = [];
-	for (const message of messages) if (message?.role === "tool") {
-		message.content = replacer(message.content);
+	for (const message of messages) if (message?.role === "tool" && typeof message.content === "string") {
+		if (message.content.startsWith(mark)) continue;
+		message.content = await replacer(message.content, {
+			mark,
+			model
+		});
 		softDeleted.push(message);
 	}
 	if (softDeleted.length > 0) logger(`软删除了${softDeleted.length}条工具调用结果消息`);
 	return softDeleted;
 };
-const softDeleteOldMediaMessages = (messages, replacer) => {
+/** 默认的软删除多模态消息策略：让大模型精简消息内容 */
+const defaultReplacerOfMediaContent = async (content, options) => {
+	const { mark, model } = options ?? {};
+	const messages = [{
+		role: "user",
+		content
+	}, {
+		role: "user",
+		content: "请用一两句话描述上方的多模态消息"
+	}];
+	let simplifiedContent = mark;
+	for await (const e of runAgent(model, messages, [])) if (e.type === "content_delta") simplifiedContent += e.delta;
+	return simplifiedContent;
+};
+const softDeleteOldMediaMessages = async (messages, options) => {
+	const { replacer = defaultReplacerOfMediaContent, mark, model } = options ?? {};
+	if (!replacer && !model) return [];
 	const mediaTypes = [
 		"image_url",
 		"input_audio",
@@ -295,7 +307,10 @@ const softDeleteOldMediaMessages = (messages, replacer) => {
 	];
 	const softDeleted = [];
 	for (const message of messages) if (message && Array.isArray(message.content) && message.content.some((part) => mediaTypes.includes(part.type))) {
-		message.content = replacer(message.content);
+		message.content = await replacer(message.content, {
+			mark,
+			model
+		});
 		softDeleted.push(message);
 	}
 	if (softDeleted.length > 0) logger(`软删除了${softDeleted.length}条旧图片/音频/视频消息`);
@@ -393,7 +408,7 @@ const hardDeleteOldMessages = (messages) => {
 * 自动优化上下文，类似AI Coding Agent的/compact命令
 */
 const compact = Object.assign(async (messages, model, options) => {
-	const { usage, keepCount = 10, ratioToCompactToolResult = .5, replacerOfToolResultContent = () => "（工具结果已消费）", ratioToCompactMedia = .6, replacerOfMediaContent = () => "（消息已过期）", ratioToClearSoftDeletedMessages = .7, ratioToSummarize = .8, summarizeOptions } = options ?? {};
+	const { usage, keepCount = 10, ratioToCompactToolResult = .5, replacerOfToolResultContent, ratioToCompactMedia = .6, replacerOfMediaContent, ratioToClearSoftDeletedMessages = .7, softDeletedMessageMark = "[已简化]", ratioToSummarize = .8, summarizeOptions } = options ?? {};
 	const context = model?.context ?? 131072;
 	const tokens = usage?.total_tokens ?? estimateTokens(messages);
 	const result = {
@@ -413,11 +428,21 @@ const compact = Object.assign(async (messages, model, options) => {
 	const reserved = messages.slice(endIndex);
 	const softDeleted = /* @__PURE__ */ new Set();
 	if (tokens > context * ratioToCompactToolResult) {
-		for (const message of softDeleteToolResults(compressible, replacerOfToolResultContent)) softDeleted.add(message);
+		const softDeletedMessages = await softDeleteToolResults(compressible, {
+			replacer: replacerOfToolResultContent,
+			mark: softDeletedMessageMark,
+			model
+		});
+		for (const message of softDeletedMessages) softDeleted.add(message);
 		result.hasCompactedToolResult = true;
 	}
 	if (tokens > context * ratioToCompactMedia) {
-		for (const message of softDeleteOldMediaMessages(compressible, replacerOfMediaContent)) softDeleted.add(message);
+		const softDeletedMessages = await softDeleteOldMediaMessages(compressible, {
+			replacer: replacerOfMediaContent,
+			mark: softDeletedMessageMark,
+			model
+		});
+		for (const message of softDeletedMessages) softDeleted.add(message);
 		result.hasCompactedMedia = true;
 	}
 	if (tokens > context * ratioToClearSoftDeletedMessages) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nickyzj2023/ai",
-  "version": "1.1.2",
+  "version": "1.2.0",
   "description": "我的“pi”，参考了pi-from-scratch",
   "type": "module",
   "main": "./dist/index.mjs",
@@ -26,8 +26,8 @@
     "access": "public"
   },
   "devDependencies": {
-    "@biomejs/biome": "^2.5.11",
-    "@types/node": "^26.4.0",
+    "@biomejs/biome": "^2.5.12",
+    "@types/node": "^26.4.1",
     "tsdown": "^0.22.14",
     "typescript": "^7.0.2"
   },