zhaoqingang
2024-12-18 ae88193c5153aa04226a8307e1294a9c75d312bb
app/models/token_model.py
@@ -4,6 +4,7 @@
from sqlalchemy import Column, Integer, DateTime, Text
from sqlalchemy.orm import Session
from app.config.const import RAGFLOW
# from Log import logger
from app.models.base_model import Base
@@ -65,6 +66,8 @@
        if db_token:
            # 记录存在,进行更新
            db_token.token = access_token
            db_token.ragflow_token = token.get(RAGFLOW, "")
            # db_token.token = access_token
            # for k, v in token.items():
            #     setattr(db_token, k.replace("app", "token"), v)
        else:
@@ -72,6 +75,7 @@
            db_token = TokenModel(
                user_id=user_id,
                token=access_token,
                ragflow_token = token.get(RAGFLOW, "")
            )
            # for k, v in token.items():
            #     setattr(db_token, k.replace("app", "token"), v)