Add summary language setting by env
This commit is contained in:
@@ -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
|
||||||
# ---
|
# ---
|
||||||
|
Reference in New Issue
Block a user