gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Replace hardcoded max tokens with DEFAULT_MAX_TOTAL_TOKENS constant

Browse source 
- Use constant in process_chunks_unified
- Update WebUI default to match (32000)
This commit is contained in:
yangdx 2025-07-26 11:23:54 +08:00
parent 7b915b34f6
commit a67f93acc9
2 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
lightrag/utils.py
View file

@ -20,6 +20,7 @@ from lightrag.constants import (
DEFAULT_LOG_BACKUP_COUNT, DEFAULT_LOG_BACKUP_COUNT,
DEFAULT_LOG_FILENAME, DEFAULT_LOG_FILENAME,
GRAPH_FIELD_SEP, GRAPH_FIELD_SEP,
DEFAULT_MAX_TOTAL_TOKENS,
DEFAULT_MAX_FILE_PATH_LENGTH, DEFAULT_MAX_FILE_PATH_LENGTH,
) )
@ -1887,7 +1888,7 @@ async def process_chunks_unified(
chunk_token_limit = getattr( chunk_token_limit = getattr(
query_param, query_param,
"max_total_tokens", "max_total_tokens",
global_config.get("MAX_TOTAL_TOKENS", 32000), global_config.get("MAX_TOTAL_TOKENS", DEFAULT_MAX_TOTAL_TOKENS),
) )
original_count = len(unique_chunks) original_count = len(unique_chunks)

2
lightrag_webui/src/components/retrieval/QuerySettings.tsx
View file

@ -270,7 +270,7 @@ export default function QuerySettings() {
onBlur={(e) => { onBlur={(e) => {
const value = e.target.value const value = e.target.value
if (value === '' || isNaN(parseInt(value))) { if (value === '' || isNaN(parseInt(value))) {
handleChange('max_total_tokens', 1000) handleChange('max_total_tokens', 32000)
} }
}} }}
min={1} min={1}