Spaces:
Build error
Build error
| # backend/routers/apply.py | |
| from fastapi import APIRouter | |
| from pydantic import BaseModel | |
| import uuid | |
| import yaml | |
| from agentctl.k8s_client import K8sClient | |
| router = APIRouter(tags=["Apply Manifest"]) | |
| class ApplyRequest(BaseModel): | |
| yaml: str | |
| class ApplyResponse(BaseModel): | |
| success: bool | |
| message: str | |
| raw: str | None = None | |
| def _inject_unique_name(yaml_text: str) -> str: | |
| """ | |
| Add a short UUID suffix to metadata.name for Job/CronJob/Deployment | |
| to avoid 409 AlreadyExists errors when users re-apply the same YAML. | |
| """ | |
| try: | |
| data = yaml.safe_load(yaml_text) | |
| if not isinstance(data, dict): | |
| return yaml_text | |
| kind = data.get("kind") | |
| if kind not in {"Job", "CronJob", "Deployment"}: | |
| return yaml_text | |
| meta = data.get("metadata") or {} | |
| base_name = meta.get("name") | |
| if not base_name: | |
| return yaml_text | |
| suffix = str(uuid.uuid4())[:5] | |
| new_name = f"{base_name}-{suffix}" | |
| meta["name"] = new_name | |
| data["metadata"] = meta | |
| # For CronJob make sure template labels are consistent | |
| if kind == "CronJob": | |
| tmpl = ( | |
| data.get("spec", {}) | |
| .get("jobTemplate", {}) | |
| .get("spec", {}) | |
| .get("template", {}) | |
| ) | |
| if isinstance(tmpl, dict): | |
| md = tmpl.get("metadata") or {} | |
| labels = md.get("labels") or {} | |
| labels["app"] = new_name | |
| md["labels"] = labels | |
| tmpl["metadata"] = md | |
| return yaml.safe_dump(data) | |
| except Exception: | |
| # on failure, just return original yaml | |
| return yaml_text | |
| def apply_manifest(req: ApplyRequest): | |
| """ | |
| Apply a YAML manifest to the Kubernetes cluster via K8sClient. | |
| """ | |
| # Lazy K8sClient instantiation so missing env vars don't crash import | |
| client = K8sClient() | |
| final_yaml = _inject_unique_name(req.yaml) | |
| result = client.apply_manifest(final_yaml) | |
| try: | |
| raw_str = str(result.raw_response) | |
| except Exception: | |
| raw_str = None | |
| return ApplyResponse( | |
| success=result.success, | |
| message=result.message, | |
| raw=raw_str, | |
| ) | |