(refactor) Apply pre-commit fixes across the workspace

Author: MRiabov

config/skills_repo.lock.json

@@ -2,6 +2,6 @@
   "branch": "main",
   "head_commit": "8d02a1c62f0fa1bf96210b2e7a3525538a109cab",
   "origin_url": "https://github.com/MRiabov/Problemologist-skills",
-  "recorded_at_utc": "2026-03-30T06:51:16Z",
+  "recorded_at_utc": "2026-03-30T06:54:20Z",
   "skills_repo_path": "skills"
 }

controller/agent/node_entry_validation.py

@@ -472,9 +472,9 @@ async def engineer_benchmark_handover_custom_check(
     contract: NodeEntryContract,  # noqa: ARG001
     state: BaseModel | Mapping[str, Any],
 ) -> list[NodeEntryValidationError]:
-    worker_session_id = _get_state_value(state, "worker_session_id") or _get_state_value(
-        state, "session_id"
-    )
+    worker_session_id = _get_state_value(
+        state, "worker_session_id"
+    ) or _get_state_value(state, "session_id")
     episode_id = _get_state_value(state, "episode_id")
     if not worker_session_id:
         return [

controller/agent/nodes/base.py

@@ -29,14 +29,14 @@
     evaluate_agent_hard_fail,
     mark_episode_execution_window_start,
 )
+from controller.agent.prompt_manager import PromptManager
+from controller.agent.provider_tool_call_adapters import extract_tool_calls
 from controller.agent.runtime_models import (
     FileListEntry,
     MessageContentBlock,
     NativeProviderMessage,
     ProviderResponseEnvelope,
 )
-from controller.agent.prompt_manager import PromptManager
-from controller.agent.provider_tool_call_adapters import extract_tool_calls
 from controller.clients.worker import WorkerClient
 from controller.middleware.remote_fs import RemoteFilesystemMiddleware
 from controller.observability.database import DatabaseCallbackHandler
@@ -880,10 +880,7 @@ def _coerce_message_text(content: Any) -> str:
                 if isinstance(block_model.text, str) and block_model.text.strip():
                     text_chunks.append(block_model.text.strip())
                     continue
-                if (
-                    isinstance(block_model.content, str)
-                    and block_model.content.strip()
-                ):
+                if isinstance(block_model.content, str) and block_model.content.strip():
                     text_chunks.append(block_model.content.strip())
             if text_chunks:
                 return "\n".join(text_chunks)
