优化ultra用户生成视频与不同的账号使用不同的 ua

src/services/flow_client.py

@@ -17,6 +17,80 @@ class FlowClient:
         self.labs_base_url = config.flow_labs_base_url  # https://labs.google/fx/api
         self.api_base_url = config.flow_api_base_url    # https://aisandbox-pa.googleapis.com/v1
         self.timeout = config.flow_timeout
+        # 缓存每个账号的 User-Agent
+        self._user_agent_cache = {}
+
+    def _generate_user_agent(self, account_id: str = None) -> str:
+        """基于账号ID生成固定的 User-Agent
+        
+        Args:
+            account_id: 账号标识（如 email 或 token_id），相同账号返回相同 UA
+            
+        Returns:
+            User-Agent 字符串
+        """
+        # 如果没有提供账号ID，生成随机UA
+        if not account_id:
+            account_id = f"random_{random.randint(1, 999999)}"
+        
+        # 如果已缓存，直接返回
+        if account_id in self._user_agent_cache:
+            return self._user_agent_cache[account_id]
+        
+        # 使用账号ID作为随机种子，确保同一账号生成相同的UA
+        import hashlib
+        seed = int(hashlib.md5(account_id.encode()).hexdigest()[:8], 16)
+        rng = random.Random(seed)
+        
+        # Chrome 版本池
+        chrome_versions = ["130.0.0.0", "131.0.0.0", "132.0.0.0", "129.0.0.0"]
+        # Firefox 版本池
+        firefox_versions = ["133.0", "132.0", "131.0", "134.0"]
+        # Safari 版本池
+        safari_versions = ["18.2", "18.1", "18.0", "17.6"]
+        # Edge 版本池
+        edge_versions = ["130.0.0.0", "131.0.0.0", "132.0.0.0"]
+
+        # 操作系统配置
+        os_configs = [
+            # Windows
+            {
+                "platform": "Windows NT 10.0; Win64; x64",
+                "browsers": [
+                    lambda r: f"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/{r.choice(chrome_versions)} Safari/537.36",
+                    lambda r: f"Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:{r.choice(firefox_versions).split('.')[0]}.0) Gecko/20100101 Firefox/{r.choice(firefox_versions)}",
+                    lambda r: f"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/{r.choice(chrome_versions)} Safari/537.36 Edg/{r.choice(edge_versions)}",
+                ]
+            },
+            # macOS
+            {
+                "platform": "Macintosh; Intel Mac OS X 10_15_7",
+                "browsers": [
+                    lambda r: f"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/{r.choice(chrome_versions)} Safari/537.36",
+                    lambda r: f"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/{r.choice(safari_versions)} Safari/605.1.15",
+                    lambda r: f"Mozilla/5.0 (Macintosh; Intel Mac OS X 14.{r.randint(0, 7)}; rv:{r.choice(firefox_versions).split('.')[0]}.0) Gecko/20100101 Firefox/{r.choice(firefox_versions)}",
+                ]
+            },
+            # Linux
+            {
+                "platform": "X11; Linux x86_64",
+                "browsers": [
+                    lambda r: f"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/{r.choice(chrome_versions)} Safari/537.36",
+                    lambda r: f"Mozilla/5.0 (X11; Linux x86_64; rv:{r.choice(firefox_versions).split('.')[0]}.0) Gecko/20100101 Firefox/{r.choice(firefox_versions)}",
+                    lambda r: f"Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:{r.choice(firefox_versions).split('.')[0]}.0) Gecko/20100101 Firefox/{r.choice(firefox_versions)}",
+                ]
+            }
+        ]
+
+        # 使用固定种子随机选择操作系统和浏览器
+        os_config = rng.choice(os_configs)
+        browser_generator = rng.choice(os_config["browsers"])
+        user_agent = browser_generator(rng)
+        
+        # 缓存结果
+        self._user_agent_cache[account_id] = user_agent
+        
+        return user_agent
 
     async def _make_request(
         self,
@@ -54,10 +128,17 @@ class FlowClient:
         if use_at and at_token:
             headers["authorization"] = f"Bearer {at_token}"
 
-        # 通用请求头
+        # 确定账号标识（优先使用 token 的前16个字符作为标识）
+        account_id = None
+        if st_token:
+            account_id = st_token[:16]  # 使用 ST 的前16个字符
+        elif at_token:
+            account_id = at_token[:16]  # 使用 AT 的前16个字符
+
+        # 通用请求头 - 基于账号生成固定的 User-Agent
         headers.update({
             "Content-Type": "application/json",
-            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
+            "User-Agent": self._generate_user_agent(account_id)
         })
 
         # Log request

src/services/generation_handler.py

@@ -644,6 +644,44 @@ class GenerationHandler:
             min_images = model_config.get("min_images", 0)
             max_images = model_config.get("max_images", 0)
 
+            # 根据账号tier自动调整模型 key
+            model_key = model_config["model_key"]
+            user_tier = token.user_paygate_tier or "PAYGATE_TIER_ONE"
+
+            # TIER_TWO 账号需要使用 ultra 版本的模型
+            if user_tier == "PAYGATE_TIER_TWO":
+                # 如果模型 key 不包含 ultra，自动添加
+                if "ultra" not in model_key:
+                    # veo_3_1_i2v_s_fast_fl -> veo_3_1_i2v_s_fast_ultra_fl
+                    # veo_3_1_t2v_fast -> veo_3_1_t2v_fast_ultra
+                    # veo_3_0_r2v_fast -> veo_3_0_r2v_fast_ultra
+                    if "_fl" in model_key:
+                        model_key = model_key.replace("_fl", "_ultra_fl")
+                    elif model_key.endswith("_fast"):
+                        model_key = model_key + "_ultra"
+                    elif "_fast_" in model_key:
+                        model_key = model_key.replace("_fast_", "_fast_ultra_")
+                    
+                    if stream:
+                        yield self._create_stream_chunk(f"TIER_TWO 账号自动切换到 ultra 模型: {model_key}\n")
+                    debug_logger.log_info(f"[VIDEO] TIER_TWO 账号，模型自动调整: {model_config['model_key']} -> {model_key}")
+
+            # TIER_ONE 账号需要使用非 ultra 版本
+            elif user_tier == "PAYGATE_TIER_ONE":
+                # 如果模型 key 包含 ultra，需要移除（避免用户误用）
+                if "ultra" in model_key:
+                    # veo_3_1_i2v_s_fast_ultra_fl -> veo_3_1_i2v_s_fast_fl
+                    # veo_3_1_t2v_fast_ultra -> veo_3_1_t2v_fast
+                    model_key = model_key.replace("_ultra_fl", "_fl").replace("_ultra", "")
+                    
+                    if stream:
+                        yield self._create_stream_chunk(f"TIER_ONE 账号自动切换到标准模型: {model_key}\n")
+                    debug_logger.log_info(f"[VIDEO] TIER_ONE 账号，模型自动调整: {model_config['model_key']} -> {model_key}")
+
+            # 更新 model_config 中的 model_key
+            model_config = dict(model_config)  # 创建副本避免修改原配置
+            model_config["model_key"] = model_key
+
             # 图片数量
             image_count = len(images) if images else 0