File size: 876 Bytes
a4b70d9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from __future__ import annotations
from ..template import OpenaiTemplate
class OpenRouter(OpenaiTemplate):
label = "OpenRouter"
url = "https://openrouter.ai"
login_url = "https://openrouter.ai/settings/keys"
api_base = "https://openrouter.ai/api/v1"
working = True
needs_auth = True
default_model = "openrouter/auto"
active_by_default = True
class OpenRouterFree(OpenRouter):
parent = "OpenRouter"
label = "OpenRouter (free)"
max_tokens = 5012
@classmethod
def get_models(cls, api_key: str = None, **kwargs):
models = super().get_models(api_key=api_key, **kwargs)
models = [model for model in models if model.endswith(":free")]
cls.model_aliases = {model.replace(":free", ""): model for model in models}
cls.default_model = models[0] if models else cls.default_model
return models |