mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-09-26 20:31:14 +08:00
25 lines
850 B
Python
25 lines
850 B
Python
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"
|
|
|
|
class OpenRouterFree(OpenRouter):
|
|
label = "OpenRouter (free)"
|
|
max_tokens = 4096
|
|
active_by_default = True
|
|
|
|
@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 |