Add monitor API endpoint for getting vertex builds

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-30 23:05:21 -03:00
commit af092b41dc

View file

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