koishi-plugin-best-cave 2.7.5 → 2.7.6

package/lib/AIManager.d.ts

@@ -30,6 +30,8 @@ export declare class AIManager {
     private logger;
     private fileManager;
     private http;
+    private requestCount;
+    private rateLimitResetTime;
     /**
      * @constructor
      * @description AIManager 类的构造函数，负责初始化依赖项，并向 Koishi 的数据库模型中注册 `cave_meta` 表。

package/lib/index.d.ts

@@ -61,6 +61,7 @@ export interface Config {
     aiEndpoint?: string;
     aiApiKey?: string;
     aiModel?: string;
+    aiTPM?: number;
     AnalysePrompt?: string;
     aiCheckPrompt?: string;
     aiAnalyseSchema?: string;

package/lib/index.js

@@ -999,6 +999,8 @@ var AIManager = class {
     __name(this, "AIManager");
   }
   http;
+  requestCount = 0;
+  rateLimitResetTime = 0;
   /**
    * @description 注册所有与 AIManager 功能相关的 Koishi 命令。
    * @param {any} cave - 主 `cave` 命令的实例，用于在其下注册子命令。
@@ -1013,21 +1015,18 @@ var AIManager = class {
         if (cavesToAnalyze.length === 0) return "无需分析回声洞";
         await session.send(`开始分析 ${cavesToAnalyze.length} 个回声洞...`);
         let successCount = 0;
-        const batchSize = 100;
-        for (let i = 0; i < cavesToAnalyze.length; i += batchSize) {
-          const batch = cavesToAnalyze.slice(i, i + batchSize);
-          this.logger.info(`[${i}/${cavesToAnalyze.length}] 正在分析 ${batch.length} 条回声洞...`);
-          for (const cave2 of batch) {
-            try {
-              await this.analyzeAndStore(cave2);
-              successCount++;
-            } catch (error) {
-              this.logger.error(`分析回声洞（${cave2.id}）时出错:`, error);
-              return `分析回声洞（${cave2.id}）时出错: ${error.message}`;
-            }
+        let failedCount = 0;
+        for (const [index, cave2] of cavesToAnalyze.entries()) {
+          this.logger.info(`[${index + 1}/${cavesToAnalyze.length}] 正在分析回声洞 (${cave2.id})...`);
+          try {
+            await this.analyzeAndStore(cave2);
+            successCount++;
+          } catch (error) {
+            failedCount++;
+            this.logger.error(`分析回声洞（${cave2.id}）时出错:`, error);
           }
         }
-        return `已分析 ${successCount} 个回声洞`;
+        return `已分析 ${successCount} 个回声洞（失败 ${failedCount} 个）`;
       } catch (error) {
         this.logger.error("分析回声洞失败:", error);
         return `操作失败: ${error.message}`;
@@ -1166,6 +1165,17 @@ var AIManager = class {
    * @throws {Error} 当 JSON Schema 解析失败、网络请求失败或 AI 返回错误时，抛出异常。
    */
   async requestAI(messages, systemPrompt, schemaString) {
+    const now = Date.now();
+    if (now > this.rateLimitResetTime) {
+      this.rateLimitResetTime = now + 6e4;
+      this.requestCount = 0;
+    }
+    if (this.requestCount >= this.config.aiTPM) {
+      const delay = this.rateLimitResetTime - now;
+      await new Promise((resolve) => setTimeout(resolve, delay));
+      this.rateLimitResetTime = Date.now() + 6e4;
+      this.requestCount = 0;
+    }
     let schema = JSON.parse(schemaString);
     const payload = {
       model: this.config.aiModel,
@@ -1178,6 +1188,7 @@ var AIManager = class {
       "Authorization": `Bearer ${this.config.aiApiKey}`
     };
     try {
+      this.requestCount++;
       const response = await this.http.post(fullUrl, payload, { headers, timeout: 9e4 });
       return JSON.parse(response.choices[0].message.content);
     } catch (error) {
@@ -1222,8 +1233,9 @@ var Config = import_koishi4.Schema.intersect([
     enableAI: import_koishi4.Schema.boolean().default(false).description("启用 AI"),
     aiEndpoint: import_koishi4.Schema.string().description("端点 (Endpoint)").role("link").default("https://generativelanguage.googleapis.com/v1beta/openai"),
     aiApiKey: import_koishi4.Schema.string().description("密钥 (Key)").role("secret"),
-    aiModel: import_koishi4.Schema.string().description("模型").default("gemini-2.5-flash"),
-    AnalysePrompt: import_koishi4.Schema.string().role("textarea").default(`你是一位内容分析专家。请分析我提供的内容，总结关键词，概括内容并进行评分。`).description("分析提示词 (Prompt)"),
+    aiModel: import_koishi4.Schema.string().description("模型 (Model)").default("gemini-2.5-flash"),
+    aiTPM: import_koishi4.Schema.number().description("每分钟请求数 (TPM)").default(60),
+    AnalysePrompt: import_koishi4.Schema.string().role("textarea").default(`你是一位内容分析专家。请分析我提供的内容，总结关键词，概括内容并进行评分。`).description("分析 Prompt"),
     aiAnalyseSchema: import_koishi4.Schema.string().role("textarea").default(
       `{
         "type": "object",
@@ -1246,8 +1258,8 @@ var Config = import_koishi4.Schema.intersect([
         },
         "required": ["keywords", "description", "rating"]
       }`
-    ).description("分析输出模式 (JSON Schema)"),
-    aiCheckPrompt: import_koishi4.Schema.string().role("textarea").default(`你是一位内容查重专家。请判断我提供的"新内容"是否与"已有内容"重复或高度相似。`).description("查重提示词 (Prompt)"),
+    ).description("分析 JSON Schema"),
+    aiCheckPrompt: import_koishi4.Schema.string().role("textarea").default(`你是一位内容查重专家。请判断我提供的"新内容"是否与"已有内容"重复或高度相似。`).description("查重 Prompt"),
     aiCheckSchema: import_koishi4.Schema.string().role("textarea").default(
       `{
         "type": "object",
@@ -1263,7 +1275,7 @@ var Config = import_koishi4.Schema.intersect([
         },
         "required": ["duplicate"]
       }`
-    ).description("查重输出模式 (JSON Schema)")
+    ).description("查重 JSON Schema")
   }).description("模型配置"),
   import_koishi4.Schema.object({
     localPath: import_koishi4.Schema.string().description("文件映射路径"),

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-best-cave",
   "description": "功能强大、高度可定制的回声洞。支持丰富的媒体类型、内容查重、人工审核、用户昵称、数据迁移以及本地/S3 双重文件存储后端。",
-  "version": "2.7.5",
+  "version": "2.7.6",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],