From 1bf568750526f24e70a7c97067a3c2b4c228b5d2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=86=D0=9C-24=20=D0=92=D0=BB=D0=B0=D0=B4=D0=B8=D1=81?=
 =?UTF-8?q?=D0=BB=D0=B0=D0=B2=20=D0=9A=D0=BE=D0=B2=D0=B0=D0=BB=D0=B5=D0=BD?=
 =?UTF-8?q?=D0=BA=D0=BE?= <kovalenkov222@gmail.com>
Date: Thu, 26 Mar 2026 15:56:34 +0000
Subject: [PATCH] fix: add session rollback in PUT

---
 store/main.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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()