mirror of
https://github.com/HKUDS/LightRAG.git
synced 2025-11-23 05:26:38 +00:00
bf43e1b8c1
- Read config from selected_rerank_func when env var missing - Make api_key optional for rerank function - Add response format validation with proper error handling - Update Cohere rerank default to official API endpoint
66 lines
2.2 KiB
Python
66 lines
2.2 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 server settings
|
|
DEFAULT_WOKERS = 2
|
|
DEFAULT_MAX_GRAPH_NODES = 1000
|
|
|
|
# Default values for extraction settings
|
|
DEFAULT_SUMMARY_LANGUAGE = "English" # Default language for summaries
|
|
DEFAULT_FORCE_LLM_SUMMARY_ON_MERGE = 4
|
|
DEFAULT_MAX_GLEANING = 1
|
|
DEFAULT_SUMMARY_MAX_TOKENS = 30000 # Default maximum token size
|
|
|
|
# Separator for graph fields
|
|
GRAPH_FIELD_SEP = "<SEP>"
|
|
|
|
# Query and retrieval configuration defaults
|
|
DEFAULT_TOP_K = 40
|
|
DEFAULT_CHUNK_TOP_K = 20
|
|
DEFAULT_MAX_ENTITY_TOKENS = 6000
|
|
DEFAULT_MAX_RELATION_TOKENS = 8000
|
|
DEFAULT_MAX_TOTAL_TOKENS = 30000
|
|
DEFAULT_COSINE_THRESHOLD = 0.2
|
|
DEFAULT_RELATED_CHUNK_NUMBER = 5
|
|
DEFAULT_KG_CHUNK_PICK_METHOD = "VECTOR"
|
|
# Deprated: history message have negtive effect on query performance
|
|
DEFAULT_HISTORY_TURNS = 0
|
|
|
|
# Rerank configuration defaults
|
|
DEFAULT_MIN_RERANK_SCORE = 0.0
|
|
DEFAULT_RERANK_BINDING = "cohere"
|
|
|
|
# File path configuration for vector and graph database(Should not be changed, used in Milvus Schema)
|
|
DEFAULT_MAX_FILE_PATH_LENGTH = 32768
|
|
|
|
# Default temperature for LLM
|
|
DEFAULT_TEMPERATURE = 1.0
|
|
|
|
# Async configuration defaults
|
|
DEFAULT_MAX_ASYNC = 4 # Default maximum async operations
|
|
DEFAULT_MAX_PARALLEL_INSERT = 2 # Default maximum parallel insert operations
|
|
|
|
# Embedding configuration defaults
|
|
DEFAULT_EMBEDDING_FUNC_MAX_ASYNC = 8 # Default max async for embedding functions
|
|
DEFAULT_EMBEDDING_BATCH_NUM = 10 # Default batch size for embedding computations
|
|
|
|
# gunicorn worker timeout(as default LLM request timeout if LLM_TIMEOUT is not set)
|
|
DEFAULT_TIMEOUT = 150
|
|
|
|
# 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
|
|
|
|
# Ollama server configuration defaults
|
|
DEFAULT_OLLAMA_MODEL_NAME = "lightrag"
|
|
DEFAULT_OLLAMA_MODEL_TAG = "latest"
|
|
DEFAULT_OLLAMA_MODEL_SIZE = 7365960935
|
|
DEFAULT_OLLAMA_CREATED_AT = "2024-01-15T00:00:00Z"
|
|
DEFAULT_OLLAMA_DIGEST = "sha256:lightrag"
|