Bug fix

Author: suncloudsmoon

Forge.cs

@@ -223,11 +223,12 @@ private static async Task ReviewButton_Click()
             }
             else
             {
+                Word.Document document = CommonUtils.GetActiveDocument(); // Hash code of the active document gets changed after each comment!
                 foreach (Word.Paragraph p in paragraphs)
                     // It isn't a paragraph if it doesn't contain a full stop.
                     if (p.Range.Text.Contains('.'))
                     {
-                        await CommentHandler.AddComment(CommonUtils.GetComments(), p.Range, Review(paragraphs, p.Range, prompt));
+                        await CommentHandler.AddComment(CommonUtils.GetComments(), p.Range, Review(paragraphs, p.Range, prompt, document));
                         hasCommented = true;
                     }
             }
@@ -303,15 +304,15 @@ private void UpdateCheckbox()
             DefaultCheckBox.Checked = (Properties.Settings.Default.DefaultModel == ThisAddIn.Model);
         }
 
-        private static AsyncCollectionResult<StreamingChatCompletionUpdate> Review(Word.Paragraphs context, Word.Range p, string prompt)
+        private static AsyncCollectionResult<StreamingChatCompletionUpdate> Review(Word.Paragraphs context, Word.Range p, string prompt, Word.Document doc = null)
         {
             var docRange = Globals.ThisAddIn.Application.ActiveDocument.Range();
             List<UserChatMessage> userChat = new List<UserChatMessage>()
             {
                 new UserChatMessage($@"Please review the following paragraph extracted from the Document: ""{CommonUtils.SubstringTokens(p.Text, (int)(ThisAddIn.ContextLength * 0.2))}"""),
                 new UserChatMessage(prompt)
             };
-            return RAGControl.AskQuestion(CommentSystemPrompt, userChat, docRange);
+            return RAGControl.AskQuestion(CommentSystemPrompt, userChat, docRange, doc);
         }
     }
 }

RAGControl.cs

@@ -390,8 +390,10 @@ public string GetRAGContext(string query, int maxTokens)
         }
 
         // UTILS
-        public static AsyncCollectionResult<StreamingChatCompletionUpdate> AskQuestion(SystemChatMessage systemPrompt, IEnumerable<ChatMessage> messages, Word.Range context)
+        public static AsyncCollectionResult<StreamingChatCompletionUpdate> AskQuestion(SystemChatMessage systemPrompt, IEnumerable<ChatMessage> messages, Word.Range context, Word.Document doc = null)
         {
+            if (doc == null)
+                doc = context.Document;
             string document = context.Text;
             int userPromptLen = GetUserPromptLen(messages);
             ChatMessage lastUserPrompt = messages.Last();
@@ -406,7 +408,7 @@ public static AsyncCollectionResult<StreamingChatCompletionUpdate> AskQuestion(S
                 document = RAGControl.GetWordDocumentAsRAG(lastUserPrompt.Content[0].Text, context);
 
             string ragQuery =
-                (constraints["rag_context"] == 0f) ? string.Empty : ThisAddIn.AllTaskPanes[Globals.ThisAddIn.Application.ActiveDocument].Item3.GetRAGContext(lastUserPrompt.Content[0].Text, (int)(ThisAddIn.ContextLength * constraints["rag_context"]));
+                (constraints["rag_context"] == 0f) ? string.Empty : ThisAddIn.AllTaskPanes[doc].Item3.GetRAGContext(lastUserPrompt.Content[0].Text, (int)(ThisAddIn.ContextLength * constraints["rag_context"]));
 
             List<ChatMessage> chatHistory = new List<ChatMessage>()
             {