orq-ai-sdk 4.2.8__py3-none-any.whl → 4.2.9__py3-none-any.whl

orq_ai_sdk/speech.py

@@ -6,7 +6,7 @@ from orq_ai_sdk._hooks import HookContext
 from orq_ai_sdk.models import createspeechop as models_createspeechop
 from orq_ai_sdk.types import OptionalNullable, UNSET
 from orq_ai_sdk.utils import get_security_from_env
-from typing import Mapping, Optional, Union
+from typing import List, Mapping, Optional, Union
 
 
 class Speech(BaseSDK):
@@ -20,6 +20,31 @@ class Speech(BaseSDK):
             models_createspeechop.CreateSpeechResponseFormat
         ] = "mp3",
         speed: Optional[float] = 1,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_createspeechop.CreateSpeechFallbacks],
+                List[models_createspeechop.CreateSpeechFallbacksTypedDict],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_createspeechop.CreateSpeechRetry,
+                models_createspeechop.CreateSpeechRetryTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_createspeechop.CreateSpeechLoadBalancer,
+                models_createspeechop.CreateSpeechLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_createspeechop.CreateSpeechTimeout,
+                models_createspeechop.CreateSpeechTimeoutTypedDict,
+            ]
+        ] = None,
         orq: Optional[
             Union[
                 models_createspeechop.CreateSpeechOrq,
@@ -48,6 +73,11 @@ class Speech(BaseSDK):
             `aria`, `roger`, `sarah`, `laura`, `charlie`, `george`, `callum`, `river`, `liam`, `charlotte`, `alice`, `matilda`, `will`, `jessica`, `eric`, `chris`, `brian`, `daniel`, `lily`, `bill`
         :param response_format: The format to audio in. Supported formats are `mp3`, `opus`, `aac`, `flac`, `wav`, and `pcm`. If a format is provided but not supported by the provider, the response will be in the default format. When the provided format is not supported by the provider, the response will be in the default format.
         :param speed: The speed of the generated audio.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
         :param orq:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -73,6 +103,17 @@ class Speech(BaseSDK):
             voice=voice,
             response_format=response_format,
             speed=speed,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateSpeechFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(retry, Optional[models.CreateSpeechRetry]),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateSpeechLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateSpeechTimeout]
+            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateSpeechOrq]),
         )
 
@@ -140,6 +181,31 @@ class Speech(BaseSDK):
             models_createspeechop.CreateSpeechResponseFormat
         ] = "mp3",
         speed: Optional[float] = 1,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_createspeechop.CreateSpeechFallbacks],
+                List[models_createspeechop.CreateSpeechFallbacksTypedDict],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_createspeechop.CreateSpeechRetry,
+                models_createspeechop.CreateSpeechRetryTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_createspeechop.CreateSpeechLoadBalancer,
+                models_createspeechop.CreateSpeechLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_createspeechop.CreateSpeechTimeout,
+                models_createspeechop.CreateSpeechTimeoutTypedDict,
+            ]
+        ] = None,
         orq: Optional[
             Union[
                 models_createspeechop.CreateSpeechOrq,
@@ -168,6 +234,11 @@ class Speech(BaseSDK):
             `aria`, `roger`, `sarah`, `laura`, `charlie`, `george`, `callum`, `river`, `liam`, `charlotte`, `alice`, `matilda`, `will`, `jessica`, `eric`, `chris`, `brian`, `daniel`, `lily`, `bill`
         :param response_format: The format to audio in. Supported formats are `mp3`, `opus`, `aac`, `flac`, `wav`, and `pcm`. If a format is provided but not supported by the provider, the response will be in the default format. When the provided format is not supported by the provider, the response will be in the default format.
         :param speed: The speed of the generated audio.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
         :param orq:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -193,6 +264,17 @@ class Speech(BaseSDK):
             voice=voice,
             response_format=response_format,
             speed=speed,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateSpeechFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(retry, Optional[models.CreateSpeechRetry]),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateSpeechLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateSpeechTimeout]
+            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateSpeechOrq]),
         )
 

orq_ai_sdk/transcriptions.py

@@ -31,6 +31,33 @@ class Transcriptions(BaseSDK):
         timestamp_granularities: Optional[
             List[models_createtranscriptionop.TimestampGranularities]
         ] = None,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_createtranscriptionop.CreateTranscriptionFallbacks],
