Update main.py

main.py

@@ -187,6 +187,7 @@ async def read_root():
     return HTMLResponse(content=html_content, status_code=200)
 
 # 聊天完成处理（保留原有逻辑，未修改）
+@app.post("/ai/v1/chat/completions")
 @app.post("/v1/chat/completions")
 async def chat_completions(request: Request, background_tasks: BackgroundTasks):
     """
@@ -472,6 +473,7 @@ async def chat_completions(request: Request, background_tasks: BackgroundTasks):
                 raise HTTPException(status_code=500, detail=f"内部服务器错误: {str(exc)}")
 
 # 图像生成处理
+@app.post("/ai/v1/images/generations")
 @app.post("/v1/images/generations")
 async def images_generations(request: Request):
     """
@@ -702,7 +704,8 @@ async def images_generations(request: Request):
 
     # 返回响应
     return JSONResponse(content=response_json, status_code=200)
-
+    
+@app.get("/ai/v1/models", response_class=JSONResponse)
 @app.get("/v1/models", response_class=JSONResponse)
 async def get_models():
     models_data = {