From 48affb01cc0674cd9d179dd8743c5ceaae80e9e6 Mon Sep 17 00:00:00 2001 From: Ty Dunn Date: Thu, 15 Jun 2023 22:37:35 -0700 Subject: creating segs from lines --- continuedev/src/continuedev/steps/core/core.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) (limited to 'continuedev/src') 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]) -- cgit v1.2.3-70-g09d2