@nickyzj2023/ai 1.1.3 → 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-B0tl6AT5.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.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-B0tl6AT5.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-B0tl6AT5.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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nickyzj2023/ai",
-  "version": "1.1.3",
+  "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"
   },