skypilot-nightly 1.0.0.dev20250916__py3-none-any.whl → 1.0.0.dev20250919__py3-none-any.whl

sky/provision/primeintellect/utils.py

@@ -0,0 +1,398 @@
+"""Prime Intellect library wrapper for SkyPilot."""
+
+import json
+import os
+import shlex
+import time
+from typing import Any, Dict, List, Optional, Tuple, Union
+import uuid
+
+import requests
+
+from sky.catalog import common as catalog_common
+from sky.utils import common_utils
+
+_df = None
+_lookup_dict = None
+
+# Base URL for Prime Intellect API (used as default if not configured).
+DEFAULT_BASE_URL = 'https://api.primeintellect.ai'
+CREDENTIALS_PATH = '~/.prime/config.json'
+INITIAL_BACKOFF_SECONDS = 10
+MAX_BACKOFF_FACTOR = 10
+MAX_ATTEMPTS = 6
+
+
+class PrimeintellectAPIError(Exception):
+    """Base exception for Prime Intellect API errors."""
+
+    def __init__(self,
+                 message: str,
+                 status_code: Optional[int] = None,
+                 response_data: Optional[Dict[str, Any]] = None):
+        super().__init__(message)
+        self.status_code = status_code
+        self.response_data = response_data
+
+
+class PrimeintellectResourcesUnavailableError(PrimeintellectAPIError):
+    """Exception for when resources are unavailable on Prime Intellect."""
+    pass
+
+
+def _parse_api_error(response: Any) -> Tuple[str, bool]:
+    """Parse API error response to extract meaningful error messages.
+
+    Returns:
+        Tuple[str, bool]:
+        - str: A human-readable error message parsed from the API response.
+        - bool: True if the error indicates resource unavailability (e.g.,
+          capacity issues or quota/limit exceeded), otherwise False.
+    """
+    try:
+        if hasattr(response, 'json'):
+            error_data = response.json()
+        else:
+            error_data = response
+
+        if isinstance(error_data, dict):
+            # Try to extract error message from common error response fields
+            error_message = error_data.get('message', '')
+            if not error_message:
+                error_message = error_data.get('error', '')
+            if not error_message:
+                error_message = error_data.get('detail', '')
+
+            # Check if it's a resource unavailability error
+            if any(keyword in error_message.lower() for keyword in [
+                    'no capacity', 'capacity', 'unavailable', 'out of stock',
+                    'insufficient', 'not available', 'quota exceeded',
+                    'limit exceeded'
+            ]):
+                return error_message, True
+
+            return error_message, False
+
+        return str(error_data), False
+    except Exception:  # pylint: disable=broad-except
+        return f'HTTP {response.status_code} {response.reason}', False
+
+
+def _try_request_with_backoff(
+        method: str,
+        url: str,
+        headers: Dict[str, str],
+        data: Optional[Union[str, Dict[str, Any]]] = None) -> Dict[str, Any]:
+    backoff = common_utils.Backoff(initial_backoff=INITIAL_BACKOFF_SECONDS,
+                                   max_backoff_factor=MAX_BACKOFF_FACTOR)
+    for i in range(MAX_ATTEMPTS):
+        timeout = 30
+        if method == 'get':
+            response = requests.get(url,
+                                    headers=headers,
+                                    params=data,
+                                    timeout=timeout)
+        elif method == 'post':
+            response = requests.post(url,
+                                     headers=headers,
+                                     json=data,
+                                     timeout=timeout)
+        elif method == 'put':
+            response = requests.put(url,
+                                    headers=headers,
+                                    json=data,
+                                    timeout=timeout)
+        elif method == 'patch':
+            response = requests.patch(url,
+                                      headers=headers,
+                                      json=data,
+                                      timeout=timeout)
+        elif method == 'delete':
+            response = requests.delete(url, headers=headers, timeout=timeout)
+        else:
+            raise ValueError(f'Unsupported requests method: {method}')
+        # If rate limited, wait and try again
+        if response.status_code == 429 and i != MAX_ATTEMPTS - 1:
+            time.sleep(backoff.current_backoff())
+            continue
+        if response.ok:
+            return response.json()
+        else:
+            # Parse the error response for meaningful messages
+            err, is_resource_unavailable = _parse_api_error(response)
+
+            # Create a more informative error message
+            if not err:
+                err = (f'API request failed: {method} {url}: '
+                       f'{response.status_code} {response.reason}')
+            else:
+                err = f'API request failed: {err}'
+
+            # Raise appropriate exception based on error type
+            if is_resource_unavailable:
+                raise PrimeintellectResourcesUnavailableError(
+                    err,
+                    status_code=response.status_code,
+                    response_data=response.json()
+                    if hasattr(response, 'json') else None)
+            else:
+                raise PrimeintellectAPIError(
+                    err,
+                    status_code=response.status_code,
+                    response_data=response.json()
+                    if hasattr(response, 'json') else None)
+    return {}
+
+
+def get_upstream_cloud_id(instance_type: str) -> Optional[str]:
+    global _df, _lookup_dict
+    if _df is None:
+        _df = catalog_common.read_catalog('primeintellect/vms.csv')
+        _lookup_dict = (
+            _df.set_index('InstanceType')['UpstreamCloudId'].to_dict())
+    return _lookup_dict.get(instance_type)
+
+
+class PrimeIntellectAPIClient:
+    """Client for interacting with Prime Intellect API."""
+
+    def __init__(self) -> None:
+        self.credentials = os.path.expanduser(CREDENTIALS_PATH)
+        assert os.path.exists(self.credentials), 'Credentials not found'
+        with open(self.credentials, 'r', encoding='utf-8') as f:
+            self._credentials = json.load(f)
+        self.api_key = self._credentials.get('api_key')
+        self.team_id = self._credentials.get('team_id')
+        self.base_url = self._credentials.get('base_url', DEFAULT_BASE_URL)
+        self.headers = {
+            'Authorization': f'Bearer {self.api_key}',
+            'Content-Type': 'application/json'
+        }
+
+    def list_instances(self, **search_kwargs) -> List[Dict[str, Any]]:
+        response = _try_request_with_backoff('get',
+                                             f'{self.base_url}/api/v1/pods',
+                                             headers=self.headers,
+                                             data=search_kwargs)
+        return response['data']
+
+    def get_instance_details(self, instance_id: str) -> Dict[str, Any]:
+        return _try_request_with_backoff(
+            'get',
+            f'{self.base_url}/api/v1/pods/{instance_id}',
+            headers=self.headers)
+
+    def launch(self,
+               name: str,
+               instance_type: str,
+               region: str,
+               availability_zone: str,
+               disk_size: int,
+               vcpus: int = 0,
+               memory: int = 0) -> Dict[str, Any]:
+        """Create a pod/instance via Prime Intellect API.
+
+        Args:
+            name: User-visible name of the pod.
+            instance_type: A catalog instance type string. The expected format
+                is:
+                "<provider>__<accelerator>__<vcpus>__<memory>[_SPOT]".
+
+                - <provider>: Upstream provider tag (e.g., "primecompute").
+                - <accelerator>:
+                  * GPU nodes: "<N>x<GPU_MODEL>", e.g., "8xH100_80GB".
+                  * CPU-only nodes: the literal "CPU_NODE".
+                - <vcpus>: Integer string for vCPU count (e.g., "104").
+                - <memory>: Integer string for memory in GB (e.g., "752").
+                - Optional suffix "_SPOT" may be present in the full string
+                  (ignored here; pricing/spot behavior is not controlled by
+                  this method).
+
+                Notes:
+                - Parsing: only the first two components (provider,
+                  accelerator) are needed to build the payload. The vCPU
+                  and memory values are provided via the ``vcpus`` and
+                  ``memory`` arguments.
+                - Catalog lookup: the full instance_type string is used to
+                  map to the catalog's UpstreamCloudId.
+                - CPU-only: accelerator "CPU_NODE" is a sentinel for
+                  "no GPU". We set gpuType='CPU_NODE' and gpuCount=1 to
+                  represent CPU-only pods.
+                - Spot: the optional "__SPOT" suffix (if present) is ignored
+                  here; pricing/spot behavior is handled elsewhere.
+
+            region: Country/region code used by Prime Intellect.
+            availability_zone: Data center ID (zone) within the region.
+            disk_size: Boot disk size in GB.
+            vcpus: Optional explicit vCPU override; if >0 it will be sent.
+            memory: Optional explicit memory override in GB; if >0 it will be
+                sent.
+
+        Returns:
+            The API response JSON as a dict.
+        """
+        cloud_id = get_upstream_cloud_id(instance_type)
+        assert cloud_id, 'cloudId cannot be None'
+        assert availability_zone, 'availability_zone cannot be None'
+
+        # Parse the instance_type. We only need the first two components:
+        # provider and accelerator info (see docstring above).
+        provider, gpu_parts, _, _ = instance_type.split('__', 3)
+        if 'CPU_NODE' in gpu_parts:
+            # Prime Intellect API uses the same schema for CPU-only and GPU
+            # pods. For CPU-only instances, we set gpuType='CPU_NODE' and
+            # gpuCount=1 as a sentinel to indicate "no GPU". This is how CPU
+            # instances are represented internally on our platform; the
+            # backend does not interpret this as having a physical GPU.
+            gpu_type = 'CPU_NODE'
+            gpu_count = 1
+        else:
+            parts = gpu_parts.split('x', 1)
+            gpu_count = int(parts[0])
+            gpu_type = parts[1]
+
+        payload: Dict[str, Any] = {
+            'pod': {
+                'name': name,
+                'cloudId': cloud_id,
+                'socket': 'PCIe',
+                'gpuType': gpu_type,
+                'gpuCount': int(gpu_count),
+                'diskSize': disk_size,
+                # Prime Intellect API historically required maxPrice.
+                # Set to 0 to indicate on-demand/non-spot pricing.
+                'maxPrice': 0,
+            },
+            'provider': {
+                'type': provider,
+            }
+        }
+
+        if vcpus > 0:
+            payload['pod']['vcpus'] = vcpus
+        if memory > 0:
+            payload['pod']['memory'] = memory
+
+        if region != 'UNSPECIFIED':
+            payload['pod']['country'] = region
+        if availability_zone != 'UNSPECIFIED':
+            payload['pod']['dataCenterId'] = availability_zone
+
+        if self.team_id is not None and self.team_id != '':
+            payload['team'] = {'teamId': self.team_id}
+
+        response = _try_request_with_backoff(
+            'post',
+            f'{self.base_url}/api/v1/pods',
+            headers=self.headers,
+            data=payload,
+        )
+        return response
+
+    def remove(self, instance_id: str) -> Dict[str, Any]:
+        return _try_request_with_backoff(
+            'delete',
+            f'{self.base_url}/api/v1/pods/{instance_id}',
+            headers=self.headers,
+        )
+
+    def list_ssh_keys(self) -> List[Dict[str, Any]]:
+        response = _try_request_with_backoff('get',
+                                             f'{self.base_url}/api/v1/ssh_keys',
+                                             headers=self.headers)
+        return response['data']
+
+    def get_or_add_ssh_key(self, ssh_pub_key: str = '') -> Dict[str, str]:
+        """Add ssh key if not already added."""
+        # Check if the public key is already added
+        ssh_keys = self.list_ssh_keys()
+        for key in ssh_keys:
+            if key['publicKey'].strip().split()[:2] == ssh_pub_key.strip(
+            ).split()[:2]:
+                return {'name': key['name'], 'ssh_key': ssh_pub_key}
+
+        # Add the public key to Prime Intellect account if not already added
+        ssh_key_name = 'skypilot-' + str(uuid.uuid4()).replace('-', '')[:8]
+        _try_request_with_backoff(
+            'post',
+            f'{self.base_url}/api/v1/ssh_keys',
+            headers=self.headers,
+            data={
+                'name': ssh_key_name,
+                'publicKey': ssh_pub_key
+            },
+        )
+        return {'name': ssh_key_name, 'ssh_key': ssh_pub_key}
+
+
+def parse_ssh_connection(ssh_connection: Any) -> Tuple[Optional[str], int]:
+    """Parse and extract SSH username and port from a connection field.
+
+    The provider may return the SSH connection in multiple shapes. This helper
+    robustly extracts the SSH username and port while tolerating extra flags or
+    various tokenizations.
+
+    Accepted formats (examples):
+    - String with port flag:
+      "ubuntu@1.2.3.4 -p 2222 [-o <flag> ...]"
+    - String without explicit port (defaults to 22):
+      "ubuntu@1.2.3.4"
+    - String with host:port:
+      "ubuntu@1.2.3.4:2222"
+    - List with a single target:
+      ["ubuntu@1.2.3.4"]
+    - List of tokens (e.g., split form):
+      ["ubuntu@1.2.3.4", "-p", "2222"]
+
+    Args:
+        ssh_connection: The raw field from the API; can be a string or a list
+            of strings.
+
+    Returns:
+        (ssh_user, ssh_port): username if found, else None; port if found,
+        else 22.
+    """
+    ssh_user: Optional[str] = None
+    ssh_port: int = 22
+
+    # Normalize into a list of tokens for easier processing.
+    tokens: List[str] = []
+    if isinstance(ssh_connection, str):
+        try:
+            tokens = shlex.split(ssh_connection)
+        except Exception:  # pylint: disable=broad-except
+            tokens = [ssh_connection]
+    elif isinstance(ssh_connection, list):
+        for elem in ssh_connection:
+            if isinstance(elem, str):
+                try:
+                    tokens.extend(shlex.split(elem))
+                except Exception:  # pylint: disable=broad-except
+                    tokens.append(elem)
+    else:
+        # Unknown type; return defaults.
+        return ssh_user, ssh_port
+
+    # Find the first token containing '@' as the user@host candidate.
+    user_host: Optional[str] = next((t for t in tokens if '@' in t), None)
+    if user_host:
+        ssh_user = user_host.split('@', 1)[0].strip()
+        # Try host:port format (after '@').
+        host_part = user_host.split('@', 1)[1]
+        if ':' in host_part:
+            _, maybe_port = host_part.rsplit(':', 1)
+            try:
+                ssh_port = int(maybe_port)
+            except ValueError:
+                pass
+
+    # Check for '-p <port>' pair anywhere in the tokens. This takes priority.
+    if '-p' in tokens:
+        idx = tokens.index('-p')
+        if idx + 1 < len(tokens):
+            try:
+                ssh_port = int(tokens[idx + 1])
+            except ValueError:
+                pass
+
+    return ssh_user, ssh_port

