skypilot-nightly 1.0.0.dev20250502__py3-none-any.whl → 1.0.0.dev20251203__py3-none-any.whl

sky/schemas/generated/jobsv1_pb2.pyi

@@ -0,0 +1,254 @@
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class JobStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    JOB_STATUS_UNSPECIFIED: _ClassVar[JobStatus]
+    JOB_STATUS_INIT: _ClassVar[JobStatus]
+    JOB_STATUS_PENDING: _ClassVar[JobStatus]
+    JOB_STATUS_SETTING_UP: _ClassVar[JobStatus]
+    JOB_STATUS_RUNNING: _ClassVar[JobStatus]
+    JOB_STATUS_FAILED_DRIVER: _ClassVar[JobStatus]
+    JOB_STATUS_SUCCEEDED: _ClassVar[JobStatus]
+    JOB_STATUS_FAILED: _ClassVar[JobStatus]
+    JOB_STATUS_FAILED_SETUP: _ClassVar[JobStatus]
+    JOB_STATUS_CANCELLED: _ClassVar[JobStatus]
+JOB_STATUS_UNSPECIFIED: JobStatus
+JOB_STATUS_INIT: JobStatus
+JOB_STATUS_PENDING: JobStatus
+JOB_STATUS_SETTING_UP: JobStatus
+JOB_STATUS_RUNNING: JobStatus
+JOB_STATUS_FAILED_DRIVER: JobStatus
+JOB_STATUS_SUCCEEDED: JobStatus
+JOB_STATUS_FAILED: JobStatus
+JOB_STATUS_FAILED_SETUP: JobStatus
+JOB_STATUS_CANCELLED: JobStatus
+
+class AddJobRequest(_message.Message):
+    __slots__ = ("job_name", "username", "run_timestamp", "resources_str", "metadata")
+    JOB_NAME_FIELD_NUMBER: _ClassVar[int]
+    USERNAME_FIELD_NUMBER: _ClassVar[int]
+    RUN_TIMESTAMP_FIELD_NUMBER: _ClassVar[int]
+    RESOURCES_STR_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    job_name: str
+    username: str
+    run_timestamp: str
+    resources_str: str
+    metadata: str
+    def __init__(self, job_name: _Optional[str] = ..., username: _Optional[str] = ..., run_timestamp: _Optional[str] = ..., resources_str: _Optional[str] = ..., metadata: _Optional[str] = ...) -> None: ...
+
+class AddJobResponse(_message.Message):
+    __slots__ = ("job_id", "log_dir")
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    LOG_DIR_FIELD_NUMBER: _ClassVar[int]
+    job_id: int
+    log_dir: str
+    def __init__(self, job_id: _Optional[int] = ..., log_dir: _Optional[str] = ...) -> None: ...
+
+class QueueJobRequest(_message.Message):
+    __slots__ = ("job_id", "codegen", "script_path", "remote_log_dir", "managed_job")
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    CODEGEN_FIELD_NUMBER: _ClassVar[int]
+    SCRIPT_PATH_FIELD_NUMBER: _ClassVar[int]
+    REMOTE_LOG_DIR_FIELD_NUMBER: _ClassVar[int]
+    MANAGED_JOB_FIELD_NUMBER: _ClassVar[int]
+    job_id: int
+    codegen: str
+    script_path: str
+    remote_log_dir: str
+    managed_job: ManagedJobInfo
+    def __init__(self, job_id: _Optional[int] = ..., codegen: _Optional[str] = ..., script_path: _Optional[str] = ..., remote_log_dir: _Optional[str] = ..., managed_job: _Optional[_Union[ManagedJobInfo, _Mapping]] = ...) -> None: ...
+
+class ManagedJobInfo(_message.Message):
+    __slots__ = ("name", "pool", "workspace", "entrypoint", "tasks", "user_id")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    POOL_FIELD_NUMBER: _ClassVar[int]
+    WORKSPACE_FIELD_NUMBER: _ClassVar[int]
+    ENTRYPOINT_FIELD_NUMBER: _ClassVar[int]
+    TASKS_FIELD_NUMBER: _ClassVar[int]
+    USER_ID_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    pool: str
+    workspace: str
+    entrypoint: str
+    tasks: _containers.RepeatedCompositeFieldContainer[ManagedJobTask]
+    user_id: str
+    def __init__(self, name: _Optional[str] = ..., pool: _Optional[str] = ..., workspace: _Optional[str] = ..., entrypoint: _Optional[str] = ..., tasks: _Optional[_Iterable[_Union[ManagedJobTask, _Mapping]]] = ..., user_id: _Optional[str] = ...) -> None: ...
+
+class ManagedJobTask(_message.Message):
+    __slots__ = ("task_id", "name", "resources_str", "metadata_json")
+    TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    RESOURCES_STR_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
+    task_id: int
+    name: str
+    resources_str: str
+    metadata_json: str
+    def __init__(self, task_id: _Optional[int] = ..., name: _Optional[str] = ..., resources_str: _Optional[str] = ..., metadata_json: _Optional[str] = ...) -> None: ...
+
+class QueueJobResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class UpdateStatusRequest(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class UpdateStatusResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class GetJobQueueRequest(_message.Message):
+    __slots__ = ("user_hash", "all_jobs")
+    USER_HASH_FIELD_NUMBER: _ClassVar[int]
+    ALL_JOBS_FIELD_NUMBER: _ClassVar[int]
+    user_hash: str
+    all_jobs: bool
+    def __init__(self, user_hash: _Optional[str] = ..., all_jobs: bool = ...) -> None: ...
+
+class JobInfo(_message.Message):
+    __slots__ = ("job_id", "job_name", "username", "submitted_at", "status", "run_timestamp", "start_at", "end_at", "resources", "pid", "log_path", "metadata")
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    JOB_NAME_FIELD_NUMBER: _ClassVar[int]
+    USERNAME_FIELD_NUMBER: _ClassVar[int]
+    SUBMITTED_AT_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    RUN_TIMESTAMP_FIELD_NUMBER: _ClassVar[int]
+    START_AT_FIELD_NUMBER: _ClassVar[int]
+    END_AT_FIELD_NUMBER: _ClassVar[int]
+    RESOURCES_FIELD_NUMBER: _ClassVar[int]
+    PID_FIELD_NUMBER: _ClassVar[int]
+    LOG_PATH_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    job_id: int
+    job_name: str
+    username: str
+    submitted_at: float
+    status: JobStatus
+    run_timestamp: str
+    start_at: float
+    end_at: float
+    resources: str
+    pid: int
+    log_path: str
+    metadata: str
+    def __init__(self, job_id: _Optional[int] = ..., job_name: _Optional[str] = ..., username: _Optional[str] = ..., submitted_at: _Optional[float] = ..., status: _Optional[_Union[JobStatus, str]] = ..., run_timestamp: _Optional[str] = ..., start_at: _Optional[float] = ..., end_at: _Optional[float] = ..., resources: _Optional[str] = ..., pid: _Optional[int] = ..., log_path: _Optional[str] = ..., metadata: _Optional[str] = ...) -> None: ...
+
+class GetJobQueueResponse(_message.Message):
+    __slots__ = ("jobs",)
+    JOBS_FIELD_NUMBER: _ClassVar[int]
+    jobs: _containers.RepeatedCompositeFieldContainer[JobInfo]
+    def __init__(self, jobs: _Optional[_Iterable[_Union[JobInfo, _Mapping]]] = ...) -> None: ...
+
+class CancelJobsRequest(_message.Message):
+    __slots__ = ("job_ids", "cancel_all", "user_hash")
+    JOB_IDS_FIELD_NUMBER: _ClassVar[int]
+    CANCEL_ALL_FIELD_NUMBER: _ClassVar[int]
+    USER_HASH_FIELD_NUMBER: _ClassVar[int]
+    job_ids: _containers.RepeatedScalarFieldContainer[int]
+    cancel_all: bool
+    user_hash: str
+    def __init__(self, job_ids: _Optional[_Iterable[int]] = ..., cancel_all: bool = ..., user_hash: _Optional[str] = ...) -> None: ...
+
+class CancelJobsResponse(_message.Message):
+    __slots__ = ("cancelled_job_ids",)
+    CANCELLED_JOB_IDS_FIELD_NUMBER: _ClassVar[int]
+    cancelled_job_ids: _containers.RepeatedScalarFieldContainer[int]
+    def __init__(self, cancelled_job_ids: _Optional[_Iterable[int]] = ...) -> None: ...
+
+class FailAllInProgressJobsRequest(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class FailAllInProgressJobsResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class TailLogsRequest(_message.Message):
+    __slots__ = ("job_id", "managed_job_id", "follow", "tail")
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    MANAGED_JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    FOLLOW_FIELD_NUMBER: _ClassVar[int]
+    TAIL_FIELD_NUMBER: _ClassVar[int]
+    job_id: int
+    managed_job_id: int
+    follow: bool
+    tail: int
+    def __init__(self, job_id: _Optional[int] = ..., managed_job_id: _Optional[int] = ..., follow: bool = ..., tail: _Optional[int] = ...) -> None: ...
+
+class TailLogsResponse(_message.Message):
+    __slots__ = ("log_line", "exit_code")
+    LOG_LINE_FIELD_NUMBER: _ClassVar[int]
+    EXIT_CODE_FIELD_NUMBER: _ClassVar[int]
+    log_line: str
+    exit_code: int
+    def __init__(self, log_line: _Optional[str] = ..., exit_code: _Optional[int] = ...) -> None: ...
+
+class GetJobStatusRequest(_message.Message):
+    __slots__ = ("job_ids",)
+    JOB_IDS_FIELD_NUMBER: _ClassVar[int]
+    job_ids: _containers.RepeatedScalarFieldContainer[int]
+    def __init__(self, job_ids: _Optional[_Iterable[int]] = ...) -> None: ...
+
+class GetJobStatusResponse(_message.Message):
+    __slots__ = ("job_statuses",)
+    class JobStatusesEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: int
+        value: JobStatus
+        def __init__(self, key: _Optional[int] = ..., value: _Optional[_Union[JobStatus, str]] = ...) -> None: ...
+    JOB_STATUSES_FIELD_NUMBER: _ClassVar[int]
+    job_statuses: _containers.ScalarMap[int, JobStatus]
+    def __init__(self, job_statuses: _Optional[_Mapping[int, JobStatus]] = ...) -> None: ...
+
+class GetJobSubmittedTimestampRequest(_message.Message):
+    __slots__ = ("job_id",)
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    job_id: int
+    def __init__(self, job_id: _Optional[int] = ...) -> None: ...
+
+class GetJobSubmittedTimestampResponse(_message.Message):
+    __slots__ = ("timestamp",)
+    TIMESTAMP_FIELD_NUMBER: _ClassVar[int]
+    timestamp: float
+    def __init__(self, timestamp: _Optional[float] = ...) -> None: ...
+
+class GetJobEndedTimestampRequest(_message.Message):
+    __slots__ = ("job_id",)
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    job_id: int
+    def __init__(self, job_id: _Optional[int] = ...) -> None: ...
+
+class GetJobEndedTimestampResponse(_message.Message):
+    __slots__ = ("timestamp",)
+    TIMESTAMP_FIELD_NUMBER: _ClassVar[int]
+    timestamp: float
+    def __init__(self, timestamp: _Optional[float] = ...) -> None: ...
+
+class GetLogDirsForJobsRequest(_message.Message):
+    __slots__ = ("job_ids",)
+    JOB_IDS_FIELD_NUMBER: _ClassVar[int]
+    job_ids: _containers.RepeatedScalarFieldContainer[int]
+    def __init__(self, job_ids: _Optional[_Iterable[int]] = ...) -> None: ...
+
+class GetLogDirsForJobsResponse(_message.Message):
+    __slots__ = ("job_log_dirs",)
+    class JobLogDirsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: int
+        value: str
+        def __init__(self, key: _Optional[int] = ..., value: _Optional[str] = ...) -> None: ...
+    JOB_LOG_DIRS_FIELD_NUMBER: _ClassVar[int]
+    job_log_dirs: _containers.ScalarMap[int, str]
+    def __init__(self, job_log_dirs: _Optional[_Mapping[int, str]] = ...) -> None: ...