zishuo/gpt4free
1
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-24 13:07:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor model alias handling in PollinationsAI; simplify logic for setting model aliases and improve video model filtering

Browse Source
This commit is contained in:
hlohaus
2025-12-10 15:13:50 +01:00
parent 1467d7ec44
commit d90ad1d00d
1 changed files with 4 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
g4f/Provider/PollinationsAI.py
View File

@@ -127,10 +127,10 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
if isinstance(model, str) or "image" in model.get("output_modalities", []):
image_models.append(alias)
if isinstance(model, dict) and alias != model.get("name"):
cls.model_aliases[alias] = model.get("name") if isinstance(model, dict) else model
cls.model_aliases[alias] = model.get("name")
cls.image_models = image_models
cls.video_models = [get_alias(model) for model in new_image_models if "video" in model.get("output_modalities", [])]
cls.video_models = [get_alias(model) for model in new_image_models if isinstance(model, dict) and "video" in model.get("output_modalities", [])]
text_response = requests.get(cls.nectar_text_models_endpoint if api_key else cls.g4f_text_models_endpoint, timeout=timeout)
if not text_response.ok:
@@ -156,23 +156,15 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
for model in models:
alias = get_alias(model)
if alias != model.get("name"):
cls.model_aliases[alias] = model.get("name")
if alias not in cls.text_models:
cls.text_models.append(alias)
if alias != model.get("name"):
cls.model_aliases[alias] = model.get("name")
elif model.get("name") not in cls.text_models:
cls.text_models.append(model.get("name"))
cls.live += 1
cls.swap_model_aliases = {v: k for k, v in cls.model_aliases.items()}
except Exception as e:
# Save default models in case of an error
if not cls.text_models:
cls.text_models = [cls.default_model]
if not cls.image_models:
cls.image_models = [cls.default_image_model]
debug.error(f"Failed to fetch models: {e}")
finally:
if api_key:
cls._gen_models_loaded = True