Update app.py

app.py

@@ -59,13 +59,14 @@ def get_model(model_name: str):
     return models[model_name]
 
 # 验证API密钥
-def verify_api_key(x_api_key: Optional[str] = Header(None)):  # 从X-API-Key头部提取
+def verify_api_key(authorization: Optional[str] = None):
     logger.info("验证API密钥")
-    logger.info(f"X-API-Key头部内容: {x_api_key}")
-    if not x_api_key:
+    logger.info(f"Authorization头部内容: {authorization}")
+    if not authorization or not authorization.startswith("Bearer "):
         logger.warning("未提供有效的API密钥格式")
         raise HTTPException(status_code=401, detail="未提供有效的API密钥")
-    if x_api_key != os.getenv("API_KEY"):
+    api_key = authorization[len("Bearer "):]
+    if api_key != os.getenv("API_KEY"):
         logger.warning("无效的API密钥")
         raise HTTPException(status_code=401, detail="无效的API密钥")
     logger.info("API密钥验证通过")
@@ -89,7 +90,7 @@ class EmbeddingResponse(BaseModel):
     model: str
     usage: dict = {"prompt_tokens": 0, "total_tokens": 0}
 
-@app.post("/embeddings", response_model=EmbeddingResponse)
+@app.post("/v1/embeddings", response_model=EmbeddingResponse)
 async def create_embedding(
     request: Request,  # 接收Request对象
     req: EmbeddingRequest,
@@ -135,7 +136,7 @@ async def create_embedding(
         raise HTTPException(status_code=500, detail=error_msg)
 
 # 健康检查接口也打印完整请求
-@app.post("/v1/embeddings")
+@app.post("/embeddings")
 async def v1_check(request: Request):
     logger.info("\n===== v1请求信息 =====")
     logger.info(f"请求方法: {request.method}")
@@ -146,7 +147,7 @@ async def v1_check(request: Request):
     logger.info("===============================\n")
     return {"status": "healthy", "models": list(MODEL_MAPPING.keys())}
 
-@app.get("/")
+@app.get("/health")
 async def health_check(request: Request):
     logger.info("\n===== 健康检查请求信息 =====")
     logger.info(f"请求方法: {request.method}")