@@ -164,6 +164,10 @@ class QueryParam:
|
|||||||
"""本地检索中实体描述的最大令牌分配。"""
|
"""本地检索中实体描述的最大令牌分配。"""
|
||||||
ids: list[str] | None = None # 仅支持PG向量数据库
|
ids: list[str] | None = None # 仅支持PG向量数据库
|
||||||
"""用于过滤RAG的ID列表。"""
|
"""用于过滤RAG的ID列表。"""
|
||||||
|
model_func: Callable[..., object] | None = None
|
||||||
|
"""查询使用的LLM模型函数。如果提供了此选项,它将代替LightRAG全局模型函数。
|
||||||
|
这允许为不同的查询模式使用不同的模型。
|
||||||
|
"""
|
||||||
...
|
...
|
||||||
```
|
```
|
||||||
|
|
||||||
|
@@ -194,6 +194,11 @@ class QueryParam:
|
|||||||
"""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 # ONLY SUPPORTED FOR PG VECTOR DBs
|
ids: list[str] | None = None # ONLY SUPPORTED FOR PG VECTOR DBs
|
||||||
"""List of ids to filter the RAG."""
|
"""List of ids to filter the RAG."""
|
||||||
|
model_func: Callable[..., object] | None = None
|
||||||
|
"""Optional override for the LLM model function to use for this specific query.
|
||||||
|
If provided, this will be used instead of the global model function.
|
||||||
|
This allows using different models for different query modes.
|
||||||
|
"""
|
||||||
...
|
...
|
||||||
```
|
```
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user