refactor: Improve code readability and formatting in configuration and service files

- Enhanced the `Settings` class in `__init__.py` for better clarity in Docker image retrieval logic.
- Reformatted the `get_image_for_language` function in `languages.py` for improved readability.
- Adjusted the `ApiKeyRecord` class in `api_key.py` to streamline dictionary comprehension.
- Improved formatting in the `HealthCheckService` class in `health.py` for better alignment.
- Refactored the `ExecutionContext` class in `orchestrator.py` to enhance code structure.

Author: usnavy13

src/config/__init__.py

@@ -399,12 +399,14 @@ def _set_supported_languages(cls, v, values):
         """Initialize supported_languages with registry-prefixed images."""
         if v:
             return v
-        
+
         registry = values.get("docker_image_registry", "code-interpreter")
         tag = values.get("docker_image_tag", "latest")
         return {
             code: {
-                "image": f"{registry}/{lang.image.rsplit(':', 1)[0]}:{tag}" if registry else f"{lang.image.rsplit(':', 1)[0]}:{tag}",
+                "image": f"{registry}/{lang.image.rsplit(':', 1)[0]}:{tag}"
+                if registry
+                else f"{lang.image.rsplit(':', 1)[0]}:{tag}",
                 "timeout_multiplier": lang.timeout_multiplier,
                 "memory_multiplier": lang.memory_multiplier,
             }
@@ -601,10 +603,13 @@ def get_image_for_language(self, code: str) -> str:
         config = self.get_language_config(code)
         if config and "image" in config:
             return config["image"]
-        
+
         # Fallback to languages.py logic if not in settings
         from .languages import get_image_for_language as get_img
-        return get_img(code, registry=self.docker_image_registry, tag=self.docker_image_tag)
+
+        return get_img(
+            code, registry=self.docker_image_registry, tag=self.docker_image_tag
+        )
 
     def get_execution_timeout(self, language: str) -> int:
         """Get execution timeout for a specific language."""

src/config/languages.py

@@ -183,7 +183,9 @@ def is_supported_language(code: str) -> bool:
 
 
 # Convenience lookups for backward compatibility during transition
-def get_image_for_language(code: str, registry: Optional[str] = None, tag: str = "latest") -> str:
+def get_image_for_language(
+    code: str, registry: Optional[str] = None, tag: str = "latest"
+) -> str:
     """Get Docker image for a language."""
     lang = get_language(code)
     if lang:

src/models/api_key.py

@@ -111,9 +111,9 @@ def from_redis_hash(cls, data: Dict[bytes, bytes]) -> "ApiKeyRecord":
         """Create from Redis hash data (bytes keys/values)."""
         # Decode bytes to strings
         decoded = {
-            k.decode() if isinstance(k, bytes) else k: (
-                v.decode() if isinstance(v, bytes) else v
-            )
+            k.decode()
+            if isinstance(k, bytes)
+            else k: (v.decode() if isinstance(v, bytes) else v)
             for k, v in data.items()
         }
 

src/services/health.py

@@ -404,7 +404,9 @@ async def check_docker(self) -> HealthCheckResult:
                 "running_containers": running_containers,
                 "registry_accessible": registry_accessible,
                 "server_version": system_info.get("ServerVersion", "unknown"),
-                "memory_total_gb": round(system_info.get("MemTotal", 0) / (1024**3), 2),
+                "memory_total_gb": round(
+                    system_info.get("MemTotal", 0) / (1024**3), 2
+                ),
                 "cpu_count": system_info.get("NCPU", 0),
             }
 

src/services/orchestrator.py

@@ -62,9 +62,9 @@ class ExecutionContext:
     generated_files: Optional[List[FileRef]] = None
     stdout: str = ""
     stderr: str = ""
-    container: Optional[Any] = (
-        None  # Container used for execution (avoids session lookup)
-    )
+    container: Optional[
+        Any
+    ] = None  # Container used for execution (avoids session lookup)
     # State persistence fields
     initial_state: Optional[str] = None
     new_state: Optional[str] = None