From 6c39cbf773145fd098bfc7b81c7fa8722c6a1338 Mon Sep 17 00:00:00 2001 From: yangdx Date: Tue, 4 Mar 2025 12:45:35 +0800 Subject: [PATCH 1/2] 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/2] 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 # ---