mirror of
https://github.com/lfnovo/open-notebook.git
synced 2026-05-05 23:37:58 +00:00
43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
import streamlit as st
|
|
|
|
from open_notebook.database.migrate import MigrationManager
|
|
|
|
# from open_notebook.config import DEFAULT_MODELS
|
|
from open_notebook.domain.models import DefaultModels
|
|
from stream_app.utils import version_sidebar
|
|
|
|
default_models = DefaultModels.load()
|
|
|
|
version_sidebar()
|
|
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()
|
|
elif (
|
|
not default_models.default_chat_model
|
|
or not default_models.default_transformation_model
|
|
):
|
|
st.warning(
|
|
"You don't have default chat and transformation models selected. Please, select them on the settings page."
|
|
)
|
|
elif not default_models.default_embedding_model:
|
|
st.warning(
|
|
"You don't have a default embedding model selected. Vector search will not be possible and your assistant will be less able to answer your queries. Please, select one on the settings page."
|
|
)
|
|
elif not default_models.default_speech_to_text_model:
|
|
st.warning(
|
|
"You don't have a default speech to text model selected. Your assistant will not be able to transcribe audio. Please, select one on the settings page."
|
|
)
|
|
elif not default_models.default_text_to_speech_model:
|
|
st.warning(
|
|
"You don't have a default text to speech model selected. Your assistant will not be able to generate audio and podcasts. Please, select one on the settings page."
|
|
)
|
|
elif not default_models.large_context_model:
|
|
st.warning(
|
|
"You don't have a large context model selected. Your assistant will not be able to process large documents. Please, select one on the settings page."
|
|
)
|
|
else:
|
|
st.switch_page("pages/2_📒_Notebooks.py")
|