admin-valentin/open-notebook
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: prevent migration check from occurring on every page load

Browse source
This commit is contained in:
LUIS NOVO 2025-05-30 15:25:11 -03:00
parent 0aacbcaccb
commit ca589e933d
1 changed files with 13 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
pages/stream_app/utils.py
View file

@ -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):