updated readme
Browse files
README.md
CHANGED
@@ -355,16 +355,26 @@ In order to run this experiment on low RAM GPU you should select small model and
|
|
355 |
```python
|
356 |
class QueryParam:
|
357 |
mode: Literal["local", "global", "hybrid", "naive", "mix"] = "global"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
358 |
only_need_context: bool = False
|
|
|
359 |
response_type: str = "Multiple Paragraphs"
|
360 |
-
|
361 |
top_k: int = 60
|
362 |
-
|
363 |
max_token_for_text_unit: int = 4000
|
364 |
-
|
365 |
max_token_for_global_context: int = 4000
|
366 |
-
|
367 |
max_token_for_local_context: int = 4000
|
|
|
|
|
368 |
```
|
369 |
|
370 |
> default value of Top_k can be change by environment variables TOP_K.
|
|
|
355 |
```python
|
356 |
class QueryParam:
|
357 |
mode: Literal["local", "global", "hybrid", "naive", "mix"] = "global"
|
358 |
+
"""Specifies the retrieval mode:
|
359 |
+
- "local": Focuses on context-dependent information.
|
360 |
+
- "global": Utilizes global knowledge.
|
361 |
+
- "hybrid": Combines local and global retrieval methods.
|
362 |
+
- "naive": Performs a basic search without advanced techniques.
|
363 |
+
- "mix": Integrates knowledge graph and vector retrieval.
|
364 |
+
"""
|
365 |
only_need_context: bool = False
|
366 |
+
"""If True, only returns the retrieved context without generating a response."""
|
367 |
response_type: str = "Multiple Paragraphs"
|
368 |
+
"""Defines the response format. Examples: 'Multiple Paragraphs', 'Single Paragraph', 'Bullet Points'."""
|
369 |
top_k: int = 60
|
370 |
+
"""Number of top items to retrieve. Represents entities in 'local' mode and relationships in 'global' mode."""
|
371 |
max_token_for_text_unit: int = 4000
|
372 |
+
"""Maximum number of tokens allowed for each retrieved text chunk."""
|
373 |
max_token_for_global_context: int = 4000
|
374 |
+
"""Maximum number of tokens allocated for relationship descriptions in global retrieval."""
|
375 |
max_token_for_local_context: int = 4000
|
376 |
+
"""Maximum number of tokens allocated for entity descriptions in local retrieval."""
|
377 |
+
...
|
378 |
```
|
379 |
|
380 |
> default value of Top_k can be change by environment variables TOP_K.
|