sky/resources.py

@@ -1331,10 +1331,18 @@ class Resources:
                     clouds.CloudImplementationFeatures.IMAGE_ID
                 })
         except exceptions.NotSupportedError as e:
+            # Provide a more helpful error message for Lambda cloud
+            if self.cloud.is_same_cloud(clouds.Lambda()):
+                with ux_utils.print_exception_no_traceback():
+                    raise ValueError(
+                        'Lambda cloud only supports Docker images. '
+                        'Please prefix your image with "docker:" '
+                        '(e.g., image_id: docker:your-image-name).') from e
             with ux_utils.print_exception_no_traceback():
                 raise ValueError(
                     'image_id is only supported for AWS/GCP/Azure/IBM/OCI/'
-                    'Kubernetes, please explicitly specify the cloud.') from e
+                    'Kubernetes. For Lambda cloud, use "docker:" prefix for '
+                    'Docker images.') from e
 
         if self._region is not None:
             # If the image_id has None as key (region-agnostic),

sky/schemas/generated/jobsv1_pb2.py

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"sky/schemas/generated/jobsv1.proto\x12\x07jobs.v1\"\x85\x01\n\rAddJobRequest\x12\x15\n\x08job_name\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x10\n\x08username\x18\x02 \x01(\t\x12\x15\n\rrun_timestamp\x18\x03 \x01(\t\x12\x15\n\rresources_str\x18\x04 \x01(\t\x12\x10\n\x08metadata\x18\x05 \x01(\tB\x0b\n\t_job_name\"1\n\x0e\x41\x64\x64JobResponse\x12\x0e\n\x06job_id\x18\x01 \x01(\x03\x12\x0f\n\x07log_dir\x18\x02 \x01(\t\"\xb3\x01\n\x0fQueueJobRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\x03\x12\x14\n\x07\x63odegen\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x13\n\x0bscript_path\x18\x03 \x01(\t\x12\x16\n\x0eremote_log_dir\x18\x04 \x01(\t\x12\x31\n\x0bmanaged_job\x18\x05 \x01(\x0b\x32\x17.jobs.v1.ManagedJobInfoH\x01\x88\x01\x01\x42\n\n\x08_codegenB\x0e\n\x0c_managed_job\"\x89\x01\n\x0eManagedJobInfo\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\x04pool\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x11\n\tworkspace\x18\x03 \x01(\t\x12\x12\n\nentrypoint\x18\x04 \x01(\t\x12&\n\x05tasks\x18\x05 \x03(\x0b\x32\x17.jobs.v1.ManagedJobTaskB\x07\n\x05_pool\"]\n\x0eManagedJobTask\x12\x0f\n\x07task_id\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x15\n\rresources_str\x18\x03 \x01(\t\x12\x15\n\rmetadata_json\x18\x04 \x01(\t\"\x12\n\x10QueueJobResponse\"\x15\n\x13UpdateStatusRequest\"\x16\n\x14UpdateStatusResponse\"L\n\x12GetJobQueueRequest\x12\x16\n\tuser_hash\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x10\n\x08\x61ll_jobs\x18\x02 \x01(\x08\x42\x0c\n\n_user_hash\"\xf4\x01\n\x07JobInfo\x12\x0e\n\x06job_id\x18\x01 \x01(\x03\x12\x10\n\x08job_name\x18\x02 \x01(\t\x12\x10\n\x08username\x18\x03 \x01(\t\x12\x14\n\x0csubmitted_at\x18\x04 \x01(\x01\x12\"\n\x06status\x18\x05 \x01(\x0e\x32\x12.jobs.v1.JobStatus\x12\x15\n\rrun_timestamp\x18\x06 \x01(\t\x12\x10\n\x08start_at\x18\x07 \x01(\x01\x12\x0e\n\x06\x65nd_at\x18\x08 \x01(\x01\x12\x11\n\tresources\x18\t \x01(\t\x12\x0b\n\x03pid\x18\n \x01(\x03\x12\x10\n\x08log_path\x18\x0b \x01(\t\x12\x10\n\x08metadata\x18\x0c \x01(\t\"5\n\x13GetJobQueueResponse\x12\x1e\n\x04jobs\x18\x01 \x03(\x0b\x32\x10.jobs.v1.JobInfo\"^\n\x11\x43\x61ncelJobsRequest\x12\x0f\n\x07job_ids\x18\x01 \x03(\x03\x12\x12\n\ncancel_all\x18\x02 \x01(\x08\x12\x16\n\tuser_hash\x18\x03 \x01(\tH\x00\x88\x01\x01\x42\x0c\n\n_user_hash\"/\n\x12\x43\x61ncelJobsResponse\x12\x19\n\x11\x63\x61ncelled_job_ids\x18\x01 \x03(\x03\"\x1e\n\x1c\x46\x61ilAllInProgressJobsRequest\"\x1f\n\x1d\x46\x61ilAllInProgressJobsResponse\"\x7f\n\x0fTailLogsRequest\x12\x13\n\x06job_id\x18\x01 \x01(\x03H\x00\x88\x01\x01\x12\x1b\n\x0emanaged_job_id\x18\x02 \x01(\x03H\x01\x88\x01\x01\x12\x0e\n\x06\x66ollow\x18\x03 \x01(\x08\x12\x0c\n\x04tail\x18\x04 \x01(\x05\x42\t\n\x07_job_idB\x11\n\x0f_managed_job_id\"7\n\x10TailLogsResponse\x12\x10\n\x08log_line\x18\x01 \x01(\t\x12\x11\n\texit_code\x18\x02 \x01(\x05\"&\n\x13GetJobStatusRequest\x12\x0f\n\x07job_ids\x18\x01 \x03(\x03\"\xa4\x01\n\x14GetJobStatusResponse\x12\x44\n\x0cjob_statuses\x18\x01 \x03(\x0b\x32..jobs.v1.GetJobStatusResponse.JobStatusesEntry\x1a\x46\n\x10JobStatusesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12!\n\x05value\x18\x02 \x01(\x0e\x32\x12.jobs.v1.JobStatus:\x02\x38\x01\"A\n\x1fGetJobSubmittedTimestampRequest\x12\x13\n\x06job_id\x18\x01 \x01(\x03H\x00\x88\x01\x01\x42\t\n\x07_job_id\"5\n GetJobSubmittedTimestampResponse\x12\x11\n\ttimestamp\x18\x01 \x01(\x02\"=\n\x1bGetJobEndedTimestampRequest\x12\x13\n\x06job_id\x18\x01 \x01(\x03H\x00\x88\x01\x01\x42\t\n\x07_job_id\"1\n\x1cGetJobEndedTimestampResponse\x12\x11\n\ttimestamp\x18\x01 \x01(\x02\"+\n\x18GetLogDirsForJobsRequest\x12\x0f\n\x07job_ids\x18\x01 \x03(\x03\"\x98\x01\n\x19GetLogDirsForJobsResponse\x12H\n\x0cjob_log_dirs\x18\x01 \x03(\x0b\x32\x32.jobs.v1.GetLogDirsForJobsResponse.JobLogDirsEntry\x1a\x31\n\x0fJobLogDirsEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01*\x8d\x02\n\tJobStatus\x12\x1a\n\x16JOB_STATUS_UNSPECIFIED\x10\x00\x12\x13\n\x0fJOB_STATUS_INIT\x10\x01\x12\x16\n\x12JOB_STATUS_PENDING\x10\x02\x12\x19\n\x15JOB_STATUS_SETTING_UP\x10\x03\x12\x16\n\x12JOB_STATUS_RUNNING\x10\x04\x12\x1c\n\x18JOB_STATUS_FAILED_DRIVER\x10\x05\x12\x18\n\x14JOB_STATUS_SUCCEEDED\x10\x06\x12\x15\n\x11JOB_STATUS_FAILED\x10\x07\x12\x1b\n\x17JOB_STATUS_FAILED_SETUP\x10\x08\x12\x18\n\x14JOB_STATUS_CANCELLED\x10\t2\x91\x07\n\x0bJobsService\x12\x39\n\x06\x41\x64\x64Job\x12\x16.jobs.v1.AddJobRequest\x1a\x17.jobs.v1.AddJobResponse\x12?\n\x08QueueJob\x12\x18.jobs.v1.QueueJobRequest\x1a\x19.jobs.v1.QueueJobResponse\x12K\n\x0cUpdateStatus\x12\x1c.jobs.v1.UpdateStatusRequest\x1a\x1d.jobs.v1.UpdateStatusResponse\x12H\n\x0bGetJobQueue\x12\x1b.jobs.v1.GetJobQueueRequest\x1a\x1c.jobs.v1.GetJobQueueResponse\x12\x45\n\nCancelJobs\x12\x1a.jobs.v1.CancelJobsRequest\x1a\x1b.jobs.v1.CancelJobsResponse\x12\x66\n\x15\x46\x61ilAllInProgressJobs\x12%.jobs.v1.FailAllInProgressJobsRequest\x1a&.jobs.v1.FailAllInProgressJobsResponse\x12\x41\n\x08TailLogs\x12\x18.jobs.v1.TailLogsRequest\x1a\x19.jobs.v1.TailLogsResponse0\x01\x12K\n\x0cGetJobStatus\x12\x1c.jobs.v1.GetJobStatusRequest\x1a\x1d.jobs.v1.GetJobStatusResponse\x12o\n\x18GetJobSubmittedTimestamp\x12(.jobs.v1.GetJobSubmittedTimestampRequest\x1a).jobs.v1.GetJobSubmittedTimestampResponse\x12\x63\n\x14GetJobEndedTimestamp\x12$.jobs.v1.GetJobEndedTimestampRequest\x1a%.jobs.v1.GetJobEndedTimestampResponse\x12Z\n\x11GetLogDirsForJobs\x12!.jobs.v1.GetLogDirsForJobsRequest\x1a\".jobs.v1.GetLogDirsForJobsResponseb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"sky/schemas/generated/jobsv1.proto\x12\x07jobs.v1\"\x85\x01\n\rAddJobRequest\x12\x15\n\x08job_name\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x10\n\x08username\x18\x02 \x01(\t\x12\x15\n\rrun_timestamp\x18\x03 \x01(\t\x12\x15\n\rresources_str\x18\x04 \x01(\t\x12\x10\n\x08metadata\x18\x05 \x01(\tB\x0b\n\t_job_name\"1\n\x0e\x41\x64\x64JobResponse\x12\x0e\n\x06job_id\x18\x01 \x01(\x03\x12\x0f\n\x07log_dir\x18\x02 \x01(\t\"\xb3\x01\n\x0fQueueJobRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\x03\x12\x14\n\x07\x63odegen\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x13\n\x0bscript_path\x18\x03 \x01(\t\x12\x16\n\x0eremote_log_dir\x18\x04 \x01(\t\x12\x31\n\x0bmanaged_job\x18\x05 \x01(\x0b\x32\x17.jobs.v1.ManagedJobInfoH\x01\x88\x01\x01\x42\n\n\x08_codegenB\x0e\n\x0c_managed_job\"\x89\x01\n\x0eManagedJobInfo\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\x04pool\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x11\n\tworkspace\x18\x03 \x01(\t\x12\x12\n\nentrypoint\x18\x04 \x01(\t\x12&\n\x05tasks\x18\x05 \x03(\x0b\x32\x17.jobs.v1.ManagedJobTaskB\x07\n\x05_pool\"]\n\x0eManagedJobTask\x12\x0f\n\x07task_id\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x15\n\rresources_str\x18\x03 \x01(\t\x12\x15\n\rmetadata_json\x18\x04 \x01(\t\"\x12\n\x10QueueJobResponse\"\x15\n\x13UpdateStatusRequest\"\x16\n\x14UpdateStatusResponse\"L\n\x12GetJobQueueRequest\x12\x16\n\tuser_hash\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x10\n\x08\x61ll_jobs\x18\x02 \x01(\x08\x42\x0c\n\n_user_hash\"\xa3\x02\n\x07JobInfo\x12\x0e\n\x06job_id\x18\x01 \x01(\x03\x12\x10\n\x08job_name\x18\x02 \x01(\t\x12\x10\n\x08username\x18\x03 \x01(\t\x12\x14\n\x0csubmitted_at\x18\x04 \x01(\x01\x12\"\n\x06status\x18\x05 \x01(\x0e\x32\x12.jobs.v1.JobStatus\x12\x15\n\rrun_timestamp\x18\x06 \x01(\t\x12\x15\n\x08start_at\x18\x07 \x01(\x01H\x00\x88\x01\x01\x12\x13\n\x06\x65nd_at\x18\x08 \x01(\x01H\x01\x88\x01\x01\x12\x11\n\tresources\x18\t \x01(\t\x12\x10\n\x03pid\x18\n \x01(\x03H\x02\x88\x01\x01\x12\x10\n\x08log_path\x18\x0b \x01(\t\x12\x10\n\x08metadata\x18\x0c \x01(\tB\x0b\n\t_start_atB\t\n\x07_end_atB\x06\n\x04_pid\"5\n\x13GetJobQueueResponse\x12\x1e\n\x04jobs\x18\x01 \x03(\x0b\x32\x10.jobs.v1.JobInfo\"^\n\x11\x43\x61ncelJobsRequest\x12\x0f\n\x07job_ids\x18\x01 \x03(\x03\x12\x12\n\ncancel_all\x18\x02 \x01(\x08\x12\x16\n\tuser_hash\x18\x03 \x01(\tH\x00\x88\x01\x01\x42\x0c\n\n_user_hash\"/\n\x12\x43\x61ncelJobsResponse\x12\x19\n\x11\x63\x61ncelled_job_ids\x18\x01 \x03(\x03\"\x1e\n\x1c\x46\x61ilAllInProgressJobsRequest\"\x1f\n\x1d\x46\x61ilAllInProgressJobsResponse\"\x7f\n\x0fTailLogsRequest\x12\x13\n\x06job_id\x18\x01 \x01(\x03H\x00\x88\x01\x01\x12\x1b\n\x0emanaged_job_id\x18\x02 \x01(\x03H\x01\x88\x01\x01\x12\x0e\n\x06\x66ollow\x18\x03 \x01(\x08\x12\x0c\n\x04tail\x18\x04 \x01(\x05\x42\t\n\x07_job_idB\x11\n\x0f_managed_job_id\"7\n\x10TailLogsResponse\x12\x10\n\x08log_line\x18\x01 \x01(\t\x12\x11\n\texit_code\x18\x02 \x01(\x05\"&\n\x13GetJobStatusRequest\x12\x0f\n\x07job_ids\x18\x01 \x03(\x03\"\xa4\x01\n\x14GetJobStatusResponse\x12\x44\n\x0cjob_statuses\x18\x01 \x03(\x0b\x32..jobs.v1.GetJobStatusResponse.JobStatusesEntry\x1a\x46\n\x10JobStatusesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12!\n\x05value\x18\x02 \x01(\x0e\x32\x12.jobs.v1.JobStatus:\x02\x38\x01\"A\n\x1fGetJobSubmittedTimestampRequest\x12\x13\n\x06job_id\x18\x01 \x01(\x03H\x00\x88\x01\x01\x42\t\n\x07_job_id\"5\n GetJobSubmittedTimestampResponse\x12\x11\n\ttimestamp\x18\x01 \x01(\x02\"=\n\x1bGetJobEndedTimestampRequest\x12\x13\n\x06job_id\x18\x01 \x01(\x03H\x00\x88\x01\x01\x42\t\n\x07_job_id\"1\n\x1cGetJobEndedTimestampResponse\x12\x11\n\ttimestamp\x18\x01 \x01(\x02\"+\n\x18GetLogDirsForJobsRequest\x12\x0f\n\x07job_ids\x18\x01 \x03(\x03\"\x98\x01\n\x19GetLogDirsForJobsResponse\x12H\n\x0cjob_log_dirs\x18\x01 \x03(\x0b\x32\x32.jobs.v1.GetLogDirsForJobsResponse.JobLogDirsEntry\x1a\x31\n\x0fJobLogDirsEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01*\x8d\x02\n\tJobStatus\x12\x1a\n\x16JOB_STATUS_UNSPECIFIED\x10\x00\x12\x13\n\x0fJOB_STATUS_INIT\x10\x01\x12\x16\n\x12JOB_STATUS_PENDING\x10\x02\x12\x19\n\x15JOB_STATUS_SETTING_UP\x10\x03\x12\x16\n\x12JOB_STATUS_RUNNING\x10\x04\x12\x1c\n\x18JOB_STATUS_FAILED_DRIVER\x10\x05\x12\x18\n\x14JOB_STATUS_SUCCEEDED\x10\x06\x12\x15\n\x11JOB_STATUS_FAILED\x10\x07\x12\x1b\n\x17JOB_STATUS_FAILED_SETUP\x10\x08\x12\x18\n\x14JOB_STATUS_CANCELLED\x10\t2\x91\x07\n\x0bJobsService\x12\x39\n\x06\x41\x64\x64Job\x12\x16.jobs.v1.AddJobRequest\x1a\x17.jobs.v1.AddJobResponse\x12?\n\x08QueueJob\x12\x18.jobs.v1.QueueJobRequest\x1a\x19.jobs.v1.QueueJobResponse\x12K\n\x0cUpdateStatus\x12\x1c.jobs.v1.UpdateStatusRequest\x1a\x1d.jobs.v1.UpdateStatusResponse\x12H\n\x0bGetJobQueue\x12\x1b.jobs.v1.GetJobQueueRequest\x1a\x1c.jobs.v1.GetJobQueueResponse\x12\x45\n\nCancelJobs\x12\x1a.jobs.v1.CancelJobsRequest\x1a\x1b.jobs.v1.CancelJobsResponse\x12\x66\n\x15\x46\x61ilAllInProgressJobs\x12%.jobs.v1.FailAllInProgressJobsRequest\x1a&.jobs.v1.FailAllInProgressJobsResponse\x12\x41\n\x08TailLogs\x12\x18.jobs.v1.TailLogsRequest\x1a\x19.jobs.v1.TailLogsResponse0\x01\x12K\n\x0cGetJobStatus\x12\x1c.jobs.v1.GetJobStatusRequest\x1a\x1d.jobs.v1.GetJobStatusResponse\x12o\n\x18GetJobSubmittedTimestamp\x12(.jobs.v1.GetJobSubmittedTimestampRequest\x1a).jobs.v1.GetJobSubmittedTimestampResponse\x12\x63\n\x14GetJobEndedTimestamp\x12$.jobs.v1.GetJobEndedTimestampRequest\x1a%.jobs.v1.GetJobEndedTimestampResponse\x12Z\n\x11GetLogDirsForJobs\x12!.jobs.v1.GetLogDirsForJobsRequest\x1a\".jobs.v1.GetLogDirsForJobsResponseb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -25,8 +25,8 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_GETJOBSTATUSRESPONSE_JOBSTATUSESENTRY']._serialized_options = b'8\001'
   _globals['_GETLOGDIRSFORJOBSRESPONSE_JOBLOGDIRSENTRY']._loaded_options = None
   _globals['_GETLOGDIRSFORJOBSRESPONSE_JOBLOGDIRSENTRY']._serialized_options = b'8\001'
