mirror of
https://github.com/HKUDS/LightRAG.git
synced 2025-11-22 13:06:10 +00:00
ebaff228aa
- Add DEFAULT_MIN_RERANK_SCORE constant (default: 0.0) - Add MIN_RERANK_SCORE environment variable support - Filter chunks with rerank scores below threshold in process_chunks_unified - Add info-level logging for filtering operations - Handle empty results gracefully after filtering - Maintain backward compatibility with non-reranked chunks
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
"""
|
|
Centralized configuration constants for LightRAG.
|
|
|
|
This module defines default values for configuration constants used across
|
|
different parts of the LightRAG system. Centralizing these values ensures
|
|
consistency and makes maintenance easier.
|
|
"""
|
|
|
|
# Default values for environment variables
|
|
DEFAULT_MAX_GLEANING = 1
|
|
DEFAULT_FORCE_LLM_SUMMARY_ON_MERGE = 4
|
|
DEFAULT_WOKERS = 2
|
|
DEFAULT_TIMEOUT = 150
|
|
|
|
# Query and retrieval configuration defaults
|
|
DEFAULT_TOP_K = 40
|
|
DEFAULT_CHUNK_TOP_K = 10
|
|
DEFAULT_MAX_ENTITY_TOKENS = 10000
|
|
DEFAULT_MAX_RELATION_TOKENS = 10000
|
|
DEFAULT_MAX_TOTAL_TOKENS = 30000
|
|
DEFAULT_HISTORY_TURNS = 0
|
|
DEFAULT_ENABLE_RERANK = True
|
|
DEFAULT_COSINE_THRESHOLD = 0.2
|
|
DEFAULT_RELATED_CHUNK_NUMBER = 5
|
|
|
|
# Rerank configuration defaults
|
|
DEFAULT_MIN_RERANK_SCORE = 0.0
|
|
|
|
# Separator for graph fields
|
|
GRAPH_FIELD_SEP = "<SEP>"
|
|
|
|
# File path configuration for vector and graph database
|
|
DEFAULT_MAX_FILE_PATH_LENGTH = 4090
|
|
|
|
# Logging configuration defaults
|
|
DEFAULT_LOG_MAX_BYTES = 10485760 # Default 10MB
|
|
DEFAULT_LOG_BACKUP_COUNT = 5 # Default 5 backups
|
|
DEFAULT_LOG_FILENAME = "lightrag.log" # Default log filename
|