diff --git a/codeflash/lsp/lsp_message.py b/codeflash/lsp/lsp_message.py index cfe15ef68..6724ac7d9 100644 --- a/codeflash/lsp/lsp_message.py +++ b/codeflash/lsp/lsp_message.py @@ -58,7 +58,7 @@ class LspMultiCodeMessage(LspMessage): files: list[LspCodeMessage] def type(self) -> str: - return "code" + return self._type def serialize(self) -> str: return super().serialize() @@ -92,6 +92,8 @@ def type(self) -> str: return "markdown" def serialize(self) -> str: - self.markdown = simplify_worktree_paths(self.markdown) - self.markdown = replace_quotes_with_backticks(self.markdown) + if "worktrees/" in self.markdown and "/" in self.markdown: + self.markdown = simplify_worktree_paths(self.markdown) + if '"' in self.markdown or "'" in self.markdown: + self.markdown = replace_quotes_with_backticks(self.markdown) return super().serialize()