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

fix: fix issue that asked for migration repeatedly

Browse source
This commit is contained in:
LUIS NOVO 2025-06-10 11:53:41 -03:00
parent e6f7a6b0e6
commit 715f5741b7
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pages/stream_app/utils.py
View file

@ -53,9 +53,9 @@ def setup_stream_state(current_notebook: Notebook) -> ChatSession:
If there is no existing thread state for this session_id, it creates a new one. If there is no existing thread state for this session_id, it creates a new one.
Finally, it acquires the existing state for the session from Langgraph state and sets it in the streamlit session state. Finally, it acquires the existing state for the session from Langgraph state and sets it in the streamlit session state.
""" """
assert ( assert current_notebook is not None and current_notebook.id, (
current_notebook is not None and current_notebook.id "Current Notebook not selected properly"
), "Current Notebook not selected properly" )
if "context_config" not in st.session_state[current_notebook.id]: if "context_config" not in st.session_state[current_notebook.id]:
st.session_state[current_notebook.id]["context_config"] = {} st.session_state[current_notebook.id]["context_config"] = {}
@ -99,7 +99,6 @@ def setup_stream_state(current_notebook: Notebook) -> ChatSession:
def check_migration(): def check_migration():
if "migration_required" not in st.session_state: if "migration_required" not in st.session_state:
st.session_state["migration_required"] = None
logger.debug("Running migration check") logger.debug("Running migration check")
mm = MigrationManager() mm = MigrationManager()
if mm.needs_migration: if mm.needs_migration:
@ -108,6 +107,7 @@ def check_migration():
if st.button("Run Migration"): if st.button("Run Migration"):
mm.run_migration_up() mm.run_migration_up()
st.success("Migration successful") st.success("Migration successful")
st.session_state["migration_required"] = False
st.rerun() st.rerun()
st.stop() st.stop()
else: else: