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

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

View file

@ -11,6 +11,7 @@ from pydantic import BaseModel, Field
from typing_extensions import Annotated, Protocol, runtime_checkable
from llama_stack.apis.common.content_types import URL, InterleavedContent
from llama_stack.apis.preprocessing import PreprocessorChain
from llama_stack.providers.utils.telemetry.trace_protocol import trace_protocol
from llama_stack.schema_utils import json_schema_type, register_schema, webmethod
@ -79,6 +80,7 @@ class RAGToolRuntime(Protocol):
documents: List[RAGDocument],
vector_db_id: str,
chunk_size_in_tokens: int = 512,
preprocessor_chain: Optional[PreprocessorChain] = None,
) -> None:
"""Index documents so they can be used by the RAG system"""
...