mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-06 02:30:41 -08:00
Code Issues Projects Releases Packages Wiki Activity

Enhance MCP server tests to reflect updated tool count; improve model fetching with timeout handling in providers

Browse source
This commit is contained in:
hlohaus 2025-11-02 08:01:20 +01:00
parent 006b8c8d50
commit af56ac0c03
6 changed files with 19 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

4
g4f/Provider/template/OpenaiTemplate.py
View file

@ -31,7 +31,7 @@ class OpenaiTemplate(AsyncGeneratorProvider, ProviderModelMixin, RaiseErrorMixin
max_tokens: int = None
@classmethod
def get_models(cls, api_key: str = None, api_base: str = None) -> list[str]:
def get_models(cls, api_key: str = None, api_base: str = None, timeout: int = None) -> list[str]:
if not cls.models:
try:
if api_base is None:
@ -42,7 +42,7 @@ class OpenaiTemplate(AsyncGeneratorProvider, ProviderModelMixin, RaiseErrorMixin
api_key = AuthManager.load_api_key(cls)
if cls.models_needs_auth and not api_key:
raise MissingAuthError('Add a "api_key"')
response = requests.get(f"{api_base}/models", headers=cls.get_headers(False, api_key), verify=cls.ssl)
response = requests.get(f"{api_base}/models", headers=cls.get_headers(False, api_key), verify=cls.ssl, timeout=timeout)
raise_for_status(response)
data = response.json()
data = data.get("data", data.get("models")) if isinstance(data, dict) else data