codeplain 0.3.2__tar.gz → 0.3.3__tar.gz

{codeplain-0.3.2 → codeplain-0.3.3}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: codeplain
-Version: 0.3.2
+Version: 0.3.3
 Summary: Transform plain language specifications into working code
 License-File: LICENSE
 Classifier: Environment :: Console
 Classifier: Intended Audience :: Developers
 Classifier: Operating System :: OS Independent
 Classifier: Topic :: Software Development :: Code Generators
-Requires-Python: >=3.11
+Requires-Python: ==3.11
 Requires-Dist: gitpython==3.1.50
 Requires-Dist: mistletoe==1.3.0
 Requires-Dist: networkx==3.6.1
@@ -16,6 +16,7 @@ Requires-Dist: python-liquid2==0.3.0
 Requires-Dist: pyyaml==6.0.3
 Requires-Dist: requests==2.34.0
 Requires-Dist: rich==15.0.0
+Requires-Dist: sentry-sdk==2.62.0
 Requires-Dist: textual>=7.5.0
 Requires-Dist: tiktoken==0.12.0
 Requires-Dist: transitions==0.9.3
@@ -103,6 +104,11 @@ After completing the installation steps above, you can immediately test the syst
   python hello_world.py
   ```
 
+### Crash reporting
+
+If the plain2code client crashes unexpectedly, it sends an anonymous crash report to Codeplain to help improve the tool. To disable crash reporting, set the `CODEPLAIN_NO_TELEMETRY=1` flag.
+
+
 ## Releasing
 
 Releases are built and published with [uv](https://docs.astral.sh/uv/). The version is read from `_version.py`.

{codeplain-0.3.2 → codeplain-0.3.3}/README.md

@@ -74,6 +74,11 @@ After completing the installation steps above, you can immediately test the syst
   python hello_world.py
   ```
 
+### Crash reporting
+
+If the plain2code client crashes unexpectedly, it sends an anonymous crash report to Codeplain to help improve the tool. To disable crash reporting, set the `CODEPLAIN_NO_TELEMETRY=1` flag.
+
+
 ## Releasing
 
 Releases are built and published with [uv](https://docs.astral.sh/uv/). The version is read from `_version.py`.

codeplain-0.3.3/_version.py

@@ -0,0 +1 @@
+__version__ = "0.3.3"

{codeplain-0.3.2 → codeplain-0.3.3}/cli_output/status.py

@@ -50,8 +50,8 @@ def _display_credit_line(plan_credits: dict) -> None:
         console.print(f"    {plan_label:10} {bar}   {remaining:2} of {total} remaining    expires {formatted_date}")
 
 
-def _display_purchased_credit_line(bucket: dict) -> None:
-    """Display a purchased credit line with progress bar."""
+def _display_bucket_credit_line(bucket: dict, label: str) -> None:
+    """Display a credit bucket line (purchased or promo) with progress bar."""
     remaining = bucket["remaining"]
     total = bucket["total"]
     expiry_date = bucket["expiry_date"]
@@ -71,12 +71,12 @@ def _display_purchased_credit_line(bucket: dict) -> None:
     bar = _create_progress_bar(remaining, total, width=30)
 
     if is_expired:
-        console.print(f"    Purchased  {bar}   expired {formatted_date}")
+        console.print(f"    {label:10} {bar}   expired {formatted_date}")
     else:
-        console.print(f"    Purchased  {bar}   {remaining:2} of {total} remaining    expires {formatted_date}")
+        console.print(f"    {label:10} {bar}   {remaining:2} of {total} remaining    expires {formatted_date}")
 
 
-def _display_status_message(plan_credits: Optional[dict], purchased_credits: list) -> None:
+def _display_status_message(plan_credits: Optional[dict], purchased_credits: list, promo_credits: list) -> None:
     """Display appropriate status message based on credit state."""
     has_remaining = False
 
@@ -93,8 +93,8 @@ def _display_status_message(plan_credits: Optional[dict], purchased_credits: lis
         if remaining > 0 and period_end > now:
             has_remaining = True
 
-    # Check if any purchased credits have remaining balance and not expired
-    for bucket in purchased_credits:
+    # Check if any purchased or promo credits have remaining balance and not expired
+    for bucket in [*purchased_credits, *promo_credits]:
         dt_str = bucket["expiry_date"].replace("Z", "+00:00")
         expiry_date = datetime.fromisoformat(dt_str)
         # If naive datetime, assume UTC
@@ -127,6 +127,7 @@ def print_status(api_key: str, api_url: str, client_version: str) -> None:
     org_owner = response.get("organization_owner_email")
     plan_credits = response.get("plan_credits")
     purchased_credits = response.get("purchased_credits", [])
+    promo_credits = response.get("promo_credits", [])
 
     # Display header information
     if client_version_valid:
@@ -153,8 +154,12 @@ def print_status(api_key: str, api_url: str, client_version: str) -> None:
 
     # Display purchased credits
     for bucket in purchased_credits:
-        _display_purchased_credit_line(bucket)
+        _display_bucket_credit_line(bucket, "Purchased")
+
+    # Display promo credits
+    for bucket in promo_credits:
+        _display_bucket_credit_line(bucket, "Promo")
 
     # Display status messages and management link
-    _display_status_message(plan_credits, purchased_credits)
+    _display_status_message(plan_credits, purchased_credits, promo_credits)
     console.print("\nTo manage your plan navigate to https://platform.codeplain.ai/plans")

{codeplain-0.3.2 → codeplain-0.3.3}/examples/example_hello_world_python/hello_world_python.plain

@@ -13,7 +13,7 @@ import:
 - Display "hello, world"
 
   ***acceptance tests***
+    
+  - :App: should exit with status code 0 indicating successful execution.
 
-  - :App: should exit with status code 0 indicating successful execution [adsa](asdsa).
-
-  - :App: should complete execution in under 1 second.
+  - :App: should complete execution in under 1 second.

{codeplain-0.3.2 → codeplain-0.3.3}/plain2code.py

@@ -49,6 +49,7 @@ from plain2code_logger import (
     get_log_file_path,
 )
 from plain2code_state import RunState
+from plain2code_telemetry import capture_crash, initialize_telemetry
 from system_config import system_config
 from tui.plain2code_tui import Plain2CodeTUI
 from tui.plain_module_render_choice_tui import PlainModuleRenderChoiceTUI
@@ -56,6 +57,27 @@ from tui.plain_module_render_choice_tui import PlainModuleRenderChoiceTUI
 DEFAULT_TEMPLATE_DIRS = importlib.resources.files("standard_template_library")
 RENDER_THREAD_SHUTDOWN_TIMEOUT = 0.7
 
+# Exceptions that represent expected, user-facing error conditions. They are
+# reported to the user directly and must never be sent to Sentry as crashes.
+EXPECTED_EXCEPTIONS = (
+    InvalidFridArgument,
+    FileNotFoundError,
+    MissingResource,
+    TemplateNotFoundError,
+    PlainSyntaxError,
+    MissingPreviousFunctionalitiesError,
+    MissingAPIKey,
+    InvalidAPIKey,
+    OutdatedClientVersion,
+    ConflictingRequirements,
+    RenderingCreditBalanceTooLow,
+    NetworkConnectionError,
+    ModuleDoesNotExistError,
+    UnsupportedResourceType,
+    GitNotInstalledError,
+    SystemExit,
+)
+
 
 def setup_logging(
     args,
@@ -324,6 +346,8 @@ def main():  # noqa: C901
         args, event_bus, run_state, args.log_to_file, args.log_file_name, args.filename, args.headless
     )
 
+    initialize_telemetry()
+
     exc_info = None
     error_message = None
 
@@ -340,36 +364,17 @@ def main():  # noqa: C901
         else:
             error_message = str(e) if str(e) else repr(e)
 
-            if not isinstance(
-                e,
-                (
-                    InvalidFridArgument,
-                    FileNotFoundError,
-                    MissingResource,
-                    TemplateNotFoundError,
-                    PlainSyntaxError,
-                    MissingPreviousFunctionalitiesError,
-                    MissingAPIKey,
-                    InvalidAPIKey,
-                    OutdatedClientVersion,
-                    ConflictingRequirements,
-                    RenderingCreditBalanceTooLow,
-                    NetworkConnectionError,
-                    ModuleDoesNotExistError,
-                    UnsupportedResourceType,
-                    GitNotInstalledError,
-                ),
-            ):
+            if not isinstance(e, EXPECTED_EXCEPTIONS):
                 exc_info = sys.exc_info()
     finally:
+        if exc_info:
+            dump_crash_logs(args, run_state)
+            capture_crash(exc_info, run_state, args)
         print_exit_summary(
             run_state,
             args.filename,
             error_message=error_message,
         )
-        if exc_info:
-            # Log traceback
-            dump_crash_logs(args, run_state)
 
     if args.headless and (exc_info is not None or not run_state.render_succeeded):
         sys.exit(1)

{codeplain-0.3.2 → codeplain-0.3.3}/plain2code_state.py

@@ -23,6 +23,10 @@ class RunState:
         self.unittest_batch_id: int = 0
         self.render_time_accumulated: int = 0
         self.last_render_start_timestamp: float = time.monotonic()
+        # Mirrored from the render state machine for crash reporting; not authoritative.
+        self.current_module: Optional[str] = None
+        self.current_frid: Optional[str] = None
+        self.current_render_state: Optional[str] = None
 
     def increment_call_count(self):
         self.call_count += 1

codeplain-0.3.3/plain2code_telemetry.py

@@ -0,0 +1,114 @@
+"""Crash reporting via Sentry.
+
+Only unexpected exceptions are reported (the caller decides which exceptions are
+expected; see EXPECTED_EXCEPTIONS in plain2code.py). Reporting is on by default
+and can be disabled by setting the CODEPLAIN_NO_TELEMETRY environment variable
+to any non-empty value.
+"""
+
+import os
+from typing import Any, Optional
+
+import sentry_sdk
+from sentry_sdk.integrations.atexit import AtexitIntegration
+from sentry_sdk.integrations.dedupe import DedupeIntegration
+from sentry_sdk.integrations.modules import ModulesIntegration
+from sentry_sdk.scrubber import DEFAULT_DENYLIST, EventScrubber
+
+from plain2code_state import RunState
+from system_config import system_config
+
+SENTRY_DSN = "https://64d0d86b50b34e2dede3e4eaf5142282@o4510793955934208.ingest.us.sentry.io/4511540621213696"
+
+NO_TELEMETRY_ENV_VAR = "CODEPLAIN_NO_TELEMETRY"
+ENVIRONMENT_ENV_VAR = "CODEPLAIN_ENV"
+DEFAULT_ENVIRONMENT = "production"
+
+FLUSH_TIMEOUT_SECONDS = 2
+
+# Local variable names whose values may contain proprietary spec or generated
+# code content and must be scrubbed from stack traces (extends Sentry's
+# default denylist, which already covers api_key, auth, secrets etc.).
+# "headers" and "x-api-key" cover the request headers local in
+# codeplain_REST_api.post_request; the default denylist only has the
+# underscore form "x_api_key".
+SCRUB_DENYLIST = DEFAULT_DENYLIST + [
+    "authorization",
+    "headers",
+    "x-api-key",
+    "plain_source",
+    "plain_source_tree",
+    "full_plain_source",
+    "existing_files_content",
+    "file_content",
+    "files_content",
+    "content",
+    "source",
+    "response_json",
+    "payload",
+]
+
+
+def telemetry_enabled() -> bool:
+    """Return True if crash reporting should be active."""
+    if os.environ.get(NO_TELEMETRY_ENV_VAR):
+        return False
+    return True
+
+
+def initialize_telemetry(**init_overrides: Any) -> bool:
+    """Initialize Sentry crash reporting. Returns True if initialized."""
+    if not telemetry_enabled():
+        return False
+
+    try:
+        init_kwargs: dict[str, Any] = dict(
+            dsn=SENTRY_DSN,
+            release=system_config.client_version,
+            environment=os.environ.get(ENVIRONMENT_ENV_VAR, DEFAULT_ENVIRONMENT),
+            send_default_pii=False,
+            server_name="",  # hostname is identifying; don't send it
+            default_integrations=False,
+            auto_enabling_integrations=False,
+            integrations=[
+                AtexitIntegration(callback=lambda pending, timeout: None),
+                DedupeIntegration(),
+                ModulesIntegration(),
+            ],
+            include_local_variables=True,
+            event_scrubber=EventScrubber(denylist=SCRUB_DENYLIST, recursive=True),
+            shutdown_timeout=FLUSH_TIMEOUT_SECONDS,
+        )
+        init_kwargs.update(init_overrides)
+        sentry_sdk.init(**init_kwargs)
+        return True
+    except Exception:
+        return False
+
+
+def capture_crash(exc_info, run_state: Optional[RunState], args) -> bool:
+    """Report an unexpected crash to Sentry. Returns True if an event was sent."""
+    if not telemetry_enabled():
+        return False
+
+    try:
+        with sentry_sdk.new_scope() as scope:
+            if run_state is not None:
+                scope.set_tag("render_id", run_state.render_id)
+                scope.set_tag("render_state", run_state.current_render_state)
+                scope.set_tag("current_module", run_state.current_module)
+                scope.set_tag("current_frid", run_state.current_frid)
+            scope.set_tag("headless", bool(getattr(args, "headless", False)))
+            scope.set_tag("unittests_script_provided", bool(getattr(args, "unittests_script", None)))
+            scope.set_tag("conformance_tests_script_provided", bool(getattr(args, "conformance_tests_script", None)))
+            scope.set_tag(
+                "prepare_environment_script_provided", bool(getattr(args, "prepare_environment_script", None))
+            )
+
+            event_id = sentry_sdk.capture_exception(exc_info[1])
+
+        sentry_sdk.flush(timeout=FLUSH_TIMEOUT_SECONDS)
+        return event_id is not None
+    except Exception:
+        # Telemetry must never break the CLI or mask the original crash.
+        return False

{codeplain-0.3.2 → codeplain-0.3.3}/pyproject.toml

@@ -7,7 +7,7 @@ name = "codeplain"
 dynamic = ["version"]
 description = "Transform plain language specifications into working code"
 readme = "README.md"
-requires-python = ">=3.11"
+requires-python = "==3.11"
 classifiers = [
     "Environment :: Console",
     "Intended Audience :: Developers",
@@ -26,6 +26,7 @@ dependencies = [
     "rich==15.0.0",
     "python-frontmatter==1.3.0",
     "networkx==3.6.1",
+    "sentry-sdk==2.62.0",
 ]
 
 [project.optional-dependencies]

{codeplain-0.3.2 → codeplain-0.3.3}/render_machine/code_renderer.py

@@ -42,6 +42,7 @@ class CodeRenderer:
     def run(self):
         """Execute the main rendering workflow."""
         self.render_context.event_bus.publish(RenderModuleStarted(module_name=self.render_context.module_name))
+        self.render_context.run_state.current_module = self.render_context.module_name
         previous_action_payload = None
         previous_state = None
 
@@ -63,6 +64,7 @@ class CodeRenderer:
                 )
             )
             previous_state = deepcopy(self.render_context.state)
+            self.render_context.run_state.current_render_state = self.render_context.state
             self.render_context.script_execution_history.should_update_script_outputs = False
 
             self.render_context.previous_action_payload = previous_action_payload

{codeplain-0.3.2 → codeplain-0.3.3}/render_machine/render_context.py

@@ -1,6 +1,6 @@
 import threading
 from copy import deepcopy
-from typing import Optional
+from typing import Callable, Optional
 
 import file_utils
 import git_utils
@@ -162,6 +162,7 @@ class RenderContext:
             linked_resources=linked_resources,
             functional_requirement_render_attempts=0,
         )
+        self.run_state.current_frid = frid
         return
 
     def check_frid_iteration_limit(self):
@@ -195,19 +196,10 @@ class RenderContext:
     def should_run_conformance_tests(self) -> bool:
         return self.conformance_tests_script is not None
 
-    def finish_frid_implementation_step(self):
-        pass
-
     def start_unittests_processing(self):
         self.unit_tests_running_context = UnitTestsRunningContext(fix_attempts=0)
         self.run_state.increment_unittest_batch_id()
 
-    def start_unittests_processing_in_implementation(self):
-        self.start_unittests_processing()
-
-    def start_unittests_processing_in_refactoring(self):
-        self.start_unittests_processing()
-
     def _get_first_frid_conformance_test_running_context(self, module: PlainModule | None):
         conformance_tests_running_context = self.conformance_tests_running_context
 
@@ -280,9 +272,6 @@ class RenderContext:
 
         return conformance_tests_running_context
 
-    def start_unittests_processing_in_conformance_tests(self):
-        self.start_unittests_processing()
-
     def finish_unittests_processing(self):
         existing_files = file_utils.list_all_text_files(self.build_folder)
 
@@ -294,39 +283,32 @@ class RenderContext:
                 self.frid_context.changed_files.add(file_name)
         self.unit_tests_running_context.fix_attempts = 1
 
