Add optional ids filter to QueryParam for RAG document filtering

This commit is contained in:
Roy
2025-03-08 21:13:29 +00:00
parent d6a426d3e6
commit a1708a2638

View File

@@ -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."""
...
```