poly-hammer-worker 0.1.1.dev10__tar.gz → 0.1.1.dev11__tar.gz

{poly_hammer_worker-0.1.1.dev10 → poly_hammer_worker-0.1.1.dev11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: poly-hammer-worker
-Version: 0.1.1.dev10
+Version: 0.1.1.dev11
 Summary: Poly Hammer self-hosted GPU worker agent
 License-Expression: MIT
 Requires-Python: >=3.13

{poly_hammer_worker-0.1.1.dev10 → poly_hammer_worker-0.1.1.dev11}/poly_hammer_worker/image_discovery.py

@@ -87,6 +87,10 @@ def discover_local_images(
             continue
 
         image_ref = _best_tag(image)
+        if not image_ref:
+            logger.debug("Skipping untagged image %s", image.short_id)
+            continue
+
         service_id = ph_labels["service-id"]
         min_cuda_version: str | None = ph_labels.get("min-cuda-version")
 
@@ -142,13 +146,12 @@ def discover_local_images(
     return results
 
 
-def _best_tag(image: Any) -> str:
-    """Return the most useful tag for an image, or fall back to short ID."""
+def _best_tag(image: Any) -> str | None:
+    """Return the most useful tag for an image, or None if untagged."""
     tags = image.tags
     if tags:
         return tags[0]
-    short_id = image.short_id
-    return short_id.replace("sha256:", "") if short_id else "<untagged>"
+    return None
 
 
 def _parse_int(value: str) -> int:

{poly_hammer_worker-0.1.1.dev10 → poly_hammer_worker-0.1.1.dev11}/poly_hammer_worker/worker_loop.py

@@ -289,11 +289,24 @@ def _execute_job(
         elapsed = time.monotonic() - start_time
 
         # ---- Report completion ------------------------------------------------
+        # The container uploads to S3 via the presigned URL but doesn't
+        # echo the key back in its result JSON.  Fall back to the s3_key
+        # from the job payload so the backend can store it.
+        result_url = result.get("result_url") or job.get("s3_key", "")
+
+        # Build result_metadata: use explicit field if present, otherwise
+        # collect useful top-level fields the container emitted.
+        result_metadata = result.get("result_metadata") or {
+            k: v
+            for k, v in result.items()
+            if k not in ("type", "status", "result_url", "result_metadata")
+        }
+
         portal.complete_job(
             job_id,
             elapsed_seconds=elapsed,
-            result_url=result.get("result_url"),
-            result_metadata=result.get("result_metadata"),
+            result_url=result_url,
+            result_metadata=result_metadata,
         )
         console.print(f"  [green]Completed in {elapsed:.1f}s[/green]")
 

{poly_hammer_worker-0.1.1.dev10 → poly_hammer_worker-0.1.1.dev11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "poly-hammer-worker"
-version = "0.1.1.dev10"
+version = "0.1.1.dev11"
 description = "Poly Hammer self-hosted GPU worker agent"
 readme = "README.md"
 requires-python = ">=3.13"