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
commit b420d2e289
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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
# --- # ---