+                List[
+                    models_createtranscriptionop.CreateTranscriptionFallbacksTypedDict
+                ],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_createtranscriptionop.CreateTranscriptionRetry,
+                models_createtranscriptionop.CreateTranscriptionRetryTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_createtranscriptionop.CreateTranscriptionLoadBalancer,
+                models_createtranscriptionop.CreateTranscriptionLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_createtranscriptionop.CreateTranscriptionTimeout,
+                models_createtranscriptionop.CreateTranscriptionTimeoutTypedDict,
+            ]
+        ] = None,
         orq: Optional[
             Union[
                 models_createtranscriptionop.CreateTranscriptionOrq,
@@ -61,6 +88,11 @@ class Transcriptions(BaseSDK):
         :param temperature: The sampling temperature, between 0 and 1. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. If set to 0, the model will use log probability to automatically increase the temperature until certain thresholds are hit.
         :param language: The language of the input audio. Supplying the input language in ISO-639-1 format will improve accuracy and latency.
         :param timestamp_granularities: The timestamp granularities to populate for this transcription. response_format must be set to verbose_json to use timestamp granularities. Either or both of these options are supported: \"word\" or \"segment\". Note: There is no additional latency for segment timestamps, but generating word timestamps incurs additional latency.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
         :param orq:
         :param file: The audio file object (not file name) to transcribe, in one of these formats: flac, mp3, mp4, mpeg, mpga, m4a, ogg, wav, or webm.
         :param retries: Override the default retry configuration for this method
@@ -93,6 +125,19 @@ class Transcriptions(BaseSDK):
             temperature=temperature,
             language=language,
             timestamp_granularities=timestamp_granularities,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateTranscriptionFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(
+                retry, Optional[models.CreateTranscriptionRetry]
+            ),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateTranscriptionLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateTranscriptionTimeout]
+            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateTranscriptionOrq]),
             file=utils.get_pydantic_model(
                 file, Optional[models.CreateTranscriptionFile]
@@ -188,6 +233,33 @@ class Transcriptions(BaseSDK):
         timestamp_granularities: Optional[
             List[models_createtranscriptionop.TimestampGranularities]
         ] = None,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_createtranscriptionop.CreateTranscriptionFallbacks],
+                List[
+                    models_createtranscriptionop.CreateTranscriptionFallbacksTypedDict
+                ],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_createtranscriptionop.CreateTranscriptionRetry,
+                models_createtranscriptionop.CreateTranscriptionRetryTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_createtranscriptionop.CreateTranscriptionLoadBalancer,
+                models_createtranscriptionop.CreateTranscriptionLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_createtranscriptionop.CreateTranscriptionTimeout,
+                models_createtranscriptionop.CreateTranscriptionTimeoutTypedDict,
+            ]
+        ] = None,
         orq: Optional[
             Union[
                 models_createtranscriptionop.CreateTranscriptionOrq,
@@ -218,6 +290,11 @@ class Transcriptions(BaseSDK):
         :param temperature: The sampling temperature, between 0 and 1. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. If set to 0, the model will use log probability to automatically increase the temperature until certain thresholds are hit.
         :param language: The language of the input audio. Supplying the input language in ISO-639-1 format will improve accuracy and latency.
         :param timestamp_granularities: The timestamp granularities to populate for this transcription. response_format must be set to verbose_json to use timestamp granularities. Either or both of these options are supported: \"word\" or \"segment\". Note: There is no additional latency for segment timestamps, but generating word timestamps incurs additional latency.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
         :param orq:
         :param file: The audio file object (not file name) to transcribe, in one of these formats: flac, mp3, mp4, mpeg, mpga, m4a, ogg, wav, or webm.
         :param retries: Override the default retry configuration for this method
@@ -250,6 +327,19 @@ class Transcriptions(BaseSDK):
             temperature=temperature,
             language=language,
             timestamp_granularities=timestamp_granularities,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateTranscriptionFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(
+                retry, Optional[models.CreateTranscriptionRetry]
+            ),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateTranscriptionLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateTranscriptionTimeout]
+            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateTranscriptionOrq]),
             file=utils.get_pydantic_model(
                 file, Optional[models.CreateTranscriptionFile]

orq_ai_sdk/translations.py

@@ -7,7 +7,7 @@ from orq_ai_sdk.models import createtranslationop as models_createtranslationop
 from orq_ai_sdk.types import OptionalNullable, UNSET
 from orq_ai_sdk.utils import get_security_from_env
 from orq_ai_sdk.utils.unmarshal_json_response import unmarshal_json_response
-from typing import Any, Mapping, Optional, Union
+from typing import Any, List, Mapping, Optional, Union
 
 
 class Translations(BaseSDK):
@@ -27,6 +27,31 @@ class Translations(BaseSDK):
             models_createtranslationop.CreateTranslationTimestampsGranularity
         ] = "word",
         temperature: Optional[float] = None,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_createtranslationop.CreateTranslationFallbacks],
