Upload rag_reponse_002.py

rag_reponse_002.py

@@ -112,7 +112,16 @@ def rag_source(docs):
     print('starting source function!')
     source = ""
     for i, doc in enumerate(docs):
-        source += f"**【信息来源 {i+1}】** " + extract_document_name(doc.metadata['source']) + '，' + f"第{docs[i].metadata['page']+1}页" + '，部分内容摘录：' + extract_sentence(doc.page_content) + '\n\n'
+        # Get the sentence or use a fallback if None is returned
+        extracted_sentence = extract_sentence(doc.page_content) or "内容较短，无法提取完整句子"
+        
+        source += (f"**【信息来源 {i+1}】** " + 
+                  extract_document_name(doc.metadata['source']) + 
+                  '，' + 
+                  f"第{docs[i].metadata['page']+1}页" + 
+                  '，部分内容摘录：' + 
+                  extracted_sentence + 
+                  '\n\n')
     print('source:', source)
     return source