koishi-plugin-cocoyyy-console 1.0.16-beta.4 → 1.0.16-beta.6

package/lib/index.d.ts

@@ -8,6 +8,7 @@ export interface Config {
     mysql_config?: my_config.MysqlConfig;
     tag_config?: my_config.SaveConfig;
     repeat_config?: my_config.RepeatConfig;
+    son_config?: my_config.ShitOrNotConfig;
     test?: string;
 }
 export declare const Config: Schema<Config>;

package/lib/index.js

@@ -30,7 +30,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
-  Config: () => Config,
+  Config: () => Config2,
   apply: () => apply,
   dev_mode: () => dev_mode,
   logger: () => logger,
@@ -38,7 +38,7 @@ __export(src_exports, {
   savePath: () => savePath
 });
 module.exports = __toCommonJS(src_exports);
-var import_koishi9 = require("koishi");
+var import_koishi10 = require("koishi");
 
 // src/utils/configs/function_config.ts
 var import_koishi = require("koishi");
@@ -47,7 +47,7 @@ var FunctionConfigSchema = import_koishi.Schema.object({
   tag_flag: import_koishi.Schema.boolean().default(true).description("标签功能是否启用"),
   repeat_flag: import_koishi.Schema.boolean().default(true).description("复读功能是否启用"),
   rbq_flag: import_koishi.Schema.boolean().default(true).description("*人功能是否启用"),
-  shit_or_not_flag: import_koishi.Schema.boolean().default(true).description("鉴别史消息功能是否启用"),
+  son_flag: import_koishi.Schema.boolean().default(true).description("鉴别史消息功能是否启用"),
   kuro_flag: import_koishi.Schema.boolean().default(true).description("鸣潮功能是否启用")
 });
 
@@ -87,6 +87,15 @@ var RedisConfigSchema = import_koishi5.Schema.object({
   redis_url: import_koishi5.Schema.string().default("127.0.0.1:6379").description("Redis连接字符串")
 });
 
+// src/utils/configs/son_config.ts
+var import_koishi6 = require("koishi");
+var ShitOrNotConfigSchema = import_koishi6.Schema.object({
+  api_url: import_koishi6.Schema.string().default("").description("Api接口Url"),
+  api_model: import_koishi6.Schema.string().default("").description("AI模型"),
+  api_key: import_koishi6.Schema.string().default("").description("API Key"),
+  config_path: import_koishi6.Schema.string().default("").description("配置文件路径")
+});
+
 // src/infra/mysql_init.ts
 var import_sequelize = require("sequelize");
 var sequelize = null;
