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

chore: improve ux in the model provider page

Browse source
This commit is contained in:
LUIS NOVO 2025-06-10 11:55:58 -03:00
parent c377ebae83
commit f35ccbbed5
1 changed files with 15 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
pages/7_🤖_Models.py
View file

@ -78,6 +78,13 @@ st.divider()
# Helper function to add model with auto-save # Helper function to add model with auto-save
def add_model_form(model_type, container_key): def add_model_form(model_type, container_key):
available_providers = esperanto_available_providers.get(model_type, []) available_providers = esperanto_available_providers.get(model_type, [])
# Sort providers alphabetically for easier navigation
available_providers.sort()
# Remove perplexity from available_providers if it exists
if "perplexity" in available_providers:
available_providers.remove("perplexity")
if not available_providers: if not available_providers:
st.info(f"No providers available for {model_type}") st.info(f"No providers available for {model_type}")
return return
@ -91,15 +98,11 @@ def add_model_form(model_type, container_key):
key=f"provider_{model_type}_{container_key}", key=f"provider_{model_type}_{container_key}",
) )
if model_type == "text_to_speech" and provider == "gemini": model_name = st.text_input(
model_name = "gemini-default" "Model Name",
st.markdown("Gemini models are pre-configured. Using the default model.") key=f"name_{model_type}_{container_key}",
else: help="gpt-4o-mini, claude, gemini, llama3, etc. For azure, use the deployment_name as the model_name",
model_name = st.text_input( )
"Model Name",
key=f"name_{model_type}_{container_key}",
help="gpt-4o-mini, claude, gemini, llama3, etc",
)
if st.form_submit_button("Add Model"): if st.form_submit_button("Add Model"):
if model_name: if model_name:
@ -125,10 +128,12 @@ def handle_default_selection(
setattr(default_models, key, selected_model.id) setattr(default_models, key, selected_model.id)
default_models.update() default_models.update()
model_manager.refresh_defaults() model_manager.refresh_defaults()
st.toast(f"Default {model_type} model set to {selected_model.name}")
elif not selected_model and current_value: elif not selected_model and current_value:
setattr(default_models, key, None) setattr(default_models, key, None)
default_models.update() default_models.update()
model_manager.refresh_defaults() model_manager.refresh_defaults()
st.toast(f"Default {model_type} model removed")
if caption: if caption:
st.caption(caption) st.caption(caption)
@ -150,7 +155,7 @@ for model in all_models:
st.markdown(""" st.markdown("""
**Model Management Guide:** For optimal performance, refer to [Which model to choose?](https://github.com/lfnovo/open-notebook/blob/main/docs/models.md) **Model Management Guide:** For optimal performance, refer to [Which model to choose?](https://github.com/lfnovo/open-notebook/blob/main/docs/models.md)
You can test models in the [Transformations](https://try-it-out.open-notebook.com) page. You can test models in the [Transformations](Transformations) page.
""") """)
# Language Models Section # Language Models Section