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 | dfa127d440f4971d0fea9dee27349f7c1e19b5cf (patch) | |
tree | 5baeb915ece64bcc2343dd675bf7007cbc6cbb7f /continuedev/src | |
parent | b5fea3b7b5f19beaf5417d167105b4909eacb14c (diff) | |
download | sncontinue-dfa127d440f4971d0fea9dee27349f7c1e19b5cf.tar.gz sncontinue-dfa127d440f4971d0fea9dee27349f7c1e19b5cf.tar.bz2 sncontinue-dfa127d440f4971d0fea9dee27349f7c1e19b5cf.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]) |