experimaestro 2.0.0b4__py3-none-any.whl → 2.0.0b17__py3-none-any.whl

experimaestro/scheduler/base.py

@@ -1,17 +1,34 @@
+import json
 import threading
 import time
+from datetime import datetime
+from pathlib import Path
 from typing import (
+    Dict,
+    List,
     Optional,
     Set,
     ClassVar,
     TYPE_CHECKING,
 )
 import asyncio
-from typing import Dict
 
 from experimaestro.scheduler import experiment
-from experimaestro.scheduler.jobs import Job, JobState, JobError
+from experimaestro.scheduler.jobs import Job, JobState, JobError, JobDependency
 from experimaestro.scheduler.services import Service
+from experimaestro.scheduler.interfaces import (
+    BaseJob,
+    BaseExperiment,
+    BaseService,
+)
+from experimaestro.scheduler.state_provider import StateProvider
+from experimaestro.scheduler.state_status import (
+    EventReader,
+    JobProgressEvent,
+    JobStateChangedEvent,
+    WatchedDirectory,
+    job_entity_id_extractor,
+)
 
 
 from experimaestro.utils import logger
@@ -19,7 +36,7 @@ from experimaestro.utils.asyncio import asyncThreadcheck
 import concurrent.futures
 
 if TYPE_CHECKING:
-    from experimaestro.server import Server
+    from experimaestro.webui import WebUIServer
     from experimaestro.settings import ServerSettings
     from experimaestro.scheduler.workspace import Workspace
 
@@ -36,18 +53,25 @@ class Listener:
         pass
 
 
-class Scheduler(threading.Thread):
-    """A job scheduler (singleton)
+class Scheduler(StateProvider, threading.Thread):
+    """A job scheduler (singleton) that provides live state
 
     The scheduler is based on asyncio for easy concurrency handling.
     This is a singleton - only one scheduler instance exists per process.
+
+    Inherits from StateProvider to allow TUI/Web interfaces to access
+    live job and experiment state during experiment execution.
     """
 
     _instance: ClassVar[Optional["Scheduler"]] = None
     _lock: ClassVar[threading.Lock] = threading.Lock()
 
+    #: Scheduler is always live
+    is_live: bool = True
+
     def __init__(self, name: str = "Global"):
-        super().__init__(name=f"Scheduler ({name})", daemon=True)
+        StateProvider.__init__(self)  # Initialize state listener management
+        threading.Thread.__init__(self, name=f"Scheduler ({name})", daemon=True)
         self._ready = threading.Event()
 
         # Name of the scheduler
@@ -62,10 +86,19 @@ class Scheduler(threading.Thread):
         # List of all jobs
         self.jobs: Dict[str, "Job"] = {}
 
+        # Services: (experiment_id, run_id) -> {service_id -> Service}
+        self.services: Dict[tuple[str, str], Dict[str, Service]] = {}
+
+        # Tags map: (experiment_id, run_id) -> {job_id -> {tag_key: tag_value}}
+        self._tags_map: dict[tuple[str, str], dict[str, dict[str, str]]] = {}
+
+        # Dependencies map: (experiment_id, run_id) -> {job_id -> [depends_on_job_ids]}
+        self._dependencies_map: dict[tuple[str, str], dict[str, list[str]]] = {}
+
         # List of jobs
         self.waitingjobs: Set[Job] = set()
 
-        # Listeners with thread-safe access
+        # Legacy listeners with thread-safe access
         self._listeners: Set[Listener] = set()
         self._listeners_lock = threading.Lock()
 
@@ -75,7 +108,12 @@ class Scheduler(threading.Thread):
         )
 
         # Server (managed by scheduler)
-        self.server: Optional["Server"] = None
+        self.server: Optional["WebUIServer"] = None
+
+        # Job event readers per workspace
+        # Uses EventReader to watch .events/jobs/ directory
+        self._job_event_readers: Dict[Path, EventReader] = {}
+        self._job_event_readers_lock = threading.Lock()
 
     @staticmethod
     def has_instance() -> bool:
@@ -114,51 +152,58 @@ class Scheduler(threading.Thread):
 
     def register_experiment(self, xp: "experiment"):
         """Register an experiment with the scheduler"""
-        # Use experiment name as key for now
-        key = xp.workdir.name
+        # Use experiment name as key (not workdir.name which is now run_id)
+        key = xp.name
         self.experiments[key] = xp
 
+        # Start watching job events for this workspace
+        self._start_job_event_reader(xp.workspace.path)
+
         logger.debug("Registered experiment %s with scheduler", key)
 
     def unregister_experiment(self, xp: "experiment"):
         """Unregister an experiment from the scheduler"""
-        key = xp.workdir.name
+        key = xp.name
         if key in self.experiments:
             del self.experiments[key]
             logger.debug("Unregistered experiment %s from scheduler", key)
 
     def start_server(
-        self, settings: "ServerSettings" = None, workspace: "Workspace" = None
+        self,
+        settings: "ServerSettings" = None,
+        workspace: "Workspace" = None,  # noqa: ARG002 - kept for backward compat
+        wait_for_quit: bool = False,
     ):
-        """Start the notification server (if not already running)
+        """Start the web server (if not already running)
 
         Args:
             settings: Server settings
-            workspace: Workspace instance (required to get workspace path)
+            workspace: Workspace instance (deprecated, not used)
+            wait_for_quit: If True, server waits for explicit quit from web UI
         """
         if self.server is None:
