feat(lightrag): 添加环境变量控制嵌入层最大 token 数量
- 在 lightrag_api_openai_compatible_demo.py 中,使用环境变量 EMBEDDING_MAX_TOKEN_SIZE 来设置嵌入层的最大 token 数量,默认值为 8192
This commit is contained in:
@@ -60,7 +60,7 @@ rag = LightRAG(
|
|||||||
working_dir=WORKING_DIR,
|
working_dir=WORKING_DIR,
|
||||||
llm_model_func=llm_model_func,
|
llm_model_func=llm_model_func,
|
||||||
embedding_func=EmbeddingFunc(embedding_dim=asyncio.run(get_embedding_dim()),
|
embedding_func=EmbeddingFunc(embedding_dim=asyncio.run(get_embedding_dim()),
|
||||||
max_token_size=8192,
|
max_token_size=os.environ.get("EMBEDDING_MAX_TOKEN_SIZE", 8192),
|
||||||
func=embedding_func),
|
func=embedding_func),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user