diff --git a/README-zh.md b/README-zh.md index 4a764cb9..784fd1f2 100644 --- a/README-zh.md +++ b/README-zh.md @@ -164,6 +164,10 @@ class QueryParam: """本地检索中实体描述的最大令牌分配。""" ids: list[str] | None = None # 仅支持PG向量数据库 """用于过滤RAG的ID列表。""" + model_func: Callable[..., object] | None = None + """查询使用的LLM模型函数。如果提供了此选项,它将代替LightRAG全局模型函数。 + 这允许为不同的查询模式使用不同的模型。 + """ ... ``` diff --git a/README.md b/README.md index 79772c8b..d3716416 100644 --- a/README.md +++ b/README.md @@ -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. + """ ... ```