-            from experimaestro.server import Server
-            from experimaestro.scheduler.state_provider import WorkspaceStateProvider
+            from experimaestro.webui import WebUIServer
 
-            if workspace is None:
-                raise ValueError("workspace parameter is required to start server")
-
-            # Get the workspace state provider singleton
-            state_provider = WorkspaceStateProvider.get_instance(
-                workspace.path, read_only=False, sync_on_start=False
-            )
-
-            self.server = Server.instance(settings, state_provider)
+            # Use the Scheduler itself as the StateProvider for live state access
+            self.server = WebUIServer.instance(settings, self, wait_for_quit)
             self.server.start()
-            logger.info("Server started by scheduler")
+            logger.info("Web server started by scheduler")
         else:
-            logger.debug("Server already running")
+            logger.debug("Web server already running")
 
     def stop_server(self):
-        """Stop the notification server"""
+        """Stop the web server"""
         if self.server is not None:
             self.server.stop()
-            logger.info("Server stopped by scheduler")
+            logger.info("Web server stopped by scheduler")
+
+    def wait_for_server_quit(self):
+        """Wait for explicit quit from web interface
+
+        Only blocks if server was started with wait_for_quit=True.
+        """
+        if self.server is not None:
+            self.server.wait()
 
     def run(self):
         """Run the event loop forever"""
@@ -197,6 +242,27 @@ class Scheduler(threading.Thread):
         with self._listeners_lock:
             self._listeners.clear()
 
+    def wait_for_notifications(self, timeout: float = 5.0) -> bool:
+        """Wait for all pending notifications to be processed.
+
+        This submits a sentinel task and waits for it to complete,
+        ensuring all previously submitted notifications have been processed.
+
+        Args:
+            timeout: Maximum time to wait in seconds
+
+        Returns:
+            True if all notifications were processed, False if timeout occurred
+        """
+        try:
+            # Submit a no-op and wait for it to complete
+            future = self._notification_executor.submit(lambda: None)
+            future.result(timeout=timeout)
+            return True
+        except concurrent.futures.TimeoutError:
+            logger.warning("Timeout waiting for notification queue to drain")
+            return False
+
     def getJobState(self, job: Job) -> "concurrent.futures.Future[JobState]":
         # Check if the job belongs to this scheduler
         if job.identifier not in self.jobs:
@@ -256,12 +322,35 @@ class Scheduler(threading.Thread):
             xp = experiment.current()
             xp.add_job(other)
 
