OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-12 17:33:11 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac: start source citation from 1

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-03-26 20:39:39 -07:00
parent dd2d757270
commit 42636347e2
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/open_webui/utils/middleware.py
View File

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

2
src/lib/utils/index.ts
View File

@ -64,7 +64,7 @@ export const replaceTokens = (content, sourceIds, char, user) => {
if (Array.isArray(sourceIds)) { if (Array.isArray(sourceIds)) {
sourceIds.forEach((sourceId, idx) => { sourceIds.forEach((sourceId, idx) => {
const regex = new RegExp(`\\[${idx}\\]`, 'g'); const regex = new RegExp(`\\[${idx}\\]`, 'g');
segment = segment.replace(regex, `<source_id data="${idx}" title="${sourceId}" />`); segment = segment.replace(regex, `<source_id data="${idx + 1}" title="${sourceId}" />`);
}); });
} }