From 6c779eef2c65092134058e394dbcc4f41c6cdd3d Mon Sep 17 00:00:00 2001 From: Francisco Javier Arceo Date: Fri, 28 Mar 2025 15:33:25 -0400 Subject: [PATCH] fix: Adding chunk_size_in_tokens to playground rag_tool insert Signed-off-by: Francisco Javier Arceo --- llama_stack/distribution/ui/page/playground/rag.py | 1 + 1 file changed, 1 insertion(+) diff --git a/llama_stack/distribution/ui/page/playground/rag.py b/llama_stack/distribution/ui/page/playground/rag.py index fded229c4..fcd0f908e 100644 --- a/llama_stack/distribution/ui/page/playground/rag.py +++ b/llama_stack/distribution/ui/page/playground/rag.py @@ -58,6 +58,7 @@ def rag_chat_page(): llama_stack_api.client.tool_runtime.rag_tool.insert( vector_db_id=vector_db_name, # Use the user-provided name documents=documents, + chunk_size_in_tokens=512, ) st.success("Vector database created successfully!")