fix: enable vim diagnostics after finish the conversation

Closed #72

rplugin/python3/CopilotChat/handlers/chat_handler.py

@@ -44,6 +44,9 @@ def chat(
         disable_end_separator: bool = False,
         model: str = "gpt-4",
     ):
+        """Disable vim diagnostics on the chat buffer"""
+        self.nvim.command(":lua vim.diagnostic.disable()")
+
         disable_separators = (
             self.nvim.eval("g:copilot_chat_disable_separators") == "yes"
         )
@@ -330,6 +333,9 @@ def _add_chat_messages(
             'require("CopilotChat.utils").log_info(...)', "Copilot answered"
         )
 
+        """ Enable vim diagnostics on the chat buffer after the chat is complete """
+        self.nvim.command(":lua vim.diagnostic.enable()")
+
     def _add_end_separator(self, model: str, disable_separators: bool = False):
         current_datetime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         model_info = f"\n#### Answer provided by Copilot (Model: `{model}`) on {current_datetime}."

rplugin/python3/CopilotChat/handlers/vsplit_chat_handler.py

@@ -37,9 +37,6 @@ def vsplit(self):
         )
         self.nvim.current.window.vars[var_key] = True
 
-        """ Disable vim diagnostics on the chat buffer """
-        self.nvim.command(":lua vim.diagnostic.disable()")
-
     def toggle_vsplit(self):
         """Toggle vsplit chat window."""
         var_key = "copilot_chat"