phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-31 07:13:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added the preprocessing chain parameter to the RAG tool insert API.

Browse source
This commit is contained in:
ilya-kolchinsky 2025-03-06 14:22:19 +01:00
parent 4c81a72214
commit 6cbc298edb
3 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/distribution/routers/routers.py
View file

@ -458,9 +458,10 @@ class ToolRuntimeRouter(ToolRuntime):
documents: List[RAGDocument],
vector_db_id: str,
chunk_size_in_tokens: int = 512,
preprocessor_chain: Optional[PreprocessorChain] = None,
) -> None:
return await self.routing_table.get_provider_impl("insert_into_memory").insert(
documents, vector_db_id, chunk_size_in_tokens
documents, vector_db_id, chunk_size_in_tokens, preprocessor_chain
)
def __init__(