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

orq_ai_sdk/models/createimageeditop.py

@@ -34,16 +34,6 @@ CreateImageEditResponseFormat = Literal[
 r"""The format in which the generated images are returned. Some of the models only return the image in base64 format."""
 
 
-class CreateImageEditFallbacksTypedDict(TypedDict):
-    model: str
-    r"""Fallback model identifier"""
-
-
-class CreateImageEditFallbacks(BaseModel):
-    model: str
-    r"""Fallback model identifier"""
-
-
 class CreateImageEditRetryTypedDict(TypedDict):
     r"""Retry configuration for the request"""
 
@@ -79,150 +69,12 @@ class CreateImageEditRetry(BaseModel):
         return m
 
 
-CreateImageEditType = Literal["exact_match",]
-
-
-class CreateImageEditCacheTypedDict(TypedDict):
-    r"""Cache configuration for the request."""
-
-    type: CreateImageEditType
-    ttl: NotRequired[float]
-    r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
-
-
-class CreateImageEditCache(BaseModel):
-    r"""Cache configuration for the request."""
-
-    type: CreateImageEditType
-
-    ttl: Optional[float] = 1800
-    r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
-
-    @model_serializer(mode="wrap")
-    def serialize_model(self, handler):
-        optional_fields = set(["ttl"])
-        serialized = handler(self)
-        m = {}
-
-        for n, f in type(self).model_fields.items():
-            k = f.alias or n
-            val = serialized.get(k)
-
-            if val != UNSET_SENTINEL:
-                if val is not None or k not in optional_fields:
-                    m[k] = val
-
-        return m
-
-
-CreateImageEditLoadBalancerType = Literal["weight_based",]
-
-
-class CreateImageEditLoadBalancerModelsTypedDict(TypedDict):
-    model: str
-    r"""Model identifier for load balancing"""
-    weight: NotRequired[float]
-    r"""Weight assigned to this model for load balancing"""
-
-
-class CreateImageEditLoadBalancerModels(BaseModel):
-    model: str
-    r"""Model identifier for load balancing"""
-
-    weight: Optional[float] = 0.5
-    r"""Weight assigned to this model for load balancing"""
-
-    @model_serializer(mode="wrap")
-    def serialize_model(self, handler):
-        optional_fields = set(["weight"])
-        serialized = handler(self)
-        m = {}
-
-        for n, f in type(self).model_fields.items():
-            k = f.alias or n
-            val = serialized.get(k)
-
-            if val != UNSET_SENTINEL:
-                if val is not None or k not in optional_fields:
-                    m[k] = val
-
-        return m
-
-
-class CreateImageEditLoadBalancer1TypedDict(TypedDict):
-    type: CreateImageEditLoadBalancerType
-    models: List[CreateImageEditLoadBalancerModelsTypedDict]
-
-
-class CreateImageEditLoadBalancer1(BaseModel):
-    type: CreateImageEditLoadBalancerType
-
-    models: List[CreateImageEditLoadBalancerModels]
-
-
-CreateImageEditLoadBalancerTypedDict = CreateImageEditLoadBalancer1TypedDict
-r"""Load balancer configuration for the request."""
-
-
-CreateImageEditLoadBalancer = CreateImageEditLoadBalancer1
-r"""Load balancer configuration for the request."""
-
-
-class CreateImageEditTimeoutTypedDict(TypedDict):
-    r"""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."""
-
-    call_timeout: float
-    r"""Timeout value in milliseconds"""
-
-
-class CreateImageEditTimeout(BaseModel):
-    r"""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."""
-
-    call_timeout: float
-    r"""Timeout value in milliseconds"""
-
-
-class CreateImageEditRouterImagesEditsRetryTypedDict(TypedDict):
-    r"""Retry configuration for the request"""
-
-    count: NotRequired[float]
-    r"""Number of retry attempts (1-5)"""
-    on_codes: NotRequired[List[float]]
-    r"""HTTP status codes that trigger retry logic"""
-
-
-class CreateImageEditRouterImagesEditsRetry(BaseModel):
-    r"""Retry configuration for the request"""
-
-    count: Optional[float] = 3
-    r"""Number of retry attempts (1-5)"""
-
-    on_codes: Optional[List[float]] = None
-    r"""HTTP status codes that trigger retry logic"""
-
-    @model_serializer(mode="wrap")
-    def serialize_model(self, handler):
-        optional_fields = set(["count", "on_codes"])
-        serialized = handler(self)
-        m = {}
-
-        for n, f in type(self).model_fields.items():
-            k = f.alias or n
-            val = serialized.get(k)
-
-            if val != UNSET_SENTINEL:
-                if val is not None or k not in optional_fields:
-                    m[k] = val
-
-        return m
-
-
-class CreateImageEditRouterImagesEditsFallbacksTypedDict(TypedDict):
+class CreateImageEditFallbacksTypedDict(TypedDict):
     model: str
     r"""Fallback model identifier"""
 
 
-class CreateImageEditRouterImagesEditsFallbacks(BaseModel):
+class CreateImageEditFallbacks(BaseModel):
     model: str
     r"""Fallback model identifier"""
 
@@ -250,21 +102,21 @@ class CreateImageEditPrompt(BaseModel):
     r"""Version of the prompt to use (currently only \"latest\" supported)"""
 
 
-CreateImageEditRouterImagesEditsType = Literal["exact_match",]
+CreateImageEditType = Literal["exact_match",]
 
 
-class CreateImageEditRouterImagesEditsCacheTypedDict(TypedDict):
+class CreateImageEditCacheTypedDict(TypedDict):
     r"""Cache configuration for the request."""
 
-    type: CreateImageEditRouterImagesEditsType
+    type: CreateImageEditType
     ttl: NotRequired[float]
     r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
 
 
-class CreateImageEditRouterImagesEditsCache(BaseModel):
+class CreateImageEditCache(BaseModel):
     r"""Cache configuration for the request."""
 
-    type: CreateImageEditRouterImagesEditsType
+    type: CreateImageEditType
 
     ttl: Optional[float] = 1800
     r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
@@ -286,17 +138,17 @@ class CreateImageEditRouterImagesEditsCache(BaseModel):
         return m
 
 
-CreateImageEditLoadBalancerRouterImagesEditsType = Literal["weight_based",]
+CreateImageEditLoadBalancerType = Literal["weight_based",]
 
 
-class CreateImageEditLoadBalancerRouterImagesEditsModelsTypedDict(TypedDict):
+class CreateImageEditLoadBalancerModelsTypedDict(TypedDict):
     model: str
     r"""Model identifier for load balancing"""
     weight: NotRequired[float]
     r"""Weight assigned to this model for load balancing"""
 
 
-class CreateImageEditLoadBalancerRouterImagesEditsModels(BaseModel):
+class CreateImageEditLoadBalancerModels(BaseModel):
     model: str
     r"""Model identifier for load balancing"""
 
@@ -320,37 +172,33 @@ class CreateImageEditLoadBalancerRouterImagesEditsModels(BaseModel):
         return m
 
 
-class CreateImageEditLoadBalancerRouterImagesEdits1TypedDict(TypedDict):
-    type: CreateImageEditLoadBalancerRouterImagesEditsType
-    models: List[CreateImageEditLoadBalancerRouterImagesEditsModelsTypedDict]
+class CreateImageEditLoadBalancer1TypedDict(TypedDict):
+    type: CreateImageEditLoadBalancerType
+    models: List[CreateImageEditLoadBalancerModelsTypedDict]
 
 
-class CreateImageEditLoadBalancerRouterImagesEdits1(BaseModel):
-    type: CreateImageEditLoadBalancerRouterImagesEditsType
+class CreateImageEditLoadBalancer1(BaseModel):
+    type: CreateImageEditLoadBalancerType
 
-    models: List[CreateImageEditLoadBalancerRouterImagesEditsModels]
+    models: List[CreateImageEditLoadBalancerModels]
 
 
-CreateImageEditRouterImagesEditsLoadBalancerTypedDict = (
-    CreateImageEditLoadBalancerRouterImagesEdits1TypedDict
-)
+CreateImageEditLoadBalancerTypedDict = CreateImageEditLoadBalancer1TypedDict
 r"""Array of models with weights for load balancing requests"""
 
 
-CreateImageEditRouterImagesEditsLoadBalancer = (
-    CreateImageEditLoadBalancerRouterImagesEdits1
-)
+CreateImageEditLoadBalancer = CreateImageEditLoadBalancer1
 r"""Array of models with weights for load balancing requests"""
 
 
-class CreateImageEditRouterImagesEditsTimeoutTypedDict(TypedDict):
+class CreateImageEditTimeoutTypedDict(TypedDict):
     r"""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."""
 
     call_timeout: float
     r"""Timeout value in milliseconds"""
 
 
-class CreateImageEditRouterImagesEditsTimeout(BaseModel):
+class CreateImageEditTimeout(BaseModel):
     r"""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."""
 
     call_timeout: float
@@ -360,9 +208,9 @@ class CreateImageEditRouterImagesEditsTimeout(BaseModel):
 class CreateImageEditOrqTypedDict(TypedDict):
     name: NotRequired[str]
     r"""The name to display on the trace. If not specified, the default system name will be used."""
-    retry: NotRequired[CreateImageEditRouterImagesEditsRetryTypedDict]
+    retry: NotRequired[CreateImageEditRetryTypedDict]
     r"""Retry configuration for the request"""
-    fallbacks: NotRequired[List[CreateImageEditRouterImagesEditsFallbacksTypedDict]]
+    fallbacks: NotRequired[List[CreateImageEditFallbacksTypedDict]]
     r"""Array of fallback models to use if primary model fails"""
     prompt: NotRequired[CreateImageEditPromptTypedDict]
     r"""Prompt configuration for the request"""
@@ -370,11 +218,11 @@ class CreateImageEditOrqTypedDict(TypedDict):
     r"""Information about the identity making the request. If the identity does not exist, it will be created automatically."""
     contact: NotRequired[PublicContactTypedDict]
     r"""@deprecated Use identity instead. Information about the contact making the request."""
-    cache: NotRequired[CreateImageEditRouterImagesEditsCacheTypedDict]
+    cache: NotRequired[CreateImageEditCacheTypedDict]
     r"""Cache configuration for the request."""
-    load_balancer: NotRequired[CreateImageEditRouterImagesEditsLoadBalancerTypedDict]
+    load_balancer: NotRequired[CreateImageEditLoadBalancerTypedDict]
     r"""Array of models with weights for load balancing requests"""
-    timeout: NotRequired[CreateImageEditRouterImagesEditsTimeoutTypedDict]
+    timeout: NotRequired[CreateImageEditTimeoutTypedDict]
     r"""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."""
 
 
@@ -382,10 +230,10 @@ class CreateImageEditOrq(BaseModel):
     name: Optional[str] = None
     r"""The name to display on the trace. If not specified, the default system name will be used."""
 
-    retry: Optional[CreateImageEditRouterImagesEditsRetry] = None
+    retry: Optional[CreateImageEditRetry] = None
     r"""Retry configuration for the request"""
 
-    fallbacks: Optional[List[CreateImageEditRouterImagesEditsFallbacks]] = None
+    fallbacks: Optional[List[CreateImageEditFallbacks]] = None
     r"""Array of fallback models to use if primary model fails"""
 
     prompt: Optional[CreateImageEditPrompt] = None
@@ -402,13 +250,13 @@ class CreateImageEditOrq(BaseModel):
     ] = None
     r"""@deprecated Use identity instead. Information about the contact making the request."""
 
-    cache: Optional[CreateImageEditRouterImagesEditsCache] = None
+    cache: Optional[CreateImageEditCache] = None
     r"""Cache configuration for the request."""
 
-    load_balancer: Optional[CreateImageEditRouterImagesEditsLoadBalancer] = None
+    load_balancer: Optional[CreateImageEditLoadBalancer] = None
     r"""Array of models with weights for load balancing requests"""
 
-    timeout: Optional[CreateImageEditRouterImagesEditsTimeout] = None
+    timeout: Optional[CreateImageEditTimeout] = None
     r"""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."""
 
     @model_serializer(mode="wrap")
@@ -457,18 +305,6 @@ class CreateImageEditRequestBodyTypedDict(TypedDict):
     r"""The format in which the generated images are returned. Some of the models only return the image in base64 format."""
     user: NotRequired[str]
     r"""A unique identifier representing your end-user, which can help to monitor and detect abuse."""
-    name: NotRequired[str]
-    r"""The name to display on the trace. If not specified, the default system name will be used."""
-    fallbacks: NotRequired[List[CreateImageEditFallbacksTypedDict]]
-    r"""Array of fallback models to use if primary model fails"""
-    retry: NotRequired[CreateImageEditRetryTypedDict]
-    r"""Retry configuration for the request"""
-    cache: NotRequired[CreateImageEditCacheTypedDict]
-    r"""Cache configuration for the request."""
-    load_balancer: NotRequired[CreateImageEditLoadBalancerTypedDict]
-    r"""Load balancer configuration for the request."""
-    timeout: NotRequired[CreateImageEditTimeoutTypedDict]
-    r"""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."""
     orq: NotRequired[CreateImageEditOrqTypedDict]
 
 
