@anyul/koishi-plugin-rss 5.2.3 → 5.2.4

package/lib/core/search-rotation.d.ts

@@ -0,0 +1,4 @@
+import { Config, SearchConfig } from '../types';
+export declare function getNextVolcengineModel(config: Config, searchConfig: SearchConfig): string;
+export declare function markVolcengineModelFailure(config: Config, searchConfig: SearchConfig, model: string): void;
+export declare function clearSearchRotationStates(): void;

package/lib/core/search-rotation.js

@@ -0,0 +1,55 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getNextVolcengineModel = getNextVolcengineModel;
+exports.markVolcengineModelFailure = markVolcengineModelFailure;
+exports.clearSearchRotationStates = clearSearchRotationStates;
+const logger_1 = require("../utils/logger");
+const modelRotationStates = new Map();
+function getModelRotationState(key, models) {
+    if (!modelRotationStates.has(key)) {
+        modelRotationStates.set(key, {
+            currentIndex: 0,
+            models,
+            lastFailureTime: 0,
+            failureCount: 0
+        });
+    }
+    return modelRotationStates.get(key);
+}
+function getNextVolcengineModel(config, searchConfig) {
+    if (searchConfig.volcengine?.models && searchConfig.volcengine.models.length > 0) {
+        const state = getModelRotationState('volcengine', searchConfig.volcengine.models);
+        const now = Date.now();
+        if (state.lastFailureTime > 0 && now - state.lastFailureTime < 60000) {
+            state.currentIndex = (state.currentIndex + 1) % state.models.length;
+            (0, logger_1.debug)(config, `模型轮询: 上次失败，切换到模型 ${state.models[state.currentIndex]}`, 'Search-Volcengine', 'info');
+        }
+        const model = state.models[state.currentIndex];
+        state.lastFailureTime = 0;
+        return model;
+    }
+    if (searchConfig.volcengine?.useAiModel !== false && config.ai?.model) {
+        return config.ai.model;
+    }
+    const defaultModels = [
+        'doubao-seed-1-6-lite-251015',
+        'doubao-seed-1-6-flash-250828'
+    ];
+    const state = getModelRotationState('volcengine-default', defaultModels);
+    const model = defaultModels[state.currentIndex];
+    state.currentIndex = (state.currentIndex + 1) % defaultModels.length;
+    return model;
+}
+function markVolcengineModelFailure(config, searchConfig, model) {
+    const key = searchConfig.volcengine?.models ? 'volcengine' : 'volcengine-default';
+    const state = modelRotationStates.get(key);
+    if (state) {
+        state.lastFailureTime = Date.now();
+        state.failureCount++;
+        state.currentIndex = (state.currentIndex + 1) % state.models.length;
+        (0, logger_1.debug)(config, `模型 ${model} 失败，切换到下一个模型 ${state.models[state.currentIndex]} (失败次数: ${state.failureCount})`, 'Search-Volcengine', 'info');
+    }
+}
+function clearSearchRotationStates() {
+    modelRotationStates.clear();
+}

package/lib/core/search-service.d.ts

@@ -0,0 +1,3 @@
+import { Config, SearchConfig } from '../types';
+import { SearchResponse } from './search-types';
+export declare function webSearch(config: Config, query: string, searchConfig: SearchConfig): Promise<SearchResponse>;

package/lib/core/search-service.js

@@ -0,0 +1,100 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.webSearch = webSearch;
+const logger_1 = require("../utils/logger");
+const search_providers_1 = require("./search-providers");
+function createErrorResponse(query, engine, error) {
+    return {
+        success: false,
+        results: [],
+        query,
+        engine,
+        error
+    };
+}
+function getAvailableAutoEngines(searchConfig) {
+    const enginePriority = searchConfig.enginePriority || ['tavily', 'volcengine', 'searxng'];
+    return enginePriority.filter(engine => {
+        switch (engine) {
+            case 'tavily':
+                return !!searchConfig.tavily?.apiKey;
+            case 'searxng':
+                return !!searchConfig.searxng?.instanceUrl;
+            case 'volcengine':
+                return !!searchConfig.volcengine?.apiKey;
+            default:
+                return false;
+        }
+    });
+}
+async function webSearch(config, query, searchConfig) {
+    if (!searchConfig.enabled) {
+        return createErrorResponse(query, 'none', '联网搜索未启用');
+    }
+    if (searchConfig.engine === 'auto') {
+        return autoSelectEngineAndSearch(config, query, searchConfig);
+    }
+    switch (searchConfig.engine) {
+        case 'tavily':
+            if (!searchConfig.tavily?.apiKey) {
+                return createErrorResponse(query, 'tavily', 'Tavily API Key 未配置');
+            }
+            return (0, search_providers_1.searchWithTavily)(config, query, searchConfig.tavily.apiKey, {
+                maxResults: searchConfig.maxResults || 5,
+                searchDepth: searchConfig.tavily.searchDepth || 'basic',
+                includeAnswer: searchConfig.tavily.includeAnswer !== false
+            });
+        case 'searxng':
+            if (!searchConfig.searxng?.instanceUrl) {
+                return createErrorResponse(query, 'searxng', 'SearXNG 实例 URL 未配置');
+            }
+            return (0, search_providers_1.searchWithSearxng)(config, query, searchConfig.searxng.instanceUrl, {
+                maxResults: searchConfig.maxResults || 5,
+                language: searchConfig.searxng.language || 'all',
+                categories: searchConfig.searxng.categories || ['general']
+            });
+        case 'volcengine':
+            if (!searchConfig.volcengine?.apiKey) {
+                return createErrorResponse(query, 'volcengine', '火山引擎 API Key 未配置');
+            }
+            return (0, search_providers_1.searchWithVolcengine)(config, query, config.ai?.baseUrl || 'https://ark.cn-beijing.volces.com/api/v3', searchConfig.volcengine.apiKey, undefined, searchConfig);
+        default:
+            return createErrorResponse(query, 'unknown', `未知的搜索引擎: ${searchConfig.engine}`);
+    }
+}
+async function autoSelectEngineAndSearch(config, query, searchConfig) {
+    const availableEngines = getAvailableAutoEngines(searchConfig);
+    if (availableEngines.length === 0) {
+        return createErrorResponse(query, 'auto', '没有配置任何可用的搜索引擎');
+    }
+    (0, logger_1.debug)(config, `自动选择搜索引擎，可用引擎: ${availableEngines.join(', ')}`, 'Search-Auto', 'info');
+    for (const engine of availableEngines) {
+        (0, logger_1.debug)(config, `尝试使用搜索引擎: ${engine}`, 'Search-Auto', 'info');
+        let result;
+        switch (engine) {
+            case 'tavily':
+                result = await (0, search_providers_1.searchWithTavily)(config, query, searchConfig.tavily.apiKey, {
+                    maxResults: searchConfig.maxResults || 5,
+                    searchDepth: searchConfig.tavily?.searchDepth || 'basic',
+                    includeAnswer: searchConfig.tavily?.includeAnswer !== false
+                });
+                break;
+            case 'searxng':
+                result = await (0, search_providers_1.searchWithSearxng)(config, query, searchConfig.searxng.instanceUrl, {
+                    maxResults: searchConfig.maxResults || 5,
+                    language: searchConfig.searxng?.language || 'all',
+                    categories: searchConfig.searxng?.categories || ['general']
+                });
+                break;
+            case 'volcengine':
+                result = await (0, search_providers_1.searchWithVolcengine)(config, query, config.ai?.baseUrl || 'https://ark.cn-beijing.volces.com/api/v3', searchConfig.volcengine.apiKey, undefined, searchConfig);
+                break;
+        }
+        if (result.success && result.results.length > 0) {
+            (0, logger_1.debug)(config, `搜索引擎 ${engine} 成功返回 ${result.results.length} 条结果`, 'Search-Auto', 'info');
+            return result;
+        }
+        (0, logger_1.debug)(config, `搜索引擎 ${engine} 失败: ${result.error || '无结果'}，尝试下一个引擎`, 'Search-Auto', 'info');
+    }
+    return createErrorResponse(query, 'auto', '所有搜索引擎都失败了');
+}

package/lib/core/search-types.d.ts

@@ -0,0 +1,39 @@
+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;
+}
+export interface TavilyResponse {
+    answer?: string;
+    query: string;
+    results: Array<{
+        title: string;
+        url: string;
+        content: string;
+        score: number;
+        published_date?: string;
+    }>;
+}
+export interface SearxngResponse {
+    query: string;
+    results: Array<{
+        title: string;
+        url: string;
+        content: string;
+        snippet?: string;
+        engine?: string;
+        score?: number;
+    }>;
+}

package/lib/core/search-types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/lib/core/search.d.ts

@@ -1,101 +1,4 @@
-/**
- * 联网搜索模块
- *
- * 支持多种搜索引擎：
- * - 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;
+export { buildPromptWithSearchContext, formatSearchResults } from './search-format';
+export { searchWithSearxng, searchWithTavily, searchWithVolcengine } from './search-providers';
+export { webSearch } from './search-service';
+export type { SearchResponse, SearchResult } from './search-types';