Merge pull request #989 from danielaskdd/Add-summary-lang-support

Add summary language setting by env
This commit is contained in:
zrguo
2025-03-04 16:04:42 +08:00
committed by GitHub

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,11 @@ 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
# --- # ---