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

XFEstudio/gpt4free

refactor(g4f/gui/server/api.py): streamline model retrieval logic

fda90aa8
kqlio67 <kqlio67@users.noreply.github.com>
提交于

代码差异

1 个文件 +2 -7
Modified g4f/gui/server/api.py +2 -7
@@ -43,13 +43,8 @@ class Api():
43 43 provider: ProviderType = __map__[provider]
44 44 if issubclass(provider, ProviderModelMixin):
45 45 return [{"model": model, "default": model == provider.default_model} for model in provider.get_models()]
46 elif provider.supports_gpt_35_turbo or provider.supports_gpt_4:
47 return [
48 *([{"model": "gpt-4", "default": not provider.supports_gpt_4}] if provider.supports_gpt_4 else []),
49 *([{"model": "gpt-3.5-turbo", "default": not provider.supports_gpt_4}] if provider.supports_gpt_35_turbo else [])
50 ]
51 46 else:
52 return [];
47 return []
53 48
54 49 @staticmethod
55 50 def get_image_models() -> list[dict]:
@@ -245,4 +240,4 @@ def get_error_message(exception: Exception) -> str:
245 240 provider = get_last_provider()
246 241 if provider is None:
247 242 return message
248 return f"{provider.__name__}: {message}"
243 return f"{provider.__name__}: {message}"