ITADN

Optimize MSDL Search Engine Configuration System

#262Pull Requestlcolok 创建于 2024-11-21已合并
L
lcolokcommented
# Optimize MSDL Search Engine Configuration System > Seamlessly switch between search engines with enhanced configuration management ![image](https://github.com/user-attachments/assets/0e64e7d1-8e70-448a-96ce-400b6bb635c6) ## Changes ### 1. Search Engine Configuration Enhancement - Refactored search engine configuration system with dedicated environment variables - Added specialized API key prompts for different search engines - Improved search engine configuration status feedback ### 2. Environment Variable Management - Introduced dedicated search engine API key variables: * `BING_SEARCH_API_KEY` * `BRAVE_SEARCH_API_KEY` * `GOOGLE_SERPER_API_KEY` * `TENCENT_SEARCH_SECRET_ID/KEY` - Maintained backward compatibility with `WEB_SEARCH_API_KEY` - Optimized environment variable file structure ### 3. Code Structure Improvements - Added dedicated environment variable fields in `SEARCH_ENGINES` configuration - Refactored Docker Compose environment variable setup logic - Enhanced API key validation rules for multiple key formats - Improved error handling in key input and validation flow ## Test Items - [x] Configuration and validation of dedicated search engine API keys - [x] API key backward compatibility - [x] Tencent Search dual-key configuration process ## Modified Files - docker/msdl/user_interaction.py - docker/msdl/utils.py - docker/msdl/translations/en.yaml - docker/msdl/translations/zh_CN.yaml --- # 为 MSDL 增加搜索引擎配置 > 现在您可以轻松切换各类搜索引擎 ![image](https://github.com/user-attachments/assets/c5dfc077-25e1-4dd7-8a8f-d6fbf68b759e) ## 变更内容 ### 1. 搜索引擎配置优化 - 重构搜索引擎配置系统,为每个搜索引擎提供专门的环境变量 - 为不同类型的 API 密钥添加专门的提示信息 - 添加搜索引擎配置成功的状态提示 ### 2. 环境变量管理改进 - 新增专门的搜索引擎 API 密钥环境变量: * `BING_SEARCH_API_KEY` * `BRAVE_SEARCH_API_KEY` * `GOOGLE_SERPER_API_KEY` * `TENCENT_SEARCH_SECRET_ID/KEY` - 保持与现有 `WEB_SEARCH_API_KEY` 的向后兼容 - 优化环境变量文件格式和结构 ### 3. 代码结构优化 - 在 `SEARCH_ENGINES` 配置中添加专用的环境变量字段 - 重构 Docker Compose 环境变量设置逻辑 - 改进 API 密钥验证规则,支持多种密钥格式 - 优化密钥输入和验证流程的错误处理 ## 测试项 - [x] 各搜索引擎专用 API 密钥的配置与验证 - [x] API 密钥向后兼容性 - [x] 腾讯搜索的双密钥配置流程 - [x] InternLM 模型配置 ## 相关文件 - docker/msdl/user_interaction.py - docker/msdl/utils.py - docker/msdl/translations/en.yaml - docker/msdl/translations/zh_CN.yaml - mindsearch/agent/models.py - docker/msdl/temp/.env
合并状态:已合并 合并于 2024-11-22 关闭于 2024-11-22 0 条评论