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

XFEstudio/gpt4free

Fix model names

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

代码差异

2 个文件 +6 -4
Modified g4f/Provider/EasyChat.py +1 -0
@@ -20,6 +20,7 @@ class EasyChat(OpenaiTemplate, AuthFileMixin):
20 20 api_endpoint = f"{api_base}/chat/completions"
21 21 working = True
22 22 active_by_default = True
23 use_model_names = True
23 24
24 25 default_model = "gpt-oss-120b-free"
25 26 model_aliases = {
Modified g4f/Provider/template/OpenaiTemplate.py +5 -4
@@ -24,6 +24,7 @@ class OpenaiTemplate(AsyncGeneratorProvider, ProviderModelMixin, RaiseErrorMixin
24 24 fallback_models = []
25 25 sort_models = True
26 26 models_needs_auth = False
27 use_model_names = False
27 28 ssl = None
28 29
29 30 @classmethod
@@ -45,11 +46,11 @@ class OpenaiTemplate(AsyncGeneratorProvider, ProviderModelMixin, RaiseErrorMixin
45 46 raise_for_status(response)
46 47 data = response.json()
47 48 data = data.get("data") if isinstance(data, dict) else data
48 cls.image_models = [model.get("id", model.get("name")) for model in data if model.get("image") or model.get("type") == "image"]
49 cls.image_models = [model.get("name") if cls.use_model_names else model.get("id", model.get("name")) for model in data if model.get("image") or model.get("type") == "image"]
49 50 cls.vision_models = cls.vision_models.copy()
50 cls.vision_models += [model.get("name", model.get("id")) for model in data if model.get("vision")]
51 cls.models = [model.get("name", model.get("id")) for model in data]
52 cls.models_count = {model.get("name", model.get("id")): len(model.get("providers", [])) for model in data if len(model.get("providers", [])) > 1}
51 cls.vision_models += [model.get("name") if cls.use_model_names else model.get("id", model.get("name")) for model in data if model.get("vision")]
52 cls.models = [model.get("name") if cls.use_model_names else model.get("id", model.get("name")) for model in data]
53 cls.models_count = {model.get("name") if cls.use_model_names else model.get("id", model.get("name")): len(model.get("providers", [])) for model in data if len(model.get("providers", [])) > 1}
53 54 if cls.sort_models:
54 55 cls.models.sort()
55 56 except Exception as e: