فهرست منبع

refac: start source citation from 1

Timothy Jaeryang Baek 1 ماه پیش
والد
کامیت
42636347e2
2فایلهای تغییر یافته به همراه2 افزوده شده و 2 حذف شده
  1. 1 1
      backend/open_webui/utils/middleware.py
  2. 1 1
      src/lib/utils/index.ts

+ 1 - 1
backend/open_webui/utils/middleware.py

@@ -847,7 +847,7 @@ 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}</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()
         prompt = get_last_user_message(form_data["messages"])

+ 1 - 1
src/lib/utils/index.ts

@@ -64,7 +64,7 @@ export const replaceTokens = (content, sourceIds, char, user) => {
 		if (Array.isArray(sourceIds)) {
 			sourceIds.forEach((sourceId, idx) => {
 				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}" />`);
 			});
 		}