v0.3: Account pool, rotation, WebUI, adapter refactor

Author: snake-aabb-wtf

.env.example

@@ -4,80 +4,45 @@
 HOST=0.0.0.0
 PORT=1800
 API_KEY=sk-web2api-placeholder
+ROTATION_STRATEGY=least-recently-used
+MAX_ERRORS_BEFORE_DISABLE=3
 
 # ============================================
-# 模型列表（逗号分隔）
-# 运行 config_tool.py 选择 HAR 文件自动生成 Profile
+# Profiles — 模型配置（不含凭证）
+# 运行 config_tool.py 或通过 WebUI 添加账号
 # ============================================
-MODEL_NAMES=gemini-3.5-flash,gemini-3.5-flash-adv,gemini-3.1-flash-lite,gemini-3.1-pro,gemini-3.1-pro-ext
+PROFILES=gemini-3.5-flash,gemini-3.5-flash-adv,gemini-3.1-pro,gemini-3.1-pro-ext,gemini-3.1-flash-lite,gemini-3.1-flash-lite-adv
 DEFAULT_MODEL=gemini-3.5-flash
 
-# ============================================
-# Profile: gemini-3.5-flash
-# model_family=1 (Flash), thinking_mode=1 (标准)
-# ============================================
 MODEL_FAMILY_gemini-3.5-flash=1
 THINKING_MODE_gemini-3.5-flash=1
-F_SID_gemini-3.5-flash=
-AT_gemini-3.5-flash=
-SN_PARAM_gemini-3.5-flash=
-BL_PARAM_gemini-3.5-flash=
-HL_gemini-3.5-flash=zh-CN
-UUID_gemini-3.5-flash=
-HASH_gemini-3.5-flash=
 
-# ============================================
-# Profile: gemini-3.5-flash-adv
-# model_family=1 (Flash), thinking_mode=2 (进阶思考)
-# ============================================
 MODEL_FAMILY_gemini-3.5-flash-adv=1
 THINKING_MODE_gemini-3.5-flash-adv=2
-F_SID_gemini-3.5-flash-adv=
-AT_gemini-3.5-flash-adv=
-SN_PARAM_gemini-3.5-flash-adv=
-BL_PARAM_gemini-3.5-flash-adv=
-HL_gemini-3.5-flash-adv=zh-CN
-UUID_gemini-3.5-flash-adv=
-HASH_gemini-3.5-flash-adv=
-
-# ============================================
-# Profile: gemini-3.1-flash-lite
-# model_family=6 (Flash Lite), thinking_mode=1 (标准)
-# ============================================
-MODEL_FAMILY_gemini-3.1-flash-lite=6
-THINKING_MODE_gemini-3.1-flash-lite=1
-F_SID_gemini-3.1-flash-lite=
-AT_gemini-3.1-flash-lite=
-SN_PARAM_gemini-3.1-flash-lite=
-BL_PARAM_gemini-3.1-flash-lite=
-HL_gemini-3.1-flash-lite=zh-CN
-UUID_gemini-3.1-flash-lite=
-HASH_gemini-3.1-flash-lite=
 
-# ============================================
-# Profile: gemini-3.1-pro
-# model_family=3 (Pro), thinking_mode=1 (标准思考)
-# ============================================
 MODEL_FAMILY_gemini-3.1-pro=3
 THINKING_MODE_gemini-3.1-pro=1
-F_SID_gemini-3.1-pro=
-AT_gemini-3.1-pro=
-SN_PARAM_gemini-3.1-pro=
-BL_PARAM_gemini-3.1-pro=
-HL_gemini-3.1-pro=zh-CN
-UUID_gemini-3.1-pro=
-HASH_gemini-3.1-pro=
 
-# ============================================
-# Profile: gemini-3.1-pro-ext
-# model_family=3 (Pro), thinking_mode=2 (扩展思考)
-# ============================================
 MODEL_FAMILY_gemini-3.1-pro-ext=3
 THINKING_MODE_gemini-3.1-pro-ext=2
-F_SID_gemini-3.1-pro-ext=
-AT_gemini-3.1-pro-ext=
-SN_PARAM_gemini-3.1-pro-ext=
-BL_PARAM_gemini-3.1-pro-ext=
-HL_gemini-3.1-pro-ext=zh-CN
-UUID_gemini-3.1-pro-ext=
-HASH_gemini-3.1-pro-ext=
+
+MODEL_FAMILY_gemini-3.1-flash-lite=6
+THINKING_MODE_gemini-3.1-flash-lite=1
+
+MODEL_FAMILY_gemini-3.1-flash-lite-adv=6
+THINKING_MODE_gemini-3.1-flash-lite-adv=2
+
+# ============================================
+# Accounts — 账号凭证
+# 运行 config_tool.py 或通过 WebUI 添加
+# 格式: ACCOUNT_<name>_<FIELD>=<value>
+# ============================================
+# ACCOUNT_myaccount_F_SID=xxx
+# ACCOUNT_myaccount_AT=xxx
+# ACCOUNT_myaccount_SN_PARAM=xxx
+# ACCOUNT_myaccount_BL_PARAM=boq_assistant-bard-web-server_20260525.09_p0
+# ACCOUNT_myaccount_HL=zh-CN
+# ACCOUNT_myaccount_UUID=
+# ACCOUNT_myaccount_HASH=
+# ACCOUNT_myaccount_ENABLED=true
+# ACCOUNT_myaccount_MODELS=gemini-3.5-flash,gemini-3.1-flash-lite

