Upload 9 files

src/jobs.py

@@ -41,15 +41,40 @@ def _launch_job(*, token: str, env: dict[str, str], bucket_source: str, flavor:
     )
 
 
+def _job_field(job: Any, name: str, default: Any = None) -> Any:
+    if isinstance(job, dict):
+        return job.get(name, default)
+    return getattr(job, name, default)
+
+
+def _job_status_stage(job: Any) -> Any:
+    status = _job_field(job, "status")
+    if isinstance(status, dict):
+        return status.get("stage") or status.get("status") or status.get("state")
+    return getattr(status, "stage", None) or getattr(status, "status", None) or getattr(status, "state", None)
+
+
+def _job_url_from_id(*, job_id: str | None, bucket_source: str) -> str:
+    if not job_id:
+        return ""
+    owner = str(bucket_source or "").split("/", 1)[0].strip()
+    return f"https://huggingface.co/jobs/{owner}/{job_id}" if owner else ""
+
+
 def _job_result(job: Any, *, run_id: str, kind: str, bucket_source: str, extra: dict[str, Any] | None = None) -> dict[str, Any]:
+    job_id = str(_job_field(job, "id") or _job_field(job, "job_id") or _job_field(job, "jobId") or "").strip()
+    job_url = str(_job_field(job, "url") or _job_field(job, "job_url") or "").strip()
+    if not job_url:
+        job_url = _job_url_from_id(job_id=job_id, bucket_source=bucket_source)
     payload: dict[str, Any] = {
         "run_id": run_id,
         "kind": kind,
-        "job_id": job.id,
-        "job_url": getattr(job, "url", None),
-        "status": getattr(getattr(job, "status", None), "stage", None),
+        "job_id": job_id,
+        "job_url": job_url,
+        "status": _job_status_stage(job),
         "bucket_source": bucket_source,
         "bucket_uri": bucket_uri_from_source(bucket_source),
+        "created_by": str(bucket_source or "").split("/", 1)[0].strip(),
     }
     if extra:
         payload.update(extra)