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

orq_ai_sdk/models/createimagevariationop.py

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

orq_ai_sdk/models/createpromptop.py

@@ -1196,8 +1196,6 @@ 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]]
@@ -1273,9 +1271,6 @@ 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."""
 
@@ -1372,7 +1367,6 @@ class PromptInput(BaseModel):
         optional_fields = set(
             [
                 "model",
-                "name",
                 "audio",
                 "frequency_penalty",
                 "max_tokens",
@@ -3411,8 +3405,6 @@ 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]]
@@ -3487,9 +3479,6 @@ 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."""
 
@@ -3593,7 +3582,6 @@ class PromptField(BaseModel):
     def serialize_model(self, handler):
         optional_fields = set(
             [
-                "name",
                 "audio",
                 "frequency_penalty",
                 "max_tokens",