baidu search

agent/prompt_loader.py

@@ -138,33 +138,59 @@ async def load_system_prompt_async(config) -> str:
 
 
 
-def replace_mcp_placeholders(mcp_settings: List[Dict], dataset_dir: str, bot_id: str, dataset_ids: List[str]) -> List[Dict]:
+def replace_mcp_placeholders(mcp_settings: List[Dict], dataset_dir: str, bot_id: str, dataset_ids: List[str], shell_env: Optional[Dict[str, str]] = None) -> List[Dict]:
     """
     替换 MCP 配置中的占位符
+
+    支持的占位符来源（优先级从高到低）：
+    1. 内置变量: {dataset_dir}, {bot_id}, {dataset_ids}
+    2. shell_env 中的自定义环境变量
+    3. 系统环境变量 (os.environ)
     """
     if not mcp_settings or not isinstance(mcp_settings, list):
         return mcp_settings
 
     dataset_id_str = ','.join(dataset_ids) if dataset_ids else ''
 
+    # 构建占位符映射：系统环境变量 < shell_env < 内置变量（优先级递增）
+    import re
+    placeholders = {}
+    placeholders.update(os.environ)
+    if shell_env:
+        placeholders.update(shell_env)
+    placeholders.update({
+        'dataset_dir': dataset_dir,
+        'bot_id': bot_id,
+        'dataset_ids': dataset_id_str,
+    })
+
+    def _safe_format(s: str) -> str:
+        """安全地替换字符串中的占位符，未匹配的占位符保持原样"""
+        try:
+            def _replacer(match):
+                key = match.group(1)
+                return placeholders.get(key, match.group(0))
+            return re.sub(r'\{(\w+)\}', _replacer, s)
+        except Exception:
+            return s
+
     def replace_placeholders_in_obj(obj):
         """递归替换对象中的占位符"""
         if isinstance(obj, dict):
             for key, value in obj.items():
                 if key == 'args' and isinstance(value, list):
-                    # 特别处理 args 列表
-                    obj[key] = [item.format(dataset_dir=dataset_dir, bot_id=bot_id, dataset_ids=dataset_id_str) if isinstance(item, str) else item
+                    obj[key] = [_safe_format(item) if isinstance(item, str) else item
                                 for item in value]
                 elif isinstance(value, (dict, list)):
                     obj[key] = replace_placeholders_in_obj(value)
                 elif isinstance(value, str):
-                    obj[key] = value.format(dataset_dir=dataset_dir, bot_id=bot_id, dataset_ids=dataset_id_str)
+                    obj[key] = _safe_format(value)
         elif isinstance(obj, list):
-            return [replace_placeholders_in_obj(item) if isinstance(item, (dict, list)) else 
-                    item.format(dataset_dir=dataset_dir, bot_id=bot_id, dataset_ids=dataset_id_str) if isinstance(item, str) else item
+            return [replace_placeholders_in_obj(item) if isinstance(item, (dict, list)) else
+                    _safe_format(item) if isinstance(item, str) else item
                     for item in obj]
         return obj
-    
+
     return replace_placeholders_in_obj(mcp_settings)
 
 async def load_mcp_settings_async(config) -> List[Dict]:
@@ -269,7 +295,8 @@ async def load_mcp_settings_async(config) -> List[Dict]:
     # 替换 MCP 配置中的 {dataset_dir} 占位符
     if dataset_dir is None:
         dataset_dir = ""
-    merged_settings = replace_mcp_placeholders(merged_settings, dataset_dir, bot_id, dataset_ids)
+    shell_env = getattr(config, 'shell_env', None) or {}
+    merged_settings = replace_mcp_placeholders(merged_settings, dataset_dir, bot_id, dataset_ids, shell_env)
     return merged_settings
 
 

skills/baidu-search/.claude-plugin/plugin.json

@@ -6,7 +6,7 @@
           "transport": "http",
           "url": "https://qianfan.baidubce.com/v2/tools/web-search/mcp",
           "headers": {
-              "Authorization": "Bearer bce-v3/ALTAK-60JvEdbSLzQEtHmfheEmk/e24d067a609bf099505afdf71dc880cd5ece5dde"
+              "Authorization": "Bearer {BAIDU_API_KEY}"
           }
       }
   }

skills/ragflow-loader/.claude-plugin/plugin.json

@@ -1,6 +1,6 @@
 {
   "name": "ragflow-loader",
-  "description": "加载ragflow的rag_retrieve的mcp和提示词注入",
+  "description": "知识库检索服务",
   "hooks": {
     "PrePrompt": [
       {

skills_developing/baidu-search-skill/_meta.json

@@ -1,6 +0,0 @@
-{
-  "ownerId": "kn7akgt520t01vgs2tzx7yk6m180kt26",
-  "slug": "baidu-search",
-  "version": "1.1.2",
-  "publishedAt": 1773394754557
-}