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."""
|
"""Maximum number of tokens allocated for relationship descriptions in global retrieval."""
|
||||||
max_token_for_local_context: int = 4000
|
max_token_for_local_context: int = 4000
|
||||||
"""Maximum number of tokens allocated for entity descriptions in local retrieval."""
|
"""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