skypilot-nightly 1.0.0.dev20241006__py3-none-any.whl → 1.0.0.dev20241007__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
sky/__init__.py CHANGED
@@ -5,7 +5,7 @@ from typing import Optional
5
5
  import urllib.request
6
6
 
7
7
  # Replaced with the current commit when building the wheels.
8
- _SKYPILOT_COMMIT_SHA = 'f4886bed755a3a6ba62554ef359fbe1dcd174d78'
8
+ _SKYPILOT_COMMIT_SHA = 'd5b6d89c83ea1ee7258f68314da4c6f8add83e04'
9
9
 
10
10
 
11
11
  def _get_git_commit():
@@ -35,7 +35,7 @@ def _get_git_commit():
35
35
 
36
36
 
37
37
  __commit__ = _get_git_commit()
38
- __version__ = '1.0.0.dev20241006'
38
+ __version__ = '1.0.0.dev20241007'
39
39
  __root_dir__ = os.path.dirname(os.path.abspath(__file__))
40
40
 
41
41
 
sky/serve/controller.py CHANGED
@@ -10,6 +10,7 @@ import traceback
10
10
  from typing import Any, Dict, List
11
11
 
12
12
  import fastapi
13
+ from fastapi import responses
13
14
  import uvicorn
14
15
 
15
16
  from sky import serve
@@ -96,7 +97,8 @@ class SkyServeController:
96
97
  def run(self) -> None:
97
98
 
98
99
  @self._app.post('/controller/load_balancer_sync')
99
- async def load_balancer_sync(request: fastapi.Request):
100
+ async def load_balancer_sync(
101
+ request: fastapi.Request) -> fastapi.Response:
100
102
  request_data = await request.json()
101
103
  # TODO(MaoZiming): Check aggregator type.
