gridlinegpu 0.1.0__tar.gz → 0.1.2__tar.gz

{gridlinegpu-0.1.0 → gridlinegpu-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gridlinegpu
-Version: 0.1.0
+Version: 0.1.2
 Summary: Python client for Gridline YAML workload routing
 Author: Gridline GPU
 License: Proprietary
@@ -22,7 +22,7 @@ fresh plan when you supply a cost ceiling and idempotency key.
 ## Install
 
 ```sh
-python -m pip install gridlinegpu
+python -m pip install 'gridlinegpu>=0.1.2,<0.2'
 ```
 
 Create a scoped token using the Gridline CLI after browser sign-in. The token stays in a
@@ -48,7 +48,8 @@ print(plan.review)  # Check selected GPU, exclusions, estimate, and maximum.
 # Launch only after reviewing the fresh plan and accepting its maximum spend.
 launch = client.launch(plan, max_cost_usd="5.00", idempotency_key="my-job-2026-09-27")
 job_id = launch["jobId"]
-print(client.wait(job_id, timeout=7200)["job"]["outcome"])
+ready = client.wait(job_id, timeout=7200)
+print(ready["job"]["outputs"], ready["job"]["outcome"])
 print(client.outputs(job_id, "results"))
 ```
 
@@ -57,3 +58,5 @@ changes a plan or retries an ambiguous paid request. If launch outcome is unknow
 inspect `LaunchOutcomeUnknown.workload_id` with the same idempotency key; do not use a
 new key. `outputs()` verifies byte count and SHA-256 before creating private files.
 Cancellation requests stop; cleanup and final provider billing can remain pending.
+For finite jobs, `wait()` returns when outputs are complete, even if cleanup and billing
+leave `outcome` pending. Check `status()` later for final settlement.

{gridlinegpu-0.1.0 → gridlinegpu-0.1.2}/README.md

@@ -7,7 +7,7 @@ fresh plan when you supply a cost ceiling and idempotency key.
 ## Install
 
 ```sh
-python -m pip install gridlinegpu
+python -m pip install 'gridlinegpu>=0.1.2,<0.2'
 ```
 
 Create a scoped token using the Gridline CLI after browser sign-in. The token stays in a
@@ -33,7 +33,8 @@ print(plan.review)  # Check selected GPU, exclusions, estimate, and maximum.
 # Launch only after reviewing the fresh plan and accepting its maximum spend.
 launch = client.launch(plan, max_cost_usd="5.00", idempotency_key="my-job-2026-09-27")
 job_id = launch["jobId"]
-print(client.wait(job_id, timeout=7200)["job"]["outcome"])
+ready = client.wait(job_id, timeout=7200)
+print(ready["job"]["outputs"], ready["job"]["outcome"])
 print(client.outputs(job_id, "results"))
 ```
 
@@ -42,3 +43,5 @@ changes a plan or retries an ambiguous paid request. If launch outcome is unknow
 inspect `LaunchOutcomeUnknown.workload_id` with the same idempotency key; do not use a
 new key. `outputs()` verifies byte count and SHA-256 before creating private files.
 Cancellation requests stop; cleanup and final provider billing can remain pending.
+For finite jobs, `wait()` returns when outputs are complete, even if cleanup and billing
+leave `outcome` pending. Check `status()` later for final settlement.

{gridlinegpu-0.1.0 → gridlinegpu-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "gridlinegpu"
-version = "0.1.0"
+version = "0.1.2"
 description = "Python client for Gridline YAML workload routing"
 readme = "README.md"
 requires-python = ">=3.10"

{gridlinegpu-0.1.0 → gridlinegpu-0.1.2}/src/gridlinegpu/client.py

@@ -10,6 +10,7 @@ import json
 import os
 import re
 import time
+import uuid
 from dataclasses import dataclass
 from datetime import datetime, timezone
 from decimal import Decimal
@@ -29,6 +30,7 @@ _KEY = re.compile(r"^[A-Za-z0-9][A-Za-z0-9._:-]{0,127}$")
 _SHA = re.compile(r"^[0-9a-f]{64}$")
 _MAX_YAML = 65_536
 _MAX_DOWNLOAD = 8 * 1024 * 1024
+_USER_AGENT = "GridlinePythonSDK/1"
 
 
 class _NoRedirect(HTTPRedirectHandler):
@@ -97,7 +99,10 @@ class GridlineClient:
             "POST", f"{_API}/validate", {"schemaVersion": 1, "yaml": source, "attachmentBindings": []}
         )
 
-    def plan(self, yaml_path: str | Path) -> PlannedWorkload:
+    def plan(self, yaml_path: str | Path, *, submission_key: str | None = None) -> PlannedWorkload:
+        """Create a fresh submission; reuse submission_key only to retry an uncertain submit."""
+        if submission_key is not None and not _KEY.fullmatch(submission_key):
+            raise ValueError("submission_key must contain 1–128 safe characters")
         path, source, document = _read_yaml(yaml_path)
         validated = self.validate(path)
         available = self.availability()
@@ -113,8 +118,9 @@ class GridlineClient:
             raise GridlineError("Invalid server attachment limit", code="remote_contract_invalid")
         bindings = [self._stage(path.parent, name, item, ceiling) for name, item in inputs]
         body = {"schemaVersion": 1, "yaml": source, "attachmentBindings": bindings}
-        stable = _sha(json.dumps(body, sort_keys=True, separators=(",", ":")).encode())
-        submitted = self._json("POST", f"{_API}/submit", body, key=f"sdk-submit-{stable}")
+        submitted = self._json(
+            "POST", f"{_API}/submit", body, key=submission_key or f"sdk-submit-{uuid.uuid4()}"
+        )
         workload_id = _id(submitted.get("workloadId"), "workloadId")
         result = self._json("POST", f"{_API}/{quote(workload_id)}/plan")
         if result.get("workloadId") != workload_id or result.get("workloadHash") != submitted.get("workloadHash"):
@@ -205,13 +211,16 @@ class GridlineClient:
         return result
 
     def wait(self, job_id: str, *, timeout: float = 3600, interval: float = 5) -> dict[str, Any]:
+        """Wait for finite-job outputs or a terminal outcome; billing may finish later."""
         if timeout <= 0 or interval <= 0:
             raise ValueError("timeout and interval must be positive")
         deadline = time.monotonic() + timeout
         while True:
             result = self.status(job_id)
-            state = result["job"].get("outcome")
-            if state in ("succeeded", "failed", "cancelled", "timed_out"):
+            job = result["job"]
+            if job.get("outcome") in ("succeeded", "failed", "cancelled", "timed_out") or (
+                job.get("execution", {}).get("mode") == "finite" and job.get("outputs") == "complete"
+            ):
                 return result
             if time.monotonic() >= deadline:
                 raise TimeoutError(f"Job {job_id} did not finish before timeout")
@@ -307,7 +316,7 @@ class GridlineClient:
                headers: dict[str, str] | None = None, maximum: int) -> bytes:
         if not isinstance(url, str) or urlparse(url).scheme != "https":
             raise GridlineError("Grant URL must use HTTPS", code="remote_contract_invalid")
