alemonjs-aichat 1.0.12 → 1.0.13

package/lib/api.js

@@ -230,6 +230,7 @@ const availableTools = {
             return images;
         }
     },
+    /** 使用 Stable Diffusion 将图片转换为提示词 */
     StableDiffusionImageToPrompt: async ({ image_url }) => {
         // 先查看redis中是否有这张图的缓存
         const cacheKey = `image:caption:${image_url}`;
@@ -287,6 +288,45 @@ const availableTools = {
         console.log("图片提示词:", prompt);
         return prompt;
     },
+    /** 搜索survivalcraft(生存战争)论坛的帖子 */
+    SearchScbbsPost: async ({ keyword }) => {
+        const api = `https://m.suancaixianyu.cn/api/post/list?title=${encodeURIComponent(keyword)}`;
+        try {
+            const response = await fetch(api);
+            if (!response.ok) {
+                return [];
+            }
+            const data = await response.json();
+            return data.data.list.map((item) => ({
+                viewUrl: `https://test.suancaixianyu.cn/#/postDetails/${item.id}`,
+                ...item,
+            }));
+        }
+        catch (error) {
+            console.error("Error searching posts:", error);
+            return [];
+        }
+    },
+    /** 使用搜索引擎搜索网页 */
+    SearchWeb: async ({ keyword, engine = "baidu", cc = "CN", type = "organic_results", }) => {
+        const apikey = value.searchApiKey ||
+            "63706fc3e6827a90e8219913297b510ff053d5238b24f64251a066e4ecea7408";
+        const api = `https://serpapi.com/search.json?engine=${engine}&q=${encodeURIComponent(keyword)}&cc=${cc}&api_key=${apikey}`;
+        try {
+            const response = await fetch(api);
+            if (!response.ok) {
+                return [];
+            }
+            const data = await response.json();
+            console.log("搜索结果", data);
+            const results = data[type] || [];
+            return type === "organic_results" ? results.slice(0, 7) : results;
+        }
+        catch (error) {
+            console.error("Error searching web:", error);
+            return [];
+        }
+    },
 };
 
 export { TTSClient, availableTools, getTimeString, uploadImage };

package/lib/config.js

@@ -226,6 +226,57 @@ const tools = [
             },
         },
     },
+    {
+        type: "function",
+        function: {
+            name: "SearchScbbsPost",
+            description: "搜索《survivalcraft》(译名《生存战争》,也简称sc)论坛的帖子, 搜索时请尽量提供精确的关键词, 关键词最好只有一个. 在这里可以搜索到sc这个游戏相关的各种模组等资源或教程帖子和更新公告等内容, 但请注意不要搜索与游戏无关的内容, 否则可能会得到不相关的结果，若是搜索不到请减少关键词的数量或更换关键词",
+            parameters: {
+                type: "object",
+                properties: {
+                    keyword: {
+                        type: "string",
+                        description: "搜索关键词",
+                    },
+                },
+                required: ["keyword"],
+            },
+        },
+    },
+    {
+        type: "function",
+        function: {
+            name: "SearchWeb",
+            description: "使用搜索引擎搜索网页内容, 你可以使用这个工具来获取最新的资讯或数据, 但请注意不要搜索敏感或违法的内容, 否则可能会得到不相关的结果, 此外,不推荐使用该工具搜索与sc这个游戏相关的内容, 因为搜索引擎的结果可能不如专门的游戏论坛或社区来的准确和有用, 在搜索时请尽量提供精确的关键词, 关键词最好只有一个. 如果你想搜索与游戏相关的内容, 建议使用SearchScbbsPost工具来搜索survivalcraft(生存战争)论坛的帖子",
+            parameters: {
+                type: "object",
+                properties: {
+                    keyword: {
+                        type: "string",
+                        description: "搜索关键词",
+                    },
+                    engine: {
+                        type: "string",
+                        description: "选择搜索引擎",
+                        enum: ["bing", "google", "baidu"],
+                        default: "baidu",
+                    },
+                    cc: {
+                        type: "string",
+                        description: "搜索国家/地区代码, 例如搜索美国就填us, 搜索中国就填cn, 搜索日本就填jp, 这个参数会影响搜索结果的地域性",
+                        default: "CN",
+                    },
+                    type: {
+                        type: "string",
+                        description: "选择结果类型, 搜索后获取对应结果, 建议使用自然结果organic_results, 其他选项可能不存在",
+                        enum: ["organic_results", "related_questions", "related_searches"],
+                        default: "organic_results",
+                    },
+                },
+                required: ["keyword"],
+            },
+        },
+    },
 ];
 
 export { addAI, addAIChatHistory, addPrompt, affectionSwitch, clearAIChatHistory, defaultAIConfig, deleteAI, deletePrompt, getAIChatHistory, getAIConfig, getAIList, getAffectionLevel, getAffectionLevelAll, getPromptList, incrementAffectionLevel, resetAffectionLevel, resetAllAffectionLevels, setAffectionLevel, setBotID, switchAI, switchPrompt, systemPrompt, tools };

