refactor(g4f/models.py): Update provider configurations for multiple …

…models
kqlio67 · Nov 19, 2024 · 56a5e0e · 56a5e0e
1 parent 6ff94bc
commit 56a5e0e
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/g4f/models.py b/g4f/models.py
@@ -211,7 +211,7 @@ def __all__() -> list[str]:
 gemini_pro = Model(
     name          = 'gemini-pro',
     base_provider = 'Google DeepMind',
-    best_provider = IterListProvider([GeminiPro, Blackbox, AIChatFree, Liaobots])
+    best_provider = IterListProvider([Blackbox, AIChatFree, GeminiPro, Liaobots])
 )
 
 gemini_flash = Model(
@@ -347,7 +347,7 @@ def __all__() -> list[str]:
 wizardlm_2_8x22b = Model(
     name = 'wizardlm-2-8x22b',
     base_provider = 'WizardLM',
-    best_provider = IterListProvider([DeepInfraChat])
+    best_provider = DeepInfraChat
 )
 
 ### Yorickvp ###
@@ -389,7 +389,7 @@ def __all__() -> list[str]:
 sonar_online = Model(
     name = 'sonar-online',
     base_provider = 'Perplexity AI',
-    best_provider = IterListProvider([PerplexityLabs])
+    best_provider = PerplexityLabs
 )
 
 sonar_chat = Model(
@@ -450,7 +450,7 @@ def __all__() -> list[str]:
 sdxl = Model(
     name = 'sdxl',
     base_provider = 'Stability AI',
-    best_provider = IterListProvider([ReplicateHome])
+    best_provider = ReplicateHome
 
 )
 
@@ -671,4 +671,4 @@ class ModelUtils:
         'any-dark': any_dark,
     }
 
-_all_models = list(ModelUtils.convert.keys())
+_all_models = list(ModelUtils.convert.keys())