announce the edit format

This commit is contained in:
Paul Gauthier 2023-12-17 12:58:48 -08:00
parent 7113a30271
commit 360846320f
3 changed files with 6 additions and 1 deletions

View file

@ -52,6 +52,7 @@ class Coder:
num_malformed_responses = 0 num_malformed_responses = 0
last_keyboard_interrupt = None last_keyboard_interrupt = None
max_apply_update_errors = 3 max_apply_update_errors = 3
edit_format = None
@classmethod @classmethod
def create( def create(
@ -149,7 +150,7 @@ class Coder:
self.main_model = main_model self.main_model = main_model
self.io.tool_output(f"Model: {main_model.name}") self.io.tool_output(f"Model: {main_model.name} using {self.edit_format} edit format")
self.show_diffs = show_diffs self.show_diffs = show_diffs

View file

@ -9,6 +9,8 @@ from .editblock_prompts import EditBlockPrompts
class EditBlockCoder(Coder): class EditBlockCoder(Coder):
edit_format = "diff"
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
self.gpt_prompts = EditBlockPrompts() self.gpt_prompts = EditBlockPrompts()
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)

View file

@ -8,6 +8,8 @@ from .wholefile_prompts import WholeFilePrompts
class WholeFileCoder(Coder): class WholeFileCoder(Coder):
edit_format = "whole"
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
self.gpt_prompts = WholeFilePrompts() self.gpt_prompts = WholeFilePrompts()
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)