add stdour and stderr for py files

Author: andrii-i

jupyter_scheduler/environments.py

@@ -66,7 +66,12 @@ def manage_environments_command(self) -> str:
         return ""
 
     def output_formats_mapping(self) -> Dict[str, str]:
-        return {"ipynb": "Notebook", "html": "HTML"}
+        return {
+            "ipynb": "Notebook",
+            "html": "HTML",
+            "stdout": "Output",
+            "stderr": "Errors",
+        }
 
 
 class StaticEnvironmentManager(EnvironmentManager):
@@ -90,7 +95,12 @@ def manage_environments_command(self) -> str:
         return ""
 
     def output_formats_mapping(self) -> Dict[str, str]:
-        return {"ipynb": "Notebook", "html": "HTML"}
+        return {
+            "ipynb": "Notebook",
+            "html": "HTML",
+            "stdout": "Output",
+            "stderr": "Errors",
+        }
 
 
 class EnvironmentRetrievalError(Exception):

jupyter_scheduler/handlers.py

@@ -308,6 +308,10 @@ async def post(self):
                 # Ensure backend ID is stored with the job
                 payload["backend"] = backend.config.id
                 scheduler = backend.scheduler
+
+                # Set default output_formats for Python backend
+                if backend.config.id == "jupyter_server_py" and not payload.get("output_formats"):
+                    payload["output_formats"] = ["stdout", "stderr"]
             else:
                 # Fallback to default scheduler (backwards compatibility)
                 scheduler = self.scheduler

jupyter_scheduler/python_executor.py

@@ -33,18 +33,15 @@ def execute(self):
             env=env,
         )
 
-        # Capture side effect files (same pattern as DefaultExecutionManager)
-        self.add_side_effects_files(staging_dir)
-
-        # Write stdout/stderr to staging directory
-        stdout_path = os.path.join(staging_dir, "stdout.log")
-        stderr_path = os.path.join(staging_dir, "stderr.log")
-
-        with fsspec.open(stdout_path, "w", encoding="utf-8") as f:
+        # Write stdout/stderr to staging_paths (keys match output_formats)
+        with fsspec.open(self.staging_paths["stdout"], "w", encoding="utf-8") as f:
             f.write(result.stdout)
-        with fsspec.open(stderr_path, "w", encoding="utf-8") as f:
+        with fsspec.open(self.staging_paths["stderr"], "w", encoding="utf-8") as f:
             f.write(result.stderr)
 
+        # Capture any additional side effect files AFTER writing stdout/stderr
+        self.add_side_effects_files(staging_dir)
+
         if result.returncode != 0:
             raise RuntimeError(
                 f"Script exited with code {result.returncode}\nstderr: {result.stderr[:500]}"