from typing import Any, Dict, Optional | |
# from huggingface_hub import SpaceStage | |
from pydantic import BaseModel, Field | |
class DeployRequest(BaseModel): | |
hf_token: str = Field(..., description="HF token with write permission") | |
git_repo_url: str = Field(...) | |
deploy_path: str = Field("/") | |
space_name: str = Field(...) | |
space_port: int = Field(7860) | |
description: str = Field("") | |
env_vars: Dict[str, str] = Field(default_factory=dict) | |
private: bool = Field(False) | |
class DeployStatus(BaseModel): | |
task_id: str | |
status: str # PENDING | IN_PROGRESS | SUCCESS | FAILED | |
detail: Optional[Any] = None | |