Add optional ids filter to QueryParam for RAG document filtering
This commit is contained in:
parent
d6a426d3e6
commit
a1708a2638
1 changed files with 2 additions and 0 deletions
|
|
@ -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."""
|
||||
...
|
||||
```
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue