Roy
commited on
Commit
·
7afdb71
1
Parent(s):
011a9ed
Add optional ids filter to QueryParam for RAG document filtering
Browse files
README.md
CHANGED
@@ -176,6 +176,8 @@ class QueryParam:
|
|
176 |
"""Maximum number of tokens allocated for relationship descriptions in global retrieval."""
|
177 |
max_token_for_local_context: int = 4000
|
178 |
"""Maximum number of tokens allocated for entity descriptions in local retrieval."""
|
|
|
|
|
179 |
...
|
180 |
```
|
181 |
|
|
|
176 |
"""Maximum number of tokens allocated for relationship descriptions in global retrieval."""
|
177 |
max_token_for_local_context: int = 4000
|
178 |
"""Maximum number of tokens allocated for entity descriptions in local retrieval."""
|
179 |
+
ids: list[str] | None = None
|
180 |
+
"""List of ids to filter the RAG."""
|
181 |
...
|
182 |
```
|
183 |
|