kill servers started by old extension versions

2 files changed, 11 insertions, 8 deletions

diff --git a/continuedev/src/continuedev/steps/core/core.py b/continuedev/src/continuedev/steps/core/core.py
index f81b3f6d..f59ef33c 100644
--- a/continuedev/src/continuedev/steps/core/core.py
+++ b/continuedev/src/continuedev/steps/core/core.py
@@ -221,18 +221,21 @@ class DefaultModelEditCodeStep(Step):
         # TODO: Keep track of start line of the range, because it's needed below for offset stuff
         rif_start_line = rif.range.start.line
         if len(rif.contents) > 0:
-            first_line = rif.contents.splitlines(keepends=True)[0]
-            while first_line.strip() == "":
+            lines = rif.contents.splitlines(keepends=True)
+            first_line = lines[0] if lines else None
+            while first_line and first_line.strip() == "":
                 file_prefix += first_line
                 rif.contents = rif.contents[len(first_line):]
-                first_line = rif.contents.splitlines(keepends=True)[0]
+                lines = rif.contents.splitlines(keepends=True)
+                first_line = lines[0] if lines else None
 
-            last_line = rif.contents.splitlines(keepends=True)[-1]
-            while last_line.strip() == "":
+            last_line = lines[-1] if lines else None
+            while last_line and last_line.strip() == "":
                 file_suffix = last_line + file_suffix
                 rif.contents = rif.contents[:len(
                     rif.contents) - len(last_line)]
-                last_line = rif.contents.splitlines(keepends=True)[-1]
+                lines = rif.contents.splitlines(keepends=True)
+                last_line = lines[-1] if lines else None
 
             while rif.contents.startswith("\n"):
                 file_prefix += "\n"
@@ -241,7 +244,7 @@ class DefaultModelEditCodeStep(Step):
                 file_suffix = "\n" + file_suffix
                 rif.contents = rif.contents[:-1]
 
-        return file_prefix, rif.contents, file_suffix, model_to_use
+                return file_prefix, rif.contents, file_suffix, model_to_use
 
     def compile_prompt(self, file_prefix: str, contents: str, file_suffix: str, sdk: ContinueSDK) -> str:
         prompt = self._prompt
diff --git a/continuedev/src/continuedev/steps/main.py b/continuedev/src/continuedev/steps/main.py
index 5caac180..def1af4e 100644
--- a/continuedev/src/continuedev/steps/main.py
+++ b/continuedev/src/continuedev/steps/main.py
@@ -249,7 +249,7 @@ class EditHighlightedCodeStep(Step):
     user_input: str = Field(
         ..., title="User Input", description="The natural language request describing how to edit the code")
     hide = True
-    description: str = "Change the contents of the currently highlighted code or open file"
+    description: str = "Change the contents of the currently highlighted code or open file. You should call this function if the user asks seems to be asking for a code change."
 
     async def describe(self, models: Models) -> Coroutine[str, None, None]:
         return "Editing code"