From 608a43402cc8775899c53d0c6f9c6d4ba7e90369 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 26 Nov 2024 07:01:14 -0800 Subject: [PATCH] refactor: move UnknownEditFormat exception to base_coder.py --- aider/coders/base_coder.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index 9893f475d..9bdd89f4b 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -22,7 +22,8 @@ from typing import List from aider import __version__, models, prompts, urls, utils from aider.analytics import Analytics from aider.commands import Commands -from aider.exceptions import LiteLLMExceptions, UnknownEditFormat +from aider.exceptions import LiteLLMExceptions +from aider.coders.base_coder import UnknownEditFormat from aider.history import ChatSummary from aider.io import ConfirmGroup, InputOutput from aider.linter import Linter @@ -37,6 +38,14 @@ from ..dump import dump # noqa: F401 from .chat_chunks import ChatChunks +class UnknownEditFormat(ValueError): + def __init__(self, edit_format, valid_formats): + self.edit_format = edit_format + self.valid_formats = valid_formats + super().__init__( + f"Unknown edit format {edit_format}. Valid formats are: {', '.join(valid_formats)}" + ) + class MissingAPIKeyError(ValueError): pass