added loop summary and updated type for RAG tool output

1 files changed, 6 insertions, 0 deletions

diff --git a/src/client/views/nodes/ChatBox/AnswerParser.ts b/src/client/views/nodes/ChatBox/AnswerParser.ts
index 05d26b8de..885114195 100644
--- a/src/client/views/nodes/ChatBox/AnswerParser.ts
+++ b/src/client/views/nodes/ChatBox/AnswerParser.ts
@@ -10,10 +10,12 @@ export class AnswerParser {
         const questionRegex = /<question>(.*?)<\/question>/g;
         const groundedTextRegex = /<grounded_text citation_index="([^"]+)">([\s\S]*?)<\/grounded_text>/g;
         const normalTextRegex = /<normal_text>([\s\S]*?)<\/normal_text>/g;
+        const loopSummaryRegex = /<loop_summary>([\s\S]*?)<\/loop_summary>/;
 
         const answerMatch = answerRegex.exec(xml);
         const citationsMatch = citationsRegex.exec(xml);
         const followUpQuestionsMatch = followUpQuestionsRegex.exec(xml);
+        const loopSummaryMatch = loopSummaryRegex.exec(xml);
 
         if (!answerMatch) {
             throw new Error('Invalid XML: Missing <answer> tag.');
@@ -31,6 +33,9 @@ export class AnswerParser {
         if (followUpQuestionsMatch) {
             rawTextContent = rawTextContent.replace(followUpQuestionsMatch[0], '').trim();
         }
+        if (loopSummaryMatch) {
+            rawTextContent = rawTextContent.replace(loopSummaryMatch[0], '').trim();
+        }
 
         // Parse citations
         let citationMatch;
@@ -112,6 +117,7 @@ export class AnswerParser {
             follow_up_questions: followUpQuestions,
             citations,
             processing_info: processingInfo,
+            loop_summary: loopSummaryMatch ? loopSummaryMatch[1].trim() : undefined,
         };
 
         return assistantResponse;