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