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

orq_ai_sdk/models/createimagevariationop.py

@@ -26,6 +26,16 @@ Size = Literal[
 r"""The size of the generated images. Must be one of `256x256`, `512x512`, or `1024x1024`."""
 
 
+class CreateImageVariationFallbacksTypedDict(TypedDict):
+    model: str
+    r"""Fallback model identifier"""
+
+
+class CreateImageVariationFallbacks(BaseModel):
+    model: str
+    r"""Fallback model identifier"""
+
+
 class CreateImageVariationRetryTypedDict(TypedDict):
     r"""Retry configuration for the request"""
 
@@ -61,12 +71,150 @@ class CreateImageVariationRetry(BaseModel):
         return m
 
 
-class CreateImageVariationFallbacksTypedDict(TypedDict):
+CreateImageVariationType = Literal["exact_match",]
+
+
+class CreateImageVariationCacheTypedDict(TypedDict):
+    r"""Cache configuration for the request."""
+
+    type: CreateImageVariationType
+    ttl: NotRequired[float]
+    r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
+
+
+class CreateImageVariationCache(BaseModel):
+    r"""Cache configuration for the request."""
+
+    type: CreateImageVariationType
+
+    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
+
+
+CreateImageVariationLoadBalancerType = Literal["weight_based",]
+
+
+class CreateImageVariationLoadBalancerModelsTypedDict(TypedDict):
+    model: str
+    r"""Model identifier for load balancing"""
+    weight: NotRequired[float]
+    r"""Weight assigned to this model for load balancing"""
+
+
+class CreateImageVariationLoadBalancerModels(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 CreateImageVariationLoadBalancer1TypedDict(TypedDict):
+    type: CreateImageVariationLoadBalancerType
+    models: List[CreateImageVariationLoadBalancerModelsTypedDict]
+
+
+class CreateImageVariationLoadBalancer1(BaseModel):
+    type: CreateImageVariationLoadBalancerType
+
+    models: List[CreateImageVariationLoadBalancerModels]
+
+
+CreateImageVariationLoadBalancerTypedDict = CreateImageVariationLoadBalancer1TypedDict
+r"""Load balancer configuration for the request."""
+
+
+CreateImageVariationLoadBalancer = CreateImageVariationLoadBalancer1
+r"""Load balancer configuration for the request."""
+
+
+class CreateImageVariationTimeoutTypedDict(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 CreateImageVariationTimeout(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 CreateImageVariationRouterImagesVariationsRetryTypedDict(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 CreateImageVariationRouterImagesVariationsRetry(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 CreateImageVariationRouterImagesVariationsFallbacksTypedDict(TypedDict):
     model: str
     r"""Fallback model identifier"""
 
 
-class CreateImageVariationFallbacks(BaseModel):
+class CreateImageVariationRouterImagesVariationsFallbacks(BaseModel):
     model: str
     r"""Fallback model identifier"""
 
@@ -94,21 +242,21 @@ class CreateImageVariationPrompt(BaseModel):
     r"""Version of the prompt to use (currently only \"latest\" supported)"""
 
 
-CreateImageVariationType = Literal["exact_match",]
+CreateImageVariationRouterImagesVariationsType = Literal["exact_match",]
 
 
-class CreateImageVariationCacheTypedDict(TypedDict):
+class CreateImageVariationRouterImagesVariationsCacheTypedDict(TypedDict):
     r"""Cache configuration for the request."""
 
-    type: CreateImageVariationType
+    type: CreateImageVariationRouterImagesVariationsType
     ttl: NotRequired[float]
     r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
 
 
-class CreateImageVariationCache(BaseModel):
+class CreateImageVariationRouterImagesVariationsCache(BaseModel):
     r"""Cache configuration for the request."""
 
-    type: CreateImageVariationType
+    type: CreateImageVariationRouterImagesVariationsType
 
     ttl: Optional[float] = 1800
     r"""Time to live for cached responses in seconds. Maximum 259200 seconds (3 days)."""
@@ -130,17 +278,17 @@ class CreateImageVariationCache(BaseModel):
         return m
 
 
-CreateImageVariationLoadBalancerType = Literal["weight_based",]
+CreateImageVariationLoadBalancerRouterImagesVariationsType = Literal["weight_based",]
 
 
-class CreateImageVariationLoadBalancerModelsTypedDict(TypedDict):
+class CreateImageVariationLoadBalancerRouterImagesVariationsModelsTypedDict(TypedDict):
     model: str
     r"""Model identifier for load balancing"""
     weight: NotRequired[float]
     r"""Weight assigned to this model for load balancing"""
 
 
-class CreateImageVariationLoadBalancerModels(BaseModel):
+class CreateImageVariationLoadBalancerRouterImagesVariationsModels(BaseModel):
     model: str
     r"""Model identifier for load balancing"""
 
@@ -164,33 +312,37 @@ class CreateImageVariationLoadBalancerModels(BaseModel):
         return m
 
 
-class CreateImageVariationLoadBalancer1TypedDict(TypedDict):
-    type: CreateImageVariationLoadBalancerType
-    models: List[CreateImageVariationLoadBalancerModelsTypedDict]
+class CreateImageVariationLoadBalancerRouterImagesVariations1TypedDict(TypedDict):
+    type: CreateImageVariationLoadBalancerRouterImagesVariationsType
+    models: List[CreateImageVariationLoadBalancerRouterImagesVariationsModelsTypedDict]
 
 
-class CreateImageVariationLoadBalancer1(BaseModel):
-    type: CreateImageVariationLoadBalancerType
+class CreateImageVariationLoadBalancerRouterImagesVariations1(BaseModel):
+    type: CreateImageVariationLoadBalancerRouterImagesVariationsType
 
-    models: List[CreateImageVariationLoadBalancerModels]
+    models: List[CreateImageVariationLoadBalancerRouterImagesVariationsModels]
 
 
-CreateImageVariationLoadBalancerTypedDict = CreateImageVariationLoadBalancer1TypedDict
+CreateImageVariationRouterImagesVariationsLoadBalancerTypedDict = (
+    CreateImageVariationLoadBalancerRouterImagesVariations1TypedDict
+)
 r"""Array of models with weights for load balancing requests"""
 
 
-CreateImageVariationLoadBalancer = CreateImageVariationLoadBalancer1
+CreateImageVariationRouterImagesVariationsLoadBalancer = (
+    CreateImageVariationLoadBalancerRouterImagesVariations1
+)
 r"""Array of models with weights for load balancing requests"""
 
 
-class CreateImageVariationTimeoutTypedDict(TypedDict):
+class CreateImageVariationRouterImagesVariationsTimeoutTypedDict(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 CreateImageVariationTimeout(BaseModel):
+class CreateImageVariationRouterImagesVariationsTimeout(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
@@ -200,9 +352,11 @@ class CreateImageVariationTimeout(BaseModel):
 class CreateImageVariationOrqTypedDict(TypedDict):
     name: NotRequired[str]
     r"""The name to display on the trace. If not specified, the default system name will be used."""
-    retry: NotRequired[CreateImageVariationRetryTypedDict]
+    retry: NotRequired[CreateImageVariationRouterImagesVariationsRetryTypedDict]
     r"""Retry configuration for the request"""
-    fallbacks: NotRequired[List[CreateImageVariationFallbacksTypedDict]]
+    fallbacks: NotRequired[
+        List[CreateImageVariationRouterImagesVariationsFallbacksTypedDict]
+    ]
     r"""Array of fallback models to use if primary model fails"""
     prompt: NotRequired[CreateImageVariationPromptTypedDict]
     r"""Prompt configuration for the request"""
@@ -210,11 +364,13 @@ class CreateImageVariationOrqTypedDict(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[CreateImageVariationCacheTypedDict]
+    cache: NotRequired[CreateImageVariationRouterImagesVariationsCacheTypedDict]
     r"""Cache configuration for the request."""
-    load_balancer: NotRequired[CreateImageVariationLoadBalancerTypedDict]
+    load_balancer: NotRequired[
+        CreateImageVariationRouterImagesVariationsLoadBalancerTypedDict
+    ]
     r"""Array of models with weights for load balancing requests"""
-    timeout: NotRequired[CreateImageVariationTimeoutTypedDict]
+    timeout: NotRequired[CreateImageVariationRouterImagesVariationsTimeoutTypedDict]
     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."""
 
 
@@ -222,10 +378,12 @@ class CreateImageVariationOrq(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[CreateImageVariationRetry] = None
+    retry: Optional[CreateImageVariationRouterImagesVariationsRetry] = None
     r"""Retry configuration for the request"""
 
-    fallbacks: Optional[List[CreateImageVariationFallbacks]] = None
+    fallbacks: Optional[List[CreateImageVariationRouterImagesVariationsFallbacks]] = (
+        None
+    )
     r"""Array of fallback models to use if primary model fails"""
 
     prompt: Optional[CreateImageVariationPrompt] = None
@@ -242,13 +400,15 @@ class CreateImageVariationOrq(BaseModel):
     ] = None
     r"""@deprecated Use identity instead. Information about the contact making the request."""
 
-    cache: Optional[CreateImageVariationCache] = None
+    cache: Optional[CreateImageVariationRouterImagesVariationsCache] = None
     r"""Cache configuration for the request."""
 
-    load_balancer: Optional[CreateImageVariationLoadBalancer] = None
+    load_balancer: Optional[CreateImageVariationRouterImagesVariationsLoadBalancer] = (
+        None
+    )
     r"""Array of models with weights for load balancing requests"""
 
-    timeout: Optional[CreateImageVariationTimeout] = None
+    timeout: Optional[CreateImageVariationRouterImagesVariationsTimeout] = 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")
@@ -295,6 +455,18 @@ class CreateImageVariationRequestBodyTypedDict(TypedDict):
     r"""The size of the generated images. Must be one of `256x256`, `512x512`, or `1024x1024`."""
     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[CreateImageVariationFallbacksTypedDict]]
+    r"""Array of fallback models to use if primary model fails"""
+    retry: NotRequired[CreateImageVariationRetryTypedDict]
+    r"""Retry configuration for the request"""
+    cache: NotRequired[CreateImageVariationCacheTypedDict]
+    r"""Cache configuration for the request."""
+    load_balancer: NotRequired[CreateImageVariationLoadBalancerTypedDict]
+    r"""Load balancer configuration for the request."""
+    timeout: NotRequired[CreateImageVariationTimeoutTypedDict]
+    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[CreateImageVariationOrqTypedDict]
 
 
@@ -321,6 +493,39 @@ class CreateImageVariationRequestBody(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[CreateImageVariationFallbacks]],
+        FieldMetadata(multipart=MultipartFormMetadata(json=True)),
+    ] = None
+    r"""Array of fallback models to use if primary model fails"""
+
+    retry: Annotated[
+        Optional[CreateImageVariationRetry],
+        FieldMetadata(multipart=MultipartFormMetadata(json=True)),
+    ] = None
+    r"""Retry configuration for the request"""
+
+    cache: Annotated[
+        Optional[CreateImageVariationCache],
+        FieldMetadata(multipart=MultipartFormMetadata(json=True)),
+    ] = None
+    r"""Cache configuration for the request."""
+
+    load_balancer: Annotated[
+        Optional[CreateImageVariationLoadBalancer],
+        FieldMetadata(multipart=MultipartFormMetadata(json=True)),
+    ] = None
+    r"""Load balancer configuration for the request."""
+
+    timeout: Annotated[
+        Optional[CreateImageVariationTimeout],
+        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[CreateImageVariationOrq],
         FieldMetadata(multipart=MultipartFormMetadata(json=True)),
@@ -328,7 +533,22 @@ class CreateImageVariationRequestBody(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = set(["image", "n", "response_format", "size", "user", "orq"])
+        optional_fields = set(
+            [
+                "image",
+                "n",
+                "response_format",
+                "size",
+                "user",
+                "name",
+                "fallbacks",
+                "retry",
+                "cache",
+                "load_balancer",
+                "timeout",
+                "orq",
+            ]
+        )
         nullable_fields = set(["n"])
         serialized = handler(self)
         m = {}

orq_ai_sdk/models/createpromptop.py

@@ -1196,6 +1196,8 @@ class PromptInputTypedDict(TypedDict):
     r"""Array of messages that make up the conversation. Each message has a role (system, user, assistant, or tool) and content."""
     model: NotRequired[str]
     r"""Model ID used to generate the response, like `openai/gpt-4o` or `anthropic/claude-3-5-sonnet-20241022`. For private models, use format: `{workspaceKey}@{provider}/{model}`. The full list of models can be found at https://docs.orq.ai/docs/ai-gateway-supported-models. Only chat models are supported."""
+    name: NotRequired[str]
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
     audio: NotRequired[Nullable[CreatePromptAudioTypedDict]]
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
     frequency_penalty: NotRequired[Nullable[float]]
@@ -1271,6 +1273,9 @@ class PromptInput(BaseModel):
     model: Optional[str] = None
     r"""Model ID used to generate the response, like `openai/gpt-4o` or `anthropic/claude-3-5-sonnet-20241022`. For private models, use format: `{workspaceKey}@{provider}/{model}`. The full list of models can be found at https://docs.orq.ai/docs/ai-gateway-supported-models. Only chat models are supported."""
 
+    name: Optional[str] = None
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
+
     audio: OptionalNullable[CreatePromptAudio] = UNSET
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
 
@@ -1367,6 +1372,7 @@ class PromptInput(BaseModel):
         optional_fields = set(
             [
                 "model",
+                "name",
                 "audio",
                 "frequency_penalty",
                 "max_tokens",
@@ -3405,6 +3411,8 @@ CreatePromptPromptsResponseMessages = Annotated[
 class PromptFieldTypedDict(TypedDict):
     r"""Prompt configuration with model and messages. Use this instead of prompt_config."""
 
+    name: NotRequired[str]
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
     audio: NotRequired[Nullable[CreatePromptPromptsAudioTypedDict]]
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
     frequency_penalty: NotRequired[Nullable[float]]
@@ -3479,6 +3487,9 @@ class PromptFieldTypedDict(TypedDict):
 class PromptField(BaseModel):
     r"""Prompt configuration with model and messages. Use this instead of prompt_config."""
 
+    name: Optional[str] = None
+    r"""The name to display on the trace. If not specified, the default system name will be used."""
+
     audio: OptionalNullable[CreatePromptPromptsAudio] = UNSET
     r"""Parameters for audio output. Required when audio output is requested with modalities: [\"audio\"]. Learn more."""
 
@@ -3582,6 +3593,7 @@ class PromptField(BaseModel):
     def serialize_model(self, handler):
         optional_fields = set(
             [
+                "name",
                 "audio",
                 "frequency_penalty",
                 "max_tokens",