Initializes and adds a new store to the storage. Invoked by the optimizer after it has selected a store to add it to Storage.

Check that SDK function RequestId[T] matches internal function return type T.

Returns the entire generated program.

Async version of get() that waits for and gets the result of a request. Args: request_id: The request ID of the request to get. Retu

In-place sets a nested key to value. Like get_nested(), if any key is not found, this will not raise an error.

Returns a user-unique service name for each test_skyserve_<name>(). Must be called from each test_skyserve_<name>().

Streams the logs of a request or a log file and gets the final result.

Returns whether the resource is launchable.

Create a test job with default values.

Get the current buffered content and clear the buffer. Returns: The buffered log lines as a single string

Check if the cluster name is a controller name. Returns: The controller if the cluster name is a controller name. Oth

Constructs the storage object. The Storage object is lazily initialized, so that when a user initializes a Storage object on client s

Lazy async engine init; delegates table creation to get_engine.

Returns a list of local command runners.

Check if this DAG represents a JobGroup. A DAG is a JobGroup if it has parallel execution mode. This is the defining characteristic t

Start any required background infrastructure. Returns: A process to join on shutdown, or None if no background process

Create a test job with default values.

Cancel the context. Idempotent.

Get the controller by controller type. Returns: The controller if the controller type is valid. Otherwise, returns No

Stop the policy server gracefully.

Establish persistent SSH connection. Returns the connection latency in seconds, or -1.0 on failure.

Initialize the test instance. Args: active_cluster_name: Name of a running cluster to use as template terminated_clus

Validates the volume.

Check replicas' status and count in sky serve status We will check cpus=2, as all our tests use cpus=2. Args: name: the name of the

This function is testing for the core functions on server side.

Capture the full traceback, bypassing any tracebacklimit.

Errors out if the cloud does not support all requested features. For instance, Lambda Cloud does not support stop, so Lambda.check_fe

Returns the current user.

Async version of get_request.

Get termination delay in seconds for a specific task. Args: task_name: The name of the task to get the delay for. Return

Helper to create mock Kubernetes node.

Helper to create a mock Kubernetes node with labels.

Patch managed_job_state functions for testing.

Serialize the SkyPilot API request.

Return the configuration for AWS S3.

Build #SBATCH directive lines from user-supplied sbatch_options. Args: sbatch_options: Dict mapping sbatch option names to values. R

Format SQL query with correct placeholders for current database. Converts '?' placeholders to '%s' for PostgreSQL, leaves as-is for SQLite.

Expand workdir to absolute path and validate it. Note: if this function is called on a remote SkyPilot API server, it must be after t

Resolve secret references to env vars and file mounts. Args: secret_refs: List of ManagedSecretRef from the task YAML.

Main entry point. Returns on success, raises on failure.

Retrieve a callable function from Azure SDK client object. Newer versions of the various client SDKs renamed function names to have a begin_

Patch global_user_state for testing.

Given a label value, returns the GPU type

Redirect the stdout and stderr of current context to a file. Args: log_file: The log file to redirect to. If None, the stdout and

Resolve volumes config to volume mounts and validate them. Raises: exceptions.VolumeNotFoundError: if any volume is not found.

Format errors into a string and log it to the console.

Returns an instance name object from the given name.

Gets the Rclone profile name for a given bucket. Args: bucket_name: The name of the bucket. Returns:

Find the best execution plan for the given DAG. Args: dag: the DAG to optimize. minimize: whether to minimize cost or

Convert to dictionary for API responses.