diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index da05bf13..f053b25e 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -183,10 +183,10 @@ class LightRAG: embedding_func: EmbeddingFunc | None = field(default=None) """Function for computing text embeddings. Must be set before use.""" - embedding_batch_num: int = field(default=32) + embedding_batch_num: int = field(default=int(os.getenv("EMBEDDING_BATCH_NUM", 32))) """Batch size for embedding computations.""" - embedding_func_max_async: int = field(default=16) + embedding_func_max_async: int = field(default=int(os.getenv("EMBEDDING_FUNC_MAX_ASYNC", 16))) """Maximum number of concurrent embedding function calls.""" embedding_cache_config: dict[str, Any] = field(