From a791022ff1311e1fb76930c398d6ff91036d0456 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 11 十二月 2024 17:57:52 +0800
Subject: [PATCH] 新增加标签功能
---
app/service/service_token.py | 23 ++++++++++++++---------
1 files changed, 14 insertions(+), 9 deletions(-)
diff --git a/app/service/service_token.py b/app/service/service_token.py
index 8fffc99..215353b 100644
--- a/app/service/service_token.py
+++ b/app/service/service_token.py
@@ -9,19 +9,24 @@
async def get_bisheng_token(db, user_id: int):
- # token = db.query(TokenModel).filter(TokenModel.user_id == user_id).first()
- token = await UserAppDao.get_data_by_id(user_id, BISHENG)
+ token = await UserAppDao(db).get_data_by_id(user_id, BISHENG)
if not token:
- return None
- return token.access_token
-
+ token = db.query(TokenModel).filter(TokenModel.user_id == user_id).first()
+ if not token:
+ return None
+ else:
+ return token.access_token
+ return token.bisheng_token
async def get_ragflow_token(db, user_id: int):
- token = await UserAppDao.get_data_by_id(user_id, RAGFLOW)
+ token = await UserAppDao(db).get_data_by_id(user_id, RAGFLOW)
if not token:
- return None
- return token.access_token
-
+ token = db.query(TokenModel).filter(TokenModel.user_id == user_id).first()
+ if not token:
+ return None
+ else:
+ return token.access_token
+ return token.ragflow_token
async def get_ragflow_new_token(db, user_id: int, app_type):
user = db.query(UserModel).filter(UserModel.id == user_id).first()
--
Gitblit v1.8.0