package/lib/data/help.json.js

@@ -1,6 +1,6 @@
 var title = "IA聊天插件";
 var name = "aichat";
-var version = "v1.0.10";
+var version = "v1.0.11";
 var by = "AlemonJS";
 var list = [
 	{

package/lib/response/config/res.js

@@ -3,7 +3,7 @@ import { useMessage, Text } from 'alemonjs';
 import { Regular } from 'alemonjs/utils';
 
 const selects = onSelects(["message.create", "private.message.create"]);
-const regular$1 = /(\/|#)(ai配置|当前提示词|查看提示词)?$/i;
+const regular$1 = /^(\/|#)(ai配置|当前提示词|查看提示词)$/i;
 const regular$2 = /(\/|#)ai列表$/i;
 const regular$3 = /(\/|#)提示词列表$/i;
 const regular$4 = /(\/|#)清空对话$/i;

package/lib/response/setting/res.js

@@ -1,6 +1,6 @@
 import { addAI, switchAI, addPrompt, deletePrompt, switchPrompt, deleteAI, affectionSwitch, getAIList, getAIConfig } from '../../config.js';
 import { redis } from '../../redis.js';
-import { useMessage, Text, ImageFile } from 'alemonjs';
+import { useMessage, Text } from 'alemonjs';
 import { Regular } from 'alemonjs/utils';
 
 const selects = onSelects(["message.create", "private.message.create"]);
@@ -18,8 +18,9 @@ const regular$13 = /(\/|#)(开启|关闭)聊天$/i;
 const regular$14 = /(\/|#)(开启|关闭)语音回复$/i;
 const regular$15 = /(\/|#)(开启|关闭)(MCP)?工具$/i;
 const regular$16 = /(\/|#)切换图床(.*)$/i;
+const regular$17 = /(\/|#)修改 ?(.+) ?模型 ?(.+)$/i;
 const test = /(\/|#)测试$/i;
-const regular = Regular.or(regular$1, regular$2, regular$3, regular$4, regular$5, regular$6, regular$7, regular$8, regular$10, regular$11, regular$13, regular$14, regular$15, regular$16, test);
+const regular = Regular.or(regular$1, regular$2, regular$3, regular$4, regular$5, regular$6, regular$7, regular$8, regular$10, regular$11, regular$13, regular$14, regular$15, regular$16, regular$17, test);
 var res = onResponse(selects, async (e, next) => {
     // 创建
     const [message] = useMessage(e);
@@ -28,7 +29,7 @@ var res = onResponse(selects, async (e, next) => {
         // uploadImageToR2("./public/image_out/1769259547338.jpg").then((url) => {
         //   console.log("上传成功，图片URL：", url);
         // });
-        message.send(format(Text("测试消息"), ImageFile("./public/image_out/1769257917858.jpg")));
+        message.send(format(Text("测试消息")));
         // const a = await availableTools.StableDiffusionGenerateImage({
         //   prompt: "a cat playing with a ball",
         // });
@@ -217,6 +218,30 @@ var res = onResponse(selects, async (e, next) => {
         message.send(format(Text(`已切换图床至 ${bedName.trim()} !`)));
         return;
     }
+    // 修改指定ai的模型
+    if (regular$17.test(e.msg)) {
+        const match = e.msg.match(regular$17);
+        if (!match) {
+            message.send(format(Text("格式错误，请按照 格式：/修改 <AI名称> 模型 <模型名称> 进行设置")));
+            return;
+        }
+        const [, , aiName, modelName] = match;
+        const aiList = await getAIList();
+        if (!aiList.includes(aiName.trim())) {
+            message.send(format(Text(`未找到 AI ${aiName.trim()} !`)));
+            return;
+        }
+        await switchAI(e.guid, aiName.trim());
+        const aiInfo = await getAIConfig(e.guid);
+        if (!aiInfo.model) {
+            message.send(format(Text(`切换失败，未找到 AI ${aiName.trim()} !`)));
+            return;
+        }
+        aiInfo.model = modelName.trim();
+        await addAI(e.guid, aiName.trim(), aiInfo.host, aiInfo.key, modelName.trim());
+        message.send(format(Text(`已将 AI ${aiName.trim()} 的模型修改为 ${modelName.trim()} !`)));
+        return;
+    }
     // 无匹配则继续下一个响应器
     next();
 });

package/lib/response/zreply/res.js

@@ -247,7 +247,7 @@ var res = onResponse(selects, async (e, next) => {
     }
     // 记录对话
     await addAIChatHistory(e.guid, res.choices?.[0].message);
-    const isOpenTTSReply = (await redis.get(`ai:tts:reply:switch:${e.guid}`)) || "0";
+    const isOpenTTSReply = (await redis.get(`chat:tts:response:${e.guid}`)) || "0";
     if (isOpenTTSReply === "1") {
         ttsClient = new TTSClient();
         await ttsClient.setModel("三月七", "生气");
@@ -290,6 +290,7 @@ var res = onResponse(selects, async (e, next) => {
     }
     catch (error) {
         console.log("解析AI回复失败，发送原始内容");
+        console.log(error);
         const msg = reply
             .replace(/\[CQ:[^\]]+\]/, "")
             .replace(/"}]/, "")
@@ -306,20 +307,25 @@ function sendTTSMessage(message, e, ttsClient) {
     ttsClient.getAudio(message).then(async (audio) => {
         console.log("audio", audio);
         if (audio) {
-            if (e.Platform == "onebot") {
-                const [client] = useClient(e, API);
-                client.sendMessage({
-                    message_type: e.value.message_type,
-                    group_id: e.value.group_id ?? undefined,
-                    message: [
-                        {
-                            type: "record",
-                            data: {
-                                file: audio.audio_url,
+            try {
+                if (e.Platform == "onebot") {
+                    const [client] = useClient(e, API);
+                    await client.sendMessage({
+                        message_type: e.value.message_type,
+                        group_id: e.value.group_id ?? undefined,
+                        message: [
+                            {
+                                type: "record",
+                                data: {
+                                    file: audio.audio_url,
+                                },
                             },
-                        },
-                    ],
-                });
+                        ],
+                    });
+                }
+            }
+            catch (error) {
+                console.log(error);
             }
         }
     });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "alemonjs-aichat",
-  "version": "1.0.12",
+  "version": "1.0.13",
   "description": "alemonjs-aichat",
   "author": "suancaixianyu",
   "license": "MIT",
@@ -20,7 +20,7 @@
     "@alemonjs/db": "^0.0.13",
     "@alemonjs/onebot": "^2.1.0-alpha.14",
     "@types/node": "^22",
-    "alemonjs": "2.1.17",
+    "alemonjs": "2.1.18",
     "cssnano": "^7",
     "jsxp": "^1.2.3",
     "lvyjs": "^0.2.25",
@@ -72,4 +72,4 @@
     "mime-types": "^3.0.2",
     "openai": "^6.16.0"
   }
-}
+}