From a1708a2638ab0ed02bc303d994823b373db9bdbe Mon Sep 17 00:00:00 2001 From: Roy Date: Sat, 8 Mar 2025 21:13:29 +0000 Subject: [PATCH] Add optional ids filter to QueryParam for RAG document filtering --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 00da54fb..6eb25e65 100644 --- a/README.md +++ b/README.md @@ -176,6 +176,8 @@ class QueryParam: """Maximum number of tokens allocated for relationship descriptions in global retrieval.""" max_token_for_local_context: int = 4000 """Maximum number of tokens allocated for entity descriptions in local retrieval.""" + ids: list[str] | None = None + """List of ids to filter the RAG.""" ... ```