Upload 9 files

Dockerfile

@@ -33,6 +33,10 @@ RUN git clone https://github.com/Theyka/Turnstile-Solver.git /app/Turnstile-Solv
 # 为 Turnstile-Solver 安装依赖
 RUN pip install --no-cache-dir -r /app/Turnstile-Solver/requirements.txt
 
+# 预下载 browserforge 所需的数据文件
+# 这可以防止在运行时因权限问题导致下载失败
+RUN python -c "from browserforge.fingerprints import FingerprintGenerator; from browserforge.headers import HeaderGenerator; print('Pre-downloaded browserforge data files.')"
+
 # 为求解器安装浏览器
 # 这将下载并修补 chromium
 RUN python -m patchright install chromium

main.py

@@ -4,6 +4,7 @@ import time
 import uuid
 import threading
 from typing import Any, Dict, List, Optional, TypedDict, Union
+from urllib.parse import urlparse
 
 import requests
 import websocket
@@ -34,9 +35,21 @@ DEBUG_MODE = os.environ.get("DEBUG_MODE", "false").lower() == "true"
 REQUEST_TIMEOUT = 120.0  # 请求超时时间，秒
 HTTP_PROXY = os.environ.get("HTTP_PROXY")
 PROXIES = {"http": HTTP_PROXY, "https": HTTP_PROXY} if HTTP_PROXY else None
-
-if PROXIES:
-    print(f"Using HTTP proxy: {HTTP_PROXY}")
+WEBSOCKET_PROXY_CONFIG = {}
+
+if HTTP_PROXY:
+    print(f"Using HTTP/HTTPS proxy: {HTTP_PROXY}")
+    parsed_proxy_url = urlparse(HTTP_PROXY)
+    WEBSOCKET_PROXY_CONFIG = {
+        "http_proxy_host": parsed_proxy_url.hostname,
+        "http_proxy_port": parsed_proxy_url.port,
+    }
+    if parsed_proxy_url.username and parsed_proxy_url.password:
+        WEBSOCKET_PROXY_CONFIG["http_proxy_auth"] = (
+            parsed_proxy_url.username,
+            parsed_proxy_url.password,
+        )
+    print(f"Using WebSocket proxy config: {WEBSOCKET_PROXY_CONFIG}")
 
 
 # Pydantic Models
@@ -462,7 +475,11 @@ def tenbin_stream_generator(model: str, prompt: str, session_id: str, execution_
     ws = None
     try:
         log_debug("Connecting to WebSocket...")
-        ws = websocket.create_connection(url, header=headers)
+        ws = websocket.create_connection(
+            url, 
+            header=headers,
+            **WEBSOCKET_PROXY_CONFIG
+        )
         ws.send(json.dumps({"type": "connection_init"}))
         init_response = ws.recv()
         log_debug(f"WebSocket init response: {init_response}")