This commit is contained in:
Timothy Jaeryang Baek
2025-04-02 20:07:09 -07:00
parent 5c5160c294
commit c0711ba0c9
2 changed files with 8 additions and 7 deletions

View File

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