From 2f7e39fb10248b9d14f2a839d4f60f619ceb8eee Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 2 Dec 2024 13:20:23 -0800 Subject: [PATCH] fix --- llama_stack/distribution/ui/page/playground/rag.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/distribution/ui/page/playground/rag.py b/llama_stack/distribution/ui/page/playground/rag.py index d187caa09..01b0d622d 100644 --- a/llama_stack/distribution/ui/page/playground/rag.py +++ b/llama_stack/distribution/ui/page/playground/rag.py @@ -73,7 +73,7 @@ def rag_chat_page(): {"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] selected_model = st.selectbox( "Choose a model",