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

chore: Add error handling for message not found in update_message endpoint

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-25 09:13:13 -03:00
commit f5fec47f76
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/api/v1/monitor.py
View file

@ -1,4 +1,5 @@
from typing import List, Optional from typing import List, Optional
from uuid import UUID
from fastapi import APIRouter, Depends, HTTPException, Query from fastapi import APIRouter, Depends, HTTPException, Query
from sqlalchemy import delete from sqlalchemy import delete
@ -84,12 +85,14 @@ async def delete_messages(
@router.post("/messages/{message_id}", response_model=MessageRead) @router.post("/messages/{message_id}", response_model=MessageRead)
async def update_message( async def update_message(
message_id: int, message_id: UUID,
message: MessageUpdate, message: MessageUpdate,
session: Session = Depends(get_session), session: Session = Depends(get_session),
): ):
try: try:
db_message = session.get(MessageTable, message_id) 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) message_dict = message.model_dump(exclude_unset=True)
db_message.sqlmodel_update(message_dict) db_message.sqlmodel_update(message_dict)
session.add(db_message) session.add(db_message)