+            # Merge transient modes: more conservative mode wins
+            # NONE(0) > TRANSIENT(1) > REMOVE(2) - lower value wins
+            was_transient = other.transient.is_transient
+            if job.transient < other.transient:
+                other.transient = job.transient
+                # If job was transient and is now non-transient, mark it as needed
+                # This flag tells aio_submit not to skip the job
+                if was_transient and not other.transient.is_transient:
+                    other._needed_transient = True
+
             # Copy watched outputs from new job to existing job
             # This ensures new callbacks are registered even for resubmitted jobs
             other.watched_outputs.extend(job.watched_outputs)
 
+            # Check if job needs to be re-started
+            need_restart = False
             if other.state.is_error():
-                logger.info("Re-submitting job")
+                logger.info("Re-submitting job (was in error state)")
+                need_restart = True
+            elif (
+                was_transient
+                and not other.transient.is_transient
+                and other.state == JobState.UNSCHEDULED
+            ):
+                # Job was transient and skipped, but now is non-transient - restart it
+                logger.info("Re-submitting job (was transient, now non-transient)")
+                need_restart = True
+
+            if need_restart:
                 # Clean up old process info so it will be re-started
                 other._process = None
                 if other.pidpath.is_file():
@@ -269,6 +358,7 @@ class Scheduler(threading.Thread):
                 # Use set_state to handle experiment statistics updates
                 other.set_state(JobState.WAITING)
                 self.notify_job_state(other)  # Notify listeners of re-submit
+                # The calling aio_submit will continue with this job and start it
             else:
                 logger.warning("Job %s already submitted", job.identifier)
 
@@ -283,13 +373,145 @@ class Scheduler(threading.Thread):
         job.submittime = time.time()
         xp.add_job(job)
 
+        # Update tags map for this experiment/run
+        if job.tags:
+            exp_run_key = (xp.name, xp.run_id)
+            if exp_run_key not in self._tags_map:
+                self._tags_map[exp_run_key] = {}
+            self._tags_map[exp_run_key][job.identifier] = dict(job.tags)
+
+        # Update dependencies map for this experiment/run
+        exp_run_key = (xp.name, xp.run_id)
+        if exp_run_key not in self._dependencies_map:
+            self._dependencies_map[exp_run_key] = {}
+        depends_on_ids = [
+            dep.origin.identifier
+            for dep in job.dependencies
+            if isinstance(dep, JobDependency)
+        ]
+        if depends_on_ids:
+            self._dependencies_map[exp_run_key][job.identifier] = depends_on_ids
+
         # Set up dependencies
         for dependency in job.dependencies:
             dependency.target = job
-            dependency.origin.dependents.add(dependency)
+            # Some dependencies (like PartialDependency) don't have an origin resource
+            if dependency.origin is not None:
+                dependency.origin.dependents.add(dependency)
 
         return None
 
