diff --git a/coder.py b/coder.py index 671342e11..ed44c6234 100755 --- a/coder.py +++ b/coder.py @@ -111,20 +111,11 @@ class Coder: self.repo = repo - def quoted_file(self, fname): - prompt = "\n" - prompt += fname - prompt += "\n```\n" - prompt += Path(fname).read_text() - prompt += "\n```\n" - return prompt - def get_files_content(self): prompt = "" for fname in self.fnames: - prompt += self.quoted_file(fname) + prompt += utils.quoted_file(fname) return prompt - def get_input(self): if self.pretty: self.console.rule() diff --git a/utils.py b/utils.py index 5e539f5aa..c110a8295 100644 --- a/utils.py +++ b/utils.py @@ -44,6 +44,14 @@ def replace_most_similar_chunk(whole, part, replace): return modified_whole +def quoted_file(fname): + prompt = "\n" + prompt += fname + prompt += "\n```\n" + prompt += Path(fname).read_text() + prompt += "\n```\n" + return prompt + def strip_quoted_wrapping(res, fname=None): if not res: return res @@ -60,4 +68,4 @@ def strip_quoted_wrapping(res, fname=None): if res and res[-1] != "\n": res += "\n" - return res + return res \ No newline at end of file