Correct format for the failed format checks.

Author: SHRUTI6991

src/google/adk/code_executors/gke_code_executor.py

@@ -18,7 +18,6 @@
 import uuid
 
 from agentic_sandbox import SandboxClient
-
 import kubernetes as k8s
 from kubernetes.watch import Watch
 
@@ -92,7 +91,7 @@ class GkeCodeExecutor(BaseCodeExecutor):
 
   kubeconfig_path: str | None = None
   kubeconfig_context: str | None = None
-  
+
   # Sandbox constants
   sandbox_template: str = "python-sandbox-template"
 
@@ -122,9 +121,10 @@ def __init__(
     self.kubeconfig_path = kubeconfig_path
     self.kubeconfig_context = kubeconfig_context
 
-    if executor_type not in ['job', 'sandbox']:
+    if executor_type not in ["job", "sandbox"]:
       raise ValueError(
-          f"Invalid executor_type: '{executor_type}'. Must be 'job' or 'sandbox'."
+          f"Invalid executor_type: '{executor_type}'. Must be 'job' or"
+          " 'sandbox'."
       )
     if self.kubeconfig_path:
       try:
@@ -166,24 +166,26 @@ def __init__(
   def _execute_in_sandbox(self, code: str) -> CodeExecutionResult:
     """Executes code using Agent Sandbox Client."""
     try:
-        with SandboxClient(
-            template_name=self.sandbox_template,
-            gateway_name=self.sandbox_gateway_name,
-            namespace=self.namespace
-        ) as sandbox:
-            # Execute the code as a python script
-            logger.debug("Executing code in sandbox:\n```\n%s\n```", code)
-            sandbox.write("script.py", code)
-            result = sandbox.run("python3 script.py")
-
-            return CodeExecutionResult(stdout=result.stdout)
+      with SandboxClient(
+          template_name=self.sandbox_template,
+          gateway_name=self.sandbox_gateway_name,
+          namespace=self.namespace,
+      ) as sandbox:
+        # Execute the code as a python script
+        logger.debug("Executing code in sandbox:\n```\n%s\n```", code)
+        sandbox.write("script.py", code)
+        result = sandbox.run("python3 script.py")
+
+        return CodeExecutionResult(stdout=result.stdout)
     except Exception as e:
-        logger.error("Sandbox execution failed", exc_info=True)
-        return CodeExecutionResult(
-            stderr=f"Sandbox execution failed: {str(e)}",
-        )
+      logger.error("Sandbox execution failed", exc_info=True)
+      return CodeExecutionResult(
+          stderr=f"Sandbox execution failed: {str(e)}",
+      )
 
-  def _execute_as_job(self, code: str, invocation_context: InvocationContext) -> CodeExecutionResult:
+  def _execute_as_job(
+      self, code: str, invocation_context: InvocationContext
+  ) -> CodeExecutionResult:
     """Orchestrates the secure execution of a code snippet on GKE."""
     job_name = f"adk-exec-{uuid.uuid4().hex[:10]}"
     configmap_name = f"code-src-{job_name}"
@@ -234,14 +236,13 @@ def execute_code(
       invocation_context: InvocationContext,
       code_execution_input: CodeExecutionInput,
   ) -> CodeExecutionResult:
-
     """Overrides the base method to route execution based on executor_type."""
     code = code_execution_input.code
     if self.executor_type == "sandbox":
-        return self._execute_in_sandbox(code)
+      return self._execute_in_sandbox(code)
     else:
-        # Fallback to existing GKE Job logic
-        return self._execute_as_job(code, invocation_context)
+      # Fallback to existing GKE Job logic
+      return self._execute_as_job(code, invocation_context)
 
   def _create_job_manifest(
       self,

tests/unittests/code_executors/test_gke_code_executor.py

@@ -330,4 +330,4 @@ def test_execute_code_forks_to_job(
     mock_k8s_clients["batch_v1"].create_namespaced_job.assert_called_once()
 
     # Verify SandboxClient was NOT used
-    mock_sandbox_client.assert_not_called()
+    mock_sandbox_client.assert_not_called()