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

model selection playground fix

Browse source
This commit is contained in:
Xi Yan 2024-12-17 18:12:59 -08:00
parent 3700022d6f
commit af8f1b3531
2 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -11,7 +11,11 @@ from modules.api import llama_stack_api
with st.sidebar: 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 = [model.identifier for model in available_models] available_models = [
model.identifier
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",
available_models, available_models,

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

@ -74,7 +74,11 @@ def rag_chat_page():
] ]
available_models = llama_stack_api.client.models.list() available_models = llama_stack_api.client.models.list()
available_models = [model.identifier for model in available_models] available_models = [
model.identifier
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",
available_models, available_models,
@ -116,8 +120,6 @@ def rag_chat_page():
with st.chat_message(message["role"]): with st.chat_message(message["role"]):
st.markdown(message["content"]) st.markdown(message["content"])
selected_model = llama_stack_api.client.models.list()[0].identifier
agent_config = AgentConfig( agent_config = AgentConfig(
model=selected_model, model=selected_model,
instructions=system_prompt, instructions=system_prompt,