dzenzzz commited on
Commit
0e99d89
·
1 Parent(s): 7f50106

updates app

Browse files
Files changed (3) hide show
  1. app.py +3 -2
  2. auth.py +0 -12
  3. requirements.txt +0 -1
app.py CHANGED
@@ -19,9 +19,10 @@ async def search(q: str):
19
 
20
 
21
  @app.middleware("http")
22
- async def combined_middleware(request: Request, call_next):
23
  api_key = request.headers.get("X-API-KEY")
24
  if api_key != API_KEY:
25
- raise HTTPException(status_code=403, detail="Forbidden")
 
26
  response = await call_next(request)
27
  return response
 
19
 
20
 
21
  @app.middleware("http")
22
+ async def api_key_authentication(request: Request, call_next):
23
  api_key = request.headers.get("X-API-KEY")
24
  if api_key != API_KEY:
25
+ raise HTTPException(status_code=401, detail="Unauthorized: Invalid API Key")
26
+
27
  response = await call_next(request)
28
  return response
auth.py DELETED
@@ -1,12 +0,0 @@
1
- from config import API_KEY
2
-
3
- from fastapi import Request, HTTPException
4
- from starlette.middleware.base import BaseHTTPMiddleware
5
-
6
- class APIKeyMiddleware(BaseHTTPMiddleware):
7
- async def dispatch(self, request: Request, call_next):
8
- api_key = request.headers.get("X-API-KEY")
9
- if api_key != API_KEY:
10
- raise HTTPException(status_code=403, detail="Forbidden")
11
- response = await call_next(request)
12
- return response
 
 
 
 
 
 
 
 
 
 
 
 
 
requirements.txt CHANGED
@@ -8,5 +8,4 @@ python-dotenv
8
  qdrant-client
9
  qdrant-client[fastembed]>=1.8.2
10
  sentence-transformers
11
- fastapi-csrf-protect
12
  nh3
 
8
  qdrant-client
9
  qdrant-client[fastembed]>=1.8.2
10
  sentence-transformers
 
11
  nh3