Upload 9 files

src/model_tester.py

@@ -181,11 +181,13 @@ class ModelTester:
         keyword: str,
         api_key: str
     ) -> Optional[Dict[str, Any]]:
-        available = self.get_available_models()
+        # 如果没有预扫描的列表，直接从API获取
         available_free = self.get_available_models(free_only=True)
+        available = self.get_available_models()
         
         candidates = []
         
+        # 优先 free 模型
         if available_free:
             if keyword:
                 matched = [m for m in available_free if keyword.lower() in m.lower()]
@@ -194,23 +196,35 @@ class ModelTester:
             if not candidates:
                 candidates.extend([(m, "free_random") for m in available_free[:10]])
         
-        if available:
-            if keyword:
-                matched = [m for m in available if keyword.lower() in m.lower()]
-                if matched:
-                    candidates.extend([(m, "matched") for m in matched])
-            if not any(c[1].startswith("free") for c in candidates):
-                if available_free:
-                    candidates.extend([(m, "free_fallback") for m in available_free[:5]])
-                else:
-                    candidates.extend([(m, "random") for m in available[:10]])
+        # 如果没有 free，添加付费模型
+        if not candidates and available:
+            candidates.extend([(m, "paid") for m in available[:10]])
+        
+        # 如果还是没有候选，实时从API获取
+        if not candidates:
+            try:
+                _, free_models = self.refresh_model_list()
+                if free_models:
+                    candidates.extend([(m, "free_api") for m in free_models[:20]])
+            except:
+                pass
         
-        for model_id, match_type in candidates:
-            full_model = f"{model_id}:free" if ":free" not in model_id else model_id
-            result = await self.try_model_direct(session, full_model, api_key)
-            if result and result.get("success"):
-                result["method"] = f"list_{match_type}"
-                return result
+        # 并发尝试所有候选模型
+        if candidates:
+            semaphore = asyncio.Semaphore(5)
+            
+            async def try_one(model_id, match_type):
+                async with semaphore:
+                    full_model = f"{model_id}:free" if ":free" not in model_id else model_id
+                    return await self.try_model_direct(session, full_model, api_key)
+            
+            tasks = [try_one(m, t) for m, t in candidates]
+            results = await asyncio.gather(*tasks, return_exceptions=True)
+            
+            for i, result in enumerate(results):
+                if isinstance(result, dict) and result.get("success"):
+                    result["method"] = f"list_{candidates[i][1]}"
+                    return result
         
         return {
             "success": False,