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

Improve mix query context format

Browse Source
This commit is contained in:
yangdx
2025-05-07 03:11:59 +08:00
parent 098846b651
commit c984ebd462
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lightrag/operate.py
View File

@@ -1249,7 +1249,7 @@ async def mix_kg_vector_query(
for c in maybe_trun_chunks: for c in maybe_trun_chunks:
chunk_text = "File path: " + c["file_path"] + "\r\n\r\n" + c["content"] chunk_text = "File path: " + c["file_path"] + "\r\n\r\n" + c["content"]
if c["created_at"]: if c["created_at"]:
chunk_text = f"[Created at: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(c['created_at']))}]\n{chunk_text}" chunk_text = f"[Created at: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(c['created_at']))}]\r\n\r\n{chunk_text}"
formatted_chunks.append(chunk_text) formatted_chunks.append(chunk_text)
logger.debug( logger.debug(
@@ -1271,12 +1271,12 @@ async def mix_kg_vector_query(
if query_param.only_need_context: if query_param.only_need_context:
context_str = f""" context_str = f"""
-----Knowledge Graph Context----- \r\n\r\n=====Knowledge Graph Context=====\r\n\r\n
{kg_context if kg_context else "No relevant knowledge graph information found"} {kg_context if kg_context else "No relevant knowledge graph information found"}
-----Vector Context----- \r\n\r\n=====Vector Context=====\r\n\r\n
{vector_context if vector_context else "No relevant text information found"} {vector_context if vector_context else "No relevant text information found"}
""".strip() """.strip()
return context_str return context_str
# 5. Construct hybrid prompt # 5. Construct hybrid prompt