XFE Git
XFE Studio Git
Git 首页 全局搜索
XFE 主站 文档 NuGet
公开
关注 0 Fork 0 Star 1
返回提交历史

XFEstudio/gpt4free

Refactor get_alias method to prioritize first alias if available

ad9069f3
hlohaus <983577+hlohaus@users.noreply.github.com>
提交于

代码差异

1 个文件 +4 -1
Modified g4f/Provider/PollinationsAI.py +4 -1
@@ -102,7 +102,10 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
102 102 @classmethod
103 103 def get_models(cls, **kwargs):
104 104 def get_alias(model: dict) -> str:
105 return model.get("aliases", model.get("name")).replace("-instruct", "").replace("qwen-", "qwen").replace("qwen", "qwen-")
105 alias = model.get("name")
106 if (model.get("aliases")):
107 alias = model.get("aliases")[0]
108 return alias.replace("-instruct", "").replace("qwen-", "qwen").replace("qwen", "qwen-")
106 109 if not cls._models_loaded:
107 110 try:
108 111 # Update of image models