+                List[models_createtranslationop.CreateTranslationFallbacksTypedDict],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_createtranslationop.CreateTranslationRetry,
+                models_createtranslationop.CreateTranslationRetryTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_createtranslationop.CreateTranslationLoadBalancer,
+                models_createtranslationop.CreateTranslationLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_createtranslationop.CreateTranslationTimeout,
+                models_createtranslationop.CreateTranslationTimeoutTypedDict,
+            ]
+        ] = None,
         orq: Optional[
             Union[
                 models_createtranslationop.CreateTranslationOrq,
@@ -55,6 +80,11 @@ class Translations(BaseSDK):
         :param num_speakers: The maximum amount of speakers talking in the uploaded file. Helps with predicting who speaks when, the maximum is 32.
         :param timestamps_granularity: The granularity of the timestamps in the transcription. Word provides word-level timestamps and character provides character-level timestamps per word.
         :param temperature: The sampling temperature, between 0 and 1. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. If set to 0, the model will use log probability to automatically increase the temperature until certain thresholds are hit.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
         :param orq:
         :param file: The audio file object (not file name) to transcribe, in one of these formats: flac, mp3, mp4, mpeg, mpga, m4a, ogg, wav, or webm.
         :param retries: Override the default retry configuration for this method
@@ -85,6 +115,19 @@ class Translations(BaseSDK):
             num_speakers=num_speakers,
             timestamps_granularity=timestamps_granularity,
             temperature=temperature,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateTranslationFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(
+                retry, Optional[models.CreateTranslationRetry]
+            ),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateTranslationLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateTranslationTimeout]
+            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateTranslationOrq]),
             file=utils.get_pydantic_model(file, Optional[models.CreateTranslationFile]),
         )
@@ -170,6 +213,31 @@ class Translations(BaseSDK):
             models_createtranslationop.CreateTranslationTimestampsGranularity
         ] = "word",
         temperature: Optional[float] = None,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_createtranslationop.CreateTranslationFallbacks],
+                List[models_createtranslationop.CreateTranslationFallbacksTypedDict],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_createtranslationop.CreateTranslationRetry,
+                models_createtranslationop.CreateTranslationRetryTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_createtranslationop.CreateTranslationLoadBalancer,
+                models_createtranslationop.CreateTranslationLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_createtranslationop.CreateTranslationTimeout,
+                models_createtranslationop.CreateTranslationTimeoutTypedDict,
+            ]
+        ] = None,
         orq: Optional[
             Union[
                 models_createtranslationop.CreateTranslationOrq,
@@ -198,6 +266,11 @@ class Translations(BaseSDK):
         :param num_speakers: The maximum amount of speakers talking in the uploaded file. Helps with predicting who speaks when, the maximum is 32.
         :param timestamps_granularity: The granularity of the timestamps in the transcription. Word provides word-level timestamps and character provides character-level timestamps per word.
         :param temperature: The sampling temperature, between 0 and 1. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. If set to 0, the model will use log probability to automatically increase the temperature until certain thresholds are hit.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
         :param orq:
         :param file: The audio file object (not file name) to transcribe, in one of these formats: flac, mp3, mp4, mpeg, mpga, m4a, ogg, wav, or webm.
         :param retries: Override the default retry configuration for this method
@@ -228,6 +301,19 @@ class Translations(BaseSDK):
             num_speakers=num_speakers,
             timestamps_granularity=timestamps_granularity,
             temperature=temperature,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateTranslationFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(
+                retry, Optional[models.CreateTranslationRetry]
+            ),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateTranslationLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateTranslationTimeout]
+            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateTranslationOrq]),
             file=utils.get_pydantic_model(file, Optional[models.CreateTranslationFile]),
         )

orq_ai_sdk/variations.py

@@ -7,7 +7,7 @@ from orq_ai_sdk.models import createimagevariationop as models_createimagevariat
 from orq_ai_sdk.types import OptionalNullable, UNSET
 from orq_ai_sdk.utils import get_security_from_env
 from orq_ai_sdk.utils.unmarshal_json_response import unmarshal_json_response
