mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-18 18:10:37 +08:00
1faeee3732
Resolved conflicts: - openai_source.py: keep dev version with abort_signal filtering - customizer.ts: keep dev version with viewMode functionality - useSessions.ts: keep dev version with pendingSessionId handling - platformUtils.js: keep dev version with correct tutorial links - AddNewPlatform.vue: keep dev version with correct docs link - FullLayout.vue: keep dev version with viewMode-based logic - VerticalHeader.vue: keep dev version with viewMode-based logic
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
"""MCP configuration management."""
|
|
|
|
import json
|
|
import os
|
|
|
|
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
|
|
|
DEFAULT_MCP_CONFIG = {"mcpServers": {}}
|
|
|
|
|
|
def get_mcp_config_path() -> str:
|
|
"""Get the path to the MCP configuration file."""
|
|
data_dir = get_astrbot_data_path()
|
|
return os.path.join(data_dir, "mcp_server.json")
|
|
|
|
|
|
def load_mcp_config() -> dict:
|
|
"""Load MCP configuration from file.
|
|
|
|
Returns:
|
|
MCP configuration dict. If file doesn't exist, returns default config.
|
|
|
|
"""
|
|
config_path = get_mcp_config_path()
|
|
if not os.path.exists(config_path):
|
|
# Create default config if not exists
|
|
os.makedirs(os.path.dirname(config_path), exist_ok=True)
|
|
with open(config_path, "w", encoding="utf-8") as f:
|
|
json.dump(DEFAULT_MCP_CONFIG, f, ensure_ascii=False, indent=4)
|
|
return DEFAULT_MCP_CONFIG
|
|
|
|
try:
|
|
with open(config_path, encoding="utf-8") as f:
|
|
return json.load(f)
|
|
except Exception:
|
|
return DEFAULT_MCP_CONFIG
|
|
|
|
|
|
def save_mcp_config(config: dict) -> bool:
|
|
"""Save MCP configuration to file.
|
|
|
|
Args:
|
|
config: MCP configuration dict to save.
|
|
|
|
Returns:
|
|
True if successful, False otherwise.
|
|
|
|
"""
|
|
config_path = get_mcp_config_path()
|
|
try:
|
|
with open(config_path, "w", encoding="utf-8") as f:
|
|
json.dump(config, f, ensure_ascii=False, indent=4)
|
|
return True
|
|
except Exception:
|
|
return False
|