diff --git a/lightrag/__init__.py b/lightrag/__init__.py index 2d660928..e4cb3e63 100644 --- a/lightrag/__init__.py +++ b/lightrag/__init__.py @@ -1,5 +1,5 @@ from .lightrag import LightRAG as LightRAG, QueryParam as QueryParam -__version__ = "1.2.3" +__version__ = "1.2.4" __author__ = "Zirui Guo" __url__ = "https://github.com/HKUDS/LightRAG" diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 6f42003d..a36934e8 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -36,7 +36,7 @@ from .operate import ( mix_kg_vector_query, naive_query, ) -from .prompt import GRAPH_FIELD_SEP +from .prompt import GRAPH_FIELD_SEP, PROMPTS from .utils import ( EmbeddingFunc, always_get_an_event_loop, @@ -236,7 +236,11 @@ class LightRAG: max_parallel_insert: int = field(default=int(os.getenv("MAX_PARALLEL_INSERT", 20))) """Maximum number of parallel insert operations.""" - addon_params: dict[str, Any] = field(default_factory=dict) + addon_params: dict[str, Any] = field( + default_factory=lambda: { + "language": os.getenv("SUMMARY_LANGUAGE", PROMPTS["DEFAULT_LANGUAGE"]) + } + ) # Storages Management # ---