-  _globals['_JOBSTATUS']._serialized_start=2138
-  _globals['_JOBSTATUS']._serialized_end=2407
+  _globals['_JOBSTATUS']._serialized_start=2185
+  _globals['_JOBSTATUS']._serialized_end=2454
   _globals['_ADDJOBREQUEST']._serialized_start=48
   _globals['_ADDJOBREQUEST']._serialized_end=181
   _globals['_ADDJOBRESPONSE']._serialized_start=183
@@ -46,41 +46,41 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_GETJOBQUEUEREQUEST']._serialized_start=718
   _globals['_GETJOBQUEUEREQUEST']._serialized_end=794
   _globals['_JOBINFO']._serialized_start=797
-  _globals['_JOBINFO']._serialized_end=1041
-  _globals['_GETJOBQUEUERESPONSE']._serialized_start=1043
-  _globals['_GETJOBQUEUERESPONSE']._serialized_end=1096
-  _globals['_CANCELJOBSREQUEST']._serialized_start=1098
-  _globals['_CANCELJOBSREQUEST']._serialized_end=1192
-  _globals['_CANCELJOBSRESPONSE']._serialized_start=1194
-  _globals['_CANCELJOBSRESPONSE']._serialized_end=1241
-  _globals['_FAILALLINPROGRESSJOBSREQUEST']._serialized_start=1243
-  _globals['_FAILALLINPROGRESSJOBSREQUEST']._serialized_end=1273
-  _globals['_FAILALLINPROGRESSJOBSRESPONSE']._serialized_start=1275
-  _globals['_FAILALLINPROGRESSJOBSRESPONSE']._serialized_end=1306
-  _globals['_TAILLOGSREQUEST']._serialized_start=1308
-  _globals['_TAILLOGSREQUEST']._serialized_end=1435
-  _globals['_TAILLOGSRESPONSE']._serialized_start=1437
-  _globals['_TAILLOGSRESPONSE']._serialized_end=1492
-  _globals['_GETJOBSTATUSREQUEST']._serialized_start=1494
-  _globals['_GETJOBSTATUSREQUEST']._serialized_end=1532
-  _globals['_GETJOBSTATUSRESPONSE']._serialized_start=1535
-  _globals['_GETJOBSTATUSRESPONSE']._serialized_end=1699
-  _globals['_GETJOBSTATUSRESPONSE_JOBSTATUSESENTRY']._serialized_start=1629
-  _globals['_GETJOBSTATUSRESPONSE_JOBSTATUSESENTRY']._serialized_end=1699
-  _globals['_GETJOBSUBMITTEDTIMESTAMPREQUEST']._serialized_start=1701
-  _globals['_GETJOBSUBMITTEDTIMESTAMPREQUEST']._serialized_end=1766
-  _globals['_GETJOBSUBMITTEDTIMESTAMPRESPONSE']._serialized_start=1768
-  _globals['_GETJOBSUBMITTEDTIMESTAMPRESPONSE']._serialized_end=1821
-  _globals['_GETJOBENDEDTIMESTAMPREQUEST']._serialized_start=1823
-  _globals['_GETJOBENDEDTIMESTAMPREQUEST']._serialized_end=1884
-  _globals['_GETJOBENDEDTIMESTAMPRESPONSE']._serialized_start=1886
-  _globals['_GETJOBENDEDTIMESTAMPRESPONSE']._serialized_end=1935
-  _globals['_GETLOGDIRSFORJOBSREQUEST']._serialized_start=1937
-  _globals['_GETLOGDIRSFORJOBSREQUEST']._serialized_end=1980
-  _globals['_GETLOGDIRSFORJOBSRESPONSE']._serialized_start=1983
-  _globals['_GETLOGDIRSFORJOBSRESPONSE']._serialized_end=2135
-  _globals['_GETLOGDIRSFORJOBSRESPONSE_JOBLOGDIRSENTRY']._serialized_start=2086
-  _globals['_GETLOGDIRSFORJOBSRESPONSE_JOBLOGDIRSENTRY']._serialized_end=2135
-  _globals['_JOBSSERVICE']._serialized_start=2410
-  _globals['_JOBSSERVICE']._serialized_end=3323
+  _globals['_JOBINFO']._serialized_end=1088
+  _globals['_GETJOBQUEUERESPONSE']._serialized_start=1090
+  _globals['_GETJOBQUEUERESPONSE']._serialized_end=1143
+  _globals['_CANCELJOBSREQUEST']._serialized_start=1145
+  _globals['_CANCELJOBSREQUEST']._serialized_end=1239
+  _globals['_CANCELJOBSRESPONSE']._serialized_start=1241
+  _globals['_CANCELJOBSRESPONSE']._serialized_end=1288
+  _globals['_FAILALLINPROGRESSJOBSREQUEST']._serialized_start=1290
+  _globals['_FAILALLINPROGRESSJOBSREQUEST']._serialized_end=1320
+  _globals['_FAILALLINPROGRESSJOBSRESPONSE']._serialized_start=1322
+  _globals['_FAILALLINPROGRESSJOBSRESPONSE']._serialized_end=1353
+  _globals['_TAILLOGSREQUEST']._serialized_start=1355
+  _globals['_TAILLOGSREQUEST']._serialized_end=1482
+  _globals['_TAILLOGSRESPONSE']._serialized_start=1484
+  _globals['_TAILLOGSRESPONSE']._serialized_end=1539
+  _globals['_GETJOBSTATUSREQUEST']._serialized_start=1541
+  _globals['_GETJOBSTATUSREQUEST']._serialized_end=1579
+  _globals['_GETJOBSTATUSRESPONSE']._serialized_start=1582
+  _globals['_GETJOBSTATUSRESPONSE']._serialized_end=1746
+  _globals['_GETJOBSTATUSRESPONSE_JOBSTATUSESENTRY']._serialized_start=1676
+  _globals['_GETJOBSTATUSRESPONSE_JOBSTATUSESENTRY']._serialized_end=1746
+  _globals['_GETJOBSUBMITTEDTIMESTAMPREQUEST']._serialized_start=1748
+  _globals['_GETJOBSUBMITTEDTIMESTAMPREQUEST']._serialized_end=1813
+  _globals['_GETJOBSUBMITTEDTIMESTAMPRESPONSE']._serialized_start=1815
+  _globals['_GETJOBSUBMITTEDTIMESTAMPRESPONSE']._serialized_end=1868
+  _globals['_GETJOBENDEDTIMESTAMPREQUEST']._serialized_start=1870
+  _globals['_GETJOBENDEDTIMESTAMPREQUEST']._serialized_end=1931
+  _globals['_GETJOBENDEDTIMESTAMPRESPONSE']._serialized_start=1933
+  _globals['_GETJOBENDEDTIMESTAMPRESPONSE']._serialized_end=1982
+  _globals['_GETLOGDIRSFORJOBSREQUEST']._serialized_start=1984
+  _globals['_GETLOGDIRSFORJOBSREQUEST']._serialized_end=2027
+  _globals['_GETLOGDIRSFORJOBSRESPONSE']._serialized_start=2030
+  _globals['_GETLOGDIRSFORJOBSRESPONSE']._serialized_end=2182
+  _globals['_GETLOGDIRSFORJOBSRESPONSE_JOBLOGDIRSENTRY']._serialized_start=2133
+  _globals['_GETLOGDIRSFORJOBSRESPONSE_JOBLOGDIRSENTRY']._serialized_end=2182
+  _globals['_JOBSSERVICE']._serialized_start=2457
+  _globals['_JOBSSERVICE']._serialized_end=3370
 # @@protoc_insertion_point(module_scope)

