From fda90aa8f5ccea22ced86937b1ac24c55835537a Mon Sep 17 00:00:00 2001 From: kqlio67 Date: Sat, 19 Oct 2024 12:56:43 +0300 Subject: refactor(g4f/gui/server/api.py): streamline model retrieval logic --- g4f/gui/server/api.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) (limited to 'g4f/gui') diff --git a/g4f/gui/server/api.py b/g4f/gui/server/api.py index 3da0fe17..64b84767 100644 --- a/g4f/gui/server/api.py +++ b/g4f/gui/server/api.py @@ -43,13 +43,8 @@ class Api(): provider: ProviderType = __map__[provider] if issubclass(provider, ProviderModelMixin): return [{"model": model, "default": model == provider.default_model} for model in provider.get_models()] - elif provider.supports_gpt_35_turbo or provider.supports_gpt_4: - return [ - *([{"model": "gpt-4", "default": not provider.supports_gpt_4}] if provider.supports_gpt_4 else []), - *([{"model": "gpt-3.5-turbo", "default": not provider.supports_gpt_4}] if provider.supports_gpt_35_turbo else []) - ] else: - return []; + return [] @staticmethod def get_image_models() -> list[dict]: @@ -245,4 +240,4 @@ def get_error_message(exception: Exception) -> str: provider = get_last_provider() if provider is None: return message - return f"{provider.__name__}: {message}" \ No newline at end of file + return f"{provider.__name__}: {message}" -- cgit v1.2.3