From 8064f88cc6e58be79f45755a8bb4ef955d4eeaf4 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 31 Jan 2024 14:58:58 -0300 Subject: [PATCH] Add order_by parameter to get_vertex_builds, get_messages, and get_transactions --- src/backend/langflow/api/v1/monitor.py | 27 +++++++++++++++++++++++++- 1 file changed, 26 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/api/v1/monitor.py b/src/backend/langflow/api/v1/monitor.py index 23f4ee9df..1c6b1b77f 100644 --- a/src/backend/langflow/api/v1/monitor.py +++ b/src/backend/langflow/api/v1/monitor.py @@ -15,6 +15,31 @@ async def get_vertex_builds( flow_id: Optional[str] = Query(None), vertex_id: Optional[str] = Query(None), valid: Optional[bool] = Query(None), + order_by: Optional[str] = Query("timestamp"), monitor_service: MonitorService = Depends(get_monitor_service), ): - return monitor_service.get_vertex_builds(flow_id=flow_id, vertex_id=vertex_id, valid=valid) + return monitor_service.get_vertex_builds(flow_id=flow_id, vertex_id=vertex_id, valid=valid, order_by=order_by) + + +@router.get("/messages") +async def get_messages( + session_id: Optional[str] = Query(None), + sender_type: Optional[str] = Query(None), + sender_name: Optional[str] = Query(None), + order_by: Optional[str] = Query("timestamp"), + monitor_service: MonitorService = Depends(get_monitor_service), +): + return monitor_service.get_messages( + sender_type=sender_type, sender_name=sender_name, session_id=session_id, order_by=order_by + ) + + +@router.get("/transactions") +async def get_transactions( + source: Optional[str] = Query(None), + target: Optional[str] = Query(None), + status: Optional[str] = Query(None), + order_by: Optional[str] = Query("timestamp"), + monitor_service: MonitorService = Depends(get_monitor_service), +): + return monitor_service.get_transactions(source=source, target=target, status=status, order_by=order_by)