mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-10-10 02:30:14 +08:00
335c971f6a
* Add multiple images support * Add multiple images support in gui * Support multiple images in legacy client and in the api Fix some model names in provider model list * Fix unittests * Add vision and providers docs
29 lines
1.2 KiB
Python
29 lines
1.2 KiB
Python
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.IsolatedAsyncioTestCase):
|
|
cache: dict = {}
|
|
|
|
async def test_provider_has_model(self):
|
|
for model, providers in __models__.values():
|
|
for provider in providers:
|
|
if issubclass(provider, ProviderModelMixin):
|
|
if model.name in provider.model_aliases:
|
|
model_name = provider.model_aliases[model.name]
|
|
else:
|
|
model_name = model.name
|
|
await asyncio.wait_for(self.provider_has_model(provider, model_name), 10)
|
|
|
|
async 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__]:
|
|
self.assertIn(model, self.cache[provider.__name__], provider.__name__) |