qa1145 commited on
Commit
b53417f
·
verified ·
1 Parent(s): d620370

Upload 9 files

Browse files
Files changed (1) hide show
  1. 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
- available_free = self.get_available_models(free_only=True)
 
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
- print(f"[try_best] Result for {full_model}: {result.get('success') if result else 'None'}")
 
 
 
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]