102
104
  request_aggregator: Dict[str, Any] = request_data.get(
@@ -104,18 +106,21 @@ class SkyServeController:
104
106
  timestamps: List[int] = request_aggregator.get('timestamps', [])
105
107
  logger.info(f'Received {len(timestamps)} inflight requests.')
106
108
  self._autoscaler.collect_request_information(request_aggregator)
107
- return {
109
+ return responses.JSONResponse(content={
108
110
  'ready_replica_urls':
109
111
  self._replica_manager.get_active_replica_urls()
110
- }
112
+ },
113
+ status_code=200)
111
114
 
112
115
  @self._app.post('/controller/update_service')
113
- async def update_service(request: fastapi.Request):
116
+ async def update_service(request: fastapi.Request) -> fastapi.Response:
114
117
  request_data = await request.json()
115
118
  try:
116
119
  version = request_data.get('version', None)
117
120
  if version is None:
118
- return {'message': 'Error: version is not specified.'}
121
+ return responses.JSONResponse(
122
+ content={'message': 'Error: version is not specified.'},
123
+ status_code=400)
119
124
  update_mode_str = request_data.get(
120
125
  'mode', serve_utils.DEFAULT_UPDATE_MODE.value)
121
126
  update_mode = serve_utils.UpdateMode(update_mode_str)
@@ -144,11 +149,13 @@ class SkyServeController:
144
149
  self._autoscaler.update_version(version,
145
150
  service,
146
151
  update_mode=update_mode)
147
- return {'message': 'Success'}
152
+ return responses.JSONResponse(content={'message': 'Success'},
153
+ status_code=200)
148
154
  except Exception as e: # pylint: disable=broad-except
149
155
  logger.error(f'Error in update_service: '
150
156
  f'{common_utils.format_exception(e)}')
151
- return {'message': 'Error'}
157
+ return responses.JSONResponse(content={'message': 'Error'},
158
+ status_code=500)
152
159
 
153
160
  threading.Thread(target=self._run_autoscaler).start()
154
161
 
sky/serve/serve_utils.py CHANGED
@@ -302,6 +302,10 @@ def update_service_encoded(service_name: str, version: int, mode: str) -> str:
302
302
  raise ValueError('The service is up-ed in an old version and does not '
303
303
  'support update. Please `sky serve down` '
304
304
  'it first and relaunch the service. ')
305
+ elif resp.status_code == 400:
306
+ raise ValueError(f'Client error during service update: {resp.text}')
307
+ elif resp.status_code == 500:
308
+ raise RuntimeError(f'Server error during service update: {resp.text}')
305
309
  elif resp.status_code != 200:
306
310
  raise ValueError(f'Failed to update service: {resp.text}')
307
311
 
@@ -77,6 +77,7 @@ def handle_returncode(returncode: int,
77
77
  command: The command that was run.
78
78
  error_msg: The error message to print.
79
79
  stderr: The stderr of the command.
80
+ stream_logs: Whether to stream logs.
80
81
  """
81
82
  echo = logger.error if stream_logs else logger.debug
82
83
  if returncode != 0:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: skypilot-nightly
3
- Version: 1.0.0.dev20241006
3
+ Version: 1.0.0.dev20241007
4
4
  Summary: SkyPilot: An intercloud broker for the clouds
5
5
  Author: SkyPilot Team
6
6
  License: Apache 2.0
@@ -1,4 +1,4 @@
1
- sky/__init__.py,sha256=Xtdc-QY_FgNqcqj_CNhowHLdYnKutlnzRxVUFB21BJc,5854
1
+ sky/__init__.py,sha256=qM97VOOUsAX0qlJn1dd3D1SoSS-VZ6VfhyEVw4EvgAc,5854
2
2
  sky/admin_policy.py,sha256=hPo02f_A32gCqhUueF0QYy1fMSSKqRwYEg_9FxScN_s,3248
3
3
  sky/authentication.py,sha256=TfKkVnmRIetATSEVQFp-rOOIRGqVig2i8faSQQt_ixA,20974
4
4
  sky/check.py,sha256=jLMIIJrseaZj1_o5WkbaD9XdyXIlCaT6pyAaIFdhdmA,9079
@@ -171,13 +171,13 @@ sky/provision/vsphere/common/vim_utils.py,sha256=EMWLS8ILpdx6XwUZ9I53y0B_1yFrRrl
171
171
  sky/serve/__init__.py,sha256=Qg_XPOtQsUxiN-Q3njHZRfzoMcQ_KKU1QthkiTbESDw,1661
172
172
  sky/serve/autoscalers.py,sha256=khY1oZ22PRaUQNsLCoNKH178X_NiJw0LSLOKr7_LNgY,30275
173
173
  sky/serve/constants.py,sha256=OansIC7a0Pwat-Y5SF43T9phad_EvyjKO3peZgKFEHk,4367
174
- sky/serve/controller.py,sha256=NgJqRESccAQCUfW6yoxZbIOxzdsByVuRqTBCyvfmyOI,7706
174
+ sky/serve/controller.py,sha256=gfE_gB7wxE1VxvnYqw_-KcMGc6X2kufl-NLR7sWdzdY,8172
175
175
  sky/serve/core.py,sha256=cW2SNMPMbGtOcqASHnL__B12BCIErUilGtFw3olQbjk,28947
176
176
  sky/serve/load_balancer.py,sha256=aUfDsgUT_fYrchCwJCeunMPXmAkwJAY58BEu-IN2FaA,11571
177
177
  sky/serve/load_balancing_policies.py,sha256=ExdwH_pxPYpJ6CkoTQCOPSa4lzwbq1LFFMKzmIu8ryk,2331
178
178
  sky/serve/replica_managers.py,sha256=dO962WZ_6YWRDpyNemY7SzC7fZHlNfoL4kUS3MaKwDo,57405
179
179
  sky/serve/serve_state.py,sha256=5BZSKKKxQRk-0mku17Ch4Veu4qOhaFvaOJY3zrZCkLw,19315
180
- sky/serve/serve_utils.py,sha256=scZzEpJtk-Si05VuzIB212DPovuMHFV2b-T-xWNZ2Yw,37785
180
+ sky/serve/serve_utils.py,sha256=im_1cJoJmufFxkBVnhK4nI6XlHvEXersQyIivNruJJc,38009
181
181
  sky/serve/service.py,sha256=fkfJvNJ2BO6rfV0TblZG-QkOXaCyZlpkwbGgrsTzf2w,11872
182
182
  sky/serve/service_spec.py,sha256=iRhW95SERvb4NWtV10uCuhgvW31HuSAmZZ55OX0WK8s,15309
183
183
  sky/setup_files/MANIFEST.in,sha256=BAR1TfVIHwBFfV3najggE8HDXTJyO3fNN0Yhu5aTitI,634
@@ -256,7 +256,7 @@ sky/utils/log_utils.py,sha256=yVu3etgKhiVYX8UG-JFPWZujxWBT4kwxZ5oAPIdjtGs,12054
256
256
  sky/utils/resources_utils.py,sha256=snByBxgx3Hnjfch2uysdAA3D-OAwrnuzTDHug36s5H4,6515
257
257
  sky/utils/rich_utils.py,sha256=5ZVhzlFx-nhqMXwv00eO9xC4rz7ibDlfD2lmGhZrJEY,1581
258
258
  sky/utils/schemas.py,sha256=QT0Fxri2o0SiWkky1DlZhA1dzQRQoB5OdVaej0wJvhc,28787
259
- sky/utils/subprocess_utils.py,sha256=zK0L3mvAkvKX1nFFI4IFEmRWX9ytpguhhxOTYUDKoDs,6507
259
+ sky/utils/subprocess_utils.py,sha256=3R54Elc2n8DQeO6Y8MCDJ6N6v27HDGpbNMIfCquqXYQ,6552
260
260
  sky/utils/timeline.py,sha256=ao_nm0y52ZQILfL7Y92c3pSEFRyPm_ElORC3DrI5BwQ,3936
261
261
  sky/utils/ux_utils.py,sha256=318TRunQCyJpJXonfiJ1SVotNA-6K4F2XgMEYjvWvsk,3264
262
262
  sky/utils/validator.py,sha256=cAFERCoC7jH0DFKepcU4x9SYmdrYL1iVmW9tXA18hvo,701
@@ -273,9 +273,9 @@ sky/utils/kubernetes/k8s_gpu_labeler_job.yaml,sha256=KPqp23B-zQ2SZK03jdHeF9fLTog
273
273
  sky/utils/kubernetes/k8s_gpu_labeler_setup.yaml,sha256=VLKT2KKimZu1GDg_4AIlIt488oMQvhRZWwsj9vBbPUg,3812
274
274
  sky/utils/kubernetes/rsync_helper.sh,sha256=Ma-N9a271fTfdgP5-8XIQL7KPf8IPUo-uY004PCdUFo,747
275
275
  sky/utils/kubernetes/ssh_jump_lifecycle_manager.py,sha256=RFLJ3k7MR5UN4SKHykQ0lV9SgXumoULpKYIAt1vh-HU,6560
276
- skypilot_nightly-1.0.0.dev20241006.dist-info/LICENSE,sha256=emRJAvE7ngL6x0RhQvlns5wJzGI3NEQ_WMjNmd9TZc4,12170
277
- skypilot_nightly-1.0.0.dev20241006.dist-info/METADATA,sha256=2utuR-W951m23X1syJR0VXTp8XiM4Gta_L2jnGEFt38,18945
278
- skypilot_nightly-1.0.0.dev20241006.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
279
- skypilot_nightly-1.0.0.dev20241006.dist-info/entry_points.txt,sha256=StA6HYpuHj-Y61L2Ze-hK2IcLWgLZcML5gJu8cs6nU4,36
280
- skypilot_nightly-1.0.0.dev20241006.dist-info/top_level.txt,sha256=qA8QuiNNb6Y1OF-pCUtPEr6sLEwy2xJX06Bd_CrtrHY,4
281
- skypilot_nightly-1.0.0.dev20241006.dist-info/RECORD,,
276
+ skypilot_nightly-1.0.0.dev20241007.dist-info/LICENSE,sha256=emRJAvE7ngL6x0RhQvlns5wJzGI3NEQ_WMjNmd9TZc4,12170
277
+ skypilot_nightly-1.0.0.dev20241007.dist-info/METADATA,sha256=DB1gJrJOXxSKRlbVrw-6rzua8zYvBLohJnJgwVEuDzk,18945
278
+ skypilot_nightly-1.0.0.dev20241007.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
279
+ skypilot_nightly-1.0.0.dev20241007.dist-info/entry_points.txt,sha256=StA6HYpuHj-Y61L2Ze-hK2IcLWgLZcML5gJu8cs6nU4,36
280
+ skypilot_nightly-1.0.0.dev20241007.dist-info/top_level.txt,sha256=qA8QuiNNb6Y1OF-pCUtPEr6sLEwy2xJX06Bd_CrtrHY,4
281
+ skypilot_nightly-1.0.0.dev20241007.dist-info/RECORD,,