+    def _start_job_event_reader(self, workspace_path: Path) -> None:
+        """Start watching job events in a workspace
+
+        Uses EventReader to watch .events/jobs/ for job progress events.
+        Job state events are emitted by the job process itself.
+        Only starts one reader per workspace.
+
+        Args:
+            workspace_path: Path to the workspace directory
+        """
+        with self._job_event_readers_lock:
+            # Already watching this workspace
+            if workspace_path in self._job_event_readers:
+                return
+
+            jobs_dir = workspace_path / ".events" / "jobs"
+
+            # Create new reader for this workspace
+            reader = EventReader(
+                [
+                    WatchedDirectory(
+                        path=jobs_dir,
+                        glob_pattern="*/event-*-*.jsonl",
+                        entity_id_extractor=job_entity_id_extractor,
+                    )
+                ]
+            )
+            reader.start_watching(
+                on_event=self._on_job_event,
+            )
+            self._job_event_readers[workspace_path] = reader
+            logger.debug("Started job event reader for %s", jobs_dir)
+
+    def _stop_job_event_reader(self, workspace_path: Optional[Path] = None) -> None:
+        """Stop watching job events
+
+        Args:
+            workspace_path: If provided, stop only this workspace's reader.
+                           If None, stop all readers.
+        """
+        with self._job_event_readers_lock:
+            if workspace_path is not None:
+                reader = self._job_event_readers.pop(workspace_path, None)
+                if reader is not None:
+                    reader.stop_watching()
+                    logger.debug("Stopped job event reader for %s", workspace_path)
+            else:
+                # Stop all readers
+                for path, reader in self._job_event_readers.items():
+                    reader.stop_watching()
+                    logger.debug("Stopped job event reader for %s", path)
+                self._job_event_readers.clear()
+
+    def _on_job_event(self, entity_id: str, event) -> None:
+        """Handle job events from EventReader
+
+        Updates job state from file-based events and notifies listeners.
+
+        Args:
+            entity_id: The job ID
+            event: The event (JobProgressEvent or JobStateChangedEvent)
+        """
+        job = self.jobs.get(entity_id)
+        if job is None:
+            logger.debug(
+                "Job event for unknown job %s",
+                entity_id,
+            )
+            return
+        logger.debug("Received event for job %s: %s", job, event)
+
+        if isinstance(event, JobProgressEvent):
+            # Update job's in-memory progress and notify legacy listeners
+            job.set_progress(event.level, event.progress, event.desc)
+            self.notify_job_state(job)
+
+        # Notify StateProvider-style listeners (TUI/WebUI)
+        state_event = JobStateChangedEvent(
+            job_id=job.identifier,
+            state=job.state.name.lower(),
+        )
+        self._notify_state_listeners_async(state_event)
+
+    def _cleanup_job_event_files(self, job: Job) -> None:
+        """Clean up old job event files from previous runs
+
+        Removes event files at .events/jobs/{task_id}/event-{job_id}-*.jsonl
+        Called when a job is about to start to ensure clean state.
+
+        Args:
+            job: The job being started
+        """
+        # Get the workspace path from the job's path
+        # job.path is workspace/jobs/task_id/job_id
+        workspace_path = job.path.parent.parent.parent
+        task_id = str(job.type.identifier)
+        job_id = job.identifier
+
+        events_dir = workspace_path / ".events" / "jobs" / task_id
+        if not events_dir.exists():
+            return
+
+        # Find and delete old event files for this job
+        pattern = f"event-{job_id}-*.jsonl"
+        for event_file in events_dir.glob(pattern):
+            try:
+                event_file.unlink()
+                logger.debug("Removed old job event file: %s", event_file)
+            except OSError as e:
+                logger.warning("Failed to remove job event file %s: %s", event_file, e)
+
     def _notify_listeners(self, notification_func, job: Job):
         """Execute notification in thread pool with error isolation.
 
@@ -310,18 +532,103 @@ class Scheduler(threading.Thread):
 
         self._notification_executor.submit(_do_notify)
 
+    def _notify_state_listeners_async(self, event):
+        """Notify StateProvider-style listeners asynchronously with error isolation.
+
+        This runs notifications in the same thread pool as _notify_listeners
+        to avoid blocking the scheduler and isolate errors.
+        """
+
+        def _do_notify():
+            # Get a snapshot of listeners with the lock
+            with self._state_listener_lock:
+                listeners_snapshot = list(self._state_listeners)
+
+            for listener in listeners_snapshot:
+                try:
+                    listener(event)
+                except Exception:
+                    logger.exception("Got an error with state listener %s", listener)
+
+        self._notification_executor.submit(_do_notify)
+
     def notify_job_submitted(self, job: Job):
         """Notify the listeners that a job has been submitted"""
         self._notify_listeners(lambda lst, j: lst.job_submitted(j), job)
 
+        # Also notify StateProvider-style listeners (for TUI etc.)
+        from experimaestro.scheduler.state_status import JobSubmittedEvent, JobTag
+
+        # Get experiment info from job's experiments list
+        for exp in job.experiments:
+            experiment_id = exp.experiment_id
+            run_id = exp.run_id
+
+            # Get tags and dependencies for this job
+            exp_run_key = (experiment_id, run_id)
+            tags_dict = self._tags_map.get(exp_run_key, {}).get(job.identifier, {})
+            tags = [JobTag(key=k, value=v) for k, v in tags_dict.items()]
+            depends_on = self._dependencies_map.get(exp_run_key, {}).get(
+                job.identifier, []
+            )
+
+            event = JobSubmittedEvent(
+                experiment_id=experiment_id,
+                run_id=run_id,
+                job_id=job.identifier,
+                tags=tags,
+                depends_on=depends_on,
+            )
+            self._notify_state_listeners_async(event)
+
     def notify_job_state(self, job: Job):
-        """Notify the listeners that a job has changed state"""
+        """Notify the listeners that a job has changed state
+
+        Note: This does NOT write to job event files. Job events are written
+        by the job process itself. The scheduler only forwards notifications
+        to listeners.
+        """
+        # Legacy listener notification (per-experiment)
         self._notify_listeners(lambda lst, j: lst.job_state(j), job)
 
-    def notify_service_add(self, service: Service):
+        # Notify StateProvider-style listeners with experiment-independent event
+        from experimaestro.scheduler.state_status import JobStateChangedEvent
+
+        event = JobStateChangedEvent(
+            job_id=job.identifier,
+            state=job.state.name.lower(),
+        )
+        self._notify_state_listeners_async(event)
+
+    def notify_service_add(
+        self, service: Service, experiment_id: str = "", run_id: str = ""
+    ):
         """Notify the listeners that a service has been added"""
         self._notify_listeners(lambda lst, s: lst.service_add(s), service)
 
+        # Store experiment info on the service for later retrieval
+        if experiment_id:
+            service._experiment_id = experiment_id
+            service._run_id = run_id or ""
+
+        # Store service in scheduler's services dict (persists after experiment ends)
+        if experiment_id:
+            key = (experiment_id, run_id or "")
+            if key not in self.services:
+                self.services[key] = {}
+            self.services[key][service.id] = service
+
+        # Also notify StateProvider-style listeners (for TUI etc.)
+        from experimaestro.scheduler.state_status import ServiceAddedEvent
+
+        if experiment_id:
+            event = ServiceAddedEvent(
+                experiment_id=experiment_id,
+                run_id=run_id or "",
+                service_id=service.id,
+            )
+            self._notify_state_listeners_async(event)
+
     async def aio_submit(self, job: Job) -> JobState:
         """Main scheduler function: submit a job, run it (if needed), and returns
         the status code
