phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

model_type=llm for filering available models for playground

Browse source
This commit is contained in:
Xi Yan 2024-12-17 19:42:38 -08:00
parent 2f9fdb0ea7
commit 75e72cf2fc
2 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/distribution/ui/page/playground/chat.py
View file

@ -12,9 +12,7 @@ with st.sidebar:
st.header("Configuration") st.header("Configuration")
available_models = llama_stack_api.client.models.list() available_models = llama_stack_api.client.models.list()
available_models = [ available_models = [
model.identifier model.identifier for model in available_models if model.model_type == "llm"
for model in available_models
if model.identifier.startswith("meta-llama")
] ]
selected_model = st.selectbox( selected_model = st.selectbox(
"Choose a model", "Choose a model",

4
llama_stack/distribution/ui/page/playground/rag.py
View file

@ -75,9 +75,7 @@ def rag_chat_page():
available_models = llama_stack_api.client.models.list() available_models = llama_stack_api.client.models.list()
available_models = [ available_models = [
model.identifier model.identifier for model in available_models if model.model_type == "llm"
for model in available_models
if model.identifier.startswith("meta-llama")
] ]
selected_model = st.selectbox( selected_model = st.selectbox(
"Choose a model", "Choose a model",