diff --git a/codeplain_REST_api.py b/codeplain_REST_api.py index dd3570a..e815d9f 100644 --- a/codeplain_REST_api.py +++ b/codeplain_REST_api.py @@ -18,7 +18,7 @@ ERROR_CODE_EXCEPTIONS = { "FunctionalRequirementTooComplex": plain2code_exceptions.FunctionalRequirementTooComplex, "ConflictingRequirements": plain2code_exceptions.ConflictingRequirements, - "CreditBalanceTooLow": plain2code_exceptions.CreditBalanceTooLow, + "CreditBalanceTooLow": plain2code_exceptions.RenderingCreditBalanceTooLow, "LLMInternalError": plain2code_exceptions.LLMInternalError, "MissingResource": plain2code_exceptions.MissingResource, "PlainSyntaxError": plain2code_exceptions.PlainSyntaxError, diff --git a/plain2code.py b/plain2code.py index 554d7f6..1b75128 100644 --- a/plain2code.py +++ b/plain2code.py @@ -23,7 +23,6 @@ from plain2code_events import RenderFailed from plain2code_exceptions import ( ConflictingRequirements, - CreditBalanceTooLow, InternalClientError, InternalServerError, InvalidAPIKey, @@ -37,6 +36,7 @@ OutdatedClientVersion, PlainSyntaxError, RenderCancelledError, + RenderingCreditBalanceTooLow, ) from plain2code_logger import ( LOGGER_NAME, @@ -346,7 +346,7 @@ def main(): # noqa: C901 exc_info = sys.exc_info() console.error(f"Conflicting requirements: {str(e)}\n") console.debug(f"Render ID: {run_state.render_id}") - except CreditBalanceTooLow as e: + except RenderingCreditBalanceTooLow as e: exc_info = sys.exc_info() console.error(f"Credit balance too low: {str(e)}\n") console.debug(f"Render ID: {run_state.render_id}") diff --git a/plain2code_exceptions.py b/plain2code_exceptions.py index f05cb9e..bb3d94f 100644 --- a/plain2code_exceptions.py +++ b/plain2code_exceptions.py @@ -9,7 +9,7 @@ class ConflictingRequirements(Exception): pass -class CreditBalanceTooLow(Exception): +class RenderingCreditBalanceTooLow(Exception): pass