orq-ai-sdk 4.2.16__py3-none-any.whl → 4.3.0rc0__py3-none-any.whl

orq_ai_sdk/edits.py

@@ -7,7 +7,7 @@ from orq_ai_sdk.models import createimageeditop as models_createimageeditop
 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, List, Mapping, Optional, Union
+from typing import Any, Mapping, Optional, Union
 
 
 class Edits(BaseSDK):
@@ -26,37 +26,6 @@ class Edits(BaseSDK):
             models_createimageeditop.CreateImageEditResponseFormat
         ] = None,
         user: Optional[str] = None,
-        name: Optional[str] = None,
-        fallbacks: Optional[
-            Union[
-                List[models_createimageeditop.CreateImageEditFallbacks],
-                List[models_createimageeditop.CreateImageEditFallbacksTypedDict],
-            ]
-        ] = None,
-        retry: Optional[
-            Union[
-                models_createimageeditop.CreateImageEditRetry,
-                models_createimageeditop.CreateImageEditRetryTypedDict,
-            ]
-        ] = None,
-        cache: Optional[
-            Union[
-                models_createimageeditop.CreateImageEditCache,
-                models_createimageeditop.CreateImageEditCacheTypedDict,
-            ]
-        ] = None,
-        load_balancer: Optional[
-            Union[
-                models_createimageeditop.CreateImageEditLoadBalancer,
-                models_createimageeditop.CreateImageEditLoadBalancerTypedDict,
-            ]
-        ] = None,
-        timeout: Optional[
-            Union[
-                models_createimageeditop.CreateImageEditTimeout,
-                models_createimageeditop.CreateImageEditTimeoutTypedDict,
-            ]
-        ] = None,
         orq: Optional[
             Union[
                 models_createimageeditop.CreateImageEditOrq,
@@ -80,12 +49,6 @@ class Edits(BaseSDK):
         :param quality: The quality of the image that will be generated. Auto will automatically select the best quality for the given model.
         :param response_format: The format in which the generated images are returned. Some of the models only return the image in base64 format.
         :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
@@ -114,22 +77,6 @@ class Edits(BaseSDK):
             quality=quality,
             response_format=response_format,
             user=user,
-            name=name,
-            fallbacks=utils.get_pydantic_model(
-                fallbacks, Optional[List[models.CreateImageEditFallbacks]]
-            ),
-            retry=utils.get_pydantic_model(
-                retry, Optional[models.CreateImageEditRetry]
-            ),
-            cache=utils.get_pydantic_model(
-                cache, Optional[models.CreateImageEditCache]
-            ),
-            load_balancer=utils.get_pydantic_model(
-                load_balancer, Optional[models.CreateImageEditLoadBalancer]
-            ),
-            timeout=utils.get_pydantic_model(
-                timeout, Optional[models.CreateImageEditTimeout]
-            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateImageEditOrq]),
         )
 
@@ -202,37 +149,6 @@ class Edits(BaseSDK):
             models_createimageeditop.CreateImageEditResponseFormat
         ] = None,
         user: Optional[str] = None,
