fix: preserve Python request options during polling

magic_hour/resources/v1/ai_clothes_changer/client.py

@@ -86,6 +86,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -214,6 +215,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_face_editor/client.py

@@ -99,6 +99,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -251,6 +252,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_gif_generator/client.py

@@ -82,6 +82,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -204,6 +205,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_headshot_generator/client.py

@@ -85,6 +85,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -209,6 +210,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_image_editor/client.py

@@ -138,6 +138,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -456,6 +457,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_image_generator/client.py

@@ -127,6 +127,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -434,6 +435,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_image_upscaler/client.py

@@ -93,6 +93,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -235,6 +236,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_meme_generator/client.py

@@ -78,6 +78,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -195,6 +196,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_qr_code_generator/client.py

@@ -76,6 +76,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -190,6 +191,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_talking_photo/client.py

@@ -105,6 +105,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -269,6 +270,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_video_editor/client.py

@@ -111,6 +111,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -311,6 +312,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_video_translator/client.py

@@ -124,6 +124,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
     def create(
@@ -380,6 +381,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
     async def create(

magic_hour/resources/v1/ai_voice_cloner/client.py

@@ -87,6 +87,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -213,6 +214,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/ai_voice_generator/client.py

@@ -76,6 +76,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -187,6 +188,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/animation/client.py

@@ -122,6 +122,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -315,6 +316,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/audio_projects/client.py

@@ -38,6 +38,7 @@ def check_result(
         wait_for_completion: bool,
         download_outputs: bool,
         download_directory: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> V1AudioProjectsGetResponseWithDownloads:
         """
         Check the result of an audio project with optional waiting and downloading.
@@ -51,12 +52,13 @@ def check_result(
             download_outputs: Whether to download the outputs
             download_directory: The directory to download the outputs to. If not provided,
                 the outputs will be downloaded to the current working directory
+            request_options: Additional options to customize each project-status HTTP request.
 
         Returns:
             V1AudioProjectsGetResponseWithDownloads: The audio project response with optional
                 downloaded file paths included
         """
-        api_response = self.get(id=id)
+        api_response = self.get(id=id, request_options=request_options)
         if not wait_for_completion:
             response = V1AudioProjectsGetResponseWithDownloads(
                 **api_response.model_dump()
@@ -68,7 +70,7 @@ def check_result(
         status = api_response.status
 
         while status not in ["complete", "error", "canceled"]:
-            api_response = self.get(id=id)
+            api_response = self.get(id=id, request_options=request_options)
             status = api_response.status
             time.sleep(poll_interval)
 
@@ -178,6 +180,7 @@ async def check_result(
         wait_for_completion: bool,
         download_outputs: bool,
         download_directory: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> V1AudioProjectsGetResponseWithDownloads:
         """
         Check the result of an audio project with optional waiting and downloading.
@@ -191,12 +194,13 @@ async def check_result(
             download_outputs: Whether to download the outputs
             download_directory: The directory to download the outputs to. If not provided,
                 the outputs will be downloaded to the current working directory
+            request_options: Additional options to customize each project-status HTTP request.
 
         Returns:
             V1AudioProjectsGetResponseWithDownloads: The audio project response with optional
                 downloaded file paths included
         """
-        api_response = await self.get(id=id)
+        api_response = await self.get(id=id, request_options=request_options)
         if not wait_for_completion:
             response = V1AudioProjectsGetResponseWithDownloads(
                 **api_response.model_dump()
@@ -208,7 +212,7 @@ async def check_result(
         status = api_response.status
 
         while status not in ["complete", "error", "canceled"]:
-            api_response = await self.get(id=id)
+            api_response = await self.get(id=id, request_options=request_options)
             status = api_response.status
             time.sleep(poll_interval)
 

magic_hour/resources/v1/audio_to_video/client.py

@@ -112,6 +112,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -308,6 +309,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/auto_subtitle_generator/client.py

@@ -104,6 +104,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -267,6 +268,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/body_swap/client.py

@@ -92,6 +92,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -225,6 +226,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response

magic_hour/resources/v1/character_replace/client.py

@@ -122,6 +122,7 @@ def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response
@@ -338,6 +339,7 @@ async def generate(
             wait_for_completion=wait_for_completion,
             download_outputs=download_outputs,
             download_directory=download_directory,
+            request_options=request_options,
         )
 
         return response