@@ -341,7 +648,7 @@ class Scheduler(threading.Thread):
         # Check that we don't have a completed job in
         # alternate directories
         for jobspath in experiment.current().alt_jobspaths:
-            # FIXME: check if done
+            # Future enhancement: check if done
             pass
 
         # Creates a link into the experiment folder
@@ -367,10 +674,6 @@ class Scheduler(threading.Thread):
                 job.set_state(JobState.RUNNING)
                 self.notify_job_state(job)
 
-                # Adds to the listeners
-                if self.server is not None:
-                    job.add_notification_server(self.server)
-
                 # And now, we wait...
                 logger.info("Got a process for job %s - waiting to complete", job)
                 code = await process.aio_code()
@@ -413,19 +716,27 @@ class Scheduler(threading.Thread):
 
         # If not done or running, start the job
         if not job.state.finished():
-            try:
-                state = await self.aio_start(job)
-                # Set endtime before set_state so database gets the timestamp
-                job.endtime = time.time()
-                job.set_state(state)
-            except Exception:
-                logger.exception("Got an exception while starting the job")
-                raise
+            # Check if this is a transient job that is not needed
+            if job.transient.is_transient and not job._needed_transient:
+                job.set_state(JobState.UNSCHEDULED)
+
+            # Start the job if not skipped (state is still WAITING)
+            if job.state == JobState.WAITING:
+                try:
+                    state = await self.aio_start(job)
+                    if state is not None:
+                        job.endtime = time.time()
+                        job.set_state(state)
+                except Exception:
+                    logger.exception("Got an exception while starting the job")
+                    raise
 
         # Job is finished - experiment statistics already updated by set_state
 
         # Write final metadata with end time and final state
