From 6c39cbf773145fd098bfc7b81c7fa8722c6a1338 Mon Sep 17 00:00:00 2001 From: yangdx Date: Tue, 4 Mar 2025 12:45:35 +0800 Subject: [PATCH 1/3] Add summary language setting by env --- lightrag/lightrag.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index a5d3c94b..ea302822 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,9 @@ 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 # --- From 06b2124dd0e9774c5d81352060d7c5ca4a4a5ce8 Mon Sep 17 00:00:00 2001 From: yangdx Date: Tue, 4 Mar 2025 14:02:14 +0800 Subject: [PATCH 2/3] Fix linting --- lightrag/lightrag.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index ea302822..27fdafda 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -236,9 +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=lambda: { - "language": os.getenv("SUMMARY_LANGUAGE", PROMPTS["DEFAULT_LANGUAGE"]) - }) + addon_params: dict[str, Any] = field( + default_factory=lambda: { + "language": os.getenv("SUMMARY_LANGUAGE", PROMPTS["DEFAULT_LANGUAGE"]) + } + ) # Storages Management # --- From 4e59a293fe792f3733a0ff836b6b7fe714ba5526 Mon Sep 17 00:00:00 2001 From: zrguo <49157727+LarFii@users.noreply.github.com> Date: Tue, 4 Mar 2025 16:19:23 +0800 Subject: [PATCH 3/3] Update __init__.py --- lightrag/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"