mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-06 02:30:41 -08:00
Code Issues Projects Releases Packages Wiki Activity

Fix unittests

Browse source
This commit is contained in:
hlohaus 2025-08-08 02:20:19 +02:00
parent 5b65101a2a
commit 2aa40bb8a8
4 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

3
etc/unittest/models.py
View file

@ -19,7 +19,7 @@ class TestProviderHasModel(unittest.TestCase):
if model.name in provider.model_aliases:
model_name = provider.model_aliases[model.name]
else:
model_name = model.name
model_name = model.get_long_name()
self.provider_has_model(provider, model_name)
def provider_has_model(self, provider: Type[BaseProvider], model: str):
@ -29,6 +29,7 @@ class TestProviderHasModel(unittest.TestCase):
except (MissingRequirementsError, MissingAuthError):
return
if self.cache[provider.__name__]:
if not model in provider.model_aliases:
self.assertIn(model, self.cache[provider.__name__], provider.__name__)
def test_all_providers_working(self):

1
g4f/Provider/Kimi.py
View file

@ -16,6 +16,7 @@ class Kimi(AsyncAuthedProvider, ProviderModelMixin):
active_by_default = True
default_model = "kimi-k2"
models = [default_model]
model_aliases = {"moonshotai/Kimi-K2-Instruct": default_model}
@classmethod
async def on_auth_async(cls, proxy: str = None, **kwargs) -> AsyncIterator:

3
g4f/Provider/PollinationsAI.py
View file

@ -88,7 +88,8 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
vision_models = [default_vision_model]
_models_loaded = False
model_aliases = {
"gpt-4.1-nano": "openai",
"gpt-4.1-mini": "openai",
"gpt-4.1-nano": "openai-fast",
"gpt-4.1": "openai-large",
"o4-mini": "openai-reasoning",
"qwen-2.5-coder-32b": "qwen-coder",

6
g4f/models.py
View file

@ -197,20 +197,20 @@ default_vision = VisionModel(
gpt_4 = Model(
name = 'gpt-4',
base_provider = 'OpenAI',
best_provider = IterListProvider([Blackbox, PollinationsAI, Copilot, Yqcloud, WeWordle, OpenaiChat])
best_provider = IterListProvider([Blackbox, Copilot, Yqcloud, WeWordle, OpenaiChat])
)
# gpt-4o
gpt_4o = VisionModel(
name = 'gpt-4o',
base_provider = 'OpenAI',
best_provider = IterListProvider([Blackbox, PollinationsAI, OpenaiChat])
best_provider = IterListProvider([Blackbox, OpenaiChat])
)
gpt_4o_mini = Model(
name = 'gpt-4o-mini',
base_provider = 'OpenAI',
best_provider = IterListProvider([Blackbox, PollinationsAI, Chatai, OIVSCodeSer2, Startnest, OpenaiChat])
best_provider = IterListProvider([Blackbox, Chatai, OIVSCodeSer2, Startnest, OpenaiChat])
)
gpt_4o_mini_audio = AudioModel(