feat: enable speakeasy retries (#201)

This commit enables the [speakeasy retries](https://www.speakeasy.com/docs/customize-sdks/retries#global-retries) feature for the SDKs.
livepeer · Sep 13, 2024 · caa4bb7 · caa4bb7
1 parent 1b4f93c
commit caa4bb7
Show file tree

Hide file tree

Showing 9 changed files with 145 additions and 60 deletions.
diff --git a/runner/app/routes/audio_to_text.py b/runner/app/routes/audio_to_text.py
@@ -64,7 +64,13 @@ def handle_pipeline_error(e: Exception) -> JSONResponse:
     operation_id="genAudioToText",
     summary="Audio To Text",
     tags=["generate"],
-    openapi_extra={"x-speakeasy-name-override": "audioToText"},
+    openapi_extra={
+        "x-speakeasy-name-override": "audioToText",
+        "x-speakeasy-retries": {
+            "strategy": "backoff",
+            "statusCodes": ["503"],
+        },
+    },
 )
 @router.post(
     "/audio-to-text/",

diff --git a/runner/app/routes/image_to_image.py b/runner/app/routes/image_to_image.py
@@ -44,7 +44,13 @@
     operation_id="genImageToImage",
     summary="Image To Image",
     tags=["generate"],
-    openapi_extra={"x-speakeasy-name-override": "imageToImage"},
+    openapi_extra={
+        "x-speakeasy-name-override": "imageToImage",
+        "x-speakeasy-retries": {
+            "strategy": "backoff",
+            "statusCodes": ["503"],
+        },
+    },
 )
 @router.post(
     "/image-to-image/",

diff --git a/runner/app/routes/image_to_video.py b/runner/app/routes/image_to_video.py
@@ -43,7 +43,13 @@
     operation_id="genImageToVideo",
     summary="Image To Video",
     tags=["generate"],
-    openapi_extra={"x-speakeasy-name-override": "imageToVideo"},
+    openapi_extra={
+        "x-speakeasy-name-override": "imageToVideo",
+        "x-speakeasy-retries": {
+            "strategy": "backoff",
+            "statusCodes": ["503"],
+        },
+    },
 )
 @router.post(
     "/image-to-video/",

diff --git a/runner/app/routes/segment_anything_2.py b/runner/app/routes/segment_anything_2.py
@@ -49,7 +49,13 @@
     operation_id="genSegmentAnything2",
     summary="Segment Anything 2",
     tags=["generate"],
-    openapi_extra={"x-speakeasy-name-override": "segmentAnything2"},
+    openapi_extra={
+        "x-speakeasy-name-override": "segmentAnything2",
+        "x-speakeasy-retries": {
+            "strategy": "backoff",
+            "statusCodes": ["503"],
+        },
+    },
 )
 @router.post(
     "/segment-anything-2/",

diff --git a/runner/app/routes/text_to_image.py b/runner/app/routes/text_to_image.py
@@ -115,7 +115,13 @@ class TextToImageParams(BaseModel):
     operation_id="genTextToImage",
     summary="Text To Image",
     tags=["generate"],
-    openapi_extra={"x-speakeasy-name-override": "textToImage"},
+    openapi_extra={
+        "x-speakeasy-name-override": "textToImage",
+        "x-speakeasy-retries": {
+            "strategy": "backoff",
+            "statusCodes": ["503"],
+        }
+    }
 )
 @router.post(
     "/text-to-image/",

diff --git a/runner/app/routes/upscale.py b/runner/app/routes/upscale.py
@@ -44,7 +44,13 @@
     operation_id="genUpscale",
     summary="Upscale",
     tags=["generate"],
-    openapi_extra={"x-speakeasy-name-override": "upscale"},
+    openapi_extra={
+        "x-speakeasy-name-override": "upscale",
+        "x-speakeasy-retries": {
+            "strategy": "backoff",
+            "statusCodes": ["503"],
+        },
+    },
 )
 @router.post(
     "/upscale/",

diff --git a/runner/gateway.openapi.yaml b/runner/gateway.openapi.yaml
@@ -2,7 +2,7 @@ openapi: 3.1.0
 info:
   title: Livepeer AI Runner
   description: An application to run AI pipelines
-  version: v0.2.0
+  version: v0.3.0
 servers:
 - url: https://dream-gateway.livepeer.cloud
   description: Livepeer Cloud Community Gateway
@@ -57,6 +57,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: textToImage
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
   /image-to-image:
     post:
       tags:
@@ -105,6 +109,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: imageToImage
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
   /image-to-video:
     post:
       tags:
@@ -153,6 +161,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: imageToVideo
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
   /upscale:
     post:
       tags:
@@ -201,6 +213,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: upscale
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
   /audio-to-text:
     post:
       tags:
@@ -255,6 +271,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: audioToText
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
   /segment-anything-2:
     post:
       tags:
@@ -303,6 +323,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: segmentAnything2
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
 components:
   schemas:
     APIError:

diff --git a/runner/openapi.yaml b/runner/openapi.yaml
@@ -2,7 +2,7 @@ openapi: 3.1.0
 info:
   title: Livepeer AI Runner
   description: An application to run AI pipelines
-  version: v0.2.0
+  version: v0.3.0
 servers:
 - url: https://dream-gateway.livepeer.cloud
   description: Livepeer Cloud Community Gateway
@@ -68,6 +68,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: textToImage
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
   /image-to-image:
     post:
       tags:
@@ -116,6 +120,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: imageToImage
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
   /image-to-video:
     post:
       tags:
@@ -164,6 +172,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: imageToVideo
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
   /upscale:
     post:
       tags:
@@ -212,6 +224,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: upscale
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
   /audio-to-text:
     post:
       tags:
@@ -266,6 +282,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: audioToText
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
   /segment-anything-2:
     post:
       tags:
@@ -314,6 +334,10 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: segmentAnything2
+      x-speakeasy-retries:
+        strategy: backoff
+        statusCodes:
+        - '503'
 components:
   schemas:
     APIError:

diff --git a/worker/runner.gen.go b/worker/runner.gen.go