注册用户权限同步

zhaoqingang

2024-12-26 fe30e85ad48a5fd787940cf9d58fb87a12b0b562

 app/service/service_token.py

@@ -8,6 +8,7 @@
from app.service.bisheng import BishengService
from app.service.difyService import DifyService
from app.service.ragflow import RagflowService
from app.task.fetch_agent import update_ragflow_user_tenant


async def get_bisheng_token(db, user_id: int):
@@ -103,7 +104,7 @@
        logger.error("注册用户获取token失败！")
        return False
    user_list = db.query(UserAppModel).filter(UserAppModel.app_type==RAGFLOW).all()
    user = db.query(UserAppModel).filter(UserAppModel.userFlag==sync_user.id, UserAppModel.app_type == RAGFLOW).first()
    user = db.query(UserAppModel).filter(UserAppModel.user_id==sync_user.id, UserAppModel.app_type == RAGFLOW).first()
    if not user:
        logger.error("注册用户获取信息失败！")
        return False
@@ -116,4 +117,5 @@
                                                      user.app_id)
            await ragflow_service.add_user_tenant(token, user.app_id,
                                                  u.email,
                                                  u.app_id)
                                                  u.app_id)
    update_ragflow_user_tenant(sync_user.id)