From 1bfb36176c2ee7bacc981247b0a79897d89151a3 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Sun, 8 Dec 2024 04:13:09 +0100 Subject: Add more flux dev image providers --- g4f/providers/base_provider.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'g4f/providers') diff --git a/g4f/providers/base_provider.py b/g4f/providers/base_provider.py index e8a47154..0cdcde90 100644 --- a/g4f/providers/base_provider.py +++ b/g4f/providers/base_provider.py @@ -98,7 +98,7 @@ class AbstractProvider(BaseProvider): default_value = f'"{param.default}"' if isinstance(param.default, str) else param.default args += f" = {default_value}" if param.default is not Parameter.empty else "" args += "," - + return f"g4f.Provider.{cls.__name__} supports: ({args}\n)" class AsyncProvider(AbstractProvider): @@ -240,6 +240,7 @@ class ProviderModelMixin: models: list[str] = [] model_aliases: dict[str, str] = {} image_models: list = None + last_model: str = None @classmethod def get_models(cls) -> list[str]: @@ -255,5 +256,6 @@ class ProviderModelMixin: model = cls.model_aliases[model] elif model not in cls.get_models() and cls.models: raise ModelNotSupportedError(f"Model is not supported: {model} in: {cls.__name__}") + cls.last_model = model debug.last_model = model return model \ No newline at end of file -- cgit v1.2.3