-from typing import Any, Mapping, Optional, Union
+from typing import Any, List, Mapping, Optional, Union
 
 
 class Variations(BaseSDK):
@@ -22,6 +22,39 @@ class Variations(BaseSDK):
         ] = "url",
         size: Optional[models_createimagevariationop.Size] = "1024x1024",
         user: Optional[str] = None,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_createimagevariationop.CreateImageVariationFallbacks],
+                List[
+                    models_createimagevariationop.CreateImageVariationFallbacksTypedDict
+                ],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_createimagevariationop.CreateImageVariationRetry,
+                models_createimagevariationop.CreateImageVariationRetryTypedDict,
+            ]
+        ] = None,
+        cache: Optional[
+            Union[
+                models_createimagevariationop.CreateImageVariationCache,
+                models_createimagevariationop.CreateImageVariationCacheTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_createimagevariationop.CreateImageVariationLoadBalancer,
+                models_createimagevariationop.CreateImageVariationLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_createimagevariationop.CreateImageVariationTimeout,
+                models_createimagevariationop.CreateImageVariationTimeoutTypedDict,
+            ]
+        ] = None,
         orq: Optional[
             Union[
                 models_createimagevariationop.CreateImageVariationOrq,
@@ -43,6 +76,12 @@ class Variations(BaseSDK):
         :param response_format: The format in which the generated images are returned. Must be one of `url` or `b64_json`. URLs are only valid for 60 minutes after the image has been generated.
         :param size: The size of the generated images. Must be one of `256x256`, `512x512`, or `1024x1024`.
         :param user: A unique identifier representing your end-user, which can help to monitor and detect abuse.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param cache: Cache configuration for the request.
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
         :param orq:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -69,6 +108,22 @@ class Variations(BaseSDK):
             response_format=response_format,
             size=size,
             user=user,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateImageVariationFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(
+                retry, Optional[models.CreateImageVariationRetry]
+            ),
+            cache=utils.get_pydantic_model(
+                cache, Optional[models.CreateImageVariationCache]
+            ),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateImageVariationLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateImageVariationTimeout]
+            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateImageVariationOrq]),
         )
 
@@ -143,6 +198,39 @@ class Variations(BaseSDK):
         ] = "url",
         size: Optional[models_createimagevariationop.Size] = "1024x1024",
         user: Optional[str] = None,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_createimagevariationop.CreateImageVariationFallbacks],
+                List[
+                    models_createimagevariationop.CreateImageVariationFallbacksTypedDict
+                ],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_createimagevariationop.CreateImageVariationRetry,
+                models_createimagevariationop.CreateImageVariationRetryTypedDict,
+            ]
+        ] = None,
+        cache: Optional[
+            Union[
+                models_createimagevariationop.CreateImageVariationCache,
+                models_createimagevariationop.CreateImageVariationCacheTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_createimagevariationop.CreateImageVariationLoadBalancer,
+                models_createimagevariationop.CreateImageVariationLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_createimagevariationop.CreateImageVariationTimeout,
+                models_createimagevariationop.CreateImageVariationTimeoutTypedDict,
+            ]
+        ] = None,
         orq: Optional[
             Union[
                 models_createimagevariationop.CreateImageVariationOrq,
@@ -164,6 +252,12 @@ class Variations(BaseSDK):
         :param response_format: The format in which the generated images are returned. Must be one of `url` or `b64_json`. URLs are only valid for 60 minutes after the image has been generated.
         :param size: The size of the generated images. Must be one of `256x256`, `512x512`, or `1024x1024`.
         :param user: A unique identifier representing your end-user, which can help to monitor and detect abuse.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param cache: Cache configuration for the request.
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
         :param orq:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -190,6 +284,22 @@ class Variations(BaseSDK):
             response_format=response_format,
             size=size,
             user=user,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateImageVariationFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(
+                retry, Optional[models.CreateImageVariationRetry]
+            ),
+            cache=utils.get_pydantic_model(
+                cache, Optional[models.CreateImageVariationCache]
+            ),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateImageVariationLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateImageVariationTimeout]
+            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateImageVariationOrq]),
         )
 

{orq_ai_sdk-4.2.8.dist-info → orq_ai_sdk-4.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orq-ai-sdk
-Version: 4.2.8
+Version: 4.2.9
 Summary: Python Client SDK for the Orq API.
 Author: Orq
 Project-URL: repository, https://github.com/orq-ai/orq-python.git