-        name: Optional[str] = None,
-        fallbacks: Optional[
-            Union[
-                List[models_createimageeditop.CreateImageEditFallbacks],
-                List[models_createimageeditop.CreateImageEditFallbacksTypedDict],
-            ]
-        ] = None,
-        retry: Optional[
-            Union[
-                models_createimageeditop.CreateImageEditRetry,
-                models_createimageeditop.CreateImageEditRetryTypedDict,
-            ]
-        ] = None,
-        cache: Optional[
-            Union[
-                models_createimageeditop.CreateImageEditCache,
-                models_createimageeditop.CreateImageEditCacheTypedDict,
-            ]
-        ] = None,
-        load_balancer: Optional[
-            Union[
-                models_createimageeditop.CreateImageEditLoadBalancer,
-                models_createimageeditop.CreateImageEditLoadBalancerTypedDict,
-            ]
-        ] = None,
-        timeout: Optional[
-            Union[
-                models_createimageeditop.CreateImageEditTimeout,
-                models_createimageeditop.CreateImageEditTimeoutTypedDict,
-            ]
-        ] = None,
         orq: Optional[
             Union[
                 models_createimageeditop.CreateImageEditOrq,
@@ -256,12 +172,6 @@ class Edits(BaseSDK):
         :param quality: The quality of the image that will be generated. Auto will automatically select the best quality for the given model.
         :param response_format: The format in which the generated images are returned. Some of the models only return the image in base64 format.
         :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
@@ -290,22 +200,6 @@ class Edits(BaseSDK):
             quality=quality,
             response_format=response_format,
             user=user,
-            name=name,
-            fallbacks=utils.get_pydantic_model(
-                fallbacks, Optional[List[models.CreateImageEditFallbacks]]
-            ),
-            retry=utils.get_pydantic_model(
-                retry, Optional[models.CreateImageEditRetry]
-            ),
-            cache=utils.get_pydantic_model(
-                cache, Optional[models.CreateImageEditCache]
-            ),
-            load_balancer=utils.get_pydantic_model(
-                load_balancer, Optional[models.CreateImageEditLoadBalancer]
-            ),
-            timeout=utils.get_pydantic_model(
-                timeout, Optional[models.CreateImageEditTimeout]
-            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateImageEditOrq]),
         )
 

orq_ai_sdk/embeddings.py

@@ -7,7 +7,7 @@ from orq_ai_sdk.models import createembeddingop as models_createembeddingop
 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 List, Mapping, Optional, Union
+from typing import Mapping, Optional, Union
 
 
 class Embeddings(BaseSDK):
@@ -22,37 +22,6 @@ class Embeddings(BaseSDK):
         encoding_format: Optional[models_createembeddingop.EncodingFormat] = "float",
         dimensions: Optional[float] = None,
         user: Optional[str] = None,
-        name: Optional[str] = None,
-        fallbacks: Optional[
-            Union[
-                List[models_createembeddingop.CreateEmbeddingFallbacks],
-                List[models_createembeddingop.CreateEmbeddingFallbacksTypedDict],
-            ]
-        ] = None,
-        retry: Optional[
-            Union[
-                models_createembeddingop.CreateEmbeddingRetry,
-                models_createembeddingop.CreateEmbeddingRetryTypedDict,
-            ]
-        ] = None,
-        cache: Optional[
-            Union[
-                models_createembeddingop.CreateEmbeddingCache,
-                models_createembeddingop.CreateEmbeddingCacheTypedDict,
-            ]
-        ] = None,
-        load_balancer: Optional[
-            Union[
-                models_createembeddingop.CreateEmbeddingLoadBalancer,
-                models_createembeddingop.CreateEmbeddingLoadBalancerTypedDict,
-            ]
-        ] = None,
-        timeout: Optional[
-            Union[
-                models_createembeddingop.CreateEmbeddingTimeout,
-                models_createembeddingop.CreateEmbeddingTimeoutTypedDict,
-            ]
-        ] = None,
         orq: Optional[
             Union[
                 models_createembeddingop.CreateEmbeddingOrq,
@@ -73,12 +42,6 @@ class Embeddings(BaseSDK):
         :param encoding_format: Type of the document element
         :param dimensions: The number of dimensions the resulting output embeddings should have.
         :param user: A unique identifier representing your end-user
-        :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
@@ -104,22 +67,6 @@ class Embeddings(BaseSDK):
             encoding_format=encoding_format,
             dimensions=dimensions,
             user=user,
-            name=name,
-            fallbacks=utils.get_pydantic_model(
-                fallbacks, Optional[List[models.CreateEmbeddingFallbacks]]
-            ),
-            retry=utils.get_pydantic_model(
-                retry, Optional[models.CreateEmbeddingRetry]
-            ),
-            cache=utils.get_pydantic_model(
-                cache, Optional[models.CreateEmbeddingCache]
-            ),
-            load_balancer=utils.get_pydantic_model(
-                load_balancer, Optional[models.CreateEmbeddingLoadBalancer]
-            ),
-            timeout=utils.get_pydantic_model(
-                timeout, Optional[models.CreateEmbeddingTimeout]
-            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateEmbeddingOrq]),
         )
 
