Moonfanz commited on
Commit
2a297e1
·
verified ·
1 Parent(s): b843443

Update app/main.py

Browse files
Files changed (1) hide show
  1. 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("/hf/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,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("/hf/v1/chat/completions", response_model=ChatCompletionResponse)
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