From 985e362544bc4a8f3a65064d181a757735750ec9 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 22 Jun 2023 08:15:20 -0700 Subject: [PATCH] renamed func -> wholefile_func --- aider/coders/__init__.py | 4 ++-- aider/coders/base_coder.py | 6 +++--- aider/coders/{func_coder.py => wholefile_func_coder.py} | 6 +++--- aider/coders/{func_prompts.py => wholefile_func_prompts.py} | 3 ++- 4 files changed, 10 insertions(+), 9 deletions(-) rename aider/coders/{func_coder.py => wholefile_func_coder.py} (96%) rename aider/coders/{func_prompts.py => wholefile_func_prompts.py} (88%) diff --git a/aider/coders/__init__.py b/aider/coders/__init__.py index 5a4126ee9..96b5ba2bd 100644 --- a/aider/coders/__init__.py +++ b/aider/coders/__init__.py @@ -1,6 +1,6 @@ from .base_coder import Coder from .editblock_coder import EditBlockCoder -from .func_coder import FunctionCoder from .wholefile_coder import WholeFileCoder +from .wholefile_func_coder import WholeFileFunctionCoder -__all__ = [Coder, EditBlockCoder, WholeFileCoder, FunctionCoder] +__all__ = [Coder, EditBlockCoder, WholeFileCoder, WholeFileFunctionCoder] diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index 1fc553124..8f7900e1f 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -49,7 +49,7 @@ class Coder: openai_api_base="https://api.openai.com/v1", **kwargs, ): - from . import EditBlockCoder, FunctionCoder, WholeFileCoder + from . import EditBlockCoder, WholeFileCoder, WholeFileFunctionCoder openai.api_key = openai_api_key openai.api_base = openai_api_base @@ -70,8 +70,8 @@ class Coder: return EditBlockCoder(main_model, io, **kwargs) elif edit_format == "whole": return WholeFileCoder(main_model, io, **kwargs) - elif edit_format == "func": - return FunctionCoder(main_model, io, **kwargs) + elif edit_format == "whole-func": + return WholeFileFunctionCoder(main_model, io, **kwargs) else: raise ValueError(f"Unknown edit format {edit_format}") diff --git a/aider/coders/func_coder.py b/aider/coders/wholefile_func_coder.py similarity index 96% rename from aider/coders/func_coder.py rename to aider/coders/wholefile_func_coder.py index 8f3203c74..85b9ae289 100644 --- a/aider/coders/func_coder.py +++ b/aider/coders/wholefile_func_coder.py @@ -5,10 +5,10 @@ from aider import diffs from ..dump import dump # noqa: F401 from .base_coder import Coder -from .func_prompts import FunctionPrompts +from .wholefile_func_prompts import WholeFileFunctionPrompts -class FunctionCoder(Coder): +class WholeFileFunctionCoder(Coder): functions = [ dict( name="write_file", @@ -47,7 +47,7 @@ class FunctionCoder(Coder): ] def __init__(self, *args, **kwargs): - self.gpt_prompts = FunctionPrompts() + self.gpt_prompts = WholeFileFunctionPrompts() super().__init__(*args, **kwargs) def update_cur_messages(self, content, edited): diff --git a/aider/coders/func_prompts.py b/aider/coders/wholefile_func_prompts.py similarity index 88% rename from aider/coders/func_prompts.py rename to aider/coders/wholefile_func_prompts.py index 0e3ea62bc..324502ed9 100644 --- a/aider/coders/func_prompts.py +++ b/aider/coders/wholefile_func_prompts.py @@ -3,7 +3,7 @@ from .prompts_base import EditorPrompts -class FunctionPrompts(EditorPrompts): +class WholeFileFunctionPrompts(EditorPrompts): main_system = """Act as an expert software developer. Take requests for changes to the supplied code. If the request is ambiguous, ask questions. @@ -21,6 +21,7 @@ NEVER return code outside the `write_file` function. redacted_edit_message = "No changes are needed." + # TODO: make this optional, since this Coder doesn't use it repo_content_prefix = ( "Below here are summaries of other files! Do not propose changes to these *read-only*" " files without asking me first.\n"