From 480507eff0dd7a13b0919dfdbad99e2edcc0b12b Mon Sep 17 00:00:00 2001
From: Rustem Galiullin <rustem.galiullin@bayanat.ai>
Date: Wed, 22 May 2024 12:30:48 +0400
Subject: [PATCH] add fastapi method tags

---
 src/hayhooks/server/handlers/deploy.py    | 2 +-
 src/hayhooks/server/handlers/draw.py      | 2 +-
 src/hayhooks/server/handlers/undeploy.py  | 2 +-
 src/hayhooks/server/utils/deploy_utils.py | 1 +
 4 files changed, 4 insertions(+), 3 deletions(-)

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()