small fixes

Author: mohammedahmed18

codeflash/lsp/lsp_logger.py

@@ -39,7 +39,6 @@ def extract_tags(msg: str) -> tuple[Optional[LspMessageTags], str]:
     parts = msg.split("|tags|")
     if len(parts) == 2:
         message_tags = LspMessageTags()
-        # Use set for O(1) lookups and remove whitespace in a single pass
         tags = {tag.strip() for tag in parts[0].split(",")}
         if "!lsp" in tags:
             message_tags.not_lsp = True

codeflash/lsp/lsp_message.py

@@ -14,7 +14,7 @@
 @dataclass
 class LspMessage:
     # to show a loading indicator if the operation is taking time like generating candidates or tests
-    takes_time: bool = field(default=False, kw_only=True)
+    takes_time: bool = field(default=False)
 
     def _loop_through(self, obj: Any) -> Any:  # noqa: ANN401
         if isinstance(obj, list):
@@ -35,8 +35,9 @@ def serialize(self) -> str:
         # Important: keep type as the first key, for making it easy and fast for the client to know if this is a lsp message before parsing it
         ordered = {"type": self.type(), **data}
         return (
-            json.dumps(ordered) + "\u241f"
-        )  # \u241F is the message delimiter becuase it can be more than one message sent over the same message
+            json.dumps(ordered)
+            + "\u241f"  # \u241F is the message delimiter becuase it can be more than one message sent over the same message, so we need something to separate each message
+        )
 
 
 @dataclass

codeflash/optimization/function_optimizer.py

@@ -1241,7 +1241,7 @@ def find_and_process_best_optimization(
             )
 
             if best_optimization:
-                logger.info("h2|tags|Best candidate 🚀:")
+                logger.info("h2|tags|Best candidate 🚀")
                 code_print(best_optimization.candidate.source_code.flat)
                 processed_benchmark_info = None
                 if self.args.benchmark:

codeflash/result/critic.py

@@ -9,7 +9,7 @@
     MIN_IMPROVEMENT_THRESHOLD,
     MIN_TESTCASE_PASSED_THRESHOLD,
 )
-from codeflash.models.models import TestType
+from codeflash.models.test_type import TestType
 
 if TYPE_CHECKING:
     from codeflash.models.models import CoverageData, OptimizedCandidateResult, OriginalCodeBaseline