feat: 添加HTTP代理支持

- 添加HTTP_PROXY环境变量配置
- 在所有requests请求中使用代理（留空则不使用）
- 支持replicate.py、app.py和auth_flow.py中的所有HTTP请求

.env.example

@@ -4,4 +4,8 @@
 OPENAI_KEYS=""
 
 # 出错次数阈值，超过此值自动禁用账号
-MAX_ERROR_COUNT=100
+MAX_ERROR_COUNT=100
+
+# HTTP代理设置（留空不使用代理）
+# 例如：HTTP_PROXY="http://127.0.0.1:7890"
+HTTP_PROXY=""

app.py

@@ -193,6 +193,12 @@ class ChatCompletionRequest(BaseModel):
 OIDC_BASE = "https://oidc.us-east-1.amazonaws.com"
 TOKEN_URL = f"{OIDC_BASE}/token"
 
+def _get_proxies() -> Optional[Dict[str, str]]:
+    proxy = os.getenv("HTTP_PROXY", "").strip()
+    if proxy:
+        return {"http": proxy, "https": proxy}
+    return None
+
 def _oidc_headers() -> Dict[str, str]:
     return {
         "content-type": "application/json",
@@ -220,7 +226,7 @@ def refresh_access_token_in_db(account_id: str) -> Dict[str, Any]:
         }
 
         try:
-            r = requests.post(TOKEN_URL, headers=_oidc_headers(), json=payload, timeout=(15, 60))
+            r = requests.post(TOKEN_URL, headers=_oidc_headers(), json=payload, timeout=(15, 60), proxies=_get_proxies())
             r.raise_for_status()
             data = r.json()
             new_access = data.get("accessToken")

auth_flow.py

@@ -1,10 +1,17 @@
 import json
 import time
 import uuid
+import os
 from typing import Dict, Tuple, Optional
 
 import requests
 
+def _get_proxies() -> Optional[Dict[str, str]]:
+    proxy = os.getenv("HTTP_PROXY", "").strip()
+    if proxy:
+        return {"http": proxy, "https": proxy}
+    return None
+
 # OIDC endpoints and constants (aligned with v1/auth_client.py)
 OIDC_BASE = "https://oidc.us-east-1.amazonaws.com"
 REGISTER_URL = f"{OIDC_BASE}/client/register"
@@ -31,7 +38,7 @@ def post_json(url: str, payload: Dict) -> requests.Response:
     # Keep JSON order and mimic body closely to v1
     payload_str = json.dumps(payload, ensure_ascii=False)
     headers = make_headers()
-    resp = requests.post(url, headers=headers, data=payload_str, timeout=(15, 60))
+    resp = requests.post(url, headers=headers, data=payload_str, timeout=(15, 60), proxies=_get_proxies())
     return resp
 
 

replicate.py

@@ -1,5 +1,6 @@
 import json
 import uuid
+import os
 from pathlib import Path
 from typing import Dict, Optional, Tuple, Iterator, List, Generator, Any
 import struct
@@ -15,6 +16,12 @@ class StreamTracker:
                 self.has_content = True
             yield item
 
+def _get_proxies() -> Optional[Dict[str, str]]:
+    proxy = os.getenv("HTTP_PROXY", "").strip()
+    if proxy:
+        return {"http": proxy, "https": proxy}
+    return None
+
 BASE_DIR = Path(__file__).resolve().parent
 TEMPLATE_PATH = BASE_DIR / "templates" / "streaming_request.json"
 
@@ -198,7 +205,8 @@ def send_chat_request(access_token: str, messages: List[Dict[str, Any]], model:
     payload_str = json.dumps(body_json, ensure_ascii=False)
     headers = _merge_headers(headers_from_log, access_token)
     session = requests.Session()
-    resp = session.post(url, headers=headers, data=payload_str, stream=True, timeout=timeout)
+    proxies = _get_proxies()
+    resp = session.post(url, headers=headers, data=payload_str, stream=True, timeout=timeout, proxies=proxies)
     if resp.status_code >= 400:
         try:
             err = resp.text