yangdx commited on
Commit
bd45820
·
1 Parent(s): f46dd17

Update README

Browse files
Files changed (2) hide show
  1. README-zh.md +5 -1
  2. README.md +6 -1
README-zh.md CHANGED
@@ -164,6 +164,10 @@ class QueryParam:
164
  """本地检索中实体描述的最大令牌分配。"""
165
  ids: list[str] | None = None # 仅支持PG向量数据库
166
  """用于过滤RAG的ID列表。"""
 
 
 
 
167
  ...
168
  ```
169
 
@@ -764,7 +768,7 @@ rag = LightRAG(
764
  create INDEX CONCURRENTLY entity_idx_node_id ON dickens."Entity" (ag_catalog.agtype_access_operator(properties, '"node_id"'::agtype));
765
  CREATE INDEX CONCURRENTLY entity_node_id_gin_idx ON dickens."Entity" using gin(properties);
766
  ALTER TABLE dickens."DIRECTED" CLUSTER ON directed_sid_idx;
767
-
768
  -- 如有必要可以删除
769
  drop INDEX entity_p_idx;
770
  drop INDEX vertex_p_idx;
 
164
  """本地检索中实体描述的最大令牌分配。"""
165
  ids: list[str] | None = None # 仅支持PG向量数据库
166
  """用于过滤RAG的ID列表。"""
167
+ model_func: Callable[..., object] | None = None
168
+ """查询使用的LLM模型函数。如果提供了此选项,它将代替LightRAG全局模型函数。
169
+ 这允许为不同的查询模式使用不同的模型。
170
+ """
171
  ...
172
  ```
173
 
 
768
  create INDEX CONCURRENTLY entity_idx_node_id ON dickens."Entity" (ag_catalog.agtype_access_operator(properties, '"node_id"'::agtype));
769
  CREATE INDEX CONCURRENTLY entity_node_id_gin_idx ON dickens."Entity" using gin(properties);
770
  ALTER TABLE dickens."DIRECTED" CLUSTER ON directed_sid_idx;
771
+
772
  -- 如有必要可以删除
773
  drop INDEX entity_p_idx;
774
  drop INDEX vertex_p_idx;
README.md CHANGED
@@ -194,6 +194,11 @@ class QueryParam:
194
  """Maximum number of tokens allocated for entity descriptions in local retrieval."""
195
  ids: list[str] | None = None # ONLY SUPPORTED FOR PG VECTOR DBs
196
  """List of ids to filter the RAG."""
 
 
 
 
 
197
  ...
198
  ```
199
 
@@ -761,7 +766,7 @@ For production level scenarios you will most likely want to leverage an enterpri
761
  create INDEX CONCURRENTLY entity_idx_node_id ON dickens."Entity" (ag_catalog.agtype_access_operator(properties, '"node_id"'::agtype));
762
  CREATE INDEX CONCURRENTLY entity_node_id_gin_idx ON dickens."Entity" using gin(properties);
763
  ALTER TABLE dickens."DIRECTED" CLUSTER ON directed_sid_idx;
764
-
765
  -- drop if necessary
766
  drop INDEX entity_p_idx;
767
  drop INDEX vertex_p_idx;
 
194
  """Maximum number of tokens allocated for entity descriptions in local retrieval."""
195
  ids: list[str] | None = None # ONLY SUPPORTED FOR PG VECTOR DBs
196
  """List of ids to filter the RAG."""
197
+ model_func: Callable[..., object] | None = None
198
+ """Optional override for the LLM model function to use for this specific query.
199
+ If provided, this will be used instead of the global model function.
200
+ This allows using different models for different query modes.
201
+ """
202
  ...
203
  ```
204
 
 
766
  create INDEX CONCURRENTLY entity_idx_node_id ON dickens."Entity" (ag_catalog.agtype_access_operator(properties, '"node_id"'::agtype));
767
  CREATE INDEX CONCURRENTLY entity_node_id_gin_idx ON dickens."Entity" using gin(properties);
768
  ALTER TABLE dickens."DIRECTED" CLUSTER ON directed_sid_idx;
769
+
770
  -- drop if necessary
771
  drop INDEX entity_p_idx;
772
  drop INDEX vertex_p_idx;