@@ -188,37 +135,6 @@ class Embeddings(BaseSDK):
         encoding_format: Optional[models_createembeddingop.EncodingFormat] = "float",
         dimensions: Optional[float] = None,
         user: Optional[str] = None,
-        name: Optional[str] = None,
-        fallbacks: Optional[
-            Union[
-                List[models_createembeddingop.CreateEmbeddingFallbacks],
-                List[models_createembeddingop.CreateEmbeddingFallbacksTypedDict],
-            ]
-        ] = None,
-        retry: Optional[
-            Union[
-                models_createembeddingop.CreateEmbeddingRetry,
-                models_createembeddingop.CreateEmbeddingRetryTypedDict,
-            ]
-        ] = None,
-        cache: Optional[
-            Union[
-                models_createembeddingop.CreateEmbeddingCache,
-                models_createembeddingop.CreateEmbeddingCacheTypedDict,
-            ]
-        ] = None,
-        load_balancer: Optional[
-            Union[
-                models_createembeddingop.CreateEmbeddingLoadBalancer,
-                models_createembeddingop.CreateEmbeddingLoadBalancerTypedDict,
-            ]
-        ] = None,
-        timeout: Optional[
-            Union[
-                models_createembeddingop.CreateEmbeddingTimeout,
-                models_createembeddingop.CreateEmbeddingTimeoutTypedDict,
-            ]
-        ] = None,
         orq: Optional[
             Union[
                 models_createembeddingop.CreateEmbeddingOrq,
@@ -239,12 +155,6 @@ class Embeddings(BaseSDK):
         :param encoding_format: Type of the document element
         :param dimensions: The number of dimensions the resulting output embeddings should have.
         :param user: A unique identifier representing your end-user
-        :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
@@ -270,22 +180,6 @@ class Embeddings(BaseSDK):
             encoding_format=encoding_format,
             dimensions=dimensions,
             user=user,
-            name=name,
-            fallbacks=utils.get_pydantic_model(
-                fallbacks, Optional[List[models.CreateEmbeddingFallbacks]]
-            ),
-            retry=utils.get_pydantic_model(
-                retry, Optional[models.CreateEmbeddingRetry]
-            ),
-            cache=utils.get_pydantic_model(
-                cache, Optional[models.CreateEmbeddingCache]
-            ),
-            load_balancer=utils.get_pydantic_model(
-                load_balancer, Optional[models.CreateEmbeddingLoadBalancer]
-            ),
-            timeout=utils.get_pydantic_model(
-                timeout, Optional[models.CreateEmbeddingTimeout]
-            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateEmbeddingOrq]),
         )
 

orq_ai_sdk/generations.py

@@ -7,7 +7,7 @@ from orq_ai_sdk.models import createimageop as models_createimageop
 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 List, Mapping, Optional, Union
+from typing import Mapping, Optional, Union
 
 
 class Generations(BaseSDK):
@@ -27,37 +27,6 @@ class Generations(BaseSDK):
         ] = UNSET,
         size: OptionalNullable[str] = UNSET,
         style: OptionalNullable[models_createimageop.Style] = UNSET,
