typo

Author: mohammedahmed18

codeflash/lsp/lsp_message.py

@@ -53,6 +53,17 @@ def serialize(self) -> str:
         return super().serialize()
 
 
+# TODO: use it instead of the lspcodemessage to display multiple files in the same message
+class LspMultiCodeMessage(LspMessage):
+    files: list[LspCodeMessage]
+
+    def type(self) -> str:
+        return "code"
+
+    def serialize(self) -> str:
+        return super().serialize()
+
+
 @dataclass
 class LspCodeMessage(LspMessage):
     code: str = ""

codeflash/result/explanation.py

@@ -97,10 +97,11 @@ def __str__(self) -> str:
             + self.raw_explanation_message
             + " \n\n"
             + (
-                "The new optimized code was tested for correctness. The results are listed below.\n" + test_report_str
-                if is_LSP_enabled()
                 # in the lsp (extension) we display the test results before the optimization summary
-                else ""
+                ""
+                if is_LSP_enabled()
+                else "The new optimized code was tested for correctness. The results are listed below.\n"
+                + test_report_str
             )
         )