File size: 1,611 Bytes
a4b70d9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
import unittest
from typing import Type
import asyncio
from g4f.models import __models__
from g4f.providers.base_provider import BaseProvider, ProviderModelMixin
from g4f.errors import MissingRequirementsError, MissingAuthError
class TestProviderHasModel(unittest.TestCase):
cache: dict = {}
def test_provider_has_model(self):
for model, providers in __models__.values():
for provider in providers:
if provider.needs_auth:
continue
if issubclass(provider, ProviderModelMixin):
provider.get_models() # Update models
if model.name in provider.model_aliases:
model_name = provider.model_aliases[model.name]
else:
model_name = model.get_long_name()
self.provider_has_model(provider, model_name)
def provider_has_model(self, provider: Type[BaseProvider], model: str):
if provider.__name__ not in self.cache:
try:
self.cache[provider.__name__] = provider.get_models()
except (MissingRequirementsError, MissingAuthError):
return
if self.cache[provider.__name__]:
if not model in provider.model_aliases:
self.assertIn(model, self.cache[provider.__name__], provider.__name__)
def test_all_providers_working(self):
for model, providers in __models__.values():
for provider in providers:
self.assertTrue(provider.working, f"{provider.__name__} in {model.name}") |