This commit is contained in:
Xi Yan 2024-12-02 13:20:23 -08:00
parent 6bdad37372
commit 2f7e39fb10

View file

@ -73,7 +73,7 @@ def rag_chat_page():
{"bank_id": bank_id, "type": "vector"} for bank_id in selected_memory_banks {"bank_id": bank_id, "type": "vector"} for bank_id in selected_memory_banks
] ]
available_models = llama_stack_api.list_models() 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]
selected_model = st.selectbox( selected_model = st.selectbox(
"Choose a model", "Choose a model",