agent功能开发增加MCP后端

This commit is contained in:
2025-11-07 22:12:23 +08:00
parent c886d78ff6
commit 30788648af
2 changed files with 27 additions and 7 deletions

View File

@@ -1549,13 +1549,23 @@ async def agent_chat(request: AgentChatRequest):
logger.info(f"Agent chat: {request.message} (user: {request.user_id})")
# ==================== 权限检查 ====================
# 仅允许 max 用户使用
if request.user_id != "max":
# 仅允许 max 用户使用(支持多种格式:字符串 "max"、数字 1、或 nickname 为 "max"
is_max_user = (
request.user_id == "max" or
request.user_id == "1" or
request.user_id == 1 or
request.user_nickname == "max"
)
if not is_max_user:
logger.warning(f"权限检查失败 - user_id: {request.user_id}, nickname: {request.user_nickname}")
raise HTTPException(
status_code=403,
detail="很抱歉,「价小前投研」功能目前仅对特定用户开放。如需使用,请联系管理员。"
)
logger.info(f"权限检查通过 - user_id: {request.user_id}, nickname: {request.user_nickname}")
# ==================== 会话管理 ====================
# 如果没有提供 session_id创建新会话
session_id = request.session_id or str(uuid.uuid4())