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

Fix cache handling and context return logic for query parameters

Browse source 
• Skip cache when only_need_prompt is set
• Update only_need_context condition logic
• Prevent cache bypass in prompt-only mode
This commit is contained in:
yangdx 2025-09-10 11:31:48 +08:00
parent 6774058670
commit e078ab7103
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
lightrag/operate.py
View file

@ -2137,7 +2137,8 @@ async def kg_query(
hashing_kv, args_hash, query, query_param.mode, cache_type="query" hashing_kv, args_hash, query, query_param.mode, cache_type="query"
) )
if cached_response is not None: if cached_response is not None:
return cached_response if not query_param.only_need_context and not query_param.only_need_prompt:
return cached_response
hl_keywords, ll_keywords = await get_keywords_from_query( hl_keywords, ll_keywords = await get_keywords_from_query(
query, query_param, global_config, hashing_kv query, query_param, global_config, hashing_kv
@ -2174,7 +2175,7 @@ async def kg_query(
chunks_vdb, chunks_vdb,
) )
if query_param.only_need_context: if query_param.only_need_context and not query_param.only_need_prompt:
return context if context is not None else PROMPTS["fail_response"] return context if context is not None else PROMPTS["fail_response"]
if context is None: if context is None:
return PROMPTS["fail_response"] return PROMPTS["fail_response"]
@ -3944,7 +3945,8 @@ async def naive_query(
hashing_kv, args_hash, query, query_param.mode, cache_type="query" hashing_kv, args_hash, query, query_param.mode, cache_type="query"
) )
if cached_response is not None: if cached_response is not None:
return cached_response if not query_param.only_need_context and not query_param.only_need_prompt:
return cached_response
tokenizer: Tokenizer = global_config["tokenizer"] tokenizer: Tokenizer = global_config["tokenizer"]
@ -4028,8 +4030,8 @@ async def naive_query(
} }
) )
text_units_str = json.dumps(text_units_context, ensure_ascii=False)
if query_param.only_need_context: if query_param.only_need_context and not query_param.only_need_prompt:
return f""" return f"""
---Document Chunks(DC)--- ---Document Chunks(DC)---