zhangxiao
2024-10-23 df971a07c4c4b9f6e5e1ba6b966102f41b54c717
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from app.models.token_model import TokenModel
 
 
def get_bisheng_token(db, user_id: int):
    token = db.query(TokenModel).filter(TokenModel.user_id == user_id).first()
    if not token:
        return None
    return token.bisheng_token
 
 
def get_ragflow_token(db, user_id: int):
    token = db.query(TokenModel).filter(TokenModel.user_id == user_id).first()
    if not token:
        return None
    return token.ragflow_token