From f95f801f35aa201cbaffd7d881c07edc9398b570 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期一, 03 三月 2025 16:03:51 +0800
Subject: [PATCH] 增加外接知识库中转接口
---
app/api/__init__.py | 12 +++++++++---
1 files changed, 9 insertions(+), 3 deletions(-)
diff --git a/app/api/__init__.py b/app/api/__init__.py
index 2f325c1..2a679e8 100644
--- a/app/api/__init__.py
+++ b/app/api/__init__.py
@@ -2,8 +2,8 @@
from urllib.parse import urlencode
import jwt
-from cryptography.fernet import Fernet
-from fastapi import FastAPI, Depends, HTTPException
+# from cryptography.fernet import Fernet
+from fastapi import FastAPI, Depends, HTTPException, Header
from fastapi.security import OAuth2PasswordBearer
from passlib.context import CryptContext
from pydantic import BaseModel
@@ -11,7 +11,7 @@
from starlette.websockets import WebSocket, WebSocketDisconnect
from Log import logger
-from app.models.app_model import AppRegisterModel
+# from app.models.app_model import AppRegisterModel
from app.models.user_model import UserModel
from app.service.auth import SECRET_KEY, ALGORITHM
from app.config.config import settings
@@ -116,6 +116,12 @@
# 璁板綍寮傚父淇℃伅锛屼絾缁х画澶勭悊鍏朵粬鏂囦欢
print(f"Error processing file URL: {e}")
+def get_api_key(authorization: str = Header(...)):
+ if not authorization.startswith("Bearer "):
+ raise HTTPException(status_code=401, detail="Invalid Authorization header format.")
+ return authorization.split(" ")[1]
+
+
if __name__=="__main__":
--
Gitblit v1.8.0