File size: 884 Bytes
ab8b5a4 7ad11ea ab8b5a4 7ad11ea ab8b5a4 7ad11ea |
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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
"""
Generate router
"""
import logging
from contextlib import asynccontextmanager
from fastapi import APIRouter
from api.schema import SuccessDetail
from .schema import InferenceRequest
@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():
"""
Inference home page.
"""
return {"success": "Welcome to the inference submodule!"}
@router.post(
"/match",
status_code=200,
response_model=SuccessDetail,
)
async def match(
inference: InferenceRequest,
):
"""
Match inference.
"""
return {"success": "Inference matched successfully!"}
|