6307deb701
- 添加用户权限检查功能，支持配置允许使用的用户列表 - 实现会话管理功能，包括会话创建、关闭、列表和切换 - 新增审计日志模块，记录所有交互信息 - 改进WebSocket连接，增加自动重连机制 - 添加健康检查端点，包含Claude服务可用性测试 - 实现会话持久化功能，重启后恢复会话状态 - 增加命令行功能支持，包括/new、/list、/close等命令 - 优化消息处理流程，支持直接传递模式
32 lines
942 B
Python
32 lines
942 B
Python
import yaml
|
|
from pathlib import Path
|
|
from typing import List
|
|
|
|
_CONFIG_PATH = Path(__file__).parent / "keyring.yaml"
|
|
|
|
|
|
def _load() -> dict:
|
|
with open(_CONFIG_PATH, "r", encoding="utf-8") as f:
|
|
return yaml.safe_load(f) or {}
|
|
|
|
|
|
_cfg = _load()
|
|
|
|
FEISHU_APP_ID: str = _cfg["FEISHU_APP_ID"]
|
|
FEISHU_APP_SECRET: str = _cfg["FEISHU_APP_SECRET"]
|
|
OPENAI_BASE_URL: str = _cfg["OPENAI_BASE_URL"]
|
|
OPENAI_API_KEY: str = _cfg["OPENAI_API_KEY"]
|
|
OPENAI_MODEL: str = _cfg.get("OPENAI_MODEL", "glm-4.7")
|
|
WORKING_DIR: Path = Path(_cfg.get("WORKING_DIR", Path.home())).expanduser().resolve()
|
|
|
|
ALLOWED_OPEN_IDS: List[str] = _cfg.get("ALLOWED_OPEN_IDS", [])
|
|
if ALLOWED_OPEN_IDS and not isinstance(ALLOWED_OPEN_IDS, list):
|
|
ALLOWED_OPEN_IDS = [str(ALLOWED_OPEN_IDS)]
|
|
|
|
|
|
def is_user_allowed(open_id: str) -> bool:
|
|
"""Check if a user is allowed to use the bot."""
|
|
if not ALLOWED_OPEN_IDS:
|
|
return True
|
|
return open_id in ALLOWED_OPEN_IDS
|