gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add conversation history to naive_query prompt

Browse Source
This commit is contained in:
yangdx
2025-01-25 16:29:18 +08:00
parent 207c0e8299
commit d1969515d3
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lightrag/operate.py
View File

@@ -1534,9 +1534,16 @@ async def naive_query(
if query_param.only_need_context: if query_param.only_need_context:
return section return section
# Process conversation history
history_context = ""
if query_param.conversation_history:
history_context = get_conversation_turns(
query_param.conversation_history, query_param.history_turns
)
sys_prompt_temp = PROMPTS["naive_rag_response"] sys_prompt_temp = PROMPTS["naive_rag_response"]
sys_prompt = sys_prompt_temp.format( sys_prompt = sys_prompt_temp.format(
content_data=section, response_type=query_param.response_type content_data=section, response_type=query_param.response_type, history=history_context
) )
if query_param.only_need_prompt: if query_param.only_need_prompt: