diff --git a/src/hayhooks/server/handlers/deploy.py b/src/hayhooks/server/handlers/deploy.py index 21e5166..57899e0 100644 --- a/src/hayhooks/server/handlers/deploy.py +++ b/src/hayhooks/server/handlers/deploy.py @@ -2,6 +2,6 @@ from hayhooks.server.utils.deploy_utils import deploy_pipeline_def, PipelineDefinition -@app.post("/deploy") +@app.post("/deploy", tags=["config"]) async def deploy(pipeline_def: PipelineDefinition): return deploy_pipeline_def(app, pipeline_def) diff --git a/src/hayhooks/server/handlers/draw.py b/src/hayhooks/server/handlers/draw.py index d78a7c3..f942b68 100644 --- a/src/hayhooks/server/handlers/draw.py +++ b/src/hayhooks/server/handlers/draw.py @@ -7,7 +7,7 @@ from hayhooks.server.pipelines import registry -@app.get("/draw/{pipeline_name}") +@app.get("/draw/{pipeline_name}", tags=["config"]) async def status(pipeline_name): pipeline = registry.get(pipeline_name) if not pipeline: diff --git a/src/hayhooks/server/handlers/undeploy.py b/src/hayhooks/server/handlers/undeploy.py index 491cf79..9985261 100644 --- a/src/hayhooks/server/handlers/undeploy.py +++ b/src/hayhooks/server/handlers/undeploy.py @@ -6,7 +6,7 @@ from hayhooks.server.pipelines import registry -@app.post("/undeploy/{pipeline_name}") +@app.post("/undeploy/{pipeline_name}", tags=["config"]) async def deploy(pipeline_name: str): if pipeline_name not in registry.get_names(): raise HTTPException(status_code=404) diff --git a/src/hayhooks/server/utils/deploy_utils.py b/src/hayhooks/server/utils/deploy_utils.py index eada5e6..a2b4119 100644 --- a/src/hayhooks/server/utils/deploy_utils.py +++ b/src/hayhooks/server/utils/deploy_utils.py @@ -37,6 +37,7 @@ async def pipeline_run(pipeline_run_req: PipelineRunRequest) -> JSONResponse: # methods=["POST"], name=pipeline_def.name, response_model=PipelineRunResponse, + tags=["pipelines"], ) app.openapi_schema = None app.setup()