| import sys | |
| from pathlib import Path | |
| from colorama import Fore, Style | |
| sys.path.append(str(Path(__file__).parent.parent)) | |
| from g4f import BaseProvider, models, Provider | |
| logging = False | |
| def main(): | |
| providers = get_providers() | |
| failed_providers = [] | |
| for _provider in providers: | |
| if _provider.needs_auth: | |
| continue | |
| print("Provider:", _provider.__name__) | |
| result = test(_provider) | |
| print("Result:", result) | |
| if _provider.working and not result: | |
| failed_providers.append(_provider) | |
| print() | |
| if failed_providers: | |
| print(f"{Fore.RED + Style.BRIGHT}Failed providers:{Style.RESET_ALL}") | |
| for _provider in failed_providers: | |
| print(f"{Fore.RED}{_provider.__name__}") | |
| else: | |
| print(f"{Fore.GREEN + Style.BRIGHT}All providers are working") | |
| def get_providers() -> list[type[BaseProvider]]: | |
| provider_names = dir(Provider) | |
| ignore_names = [ | |
| "annotations", | |
| "base_provider", | |
| "retry_provider", | |
| "BaseProvider", | |
| "AsyncProvider", | |
| "AsyncGeneratorProvider", | |
| "RetryProvider", | |
| ] | |
| return [ | |
| getattr(Provider, provider_name) | |
| for provider_name in provider_names | |
| if not provider_name.startswith("__") and provider_name not in ignore_names | |
| ] | |
| def create_response(_provider: type[BaseProvider]) -> str: | |
| model = models.gpt_35_turbo.name if _provider.supports_gpt_35_turbo else models.default.name | |
| response = _provider.create_completion( | |
| model=model, | |
| messages=[{"role": "user", "content": "Hello, who are you? Answer in detail much as possible."}], | |
| stream=False, | |
| ) | |
| return "".join(response) | |
| def test(_provider: type[BaseProvider]) -> bool: | |
| try: | |
| response = create_response(_provider) | |
| assert type(response) is str | |
| assert len(response) > 0 | |
| return response | |
| except Exception as e: | |
| if logging: | |
| print(e) | |
| return False | |
| if __name__ == "__main__": | |
| main() | |