diff --git a/router/main.py b/router/main.py index 4a163d22..cd50c1b4 100644 --- a/router/main.py +++ b/router/main.py @@ -13,6 +13,7 @@ from .db import init_db from .discovery import providers_router from .models import MODELS, update_sats_pricing from .proxy import proxy_router +from .route.models import models_router __version__ = "0.0.1" @@ -69,6 +70,7 @@ async def info() -> dict: } +app.include_router(models_router) app.include_router(admin_router) app.include_router(wallet_router) app.include_router(providers_router) diff --git a/router/route/__init__.py b/router/route/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/router/route/models.py b/router/route/models.py new file mode 100644 index 00000000..4484ee85 --- /dev/null +++ b/router/route/models.py @@ -0,0 +1,12 @@ +from typing import List + +from fastapi import APIRouter + +from router.models import MODELS, Model + +models_router = APIRouter(prefix="/api/proxy") + + +@models_router.get("/models") +async def get_models() -> List[Model]: + return MODELS