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

prevent users from picking a model type not supported by the provider

Browse source
This commit is contained in:
LUIS NOVO 2024-10-30 14:44:48 -03:00
parent aca75221a7
commit 99f177044f
2 changed files with 26 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
open_notebook/models/__init__.py
View file

@ -30,6 +30,7 @@ MODEL_CLASS_MAP = {
"speech_to_text": { "speech_to_text": {
"openai": OpenAISpeechToTextModel, "openai": OpenAISpeechToTextModel,
}, },
"text_to_speech": {"openai": None, "elevenlabs": None},
} }

41
pages/9_⚙️_Settings.py
View file

@ -3,6 +3,7 @@ import os
import streamlit as st import streamlit as st
from open_notebook.domain.models import DefaultModels, Model from open_notebook.domain.models import DefaultModels, Model
from open_notebook.models import MODEL_CLASS_MAP
from stream_app.utils import version_sidebar from stream_app.utils import version_sidebar
st.set_page_config( st.set_page_config(
@ -19,10 +20,10 @@ provider_status = {}
model_types = [ model_types = [
# "vision", # "vision",
"text generation", "language",
"embedding", "embedding",
"text to speech", "text_to_speech",
"speech to text", "speech_to_text",
] ]
provider_status["ollama"] = os.environ.get("OLLAMA_API_BASE") is not None provider_status["ollama"] = os.environ.get("OLLAMA_API_BASE") is not None
@ -65,20 +66,30 @@ with model_tab:
f"Unavailable Providers: {', '.join(unavailable_providers)}. Please check docs page if you wish to enable them." f"Unavailable Providers: {', '.join(unavailable_providers)}. Please check docs page if you wish to enable them."
) )
model_name = st.text_input("Model Name", "") model_name = st.text_input("Model Name", "")
model_type = st.selectbox("Model Type", model_types)
if st.button("Save"): # Filter model types based on provider availability in MODEL_CLASS_MAP
model = Model(name=model_name, provider=provider, type=model_type) available_model_types = []
model.save() for model_type in model_types:
st.success("Saved") if model_type in MODEL_CLASS_MAP and provider in MODEL_CLASS_MAP[model_type]:
available_model_types.append(model_type)
if not available_model_types:
st.error(f"No compatible model types available for provider: {provider}")
else:
model_type = st.selectbox("Model Type", available_model_types)
if st.button("Save"):
model = Model(name=model_name, provider=provider, type=model_type)
model.save()
st.success("Saved")
st.divider() st.divider()
all_models = Model.get_all() all_models = Model.get_all()
st.subheader("Configured Models") st.subheader("Configured Models")
model_types_available = { model_types_available = {
# "vision": False, # "vision": False,
"text generation": False, "language": False,
"embedding": False, "embedding": False,
"text to speech": False, "text_to_speech": False,
"speech to text": False, "speech_to_text": False,
} }
for model in all_models: for model in all_models:
model_types_available[model.type] = True model_types_available[model.type] = True
@ -107,16 +118,14 @@ def get_selected_index(models, model_id, default=0):
with model_defaults_tab: with model_defaults_tab:
default_models = DefaultModels.load().model_dump() default_models = DefaultModels.load().model_dump()
all_models = Model.get_all() all_models = Model.get_all()
text_generation_models = [ text_generation_models = [model for model in all_models if model.type == "language"]
model for model in all_models if model.type == "text generation"
]
text_to_speech_models = [ text_to_speech_models = [
model for model in all_models if model.type == "text to speech" model for model in all_models if model.type == "text_to_speech"
] ]
speech_to_text_models = [ speech_to_text_models = [
model for model in all_models if model.type == "speech to text" model for model in all_models if model.type == "speech_to_text"
] ]
vision_models = [model for model in all_models if model.type == "vision"] vision_models = [model for model in all_models if model.type == "vision"]
embedding_models = [model for model in all_models if model.type == "embedding"] embedding_models = [model for model in all_models if model.type == "embedding"]