sky/schemas/generated/servev1_pb2.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: sky/schemas/generated/servev1.proto
+# Protobuf Python Version: 5.26.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#sky/schemas/generated/servev1.proto\x12\x08serve.v1\"\x1d\n\x0cServiceNames\x12\r\n\x05names\x18\x01 \x03(\t\"s\n\rServiceStatus\x12\x33\n\x06status\x18\x01 \x03(\x0b\x32#.serve.v1.ServiceStatus.StatusEntry\x1a-\n\x0bStatusEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"m\n\x17GetServiceStatusRequest\x12\x32\n\rservice_names\x18\x01 \x01(\x0b\x32\x16.serve.v1.ServiceNamesH\x00\x88\x01\x01\x12\x0c\n\x04pool\x18\x02 \x01(\x08\x42\x10\n\x0e_service_names\"E\n\x18GetServiceStatusResponse\x12)\n\x08statuses\x18\x01 \x03(\x0b\x32\x17.serve.v1.ServiceStatus\")\n\x11\x41\x64\x64VersionRequest\x12\x14\n\x0cservice_name\x18\x01 \x01(\t\"%\n\x12\x41\x64\x64VersionResponse\x12\x0f\n\x07version\x18\x01 \x01(\x03\"}\n\x18TerminateServicesRequest\x12\x32\n\rservice_names\x18\x01 \x01(\x0b\x32\x16.serve.v1.ServiceNamesH\x00\x88\x01\x01\x12\r\n\x05purge\x18\x02 \x01(\x08\x12\x0c\n\x04pool\x18\x03 \x01(\x08\x42\x10\n\x0e_service_names\",\n\x19TerminateServicesResponse\x12\x0f\n\x07message\x18\x01 \x01(\t\"R\n\x17TerminateReplicaRequest\x12\x14\n\x0cservice_name\x18\x01 \x01(\t\x12\x12\n\nreplica_id\x18\x02 \x01(\x03\x12\r\n\x05purge\x18\x03 \x01(\x08\"+\n\x18TerminateReplicaResponse\x12\x0f\n\x07message\x18\x01 \x01(\t\"T\n\x1eWaitServiceRegistrationRequest\x12\x14\n\x0cservice_name\x18\x01 \x01(\t\x12\x0e\n\x06job_id\x18\x02 \x01(\x03\x12\x0c\n\x04pool\x18\x03 \x01(\x08\"2\n\x1fWaitServiceRegistrationResponse\x12\x0f\n\x07lb_port\x18\x01 \x01(\x03\"Y\n\x14UpdateServiceRequest\x12\x14\n\x0cservice_name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\x03\x12\x0c\n\x04mode\x18\x03 \x01(\t\x12\x0c\n\x04pool\x18\x04 \x01(\x08\"\x17\n\x15UpdateServiceResponse2\xad\x04\n\x0cServeService\x12Y\n\x10GetServiceStatus\x12!.serve.v1.GetServiceStatusRequest\x1a\".serve.v1.GetServiceStatusResponse\x12G\n\nAddVersion\x12\x1b.serve.v1.AddVersionRequest\x1a\x1c.serve.v1.AddVersionResponse\x12\\\n\x11TerminateServices\x12\".serve.v1.TerminateServicesRequest\x1a#.serve.v1.TerminateServicesResponse\x12Y\n\x10TerminateReplica\x12!.serve.v1.TerminateReplicaRequest\x1a\".serve.v1.TerminateReplicaResponse\x12n\n\x17WaitServiceRegistration\x12(.serve.v1.WaitServiceRegistrationRequest\x1a).serve.v1.WaitServiceRegistrationResponse\x12P\n\rUpdateService\x12\x1e.serve.v1.UpdateServiceRequest\x1a\x1f.serve.v1.UpdateServiceResponseb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sky.schemas.generated.servev1_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_SERVICESTATUS_STATUSENTRY']._loaded_options = None
+  _globals['_SERVICESTATUS_STATUSENTRY']._serialized_options = b'8\001'
+  _globals['_SERVICENAMES']._serialized_start=49
+  _globals['_SERVICENAMES']._serialized_end=78
+  _globals['_SERVICESTATUS']._serialized_start=80
+  _globals['_SERVICESTATUS']._serialized_end=195
+  _globals['_SERVICESTATUS_STATUSENTRY']._serialized_start=150
+  _globals['_SERVICESTATUS_STATUSENTRY']._serialized_end=195
+  _globals['_GETSERVICESTATUSREQUEST']._serialized_start=197
+  _globals['_GETSERVICESTATUSREQUEST']._serialized_end=306
+  _globals['_GETSERVICESTATUSRESPONSE']._serialized_start=308
+  _globals['_GETSERVICESTATUSRESPONSE']._serialized_end=377
+  _globals['_ADDVERSIONREQUEST']._serialized_start=379
+  _globals['_ADDVERSIONREQUEST']._serialized_end=420
+  _globals['_ADDVERSIONRESPONSE']._serialized_start=422
+  _globals['_ADDVERSIONRESPONSE']._serialized_end=459
+  _globals['_TERMINATESERVICESREQUEST']._serialized_start=461
+  _globals['_TERMINATESERVICESREQUEST']._serialized_end=586
+  _globals['_TERMINATESERVICESRESPONSE']._serialized_start=588
+  _globals['_TERMINATESERVICESRESPONSE']._serialized_end=632
+  _globals['_TERMINATEREPLICAREQUEST']._serialized_start=634
+  _globals['_TERMINATEREPLICAREQUEST']._serialized_end=716
+  _globals['_TERMINATEREPLICARESPONSE']._serialized_start=718
+  _globals['_TERMINATEREPLICARESPONSE']._serialized_end=761
+  _globals['_WAITSERVICEREGISTRATIONREQUEST']._serialized_start=763
+  _globals['_WAITSERVICEREGISTRATIONREQUEST']._serialized_end=847
+  _globals['_WAITSERVICEREGISTRATIONRESPONSE']._serialized_start=849
+  _globals['_WAITSERVICEREGISTRATIONRESPONSE']._serialized_end=899
+  _globals['_UPDATESERVICEREQUEST']._serialized_start=901
+  _globals['_UPDATESERVICEREQUEST']._serialized_end=990
+  _globals['_UPDATESERVICERESPONSE']._serialized_start=992
+  _globals['_UPDATESERVICERESPONSE']._serialized_end=1015
+  _globals['_SERVESERVICE']._serialized_start=1018
+  _globals['_SERVESERVICE']._serialized_end=1575
+# @@protoc_insertion_point(module_scope)