account_pool.py

@@ -0,0 +1,126 @@
+import json
+import time
+import os
+from dataclasses import dataclass, field
+from typing import Optional
+
+
+@dataclass
+class Account:
+    name: str
+    f_sid: str = ""
+    at: str = ""
+    sn_param: str = ""
+    bl_param: str = ""
+    hl: str = "zh-CN"
+    session_uuid: str = ""
+    request_hash: str = ""
+    headers: dict = field(default_factory=dict)
+    enabled: bool = True
+    error_count: int = 0
+    last_used: float = 0.0
+    bound_models: list = field(default_factory=list)
+
+
+class AccountPool:
+    def __init__(self, strategy: str = "least-recently-used", max_errors: int = 3):
+        self.accounts: list[Account] = []
+        self.strategy = strategy
+        self.max_errors = max_errors
+        self._stats = {"total_requests": 0, "success": 0, "failures": 0, "retries": 0}
+
+    def add(self, account: Account):
+        self.accounts.append(account)
+
+    def select(self, model_name: str) -> Optional[Account]:
+        candidates = [
+            a for a in self.accounts
+            if a.enabled and a.error_count < self.max_errors and model_name in a.bound_models
+        ]
+        if not candidates:
+            return None
+        if self.strategy == "least-recently-used":
+            return min(candidates, key=lambda a: a.last_used)
+        return candidates[0]
+
+    def record_success(self, account: Account):
+        account.error_count = 0
+        account.last_used = time.time()
+        self._stats["success"] += 1
+
+    def record_failure(self, account: Account):
+        account.error_count += 1
+        self._stats["failures"] += 1
+
+    def record_retry(self):
+        self._stats["retries"] += 1
+
+    def record_request(self):
+        self._stats["total_requests"] += 1
+
+    def stats(self) -> dict:
+        return {
+            **self._stats,
+            "accounts_total": len(self.accounts),
+            "accounts_enabled": sum(1 for a in self.accounts if a.enabled),
+            "accounts_disabled": sum(1 for a in self.accounts if not a.enabled),
+            "accounts_exhausted": sum(1 for a in self.accounts if a.error_count >= self.max_errors),
+            "accounts": [
+                {
+                    "name": a.name,
+                    "enabled": a.enabled,
+                    "error_count": a.error_count,
+                    "last_used": a.last_used,
+                    "bound_models": a.bound_models,
+                }
+                for a in self.accounts
+            ],
+        }
+
+    def load_from_env(self):
+        prefix = "ACCOUNT_"
+        accounts_data = {}
+        for key, val in sorted(os.environ.items()):
+            if not key.startswith(prefix):
+                continue
+            parts = key[len(prefix):].split("_", 1)
+            if len(parts) != 2:
+                continue
+            acct_name, field = parts
+            if acct_name not in accounts_data:
+                accounts_data[acct_name] = {"name": acct_name}
+            if field == "ENABLED":
+                accounts_data[acct_name]["enabled"] = val.lower() in ("true", "1", "yes")
+            elif field == "HEADER":
+                hname, hval = val.split("=", 1) if "=" in val else (val, "")
+                if "headers" not in accounts_data[acct_name]:
+                    accounts_data[acct_name]["headers"] = {}
+                accounts_data[acct_name]["headers"][hname] = hval
+            elif field == "MODELS":
+                accounts_data[acct_name]["bound_models"] = [m.strip() for m in val.split(",") if m.strip()]
+            else:
+                accounts_data[acct_name][field.lower()] = val
+
+        for data in accounts_data.values():
+            self.add(Account(**data))
+
+    def get_account(self, name: str) -> Optional[Account]:
+        for a in self.accounts:
+            if a.name == name:
+                return a
+        return None
+
+    def to_env_lines(self) -> list[str]:
+        lines = []
+        for a in self.accounts:
+            lines.append(f"ACCOUNT_{a.name}_F_SID={a.f_sid}")
+            lines.append(f"ACCOUNT_{a.name}_AT={a.at}")
+            lines.append(f"ACCOUNT_{a.name}_SN_PARAM={a.sn_param}")
+            lines.append(f"ACCOUNT_{a.name}_BL_PARAM={a.bl_param}")
+            lines.append(f"ACCOUNT_{a.name}_HL={a.hl}")
+            lines.append(f"ACCOUNT_{a.name}_UUID={a.session_uuid}")
+            lines.append(f"ACCOUNT_{a.name}_HASH={a.request_hash}")
+            lines.append(f"ACCOUNT_{a.name}_ENABLED={'true' if a.enabled else 'false'}")
+            lines.append(f"ACCOUNT_{a.name}_MODELS={','.join(a.bound_models)}")
+            lines.append("")
+        return lines