fix: add session rollback in PUT

2026-03-26 15:56:34 +00:00
parent 0d9dcef994
commit 1bf5687505
1 changed files with 5 additions and 1 deletions
--- 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()