ADDED AUTH

auth.py

@@ -0,0 +1,13 @@
+from fastapi import HTTPException, Security
+from fastapi.security import api_key
+
+api_key_header = api_key.APIKeyHeader(name="X-API-KEY")
+
+
+async def validate_api_key(key: str = Security(api_key_header)):
+    if key != settings.API_KEY:
+        raise HTTPException(
+            status_code=401, detail="Unauthorized - API Key is wrong"
+        )
+    return None
+

main.py

@@ -1,4 +1,5 @@
-from fastapi import FastAPI, HTTPException, UploadFile, File
+from fastapi import FastAPI, HTTPException, UploadFile, File,Request,Depends,status
+from fastapi.security import OAuth2PasswordBearer
 from pydantic import BaseModel, Json
 from uuid import uuid4, UUID
 from typing import Optional
@@ -38,7 +39,34 @@ if index_name not in existing_indexes:
 
 index = pc.Index(index_name)
 
-app = FastAPI()
+
+
+api_keys = [os.environ.get("FASTAPI_API_KEY")]
+
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")  # use token authentication
+
+
+def api_key_auth(api_key: str = Depends(oauth2_scheme)):
+    if api_key not in api_keys:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Forbidden"
+        )
+
+app = FastAPI(dependencies=[Depends(api_key_auth)])
+
+# FASTAPI_KEY_NAME = os.environ.get("FASTAPI_KEY_NAME")
+# FASTAPI_API_KEY = os.environ.get("FASTAPI_API_KEY")
+
+
+# @app.middleware("http")
+# async def api_key_middleware(request: Request, call_next):
+#     if request.url.path not in ["/","/docs","/openapi.json"]:
+#         api_key = request.headers.get(FASTAPI_KEY_NAME)
+#         if api_key != FASTAPI_API_KEY:
+#             raise HTTPException(status_code=403, detail="invalid API key :/")
+#     response = await call_next(request)
+#     return response
 
 class StyleWriter(BaseModel):
     style: str

rag.py

@@ -149,7 +149,6 @@ def prompt_reformatting(prompt:str,context,query:str,style="formel",tonality="ne
     
     docs_names = []
     for chunk in context:
-        print(chunk.metadata)
         chunk_name = chunk.metadata["filename"]
         if chunk_name not in docs_names:
             docs_names.append(chunk_name)