-        name: Optional[str] = None,
-        fallbacks: Optional[
-            Union[
-                List[models_createimageop.CreateImageFallbacks],
-                List[models_createimageop.CreateImageFallbacksTypedDict],
-            ]
-        ] = None,
-        retry: Optional[
-            Union[
-                models_createimageop.CreateImageRetry,
-                models_createimageop.CreateImageRetryTypedDict,
-            ]
-        ] = None,
-        cache: Optional[
-            Union[
-                models_createimageop.CreateImageCache,
-                models_createimageop.CreateImageCacheTypedDict,
-            ]
-        ] = None,
-        load_balancer: Optional[
-            Union[
-                models_createimageop.CreateImageLoadBalancer,
-                models_createimageop.CreateImageLoadBalancerTypedDict,
-            ]
-        ] = None,
-        timeout: Optional[
-            Union[
-                models_createimageop.CreateImageTimeout,
-                models_createimageop.CreateImageTimeoutTypedDict,
-            ]
-        ] = None,
         orq: Optional[
             Union[
                 models_createimageop.CreateImageOrq,
@@ -84,12 +53,6 @@ class Generations(BaseSDK):
         :param response_format: The format in which generated images are returned. Must be one of `url` or `b64_json`. This parameter isn't supported for `gpt-image-1` which will always return base64-encoded images.
         :param size: The size of the generated images. Must be one of the specified sizes for each model.
         :param style: The style of the generated images. This parameter is only supported for `openai/dall-e-3`. Must be one of `vivid` or `natural`.
-        :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
@@ -121,18 +84,6 @@ class Generations(BaseSDK):
             response_format=response_format,
             size=size,
             style=style,
-            name=name,
-            fallbacks=utils.get_pydantic_model(
-                fallbacks, Optional[List[models.CreateImageFallbacks]]
-            ),
-            retry=utils.get_pydantic_model(retry, Optional[models.CreateImageRetry]),
-            cache=utils.get_pydantic_model(cache, Optional[models.CreateImageCache]),
-            load_balancer=utils.get_pydantic_model(
-                load_balancer, Optional[models.CreateImageLoadBalancer]
-            ),
-            timeout=utils.get_pydantic_model(
-                timeout, Optional[models.CreateImageTimeout]
-            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateImageOrq]),
         )
 
@@ -206,37 +157,6 @@ class Generations(BaseSDK):
         ] = UNSET,
         size: OptionalNullable[str] = UNSET,
         style: OptionalNullable[models_createimageop.Style] = UNSET,
-        name: Optional[str] = None,
-        fallbacks: Optional[
-            Union[
-                List[models_createimageop.CreateImageFallbacks],
-                List[models_createimageop.CreateImageFallbacksTypedDict],
-            ]
-        ] = None,
-        retry: Optional[
-            Union[
-                models_createimageop.CreateImageRetry,
-                models_createimageop.CreateImageRetryTypedDict,
-            ]
-        ] = None,
-        cache: Optional[
-            Union[
-                models_createimageop.CreateImageCache,
-                models_createimageop.CreateImageCacheTypedDict,
-            ]
-        ] = None,
-        load_balancer: Optional[
-            Union[
-                models_createimageop.CreateImageLoadBalancer,
-                models_createimageop.CreateImageLoadBalancerTypedDict,
-            ]
-        ] = None,
-        timeout: Optional[
-            Union[
-                models_createimageop.CreateImageTimeout,
-                models_createimageop.CreateImageTimeoutTypedDict,
-            ]
-        ] = None,
         orq: Optional[
             Union[
                 models_createimageop.CreateImageOrq,
@@ -263,12 +183,6 @@ class Generations(BaseSDK):
         :param response_format: The format in which generated images are returned. Must be one of `url` or `b64_json`. This parameter isn't supported for `gpt-image-1` which will always return base64-encoded images.
         :param size: The size of the generated images. Must be one of the specified sizes for each model.
         :param style: The style of the generated images. This parameter is only supported for `openai/dall-e-3`. Must be one of `vivid` or `natural`.
-        :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
@@ -300,18 +214,6 @@ class Generations(BaseSDK):
             response_format=response_format,
             size=size,
             style=style,
-            name=name,
-            fallbacks=utils.get_pydantic_model(
-                fallbacks, Optional[List[models.CreateImageFallbacks]]
-            ),
-            retry=utils.get_pydantic_model(retry, Optional[models.CreateImageRetry]),
-            cache=utils.get_pydantic_model(cache, Optional[models.CreateImageCache]),
-            load_balancer=utils.get_pydantic_model(
-                load_balancer, Optional[models.CreateImageLoadBalancer]
-            ),
-            timeout=utils.get_pydantic_model(
-                timeout, Optional[models.CreateImageTimeout]
-            ),
             orq=utils.get_pydantic_model(orq, Optional[models.CreateImageOrq]),
         )