Add summary language setting by env

This commit is contained in:
yangdx
2025-03-04 12:45:35 +08:00
parent acb074d0f8
commit 6c39cbf773

View File

@@ -36,7 +36,7 @@ from .operate import (
mix_kg_vector_query, mix_kg_vector_query,
naive_query, naive_query,
) )
from .prompt import GRAPH_FIELD_SEP from .prompt import GRAPH_FIELD_SEP, PROMPTS
from .utils import ( from .utils import (
EmbeddingFunc, EmbeddingFunc,
always_get_an_event_loop, always_get_an_event_loop,
@@ -236,7 +236,9 @@ class LightRAG:
max_parallel_insert: int = field(default=int(os.getenv("MAX_PARALLEL_INSERT", 20))) max_parallel_insert: int = field(default=int(os.getenv("MAX_PARALLEL_INSERT", 20)))
"""Maximum number of parallel insert operations.""" """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 # Storages Management
# --- # ---