@@ -936,7 +933,9 @@ def _assistant_message_with_tool_calls(
         with suppress(Exception):
             message_model = NativeProviderMessage.model_validate(message)
         provider_specific_fields = (
-            message_model.provider_specific_fields if message_model is not None else None
+            message_model.provider_specific_fields
+            if message_model is not None
+            else None
         )
         if isinstance(provider_specific_fields, dict) and provider_specific_fields:
             assistant_message["provider_specific_fields"] = provider_specific_fields

controller/agent/provider_tool_call_adapters/native_adapter.py

@@ -1,13 +1,14 @@
 from contextlib import suppress
 from typing import Any
 
-from .base import ParsedToolCalls
 from controller.agent.runtime_models import (
     NativeProviderMessage,
     NativeToolCall,
     NativeToolCallFunction,
 )
 
+from .base import ParsedToolCalls
+
 
 class NativeToolCallAdapter:
     name = "native"
@@ -42,10 +43,7 @@ def extract(
                 continue
             function = tool_call_model.function or NativeToolCallFunction()
             tool_name = (
-                function.name
-                or tool_call_model.name
-                or tool_call_model.tool_name
-                or ""
+                function.name or tool_call_model.name or tool_call_model.tool_name or ""
             ).strip()
             raw_arguments = (
                 function.arguments

controller/agent/runtime_models.py

@@ -55,7 +55,9 @@ class ProviderChoice(BaseModel):
 
 
 class ProviderResponseEnvelope(BaseModel):
-    model_config = ConfigDict(from_attributes=True, populate_by_name=True, extra="ignore")
+    model_config = ConfigDict(
+        from_attributes=True, populate_by_name=True, extra="ignore"
+    )
 
     response_ms: float | int | None = None
     hidden_params: dict[str, Any] | None = Field(default=None, alias="_hidden_params")

controller/middleware/remote_fs.py

@@ -12,6 +12,7 @@
 from temporalio.common import WorkflowIDConflictPolicy
 from temporalio.exceptions import WorkflowAlreadyStartedError
 
+from controller.agent.runtime_models import FileListEntry
 from controller.clients.worker import WorkerClient
 from controller.observability.middleware_helper import (
     broadcast_file_update,
@@ -28,7 +29,6 @@
     HeavyValidationWorkflow,
     HeavyVerifyWorkflow,
 )
-from controller.agent.runtime_models import FileListEntry
 from shared.agents.config import resolve_agents_config_path
 from shared.enums import AgentName, ManufacturingMethod
 from shared.observability.schemas import (

tests/integration/agent/p1/test_int_181_to_183.py

@@ -14,12 +14,12 @@
 from tests.integration.agent.helpers import (
     CONTROLLER_URL,
     get_controller_log_path,
+    integration_workspace_session_id,
     read_log_segment,
     run_agent_episode,
     seed_benchmark_assembly_definition,
     seed_execution_reviewer_handover,
     strip_ansi,
-    integration_workspace_session_id,
     wait_for_episode_terminal,
     wait_for_queue_empty,
 )

tests/integration/architecture_p0/test_codex_runner_mode.py

@@ -266,7 +266,9 @@ async def fail_reviewer_chain_for_judge(**_: object) -> list[dict[str, object]]:
 
 
 @pytest.mark.integration_p0
-def test_run_evals_codex_readable_logs_mirror_imported_transcript(tmp_path, monkeypatch):
+def test_run_evals_codex_readable_logs_mirror_imported_transcript(
+    tmp_path, monkeypatch
+):
     readable_log = tmp_path / "readable_agent_logs.log"
     session_root = tmp_path / "sessions"
     transcript_path = tmp_path / "transcript.log"
@@ -302,7 +304,10 @@ def test_run_evals_codex_readable_logs_mirror_imported_transcript(tmp_path, monk
 
     assert "CODEX_SESSION_TRACE_IMPORTED session_id=codex-123" in readable_text
     assert "SESSION_META id=codex-123 cwd=/workspace" in readable_text
-    assert "MESSAGE role=user phase=prompt text=Workspace: current directory" in readable_text
+    assert (
+        "MESSAGE role=user phase=prompt text=Workspace: current directory"
+        in readable_text
+    )
     assert 'TOOL_CALL python args={"code":"print(1)"}' in readable_text
     assert readable_text == session_text
 

worker_heavy/simulation/loop.py

@@ -29,8 +29,8 @@
 from worker_heavy.simulation.factory import get_physics_backend
 from worker_heavy.simulation.media import MediaRecorder
 from worker_heavy.simulation.metrics import MetricCollector
-from worker_heavy.simulation.objectives import ObjectiveEvaluator
 from worker_heavy.simulation.naming import moved_object_scene_name
+from worker_heavy.simulation.objectives import ObjectiveEvaluator
 from worker_heavy.utils.dfm import (
     resolve_requested_quantity,
     validate_and_price_assembly,

worker_heavy/simulation/mujoco_backend.py

@@ -2,6 +2,8 @@
 import threading
 from typing import Any
 
+import structlog
+
 from worker_heavy.simulation._mujoco_env import ensure_headless_mujoco
 
 ensure_headless_mujoco()
@@ -10,6 +12,9 @@
 import numpy as np
 
 from shared.enums import FailureReason
+
+# ... (rest of imports)
+logger = structlog.get_logger(__name__)
 from shared.models.simulation import (
     FluidMetricResult,
     RendererCapabilities,
@@ -303,7 +308,9 @@ def render_camera_modalities(
                 include_segmentation=include_segmentation,
                 session_id=self.session_id,
                 rgb_shape=tuple(frame.shape) if frame is not None else None,
-                depth_shape=tuple(depth_frame.shape) if depth_frame is not None else None,
+                depth_shape=tuple(depth_frame.shape)
+                if depth_frame is not None
+                else None,
                 segmentation_shape=(
                     tuple(segmentation_frame.shape)
                     if segmentation_frame is not None