From ca589e933dc748c01db1b4ba94d33e4ec6b1229a Mon Sep 17 00:00:00 2001
From: LUIS NOVO <lfnovo@gmail.com>
Date: Fri, 30 May 2025 15:25:11 -0300
Subject: [PATCH] fix: prevent migration check from occurring on every page
 load

---
 pages/stream_app/utils.py | 22 +++++++++++++---------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/pages/stream_app/utils.py b/pages/stream_app/utils.py
index ae3c3ee..926b468 100644
--- a/pages/stream_app/utils.py
+++ b/pages/stream_app/utils.py
@@ -98,15 +98,19 @@ def setup_stream_state(current_notebook: Notebook) -> ChatSession:
 
 
 def check_migration():
-    logger.critical("Running migration check")
-    mm = MigrationManager()
-    if mm.needs_migration:
-        st.warning("The Open Notebook database needs a migration to run properly.")
-        if st.button("Run Migration"):
-            mm.run_migration_up()
-            st.success("Migration successful")
-            st.rerun()
-        st.stop()
+    if "migration_required" not in st.session_state:
+        st.session_state["migration_required"] = None
+        logger.critical("Running migration check")
+        mm = MigrationManager()
+        if mm.needs_migration:
+            st.warning("The Open Notebook database needs a migration to run properly.")
+            if st.button("Run Migration"):
+                mm.run_migration_up()
+                st.success("Migration successful")
+                st.rerun()
+            st.stop()
+        else:
+            st.session_state["migration_required"] = False
 
 
 def check_models(only_mandatory=True, stop_on_error=True):