diffio 0.1.41__tar.gz → 0.1.43__tar.gz

{diffio-0.1.41/src/diffio.egg-info → diffio-0.1.43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffio
-Version: 0.1.41
+Version: 0.1.43
 Summary: Python SDK for the Diffio API.
 Author: Diffio
 License: MIT
@@ -89,11 +89,17 @@ generation = client.create_generation(
     apiProjectId=project.apiProjectId,
     model="diffio-3.5",
     sampling={"steps": 12, "guidance": 1.5},
+    idempotencyKey="restore-job-2026-001",
 )
 
 print(generation.generationId)
+print(generation.idempotentReplay)
 ```
 
+Use one stable `idempotencyKey` for every retry of the same logical generation request.
+The response's optional `idempotentReplay` value is `True` when the API returns the
+result of an earlier request with that key. Use a new key for a different generation.
+
 ## Audio isolation helper
 
 ```py

{diffio-0.1.41 → diffio-0.1.43}/README.md

@@ -61,11 +61,17 @@ generation = client.create_generation(
     apiProjectId=project.apiProjectId,
     model="diffio-3.5",
     sampling={"steps": 12, "guidance": 1.5},
+    idempotencyKey="restore-job-2026-001",
 )
 
 print(generation.generationId)
+print(generation.idempotentReplay)
 ```
 
+Use one stable `idempotencyKey` for every retry of the same logical generation request.
+The response's optional `idempotentReplay` value is `True` when the API returns the
+result of an earlier request with that key. Use a new key for a different generation.
+
 ## Audio isolation helper
 
 ```py

{diffio-0.1.41 → diffio-0.1.43}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "diffio"
-version = "0.1.41"
+version = "0.1.43"
 description = "Python SDK for the Diffio API."
 readme = "README.md"
 requires-python = ">=3.8"

{diffio-0.1.41 → diffio-0.1.43}/src/diffio/__init__.py

@@ -70,4 +70,4 @@ __all__ = [
     "WebhooksClient",
 ]
 
-__version__ = "0.1.41"
+__version__ = "0.1.43"

{diffio-0.1.41 → diffio-0.1.43}/src/diffio/client.py

@@ -36,7 +36,6 @@ API_PREFIX = "v1"
 MODEL_ENDPOINTS = {
     "diffio-2": "diffio-2.0-generation",
     "diffio-2-flash": "diffio-2.0-flash-generation",
-    "diffio-3": "diffio-3.0-generation",
     "diffio-3.4": "diffio-3.4-generation",
     "diffio-3.5": "diffio-3.5-generation",
 }
@@ -309,6 +308,7 @@ class DiffioClient:
         model="diffio-2",
         sampling=None,
         params=None,
+        idempotencyKey=None,
         requestOptions=None,
     ):
         endpoint = MODEL_ENDPOINTS.get(model)
@@ -320,6 +320,8 @@ class DiffioClient:
             payload["sampling"] = sampling
         if params:
             payload["params"] = params
+        if idempotencyKey is not None:
+            payload["idempotencyKey"] = idempotencyKey
 
         response = self._request("POST", endpoint, json_payload=payload, requestOptions=requestOptions)
         return CreateGenerationResponse.from_dict(response)
@@ -833,6 +835,7 @@ class GenerationsClient:
         model="diffio-2",
         sampling=None,
         params=None,
+        idempotencyKey=None,
         requestOptions=None,
     ):
         return self._parent.create_generation(
@@ -840,6 +843,7 @@ class GenerationsClient:
             model=model,
             sampling=sampling,
             params=params,
+            idempotencyKey=idempotencyKey,
             requestOptions=requestOptions,
         )
 
@@ -945,6 +949,7 @@ class GenerationsClient:
         model="diffio-2",
         sampling=None,
         params=None,
+        idempotencyKey=None,
         pollInterval=2.0,
         timeout=600.0,
         onProgress=None,
@@ -958,6 +963,7 @@ class GenerationsClient:
             model=model,
             sampling=sampling,
             params=params,
+            idempotencyKey=idempotencyKey,
             requestOptions=requestOptions,
         )
         progress = self.wait_for_complete(

{diffio-0.1.41 → diffio-0.1.43}/src/diffio/types.py

@@ -76,11 +76,19 @@ class ListProjectsResponse:
 
 
 class CreateGenerationResponse:
-    def __init__(self, generationId, apiProjectId, modelKey, status):
+    def __init__(
+        self,
+        generationId,
+        apiProjectId,
+        modelKey,
+        status,
+        idempotentReplay=None,
+    ):
         self.generationId = generationId
         self.apiProjectId = apiProjectId
         self.modelKey = modelKey
         self.status = status
+        self.idempotentReplay = idempotentReplay
 
     @classmethod
     def from_dict(cls, data):
@@ -89,6 +97,7 @@ class CreateGenerationResponse:
             apiProjectId=data["apiProjectId"],
             modelKey=data["modelKey"],
             status=data["status"],
+            idempotentReplay=data.get("idempotentReplay"),
         )
 
 
@@ -419,7 +428,7 @@ class AudioIsolationResult:
         self.generation = generation
 
 
-ModelKey = ("diffio-2", "diffio-2-flash", "diffio-3", "diffio-3.4", "diffio-3.5")
+ModelKey = ("diffio-2", "diffio-2-flash", "diffio-3.4", "diffio-3.5")
 DownloadType = ("audio", "mp3", "video", "transcript")
 WebhookMode = ("test", "live")
 WebhookEventType = (

{diffio-0.1.41 → diffio-0.1.43/src/diffio.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffio
-Version: 0.1.41
+Version: 0.1.43
 Summary: Python SDK for the Diffio API.
 Author: Diffio
 License: MIT
@@ -89,11 +89,17 @@ generation = client.create_generation(
     apiProjectId=project.apiProjectId,
     model="diffio-3.5",
     sampling={"steps": 12, "guidance": 1.5},
+    idempotencyKey="restore-job-2026-001",
 )
 
 print(generation.generationId)
+print(generation.idempotentReplay)
 ```
 
+Use one stable `idempotencyKey` for every retry of the same logical generation request.
+The response's optional `idempotentReplay` value is `True` when the API returns the
+result of an earlier request with that key. Use a new key for a different generation.
+
 ## Audio isolation helper
 
 ```py

{diffio-0.1.41 → diffio-0.1.43}/tests/test_client.py

@@ -5,7 +5,8 @@ from pathlib import Path
 import httpx
 import pytest
 
-from diffio import DiffioClient
+from diffio import DiffioClient, ModelKey
+from diffio.client import MODEL_ENDPOINTS
 
 
 def test_create_project_payload_and_headers(tmp_path: Path):
@@ -308,7 +309,143 @@ def test_create_generation_routes_to_diffio_3_5_endpoint():
 
     assert received["path"] == "/v1/diffio-3.5-generation"
     assert received["payload"]["apiProjectId"] == "proj_35"
+    assert "idempotencyKey" not in received["payload"]
     assert response.modelKey == "diffio-3.5"
+    assert response.idempotentReplay is None
+
+
+def test_create_generation_sends_idempotency_key_and_parses_replay():
+    received = {}
+
+    def handler(request):
+        received["path"] = request.url.path
+        received["payload"] = json.loads(request.content.decode("utf-8"))
+        return httpx.Response(
+            200,
+            json={
+                "generationId": "gen_original",
+                "apiProjectId": "proj_123",
+                "modelKey": "diffio-2",
+                "status": "queued",
+                "idempotentReplay": True,
+            },
+        )
+
+    transport = httpx.MockTransport(handler)
+    http_client = httpx.Client(base_url="https://api.test", transport=transport)
+    client = DiffioClient(apiKey="diffio_live_test", baseUrl="https://api.test", httpClient=http_client)
+
+    response = client.create_generation(
+        apiProjectId="proj_123",
+        idempotencyKey="restore-job-2026-001",
+    )
+
+    assert received["path"] == "/v1/diffio-2.0-generation"
+    assert received["payload"] == {
+        "apiProjectId": "proj_123",
+        "idempotencyKey": "restore-job-2026-001",
+    }
+    assert response.generationId == "gen_original"
+    assert response.idempotentReplay is True
+
+
+def test_generations_create_forwards_idempotency_key():
+    received = {}
+
+    def handler(request):
+        received["payload"] = json.loads(request.content.decode("utf-8"))
+        return httpx.Response(
+            200,
+            json={
+                "generationId": "gen_123",
+                "apiProjectId": "proj_123",
+                "modelKey": "diffio-2",
+                "status": "queued",
+            },
+        )
+
+    transport = httpx.MockTransport(handler)
+    http_client = httpx.Client(base_url="https://api.test", transport=transport)
+    client = DiffioClient(apiKey="diffio_live_test", baseUrl="https://api.test", httpClient=http_client)
+
+    response = client.generations.create(
+        apiProjectId="proj_123",
+        idempotencyKey="restore-job-2026-wrapper",
+    )
+
+    assert received["payload"]["idempotencyKey"] == "restore-job-2026-wrapper"
+    assert response.idempotentReplay is None
+
+
+def test_generations_create_and_wait_forwards_idempotency_key():
+    received = {}
+
+    def handler(request):
+        if request.url.path == "/v1/diffio-2.0-generation":
+            received["createPayload"] = json.loads(request.content.decode("utf-8"))
+            return httpx.Response(
+                200,
+                json={
+                    "generationId": "gen_123",
+                    "apiProjectId": "proj_123",
+                    "modelKey": "diffio-2",
+                    "status": "queued",
+                    "idempotentReplay": False,
+                },
+            )
+        if request.url.path == "/v1/get_generation_progress":
+            return httpx.Response(
+                200,
+                json={
+                    "generationId": "gen_123",
+                    "apiProjectId": "proj_123",
+                    "status": "complete",
+                    "hasVideo": False,
+                    "preProcessing": {
+                        "jobId": "job-pre",
+                        "jobState": "SUCCEEDED",
+                        "status": "complete",
+                        "progress": 100,
+                        "statusMessage": None,
+                        "error": None,
+                        "errorDetails": None,
+                    },
+                    "inference": {
+                        "jobId": "job-run",
+                        "jobState": "SUCCEEDED",
+                        "status": "complete",
+                        "progress": 100,
+                        "statusMessage": None,
+                        "error": None,
+                        "errorDetails": None,
+                    },
+                },
+            )
+        return httpx.Response(404, json={"error": "not found"})
+
+    transport = httpx.MockTransport(handler)
+    http_client = httpx.Client(base_url="https://api.test", transport=transport)
+    client = DiffioClient(apiKey="diffio_live_test", baseUrl="https://api.test", httpClient=http_client)
+
+    generation, progress = client.generations.create_and_wait(
+        apiProjectId="proj_123",
+        idempotencyKey="restore-job-2026-002",
+        pollInterval=0,
+    )
+
+    assert received["createPayload"] == {
+        "apiProjectId": "proj_123",
+        "idempotencyKey": "restore-job-2026-002",
+    }
+    assert generation.idempotentReplay is False
+    assert progress.status == "complete"
+
+
+def test_advertised_models_match_runtime_model_support():
+    assert set(ModelKey) == set(MODEL_ENDPOINTS)
+    assert "diffio-3" not in ModelKey
+    assert "diffio-3.0-generation" not in MODEL_ENDPOINTS.values()
+    assert MODEL_ENDPOINTS["diffio-3.4"] == "diffio-3.4-generation"
 
 
 def test_restore_audio_runs_full_flow_and_downloads(tmp_path: Path, monkeypatch):
@@ -439,6 +576,18 @@ def test_create_generation_rejects_unknown_model():
         client.create_generation(apiProjectId="proj", model="diffio-unknown")
 
 
+def test_create_generation_rejects_diffio_3():
+    def handler(request):
+        raise AssertionError("Unsupported models must be rejected before an HTTP request")
+
+    transport = httpx.MockTransport(handler)
+    http_client = httpx.Client(base_url="https://api.test", transport=transport)
+    client = DiffioClient(apiKey="diffio_live_test", baseUrl="https://api.test", httpClient=http_client)
+
+    with pytest.raises(ValueError, match="Unsupported model: diffio-3"):
+        client.create_generation(apiProjectId="proj", model="diffio-3")
+
+
 def test_get_generation_progress_payload_and_response():
     received = {}