diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py
index ae626a1b..450f6a3d 100644
--- a/skyvern/forge/sdk/routes/agent_protocol.py
+++ b/skyvern/forge/sdk/routes/agent_protocol.py
@@ -221,6 +221,7 @@ async def run_task_v1(
 
 @base_router.get(
     "/tasks/{task_id}",
+    tags=["agent"],
     response_model=TaskResponse,
     openapi_extra={
         "x-fern-sdk-group-name": "agent",