@anyul/koishi-plugin-rss 5.0.5 → 5.1.0

package/lib/config.js

@@ -94,6 +94,59 @@ exports.Config = koishi_1.Schema.object({
         maxInputLength: koishi_1.Schema.number().description('发送给 AI 的最大字数限制').default(2000),
         timeout: koishi_1.Schema.number().description('AI 请求超时时间(毫秒)').default(30000),
     }).description('AI 摘要设置'),
+    search: koishi_1.Schema.intersect([
+        koishi_1.Schema.object({
+            enabled: koishi_1.Schema.boolean().description('启用联网搜索增强 AI 摘要').default(false),
+            engine: koishi_1.Schema.union(['tavily', 'searxng', 'volcengine', 'auto']).description('搜索引擎选择（auto=自动按优先级选择）').default('tavily'),
+            maxResults: koishi_1.Schema.number().description('最大搜索结果数').default(5).min(1).max(10),
+            enginePriority: koishi_1.Schema.array(koishi_1.Schema.union(['tavily', 'searxng', 'volcengine'])).description('引擎优先级（当 engine 为 auto 时使用）').default(['tavily', 'volcengine', 'searxng']),
+        }),
+        koishi_1.Schema.union([
+            koishi_1.Schema.object({
+                engine: koishi_1.Schema.const('tavily').required(),
+                tavily: koishi_1.Schema.intersect([
+                    koishi_1.Schema.object({ enabled: koishi_1.Schema.const(true).required() }),
+                    koishi_1.Schema.union([
+                        koishi_1.Schema.object({
+                            apiKey: koishi_1.Schema.string().role('secret').description('Tavily API Key（获取地址: https://tavily.com）').required(),
+                            searchDepth: koishi_1.Schema.union(['basic', 'advanced']).description('搜索深度').default('basic'),
+                            includeAnswer: koishi_1.Schema.boolean().description('是否包含 AI 生成的搜索答案').default(true),
+                        }),
+                        koishi_1.Schema.object({}),
+                    ]),
+                ]),
+            }),
+            koishi_1.Schema.object({
+                engine: koishi_1.Schema.const('searxng').required(),
+                searxng: koishi_1.Schema.intersect([
+                    koishi_1.Schema.object({ enabled: koishi_1.Schema.const(true).required() }),
+                    koishi_1.Schema.union([
+                        koishi_1.Schema.object({
+                            instanceUrl: koishi_1.Schema.string().role('link').description('SearXNG 实例 URL（自建或公共实例）').required(),
+                            language: koishi_1.Schema.string().description('搜索语言（例如: all, zh, en）').default('all'),
+                            categories: koishi_1.Schema.array(koishi_1.Schema.union(['general', 'news', 'images', 'videos'])).description('搜索类别').default(['general']),
+                        }),
+                        koishi_1.Schema.object({}),
+                    ]),
+                ]),
+            }),
+            koishi_1.Schema.object({
+                engine: koishi_1.Schema.const('volcengine').required(),
+                volcengine: koishi_1.Schema.intersect([
+                    koishi_1.Schema.object({ enabled: koishi_1.Schema.const(true).required() }),
+                    koishi_1.Schema.union([
+                        koishi_1.Schema.object({
+                            apiKey: koishi_1.Schema.string().role('secret').description('火山引擎 API Key（与 AI 配置中的 API Key 相同，或单独配置）').required(),
+                            models: koishi_1.Schema.array(koishi_1.Schema.string()).description('模型列表（支持轮询，留空则使用默认模型）').default([]),
+                            useAiModel: koishi_1.Schema.boolean().description('是否使用 AI 配置中的 model（优先级高于 models）').default(true),
+                        }),
+                        koishi_1.Schema.object({}),
+                    ]),
+                ]),
+            }),
+            koishi_1.Schema.object({}),
+        ]),
+    ]).description('联网搜索设置（AI 摘要增强）'),
     cache: koishi_1.Schema.object({
         enabled: koishi_1.Schema.boolean().description('启用消息缓存').default(true),
         maxSize: koishi_1.Schema.number().description('最大缓存消息条数').default(100),

package/lib/core/ai.js

@@ -50,6 +50,7 @@ const https_proxy_agent_1 = require("https-proxy-agent");
 const cheerio = __importStar(require("cheerio"));
 const crypto = __importStar(require("crypto"));
 const logger_1 = require("../utils/logger");
+const search_1 = require("./search");
 /**
  * AI 摘要缓存管理器
  */
@@ -247,9 +248,31 @@ async function getAiSummary(config, title, contentHtml) {
         return cachedSummary;
     }
     // 构建 Prompt
-    const prompt = config.ai.prompt
+    let prompt = config.ai.prompt
         .replace('{{title}}', title || '')
         .replace('{{content}}', plainText);
+    // 如果启用了联网搜索，进行搜索并增强 Prompt
+    if (config.search?.enabled) {
+        try {
+            // 生成搜索查询（使用标题作为查询）
+            const searchQuery = title || plainText.substring(0, 100);
+            (0, logger_1.debug)(config, `正在联网搜索: ${searchQuery}`, 'AI-Search', 'info');
+            // 执行搜索
+            const searchResults = await (0, search_1.webSearch)(config, searchQuery, config.search);
+            // 如果搜索成功，将搜索结果添加到 Prompt 中
+            if (searchResults.success && searchResults.results.length > 0) {
+                (0, logger_1.debug)(config, `联网搜索成功，找到 ${searchResults.results.length} 条结果`, 'AI-Search', 'details');
+                prompt = (0, search_1.buildPromptWithSearchContext)(prompt, searchResults, searchQuery);
+            }
+            else if (searchResults.error) {
+                (0, logger_1.debug)(config, `联网搜索失败: ${searchResults.error}`, 'AI-Search', 'info');
+            }
+        }
+        catch (error) {
+            (0, logger_1.debug)(config, `联网搜索异常: ${error.message}`, 'AI-Search', 'error');
+            // 搜索失败时不影响摘要生成，继续使用原始 Prompt
+        }
+    }
     // 调用 AI API
     const result = await callAiApi(config, prompt, `单条摘要: ${title}`);
     // 缓存成功的结果
@@ -283,7 +306,7 @@ async function getBatchAiSummary(config, items) {
             return '';
         }
         // 构建批量 Prompt
-        const prompt = `请简要总结以下 ${cleanedItems.length} 条新闻/文章的核心内容，要求：
+        let prompt = `请简要总结以下 ${cleanedItems.length} 条新闻/文章的核心内容，要求：
 1. 语言简洁流畅，每条总结不超过30字
 2. 按顺序总结，使用数字编号
 3. 突出重点信息
@@ -294,6 +317,28 @@ ${index + 1}. 标题：${item.title}
 `).join('\n')}
 
 总结：`;
+        // 如果启用了联网搜索，对第一个或最重要的标题进行搜索
+        if (config.search?.enabled && cleanedItems.length > 0) {
+            try {
+                // 使用第一条内容的标题作为搜索查询
+                const searchQuery = cleanedItems[0].title;
+                (0, logger_1.debug)(config, `批量摘要 - 正在联网搜索: ${searchQuery}`, 'AI-Search', 'info');
+                // 执行搜索
+                const searchResults = await (0, search_1.webSearch)(config, searchQuery, config.search);
+                // 如果搜索成功，将搜索结果添加到 Prompt 中
+                if (searchResults.success && searchResults.results.length > 0) {
+                    (0, logger_1.debug)(config, `批量摘要 - 联网搜索成功，找到 ${searchResults.results.length} 条结果`, 'AI-Search', 'details');
+                    prompt = (0, search_1.buildPromptWithSearchContext)(prompt, searchResults, searchQuery);
+                }
+                else if (searchResults.error) {
+                    (0, logger_1.debug)(config, `批量摘要 - 联网搜索失败: ${searchResults.error}`, 'AI-Search', 'info');
+                }
+            }
+            catch (error) {
+                (0, logger_1.debug)(config, `批量摘要 - 联网搜索异常: ${error.message}`, 'AI-Search', 'error');
+                // 搜索失败时不影响摘要生成，继续使用原始 Prompt
+            }
+        }
         // 调用 AI API
         const result = await callAiApi(config, prompt, `批量摘要: ${cleanedItems.length}条`);
         if (result.success && result.summary) {

package/lib/core/search.d.ts

@@ -0,0 +1,101 @@
+/**
+ * 联网搜索模块
+ *
+ * 支持多种搜索引擎：
+ * - Tavily: 专业的 AI 搜索引擎
+ * - Searxng: 开源隐私友好的元搜索引擎
+ * - Volcengine: 火山引擎联网搜索（支持模型轮询）
+ *
+ * @module core/search
+ */
+import { Config, SearchConfig } from '../types';
+/**
+ * 搜索结果接口
+ */
+export interface SearchResult {
+    title: string;
+    url: string;
+    snippet?: string;
+    content?: string;
+    score?: number;
+    publishedDate?: string;
+    source?: string;
+}
+/**
+ * 搜索响应接口
+ */
+export interface SearchResponse {
+    success: boolean;
+    results: SearchResult[];
+    query: string;
+    engine: string;
+    model?: string;
+    error?: string;
+}
+/**
+ * Tavily 搜索引擎
+ *
+ * @param config - 插件配置
+ * @param query - 搜索查询
+ * @param apiKey - Tavily API Key
+ * @returns 搜索结果
+ */
+export declare function searchWithTavily(config: Config, query: string, apiKey: string, options?: {
+    maxResults?: number;
+    searchDepth?: 'basic' | 'advanced';
+    includeAnswer?: boolean;
+}): Promise<SearchResponse>;
+/**
+ * SearXNG 搜索引擎
+ *
+ * @param config - 插件配置
+ * @param query - 搜索查询
+ * @param instanceUrl - SearXNG 实例 URL
+ * @returns 搜索结果
+ */
+export declare function searchWithSearxng(config: Config, query: string, instanceUrl: string, options?: {
+    maxResults?: number;
+    language?: string;
+    categories?: Array<'general' | 'news' | 'images' | 'videos'>;
+}): Promise<SearchResponse>;
+/**
+ * 火山引擎联网搜索（支持模型轮询）
+ *
+ * @param config - 插件配置
+ * @param query - 搜索查询
+ * @param baseUrl - API Base URL
+ * @param apiKey - API Key
+ * @param model - 模型名称（可选，如果不指定则使用轮询）
+ * @param searchConfig - 搜索配置（用于模型轮询）
+ * @returns 搜索结果
+ */
+export declare function searchWithVolcengine(config: Config, query: string, baseUrl: string, apiKey: string, model?: string, searchConfig?: SearchConfig): Promise<SearchResponse>;
+/**
+ * 统一搜索接口
+ *
+ * 根据配置自动选择搜索引擎并执行搜索
+ * - 如果 engine 为 'auto'，则根据配置的 API Keys 按优先级自动选择
+ * - 支持多引擎配置时智能选择
+ *
+ * @param config - 插件配置
+ * @param query - 搜索查询
+ * @param searchConfig - 搜索配置
+ * @returns 搜索结果
+ */
+export declare function webSearch(config: Config, query: string, searchConfig: SearchConfig): Promise<SearchResponse>;
+/**
+ * 将搜索结果格式化为 AI 可读的文本
+ *
+ * @param response - 搜索响应
+ * @returns 格式化的文本
+ */
+export declare function formatSearchResults(response: SearchResponse): string;
+/**
+ * 构建带搜索上下文的 AI Prompt
+ *
+ * @param originalPrompt - 原始提示词
+ * @param searchResults - 搜索结果
+ * @param searchQuery - 搜索查询
+ * @returns 增强后的提示词
+ */
+export declare function buildPromptWithSearchContext(originalPrompt: string, searchResults: SearchResponse, searchQuery: string): string;