refac: citations

This commit is contained in:
Timothy Jaeryang Baek
2025-02-15 17:24:29 -08:00
parent 3d0c06ccee
commit bbda717b69
2 changed files with 17 additions and 2 deletions

View File

@@ -772,7 +772,7 @@ async def process_chat_payload(request, form_data, metadata, user, model):
if "document" in source:
for doc_idx, doc_context in enumerate(source["document"]):
context_string += f"<source><source_id>{doc_idx}</source_id><source_context>{doc_context}</source_context></source>\n"
context_string += f"<source><source_id>{source_idx}</source_id><source_context>{doc_context}</source_context></source>\n"
context_string = context_string.strip()
prompt = get_last_user_message(form_data["messages"])