diff options
author | Ty Dunn <ty@tydunn.com> | 2023-06-15 22:37:35 -0700 |
---|---|---|
committer | Ty Dunn <ty@tydunn.com> | 2023-06-15 22:37:35 -0700 |
commit | 48affb01cc0674cd9d179dd8743c5ceaae80e9e6 (patch) | |
tree | 6efd2e5b701592ca9ed0392428d00ecd2182fa1c /continuedev/src | |
parent | 6899bfb475ca1ec423e542540fa2788393e412ac (diff) | |
download | sncontinue-48affb01cc0674cd9d179dd8743c5ceaae80e9e6.tar.gz sncontinue-48affb01cc0674cd9d179dd8743c5ceaae80e9e6.tar.bz2 sncontinue-48affb01cc0674cd9d179dd8743c5ceaae80e9e6.zip |
creating segs from lines
Diffstat (limited to 'continuedev/src')
-rw-r--r-- | continuedev/src/continuedev/steps/core/core.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/continuedev/src/continuedev/steps/core/core.py b/continuedev/src/continuedev/steps/core/core.py index ee3ef9a7..88dc8d72 100644 --- a/continuedev/src/continuedev/steps/core/core.py +++ b/continuedev/src/continuedev/steps/core/core.py @@ -224,12 +224,11 @@ class DefaultModelEditCodeStep(Step): break else: raise Exception("Unknown default model") + + code_before = "".join(full_file_contents_lst[min_end_line:]) + code_after = "".join(full_file_contents_lst[:max_start_line]) - start_index, end_index = rif.range.indices_in_string( - full_file_contents) - - segs = [full_file_contents[:start_index], - full_file_contents[end_index:]] + segs = [code_before, code_after] prompt = self._prompt.format( code=rif.contents, user_request=self.user_input, file_prefix=segs[0], file_suffix=segs[1]) |