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

4
pages/stream_app/utils.py
View file

@ -98,6 +98,8 @@ def setup_stream_state(current_notebook: Notebook) -> ChatSession:
def check_migration(): def check_migration():
if "migration_required" not in st.session_state:
st.session_state["migration_required"] = None
logger.critical("Running migration check") logger.critical("Running migration check")
mm = MigrationManager() mm = MigrationManager()
if mm.needs_migration: if mm.needs_migration:
@ -107,6 +109,8 @@ def check_migration():
st.success("Migration successful") st.success("Migration successful")
st.rerun() st.rerun()
st.stop() st.stop()
else:
st.session_state["migration_required"] = False
def check_models(only_mandatory=True, stop_on_error=True): def check_models(only_mandatory=True, stop_on_error=True):