refactor: extract job execution logic into _run_job method

Author: hiroyukinakazato-db

src/databricks/labs/lakebridge/transpiler/switch_runner.py

@@ -83,42 +83,7 @@ def run(
         )
         logger.info(f"Triggering Switch job with job_id: {job_id}")
 
-        if wait_for_completion:
-            run = self._ws.jobs.run_now_and_wait(job_id, notebook_params=job_params)
-
-            if not run.run_id:
-                raise SystemExit(f"Job {job_id} execution failed.")
-
-            job_run_url = f"{self._ws.config.host}/jobs/{job_id}/runs/{run.run_id}"
-            logger.info(f"Switch LLM transpilation job completed: {job_run_url}")
-
-            return [
-                {
-                    "job_id": job_id,
-                    "run_id": run.run_id,
-                    "run_url": job_run_url,
-                    "state": (
-                        run.state.life_cycle_state.value if run.state and run.state.life_cycle_state else "UNKNOWN"
-                    ),
-                    "result_state": run.state.result_state.value if run.state and run.state.result_state else None,
-                }
-            ]
-
-        wait = self._ws.jobs.run_now(job_id, notebook_params=job_params)
-
-        if not wait.run_id:
-            raise SystemExit(f"Job {job_id} execution failed.")
-
-        job_run_url = f"{self._ws.config.host}/jobs/{job_id}/runs/{wait.run_id}"
-        logger.info(f"Switch LLM transpilation job started: {job_run_url}")
-
-        return [
-            {
-                "job_id": job_id,
-                "run_id": wait.run_id,
-                "run_url": job_run_url,
-            }
-        ]
+        return self._run_job(job_id, job_params, wait_for_completion)
 
     def _upload_to_volume(
         self,
@@ -180,3 +145,47 @@ def _build_job_parameters(
             "builtin_prompt": source_dialect,
             **switch_options,
         }
+
+    def _run_job(
+        self,
+        job_id: int,
+        job_params: dict[str, str],
+        wait_for_completion: bool,
+    ) -> RootJsonValue:
+        """Execute Switch job and return run information."""
+        if wait_for_completion:
+            run = self._ws.jobs.run_now_and_wait(job_id, notebook_params=job_params)
+
+            if not run.run_id:
+                raise SystemExit(f"Job {job_id} execution failed.")
+
+            job_run_url = f"{self._ws.config.host}/jobs/{job_id}/runs/{run.run_id}"
+            logger.info(f"Switch LLM transpilation job completed: {job_run_url}")
+
+            return [
+                {
+                    "job_id": job_id,
+                    "run_id": run.run_id,
+                    "run_url": job_run_url,
+                    "state": (
+                        run.state.life_cycle_state.value if run.state and run.state.life_cycle_state else "UNKNOWN"
+                    ),
+                    "result_state": run.state.result_state.value if run.state and run.state.result_state else None,
+                }
+            ]
+
+        wait = self._ws.jobs.run_now(job_id, notebook_params=job_params)
+
+        if not wait.run_id:
+            raise SystemExit(f"Job {job_id} execution failed.")
+
+        job_run_url = f"{self._ws.config.host}/jobs/{job_id}/runs/{wait.run_id}"
+        logger.info(f"Switch LLM transpilation job started: {job_run_url}")
+
+        return [
+            {
+                "job_id": job_id,
+                "run_id": wait.run_id,
+                "run_url": job_run_url,
+            }
+        ]