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

Include user query in prompt returns

Browse source
This commit is contained in:
yangdx 2025-09-21 15:24:20 +08:00
parent 6eb37e270a
commit 8f0fb3c9eb
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lightrag/operate.py
View file

@ -2391,7 +2391,7 @@ async def kg_query(
) )
if query_param.only_need_prompt: if query_param.only_need_prompt:
return sys_prompt return "\n\n".join([sys_prompt, "---User Query---", user_query])
tokenizer: Tokenizer = global_config["tokenizer"] tokenizer: Tokenizer = global_config["tokenizer"]
len_of_prompts = len(tokenizer.encode(query + sys_prompt)) len_of_prompts = len(tokenizer.encode(query + sys_prompt))
@ -4248,7 +4248,7 @@ async def naive_query(
) )
if query_param.only_need_prompt: if query_param.only_need_prompt:
return sys_prompt return "\n\n".join([sys_prompt, "---User Query---", user_query])
len_of_prompts = len(tokenizer.encode(query + sys_prompt)) len_of_prompts = len(tokenizer.encode(query + sys_prompt))
logger.debug( logger.debug(