🐛 Knowledge base summary failed. #1619

Author: jiangpeiling

backend/utils/document_vector_utils.py

@@ -424,9 +424,8 @@ def summarize_document(document_content: str, filename: str, language: str = LAN
                 {"role": MESSAGE_ROLE["USER"], "content": user_prompt}
             ]
 
-            # Call LLM
+            # Call LLM, allow more tokens for generation
             response = llm(messages, max_tokens=max_words * 2)
-            # Allow more tokens for generation
             if not response or not response.content:
                 return ""
             return response.content.strip()

frontend/services/knowledgeBaseService.ts

@@ -469,17 +469,20 @@ class KnowledgeBaseService {
                   onProgress(data.message);
                 }
               } else if (data.status === "completed") {
-                // Check if message field exists when status is completed
-                if (!data.message || data.message.trim() === "") {
+                // On completed, check if the accumulated summary is empty
+                if (!summary || summary.trim() === "") {
                   // No summary was generated, throw internationalized error
                   const errorMessage = i18n.t(
                     "knowledgeBase.summary.notGenerated"
                   );
                   throw new Error(errorMessage);
                 }
-                summary += data.message;
-                if (onProgress) {
-                  onProgress(data.message);
+                // If there is a final message, append it
+                if (data.message && data.message.trim() !== "") {
+                  summary += data.message;
+                  if (onProgress) {
+                    onProgress(data.message);
+                  }
                 }
               } else if (data.status === "error") {
                 throw new Error(data.message);