diff --git a/chain_service/controllers/__main__.py b/chain_service/controllers/__main__.py index 9d56b05..3552275 100644 --- a/chain_service/controllers/__main__.py +++ b/chain_service/controllers/__main__.py @@ -1,8 +1,13 @@ -from fastapi import FastAPI, APIRouter from .chain import router as chain_router +from .run_chain import router as run_chain_router + +from fastapi import FastAPI, APIRouter def setup_controllers(application: FastAPI): router = APIRouter(prefix="/api") + router.include_router(chain_router) + router.include_router(run_chain_router) + application.include_router(router) diff --git a/chain_service/controllers/run_chain.py b/chain_service/controllers/run_chain.py new file mode 100644 index 0000000..f4178ca --- /dev/null +++ b/chain_service/controllers/run_chain.py @@ -0,0 +1,15 @@ +from loguru import logger +from fastapi import APIRouter, HTTPException + +router = APIRouter() + + +@router.post("/run_chain") +async def run_chain_controller(): + try: + logger.info("Got chain to run") + return {"works": True} + + except Exception: + logger.exception("Error during run chain") + return HTTPException(status_code=500, detail="Error during run chain")