From 6e615545bb63dd1330b742e2a8080a29f6d03b52 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=8A=AA=E5=8A=9B=E5=A5=8B=E6=96=97=E7=9A=84=E9=98=BF?= =?UTF-8?q?=E4=B8=89?= <59323683+zh19990906@users.noreply.github.com> Date: Tue, 24 Sep 2024 17:56:50 +0800 Subject: [PATCH] Update monitor.py Clearing should be about clearing all conversations, not restricting the deletion of only certain ones. --- src/backend/base/langflow/api/v1/monitor.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/api/v1/monitor.py b/src/backend/base/langflow/api/v1/monitor.py index 8ec4d3c81f2..269522593c6 100644 --- a/src/backend/base/langflow/api/v1/monitor.py +++ b/src/backend/base/langflow/api/v1/monitor.py @@ -73,12 +73,16 @@ async def get_messages( @router.delete("/messages", status_code=204) async def delete_messages( - message_ids: list[UUID], + message_ids: list[UUID] | None, + flow_id: str | None = Query(None), session: Session = Depends(get_session), current_user: User = Depends(get_current_active_user), ): try: - session.exec(delete(MessageTable).where(MessageTable.id.in_(message_ids))) # type: ignore + if flow_id: + session.exec(delete(MessageTable).where(MessageTable.flow_id == flow_id)) + else: + session.exec(delete(MessageTable).where(MessageTable.id.in_(message_ids))) # type: ignore session.commit() except Exception as e: raise HTTPException(status_code=500, detail=str(e))