-    def finish_unittests_processing_during_implementation(self):
-        self.finish_unittests_processing()
-
-    def start_fixing_unit_tests(self):
+    def start_fixing_unit_tests(self, on_limit_exceeded: Callable):
         self.unit_tests_running_context.fix_attempts += 1
-
         if self.unit_tests_running_context.fix_attempts > MAX_UNITTEST_FIX_ATTEMPTS:
-            self.machine.dispatch(triggers.RESTART_FRID_PROCESSING)
+            on_limit_exceeded()
 
-    def start_fixing_unit_tests_in_conformance_tests(self):
-        self.unit_tests_running_context.fix_attempts += 1
+    def _on_unit_test_limit_exceeded_in_implementation(self):
+        self.machine.dispatch(triggers.RESTART_FRID_PROCESSING)
 
-        if self.unit_tests_running_context.fix_attempts > MAX_UNITTEST_FIX_ATTEMPTS:
-            self.functional_requirements_render_attempts_failed_unit_during_conformance_tests += 1
-            if (
-                self.functional_requirements_render_attempts_failed_unit_during_conformance_tests
-                >= MAX_FUNCTIONAL_REQUIREMENT_RENDER_ATTEMPTS_FAILED_UNIT_DURING_CONFORMANCE_TESTS
-            ):
-                error_msg = f"Failed to adjust unit tests after implementation code was update while fixing conformance tests for functionality {self.frid_context.frid} for the {MAX_FUNCTIONAL_REQUIREMENT_RENDER_ATTEMPTS_FAILED_UNIT_DURING_CONFORMANCE_TESTS} times."
-                self.dispatch_error(error_msg)
-            else:
-                console.info(
-                    f"Failed to adjust unit tests after implementation code was update while fixing conformance tests for functionality {self.frid_context.frid}."
-                )
-                console.info(f"Restarting rendering the functionality {self.frid_context.frid} from scratch.")
-                self.machine.dispatch(triggers.RESTART_FRID_PROCESSING)
-
-    def start_fixing_unit_tests_in_refactoring(self):
-        self.unit_tests_running_context.fix_attempts += 1
+    def _on_unit_test_limit_exceeded_in_conformance_tests(self):
+        self.functional_requirements_render_attempts_failed_unit_during_conformance_tests += 1
+        if (
+            self.functional_requirements_render_attempts_failed_unit_during_conformance_tests
+            >= MAX_FUNCTIONAL_REQUIREMENT_RENDER_ATTEMPTS_FAILED_UNIT_DURING_CONFORMANCE_TESTS
+        ):
+            error_msg = f"Failed to adjust the unit tests after implementation code was update while fixing the conformance tests for functionality {self.frid_context.frid} for the {MAX_FUNCTIONAL_REQUIREMENT_RENDER_ATTEMPTS_FAILED_UNIT_DURING_CONFORMANCE_TESTS} times."
+            self.dispatch_error(error_msg)
+        else:
+            console.info(
+                f"Failed to adjust the unit tests after implementation code was updated while fixing the conformance tests for functionality {self.frid_context.frid}."
+            )
+            console.info(f"Restarting rendering the functionality {self.frid_context.frid} from scratch.")
+            self.machine.dispatch(triggers.RESTART_FRID_PROCESSING)
 
