130 lines
4.4 KiB
Python
130 lines
4.4 KiB
Python
"""从 Hermes 配置文件中读取各搜索源所需密钥"""
|
||
|
||
import os
|
||
import re
|
||
import yaml
|
||
|
||
HERMES_CONFIG = '/root/.hermes/config.yaml'
|
||
HERMES_ENV = '/root/.hermes/.env'
|
||
SEARCH_HUB_CONFIG = '/root/agentspace/projects/search-hub/config.yaml'
|
||
|
||
|
||
def load_config():
|
||
"""加载所有搜索源配置，返回 dict"""
|
||
# 从 hermes config 读取
|
||
with open(HERMES_CONFIG, 'r') as f:
|
||
config = yaml.safe_load(f)
|
||
|
||
# 读取 .env
|
||
env_vars = {}
|
||
if os.path.exists(HERMES_ENV):
|
||
with open(HERMES_ENV, 'r') as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if line and not line.startswith('#'):
|
||
if '=' in line:
|
||
k, v = line.split('=', 1)
|
||
env_vars[k.strip()] = v.strip()
|
||
|
||
def resolve_api_key(raw_key):
|
||
if not raw_key:
|
||
return None
|
||
m = re.match(r'\$\{(\w+)\}', str(raw_key))
|
||
if m:
|
||
var_name = m.group(1)
|
||
return env_vars.get(var_name) or os.environ.get(var_name)
|
||
return raw_key
|
||
|
||
# === Tavily ===
|
||
tavily_cfg = config.get('search', {}).get('tavily', {})
|
||
tavily_api_key = resolve_api_key(tavily_cfg.get('api_key'))
|
||
tavily_base_url = tavily_cfg.get('base_url', 'https://api.tavily.com')
|
||
tavily_max_results = tavily_cfg.get('default_max_results', 10)
|
||
tavily_depth = tavily_cfg.get('default_search_depth', 'basic')
|
||
|
||
# === OpenCodeZen ===
|
||
opencodezen_key = None
|
||
opencodezen_base_url = 'https://opencode.ai/zen/go/v1'
|
||
opencodezen_model = 'deepseek-v4-flash'
|
||
|
||
for provider in config.get('custom_providers', []):
|
||
if provider.get('name') == 'OpenCodeZen':
|
||
opencodezen_key = resolve_api_key(provider.get('api_key'))
|
||
opencodezen_base_url = provider.get('base_url', opencodezen_base_url)
|
||
opencodezen_model = provider.get('model', opencodezen_model)
|
||
break
|
||
|
||
if not opencodezen_key:
|
||
opencodezen_key = env_vars.get('OPENCODE_ZEN_API_KEY')
|
||
|
||
# === Search Hub 自身配置（可选，可覆盖 Hermes 配置）===
|
||
hub_config = {}
|
||
if os.path.exists(SEARCH_HUB_CONFIG):
|
||
with open(SEARCH_HUB_CONFIG, 'r') as f:
|
||
hub_config = yaml.safe_load(f) or {}
|
||
|
||
# 应用本地配置覆盖
|
||
local_sources = hub_config.get('sources', {})
|
||
|
||
# Tavily 覆盖
|
||
if 'tavily' in local_sources:
|
||
tavily_local = local_sources['tavily']
|
||
if tavily_local.get('api_key'):
|
||
tavily_api_key = tavily_local['api_key']
|
||
if tavily_local.get('base_url'):
|
||
tavily_base_url = tavily_local['base_url']
|
||
|
||
# OpenCodeZen/AI 覆盖
|
||
if 'ai' in local_sources:
|
||
ai_local = local_sources['ai']
|
||
if ai_local.get('api_key'):
|
||
opencodezen_key = ai_local['api_key']
|
||
if ai_local.get('base_url'):
|
||
opencodezen_base_url = ai_local['base_url']
|
||
if ai_local.get('model'):
|
||
opencodezen_model = ai_local['model']
|
||
|
||
# 百度配置（来自本地 config.yaml）
|
||
baidu_api_key = ''
|
||
baidu_intelligent_url = 'https://qianfan.baidubce.com/v2/ai_search/chat/completions'
|
||
baidu_web_search_url = 'https://qianfan.baidubce.com/v2/ai_search/web_search'
|
||
|
||
if 'baidu' in local_sources:
|
||
baidu_local = local_sources['baidu']
|
||
if baidu_local.get('api_key'):
|
||
baidu_api_key = baidu_local['api_key']
|
||
if baidu_local.get('intelligent_url'):
|
||
baidu_intelligent_url = baidu_local['intelligent_url']
|
||
if baidu_local.get('web_search_url'):
|
||
baidu_web_search_url = baidu_local['web_search_url']
|
||
|
||
# === SearXNG ===
|
||
searxng_base_url = 'http://localhost:8888'
|
||
if 'searxng' in local_sources:
|
||
searxng_local = local_sources['searxng']
|
||
if searxng_local.get('base_url'):
|
||
searxng_base_url = searxng_local['base_url']
|
||
|
||
return {
|
||
'tavily': {
|
||
'api_key': tavily_api_key,
|
||
'base_url': tavily_base_url,
|
||
'max_results': tavily_max_results,
|
||
'depth': tavily_depth,
|
||
},
|
||
'opencodezen': {
|
||
'api_key': opencodezen_key,
|
||
'base_url': opencodezen_base_url,
|
||
'model': opencodezen_model,
|
||
},
|
||
'hub': hub_config,
|
||
'baidu': {
|
||
'api_key': baidu_api_key,
|
||
'intelligent_url': baidu_intelligent_url,
|
||
'web_search_url': baidu_web_search_url,
|
||
},
|
||
'searxng': {
|
||
'base_url': searxng_base_url,
|
||
},
|
||
}
|