From c795d4b39ce86ad9b502044d96df9392c0c7dca7 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期一, 25 十一月 2024 17:39:56 +0800
Subject: [PATCH] ragflow和bisheng的会话时间统一

---
 app/service/user.py |    6 ++----
 1 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/app/service/user.py b/app/service/user.py
index 6cf7c1e..f5cc7ea 100644
--- a/app/service/user.py
+++ b/app/service/user.py
@@ -8,7 +8,7 @@
 from app.service.bisheng import BishengService
 from app.service.common.app_register import AppRegister
 from app.service.ragflow import RagflowService
-from app.service.service_token import get_ragflow_token, get_bisheng_token
+from app.service.service_token import get_ragflow_token, get_bisheng_token, get_ragflow_new_token
 
 
 async def get_user_list(db, page_index: int, page_size: int, keyword: str, role_key:str, user_id):
@@ -104,11 +104,9 @@
     try:
         user = db.query(UserModel).filter(UserModel.id == user_id).first()
         pwd = user.decrypted_password()
-        print(pwd)
-        print(new_password)
         for app_type in AppRegister(db).get_app():
             if app_type == AgentType.RAGFLOW:
-                token = get_ragflow_token(db, user_id)
+                token = await get_ragflow_new_token(db, user_id, app_type)
                 ragflow_service = RagflowService(settings.fwr_base_url)
                 await ragflow_service.set_user_password(token, pwd, new_password)
             elif app_type == AgentType.BISHENG:

--
Gitblit v1.8.0