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

Swap model aliases for consistency and enhance model retrieval logic

Browse source
This commit is contained in:
hlohaus 2025-09-21 12:48:55 +02:00
parent bb02eeb481
commit 9bed8da354
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
g4f/Provider/PollinationsAI.py
View file

@ -88,7 +88,7 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
vision_models = [default_vision_model]
_models_loaded = False
model_aliases = {
"openai-fast": "gpt-4.1-nano",
"gpt-4.1-nano": "openai-fast",
"llama-4-scout": "llamascout",
"deepseek-r1": "deepseek-reasoning",
"sdxl-turbo": "turbo",
@ -99,6 +99,7 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
"flux": "flux",
"flux-kontext": "kontext",
}
swap_model_aliases = {v: k for k, v in model_aliases.items()}
@classmethod
def get_models(cls, **kwargs):
@ -106,8 +107,8 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
alias = model.get("name")
if (model.get("aliases")):
alias = model.get("aliases")[0]
elif alias in cls.model_aliases:
alias = cls.model_aliases[alias]
elif alias in cls.swap_model_aliases:
alias = cls.swap_model_aliases[alias]
return alias.replace("-instruct", "").replace("qwen-", "qwen").replace("qwen", "qwen-")
if not cls._models_loaded:
try:
@ -383,6 +384,9 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
if cls.login_url in str(item):
raise MissingAuthError(item)
raise item
else:
finished += 1
yield Reasoning(label=f"Image {finished}/{n} failed after {time.time() - start:.2f}s: {item}")
else:
finished += 1
yield Reasoning(label=f"Image {finished}/{n} generated in {time.time() - start:.2f}s")