From 0d9dcef9946c7b8b5e668ed26ca497694b70259a 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:52:33 +0000
Subject: [PATCH] fix: close connection after GET in websocket

---
 store/main.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/store/main.py b/store/main.py
index 8ebef39..2ecae0d 100644
--- a/store/main.py
+++ b/store/main.py
@@ -48,7 +48,9 @@ async def websocket_endpoint(websocket: WebSocket, user_id: int):
     try:
         # send already available data
         r = processed_agent_data.select().order_by(processed_agent_data.c.timestamp)
+        session = SessionLocal()
         stored_data = SessionLocal().execute(r).fetchall()
+        session.close() 
 
         jsonable_data = [{c.name: getattr(i, c.name) for c in processed_agent_data.columns} for i in stored_data]
         for i in jsonable_data: