新增加标签功能

zhaoqingang

2024-12-11 a791022ff1311e1fb76930c398d6ff91036d0456

 app/service/service_token.py

@@ -9,19 +9,24 @@


async def get_bisheng_token(db, user_id: int):
    # token = db.query(TokenModel).filter(TokenModel.user_id == user_id).first()
    token = await UserAppDao.get_data_by_id(user_id, BISHENG)
    token = await UserAppDao(db).get_data_by_id(user_id, BISHENG)
    if not token:
        return None
    return token.access_token

        token = db.query(TokenModel).filter(TokenModel.user_id == user_id).first()
        if not token:
            return None
    else:
        return token.access_token
    return token.bisheng_token

async def get_ragflow_token(db, user_id: int):
    token = await UserAppDao.get_data_by_id(user_id, RAGFLOW)
    token = await UserAppDao(db).get_data_by_id(user_id, RAGFLOW)
    if not token:
        return None
    return token.access_token

        token = db.query(TokenModel).filter(TokenModel.user_id == user_id).first()
        if not token:
            return None
    else:
        return token.access_token
    return token.ragflow_token

async def get_ragflow_new_token(db, user_id: int, app_type):
    user = db.query(UserModel).filter(UserModel.id == user_id).first()