Merge pull request #989 from danielaskdd/Add-summary-lang-support
Add summary language setting by env
This commit is contained in:
commit
b420d2e289
1 changed files with 6 additions and 2 deletions
|
|
@ -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
|
||||||
# ---
|
# ---
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue