返回提交历史
Modified
etc/unittest/models.py
+1
-1
Modified
g4f/models.py
+13
-30
XFEstudio/gpt4free
Fix unittests
a06d651d
代码差异
2 个文件
+14
-31
@@ -36,7 +36,7 @@ class TestProviderHasModel(unittest.TestCase):
36
36
def provider_has_model(self, provider: Type[BaseProvider], model: str):
37
37
if provider.__name__ not in self.cache:
38
38
try:
39
self.cache[provider.__name__] = provider.get_models()
39
self.cache[provider.__name__] = list(provider.get_models())
40
40
except (MissingRequirementsError, MissingAuthError):
41
41
return
42
42
if self.cache[provider.__name__]:
@@ -133,7 +133,6 @@ default_vision = VisionModel(
133
133
"Together",
134
134
"HuggingSpace",
135
135
"GeminiPro",
136
"HuggingFaceAPI",
137
136
"Ollama",
138
137
"OpenaiAccount",
139
138
"Gemini",
@@ -246,7 +245,7 @@ gpt_oss_120b = Model(
246
245
name = 'gpt-oss-120b',
247
246
long_name = 'openai/gpt-oss-120b',
248
247
base_provider = 'OpenAI',
249
best_provider = IterListProvider(["Together", "HuggingFace", "OpenRouter", "Groq"])
248
best_provider = IterListProvider(["Together", "OpenRouter", "Groq"])
250
249
)
251
250
252
251
# dall-e
@@ -330,7 +329,7 @@ llama_3_2_3b = Model(
330
329
llama_3_2_11b = VisionModel(
331
330
name = "llama-3.2-11b",
332
331
base_provider = "Meta Llama",
333
best_provider = IterListProvider(["Together", "HuggingChat", "HuggingFace"])
332
best_provider = None
334
333
)
335
334
336
335
llama_3_2_90b = Model(
@@ -343,7 +342,7 @@ llama_3_2_90b = Model(
343
342
llama_3_3_70b = Model(
344
343
name = "llama-3.3-70b",
345
344
base_provider = "Meta Llama",
346
best_provider = IterListProvider(["Together", "HuggingChat", "HuggingFace"])
345
best_provider = None
347
346
)
348
347
349
348
# llama-4
@@ -375,7 +374,7 @@ mixtral_8x7b = Model(
375
374
mistral_nemo = Model(
376
375
name = "mistral-nemo",
377
376
base_provider = "Mistral AI",
378
best_provider = IterListProvider(["HuggingChat", "HuggingFace"])
377
best_provider = None
379
378
)
380
379
381
380
mistral_small_24b = Model(
@@ -402,7 +401,7 @@ hermes_2_dpo = Model(
402
401
phi_3_5_mini = Model(
403
402
name = "phi-3.5-mini",
404
403
base_provider = "Microsoft",
405
best_provider = "HuggingChat"
404
best_provider = None
406
405
)
407
406
408
407
@@ -505,13 +504,13 @@ gemma_3n_e4b = Model(
505
504
command_r = Model(
506
505
name = 'command-r',
507
506
base_provider = 'CohereForAI',
508
best_provider = "HuggingSpace"
507
best_provider = None
509
508
)
510
509
511
510
command_r_plus = Model(
512
511
name = 'command-r-plus',
513
512
base_provider = 'CohereForAI',
514
best_provider = IterListProvider(["HuggingSpace", "HuggingChat"])
513
best_provider = None
515
514
)
516
515
517
516
command_r7b = Model(
@@ -544,7 +543,7 @@ qwen_2_72b = Model(
544
543
qwen_2_vl_7b = VisionModel(
545
544
name = "qwen-2-vl-7b",
546
545
base_provider = 'Qwen',
547
best_provider = "HuggingFaceAPI"
546
best_provider = None
548
547
)
549
548
550
549
qwen_2_vl_72b = VisionModel(
@@ -617,7 +616,7 @@ deepseek_v3 = Model(
617
616
deepseek_r1 = Model(
618
617
name = 'deepseek-r1',
619
618
base_provider = 'DeepSeek',
620
best_provider = IterListProvider(["PollinationsAI", "Together", "HuggingChat", "HuggingFace"])
619
best_provider = IterListProvider(["PollinationsAI", "Together"])
621
620
)
622
621
623
622
@@ -669,7 +668,7 @@ grok_3_r1 = Model(
669
668
kimi = Model(
670
669
name = 'kimi-k2',
671
670
base_provider = 'kimi.com',
672
best_provider = IterListProvider(["HuggingFace", "Groq"]),
671
best_provider = IterListProvider(["Groq"]),
673
672
long_name = "moonshotai/Kimi-K2-Instruct"
674
673
)
675
674
@@ -708,7 +707,7 @@ r1_1776 = Model(
708
707
nemotron_70b = Model(
709
708
name = 'nemotron-70b',
710
709
base_provider = 'Nvidia',
711
best_provider = IterListProvider(["Together", "HuggingChat", "HuggingFace"])
710
best_provider = IterListProvider(["Together", "HuggingChat"])
712
711
)
713
712
714
713
### Cognitive Computations ###
@@ -754,13 +753,13 @@ flux_pro = ImageModel(
754
753
flux_dev = ImageModel(
755
754
name = 'flux-dev',
756
755
base_provider = 'Black Forest Labs',
757
best_provider = IterListProvider(["PollinationsImage", "HuggingSpace", "Together", "HuggingChat", "HuggingFace"])
756
best_provider = IterListProvider(["PollinationsImage", "HuggingSpace", "Together", "HuggingChat"])
758
757
)
759
758
760
759
flux_schnell = ImageModel(
761
760
name = 'flux-schnell',
762
761
base_provider = 'Black Forest Labs',
763
best_provider = IterListProvider(["PollinationsImage", "Together", "HuggingChat", "HuggingFace"])
762
best_provider = IterListProvider(["PollinationsImage", "Together", "HuggingChat"])
764
763
)
765
764
766
765
flux_redux = ImageModel(
@@ -823,22 +822,6 @@ ModelRegistry._aliases["gemini"] = "gemini-2.0"
823
822
# Fill the convert dictionary
824
823
ModelUtils.convert = ModelRegistry.all_models()
825
824
826
# Demo models configuration
827
demo_models = {
828
llama_3_2_11b.name: [llama_3_2_11b, ["HuggingChat"]],
829
qwen_2_vl_7b.name: [qwen_2_vl_7b, ["HuggingFaceAPI"]],
830
deepseek_r1.name: [deepseek_r1, ["HuggingFace", "PollinationsAI"]],
831
command_r.name: [command_r, ["HuggingSpace"]],
832
command_r_plus.name: [command_r_plus, ["HuggingSpace"]],
833
command_r7b.name: [command_r7b, ["HuggingSpace"]],
834
qwen_2_5_coder_32b.name: [qwen_2_5_coder_32b, ["HuggingFace"]],
835
qwq_32b.name: [qwq_32b, ["HuggingFace"]],
836
llama_3_3_70b.name: [llama_3_3_70b, ["HuggingFace"]],
837
sd_3_5_large.name: [sd_3_5_large, ["HuggingSpace", "HuggingFace"]],
838
flux_dev.name: [flux_dev, ["PollinationsImage", "HuggingFace", "HuggingSpace"]],
839
flux_schnell.name: [flux_schnell, ["PollinationsImage", "HuggingFace", "HuggingSpace"]],
840
}
841
842
825
# Create a list of all models and their providers
843
826
def _get_best_providers(model: Model) -> List:
844
827
"""Get list of working providers for a model"""