INNER CODE UNIT · Python
on_model_selection_change
mrwadams/stride-gpt · apps/web/main.py:938
def on_model_selection_change():
"""Update token limit when specific model is selected"""
if "model_provider" not in st.session_state or "selected_model" not in st.session_state:
return
model_info = get_model(st.session_state.model_provider, st.session_state.selected_model)
if model_info:
st.session_state.token_limit = model_info.default_tokens
# Reset the current_model_key to force token limit update in Advanced Settings
if "current_model_key" in st.session_state:
del st.session_state.current_model_key
# ------------------ Sidebar ------------------ #
st.sidebar.image("stride-gpt-logo.png", width=250)