@@ -314,6 +323,10 @@ function getMenuList(command = null, functionConfig) {
       if (!functionConfig.rbq_flag)
         return "*人功能已关闭";
       return getRbqFuncMenu(command);
+    case "son":
+      if (!functionConfig.son_flag)
+        return "son功能已关闭";
+      return getSonFuncMenu(command);
     case "kuro":
       if (!functionConfig.kuro_flag)
         return "库洛相关功能已关闭";
@@ -404,6 +417,25 @@ function getRbqFuncMenu(command = null) {
 输入对应指令使用*人功能，如果遇到问题请联系开发人员。`;
 }
 __name(getRbqFuncMenu, "getRbqFuncMenu");
+var sonFuncMenuList = [
+  {
+    name: "sonlist",
+    description: "查看son判断模式列表",
+    command: "sonlist"
+  },
+  {
+    name: "son",
+    description: "判断是否为史",
+    command: "son [判断模式]"
+  }
+];
+function getSonFuncMenu(command = null) {
+  return `[所有命令都需要@bot]
+son功能菜单:
+	${sonFuncMenuList.map((item) => item.command + ": " + item.description).join("\n	")}
+输入对应指令使用son功能，如果遇到问题请联系开发人员。`;
+}
+__name(getSonFuncMenu, "getSonFuncMenu");
 var kuroFuncMenuList = [
   {
     name: "login",
@@ -436,7 +468,7 @@ function registerMenuCommands(ctx, connected, functionConfig) {
 __name(registerMenuCommands, "registerMenuCommands");
 
 // src/services/tag_func/tag_commands.ts
-var import_koishi7 = require("koishi");
+var import_koishi8 = require("koishi");
 
 // src/services/tag_func/tag_service.ts
 var import_sequelize4 = require("sequelize");
@@ -444,7 +476,7 @@ var import_fs2 = require("fs");
 var import_path2 = require("path");
 
 // src/services/tag_func/img_service.ts
-var import_koishi6 = require("koishi");
+var import_koishi7 = require("koishi");
 var import_path = require("path");
 var import_fs = require("fs");
 async function loadImageFromUrl(ctx, url) {
@@ -519,7 +551,7 @@ async function randomImgByTag(tag_name, guild_id) {
     if (ImgList.length <= 0) {
       throw new Error("Don't have image!");
     }
-    const random = new import_koishi6.Random(() => Math.random());
+    const random = new import_koishi7.Random(() => Math.random());
     let result = ImgList[random.int(0, ImgList.length)];
     const filePath = result.get("img_url");
     return filePath;
@@ -763,8 +795,8 @@ function registerTagCommands(ctx, connected, savePath2) {
     let exec = await randomImgByTag(tag, session.guildId);
     if (!exec)
       return `获取图片失败`;
-    let path = (0, import_url.pathToFileURL)(exec).href;
-    return import_koishi7.h.image(path);
+    let path2 = (0, import_url.pathToFileURL)(exec).href;
+    return import_koishi8.h.image(path2);
   });
   ctx.command("bind <参数>", "标签绑定别名").action(async ({ session }, ...args) => {
     if (!dev_mode) {
@@ -907,7 +939,7 @@ function registerRepeatMiddleware(ctx, config) {
 __name(registerRepeatMiddleware, "registerRepeatMiddleware");
 
 // src/services/rbq_func/rbq_command.ts
-var import_koishi8 = require("koishi");
+var import_koishi9 = require("koishi");
 
 // src/models/rbq_persons.ts
 var import_sequelize5 = require("sequelize");
@@ -1232,7 +1264,7 @@ function registerRbqCommands(ctx, connected) {
     if (!list || list.length === 0) {
       return "当前群聊还没有设置任何rbq";
     }
-    return "当前群聊rbq列表:\n" + list.map((item) => (0, import_koishi8.h)("at", { id: item })).join(",");
+    return "当前群聊rbq列表:\n" + list.map((item) => (0, import_koishi9.h)("at", { id: item })).join(",");
   });
   ctx.command("rbqadd <参数>", "添加rbq").action(async ({ session }, ...args) => {
     if (!dev_mode) {
@@ -1250,7 +1282,7 @@ function registerRbqCommands(ctx, connected) {
     exec = await create_person(session.guildId, uid);
     if (!exec.result)
       return `添加rbq失败: ${exec.error}`;
-    return "已添加rbq:" + (0, import_koishi8.h)("at", { id: uid });
+    return "已添加rbq:" + (0, import_koishi9.h)("at", { id: uid });
   });
   ctx.command("rbqinstead <参数>", "替换rbq").action(async ({ session }, ...args) => {
     if (!dev_mode) {
@@ -1275,12 +1307,12 @@ function registerRbqCommands(ctx, connected) {
       exec = await instead_person(session.guildId, source_uid, new_target);
       if (!exec.result)
         return `替换rbq失败：${exec.error}`;
-      return "倒反天罡！进去吧你，已替换rbq：\n" + (0, import_koishi8.h)("at", { id: source_uid }) + " -> " + (0, import_koishi8.h)("at", { id: new_target });
+      return "倒反天罡！进去吧你，已替换rbq：\n" + (0, import_koishi9.h)("at", { id: source_uid }) + " -> " + (0, import_koishi9.h)("at", { id: new_target });
     }
     exec = await instead_person(session.guildId, source_uid, target_uid);
     if (!exec.result)
       return `替换rbq失败：${exec.error}`;
-    return (0, import_koishi8.h)("p", "已替换rbq：", (0, import_koishi8.h)("at", { id: source_uid }), " -> ", (0, import_koishi8.h)("at", { id: target_uid }));
+    return (0, import_koishi9.h)("p", "已替换rbq：", (0, import_koishi9.h)("at", { id: source_uid }), " -> ", (0, import_koishi9.h)("at", { id: target_uid }));
   });
   ctx.command("rbqadd_txt <参数>", "添加自定义文本").action(async ({ session }, ...args) => {
     if (!dev_mode) {
@@ -1307,10 +1339,10 @@ function registerRbqCommands(ctx, connected) {
         const contents = await get_content_list(session.guildId, target_uid);
         let content2 = "*死你";
         if (contents && contents.length > 0) {
-          const random = new import_koishi8.Random(() => Math.random());
+          const random = new import_koishi9.Random(() => Math.random());
           content2 = contents[random.int(0, contents.length)];
         }
-        return (0, import_koishi8.h)("at", { id: target_uid }) + "还想自己加文本？！\n" + content2;
+        return (0, import_koishi9.h)("at", { id: target_uid }) + "还想自己加文本？！\n" + content2;
       } catch (error) {
         logger.error("[rbq rbqadd_txt] Error:", error);
         return "不能给自己设置文本，请输入其他qq号";
@@ -1340,10 +1372,10 @@ function registerRbqCommands(ctx, connected) {
       const contents = await get_content_list(guildId, matchedUid);
       let content = "*死你";
       if (contents && contents.length > 0) {
-        const random = new import_koishi8.Random(() => Math.random());
+        const random = new import_koishi9.Random(() => Math.random());
         content = contents[random.int(0, contents.length)];
       }
-      return (0, import_koishi8.h)("at", { id: matchedUid }) + " " + content;
+      return (0, import_koishi9.h)("at", { id: matchedUid }) + " " + content;
     } catch (error) {
       logger.error("[rbq middleware] Error:", error);
       return next();
@@ -1354,23 +1386,14 @@ __name(registerRbqCommands, "registerRbqCommands");
 
 // src/services/test_command.ts
 var import_tesseract = require("tesseract.js");
-var CONSTANTS = {
-  ESU: "前提条件"
-};
-var chatHistory = [];
 function registerTestCommands(ctx, config) {
   ctx.command("test1 <prompt:text>", "调用 OpenRouter 模型进行测试").action(async ({ session }, prompt) => {
     const apiKey = config?.test || process.env.OPENROUTER_API_KEY;
     if (!apiKey) return "未配置 OpenRouter API Key（config.test 或环境变量 OPENROUTER_API_KEY）";
     const content = prompt?.trim();
     if (!content) return "请输入要测试的内容";
-    chatHistory.push({ role: "user", content });
-    const messages = [
-      { role: "system", content: CONSTANTS.ESU },
-      ...chatHistory
-    ];
   });
-  ctx.command("test <path>", "测试用").action(async ({ session }, path) => {
+  ctx.command("test <path>", "测试用").action(async ({ session }, path2) => {
     const { content, uid, userId } = session;
     if (session.quote) {
       const quoteContent = session.quote.content;
@@ -1396,8 +1419,197 @@ function registerTestCommands(ctx, config) {
 }
 __name(registerTestCommands, "registerTestCommands");
 
-// src/services/kuro_func/kuro_service.ts
+// src/services/son_func/shit_or_not_command.ts
+var import_fs3 = require("fs");
+var yaml = __toESM(require("js-yaml"));
+var import_path3 = __toESM(require("path"));
+
+// src/services/son_func/shit_or_not_service.ts
+var import_tesseract2 = require("tesseract.js");
+
+// src/utils/ai_axios.ts
 var import_axios = __toESM(require("axios"));
+async function callOpenRouter(api_url, api_key, api_model = "gpt-4o-mini", messages) {
+  try {
+    const { data } = await import_axios.default.post(
+      api_url,
+      {
+        api_model,
+        messages
+      },
+      {
+        headers: {
+          Authorization: `Bearer ${api_key}`,
+          "Content-Type": "application/json"
+        }
+      }
+    );
+    const choices = data?.choices ?? [];
+    const reply = choices[0]?.message?.content;
+    let responseText;
+    let err = null;
+    if (typeof reply === "string") {
+      responseText = reply.trim();
+    } else if (Array.isArray(reply)) {
+      responseText = reply.map((part) => part?.text ?? "").join("").trim();
+      if (!responseText) {
+        err = "OpenRouter 返回空内容";
+      }
+    } else {
+      err = "OpenRouter 返回了未知格式，请检查日志";
+    }
+    return { err, resp: responseText };
+  } catch (error) {
+    const err = error;
+    const status = err.response?.status;
+    const detail = err.response?.data?.error || err.response?.data?.message || err.message;
+    logger.error("[callOpenRouter Error]: OpenRouter 请求失败" + error);
+    const errorMessage = `test error${status ? `(${status})` : ""}: ${detail}`;
+    return { err: errorMessage, resp: null };
+  }
+}
+__name(callOpenRouter, "callOpenRouter");
+
+// src/services/son_func/shit_or_not_service.ts
+var chatHistory = [];
+function decodeHtmlEntities2(value) {
+  return value.replace(/&amp;/g, "&").replace(/&lt;/g, "<").replace(/&gt;/g, ">").replace(/&quot;/g, '"').replace(/&#39;/g, "'");
+}
+__name(decodeHtmlEntities2, "decodeHtmlEntities");
+function extractImageAndText(content) {
+  const imageUrls = [];
+  const imgTagRegex = /<img\b[^>]*>/gi;
+  const stripped = content.replace(imgTagRegex, (tag) => {
+    const srcMatch = tag.match(/src=["']([^"']+)["']/i);
+    if (srcMatch?.[1]) {
+      imageUrls.push(decodeHtmlEntities2(srcMatch[1]));
+    }
+    return " ";
+  });
+  const text = stripped.replace(/\s+/g, " ").trim();
+  return { imageUrls, text };
+}
+__name(extractImageAndText, "extractImageAndText");
+async function checkShitOrNot(session, parttern_msg, config) {
+  chatHistory = [];
+  chatHistory.push({ role: "user", content: [{ type: "text", text: parttern_msg }] });
+  const { err, resp } = await callOpenRouter(config.api_url, config.api_key, config.api_model, chatHistory);
+  if (err != null) {
+    logger.error("[callOpenRouter Error]: " + err);
+    return null;
+  }
+  chatHistory.push({ role: "assistant", content: [{ type: "text", text: resp }] });
+  const quoteContent = session.quote.content;
+  const { imageUrls, text } = extractImageAndText(quoteContent);
+  if (!text && imageUrls.length === 0) {
+    logger.warn("[checkShitOrNot Warn]: 引用消息中未发现文本或图片");
+    return "引用消息中未发现文本或图片";
+  }
+  const targetContent = [
+    ...text ? [{ type: "text", text }] : [],
+    ...imageUrls.map((url) => ({ type: "image_url", image_url: { url } }))
+  ];
+  chatHistory.push({ role: "user", content: targetContent });
+  const secondCall = await callOpenRouter(config.api_url, config.api_key, config.api_model, chatHistory);
+  if (secondCall.err != null) {
+    logger.error("[callOpenRouter Error]: " + secondCall.err);
+    return null;
+  }
+  return secondCall.resp;
+}
+__name(checkShitOrNot, "checkShitOrNot");
+
+// src/services/son_func/shit_or_not_command.ts
+var local_config = null;
+var mode_intro = {
+  "esu": "【图文】ohno妈咪饿素啊😨，恶俗慎用！🈲",
+  "nov": "【图文】来段微小说📕",
+  "lit": "【图文】文楠魅力时刻📕👨",
+  "zhi": "【图文】分享你刚编的故事🤓",
+  "boa": "【图文】《答案之书》🔮",
+  "scp": "【图文】控制-收容-保护，献上你的异常👽",
+  "phi": "【图文】我思，故我在🧠",
+  "see": "【图像】让我康康怎么个事👀",
+  "ccb": "【图像】压抑笑传之踩踩背🐘"
+};
+function loadYamlConfig(configPath) {
+  if (!configPath) {
+    logger.error("[loadYamlConfig Error]: SON 配置文件路径未配置");
+    return null;
+  }
+  try {
+    const trimmedPath = configPath.trim();
+    const fullPath = import_path3.default.isAbsolute(trimmedPath) ? trimmedPath : import_path3.default.resolve(process.cwd(), trimmedPath);
+    if (!(0, import_fs3.existsSync)(fullPath)) {
+      logger.error(`[loadYamlConfig Error]: SON 配置文件不存在: ${fullPath}`);
+      return null;
+    }
+    const fileContent = (0, import_fs3.readFileSync)(fullPath, "utf-8");
+    const config = yaml.load(fileContent);
+    return config;
+  } catch (error) {
+    logger.error(`[loadYamlConfig Error]: SON 读取配置文件失败: ${configPath}`, error);
+    return null;
+  }
+}
+__name(loadYamlConfig, "loadYamlConfig");
+function registerShitOrNotCommands(ctx, config) {
+  if (config?.config_path) {
+    local_config = loadYamlConfig(config.config_path);
+    if (local_config) {
+      logger.info("[loadYamlConfig Info]: SON 成功加载配置文件");
+    } else {
+      logger.error("[loadYamlConfig Error]: SON 配置文件加载失败，将使用默认配置");
+    }
+  } else {
+    logger.error("[loadYamlConfig Error]: 未配置 SON 配置文件路径");
+  }
+  ctx.command("sonlist <参数>", "查看son判断模式列表").action(() => {
+    return `[所有命令都需要@bot]
+son判断模式列表:
+	${Object.keys(mode_intro).map((item) => item + ": " + mode_intro[item]).join("\n	")}`;
+  });
+  ctx.command("son <参数>", "判定是否为史").action(async ({ session }, ...args) => {
+    if (!dev_mode) {
+      if (!is_at_bot_quote(session)) return "请提供正确格式，如：1 [引用消息] @bot son [判断模式]";
+    } else {
+      if (!session.quote) return "请提供正确格式，如：2 [引用消息] @bot son [判断模式]";
+    }
+    const parttern = args?.[0];
+    if (!parttern) return "请提供正确格式，如：3 [引用消息] @bot son [判断模式]";
+    if (!local_config) return "未加载配置文件，请查看日志";
+    const parttern_msg = local_config.shit_or_not?.prompts[parttern];
+    if (!parttern_msg) return "未找到判断模式，请查看日志";
+    let exec = await checkShitOrNot(session, parttern_msg, config);
+    return exec;
+  });
+}
+__name(registerShitOrNotCommands, "registerShitOrNotCommands");
+function is_at_bot_quote(session) {
+  if (!session.quote)
+    return false;
+  const { content, userId } = session;
+  const quote = session.quote;
+  const quoteUserId = quote.user.id;
+  const quoteContent = quote.content;
+  const botId = session?.bot?.selfId ?? session?.selfId;
+  if (dev_mode)
+    logger.info(`command: quoteContent: ${quoteContent}, quoteUserId: ${quoteUserId},content: ${content} userId: ${userId}`);
+  const elements = quote.elements || [];
+  for (const elem of elements) {
+    if (elem.type === "at") {
+      const atId = elem.attrs.id;
+      if (String(atId) === String(botId)) {
+        return true;
+      }
+    }
+  }
+  return false;
+}
+__name(is_at_bot_quote, "is_at_bot_quote");
+
+// src/services/kuro_func/kuro_service.ts
+var import_axios2 = __toESM(require("axios"));
 var import_qs = __toESM(require("qs"));
 
 // src/models/kuro_user.ts
@@ -1506,7 +1718,7 @@ __name(createUpdateUser, "createUpdateUser");
 
 // src/services/kuro_func/kuro_service.ts
 var userTokenMap = /* @__PURE__ */ new Map();
-var CONSTANTS2 = {
+var CONSTANTS = {
   BASE_URL: "https://api.kurobbs.com",
   LOGIN_URL: "/user/sdkLogin",
   GAME_DATA_URL: "/gamer/role/list",
@@ -1530,11 +1742,11 @@ var CONSTANTS2 = {
     "source": "ios"
   }
 };
-var wavesApi = import_axios.default.create();
+var wavesApi = import_axios2.default.create();
 wavesApi.interceptors.request.use(
   async (config) => {
     if (config.url.startsWith("/")) {
-      config.url = CONSTANTS2.BASE_URL + config.url;
+      config.url = CONSTANTS.BASE_URL + config.url;
     }
     return config;
   },
@@ -1588,7 +1800,7 @@ async function getToken(session, mobile, code) {
     code
   });
   try {
-    const user_response = await wavesApi.post(CONSTANTS2.LOGIN_URL, data, { headers: { ...CONSTANTS2.REQUEST_HEADERS_BASE, devCode } });
+    const user_response = await wavesApi.post(CONSTANTS.LOGIN_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, devCode } });
     if (user_response.data.code != 200) {
       logger.error(`验证码登录失败: ${user_response.data.msg}`);
       return { status: false, msg: user_response.data.msg };
@@ -1628,7 +1840,7 @@ async function getGameData(token) {
     let data = import_qs.default.stringify({
       "gameId": 3
     });
-    const response = await wavesApi.post(CONSTANTS2.GAME_DATA_URL, data, { headers: { ...CONSTANTS2.REQUEST_HEADERS_BASE, "token": token } });
+    const response = await wavesApi.post(CONSTANTS.GAME_DATA_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, "token": token } });
     if (response.data.code === 200) {
       return { status: true, data: response.data.data[0] };
     } else {
@@ -1647,7 +1859,7 @@ async function isAvailable(serverId, roleId, token) {
     "roleId": roleId
   });
   try {
-    const response = await wavesApi.post(CONSTANTS2.TOKEN_REFRESH_URL, data, { headers: { ...CONSTANTS2.REQUEST_HEADERS_BASE, "token": token } });
+    const response = await wavesApi.post(CONSTANTS.TOKEN_REFRESH_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, "token": token } });
     if (response.data.code === 220) {
       logger.info(`${roleId} 获取可用性成功，账号已过期`);
       return { status: false, msg: "账号已过期" };
@@ -1668,7 +1880,7 @@ async function refreshData(bat, serverId, roleId, token) {
     "roleId": roleId
   });
   try {
-    const response = await wavesApi.post(CONSTANTS2.REFRESH_URL, data, { headers: { ...CONSTANTS2.REQUEST_HEADERS_BASE, "token": token, "b-at": bat } });
+    const response = await wavesApi.post(CONSTANTS.REFRESH_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, "token": token, "b-at": bat } });
     if (response.data.code === 10902 || response.data.code === 200) {
       return { status: true, data: response.data.data };
     } else {
@@ -1695,7 +1907,7 @@ async function signIn(qq_uid) {
     "reqMonth": ((/* @__PURE__ */ new Date()).getMonth() + 1).toString().padStart(2, "0")
   });
   try {
-    const response = await wavesApi.post(CONSTANTS2.SIGNIN_URL, data, { headers: { ...CONSTANTS2.REQUEST_HEADERS_BASE, "token": userInfo.token, devcode: "", "b-at": userInfo.bat } });
+    const response = await wavesApi.post(CONSTANTS.SIGNIN_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, "token": userInfo.token, devcode: "", "b-at": userInfo.bat } });
     if (response.data.code === 200) {
       if (response.data.data === null) {
         logger.info(`[signIn Info]: ${qq_uid} 签到失败，返回空数据`);
@@ -1724,7 +1936,7 @@ async function getRoleData(qq_uid) {
     "roleId": userInfo.role_id
   });
   try {
-    const response = await wavesApi.post(CONSTANTS2.ROLE_DATA_URL, data, { headers: { ...CONSTANTS2.REQUEST_HEADERS_BASE, "token": userInfo.token, "b-at": userInfo.bat } });
+    const response = await wavesApi.post(CONSTANTS.ROLE_DATA_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, "token": userInfo.token, "b-at": userInfo.bat } });
     if (response.data.code === 10902 || response.data.code === 200) {
       response.data.data = JSON.parse(response.data.data);
       if (response.data.data === null || !response.data.data.showToGuest) {
@@ -1807,15 +2019,16 @@ __name(registerKuroCommands, "registerKuroCommands");
 // src/index.ts
 var name = "cocoyyy-console";
 var dev_mode;
-var Config = import_koishi9.Schema.object({
+var Config2 = import_koishi10.Schema.object({
   function_config: FunctionConfigSchema.description("功能开关配置"),
   redis_config: RedisConfigSchema.description("Redis配置"),
   mysql_config: MysqlConfigSchema.description("MySQL 数据库配置"),
   tag_config: SaveConfigSchema.description("tag图片保存配置"),
   repeat_config: RepeatConfigSchema.description("复读配置"),
-  test: import_koishi9.Schema.string().description("测试")
+  son_config: ShitOrNotConfigSchema.description("SON配置"),
+  test: import_koishi10.Schema.string().description("测试")
 });
-var logger = new import_koishi9.Logger(name);
+var logger = new import_koishi10.Logger(name);
 var savePath = null;
 async function apply(ctx, config) {
   ctx = ctx.guild();
@@ -1835,6 +2048,8 @@ async function apply(ctx, config) {
     registerTagCommands(ctx, connected, savePath);
   if (config.function_config.repeat_flag)
     registerRepeatMiddleware(ctx, config?.repeat_config);
+  if (config.function_config.son_flag)
+    registerShitOrNotCommands(ctx, config?.son_config);
   if (config.function_config.rbq_flag)
     registerRbqCommands(ctx, connected);
   if (config.function_config)

package/lib/services/son_func/shit_or_not_command.d.ts

@@ -1 +1,4 @@
-export {};
+import { Context } from 'koishi';
+import { ShitOrNotConfig } from '../../utils/config';
+declare function registerShitOrNotCommands(ctx: Context, config: ShitOrNotConfig): void;
+export { registerShitOrNotCommands };

package/lib/services/son_func/shit_or_not_service.d.ts

@@ -1,5 +1,5 @@
-import { OpenRouterMessage } from '../../utils/ai_axios';
 import { Session } from 'koishi';
+import { ShitOrNotConfig } from '../../utils/config';
 declare function ocrImage(buffer: Buffer): Promise<string>;
-declare function checkShitOrNot(session: Session, parttern_msg: string): Promise<OpenRouterMessage[]>;
+declare function checkShitOrNot(session: Session, parttern_msg: string, config: ShitOrNotConfig): Promise<string>;
 export { checkShitOrNot, ocrImage };

package/lib/utils/ai_axios.d.ts

@@ -1,17 +1,15 @@
 export interface OpenRouterMessage {
     role: 'system' | 'user' | 'assistant';
-    content: string;
+    content: Array<{
+        type: 'text' | 'image_url';
+        text?: string;
+        image_url?: {
+            url: string;
+        };
+    }>;
 }
 export interface OpenRouterResult {
     err: string | null;
     resp: string | null;
 }
-/**
- * 调用 OpenRouter API 进行对话
- * @param apiKey OpenRouter API Key
- * @param messages 消息数组，包含系统消息和用户消息
- * @param model 模型名称，默认为 gpt-4o-mini
- * @returns 返回包含错误和响应的对象 { err: string | null, resp: string | null }
- *          err 不为 null 时表示出错，resp 不为 null 时表示成功
- */
 export declare function callOpenRouter(api_url: string, api_key: string, api_model: string, messages: OpenRouterMessage[]): Promise<OpenRouterResult>;

package/lib/utils/config.d.ts

@@ -3,3 +3,4 @@ export * from './configs/mysql_config';
 export * from './configs/tag_config';
 export * from './configs/repeat_config';
 export * from './configs/redis_config';
+export * from './configs/son_config';

package/lib/utils/configs/function_config.d.ts

@@ -4,7 +4,7 @@ interface FunctionConfig {
     tag_flag: boolean;
     repeat_flag: boolean;
     rbq_flag: boolean;
-    shit_or_not_flag: boolean;
+    son_flag: boolean;
     kuro_flag: boolean;
 }
 declare const FunctionConfigSchema: Schema<FunctionConfig>;

package/lib/utils/configs/son_config.d.ts

@@ -3,6 +3,7 @@ interface ShitOrNotConfig {
     api_url?: string;
     api_model?: string;
     api_key?: string;
+    config_path?: string;
 }
 declare const ShitOrNotConfigSchema: Schema<ShitOrNotConfig>;
 export { ShitOrNotConfig, ShitOrNotConfigSchema, };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-cocoyyy-console",
   "description": "自用koishi插件，功能包含复读，记录黑历史，*人等",
-  "version": "1.0.16-beta.4",
+  "version": "1.0.16-beta.6",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "contributors": [
@@ -26,6 +26,7 @@
     "qs": "^6.14.0",
     "sequelize": "^6.37.3",
     "ioredis": "^5.8.2",
-    "tesseract.js": "^5.1.0"
+    "tesseract.js": "^5.1.0",
+    "js-yaml": "^4.1.1"
   }
-}
+}