-        if self.unit_tests_running_context.fix_attempts > MAX_UNITTEST_FIX_ATTEMPTS:
-            git_utils.revert_changes(self.build_folder)
-            self.machine.dispatch(triggers.START_NEW_REFACTORING_ITERATION)
+    def _on_unit_test_limit_exceeded_in_refactoring(self):
+        git_utils.revert_changes(self.build_folder)
+        self.machine.dispatch(triggers.START_NEW_REFACTORING_ITERATION)
 
     def start_refactoring_code(self):
 
@@ -341,9 +323,6 @@ class RenderContext:
             )
             self.machine.dispatch(triggers.PROCEED_FRID_PROCESSING)
 
-    def finish_refactoring_code(self):
-        pass
-
     def start_testing_environment_preparation(self):
         if (
             self.prepare_environment_script is None

{codeplain-0.3.2 → codeplain-0.3.3}/render_machine/state_machine_config.py

@@ -5,7 +5,6 @@ This module defines the hierarchical state machine structure, transitions, and a
 used by the CodeRenderer to orchestrate the code generation workflow.
 """
 
-from dataclasses import dataclass
 from typing import Any, Callable, Dict, List
 
 import git_utils
@@ -30,46 +29,6 @@ from render_machine.render_context import RenderContext
 from render_machine.states import States
 
 
-@dataclass
-class UnitTestsStateConfig:
-    """Dataclass for unit test state configuration."""
-
-    unit_tests_failed_on_enter_function: Callable
-    on_enter_action: Callable
-    on_exit_action: Callable
-
-
-class UnitTestsConfig:
-    """Provides configurations for different unit test scenarios."""
-
-    @staticmethod
-    def for_refactoring(render_context: RenderContext) -> UnitTestsStateConfig:
-        """Configuration for unit tests during refactoring."""
-        return UnitTestsStateConfig(
-            unit_tests_failed_on_enter_function=render_context.start_fixing_unit_tests_in_refactoring,
-            on_enter_action=render_context.start_unittests_processing_in_refactoring,
-            on_exit_action=render_context.finish_unittests_processing,
-        )
-
-    @staticmethod
-    def for_conformance_tests(render_context: RenderContext) -> UnitTestsStateConfig:
-        """Configuration for unit tests during conformance tests."""
-        return UnitTestsStateConfig(
-            unit_tests_failed_on_enter_function=render_context.start_fixing_unit_tests_in_conformance_tests,
-            on_enter_action=render_context.start_unittests_processing_in_conformance_tests,
-            on_exit_action=render_context.finish_unittests_processing,
-        )
-
-    @staticmethod
-    def for_implementation(render_context: RenderContext) -> UnitTestsStateConfig:
-        """Configuration for unit tests during initial implementation."""
-        return UnitTestsStateConfig(
-            unit_tests_failed_on_enter_function=render_context.start_fixing_unit_tests,
-            on_enter_action=render_context.start_unittests_processing_in_implementation,
-            on_exit_action=render_context.finish_unittests_processing_during_implementation,
-        )
-
-
 class StateMachineConfig:
     """Configuration class for the render state machine."""
 
@@ -137,26 +96,21 @@ class StateMachineConfig:
             AnalyzeSpecificationAmbiguity.SUCCESSFUL_OUTCOME: triggers.PROCEED_FRID_PROCESSING,
         }
 
-    def get_processing_unit_tests_states(self, config: UnitTestsStateConfig) -> Dict[str, Any]:
-        """Create the processing unit tests state configuration based on the provided configuration.
-
-        Args:
-            config: A dataclass containing the configuration for the unit test state.
-
-        Returns:
-            Dictionary defining the processing unit tests hierarchical state.
-        """
-        children = [
-            States.UNIT_TESTS_READY.value,
-            {"name": States.UNIT_TESTS_FAILED.value, "on_enter": config.unit_tests_failed_on_enter_function},
-        ]
-
+    def get_processing_unit_tests_states(
+        self, render_context: RenderContext, on_limit_exceeded: Callable
+    ) -> Dict[str, Any]:
         return {
             "name": States.PROCESSING_UNIT_TESTS.value,
             "initial": States.UNIT_TESTS_READY.value,
-            "on_enter": config.on_enter_action,
-            "on_exit": config.on_exit_action,
-            "children": children,
+            "on_enter": render_context.start_unittests_processing,
+            "on_exit": render_context.finish_unittests_processing,
+            "children": [
+                States.UNIT_TESTS_READY.value,
+                {
+                    "name": States.UNIT_TESTS_FAILED.value,
+                    "on_enter": lambda: render_context.start_fixing_unit_tests(on_limit_exceeded),
+                },
+            ],
         }
 
     def get_postprocessing_conformance_tests_states(self) -> Dict[str, Any]:
@@ -191,7 +145,9 @@ class StateMachineConfig:
                     "on_enter": render_context.start_fixing_conformance_tests,
                     "on_exit": render_context.finish_fixing_conformance_tests,
                 },
-                self.get_processing_unit_tests_states(UnitTestsConfig.for_conformance_tests(render_context)),
+                self.get_processing_unit_tests_states(
+                    render_context, render_context._on_unit_test_limit_exceeded_in_conformance_tests
+                ),
                 self.get_postprocessing_conformance_tests_states(),
             ],
         }
@@ -209,10 +165,11 @@ class StateMachineConfig:
             "name": States.REFACTORING_CODE.value,
             "initial": States.READY_FOR_REFACTORING.value,
             "on_enter": render_context.start_refactoring_code,
-            "on_exit": render_context.finish_refactoring_code,
             "children": [
                 States.READY_FOR_REFACTORING.value,
-                self.get_processing_unit_tests_states(UnitTestsConfig.for_refactoring(render_context)),
+                self.get_processing_unit_tests_states(
+                    render_context, render_context._on_unit_test_limit_exceeded_in_refactoring
+                ),
                 States.STEP_COMPLETED.value,
             ],
         }
@@ -225,12 +182,14 @@ class StateMachineConfig:
                 "on_enter": render_context.start_implementing_frid,
                 "on_exit": render_context.finish_implementing_frid,
                 "children": [
-                    {"name": States.STEP_COMPLETED.value, "on_exit": render_context.finish_frid_implementation_step},
+                    {"name": States.STEP_COMPLETED.value},
                     {
                         "name": States.READY_FOR_FRID_IMPLEMENTATION.value,
                         "on_enter": render_context.check_frid_iteration_limit,
                     },
-                    self.get_processing_unit_tests_states(UnitTestsConfig.for_implementation(render_context)),
+                    self.get_processing_unit_tests_states(
+                        render_context, render_context._on_unit_test_limit_exceeded_in_implementation
+                    ),
                     refactoring_code_states,
                     self.get_processing_conformance_tests_states(render_context),
                     States.FRID_FULLY_IMPLEMENTED.value,

{codeplain-0.3.2 → codeplain-0.3.3}/render_machine/triggers.py

@@ -7,7 +7,7 @@ that controls the code rendering process.
 
 # Trigger constants for state machine transitions
 START_RENDER = "start_render"
-RENDER_FUNCTIONAL_REQUIREMENT = "render_functional_requirement"  # todo: rename
+RENDER_FUNCTIONAL_REQUIREMENT = "render_functional_requirement"
 PROCEED_FRID_PROCESSING = "proceed_frid_processing"
 MARK_UNIT_TESTS_FAILED = "mark_unit_tests_failed"
 MARK_UNIT_TESTS_PASSED = "mark_unit_tests_passed"

{codeplain-0.3.2 → codeplain-0.3.3}/requirements.txt

@@ -10,6 +10,7 @@ pytest==8.3.4
 textual>=7.5.0
 networkx==3.6.1
 transitions==0.9.3
+sentry-sdk==2.62.0
 
 
 # Development dependencies