summaryrefslogtreecommitdiffstats
path: root/tool/readme_table.py
diff options
context:
space:
mode:
Diffstat (limited to 'tool/readme_table.py')
-rw-r--r--tool/readme_table.py27
1 files changed, 3 insertions, 24 deletions
diff --git a/tool/readme_table.py b/tool/readme_table.py
index 522c66a7..dc7c85f3 100644
--- a/tool/readme_table.py
+++ b/tool/readme_table.py
@@ -5,10 +5,9 @@ from urllib.parse import urlparse
sys.path.append(str(Path(__file__).parent.parent))
-from g4f import models, Provider
-from g4f.Provider.base_provider import BaseProvider, AsyncProvider
-from testing.test_providers import test
-
+from g4f import models
+from g4f.Provider.base_provider import AsyncProvider
+from testing.test_providers import test, get_providers
def print_imports():
print("##### Providers:")
@@ -68,26 +67,6 @@ def print_providers():
)
print("\n".join(lines))
-
-def get_provider_names() -> list[str]:
- provider_names = dir(Provider)
- ignore_names = [
- "base_provider",
- "BaseProvider",
- "AsyncProvider",
- "AsyncGeneratorProvider"
- ]
- return [
- provider_name
- for provider_name in provider_names
- if not provider_name.startswith("__") and provider_name not in ignore_names
- ]
-
-
-def get_providers() -> list[type[BaseProvider]]:
- return [getattr(Provider, provider_name) for provider_name in get_provider_names()]
-
-
def print_models():
base_provider_names = {
"cohere": "Cohere",