admin-valentin/arcade-mcp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Cleanup deploy exceptions (#410)

Browse source
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
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
arcade/arcade/cli/deployment.py
View file

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