diffio 0.1.92__tar.gz → 0.1.93__tar.gz

{diffio-0.1.92/src/diffio.egg-info → diffio-0.1.93}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffio
-Version: 0.1.92
+Version: 0.1.93
 Summary: Python SDK for the Diffio API.
 Author: Diffio
 License: MIT
@@ -95,7 +95,7 @@ project = client.create_project(
 
 generation = client.create_generation(
     apiProjectId=project.apiProjectId,
-    model="diffio-3.5",
+    model="diffio-4.0-flash",
     sampling={"steps": 12, "guidance": 1.5},
     idempotencyKey="restore-job-2026-001",
     requestOptions={"maxRetries": 2},
@@ -117,7 +117,7 @@ from diffio import DiffioClient
 client = DiffioClient(apiKey="diffio_live_...")
 result = client.audio_isolation.isolate(
     filePath="sample.wav",
-    model="diffio-3.5",
+    model="diffio-4.0-flash",
     sampling={"steps": 12, "guidance": 1.5},
 )
 
@@ -134,7 +134,7 @@ from diffio import DiffioClient
 client = DiffioClient(apiKey="diffio_live_...")
 audio_bytes, info = client.restore_audio(
     filePath="sample.wav",
-    model="diffio-3.5",
+    model="diffio-4.0-flash",
     sampling={"steps": 12, "guidance": 1.5},
     onProgress=lambda progress: print(progress.status),
 )

{diffio-0.1.92 → diffio-0.1.93}/README.md

@@ -67,7 +67,7 @@ project = client.create_project(
 
 generation = client.create_generation(
     apiProjectId=project.apiProjectId,
-    model="diffio-3.5",
+    model="diffio-4.0-flash",
     sampling={"steps": 12, "guidance": 1.5},
     idempotencyKey="restore-job-2026-001",
     requestOptions={"maxRetries": 2},
@@ -89,7 +89,7 @@ from diffio import DiffioClient
 client = DiffioClient(apiKey="diffio_live_...")
 result = client.audio_isolation.isolate(
     filePath="sample.wav",
-    model="diffio-3.5",
+    model="diffio-4.0-flash",
     sampling={"steps": 12, "guidance": 1.5},
 )
 
@@ -106,7 +106,7 @@ from diffio import DiffioClient
 client = DiffioClient(apiKey="diffio_live_...")
 audio_bytes, info = client.restore_audio(
     filePath="sample.wav",
-    model="diffio-3.5",
+    model="diffio-4.0-flash",
     sampling={"steps": 12, "guidance": 1.5},
     onProgress=lambda progress: print(progress.status),
 )

{diffio-0.1.92 → diffio-0.1.93}/pyproject.toml

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

{diffio-0.1.92 → diffio-0.1.93}/src/diffio/__init__.py

@@ -74,4 +74,4 @@ __all__ = [
     "WebhooksClient",
 ]
 
-__version__ = "0.1.92"
+__version__ = "0.1.93"

{diffio-0.1.92 → diffio-0.1.93}/src/diffio/client.py

@@ -38,6 +38,8 @@ MODEL_ENDPOINTS = {
     "diffio-2-flash": "diffio-2.0-flash-generation",
     "diffio-3.4": "diffio-3.4-generation",
     "diffio-3.5": "diffio-3.5-generation",
+    "diffio-4.0-flash": "diffio-4.0-flash-generation",
+    "diffio-4.0-pro": "diffio-4.0-pro-generation",
 }
 DEFAULT_RETRY_STATUS_CODES = [408, 429, 500, 502, 503, 504]
 DEFAULT_RETRY_BACKOFF = 0.5
@@ -305,7 +307,7 @@ class DiffioClient:
         self,
         *,
         apiProjectId,
-        model="diffio-2",
+        model="diffio-4.0-flash",
         sampling=None,
         params=None,
         idempotencyKey=None,
@@ -717,7 +719,7 @@ class DiffioClient:
         contentType=None,
         contentLength=None,
         fileFormat=None,
-        model="diffio-2",
+        model="diffio-4.0-flash",
         sampling=None,
         projectParams=None,
         generationParams=None,
@@ -852,7 +854,7 @@ class GenerationsClient:
         self,
         *,
         apiProjectId,
-        model="diffio-2",
+        model="diffio-4.0-flash",
         sampling=None,
         params=None,
         idempotencyKey=None,
@@ -966,7 +968,7 @@ class GenerationsClient:
         self,
         *,
         apiProjectId,
-        model="diffio-2",
+        model="diffio-4.0-flash",
         sampling=None,
         params=None,
         idempotencyKey=None,
@@ -1090,7 +1092,7 @@ class AudioIsolationClient:
         contentType=None,
         contentLength=None,
         fileFormat=None,
-        model="diffio-2",
+        model="diffio-4.0-flash",
         sampling=None,
         projectParams=None,
         generationParams=None,
@@ -1115,7 +1117,7 @@ class AudioIsolationClient:
         contentType=None,
         contentLength=None,
         fileFormat=None,
-        model="diffio-2",
+        model="diffio-4.0-flash",
         sampling=None,
         projectParams=None,
         generationParams=None,
@@ -1147,7 +1149,7 @@ class AudioIsolationClient:
         contentType=None,
         contentLength=None,
         fileFormat=None,
-        model="diffio-2",
+        model="diffio-4.0-flash",
         sampling=None,
         projectParams=None,
         generationParams=None,
@@ -1271,7 +1273,7 @@ class AudioIsolationClient:
         contentType=None,
         contentLength=None,
         fileFormat=None,
-        model="diffio-2",
+        model="diffio-4.0-flash",
         sampling=None,
         projectParams=None,
         generationParams=None,

{diffio-0.1.92 → diffio-0.1.93}/src/diffio/types.py

@@ -451,7 +451,7 @@ class AudioIsolationResult:
         self.generation = generation
 
 
-ModelKey = ("diffio-2", "diffio-2-flash", "diffio-3.4", "diffio-3.5")
+ModelKey = ("diffio-2", "diffio-2-flash", "diffio-3.4", "diffio-3.5", "diffio-4.0-flash", "diffio-4.0-pro")
 DownloadType = ("audio", "mp3", "video", "transcript")
 WebhookMode = ("test", "live")
 WebhookEventType = (

{diffio-0.1.92 → diffio-0.1.93/src/diffio.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffio
-Version: 0.1.92
+Version: 0.1.93
 Summary: Python SDK for the Diffio API.
 Author: Diffio
 License: MIT
@@ -95,7 +95,7 @@ project = client.create_project(
 
 generation = client.create_generation(
     apiProjectId=project.apiProjectId,
-    model="diffio-3.5",
+    model="diffio-4.0-flash",
     sampling={"steps": 12, "guidance": 1.5},
     idempotencyKey="restore-job-2026-001",
     requestOptions={"maxRetries": 2},
@@ -117,7 +117,7 @@ from diffio import DiffioClient
 client = DiffioClient(apiKey="diffio_live_...")
 result = client.audio_isolation.isolate(
     filePath="sample.wav",
-    model="diffio-3.5",
+    model="diffio-4.0-flash",
     sampling={"steps": 12, "guidance": 1.5},
 )
 
@@ -134,7 +134,7 @@ from diffio import DiffioClient
 client = DiffioClient(apiKey="diffio_live_...")
 audio_bytes, info = client.restore_audio(
     filePath="sample.wav",
-    model="diffio-3.5",
+    model="diffio-4.0-flash",
     sampling={"steps": 12, "guidance": 1.5},
     onProgress=lambda progress: print(progress.status),
 )

{diffio-0.1.92 → diffio-0.1.93}/src/diffio.egg-info/SOURCES.txt

@@ -12,5 +12,6 @@ src/diffio.egg-info/dependency_links.txt
 src/diffio.egg-info/requires.txt
 src/diffio.egg-info/top_level.txt
 tests/test_client.py
+tests/test_diffio40_models.py
 tests/test_emulator_e2e.py
 tests/test_transcription.py

{diffio-0.1.92 → diffio-0.1.93}/tests/test_client.py

@@ -248,7 +248,7 @@ def test_audio_isolation_isolate_runs_full_flow(tmp_path: Path):
             assert request.headers.get("Content-Type") == "audio/wav"
             return httpx.Response(200)
 
-        if request.url.path == "/v1/diffio-2.0-generation":
+        if request.url.path in ("/v1/diffio-2.0-generation", "/v1/diffio-4.0-flash-generation"):
             payload = json.loads(request.content.decode("utf-8"))
             assert payload["apiProjectId"] == "proj_abc"
             assert payload["sampling"]["steps"] == 10
@@ -257,7 +257,7 @@ def test_audio_isolation_isolate_runs_full_flow(tmp_path: Path):
                 json={
                     "generationId": "gen_123",
                     "apiProjectId": "proj_abc",
-                    "modelKey": "diffio-2",
+                    "modelKey": "diffio-4.0-flash" if "4.0-flash" in request.url.path else "diffio-2",
                     "status": "queued",
                 },
             )
@@ -326,7 +326,7 @@ def test_create_generation_sends_idempotency_key_and_parses_replay():
             json={
                 "generationId": "gen_original",
                 "apiProjectId": "proj_123",
-                "modelKey": "diffio-2",
+                "modelKey": "diffio-4.0-flash" if "4.0-flash" in request.url.path else "diffio-2",
                 "status": "queued",
                 "idempotentReplay": True,
             },
@@ -341,7 +341,7 @@ def test_create_generation_sends_idempotency_key_and_parses_replay():
         idempotencyKey="restore-job-2026-001",
     )
 
-    assert received["path"] == "/v1/diffio-2.0-generation"
+    assert received["path"] == "/v1/diffio-4.0-flash-generation"
     assert received["payload"] == {
         "apiProjectId": "proj_123",
         "idempotencyKey": "restore-job-2026-001",
@@ -360,7 +360,7 @@ def test_generations_create_forwards_idempotency_key():
             json={
                 "generationId": "gen_123",
                 "apiProjectId": "proj_123",
-                "modelKey": "diffio-2",
+                "modelKey": "diffio-4.0-flash" if "4.0-flash" in request.url.path else "diffio-2",
                 "status": "queued",
             },
         )
@@ -382,14 +382,14 @@ def test_generations_create_and_wait_forwards_idempotency_key():
     received = {}
 
     def handler(request):
-        if request.url.path == "/v1/diffio-2.0-generation":
+        if request.url.path in ("/v1/diffio-2.0-generation", "/v1/diffio-4.0-flash-generation"):
             received["createPayload"] = json.loads(request.content.decode("utf-8"))
             return httpx.Response(
                 200,
                 json={
                     "generationId": "gen_123",
                     "apiProjectId": "proj_123",
-                    "modelKey": "diffio-2",
+                    "modelKey": "diffio-4.0-flash" if "4.0-flash" in request.url.path else "diffio-2",
                     "status": "queued",
                     "idempotentReplay": False,
                 },
@@ -474,7 +474,7 @@ def test_restore_audio_runs_full_flow_and_downloads(tmp_path, monkeypatch, trans
             assert request.headers.get("Content-Type") == "audio/wav"
             return httpx.Response(200)
 
-        if request.url.path == "/v1/diffio-2.0-generation":
+        if request.url.path in ("/v1/diffio-2.0-generation", "/v1/diffio-4.0-flash-generation"):
             payload = json.loads(request.content.decode("utf-8"))
             assert payload["apiProjectId"] == "proj_abc"
             assert payload["sampling"]["steps"] == 10
@@ -483,7 +483,7 @@ def test_restore_audio_runs_full_flow_and_downloads(tmp_path, monkeypatch, trans
                 json={
                     "generationId": "gen_123",
                     "apiProjectId": "proj_abc",
-                    "modelKey": "diffio-2",
+                    "modelKey": "diffio-4.0-flash" if "4.0-flash" in request.url.path else "diffio-2",
                     "status": "queued",
                 },
             )
@@ -861,7 +861,7 @@ def test_list_project_generations_payload_and_response():
                     {
                         "generationId": "gen_123",
                         "status": "processing",
-                        "modelKey": "diffio-2",
+                        "modelKey": "diffio-4.0-flash" if "4.0-flash" in request.url.path else "diffio-2",
                         "progress": None,
                         "createdAt": "2026-01-05T12:40:00Z",
                         "updatedAt": "2026-01-05T12:41:00Z",

diffio-0.1.93/tests/test_diffio40_models.py

@@ -0,0 +1,48 @@
+import json
+
+import httpx
+import pytest
+
+from diffio import DiffioClient
+
+
+@pytest.mark.parametrize(
+    ("model", "path"),
+    [
+        ("diffio-4.0-flash", "/v1/diffio-4.0-flash-generation"),
+        ("diffio-4.0-pro", "/v1/diffio-4.0-pro-generation"),
+    ],
+)
+def test_create_generation_routes_each_diffio_4_0_model_to_its_endpoint(model, path):
+    received = {}
+
+    def handler(request: httpx.Request) -> httpx.Response:
+        received["path"] = request.url.path
+        received["payload"] = json.loads(request.content.decode("utf-8"))
+        return httpx.Response(
+            200,
+            json={"generationId": "gen_40", "apiProjectId": "proj_40", "modelKey": model, "status": "queued"},
+        )
+
+    http_client = httpx.Client(base_url="https://api.test", transport=httpx.MockTransport(handler))
+    client = DiffioClient(apiKey="diffio_live_test", baseUrl="https://api.test", httpClient=http_client)
+
+    response = client.create_generation(apiProjectId="proj_40", model=model)
+
+    assert received["path"] == path
+    assert received["payload"]["apiProjectId"] == "proj_40"
+    assert response.modelKey == model
+
+
+def test_creating_a_generation_without_a_model_uses_diffio_40_flash():
+    """The published SDK defaulted to the deprecated diffio-2, so docs examples broke."""
+    import inspect
+
+    from diffio.client import DiffioClient
+
+    for name in ("create_generation", "restore"):
+        method = getattr(DiffioClient, name, None)
+        if method is None:
+            continue
+        default = inspect.signature(method).parameters["model"].default
+        assert default == "diffio-4.0-flash", f"{name} still defaults to {default}"