woody
/
legco_ai_assistant
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(backend): use [filename, page N] citation labels in RAG context (sub-phase 2.6) 

Replace numeric [1] labels with [filename, page N] format in context chunks.
Update LLM prompt to instruct inline citation using bracket labels.
Enables traceable source references in generated answers.

Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent)

Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
Woody 2026-04-24 17:52:54 +08:00
parent 06f016c83d
commit e78b670baa
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/app/services/rag.py
View File

@ -95,8 +95,10 @@ class RAGService:
for i, (chunk, meta) in enumerate(zip(chunks, metadata_list)): for i, (chunk, meta) in enumerate(zip(chunks, metadata_list)):
source = meta.get("filename", "unknown") source = meta.get("filename", "unknown")
summary = meta.get("content_summary", "") summary = meta.get("content_summary", "")
page_num = meta.get("page_number")
citation_label = f"{source}, page {page_num}" if page_num else source
context_parts.append( context_parts.append(
f"[{i + 1}] Source: {source}\n" f"[{citation_label}] Source: {source}\n"
f"Summary: {summary}\n" f"Summary: {summary}\n"
f"Content: {chunk}\n" f"Content: {chunk}\n"
) )
@ -108,7 +110,8 @@ class RAGService:
f"Answer the question using ONLY these document chunks. " f"Answer the question using ONLY these document chunks. "
f"Do not use any external knowledge. " f"Do not use any external knowledge. "
f"Format your answer as bullet points. " f"Format your answer as bullet points. "
f"Cite the source name in [ ] for each point.\n\n" f"Cite your sources inline using the exact bracket labels provided, "
f"e.g. [filename, page N]. Place the citation at the end of each relevant point.\n\n"
f"Document chunks:\n{context}\n\n" f"Document chunks:\n{context}\n\n"
f"Answer:" f"Answer:"
) )