shuiyu1123
78fea7ccdd
- 重构 core/ 底层 (7文件): app/runner/engine/parser/history/safety/display - FC原生调用为主，prompt-JSON保底 - 移除所有工具返回截断 (ai_search/git_tools/cmd_executor/browser_auto/network_tools/sub_agent/tool_search) - 删除无用文件 (旧服务器/WebUI/彩蛋/启动器等21个文件) - 清理 constants.py 死代码 - 版本号更新为 7.0.0
109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
import os
|
|
import json
|
|
import logging
|
|
from colorama import Fore, Style
|
|
|
|
# 配置文件路径
|
|
CONFIG_FILE = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "config.json")
|
|
|
|
|
|
def load_config():
|
|
"""加载配置文件"""
|
|
try:
|
|
with open(CONFIG_FILE, 'r', encoding='utf-8') as f:
|
|
return json.load(f)
|
|
except FileNotFoundError:
|
|
return {"api": {"engines": {}}, "system": {}, "plugins": {}}
|
|
except json.JSONDecodeError:
|
|
return {"api": {"engines": {}}, "system": {}, "plugins": {}}
|
|
|
|
|
|
def save_config(config):
|
|
"""保存配置文件"""
|
|
try:
|
|
with open(CONFIG_FILE, 'w', encoding='utf-8') as f:
|
|
json.dump(config, f, ensure_ascii=False, indent=2)
|
|
except Exception as e:
|
|
print(f"保存配置失败: {e}")
|
|
|
|
|
|
def get_system_config(key, default=None):
|
|
"""获取系统配置"""
|
|
config = load_config()
|
|
return config.get("system", {}).get(key, default)
|
|
|
|
|
|
def set_system_config(key, value):
|
|
"""设置系统配置"""
|
|
config = load_config()
|
|
if "system" not in config:
|
|
config["system"] = {}
|
|
config["system"][key] = value
|
|
save_config(config)
|
|
|
|
|
|
# ── 引擎配置档案管理 ──
|
|
# 存储结构: config["engine_profiles"][engine_name] = [ {name, base_url, api_key, model}, ... ]
|
|
|
|
def get_profiles(engine_name):
|
|
"""获取指定引擎的所有配置档案"""
|
|
config = load_config()
|
|
return config.get("engine_profiles", {}).get(engine_name, [])
|
|
|
|
|
|
def save_profile(engine_name, profile):
|
|
"""保存/更新一个配置档案（按 name 去重，已存在则覆盖）"""
|
|
config = load_config()
|
|
if "engine_profiles" not in config:
|
|
config["engine_profiles"] = {}
|
|
profiles = config["engine_profiles"].setdefault(engine_name, [])
|
|
# 按 name 去重
|
|
for i, p in enumerate(profiles):
|
|
if p.get("name") == profile.get("name"):
|
|
profiles[i] = profile
|
|
break
|
|
else:
|
|
profiles.append(profile)
|
|
config["engine_profiles"][engine_name] = profiles
|
|
save_config(config)
|
|
|
|
|
|
def delete_profile(engine_name, profile_name):
|
|
"""删除指定引擎的一个配置档案"""
|
|
config = load_config()
|
|
profiles = config.get("engine_profiles", {}).get(engine_name, [])
|
|
profiles = [p for p in profiles if p.get("name") != profile_name]
|
|
if "engine_profiles" not in config:
|
|
config["engine_profiles"] = {}
|
|
config["engine_profiles"][engine_name] = profiles
|
|
save_config(config)
|
|
|
|
|
|
def apply_profile(engine_name, profile):
|
|
"""将档案应用到 config.json 的 api.engines 节点（下次启动或重载时生效）"""
|
|
config = load_config()
|
|
if "api" not in config:
|
|
config["api"] = {}
|
|
if "engines" not in config["api"]:
|
|
config["api"]["engines"] = {}
|
|
engine_cfg = config["api"]["engines"].setdefault(engine_name, {})
|
|
if "base_url" in profile:
|
|
engine_cfg["base_url"] = profile["base_url"]
|
|
if "api_key" in profile:
|
|
engine_cfg["api_key"] = profile["api_key"]
|
|
if "model" in profile:
|
|
engine_cfg["model"] = profile["model"]
|
|
save_config(config)
|
|
|
|
|
|
# 配置日志
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
handlers=[
|
|
logging.FileHandler('xiaoli_cli.log', encoding='utf-8'),
|
|
logging.StreamHandler() # 同时输出到控制台
|
|
]
|
|
)
|
|
logger = logging.getLogger(__name__)
|