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

Refactor get_messages and get_transactions in monitor.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-21 16:00:13 -03:00
commit b7d175af2f
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -41,14 +41,17 @@ async def delete_vertex_builds(
@router.get("/messages") @router.get("/messages")
async def get_messages( async def get_messages(
session_id: Optional[str] = Query(None), session_id: Optional[str] = Query(None),
sender_type: Optional[str] = Query(None), sender: Optional[str] = Query(None),
sender_name: Optional[str] = Query(None), sender_name: Optional[str] = Query(None),
order_by: Optional[str] = Query("timestamp"), order_by: Optional[str] = Query("timestamp"),
monitor_service: MonitorService = Depends(get_monitor_service), monitor_service: MonitorService = Depends(get_monitor_service),
): ):
try: try:
return monitor_service.get_messages( 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: except Exception as e:
raise HTTPException(status_code=500, detail=str(e)) raise HTTPException(status_code=500, detail=str(e))
@ -63,6 +66,8 @@ async def get_transactions(
monitor_service: MonitorService = Depends(get_monitor_service), monitor_service: MonitorService = Depends(get_monitor_service),
): ):
try: 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: except Exception as e:
raise HTTPException(status_code=500, detail=str(e)) raise HTTPException(status_code=500, detail=str(e))