jleu3482/railseek6
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

workspace working

Browse Source
This commit is contained in:
jleu3482
2026-01-12 22:31:11 +08:00
parent 2738a822d1
commit 370fe6368a
149 changed files with 4648 additions and 660 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
LightRAG-main/lightrag/operate.py
View File

@@ -4008,6 +4008,7 @@ async def _build_llm_context(
if not entities_context and not relations_context:
if return_raw_data:
# Return empty raw data structure when no entities/relations
logger.debug(f"[_build_query_context] Early return (mode != mix) with empty raw data")
empty_raw_data = _convert_to_user_format(
[],
[],
@@ -4122,12 +4123,34 @@ async def _build_query_context(
chunks_vdb,
)
logger.debug(f"[_build_query_context] Early return check: entities={len(search_result['final_entities'])}, relations={len(search_result['final_relations'])}, chunk_tracking={search_result.get('chunk_tracking')}, mode={query_param.mode}, return_raw_data={return_raw_data}")
if not search_result["final_entities"] and not search_result["final_relations"]:
if query_param.mode != "mix":
return None
if return_raw_data:
# Return empty raw data structure when no entities/relations
empty_raw_data = _convert_to_user_format(
[],
[],
[],
query_param.mode,
)
return None, empty_raw_data
else:
return None
else:
if not search_result["chunk_tracking"]:
return None
if return_raw_data:
# Return empty raw data structure when no entities/relations/chunks
logger.debug(f"[_build_query_context] Early return (mode=mix, no chunk_tracking) with empty raw data")
empty_raw_data = _convert_to_user_format(
[],
[],
[],
query_param.mode,
)
return None, empty_raw_data
else:
return None
# Stage 2: Apply token truncation for LLM efficiency
truncation_result = await _apply_token_truncation(
@@ -4155,7 +4178,17 @@ async def _build_query_context(
and not truncation_result["entities_context"]
and not truncation_result["relations_context"]
):
return None
if return_raw_data:
# Return empty raw data structure when no entities/relations/chunks
empty_raw_data = _convert_to_user_format(
[],
[],
[],
query_param.mode,
)
return None, empty_raw_data
else:
return None
# Stage 4: Build final LLM context with dynamic token processing