diff --git a/store/main.py b/store/main.py index 2ecae0d..08a5979 100644 --- a/store/main.py +++ b/store/main.py @@ -200,8 +200,12 @@ def update_processed_agent_data(processed_agent_data_id: int, data: ProcessedAge session.commit() updated_result = session.execute(query).fetchone() - return ProcessedAgentDataInDB(**updated_result._mapping) + + except Exception as err: + session.rollback() + print(f"Database error: {err}") + raise HTTPException(status_code=500, detail="Internal Server Error") finally: session.close()