From ae88193c5153aa04226a8307e1294a9c75d312bb Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期三, 18 十二月 2024 18:30:03 +0800 Subject: [PATCH] add dialog sss --- app/models/token_model.py | 16 ++++++++++------ 1 files changed, 10 insertions(+), 6 deletions(-) diff --git a/app/models/token_model.py b/app/models/token_model.py index aa37d12..736720c 100644 --- a/app/models/token_model.py +++ b/app/models/token_model.py @@ -4,7 +4,8 @@ from sqlalchemy import Column, Integer, DateTime, Text from sqlalchemy.orm import Session -from Log import logger +from app.config.const import RAGFLOW +# from Log import logger from app.models.base_model import Base @@ -65,16 +66,19 @@ if db_token: # 璁板綍瀛樺湪锛岃繘琛屾洿鏂� db_token.token = access_token - for k, v in token.items(): - setattr(db_token, k.replace("app", "token"), v) + 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: # 璁板綍涓嶅瓨鍦紝杩涜鎻掑叆 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) + # for k, v in token.items(): + # setattr(db_token, k.replace("app", "token"), v) db.add(db_token) # 鎻愪氦浜嬪姟 @@ -82,7 +86,7 @@ db.refresh(db_token) except Exception as e: - logger.error(e) + # logger.error(e) # 寮傚父澶勭悊 db.rollback() # 鍥炴粴浜嬪姟 -- Gitblit v1.8.0