fix: Standardize code formatting and improve readability across multiple files

- Added missing commas in field descriptions within `exec.py`, `files.py`, and `state_archival.py` to ensure proper syntax.
- Reformatted multi-line expressions in `file.py`, `orchestrator.py`, and `runner.py` for better readability.
- Updated test functions in `test_exec_languages.py`, `test_exec_workflow.py`, and `test_files.py` to follow consistent formatting practices.
- Cleaned up unnecessary blank lines and improved alignment in various test files to enhance overall code clarity.

Author: usnavy13

src/models/exec.py

@@ -25,7 +25,7 @@ class RequestFile(BaseModel):
     name: str
     restore_state: bool = Field(
         default=False,
-        description="If true, restore Python state from when this file was last used"
+        description="If true, restore Python state from when this file was last used",
     )
 
 

src/models/files.py

@@ -42,16 +42,15 @@ class FileInfo(BaseModel):
     path: str = Field(..., description="File path in the session")
     # State restoration fields (for Python state-file linking)
     execution_id: Optional[str] = Field(
-        default=None,
-        description="ID of the execution that created/last used this file"
+        default=None, description="ID of the execution that created/last used this file"
     )
     state_hash: Optional[str] = Field(
         default=None,
-        description="SHA256 hash of the Python state when this file was last used"
+        description="SHA256 hash of the Python state when this file was last used",
     )
     last_used_at: Optional[datetime] = Field(
         default=None,
-        description="Timestamp of when this file was last used in an execution"
+        description="Timestamp of when this file was last used in an execution",
     )
 
     class Config:

src/services/execution/runner.py

