From ec9df59828ce6c244e155650a88c99538ddafb91 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Mon, 16 Dec 2024 01:59:30 +0100 Subject: Add get_models to GeminiPro provider --- g4f/providers/base_provider.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'g4f/providers/base_provider.py') diff --git a/g4f/providers/base_provider.py b/g4f/providers/base_provider.py index 0cdcde90..e2c356e3 100644 --- a/g4f/providers/base_provider.py +++ b/g4f/providers/base_provider.py @@ -243,19 +243,20 @@ class ProviderModelMixin: last_model: str = None @classmethod - def get_models(cls) -> list[str]: + def get_models(cls, **kwargs) -> list[str]: if not cls.models and cls.default_model is not None: return [cls.default_model] return cls.models @classmethod - def get_model(cls, model: str) -> str: + def get_model(cls, model: str, **kwargs) -> str: if not model and cls.default_model is not None: model = cls.default_model elif model in cls.model_aliases: 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__}") + else: + if model not in cls.get_models(**kwargs) 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