diff options
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/src/continuedev/steps/core/core.py | 17 | ||||
| -rw-r--r-- | continuedev/src/continuedev/steps/main.py | 2 |
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" |