From af092b41dce498b27c46b71db71382cedd2504dd Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 30 Jan 2024 23:05:21 -0300 Subject: [PATCH] Add monitor API endpoint for getting vertex builds --- src/backend/langflow/api/v1/monitor.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 src/backend/langflow/api/v1/monitor.py diff --git a/src/backend/langflow/api/v1/monitor.py b/src/backend/langflow/api/v1/monitor.py new file mode 100644 index 000000000..23f4ee9df --- /dev/null +++ b/src/backend/langflow/api/v1/monitor.py @@ -0,0 +1,20 @@ +from typing import Optional + +from fastapi import APIRouter, Depends, Query + +from langflow.services.deps import get_monitor_service +from langflow.services.monitor.schema import VertexBuildModel +from langflow.services.monitor.service import MonitorService + +router = APIRouter(prefix="/monitor", tags=["Monitor"]) + + +# Get vertex_builds data from the monitor service +@router.get("/builds", response_model=list[VertexBuildModel]) +async def get_vertex_builds( + flow_id: Optional[str] = Query(None), + vertex_id: Optional[str] = Query(None), + valid: Optional[bool] = Query(None), + monitor_service: MonitorService = Depends(get_monitor_service), +): + return monitor_service.get_vertex_builds(flow_id=flow_id, vertex_id=vertex_id, valid=valid)