yuangs 1.0.7 → 1.0.9

package/cli.js

@@ -14,8 +14,9 @@ function printHelp() {
     console.log(`  ${chalk.green('pong')}              打开 Pong 游戏`);
     console.log(`  ${chalk.green('list')}              列出所有应用链接`);
     console.log(`  ${chalk.green('ai')} "<问题>"      向 AI 提问`);
+    console.log(`    ${chalk.gray('--model <模型名称>')}  指定 AI 模型 (可选)`);
     console.log(`  ${chalk.green('help')}              显示帮助信息\n`);
-    console.log(chalk.gray('AI 示例: yuangs ai "你好，请问你是谁？"'));
+    console.log(chalk.gray('AI 示例: yuangs ai "你好" --model gemini-pro'));
     console.log(chalk.gray('普通示例: yuangs shici\n'));
 }
 
@@ -25,7 +26,20 @@ function printSuccess(app, url) {
 }
 
 async function handleAICommand() {
-    const question = args.slice(1).join(' ');
+    const commandArgs = args.slice(1);
+    const modelIndex = commandArgs.indexOf('--model');
+
+    let model = null; // Default model will be handled in index.js
+    let questionParts = commandArgs;
+
+    if (modelIndex !== -1 && commandArgs.length > modelIndex + 1) {
+        model = commandArgs[modelIndex + 1];
+        // Filter out --model and its value
+        questionParts = commandArgs.filter((_, index) => index !== modelIndex && index !== modelIndex + 1);
+    }
+
+    const question = questionParts.join(' ');
+
     if (!question) {
         console.log(chalk.red('错误: 请在 ai 命令后输入你的问题。'));
         printHelp();
@@ -39,7 +53,7 @@ async function handleAICommand() {
     }, 100);
 
     try {
-        const answer = await yuangs.getAIAnswer(question);
+        const answer = await yuangs.getAIAnswer(question, model);
         clearInterval(interval);
         process.stdout.clearLine(0);
         process.stdout.cursorTo(0);

package/index.js

@@ -17,7 +17,7 @@ function openUrl(url) {
     exec(command);
 }
 
-async function getAIAnswer(question) {
+async function getAIAnswer(question, model) {
     const url = 'https://aiproxy.want.biz/ai/explain';
     const prompt = question;
 
@@ -30,7 +30,7 @@ async function getAIAnswer(question) {
 
     const data = {
         text: prompt,
-        model: "gemini-flash-lite-latest"
+        model: model || "gemini-flash-lite-latest"
     };
 
     try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yuangs",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "description": "苑广山的个人应用集合 CLI（彩色版）",
   "main": "index.js",
   "bin": {