genblaze-replicate 0.2.1__tar.gz → 0.2.2__tar.gz

{genblaze_replicate-0.2.1 → genblaze_replicate-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genblaze-replicate
-Version: 0.2.1
+Version: 0.2.2
 Summary: Replicate provider adapter for genblaze
 Project-URL: Homepage, https://github.com/backblaze-labs/genblaze
 Project-URL: Documentation, https://github.com/backblaze-labs/genblaze

{genblaze_replicate-0.2.1 → genblaze_replicate-0.2.2}/genblaze_replicate/provider.py

@@ -29,6 +29,7 @@ from genblaze_core.providers import (
     route_by_media_type,
     validate_asset_url,
 )
+from genblaze_core.providers.retry import retry_after_from_response
 from genblaze_core.runnable.config import RunnableConfig
 
 from ._errors import map_replicate_error
@@ -123,6 +124,7 @@ class ReplicateProvider(BaseProvider):
             raise ProviderError(
                 f"Replicate submit failed: {exc}",
                 error_code=map_replicate_error(exc),
+                retry_after=retry_after_from_response(exc),
             ) from exc
 
     def poll(self, prediction_id: Any, config: RunnableConfig | None = None) -> bool:
@@ -137,6 +139,7 @@ class ReplicateProvider(BaseProvider):
             raise ProviderError(
                 f"Replicate poll failed: {exc}",
                 error_code=map_replicate_error(exc),
+                retry_after=retry_after_from_response(exc),
             ) from exc
 
     def fetch_output(self, prediction_id: Any, step: Step) -> Step:
@@ -220,4 +223,5 @@ class ReplicateProvider(BaseProvider):
             raise ProviderError(
                 f"Replicate fetch_output failed: {exc}",
                 error_code=map_replicate_error(exc),
+                retry_after=retry_after_from_response(exc),
             ) from exc

{genblaze_replicate-0.2.1 → genblaze_replicate-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "genblaze-replicate"
-version = "0.2.1"
+version = "0.2.2"
 description = "Replicate provider adapter for genblaze"
 authors = [{name = "Jeronimo De Leon", email = "jdeleon@backblaze.com"}]
 readme = "README.md"