From afc68af3a67b2f936429feb3a68c1f09e03ce930 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 11 四月 2025 10:07:24 +0800
Subject: [PATCH] 智能体中心不显示自研智能体

---
 app/api/__init__.py |    8 ++++++++
 1 files changed, 8 insertions(+), 0 deletions(-)

diff --git a/app/api/__init__.py b/app/api/__init__.py
index 6cb4b05..de2282f 100644
--- a/app/api/__init__.py
+++ b/app/api/__init__.py
@@ -68,6 +68,14 @@
 def get_current_user(token: str = Depends(oauth2_scheme)):
     try:
         payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
+        expired_time = payload.get("lex")
+        if not expired_time:
+            raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED,  detail="浠ょ墝鏃犳晥鎴栧凡杩囨湡",
+            headers={"WWW-Authenticate": "Bearer"})
+        if datetime.strptime(expired_time, "%Y-%m-%d %H:%M:%S") < datetime.now():
+            raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED,  detail="绯荤粺鎺堟潈宸茶繃鏈燂紒",
+            headers={"WWW-Authenticate": "Bearer"})
+
         username: str = payload.get("sub")
         if username is None:
             raise HTTPException(

--
Gitblit v1.8.0