@@ -164,6 +164,10 @@ class QueryParam:
|
||||
"""本地检索中实体描述的最大令牌分配。"""
|
||||
ids: list[str] | None = None # 仅支持PG向量数据库
|
||||
"""用于过滤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."""
|
||||
ids: list[str] | None = None # ONLY SUPPORTED FOR PG VECTOR DBs
|
||||
"""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