Update app.py

app.py

@@ -243,6 +243,12 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# 设置 API 前缀
+API_PREFIX = "/ai"
+
+# 创建一个路由器
+router = APIRouter(prefix=API_PREFIX)
+
 # 认证依赖
 def verify_api_key(authorization: str = None):
     if config.API_KEY:
@@ -294,14 +300,16 @@ class ChatCompletionRequest(BaseModel):
     temperature: Optional[float] = 0.1
     top_p: Optional[float] = 1.0
 
-@app.post(f"{config.API_PREFIX}/v1/chat/completions")
-async def chat_completions(request: ChatCompletionRequest, authorization: str = Depends(verify_api_key)):
+@router.post("/v1/chat/completions")
+async def chat_completions(request: ChatCompletionRequest):
     try:
         rules, content = process_messages(request.messages)
         return await grpc_to_pieces(request.model, content, rules, request.stream, request.temperature, request.top_p)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+app.include_router(router)
+
 def process_messages(messages):
     rules = ''
     content = ''