import sys from pathlib import Path from colorama import Fore, Style sys.path.append(str(Path(__file__).parent.parent)) from g4f import Provider, ProviderType, models from g4f.Provider import __providers__ 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[ProviderType]: return [ provider for provider in __providers__ if provider.__name__ not in dir(Provider.deprecated) and provider.__name__ not in dir(Provider.unfinished) ] def create_response(provider: ProviderType) -> str: response = provider.create_completion( model=models.default.name, messages=[{"role": "user", "content": "Hello, who are you? Answer in detail much as possible."}], stream=False, ) return "".join(response) def test(provider: ProviderType) -> bool: try: response = create_response(provider) assert type(response) is str assert len(response) > 0 return response except Exception: return False if __name__ == "__main__": main()