orq-ai-sdk 4.2.0rc49__py3-none-any.whl → 4.2.12__py3-none-any.whl

orq_ai_sdk/models/createimageop.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .publiccontact import PublicContact, PublicContactTypedDict
+from .publicidentity import PublicIdentity, PublicIdentityTypedDict
 from orq_ai_sdk.types import (
     BaseModel,
     Nullable,
@@ -9,9 +10,10 @@ from orq_ai_sdk.types import (
     UNSET,
     UNSET_SENTINEL,
 )
+import pydantic
 from pydantic import model_serializer
-from typing import Any, Dict, List, Literal, Optional
-from typing_extensions import NotRequired, TypedDict, deprecated
+from typing import List, Literal, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 Background = Literal[
@@ -62,6 +64,16 @@ Style = Literal[
 r"""The style of the generated images. This parameter is only supported for `openai/dall-e-3`. Must be one of `vivid` or `natural`."""
 
 
+class CreateImageFallbacksTypedDict(TypedDict):
+    model: str
+    r"""Fallback model identifier"""
+
+
+class CreateImageFallbacks(BaseModel):
+    model: str
+    r"""Fallback model identifier"""
+
+
 class CreateImageRetryTypedDict(TypedDict):
     r"""Retry configuration for the request"""
 
@@ -97,63 +109,28 @@ class CreateImageRetry(BaseModel):
         return m
 
 
-class CreateImageFallbacksTypedDict(TypedDict):
-    model: str
-    r"""Fallback model identifier"""
-
-
-class CreateImageFallbacks(BaseModel):
-    model: str
-    r"""Fallback model identifier"""
-
-
-@deprecated(
-    "warning: ** DEPRECATED ** - This will be removed in a future release, please migrate away from it as soon as possible."
-)
-class CreateImageContactTypedDict(TypedDict):
-    r"""@deprecated Use identity instead. Information about the contact making the request."""
+CreateImageType = Literal["exact_match",]
 
-    id: str
-    r"""Unique identifier for the contact"""
-    display_name: NotRequired[str]
-    r"""Display name of the contact"""
-    email: NotRequired[str]
-    r"""Email address of the contact"""
-    metadata: NotRequired[List[Dict[str, Any]]]
-    r"""A hash of key/value pairs containing any other data about the contact"""
-    logo_url: NotRequired[str]
-    r"""URL to the contact's avatar or logo"""
-    tags: NotRequired[List[str]]
-    r"""A list of tags associated with the contact"""
-
-
-@deprecated(
-    "warning: ** DEPRECATED ** - This will be removed in a future release, please migrate away from it as soon as possible."
-)
-class CreateImageContact(BaseModel):
-    r"""@deprecated Use identity instead. Information about the contact making the request."""
 
-    id: str
-    r"""Unique identifier for the contact"""
+class CreateImageCacheTypedDict(TypedDict):
+    r"""Cache configuration for the request."""
 
-    display_name: Optional[str] = None
-    r"""Display name of the contact"""
+    type: CreateImageType
+    ttl: NotRequired[float]
+    r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
 
-    email: Optional[str] = None
-    r"""Email address of the contact"""
 
-    metadata: Optional[List[Dict[str, Any]]] = None
-    r"""A hash of key/value pairs containing any other data about the contact"""
+class CreateImageCache(BaseModel):
+    r"""Cache configuration for the request."""
 
-    logo_url: Optional[str] = None
-    r"""URL to the contact's avatar or logo"""
+    type: CreateImageType
 
-    tags: Optional[List[str]] = None
-    r"""A list of tags associated with the contact"""
+    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(["display_name", "email", "metadata", "logo_url", "tags"])
+        optional_fields = set(["ttl"])
         serialized = handler(self)
         m = {}
 
@@ -168,28 +145,26 @@ class CreateImageContact(BaseModel):
         return m
 
 
-CreateImageType = Literal["exact_match",]
-
-
-class CreateImageCacheTypedDict(TypedDict):
-    r"""Cache configuration for the request."""
+CreateImageLoadBalancerType = Literal["weight_based",]
 
-    type: CreateImageType
-    ttl: NotRequired[float]
-    r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
 
+class CreateImageLoadBalancerModelsTypedDict(TypedDict):
+    model: str
+    r"""Model identifier for load balancing"""
+    weight: NotRequired[float]
+    r"""Weight assigned to this model for load balancing"""
 
-class CreateImageCache(BaseModel):
-    r"""Cache configuration for the request."""
 
-    type: CreateImageType
+class CreateImageLoadBalancerModels(BaseModel):
+    model: str
+    r"""Model identifier for load balancing"""
 
-    ttl: Optional[float] = 1800
-    r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
+    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(["ttl"])
+        optional_fields = set(["weight"])
         serialized = handler(self)
         m = {}
 
@@ -204,20 +179,25 @@ class CreateImageCache(BaseModel):
         return m
 
 
-CreateImageLoadBalancerType = Literal["weight_based",]
+class CreateImageLoadBalancer1TypedDict(TypedDict):
+    type: CreateImageLoadBalancerType
+    models: List[CreateImageLoadBalancerModelsTypedDict]
 
 
-class CreateImageLoadBalancer1TypedDict(TypedDict):
+class CreateImageLoadBalancer1(BaseModel):
     type: CreateImageLoadBalancerType
+
+    models: List[CreateImageLoadBalancerModels]
+
+
+class CreateImageLoadBalancerRouterImagesGenerationsModelsTypedDict(TypedDict):
     model: str
     r"""Model identifier for load balancing"""
     weight: NotRequired[float]
     r"""Weight assigned to this model for load balancing"""
 
 
-class CreateImageLoadBalancer1(BaseModel):
-    type: CreateImageLoadBalancerType
-
+class CreateImageLoadBalancerRouterImagesGenerationsModels(BaseModel):
     model: str
     r"""Model identifier for load balancing"""
 
@@ -241,10 +221,23 @@ class CreateImageLoadBalancer1(BaseModel):
         return m
 
 
+class CreateImageLoadBalancerRouterImagesGenerations1TypedDict(TypedDict):
+    type: CreateImageLoadBalancerRouterImagesGenerationsType
+    models: List[CreateImageLoadBalancerRouterImagesGenerationsModelsTypedDict]
+
+
+class CreateImageLoadBalancerRouterImagesGenerations1(BaseModel):
+    type: CreateImageLoadBalancerRouterImagesGenerationsType
+
+    models: List[CreateImageLoadBalancerRouterImagesGenerationsModels]
+
+
 CreateImageLoadBalancerTypedDict = CreateImageLoadBalancer1TypedDict
+r"""Load balancer configuration for the request."""
 
 
 CreateImageLoadBalancer = CreateImageLoadBalancer1
+r"""Load balancer configuration for the request."""
 
 
 class CreateImageTimeoutTypedDict(TypedDict):
@@ -261,21 +254,132 @@ class CreateImageTimeout(BaseModel):
     r"""Timeout value in milliseconds"""
 
 
+class CreateImageRouterImagesGenerationsRetryTypedDict(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 CreateImageRouterImagesGenerationsRetry(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 CreateImageRouterImagesGenerationsFallbacksTypedDict(TypedDict):
+    model: str
+    r"""Fallback model identifier"""
+
+
+class CreateImageRouterImagesGenerationsFallbacks(BaseModel):
+    model: str
+    r"""Fallback model identifier"""
+
+
+CreateImageRouterImagesGenerationsType = Literal["exact_match",]
+
+
+class CreateImageRouterImagesGenerationsCacheTypedDict(TypedDict):
+    r"""Cache configuration for the request."""
+
+    type: CreateImageRouterImagesGenerationsType
+    ttl: NotRequired[float]
+    r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
+
+
+class CreateImageRouterImagesGenerationsCache(BaseModel):
+    r"""Cache configuration for the request."""
+
+    type: CreateImageRouterImagesGenerationsType
+
+    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
+
+
+CreateImageLoadBalancerRouterImagesGenerationsType = Literal["weight_based",]
+
+
+CreateImageRouterImagesGenerationsLoadBalancerTypedDict = (
+    CreateImageLoadBalancerRouterImagesGenerations1TypedDict
+)
+r"""Array of models with weights for load balancing requests"""
+
+
+CreateImageRouterImagesGenerationsLoadBalancer = (
+    CreateImageLoadBalancerRouterImagesGenerations1
+)
+r"""Array of models with weights for load balancing requests"""
+
+
+class CreateImageRouterImagesGenerationsTimeoutTypedDict(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 CreateImageRouterImagesGenerationsTimeout(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 CreateImageOrqTypedDict(TypedDict):
     name: NotRequired[str]
     r"""The name to display on the trace. If not specified, the default system name will be used."""
-    retry: NotRequired[CreateImageRetryTypedDict]
+    retry: NotRequired[CreateImageRouterImagesGenerationsRetryTypedDict]
     r"""Retry configuration for the request"""
-    fallbacks: NotRequired[List[CreateImageFallbacksTypedDict]]
+    fallbacks: NotRequired[List[CreateImageRouterImagesGenerationsFallbacksTypedDict]]
     r"""Array of fallback models to use if primary model fails"""
-    identity: NotRequired[PublicContactTypedDict]
+    identity: NotRequired[PublicIdentityTypedDict]
     r"""Information about the identity making the request. If the identity does not exist, it will be created automatically."""
-    contact: NotRequired[CreateImageContactTypedDict]
-    cache: NotRequired[CreateImageCacheTypedDict]
+    contact: NotRequired[PublicContactTypedDict]
+    r"""@deprecated Use identity instead. Information about the contact making the request."""
+    cache: NotRequired[CreateImageRouterImagesGenerationsCacheTypedDict]
     r"""Cache configuration for the request."""
-    load_balancer: NotRequired[List[CreateImageLoadBalancerTypedDict]]
+    load_balancer: NotRequired[CreateImageRouterImagesGenerationsLoadBalancerTypedDict]
     r"""Array of models with weights for load balancing requests"""
-    timeout: NotRequired[CreateImageTimeoutTypedDict]
+    timeout: NotRequired[CreateImageRouterImagesGenerationsTimeoutTypedDict]
     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."""
 
 
@@ -283,24 +387,30 @@ class CreateImageOrq(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[CreateImageRetry] = None
+    retry: Optional[CreateImageRouterImagesGenerationsRetry] = None
     r"""Retry configuration for the request"""
 
-    fallbacks: Optional[List[CreateImageFallbacks]] = None
+    fallbacks: Optional[List[CreateImageRouterImagesGenerationsFallbacks]] = None
     r"""Array of fallback models to use if primary model fails"""
 
-    identity: Optional[PublicContact] = None
+    identity: Optional[PublicIdentity] = None
     r"""Information about the identity making the request. If the identity does not exist, it will be created automatically."""
 
-    contact: Optional[CreateImageContact] = None
+    contact: Annotated[
+        Optional[PublicContact],
+        pydantic.Field(
+            deprecated="warning: ** DEPRECATED ** - This will be removed in a future release, please migrate away from it as soon as possible."
+        ),
+    ] = None
+    r"""@deprecated Use identity instead. Information about the contact making the request."""
 
-    cache: Optional[CreateImageCache] = None
+    cache: Optional[CreateImageRouterImagesGenerationsCache] = None
     r"""Cache configuration for the request."""
 
-    load_balancer: Optional[List[CreateImageLoadBalancer]] = None
+    load_balancer: Optional[CreateImageRouterImagesGenerationsLoadBalancer] = None
     r"""Array of models with weights for load balancing requests"""
 
-    timeout: Optional[CreateImageTimeout] = None
+    timeout: Optional[CreateImageRouterImagesGenerationsTimeout] = 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")
@@ -356,6 +466,18 @@ class CreateImageRequestBodyTypedDict(TypedDict):
     r"""The size of the generated images. Must be one of the specified sizes for each model."""
     style: NotRequired[Nullable[Style]]
     r"""The style of the generated images. This parameter is only supported for `openai/dall-e-3`. Must be one of `vivid` or `natural`."""
+    name: NotRequired[str]
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
+    fallbacks: NotRequired[List[CreateImageFallbacksTypedDict]]
+    r"""Array of fallback models to use if primary model fails"""
+    retry: NotRequired[CreateImageRetryTypedDict]
+    r"""Retry configuration for the request"""
+    cache: NotRequired[CreateImageCacheTypedDict]
+    r"""Cache configuration for the request."""
+    load_balancer: NotRequired[CreateImageLoadBalancerTypedDict]
+    r"""Load balancer configuration for the request."""
+    timeout: NotRequired[CreateImageTimeoutTypedDict]
+    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[CreateImageOrqTypedDict]
 
 
@@ -395,6 +517,24 @@ class CreateImageRequestBody(BaseModel):
     style: OptionalNullable[Style] = UNSET
     r"""The style of the generated images. This parameter is only supported for `openai/dall-e-3`. Must be one of `vivid` or `natural`."""
 
+    name: Optional[str] = None
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
+
+    fallbacks: Optional[List[CreateImageFallbacks]] = None
+    r"""Array of fallback models to use if primary model fails"""
+
+    retry: Optional[CreateImageRetry] = None
+    r"""Retry configuration for the request"""
+
+    cache: Optional[CreateImageCache] = None
+    r"""Cache configuration for the request."""
+
+    load_balancer: Optional[CreateImageLoadBalancer] = None
+    r"""Load balancer configuration for the request."""
+
+    timeout: Optional[CreateImageTimeout] = 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: Optional[CreateImageOrq] = None
 
     @model_serializer(mode="wrap")
@@ -410,6 +550,12 @@ class CreateImageRequestBody(BaseModel):
                 "response_format",
                 "size",
                 "style",
+                "name",
+                "fallbacks",
+                "retry",
+                "cache",
+                "load_balancer",
+                "timeout",
                 "orq",
             ]
         )