fix: resolve test failures by reverting circular import changes and updating AI test mocks

Author: ArthurCRodrigues

autograder/autograder.py

@@ -1,13 +1,11 @@
 import logging
 
-from sandbox_manager.manager import get_sandbox_manager
 from autograder.models.abstract.step import Step
 from autograder.models.dataclass.step_result import StepName
 from autograder.models.pipeline_execution import PipelineExecution, PipelineStatus
 from autograder.steps.step_registry import StepRegistry
 from autograder.models.dataclass.submission import Submission
 from autograder.services.template_library_service import TemplateLibraryService
-from autograder.steps.load_template_step import TemplateLoaderStep
 
 logger = logging.getLogger(__name__)
 
@@ -104,6 +102,7 @@ def _cleanup_sandbox(self, pipeline_execution: PipelineExecution) -> None:
         try:
             sandbox = pipeline_execution.sandbox
             if sandbox:
+                from sandbox_manager.manager import get_sandbox_manager
                 manager = get_sandbox_manager()
                 language = pipeline_execution.submission.language
                 manager.destroy_sandbox(language, sandbox)
@@ -159,6 +158,7 @@ def build_pipeline(  # pylint: disable=too-many-arguments,too-many-locals
         templates.append(template_service.load_custom_template(custom_template))
     elif template_name:
         # Normalize template names (can be string, comma-separated string, or list)
+        from autograder.steps.load_template_step import TemplateLoaderStep
         names = TemplateLoaderStep.normalize_template_names(template_name)
         for name in names:
             templates.append(template_service.load_builtin_template(name))

autograder/services/sandbox_service.py

@@ -1,6 +1,5 @@
 import logging
 from typing import Optional, Any
-from sandbox_manager.manager import get_sandbox_manager
 from autograder.models.dataclass.submission import Submission
 from sandbox_manager.sandbox_container import SandboxContainer
 from sandbox_manager.models.sandbox_models import Language, ResponseCategory, CommandResponse
@@ -28,6 +27,7 @@ def create_sandbox(self, submission: Submission) -> Optional[SandboxContainer]:
             raise ValueError("Submission language is required for sandbox creation")
 
         try:
+            from sandbox_manager.manager import get_sandbox_manager
             sandbox_manager = get_sandbox_manager()
             sandbox = sandbox_manager.get_sandbox(submission.language)
             self.logger.debug("Sandbox created for language %s", submission.language)
@@ -53,6 +53,7 @@ def release_sandbox(self, language: Language, sandbox: SandboxContainer):
         Releases a sandbox back to the manager pool.
         """
         try:
+            from sandbox_manager.manager import get_sandbox_manager
             sandbox_manager = get_sandbox_manager()
             sandbox_manager.release_sandbox(language, sandbox)
             self.logger.info("Sandbox released for language %s", language)

tests/unit/pipeline/test_ai_batch_step.py

@@ -127,7 +127,7 @@ def test_fallback_called_when_no_precomputed(self):
         )
 
         with patch(
-            "autograder.utils.executors.ai_executor.AiExecutor"
+            "autograder.models.abstract.ai_test_function.AiExecutor"
         ) as mock_executor:
             mock_executor.return_value.run.return_value = {"ai_code_review": fallback_result}
             result = func.execute(files=[], sandbox=None)
@@ -139,7 +139,7 @@ def test_fallback_returns_zero_result_on_empty_api_response(self):
         func = _ConcreteAiTest()
 
         with patch(
-            "autograder.utils.executors.ai_executor.AiExecutor"
+            "autograder.models.abstract.ai_test_function.AiExecutor"
         ) as mock_executor:
             mock_executor.return_value.run.return_value = {}
             result = func.execute(files=None, sandbox=None)