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