-        job.write_metadata()
+        # Only for jobs that actually started (starttime is set in aio_start)
+        if job.starttime is not None:
+            job.status_path.write_text(json.dumps(job.state_dict()))
 
         if job in self.waitingjobs:
             self.waitingjobs.remove(job)
@@ -457,7 +768,7 @@ class Scheduler(threading.Thread):
             or process creation
         """
         from experimaestro.scheduler.jobs import JobStateError
-        from experimaestro.locking import Locks, LockError
+        from experimaestro.locking import DynamicDependencyLocks, LockError
         from experimaestro.scheduler.jobs import JobFailureStatus
 
         # Assert preconditions
@@ -489,7 +800,7 @@ class Scheduler(threading.Thread):
                     return JobStateError(JobFailureStatus.DEPENDENCY)
 
             # We first lock the job before proceeding
-            with Locks() as locks:
+            with DynamicDependencyLocks() as locks:
                 logger.debug("[starting] Locking job %s", job)
                 async with job.launcher.connector.lock(job.lockpath):
                     logger.debug("[starting] Locked job %s", job)
@@ -562,12 +873,15 @@ class Scheduler(threading.Thread):
                         if not directory.is_dir():
                             directory.mkdir(parents=True, exist_ok=True)
 
+                        # Clean up old job event files from previous runs
+                        self._cleanup_job_event_files(job)
+
                         # Write metadata with submit and start time (after directory creation)
-                        job.write_metadata()
+                        job.status_path.parent.mkdir(parents=True, exist_ok=True)
+                        job.status_path.write_text(json.dumps(job.state_dict()))
 
-                        # Sets up the notification URL
-                        if self.server is not None:
-                            job.add_notification_server(self.server)
+                        # Notify locks before job starts (e.g., create symlinks)
+                        await locks.aio_job_before_start(job)
 
                     except Exception:
                         logger.warning("Error while locking job", exc_info=True)
@@ -576,6 +890,30 @@ class Scheduler(threading.Thread):
                     try:
                         # Runs the job
                         process = await job.aio_run()
+
+                        # Notify locks that job has started
+                        await locks.aio_job_started(job, process)
+
+                        # Write locks.json for job process (if there are dynamic locks)
+                        if locks.locks:
+                            import tempfile
+
+                            locks_path = job.path / "locks.json"
+                            locks_data = {"dynamic_locks": locks.to_json()}
+                            # Atomic write: write to temp file then rename
+                            with tempfile.NamedTemporaryFile(
+                                mode="w",
+                                dir=job.path,
+                                prefix=".locks.",
+                                suffix=".json",
+                                delete=False,
+                            ) as tmp:
+                                json.dump(locks_data, tmp)
+                                tmp_path = tmp.name
+                            # Rename is atomic on POSIX
+                            import os
+
+                            os.rename(tmp_path, locks_path)
                     except Exception:
                         logger.warning("Error while starting job", exc_info=True)
                         return JobState.ERROR
@@ -633,6 +971,9 @@ class Scheduler(threading.Thread):
                     )
                     state = JobState.ERROR
 
+                # Notify locks that job has finished (before releasing)
+                await locks.aio_job_finished(job)
+
             # Locks are released here after job completes
 
             # Check if we should restart a resumable task that timed out
@@ -672,3 +1013,259 @@ class Scheduler(threading.Thread):
             # Notify scheduler listeners of job state after job completes
             self.notify_job_state(job)
             return state
+
+    # =========================================================================
+    # StateProvider abstract method implementations
+    # =========================================================================
+
+    def get_experiments(
+        self,
+        since: Optional[datetime] = None,  # noqa: ARG002
+    ) -> List[BaseExperiment]:
+        """Get list of all live experiments"""
+        # Note: 'since' filter not applicable for live scheduler
+        return list(self.experiments.values())
+
+    def get_experiment(self, experiment_id: str) -> Optional[BaseExperiment]:
+        """Get a specific experiment by ID"""
+        return self.experiments.get(experiment_id)
+
+    def get_experiment_runs(self, experiment_id: str) -> List[BaseExperiment]:
+        """Get all runs for an experiment
+
+        For a live scheduler, returns the live experiment directly.
+        """
+        exp = self.experiments.get(experiment_id)
+        if not exp:
+            return []
+
+        # Return the live experiment (it already implements BaseExperiment)
+        return [exp]
+
+    def get_current_run(self, experiment_id: str) -> Optional[str]:
+        """Get the current run ID for an experiment"""
+        exp = self.experiments.get(experiment_id)
+        return exp.run_id if exp else None
+
+    def get_jobs(
+        self,
+        experiment_id: Optional[str] = None,
+        run_id: Optional[str] = None,  # noqa: ARG002 - not used in live scheduler
+        task_id: Optional[str] = None,
+        state: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+        since: Optional[datetime] = None,  # noqa: ARG002 - not used in live scheduler
+    ) -> List[BaseJob]:
+        """Query jobs with optional filters"""
+        jobs: List[BaseJob] = list(self.jobs.values())
+
+        # Filter by experiment
+        if experiment_id:
+            exp = self.experiments.get(experiment_id)
+            if exp:
+                jobs = [j for j in jobs if j.experiments and exp in j.experiments]
+            else:
+                jobs = []
+
+        # Filter by task_id
+        if task_id:
+            jobs = [j for j in jobs if j.task_id == task_id]
+
+        # Filter by state
+        if state:
+            jobs = [j for j in jobs if j.state.name.lower() == state.lower()]
+
+        # Filter by tags (all tags must match)
+        if tags:
+            jobs = [j for j in jobs if all(j.tags.get(k) == v for k, v in tags.items())]
+
+        return jobs
+
+    def get_job(
+        self,
+        job_id: str,
+        experiment_id: str,  # noqa: ARG002 - job_id is sufficient in live scheduler
+        run_id: Optional[str] = None,  # noqa: ARG002 - job_id is sufficient in live scheduler
+    ) -> Optional[BaseJob]:
+        """Get a specific job"""
+        return self.jobs.get(job_id)
+
+    def get_all_jobs(
+        self,
+        state: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+        since: Optional[datetime] = None,  # noqa: ARG002 - not used in live scheduler
+    ) -> List[BaseJob]:
+        """Get all jobs across all experiments"""
+        jobs: List[BaseJob] = list(self.jobs.values())
+
+        if state:
+            jobs = [j for j in jobs if j.state.name.lower() == state.lower()]
+
+        if tags:
+            jobs = [j for j in jobs if all(j.tags.get(k) == v for k, v in tags.items())]
+
+        return jobs
+
+    def get_services(
+        self,
+        experiment_id: Optional[str] = None,
+        run_id: Optional[str] = None,
+    ) -> List[BaseService]:
+        """Get services for an experiment
+
+        Services are stored in the scheduler and persist after experiments finish.
+        """
+        if experiment_id is None:
+            # Return all services from all experiments
+            services = []
+            for services_dict in self.services.values():
+                services.extend(services_dict.values())
+            return services
+
+        # Get services for specific experiment
+        services = []
+        if run_id is not None:
+            # Specific run requested
+            key = (experiment_id, run_id)
+            services_dict = self.services.get(key, {})
+            services = list(services_dict.values())
+        else:
+            # No run_id specified - return services from all runs of this experiment
+            for (exp_id, _run_id), services_dict in self.services.items():
+                if exp_id == experiment_id:
+                    services.extend(services_dict.values())
+
+        logger.debug(
+            "get_services(%s, %s): returning %d services",
+            experiment_id,
+            run_id,
+            len(services),
+        )
+        return services
+
+    def get_tags_map(
+        self,
+        experiment_id: str,
+        run_id: Optional[str] = None,
+    ) -> dict[str, dict[str, str]]:
+        """Get tags map for jobs in an experiment/run
+
+        Returns a map from job_id to {tag_key: tag_value}.
+        """
+        exp = self.experiments.get(experiment_id)
+        if not exp:
+            return {}
+
+        # Use current run if not specified
+        if run_id is None:
+            run_id = exp.run_id
+
+        exp_run_key = (experiment_id, run_id)
+        return self._tags_map.get(exp_run_key, {})
+
+    def get_dependencies_map(
+        self,
+        experiment_id: str,
+        run_id: Optional[str] = None,
+    ) -> dict[str, list[str]]:
+        """Get dependencies map for jobs in an experiment/run
+
+        Returns a map from job_id to list of job_ids it depends on.
+        """
+        exp = self.experiments.get(experiment_id)
+        if not exp:
+            return {}
+
+        # Use current run if not specified
+        if run_id is None:
+            run_id = exp.run_id
+
+        exp_run_key = (experiment_id, run_id)
+        return self._dependencies_map.get(exp_run_key, {})
+
+    def kill_job(self, job: BaseJob, perform: bool = False) -> bool:
+        """Kill a running job
+
+        For the scheduler, this is a live operation.
+        """
+        if not perform:
+            # Just check if the job can be killed
+            return job.state == JobState.RUNNING
+
+        if job.state != JobState.RUNNING:
+            return False
+
+        # Get the actual Job from our jobs dict
+        actual_job = self.jobs.get(job.identifier)
+        if actual_job is None:
+            return False
+
+        # Try to kill the process via the process attribute
+        process = getattr(actual_job, "process", None)
+        if process is not None:
+            try:
+                process.kill()
+                return True
+            except Exception:
+                logger.exception("Failed to kill job %s", job.identifier)
+        return False
+
+    def clean_job(
+        self,
+        job: BaseJob,  # noqa: ARG002
+        perform: bool = False,  # noqa: ARG002
+    ) -> bool:
+        """Clean a finished job
+
+        For the scheduler, jobs are automatically cleaned when they finish.
+        """
+        # Live scheduler doesn't support cleaning jobs
+        return False
+
+    def get_process_info(self, job: BaseJob):
+        """Get process information for a job
+
+        For the scheduler, we can access the actual Job and read its PID file.
+        """
+        from experimaestro.scheduler.state_provider import ProcessInfo
+
+        # Get the actual Job from our jobs dict
+        actual_job = self.jobs.get(job.identifier)
+        if actual_job is None:
+            return None
+
+        # Try to read the PID file
+        try:
+            pidpath = getattr(actual_job, "pidpath", None)
+            if pidpath is None or not pidpath.exists():
+                return None
+
+            pinfo = json.loads(pidpath.read_text())
+            pid = pinfo.get("pid")
+            proc_type = pinfo.get("type", "unknown")
+
+            if pid is None:
+                return None
+
+            # Check if running based on job state
+            running = actual_job.state == JobState.RUNNING
+
+            return ProcessInfo(pid=pid, type=proc_type, running=running)
+        except Exception:
+            return None
+
+    def close(self) -> None:
+        """Close the state provider and clean up resources"""
+        # Stop all job event readers
+        self._stop_job_event_reader()
+
+    @property
+    def read_only(self) -> bool:
+        """Live scheduler is read-write"""
+        return False
+
+    @property
+    def is_remote(self) -> bool:
+        """Live scheduler is local"""
+        return False