From af6594f6705af6c8b98cef81588b247fced54f10 Mon Sep 17 00:00:00 2001 From: Francisco Arceo Date: Fri, 28 Mar 2025 13:56:25 -0600 Subject: [PATCH] fix: Adding chunk_size_in_tokens to playground rag_tool insert (#1826) # What does this PR do? Adding chunk_size_in_tokens to playground rag_tool insert. # Closes #1825 ## Test Plan Tested locally. [//]: # (## Documentation) 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!")