koishi-plugin-cocoyyy-console 1.0.15 → 1.0.16-beta.2

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;
+    test?: string;
 }
 export declare const Config: Schema<Config>;
 export declare let logger: Logger;

package/lib/index.js

@@ -321,7 +321,7 @@ function getMenuList(command = null, functionConfig) {
       return `[所有命令都需要@bot]
 当前可用功能列表:
 	${menuList.map((item) => item.name + ": " + item.description).join("\n	")}
-输入"help 功能名"查看特定功能的指令和使用示例。`;
+输入"help [功能名]"查看特定功能的指令和使用示例。`;
   }
 }
 __name(getMenuList, "getMenuList");
@@ -1351,38 +1351,95 @@ function registerRbqCommands(ctx, connected) {
 }
 __name(registerRbqCommands, "registerRbqCommands");
 
-// src/services/test_command.ts
-function registerTestCommands(ctx) {
-  ctx.command("test <参数>", "test").action(async ({ session }, phoneNumber, code) => {
-    try {
-    } catch (err) {
-      console.error(err);
-      return `test error: ${err.message}`;
-    }
-  });
-}
-__name(registerTestCommands, "registerTestCommands");
-
-// src/infra/redis_init.ts
-var import_ioredis = __toESM(require("ioredis"));
-var redis;
-function init_redis(config) {
+// src/utils/ai_axios.ts
+var import_axios = __toESM(require("axios"));
+async function callOpenRouter(apiKey, messages, model = "gpt-4o-mini") {
   try {
-    if (!redis) {
-      redis = new import_ioredis.default(config.redis_url);
+    const { data } = await import_axios.default.post(
+      "https://openrouter.ai/api/v1/chat/completions",
+      {
+        model,
+        messages
+      },
+      {
+        headers: {
+          Authorization: `Bearer ${apiKey}`,
+          "Content-Type": "application/json"
+        }
+      }
+    );
+    const choices = data?.choices ?? [];
+    const reply = choices[0]?.message?.content;
+    let responseText;
+    if (typeof reply === "string") {
+      responseText = reply.trim();
+    } else if (Array.isArray(reply)) {
+      responseText = reply.map((part) => part?.text ?? "").join("").trim();
+      if (!responseText) {
+        responseText = "OpenRouter 返回空内容";
+      }
+    } else {
+      responseText = "OpenRouter 返回了未知格式，请检查日志";
     }
-    redis.ping();
-    logger.info("redis连接成功");
-    return true;
+    return { err: null, resp: responseText };
   } catch (error) {
-    logger.error("redis连接失败: " + error);
-    return false;
+    const err = error;
+    const status = err.response?.status;
+    const detail = err.response?.data?.error || err.response?.data?.message || err.message;
+    logger.error(error, "OpenRouter 请求失败");
+    const errorMessage = `test error${status ? `(${status})` : ""}: ${detail}`;
+    return { err: errorMessage, resp: null };
   }
 }
-__name(init_redis, "init_redis");
+__name(callOpenRouter, "callOpenRouter");
+
+// src/services/test_command.ts
+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
+    ];
+    const { err, resp } = await callOpenRouter(apiKey, messages);
+    if (err) {
+      return err;
+    }
+    if (!resp) {
+      return "OpenRouter 返回空响应";
+    }
+    chatHistory.push({ role: "assistant", content: resp });
+    return resp;
+  });
+  ctx.command("test <text>", "调用 OpenRouter 模型进行测试").action(async ({ session }, text) => {
+    return text;
+  });
+  ctx.middleware((session, next) => {
+    const { content, uid, userId } = session;
+    const cid = session.cid;
+    if (session.guildId != "829263238") return next();
+    if (ctx.bots[uid]) return next();
+    if (session.quote) {
+      const quoteContent = session.quote.content;
+      const quoteUserId = session.quote.user.id;
+      logger.error(`quoteContent: ${quoteContent}, quoteUserId: ${quoteUserId},content: ${content} userId: ${userId}`);
+    } else {
+      logger.error(`content: ${content} userId: ${userId}`);
+    }
+  });
+}
+__name(registerTestCommands, "registerTestCommands");
 
 // src/services/kuro_func/kuro_service.ts
-var import_axios = __toESM(require("axios"));
+var import_axios2 = __toESM(require("axios"));
 var import_qs = __toESM(require("qs"));
 
 // src/models/kuro_user.ts
@@ -1491,7 +1548,7 @@ __name(createUpdateUser, "createUpdateUser");
 
 // src/services/kuro_func/kuro_service.ts
 var userTokenMap = /* @__PURE__ */ new Map();
-var CONSTANTS = {
+var CONSTANTS2 = {
   BASE_URL: "https://api.kurobbs.com",
   LOGIN_URL: "/user/sdkLogin",
   GAME_DATA_URL: "/gamer/role/list",
@@ -1515,11 +1572,11 @@ var CONSTANTS = {
     "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 = CONSTANTS.BASE_URL + config.url;
+      config.url = CONSTANTS2.BASE_URL + config.url;
     }
     return config;
   },
@@ -1573,7 +1630,7 @@ async function getToken(session, mobile, code) {
     code
   });
   try {
-    const user_response = await wavesApi.post(CONSTANTS.LOGIN_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, devCode } });
+    const user_response = await wavesApi.post(CONSTANTS2.LOGIN_URL, data, { headers: { ...CONSTANTS2.REQUEST_HEADERS_BASE, devCode } });
     if (user_response.data.code != 200) {
       logger.error(`验证码登录失败: ${user_response.data.msg}`);
       return { status: false, msg: user_response.data.msg };
@@ -1613,7 +1670,7 @@ async function getGameData(token) {
     let data = import_qs.default.stringify({
       "gameId": 3
     });
-    const response = await wavesApi.post(CONSTANTS.GAME_DATA_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, "token": token } });
+    const response = await wavesApi.post(CONSTANTS2.GAME_DATA_URL, data, { headers: { ...CONSTANTS2.REQUEST_HEADERS_BASE, "token": token } });
     if (response.data.code === 200) {
       return { status: true, data: response.data.data[0] };
     } else {
@@ -1632,7 +1689,7 @@ async function isAvailable(serverId, roleId, token) {
     "roleId": roleId
   });
   try {
-    const response = await wavesApi.post(CONSTANTS.TOKEN_REFRESH_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, "token": token } });
+    const response = await wavesApi.post(CONSTANTS2.TOKEN_REFRESH_URL, data, { headers: { ...CONSTANTS2.REQUEST_HEADERS_BASE, "token": token } });
     if (response.data.code === 220) {
       logger.info(`${roleId} 获取可用性成功，账号已过期`);
       return { status: false, msg: "账号已过期" };
@@ -1653,7 +1710,7 @@ async function refreshData(bat, serverId, roleId, token) {
     "roleId": roleId
   });
   try {
-    const response = await wavesApi.post(CONSTANTS.REFRESH_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, "token": token, "b-at": bat } });
+    const response = await wavesApi.post(CONSTANTS2.REFRESH_URL, data, { headers: { ...CONSTANTS2.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 {
@@ -1680,7 +1737,7 @@ async function signIn(qq_uid) {
     "reqMonth": ((/* @__PURE__ */ new Date()).getMonth() + 1).toString().padStart(2, "0")
   });
   try {
-    const response = await wavesApi.post(CONSTANTS.SIGNIN_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, "token": userInfo.token, devcode: "", "b-at": userInfo.bat } });
+    const response = await wavesApi.post(CONSTANTS2.SIGNIN_URL, data, { headers: { ...CONSTANTS2.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} 签到失败，返回空数据`);
@@ -1709,7 +1766,7 @@ async function getRoleData(qq_uid) {
     "roleId": userInfo.role_id
   });
   try {
-    const response = await wavesApi.post(CONSTANTS.ROLE_DATA_URL, data, { headers: { ...CONSTANTS.REQUEST_HEADERS_BASE, "token": userInfo.token, "b-at": userInfo.bat } });
+    const response = await wavesApi.post(CONSTANTS2.ROLE_DATA_URL, data, { headers: { ...CONSTANTS2.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) {
@@ -1797,7 +1854,8 @@ var Config = import_koishi9.Schema.object({
   redis_config: RedisConfigSchema.description("Redis配置"),
   mysql_config: MysqlConfigSchema.description("MySQL 数据库配置"),
   tag_config: SaveConfigSchema.description("tag图片保存配置"),
-  repeat_config: RepeatConfigSchema.description("复读配置")
+  repeat_config: RepeatConfigSchema.description("复读配置"),
+  test: import_koishi9.Schema.string().description("测试")
 });
 var logger = new import_koishi9.Logger(name);
 var savePath = null;
@@ -1805,7 +1863,6 @@ async function apply(ctx, config) {
   ctx = ctx.guild();
   dev_mode = config.function_config.dev_mode;
   initSequelize(config?.mysql_config);
-  init_redis(config?.redis_config);
   const connected = await testConnect();
   if (connected) {
     await getTagsModel().sync();
@@ -1825,7 +1882,7 @@ async function apply(ctx, config) {
   if (config.function_config)
     registerKuroCommands(ctx, connected);
   if (dev_mode)
-    registerTestCommands(ctx);
+    registerTestCommands(ctx, config);
 }
 __name(apply, "apply");
 // Annotate the CommonJS export names for ESM import in node:

package/lib/services/test_command.d.ts

@@ -1,3 +1,4 @@
 import { Context } from 'koishi';
-declare function registerTestCommands(ctx: Context): void;
+import { Config } from '..';
+declare function registerTestCommands(ctx: Context, config: Config): void;
 export { registerTestCommands };

package/lib/utils/ai_axios.d.ts

@@ -0,0 +1,17 @@
+export interface OpenRouterMessage {
+    role: 'system' | 'user' | 'assistant';
+    content: 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(apiKey: string, messages: OpenRouterMessage[], model?: string): Promise<OpenRouterResult>;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-cocoyyy-console",
   "description": "自用koishi插件，功能包含复读，记录黑历史，*人等",
-  "version": "1.0.15",
+  "version": "1.0.16-beta.2",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "contributors": [