mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-15 14:51:19 -08:00
Code Issues Projects Releases Packages Wiki Activity

Fix unittests

Browse source
This commit is contained in:
hlohaus 2025-03-29 20:56:09 +01:00
parent 5bf80ec280
commit 714d02006e
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
g4f/Provider/PollinationsAI.py
View file

@ -48,7 +48,7 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
default_vision_model = default_model
text_models = [default_model]
image_models = [default_image_model]
extra_image_models = ["flux-pro", "flux-dev", "flux-schnell", "midjourney", "dall-e-3"]
extra_image_models = ["flux-pro", "flux-dev", "flux-schnell", "midjourney", "dall-e-3", "turbo"]
vision_models = [default_vision_model, "gpt-4o-mini", "o3-mini", "openai", "openai-large"]
extra_text_models = vision_models
_models_loaded = False
@ -113,10 +113,8 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
combined_text = (
cls.text_models + # Already contains the default
cls.extra_text_models +
[
model for model in original_text_models
if model not in cls.extra_text_models
]
original_text_models +
cls.vision_models
)
cls.text_models = list(dict.fromkeys(combined_text))