16 lines
504 B
Python
16 lines
504 B
Python
"""Single-endpoint gateway that fans requests out to whichever backend
|
|
provider is configured."""
|
|
|
|
from .providers import PROVIDERS
|
|
|
|
|
|
class Gateway:
|
|
def __init__(self, default_provider="gemini"):
|
|
self.default_provider = default_provider
|
|
|
|
def route(self, provider_name=None):
|
|
provider_name = provider_name or self.default_provider
|
|
if provider_name not in PROVIDERS:
|
|
raise ValueError(f"unknown provider: {provider_name}")
|
|
return PROVIDERS[provider_name]
|