jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor get_vertex_builds API response model

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-31 15:38:55 -03:00
commit ba8b608784
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/langflow/api/v1/monitor.py
View file

@ -2,14 +2,14 @@ from typing import Optional
from fastapi import APIRouter, Depends, HTTPException, Query from fastapi import APIRouter, Depends, HTTPException, Query
from langflow.services.deps import get_monitor_service from langflow.services.deps import get_monitor_service
from langflow.services.monitor.schema import VertexBuildModel from langflow.services.monitor.schema import VertexBuildMapModel
from langflow.services.monitor.service import MonitorService from langflow.services.monitor.service import MonitorService
router = APIRouter(prefix="/monitor", tags=["Monitor"]) router = APIRouter(prefix="/monitor", tags=["Monitor"])
# Get vertex_builds data from the monitor service # Get vertex_builds data from the monitor service
@router.get("/builds", response_model=list[VertexBuildModel]) @router.get("/builds", response_model=VertexBuildMapModel)
async def get_vertex_builds( async def get_vertex_builds(
flow_id: Optional[str] = Query(None), flow_id: Optional[str] = Query(None),
vertex_id: Optional[str] = Query(None), vertex_id: Optional[str] = Query(None),
@ -18,7 +18,11 @@ async def get_vertex_builds(
monitor_service: MonitorService = Depends(get_monitor_service), monitor_service: MonitorService = Depends(get_monitor_service),
): ):
try: try:
return monitor_service.get_vertex_builds(flow_id=flow_id, vertex_id=vertex_id, valid=valid, order_by=order_by) vertex_build_dicts = monitor_service.get_vertex_builds(
flow_id=flow_id, vertex_id=vertex_id, valid=valid, order_by=order_by
)
vertex_build_map = VertexBuildMapModel.from_list_of_dicts(vertex_build_dicts)
return vertex_build_map
except Exception as e: except Exception as e:
raise HTTPException(status_code=500, detail=str(e)) raise HTTPException(status_code=500, detail=str(e))