@@ -501,39 +337,6 @@ class CreateImageEditRequestBody(BaseModel):
     user: Annotated[Optional[str], FieldMetadata(multipart=True)] = None
     r"""A unique identifier representing your end-user, which can help to monitor and detect abuse."""
 
-    name: Annotated[Optional[str], FieldMetadata(multipart=True)] = None
-    r"""The name to display on the trace. If not specified, the default system name will be used."""
-
-    fallbacks: Annotated[
-        Optional[List[CreateImageEditFallbacks]],
-        FieldMetadata(multipart=MultipartFormMetadata(json=True)),
-    ] = None
-    r"""Array of fallback models to use if primary model fails"""
-
-    retry: Annotated[
-        Optional[CreateImageEditRetry],
-        FieldMetadata(multipart=MultipartFormMetadata(json=True)),
-    ] = None
-    r"""Retry configuration for the request"""
-
-    cache: Annotated[
-        Optional[CreateImageEditCache],
-        FieldMetadata(multipart=MultipartFormMetadata(json=True)),
-    ] = None
-    r"""Cache configuration for the request."""
-
-    load_balancer: Annotated[
-        Optional[CreateImageEditLoadBalancer],
-        FieldMetadata(multipart=MultipartFormMetadata(json=True)),
-    ] = None
-    r"""Load balancer configuration for the request."""
-
-    timeout: Annotated[
-        Optional[CreateImageEditTimeout],
-        FieldMetadata(multipart=MultipartFormMetadata(json=True)),
-    ] = None
-    r"""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."""
-
     orq: Annotated[
         Optional[CreateImageEditOrq],
         FieldMetadata(multipart=MultipartFormMetadata(json=True)),
@@ -542,21 +345,7 @@ class CreateImageEditRequestBody(BaseModel):
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = set(
-            [
-                "image",
-                "n",
-                "size",
-                "quality",
-                "response_format",
-                "user",
-                "name",
-                "fallbacks",
-                "retry",
-                "cache",
-                "load_balancer",
-                "timeout",
-                "orq",
-            ]
+            ["image", "n", "size", "quality", "response_format", "user", "orq"]
         )
         nullable_fields = set(["n", "size", "quality"])
         serialized = handler(self)