Update main.py

main.py

@@ -21,7 +21,6 @@ app.add_middleware(
 # 读取原项目固定命名的 Secrets
 # ──────────────────────────────────────────
 
-# 原项目的 Provider Key 映射表
 PROVIDER_MAP = {
     "GOOGLE_API_KEY":        "google",
     "GROQ_API_KEY":          "groq",
@@ -38,7 +37,6 @@ PROVIDER_MAP = {
     "ZHIPU_API_KEY":         "zhipu",
 }
 
-# Provider 对应的 base_url 和支持的模型
 PROVIDER_CONFIG = {
     "google": {
         "base_url": "https://generativelanguage.googleapis.com/v1beta/openai",
@@ -133,19 +131,14 @@ PROVIDER_CONFIG = {
 
 
 def load_config():
-    """从环境变量读取所有配置"""
-
-    # ── 读取用户 API Keys（用于鉴权） ──
     raw_keys = os.getenv("API_KEYS", "")
     api_keys = set(k.strip() for k in raw_keys.split(",") if k.strip())
 
-    # ── 读取 Provider Keys ──
     providers = {}
     for env_name, provider_name in PROVIDER_MAP.items():
         key_value = os.getenv(env_name, "").strip()
         if not key_value:
-            continue  # 没配置该 Provider，跳过
-
+            continue
         cfg = PROVIDER_CONFIG.get(provider_name, {})
         providers[provider_name] = {
             "api_key":  key_value,
@@ -156,7 +149,6 @@ def load_config():
 
     logger.info(f"✅ 加载了 {len(api_keys)} 个 API Key")
     logger.info(f"✅ 加载了 {len(providers)} 个 Provider: {list(providers.keys())}")
-
     return api_keys, providers
 
 
@@ -173,19 +165,15 @@ def verify_api_key(authorization: str = Header(...)):
     return token
 
 # ──────────────────────────────────────────
-# 路由
+# ★ 用 APIRouter 统一加 /v1 前缀
 # ──────────────────────────────────────────
 
-@app.get("/health")
-async def health():
-    return {
-        "status":    "ok",
-        "keys":      len(API_KEYS),
-        "providers": list(PROVIDERS.keys()),
-    }
+from fastapi import APIRouter
+
+router = APIRouter(prefix="/v1")  # ← 关键！所有路由自动带 /v1
 
 
-@app.get("/v1/models")
+@router.get("/models")
 async def list_models(_: str = Depends(verify_api_key)):
     data = []
     for p_cfg in PROVIDERS.values():
@@ -194,7 +182,7 @@ async def list_models(_: str = Depends(verify_api_key)):
     return {"object": "list", "data": data}
 
 
-@app.post("/v1/chat/completions")
+@router.post("/chat/completions")
 async def chat_completions(
     body: dict,
     _: str = Depends(verify_api_key)
@@ -222,9 +210,18 @@ async def chat_completions(
         return resp.json()
 
 
+# ★ 把 router 注册到 app
+app.include_router(router)
+
+
+# 健康检查不需要 /v1 前缀，保持根路径
 @app.get("/health")
 async def health():
-    return {"status": "ok"}
+    return {
+        "status":    "ok",
+        "keys":      len(API_KEYS),
+        "providers": list(PROVIDERS.keys()),
+    }
 
 
 if __name__ == "__main__":