Merge pull request #1184 from danielaskdd/main

Update README
This commit is contained in:
Daniel.y
2025-03-25 18:54:26 +08:00
committed by GitHub
2 changed files with 9 additions and 0 deletions

View File

@@ -164,6 +164,10 @@ class QueryParam:
"""本地检索中实体描述的最大令牌分配。"""
ids: list[str] | None = None # 仅支持PG向量数据库
"""用于过滤RAG的ID列表。"""
model_func: Callable[..., object] | None = None
"""查询使用的LLM模型函数。如果提供了此选项它将代替LightRAG全局模型函数。
这允许为不同的查询模式使用不同的模型。
"""
...
```

View File

@@ -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.
"""
...
```