From b7d175af2f4753c22bfdeb455aae05ebd5ad7112 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 21 Feb 2024 16:00:13 -0300 Subject: [PATCH] Refactor get_messages and get_transactions in monitor.py --- src/backend/langflow/api/v1/monitor.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/api/v1/monitor.py b/src/backend/langflow/api/v1/monitor.py index 3c4749c0d..98c96d34d 100644 --- a/src/backend/langflow/api/v1/monitor.py +++ b/src/backend/langflow/api/v1/monitor.py @@ -41,14 +41,17 @@ async def delete_vertex_builds( @router.get("/messages") async def get_messages( session_id: Optional[str] = Query(None), - sender_type: Optional[str] = Query(None), + sender: Optional[str] = Query(None), sender_name: Optional[str] = Query(None), order_by: Optional[str] = Query("timestamp"), monitor_service: MonitorService = Depends(get_monitor_service), ): try: return monitor_service.get_messages( - sender_type=sender_type, sender_name=sender_name, session_id=session_id, order_by=order_by + sender=sender, + sender_name=sender_name, + session_id=session_id, + order_by=order_by, ) except Exception as e: raise HTTPException(status_code=500, detail=str(e)) @@ -63,6 +66,8 @@ async def get_transactions( monitor_service: MonitorService = Depends(get_monitor_service), ): try: - return monitor_service.get_transactions(source=source, target=target, status=status, order_by=order_by) + return monitor_service.get_transactions( + source=source, target=target, status=status, order_by=order_by + ) except Exception as e: raise HTTPException(status_code=500, detail=str(e))