12.10 16

zhaoqingang

2024-12-10 08c8e8c9a4d65677de6a493446a605d70efee631

 app/service/service_token.py

@@ -1,23 +1,26 @@
from Log import logger
from app.config.config import settings
from app.config.const import BISHENG, RAGFLOW
from app.models import UserModel
from app.models.token_model import TokenModel
from app.service.auth import UserAppDao
from app.service.bisheng import BishengService
from app.service.ragflow import RagflowService


def get_bisheng_token(db, user_id: int):
    token = db.query(TokenModel).filter(TokenModel.user_id == user_id).first()
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)
    if not token:
        return None
    return token.bisheng_token
    return token.access_token


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


async def get_ragflow_new_token(db, user_id: int, app_type):