Cleanup deploy exceptions (#410)

This commit is contained in:
Sterling Dreyer 2025-05-23 17:21:47 -07:00 committed by GitHub
parent 5189a4701b
commit f8907684b7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -8,10 +8,9 @@ import time
from pathlib import Path
from typing import Any
import httpx
import toml
from arcadepy import Arcade, NotFoundError
from httpx import Client
from httpx import Client, ConnectError, HTTPStatusError, TimeoutException
from packaging.requirements import Requirement
from pydantic import BaseModel, field_serializer, field_validator, model_validator
from rich.console import Console
@ -130,11 +129,13 @@ class Request(BaseModel):
timeout=10,
)
worker_resp.raise_for_status()
except httpx.TimeoutException:
except TimeoutException:
time.sleep(1)
continue
except httpx.ConnectError as e:
except ConnectError as e:
raise ValueError(f"Failed to connect to cloud: {e}")
except HTTPStatusError as e:
raise ValueError(f"Failed to start worker: {e.response.json()}")
except Exception as e:
raise ValueError(f"Failed to start worker: {e}")
status = worker_resp.json()["data"]["status"]
@ -152,11 +153,14 @@ class Request(BaseModel):
timeout=360,
)
cloud_response.raise_for_status()
except httpx.ConnectError as e:
except ConnectError as e:
raise ValueError(f"Failed to connect to cloud: {e}")
except HTTPStatusError as e:
raise ValueError(f"Failed to start worker: {e.response.json()}")
except Exception as e:
# change this so it handles errors that aren't just from cloud
raise ValueError(f"Failed to start worker: {e}")
parse_deployment_response(cloud_response.json())
worker_data = self.poll_worker_status(cloud_client, self.name)