-        request = Request(url, data=data, headers=headers or {}, method=method)
+        request = Request(url, data=data, headers={"User-Agent": _USER_AGENT, **(headers or {})}, method=method)
         try:
             with self._opener.open(request, timeout=self._timeout) as response:
                 raw = response.read(maximum + 1)

{gridlinegpu-0.1.0 → gridlinegpu-0.1.2}/src/gridlinegpu.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gridlinegpu
-Version: 0.1.0
+Version: 0.1.2
 Summary: Python client for Gridline YAML workload routing
 Author: Gridline GPU
 License: Proprietary
@@ -22,7 +22,7 @@ fresh plan when you supply a cost ceiling and idempotency key.
 ## Install
 
 ```sh
-python -m pip install gridlinegpu
+python -m pip install 'gridlinegpu>=0.1.2,<0.2'
 ```
 
 Create a scoped token using the Gridline CLI after browser sign-in. The token stays in a
@@ -48,7 +48,8 @@ print(plan.review)  # Check selected GPU, exclusions, estimate, and maximum.
 # Launch only after reviewing the fresh plan and accepting its maximum spend.
 launch = client.launch(plan, max_cost_usd="5.00", idempotency_key="my-job-2026-09-27")
 job_id = launch["jobId"]
-print(client.wait(job_id, timeout=7200)["job"]["outcome"])
+ready = client.wait(job_id, timeout=7200)
+print(ready["job"]["outputs"], ready["job"]["outcome"])
 print(client.outputs(job_id, "results"))
 ```
 
@@ -57,3 +58,5 @@ changes a plan or retries an ambiguous paid request. If launch outcome is unknow
 inspect `LaunchOutcomeUnknown.workload_id` with the same idempotency key; do not use a
 new key. `outputs()` verifies byte count and SHA-256 before creating private files.
 Cancellation requests stop; cleanup and final provider billing can remain pending.
+For finite jobs, `wait()` returns when outputs are complete, even if cleanup and billing
+leave `outcome` pending. Check `status()` later for final settlement.

{gridlinegpu-0.1.0 → gridlinegpu-0.1.2}/tests/test_client.py

@@ -31,6 +31,7 @@ class FakeOpener:
     def __init__(self):
         self.calls = []
         self.changed = False
+        self.job = {"jobId": "job_1", "outcome": "succeeded"}
         self.expiry = (datetime.now(timezone.utc) + timedelta(minutes=2)).isoformat()
 
     def open(self, request, timeout):
@@ -55,7 +56,7 @@ class FakeOpener:
         elif path == "/api/workloads/yaml/yaml_1/launch":
             result = {"workloadId": "yaml_1", "planHash": "c" * 64, "jobId": "job_1"}
         elif path == "/api/workloads/yaml/jobs/job_1":
-            result = {"job": {"jobId": "job_1", "outcome": "succeeded"}}
+            result = {"job": self.job}
         elif path == "/api/workloads/yaml/jobs/job_1/output":
             result = {"jobId": "job_1", "state": "available", "manifest": {"outputs": [{
                 "name": "result.json", "artifactId": "output_1", "sha256": self.output_digest,
@@ -103,6 +104,21 @@ class ClientTest(unittest.TestCase):
             self.client.launch(plan, max_cost_usd="5", idempotency_key="same-key")
         self.assertFalse(any(path.endswith("/launch") for _, path, _, _ in self.opener.calls))
 
+    def test_identical_yaml_creates_fresh_submission_unless_retry_key_is_explicit(self):
+        for _ in range(2):
+            self.client.plan(self.path)
+        keys = [next(value for name, value in headers.items() if name.lower() == "idempotency-key")
+                for _, path, headers, _ in self.opener.calls
+                if path.endswith("/submit")]
+        self.assertEqual(len(keys), 2)
+        self.assertNotEqual(keys[0], keys[1])
+        self.client.plan(self.path, submission_key="retry-this-submit")
+        self.client.plan(self.path, submission_key="retry-this-submit")
+        keys = [next(value for name, value in headers.items() if name.lower() == "idempotency-key")
+                for _, path, headers, _ in self.opener.calls
+                if path.endswith("/submit")]
+        self.assertEqual(keys[-2:], ["retry-this-submit", "retry-this-submit"])
+
     def test_cost_ceiling_and_yaml_change_cannot_launch(self):
         plan = self.client.plan(self.path)
         with self.assertRaisesRegex(GridlineError, "ceiling"):
@@ -111,6 +127,21 @@ class ClientTest(unittest.TestCase):
         with self.assertRaisesRegex(GridlineError, "changed"):
             self.client.launch(plan, max_cost_usd="5", idempotency_key="same-key")
 
+    def test_requests_identify_sdk_to_production_edge(self):
+        self.client.availability()
+        headers = self.opener.calls[-1][2]
+        self.assertEqual(next(value for name, value in headers.items() if name.lower() == "user-agent"),
+                         "GridlinePythonSDK/1")
+
+    def test_wait_returns_when_finite_outputs_are_complete_before_billing(self):
+        self.opener.job = {"jobId": "job_1", "outcome": "pending", "execution": {"mode": "finite"},
+                           "outputs": "complete", "cleanup": "pending", "billing": "pending"}
+        result = self.client.wait("job_1", timeout=1)
+        self.assertEqual(result["job"]["outcome"], "pending")
+        self.assertEqual(result["job"]["outputs"], "complete")
+        self.assertEqual(self.client.outputs("job_1", Path(self.folder.name) / "results")["state"],
+                         "available")
+
 
 if __name__ == "__main__":
     unittest.main()