zishuo/gpt4free
1
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-24 13:07:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add support for additional parameters in get_models method for BlackboxPro and GeminiPro classes

Browse Source
This commit is contained in:
hlohaus
2025-12-09 19:12:58 +01:00
parent 0366acbcd1
commit 1032b165fe
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
g4f/Provider/needs_auth/BlackboxPro.py
View File

@@ -1992,7 +1992,7 @@ class BlackboxPro(AsyncGeneratorProvider, ProviderModelMixin):
return cls.fallback_models
@classmethod
def get_models(cls) -> list:
def get_models(cls, **kwargs) -> list:
"""
Returns a list of available models based on authorization status.
Authorized users get the full list of models.

2
g4f/Provider/needs_auth/GeminiPro.py
View File

@@ -42,7 +42,7 @@ class GeminiPro(AsyncGeneratorProvider, ProviderModelMixin):
]
@classmethod
def get_models(cls, api_key: str = None, api_base: str = api_base) -> list[str]:
def get_models(cls, api_key: str = None, api_base: str = api_base, **kwargs) -> list[str]:
if not api_key:
return cls.fallback_models
if not cls.models: