File size: 613 Bytes
ab8b5a4 |
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 27 28 29 30 31 32 33 34 35 36 |
"""
Generate router
"""
import logging
from contextlib import asynccontextmanager
from fastapi import APIRouter
from api.schema import SuccessDetail
@asynccontextmanager
async def lifespan(app: APIRouter):
logger = logging.getLogger("uvicorn")
logger.info("Starting inference router...")
yield
logger.info("Shutting down inference router...")
router = APIRouter(
lifespan=lifespan,
)
@router.get(
"/",
status_code=200,
response_model=SuccessDetail,
)
async def home():
"""
Submodule home page.
"""
return {"success": "Welcome to the inference submodule!"}
|