feat(lightrag): 添加环境变量控制嵌入层最大 token 数量

- 在 lightrag_api_openai_compatible_demo.py 中,使用环境变量 EMBEDDING_MAX_TOKEN_SIZE 来设置嵌入层的最大 token 数量,默认值为 8192
This commit is contained in:
90houlaoheshang
2024-11-06 10:48:59 +08:00
parent 5099a2e763
commit d644ee52f5

View File

@@ -60,7 +60,7 @@ rag = LightRAG(
working_dir=WORKING_DIR,
llm_model_func=llm_model_func,
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),
)