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,
|
||||
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),
|
||||
)
|
||||
|
||||
|
Reference in New Issue
Block a user