@@ -191,8 +191,11 @@ async def execute(
             else:
                 # Standard execution (no state persistence)
                 exit_code, stdout, stderr = await self._execute_code_in_container(
-                    container, request.code, request.language, request.timeout,
-                    args=request.args
+                    container,
+                    request.code,
+                    request.language,
+                    request.timeout,
+                    args=request.args,
                 )
             end_time = datetime.utcnow()
 
@@ -465,7 +468,9 @@ async def _execute_code_in_container(
             logger.debug(
                 "Using REPL executor", container_id=container.id[:12], language=language
             )
-            return await self._execute_via_repl(container, code, execution_timeout, args=args)
+            return await self._execute_via_repl(
+                container, code, execution_timeout, args=args
+            )
 
         # Standard execution path for non-REPL containers
         exec_command = lang_config.execution_command
@@ -539,8 +544,11 @@ def _is_repl_container(self, container: Container, language: str) -> bool:
             return False
 
     async def _execute_via_repl(
-        self, container: Container, code: str, timeout: int,
-        args: Optional[List[str]] = None
+        self,
+        container: Container,
+        code: str,
+        timeout: int,
+        args: Optional[List[str]] = None,
     ) -> Tuple[int, str, str]:
         """Execute code via REPL server in container.
 

src/services/file.py

@@ -591,7 +591,9 @@ async def store_uploaded_file(
                 "size": len(content),
                 "path": f"/{filename}",
                 "type": "upload",  # Mark as uploaded file
-                "is_agent_file": "1" if is_agent_file else "0",  # Read-only if agent file
+                "is_agent_file": (
+                    "1" if is_agent_file else "0"
+                ),  # Read-only if agent file
             }
 
             await self._store_file_metadata(session_id, file_id, metadata)
@@ -744,9 +746,7 @@ async def cleanup_orphan_objects(self, batch_limit: int = 1000) -> int:
             logger.error("Orphan MinIO objects cleanup failed", error=str(e))
             return 0
 
-    async def get_file_state_hash(
-        self, session_id: str, file_id: str
-    ) -> Optional[str]:
+    async def get_file_state_hash(self, session_id: str, file_id: str) -> Optional[str]:
         """Get the state hash associated with a file.
 
         Args:

src/services/orchestrator.py

@@ -335,7 +335,9 @@ async def _mount_explicit_files(
         mounted = []
         mounted_ids = set()
         file_refs = []  # Track for state-file linking
-        restore_state_hash = None  # Hash of state to restore (from first restore_state file)
+        restore_state_hash = (
+            None  # Hash of state to restore (from first restore_state file)
+        )
 
         for file_ref in ctx.request.files:
             # Get file info
@@ -374,10 +376,12 @@ async def _mount_explicit_files(
             mounted_ids.add(key)
 
             # Track file reference for state-file linking
-            file_refs.append({
-                "session_id": file_ref.session_id,
-                "file_id": file_info.file_id,
-            })
+            file_refs.append(
+                {
+                    "session_id": file_ref.session_id,
+                    "file_id": file_info.file_id,
+                }
+            )
 
             # Check for restore_state flag (only for Python, use first file's state)
             if (
@@ -443,10 +447,12 @@ async def _auto_mount_session_files(
             mounted_ids.add(key)
 
             # Track file reference for state-file linking
-            file_refs.append({
-                "session_id": ctx.session_id,
-                "file_id": file_info.file_id,
-            })
+            file_refs.append(
+                {
+                    "session_id": ctx.session_id,
+                    "file_id": file_info.file_id,
+                }
+            )
 
         # Store file refs for later state_hash update
         ctx.mounted_file_refs = file_refs
@@ -461,9 +467,7 @@ async def _auto_mount_session_files(
 
         return mounted
 
-    async def _load_state_by_hash(
-        self, ctx: ExecutionContext, state_hash: str
-    ) -> None:
+    async def _load_state_by_hash(self, ctx: ExecutionContext, state_hash: str) -> None:
         """Load state by its hash for state-file restoration.
 
         Tries Redis first, then MinIO cold storage.
@@ -472,9 +476,15 @@ async def _load_state_by_hash(
             # Try Redis first
             state = await self.state_service.get_state_by_hash(state_hash)
 
-            if not state and self.state_archival_service and settings.state_archive_enabled:
+            if (
+                not state
+                and self.state_archival_service
+                and settings.state_archive_enabled
+            ):
                 # Try MinIO cold storage
-                state = await self.state_archival_service.restore_state_by_hash(state_hash)
+                state = await self.state_archival_service.restore_state_by_hash(
+                    state_hash
+                )
 
             if state:
                 ctx.initial_state = state
@@ -820,11 +830,13 @@ async def _handle_generated_files(self, ctx: ExecutionContext) -> List[FileRef]:
                     state_hash=ctx.new_state_hash,  # Link file to current state
                 )
 
-                generated.append(FileRef(
-                    id=file_id,
-                    name=filename,
-                    session_id=ctx.session_id,  # Include for cross-message persistence
-                ))
+                generated.append(
+                    FileRef(
+                        id=file_id,
+                        name=filename,
+                        session_id=ctx.session_id,  # Include for cross-message persistence
+                    )
+                )
                 logger.info(
                     "Generated file stored",
                     session_id=ctx.session_id,

src/services/state_archival.py

@@ -415,9 +415,7 @@ def _get_state_by_hash_object_key(self, state_hash: str) -> str:
         """Generate MinIO object key for a hash-indexed state."""
         return f"{self.STATE_BY_HASH_PREFIX}/{state_hash}/state.dat"
 
-    async def archive_state_by_hash(
-        self, state_hash: str, state_data: str
-    ) -> bool:
+    async def archive_state_by_hash(self, state_hash: str, state_data: str) -> bool:
         """Archive a state indexed by its hash to MinIO.
 
         Args:
@@ -498,7 +496,9 @@ async def restore_state_by_hash(self, state_hash: str) -> Optional[str]:
                 response.release_conn()
             except S3Error as e:
                 if e.code == "NoSuchKey":
-                    logger.debug("No archived state found by hash", hash=state_hash[:12])
+                    logger.debug(
+                        "No archived state found by hash", hash=state_hash[:12]
+                    )
                     return None
                 raise
 

tests/functional/conftest.py

@@ -34,7 +34,7 @@
     "ts": ("console.log('ts: sum(1..10)=' + (1+2+3+4+5+6+7+8+9+10));", "55"),
     "go": (
         'package main\n\nimport (\n\t"fmt"\n)\n\nfunc main() {\n\ts := 0\n\t'
-        'for i := 1; i <= 10; i++ {\n\t\ts += i\n\t}\n\t'
+        "for i := 1; i <= 10; i++ {\n\t\ts += i\n\t}\n\t"
         'fmt.Printf("go: sum(1..10)=%d\\n", s)\n}',
         "55",
     ),
@@ -44,7 +44,7 @@
         "55",
     ),
     "c": (
-        '#include <stdio.h>\nint main(){int s=0; for(int i=1;i<=10;i++) s+=i; '
+        "#include <stdio.h>\nint main(){int s=0; for(int i=1;i<=10;i++) s+=i; "
         'printf("c: sum(1..10)=%d\\n", s); return 0;}',
         "55",
     ),

tests/functional/test_exec_languages.py

@@ -29,9 +29,9 @@ async def test_language_execution(
         latency = time.perf_counter() - start
 
         # Basic assertions
-        assert response.status_code == 200, (
-            f"Failed for {language_test_case['lang']}: {response.text}"
-        )
+        assert (
+            response.status_code == 200
+        ), f"Failed for {language_test_case['lang']}: {response.text}"
 
         data = response.json()
 
@@ -61,7 +61,9 @@ class TestPythonExecution:
     """Specific tests for Python execution features."""
 
     @pytest.mark.asyncio
-    async def test_python_with_imports(self, async_client, auth_headers, unique_entity_id):
+    async def test_python_with_imports(
+        self, async_client, auth_headers, unique_entity_id
+    ):
         """Test Python execution with standard library imports."""
         response = await async_client.post(
             "/exec",
@@ -78,7 +80,9 @@ async def test_python_with_imports(self, async_client, auth_headers, unique_enti
         assert '{"ok": true}' in stdout or "{'ok': true}" in stdout.replace('"', "'")
 
     @pytest.mark.asyncio
-    async def test_python_with_numpy(self, async_client, auth_headers, unique_entity_id):
+    async def test_python_with_numpy(
+        self, async_client, auth_headers, unique_entity_id
+    ):
         """Test Python execution with NumPy."""
         response = await async_client.post(
             "/exec",
@@ -94,7 +98,9 @@ async def test_python_with_numpy(self, async_client, auth_headers, unique_entity
         assert "mean=3.0" in response.json()["stdout"]
 
     @pytest.mark.asyncio
-    async def test_python_error_in_stderr(self, async_client, auth_headers, unique_entity_id):
+    async def test_python_error_in_stderr(
+        self, async_client, auth_headers, unique_entity_id
+    ):
         """Test that Python errors appear in stderr, not as HTTP error."""
         response = await async_client.post(
             "/exec",

tests/functional/test_exec_workflow.py

@@ -14,7 +14,11 @@ async def test_execution_creates_session(
         response = await async_client.post(
             "/exec",
             headers=auth_headers,
-            json={"code": "print('hello')", "lang": "py", "entity_id": unique_entity_id},
+            json={
+                "code": "print('hello')",
+                "lang": "py",
+                "entity_id": unique_entity_id,
+            },
         )
 
         assert response.status_code == 200
@@ -170,7 +174,11 @@ async def test_exec_response_includes_state_fields(
         r = await async_client.post(
             "/exec",
             headers=auth_headers,
-            json={"code": "data = [1,2,3]", "lang": "py", "entity_id": unique_entity_id},
+            json={
+                "code": "data = [1,2,3]",
+                "lang": "py",
+                "entity_id": unique_entity_id,
+            },
         )
 
         assert r.status_code == 200

tests/functional/test_files.py

@@ -7,7 +7,9 @@ class TestFileUpload:
     """Test POST /upload."""
 
     @pytest.mark.asyncio
-    async def test_upload_single_file(self, async_client, auth_headers, unique_entity_id):
+    async def test_upload_single_file(
+        self, async_client, auth_headers, unique_entity_id
+    ):
         """Upload a single file using 'files' field."""
         files = {"files": ("test.txt", b"Hello World", "text/plain")}
         data = {"entity_id": unique_entity_id}