agent功能开发增加MCP后端
This commit is contained in:
@@ -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())
|
||||
|
||||
Reference in New Issue
Block a user