Spaces:
Running
Running
Upload 9 files
Browse files- src/model_tester.py +10 -2
src/model_tester.py
CHANGED
|
@@ -134,6 +134,10 @@ class ModelTester:
|
|
| 134 |
if free_only:
|
| 135 |
return self._available_free_models
|
| 136 |
return self._available_models
|
|
|
|
|
|
|
|
|
|
|
|
|
| 137 |
|
| 138 |
async def try_model_direct(
|
| 139 |
self,
|
|
@@ -188,7 +192,8 @@ class ModelTester:
|
|
| 188 |
except Exception as e:
|
| 189 |
print(f"[try_best] refresh_model_list failed: {e}")
|
| 190 |
|
| 191 |
-
|
|
|
|
| 192 |
print(f"[try_best] Found {len(available_free)} free models")
|
| 193 |
|
| 194 |
# 第二步:用关键词匹配模型
|
|
@@ -234,7 +239,10 @@ class ModelTester:
|
|
| 234 |
full_model = f"{model_id}:free" if ":free" not in model_id else model_id
|
| 235 |
print(f"[try_best] Testing model: {full_model}")
|
| 236 |
result = await self.try_model_direct(session, full_model, api_key)
|
| 237 |
-
|
|
|
|
|
|
|
|
|
|
| 238 |
return result
|
| 239 |
|
| 240 |
tasks = [try_one(m, t) for m, t in candidates]
|
|
|
|
| 134 |
if free_only:
|
| 135 |
return self._available_free_models
|
| 136 |
return self._available_models
|
| 137 |
+
|
| 138 |
+
def get_all_free_models(self) -> List[str]:
|
| 139 |
+
"""Get all free models from API list (not tested)"""
|
| 140 |
+
return self._free_models
|
| 141 |
|
| 142 |
async def try_model_direct(
|
| 143 |
self,
|
|
|
|
| 192 |
except Exception as e:
|
| 193 |
print(f"[try_best] refresh_model_list failed: {e}")
|
| 194 |
|
| 195 |
+
# 使用所有free模型,而不是已测试的
|
| 196 |
+
available_free = self.get_all_free_models()
|
| 197 |
print(f"[try_best] Found {len(available_free)} free models")
|
| 198 |
|
| 199 |
# 第二步:用关键词匹配模型
|
|
|
|
| 239 |
full_model = f"{model_id}:free" if ":free" not in model_id else model_id
|
| 240 |
print(f"[try_best] Testing model: {full_model}")
|
| 241 |
result = await self.try_model_direct(session, full_model, api_key)
|
| 242 |
+
if result:
|
| 243 |
+
print(f"[try_best] Result for {full_model}: success={result.get('success')}, error={result.get('error', 'none')}")
|
| 244 |
+
else:
|
| 245 |
+
print(f"[try_best] Result for {full_model}: None")
|
| 246 |
return result
|
| 247 |
|
| 248 |
tasks = [try_one(m, t) for m, t in candidates]
|