From f5fec47f76c43fc6d915569a933563b7e94cd936 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 25 Jun 2024 09:13:13 -0300 Subject: [PATCH] chore: Add error handling for message not found in update_message endpoint --- src/backend/base/langflow/api/v1/monitor.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/api/v1/monitor.py b/src/backend/base/langflow/api/v1/monitor.py index c4e595f63..244c9a2fa 100644 --- a/src/backend/base/langflow/api/v1/monitor.py +++ b/src/backend/base/langflow/api/v1/monitor.py @@ -1,4 +1,5 @@ from typing import List, Optional +from uuid import UUID from fastapi import APIRouter, Depends, HTTPException, Query from sqlalchemy import delete @@ -84,12 +85,14 @@ async def delete_messages( @router.post("/messages/{message_id}", response_model=MessageRead) async def update_message( - message_id: int, + message_id: UUID, message: MessageUpdate, session: Session = Depends(get_session), ): try: db_message = session.get(MessageTable, message_id) + if not db_message: + raise HTTPException(status_code=404, detail="Message not found") message_dict = message.model_dump(exclude_unset=True) db_message.sqlmodel_update(message_dict) session.add(db_message)