Update app/main.py
Browse files- app/main.py +4 -2
app/main.py
CHANGED
|
@@ -151,7 +151,7 @@ async def startup_event():
|
|
| 151 |
log_msg = format_log_message('INFO', "Available models loaded.")
|
| 152 |
logger.info(log_msg)
|
| 153 |
|
| 154 |
-
@app.get("/
|
| 155 |
def list_models():
|
| 156 |
log_msg = format_log_message('INFO', "Received request to list models", extra={'request_type': 'list_models', 'status_code': 200})
|
| 157 |
logger.info(log_msg)
|
|
@@ -161,6 +161,8 @@ def list_models():
|
|
| 161 |
async def verify_password(request: Request):
|
| 162 |
if PASSWORD:
|
| 163 |
auth_header = request.headers.get("Authorization")
|
|
|
|
|
|
|
| 164 |
if not auth_header or not auth_header.startswith("Bearer "):
|
| 165 |
raise HTTPException(
|
| 166 |
status_code=401, detail="Unauthorized: Missing or invalid token")
|
|
@@ -310,7 +312,7 @@ async def process_request(chat_request: ChatCompletionRequest, http_request: Req
|
|
| 310 |
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=msg)
|
| 311 |
|
| 312 |
|
| 313 |
-
@app.post("/
|
| 314 |
async def chat_completions(request: ChatCompletionRequest, http_request: Request, _: None = Depends(verify_password)):
|
| 315 |
return await process_request(request, http_request, "stream" if request.stream else "non-stream")
|
| 316 |
|
|
|
|
| 151 |
log_msg = format_log_message('INFO', "Available models loaded.")
|
| 152 |
logger.info(log_msg)
|
| 153 |
|
| 154 |
+
@app.get("/v1/models", response_model=ModelList)
|
| 155 |
def list_models():
|
| 156 |
log_msg = format_log_message('INFO', "Received request to list models", extra={'request_type': 'list_models', 'status_code': 200})
|
| 157 |
logger.info(log_msg)
|
|
|
|
| 161 |
async def verify_password(request: Request):
|
| 162 |
if PASSWORD:
|
| 163 |
auth_header = request.headers.get("Authorization")
|
| 164 |
+
if not auth_header: logger.info("no auth header")
|
| 165 |
+
if not auth_header.startswith("Bearer "): logger.info("not start with bearer")
|
| 166 |
if not auth_header or not auth_header.startswith("Bearer "):
|
| 167 |
raise HTTPException(
|
| 168 |
status_code=401, detail="Unauthorized: Missing or invalid token")
|
|
|
|
| 312 |
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=msg)
|
| 313 |
|
| 314 |
|
| 315 |
+
@app.post("/v1/chat/completions", response_model=ChatCompletionResponse)
|
| 316 |
async def chat_completions(request: ChatCompletionRequest, http_request: Request, _: None = Depends(verify_password)):
|
| 317 |
return await process_request(request, http_request, "stream" if request.stream else "non-stream")
|
| 318 |
|