Update openai_ondemand_adapter.py

openai_ondemand_adapter.py

@@ -8,37 +8,32 @@ import logging
 import os
 
 # ====== 读取 Huggingface Secret 配置的私有key =======
-PRIVATE_KEY = os.environ.get("PRIVATE_KEY", "")
-PRIVATE_KEY = os.environ.get("PRIVATE_KEY", "")
-SAFE_HEADERS = ["Authorization", "X-API-KEY"]
+PRIVATE_KEY = os.environ.get("PRIVATE_KEY", "114514")
+SAFE_HEADER = "X-API-KEY"
 
 # 全局接口访问权限检查
 def check_private_key():
-    # 可以在这里放宽部分接口，比如首页等
     if request.path in ["/", "/favicon.ico"]:
         return
-    key = None
-    for header in SAFE_HEADERS:
-        key = request.headers.get(header)
-        if key:
-            if header == "Authorization" and key.startswith("Bearer "):
-                key = key[len("Bearer "):].strip()
-            break
+    key = request.headers.get(SAFE_HEADER)
     if not key or key != PRIVATE_KEY:
-        return jsonify({"error": "Unauthorized, must provide correct Authorization or X-API-KEY"}), 401
+        return jsonify({"error": "Unauthorized, must provide correct X-API-KEY"}), 401
 
 # 应用所有API鉴权
 app = Flask(__name__)
 app.before_request(check_private_key)
 
 # ========== KEY池（每行一个）==========
-ONDEMAND_APIKEYS = os.environ.get("ONDEMAND_APIKEYS", "").split(",") if os.environ.get("ONDEMAND_APIKEYS") else []
-BAD_KEY_RETRY_INTERVAL = 600 # 秒
+ONDEMAND_APIKEYS = [
+    "KEY1",
+    "KEY2",
+]
+BAD_KEY_RETRY_INTERVAL = 600  # 秒
 SESSION_TIMEOUT = 600  # 对话超时时间（10分钟）
 
 # ========== OnDemand模型映射 ==========
 MODEL_MAP = {
-    "o3-mini": "predefined-openai-gpto3-mini",
+    "gpto3-mini": "predefined-openai-gpto3-mini",
     "gpt-4o": "predefined-openai-gpt4o",
     "gpt-4.1": "predefined-openai-gpt4.1",
     "gpt-4.1-mini": "predefined-openai-gpt4.1-mini",
@@ -46,7 +41,7 @@ MODEL_MAP = {
     "gpt-4o-mini": "predefined-openai-gpt4o-mini",
     "deepseek-v3": "predefined-deepseek-v3",
     "deepseek-r1": "predefined-deepseek-r1",
-    "claude-3-7-sonnet": "predefined-claude-3.7-sonnet",
+    "claude-3.7-sonnet": "predefined-claude-3.7-sonnet",
     "gemini-2.0-flash": "predefined-gemini-2.0-flash",
 }
 DEFAULT_ONDEMAND_MODEL = "predefined-openai-gpt4o"