orq-ai-sdk 4.2.0rc28__py3-none-any.whl → 4.2.6__py3-none-any.whl

orq_ai_sdk/models/deploymentgetconfigop.py

@@ -10,6 +10,7 @@ from .imagecontentpartschema import (
     ImageContentPartSchema,
     ImageContentPartSchemaTypedDict,
 )
+from .publicidentity import PublicIdentity, PublicIdentityTypedDict
 from .reasoningpartschema import ReasoningPartSchema, ReasoningPartSchemaTypedDict
 from .redactedreasoningpartschema import (
     RedactedReasoningPartSchema,
@@ -98,6 +99,22 @@ class DeploymentGetConfigPrefixMessagesCacheControl(BaseModel):
     Defaults to `5m`. Only supported by `Anthropic` Claude models.
     """
 
+    @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
+
 
 class PrefixMessagesToolMessageTypedDict(TypedDict):
     role: DeploymentGetConfigPrefixMessagesDeploymentsRequestRequestBody5Role
@@ -125,31 +142,26 @@ class PrefixMessagesToolMessage(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["cache_control"]
-        nullable_fields = ["tool_call_id"]
-        null_default_fields = []
-
+        optional_fields = set(["cache_control"])
+        nullable_fields = set(["tool_call_id"])
         serialized = handler(self)
-
         m = {}
 
         for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
-            serialized.pop(k, None)
-
-            optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (
-                self.__pydantic_fields_set__.intersection({n})
-                or k in null_default_fields
-            )  # pylint: disable=no-member
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields or (optional_nullable and is_set)
-            ):
-                m[k] = val
+            is_nullable_and_explicitly_set = (
+                k in nullable_fields
+                and (self.__pydantic_fields_set__.intersection({n}))  # pylint: disable=no-member
+            )
+
+            if val != UNSET_SENTINEL:
+                if (
+                    val is not None
+                    or k not in optional_fields
+                    or is_nullable_and_explicitly_set
+                ):
+                    m[k] = val
 
         return m
 
@@ -228,6 +240,22 @@ class DeploymentGetConfigPrefixMessagesFunction(BaseModel):
     arguments: Optional[str] = None
     r"""The arguments to call the function with, as generated by the model in JSON format. Note that the model does not always generate valid JSON, and may hallucinate parameters not defined by your function schema. Validate the arguments in your code before calling your function."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name", "arguments"])
+        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 PrefixMessagesToolCallsTypedDict(TypedDict):
     id: str
@@ -251,6 +279,22 @@ class PrefixMessagesToolCalls(BaseModel):
     thought_signature: Optional[str] = None
     r"""Encrypted representation of the model internal reasoning state during function calling. Required by Gemini 3 models when continuing a conversation after a tool call."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["thought_signature"])
+        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 PrefixMessagesAssistantMessageTypedDict(TypedDict):
     role: DeploymentGetConfigPrefixMessagesDeploymentsRequestRequestBodyRole
@@ -294,31 +338,26 @@ class PrefixMessagesAssistantMessage(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["content", "refusal", "name", "audio", "tool_calls"]
-        nullable_fields = ["content", "refusal", "audio"]
-        null_default_fields = []
-
+        optional_fields = set(["content", "refusal", "name", "audio", "tool_calls"])
+        nullable_fields = set(["content", "refusal", "audio"])
         serialized = handler(self)
-
         m = {}
 
         for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
-            serialized.pop(k, None)
-
-            optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (
-                self.__pydantic_fields_set__.intersection({n})
-                or k in null_default_fields
-            )  # pylint: disable=no-member
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields or (optional_nullable and is_set)
-            ):
-                m[k] = val
+            is_nullable_and_explicitly_set = (
+                k in nullable_fields
+                and (self.__pydantic_fields_set__.intersection({n}))  # pylint: disable=no-member
+            )
+
+            if val != UNSET_SENTINEL:
+                if (
+                    val is not None
+                    or k not in optional_fields
+                    or is_nullable_and_explicitly_set
+                ):
+                    m[k] = val
 
         return m
 
@@ -374,6 +413,22 @@ class DeploymentGetConfig2CacheControl(BaseModel):
     Defaults to `5m`. Only supported by `Anthropic` Claude models.
     """
 
+    @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
+
 
 class DeploymentGetConfig24TypedDict(TypedDict):
     type: DeploymentGetConfig2Type
@@ -392,6 +447,22 @@ class DeploymentGetConfig24(BaseModel):
 
     cache_control: Optional[DeploymentGetConfig2CacheControl] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["cache_control"])
+        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
+
 
 DeploymentGetConfigContent2TypedDict = TypeAliasType(
     "DeploymentGetConfigContent2TypedDict",
@@ -448,6 +519,22 @@ class PrefixMessagesUserMessage(BaseModel):
     name: Optional[str] = None
     r"""An optional name for the participant. Provides the model information to differentiate between participants of the same role."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name"])
+        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
+
 
 DeploymentGetConfigPrefixMessagesDeploymentsRole = Literal["developer",]
 r"""The role of the messages author, in this case  `developer`."""
@@ -486,6 +573,22 @@ class PrefixMessagesDeveloperMessage(BaseModel):
     name: Optional[str] = None
     r"""An optional name for the participant. Provides the model information to differentiate between participants of the same role."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name"])
+        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
+
 
 DeploymentGetConfigPrefixMessagesRole = Literal["system",]
 r"""The role of the messages author, in this case `system`."""
@@ -527,6 +630,22 @@ class PrefixMessagesSystemMessage(BaseModel):
     name: Optional[str] = None
     r"""An optional name for the participant. Provides the model information to differentiate between participants of the same role."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name"])
+        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
+
 
 DeploymentGetConfigPrefixMessagesTypedDict = TypeAliasType(
     "DeploymentGetConfigPrefixMessagesTypedDict",
@@ -628,6 +747,22 @@ class DeploymentGetConfigMessagesCacheControl(BaseModel):
     Defaults to `5m`. Only supported by `Anthropic` Claude models.
     """
 
+    @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
+
 
 class DeploymentGetConfigMessagesToolMessageTypedDict(TypedDict):
     role: DeploymentGetConfigMessagesDeploymentsRequestRequestBody5Role
@@ -653,31 +788,26 @@ class DeploymentGetConfigMessagesToolMessage(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["cache_control"]
-        nullable_fields = ["tool_call_id"]
-        null_default_fields = []
-
+        optional_fields = set(["cache_control"])
+        nullable_fields = set(["tool_call_id"])
         serialized = handler(self)
-
         m = {}
 
         for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
-            serialized.pop(k, None)
-
-            optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (
-                self.__pydantic_fields_set__.intersection({n})
-                or k in null_default_fields
-            )  # pylint: disable=no-member
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields or (optional_nullable and is_set)
-            ):
-                m[k] = val
+            is_nullable_and_explicitly_set = (
+                k in nullable_fields
+                and (self.__pydantic_fields_set__.intersection({n}))  # pylint: disable=no-member
+            )
+
+            if val != UNSET_SENTINEL:
+                if (
+                    val is not None
+                    or k not in optional_fields
+                    or is_nullable_and_explicitly_set
+                ):
+                    m[k] = val
 
         return m
 
@@ -759,6 +889,22 @@ class DeploymentGetConfigMessagesFunction(BaseModel):
     arguments: Optional[str] = None
     r"""The arguments to call the function with, as generated by the model in JSON format. Note that the model does not always generate valid JSON, and may hallucinate parameters not defined by your function schema. Validate the arguments in your code before calling your function."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name", "arguments"])
+        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 DeploymentGetConfigMessagesToolCallsTypedDict(TypedDict):
     id: str
@@ -782,6 +928,22 @@ class DeploymentGetConfigMessagesToolCalls(BaseModel):
     thought_signature: Optional[str] = None
     r"""Encrypted representation of the model internal reasoning state during function calling. Required by Gemini 3 models when continuing a conversation after a tool call."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["thought_signature"])
+        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 DeploymentGetConfigMessagesAssistantMessageTypedDict(TypedDict):
     role: DeploymentGetConfigMessagesDeploymentsRequestRequestBodyRole
@@ -825,31 +987,26 @@ class DeploymentGetConfigMessagesAssistantMessage(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["content", "refusal", "name", "audio", "tool_calls"]
-        nullable_fields = ["content", "refusal", "audio"]
-        null_default_fields = []
-
+        optional_fields = set(["content", "refusal", "name", "audio", "tool_calls"])
+        nullable_fields = set(["content", "refusal", "audio"])
         serialized = handler(self)
-
         m = {}
 
         for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
-            serialized.pop(k, None)
-
-            optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (
-                self.__pydantic_fields_set__.intersection({n})
-                or k in null_default_fields
-            )  # pylint: disable=no-member
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields or (optional_nullable and is_set)
-            ):
-                m[k] = val
+            is_nullable_and_explicitly_set = (
+                k in nullable_fields
+                and (self.__pydantic_fields_set__.intersection({n}))  # pylint: disable=no-member
+            )
+
+            if val != UNSET_SENTINEL:
+                if (
+                    val is not None
+                    or k not in optional_fields
+                    or is_nullable_and_explicitly_set
+                ):
+                    m[k] = val
 
         return m
 
@@ -905,6 +1062,22 @@ class DeploymentGetConfig2DeploymentsCacheControl(BaseModel):
     Defaults to `5m`. Only supported by `Anthropic` Claude models.
     """
 
+    @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
+
 
 class DeploymentGetConfig2Deployments4TypedDict(TypedDict):
     type: DeploymentGetConfig2DeploymentsRequestType
@@ -923,6 +1096,22 @@ class DeploymentGetConfig2Deployments4(BaseModel):
 
     cache_control: Optional[DeploymentGetConfig2DeploymentsCacheControl] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["cache_control"])
+        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
+
 
 DeploymentGetConfigContentDeploymentsRequestRequestBody2TypedDict = TypeAliasType(
     "DeploymentGetConfigContentDeploymentsRequestRequestBody2TypedDict",
@@ -979,6 +1168,22 @@ class DeploymentGetConfigMessagesUserMessage(BaseModel):
     name: Optional[str] = None
     r"""An optional name for the participant. Provides the model information to differentiate between participants of the same role."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name"])
+        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
+
 
 DeploymentGetConfigMessagesDeploymentsRole = Literal["developer",]
 r"""The role of the messages author, in this case  `developer`."""
@@ -1017,6 +1222,22 @@ class DeploymentGetConfigMessagesDeveloperMessage(BaseModel):
     name: Optional[str] = None
     r"""An optional name for the participant. Provides the model information to differentiate between participants of the same role."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name"])
+        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
+
 
 DeploymentGetConfigMessagesRole = Literal["system",]
 r"""The role of the messages author, in this case `system`."""
@@ -1058,6 +1279,22 @@ class DeploymentGetConfigMessagesSystemMessage(BaseModel):
     name: Optional[str] = None
     r"""An optional name for the participant. Provides the model information to differentiate between participants of the same role."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name"])
+        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
+
 
 DeploymentGetConfigMessagesTypedDict = TypeAliasType(
     "DeploymentGetConfigMessagesTypedDict",
@@ -1106,6 +1343,22 @@ class DeploymentGetConfigMetadata(BaseModel):
     page_number: Optional[float] = None
     r"""The page number the text is from."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["file_name", "file_type", "page_number"])
+        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 DeploymentGetConfigDocumentsTypedDict(TypedDict):
     text: str
@@ -1121,6 +1374,22 @@ class DeploymentGetConfigDocuments(BaseModel):
     metadata: Optional[DeploymentGetConfigMetadata] = None
     r"""Metadata about the document"""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["metadata"])
+        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 DeploymentGetConfigInvokeOptionsTypedDict(TypedDict):
     include_retrievals: NotRequired[bool]
@@ -1141,6 +1410,22 @@ class DeploymentGetConfigInvokeOptions(BaseModel):
     mock_response: Optional[str] = None
     r"""A mock response to use instead of calling the LLM API. This is useful for testing purposes. When provided, the system will return a response object with this content as the completion, without making an actual API call to the LLM provider. This works for both streaming and non-streaming requests. Mock responses will not generate logs, traces or be counted for your plan usage."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["include_retrievals", "include_usage", "mock_response"])
+        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 DeploymentGetConfigThreadTypedDict(TypedDict):
     id: str
@@ -1156,6 +1441,22 @@ class DeploymentGetConfigThread(BaseModel):
     tags: Optional[List[str]] = None
     r"""Optional tags to differentiate or categorize threads"""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["tags"])
+        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 DeploymentGetConfigOrExistsTypedDict(TypedDict):
     r"""Exists"""
@@ -1305,38 +1606,6 @@ class DeploymentGetConfigOrEq(BaseModel):
     eq: DeploymentGetConfigOrDeploymentsEq
 
 
-DeploymentGetConfigKnowledgeFilterDeploymentsOrTypedDict = TypeAliasType(
-    "DeploymentGetConfigKnowledgeFilterDeploymentsOrTypedDict",
-    Union[
-        DeploymentGetConfigOrEqTypedDict,
-        DeploymentGetConfigOrNeTypedDict,
-        DeploymentGetConfigOrGtTypedDict,
-        DeploymentGetConfigOrGteTypedDict,
-        DeploymentGetConfigOrLtTypedDict,
-        DeploymentGetConfigOrLteTypedDict,
-        DeploymentGetConfigOrInTypedDict,
-        DeploymentGetConfigOrNinTypedDict,
-        DeploymentGetConfigOrExistsTypedDict,
-    ],
-)
-
-
-DeploymentGetConfigKnowledgeFilterDeploymentsOr = TypeAliasType(
-    "DeploymentGetConfigKnowledgeFilterDeploymentsOr",
-    Union[
-        DeploymentGetConfigOrEq,
-        DeploymentGetConfigOrNe,
-        DeploymentGetConfigOrGt,
-        DeploymentGetConfigOrGte,
-        DeploymentGetConfigOrLt,
-        DeploymentGetConfigOrLte,
-        DeploymentGetConfigOrIn,
-        DeploymentGetConfigOrNin,
-        DeploymentGetConfigOrExists,
-    ],
-)
-
-
 class DeploymentGetConfigKnowledgeFilterOrTypedDict(TypedDict):
     r"""Or"""
 
@@ -1697,6 +1966,38 @@ class DeploymentGetConfig1Eq(BaseModel):
     eq: DeploymentGetConfig1DeploymentsEq
 
 
+DeploymentGetConfigKnowledgeFilterDeploymentsOrTypedDict = TypeAliasType(
+    "DeploymentGetConfigKnowledgeFilterDeploymentsOrTypedDict",
+    Union[
+        DeploymentGetConfigOrEqTypedDict,
+        DeploymentGetConfigOrNeTypedDict,
+        DeploymentGetConfigOrGtTypedDict,
+        DeploymentGetConfigOrGteTypedDict,
+        DeploymentGetConfigOrLtTypedDict,
+        DeploymentGetConfigOrLteTypedDict,
+        DeploymentGetConfigOrInTypedDict,
+        DeploymentGetConfigOrNinTypedDict,
+        DeploymentGetConfigOrExistsTypedDict,
+    ],
+)
+
+
+DeploymentGetConfigKnowledgeFilterDeploymentsOr = TypeAliasType(
+    "DeploymentGetConfigKnowledgeFilterDeploymentsOr",
+    Union[
+        DeploymentGetConfigOrEq,
+        DeploymentGetConfigOrNe,
+        DeploymentGetConfigOrGt,
+        DeploymentGetConfigOrGte,
+        DeploymentGetConfigOrLt,
+        DeploymentGetConfigOrLte,
+        DeploymentGetConfigOrIn,
+        DeploymentGetConfigOrNin,
+        DeploymentGetConfigOrExists,
+    ],
+)
+
+
 KnowledgeFilter1TypedDict = TypeAliasType(
     "KnowledgeFilter1TypedDict",
     Union[
@@ -1762,6 +2063,8 @@ class DeploymentGetConfigRequestBodyTypedDict(TypedDict):
     r"""A list of messages to include after the `System` message, but before the  `User` and `Assistant` pairs configured in your deployment."""
     messages: NotRequired[List[DeploymentGetConfigMessagesTypedDict]]
     r"""A list of messages to send to the deployment."""
+    identity: NotRequired[PublicIdentityTypedDict]
+    r"""Information about the identity making the request. If the identity does not exist, it will be created automatically."""
     file_ids: NotRequired[List[str]]
     r"""A list of file IDs that are associated with the deployment request."""
     metadata: NotRequired[Dict[str, Any]]
@@ -1792,6 +2095,9 @@ class DeploymentGetConfigRequestBody(BaseModel):
     messages: Optional[List[DeploymentGetConfigMessages]] = None
     r"""A list of messages to send to the deployment."""
 
+    identity: Optional[PublicIdentity] = None
+    r"""Information about the identity making the request. If the identity does not exist, it will be created automatically."""
+
     file_ids: Optional[List[str]] = None
     r"""A list of file IDs that are associated with the deployment request."""
 
@@ -1811,6 +2117,37 @@ class DeploymentGetConfigRequestBody(BaseModel):
     knowledge_filter: Optional[DeploymentGetConfigKnowledgeFilter] = None
     r"""A filter to apply to the knowledge base chunk metadata when using  knowledge bases in the deployment."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(
+            [
+                "inputs",
+                "context",
+                "prefix_messages",
+                "messages",
+                "identity",
+                "file_ids",
+                "metadata",
+                "extra_params",
+                "documents",
+                "invoke_options",
+                "thread",
+                "knowledge_filter",
+            ]
+        )
+        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
+
 
 DeploymentGetConfigType = Literal[
     "chat",
@@ -1820,6 +2157,7 @@ DeploymentGetConfigType = Literal[
     "tts",
     "stt",
     "rerank",
+    "ocr",
     "moderation",
     "vision",
 ]
@@ -1867,6 +2205,22 @@ class DeploymentGetConfig2File(BaseModel):
     filename: Optional[str] = None
     r"""The name of the file, used when passing the file to the model as a string."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["file_data", "uri", "mimeType", "filename"])
+        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 DeploymentGetConfig23TypedDict(TypedDict):
     type: DeploymentGetConfig2DeploymentsResponse200ApplicationJSONType
@@ -1903,6 +2257,22 @@ class DeploymentGetConfig2ImageURL(BaseModel):
     detail: Optional[str] = None
     r"""Specifies the detail level of the image. Currently only supported with OpenAI models"""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["id", "detail"])
+        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 DeploymentGetConfig22TypedDict(TypedDict):
     r"""The image part of the prompt message. Only supported with vision models."""
@@ -2003,6 +2373,22 @@ class DeploymentGetConfigToolCalls(BaseModel):
 
     index: Optional[float] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["id", "index"])
+        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 DeploymentGetConfigDeploymentsMessagesTypedDict(TypedDict):
     role: DeploymentGetConfigRole
@@ -2026,31 +2412,26 @@ class DeploymentGetConfigDeploymentsMessages(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["tool_calls", "tool_call_id"]
-        nullable_fields = ["content", "tool_call_id"]
-        null_default_fields = []
-
+        optional_fields = set(["tool_calls", "tool_call_id"])
+        nullable_fields = set(["content", "tool_call_id"])
         serialized = handler(self)
-
         m = {}
 
         for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
-            serialized.pop(k, None)
-
-            optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (
-                self.__pydantic_fields_set__.intersection({n})
-                or k in null_default_fields
-            )  # pylint: disable=no-member
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields or (optional_nullable and is_set)
-            ):
-                m[k] = val
+            is_nullable_and_explicitly_set = (
+                k in nullable_fields
+                and (self.__pydantic_fields_set__.intersection({n}))  # pylint: disable=no-member
+            )
+
+            if val != UNSET_SENTINEL:
+                if (
+                    val is not None
+                    or k not in optional_fields
+                    or is_nullable_and_explicitly_set
+                ):
+                    m[k] = val
 
         return m
 
@@ -2064,7 +2445,7 @@ DeploymentGetConfigFormat = Literal[
 r"""Only supported on `image` models."""
 
 
-DeploymentGetConfigResponseFormat6 = Literal[
+ResponseFormat6 = Literal[
     "json",
     "text",
     "srt",
@@ -2073,13 +2454,13 @@ DeploymentGetConfigResponseFormat6 = Literal[
 ]
 
 
-DeploymentGetConfigResponseFormat5 = Literal[
+ResponseFormat5 = Literal[
     "url",
     "base64_json",
 ]
 
 
-DeploymentGetConfigResponseFormat4 = Literal[
+ResponseFormat4 = Literal[
     "mp3",
     "opus",
     "aac",
@@ -2092,22 +2473,22 @@ DeploymentGetConfigResponseFormat4 = Literal[
 DeploymentGetConfigResponseFormatDeploymentsType = Literal["text",]
 
 
-class DeploymentGetConfigResponseFormat3TypedDict(TypedDict):
+class ResponseFormat3TypedDict(TypedDict):
     type: DeploymentGetConfigResponseFormatDeploymentsType
 
 
-class DeploymentGetConfigResponseFormat3(BaseModel):
+class ResponseFormat3(BaseModel):
     type: DeploymentGetConfigResponseFormatDeploymentsType
 
 
 DeploymentGetConfigResponseFormatType = Literal["json_object",]
 
 
-class DeploymentGetConfigResponseFormat2TypedDict(TypedDict):
+class ResponseFormat2TypedDict(TypedDict):
     type: DeploymentGetConfigResponseFormatType
 
 
-class DeploymentGetConfigResponseFormat2(BaseModel):
+class ResponseFormat2(BaseModel):
     type: DeploymentGetConfigResponseFormatType
 
 
@@ -2130,30 +2511,62 @@ class DeploymentGetConfigResponseFormatJSONSchema(BaseModel):
 
     strict: Optional[bool] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["description", "strict"])
+        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 DeploymentGetConfigResponseFormat1TypedDict(TypedDict):
+
+class ResponseFormat1TypedDict(TypedDict):
     type: DeploymentGetConfigResponseFormatDeploymentsResponseType
     json_schema: DeploymentGetConfigResponseFormatJSONSchemaTypedDict
     display_name: NotRequired[str]
 
 
-class DeploymentGetConfigResponseFormat1(BaseModel):
+class ResponseFormat1(BaseModel):
     type: DeploymentGetConfigResponseFormatDeploymentsResponseType
 
     json_schema: DeploymentGetConfigResponseFormatJSONSchema
 
     display_name: Optional[str] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["display_name"])
+        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
+
 
 DeploymentGetConfigResponseFormatTypedDict = TypeAliasType(
     "DeploymentGetConfigResponseFormatTypedDict",
     Union[
-        DeploymentGetConfigResponseFormat2TypedDict,
-        DeploymentGetConfigResponseFormat3TypedDict,
-        DeploymentGetConfigResponseFormat1TypedDict,
-        DeploymentGetConfigResponseFormat4,
-        DeploymentGetConfigResponseFormat5,
-        DeploymentGetConfigResponseFormat6,
+        ResponseFormat2TypedDict,
+        ResponseFormat3TypedDict,
+        ResponseFormat1TypedDict,
+        ResponseFormat4,
+        ResponseFormat5,
+        ResponseFormat6,
     ],
 )
 r"""An object specifying the format that the model must output.
@@ -2169,12 +2582,12 @@ Important: when using JSON mode, you must also instruct the model to produce JSO
 DeploymentGetConfigResponseFormat = TypeAliasType(
     "DeploymentGetConfigResponseFormat",
     Union[
-        DeploymentGetConfigResponseFormat2,
-        DeploymentGetConfigResponseFormat3,
-        DeploymentGetConfigResponseFormat1,
-        DeploymentGetConfigResponseFormat4,
-        DeploymentGetConfigResponseFormat5,
-        DeploymentGetConfigResponseFormat6,
+        ResponseFormat2,
+        ResponseFormat3,
+        ResponseFormat1,
+        ResponseFormat4,
+        ResponseFormat5,
+        ResponseFormat6,
     ],
 )
 r"""An object specifying the format that the model must output.
@@ -2187,7 +2600,7 @@ Important: when using JSON mode, you must also instruct the model to produce JSO
 """
 
 
-DeploymentGetConfigPhotoRealVersion = Literal[
+PhotoRealVersion = Literal[
     "v1",
     "v2",
 ]
@@ -2212,7 +2625,7 @@ DeploymentGetConfigReasoningEffort = Literal[
 r"""Constrains effort on reasoning for reasoning models. Reducing reasoning effort can result in faster responses and fewer tokens used on reasoning in a response."""
 
 
-DeploymentGetConfigVerbosity = Literal[
+Verbosity = Literal[
     "low",
     "medium",
     "high",
@@ -2263,7 +2676,7 @@ class DeploymentGetConfigParametersTypedDict(TypedDict):
 
     Important: when using JSON mode, you must also instruct the model to produce JSON yourself via a system or user message. Without this, the model may generate an unending stream of whitespace until the generation reaches the token limit, resulting in a long-running and seemingly \"stuck\" request. Also note that the message content may be partially cut off if finish_reason=\"length\", which indicates the generation exceeded max_tokens or the conversation exceeded the max context length.
     """
-    photo_real_version: NotRequired[DeploymentGetConfigPhotoRealVersion]
+    photo_real_version: NotRequired[PhotoRealVersion]
     r"""The version of photoReal to use. Must be v1 or v2. Only available for `leonardoai` provider"""
     encoding_format: NotRequired[DeploymentGetConfigEncodingFormat]
     r"""The format to return the embeddings"""
@@ -2271,7 +2684,7 @@ class DeploymentGetConfigParametersTypedDict(TypedDict):
     r"""Constrains effort on reasoning for reasoning models. Reducing reasoning effort can result in faster responses and fewer tokens used on reasoning in a response."""
     budget_tokens: NotRequired[float]
     r"""Gives the model enhanced reasoning capabilities for complex tasks. A value of 0 disables thinking. The minimum budget tokens for thinking are 1024. The Budget Tokens should never exceed the Max Tokens parameter. Only supported by `Anthropic`"""
-    verbosity: NotRequired[DeploymentGetConfigVerbosity]
+    verbosity: NotRequired[Verbosity]
     r"""Controls the verbosity of the model output."""
     thinking_level: NotRequired[DeploymentGetConfigThinkingLevel]
     r"""The level of thinking to use for the model. Only supported by `Google AI`"""
@@ -2336,8 +2749,7 @@ class DeploymentGetConfigParameters(BaseModel):
     """
 
     photo_real_version: Annotated[
-        Optional[DeploymentGetConfigPhotoRealVersion],
-        pydantic.Field(alias="photoRealVersion"),
+        Optional[PhotoRealVersion], pydantic.Field(alias="photoRealVersion")
     ] = None
     r"""The version of photoReal to use. Must be v1 or v2. Only available for `leonardoai` provider"""
 
@@ -2355,7 +2767,7 @@ class DeploymentGetConfigParameters(BaseModel):
     )
     r"""Gives the model enhanced reasoning capabilities for complex tasks. A value of 0 disables thinking. The minimum budget tokens for thinking are 1024. The Budget Tokens should never exceed the Max Tokens parameter. Only supported by `Anthropic`"""
 
-    verbosity: Optional[DeploymentGetConfigVerbosity] = None
+    verbosity: Optional[Verbosity] = None
     r"""Controls the verbosity of the model output."""
 
     thinking_level: Annotated[
@@ -2366,51 +2778,48 @@ class DeploymentGetConfigParameters(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = [
-            "temperature",
-            "maxTokens",
-            "topK",
-            "topP",
-            "frequencyPenalty",
-            "presencePenalty",
-            "numImages",
-            "seed",
-            "format",
-            "dimensions",
-            "quality",
-            "style",
-            "responseFormat",
-            "photoRealVersion",
-            "encoding_format",
-            "reasoningEffort",
-            "budgetTokens",
-            "verbosity",
-            "thinkingLevel",
-        ]
-        nullable_fields = ["responseFormat"]
-        null_default_fields = []
-
+        optional_fields = set(
+            [
+                "temperature",
+                "maxTokens",
+                "topK",
+                "topP",
+                "frequencyPenalty",
+                "presencePenalty",
+                "numImages",
+                "seed",
+                "format",
+                "dimensions",
+                "quality",
+                "style",
+                "responseFormat",
+                "photoRealVersion",
+                "encoding_format",
+                "reasoningEffort",
+                "budgetTokens",
+                "verbosity",
+                "thinkingLevel",
+            ]
+        )
+        nullable_fields = set(["responseFormat"])
         serialized = handler(self)
-
         m = {}
 
         for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
-            serialized.pop(k, None)
-
-            optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (
-                self.__pydantic_fields_set__.intersection({n})
-                or k in null_default_fields
-            )  # pylint: disable=no-member
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields or (optional_nullable and is_set)
-            ):
-                m[k] = val
+            is_nullable_and_explicitly_set = (
+                k in nullable_fields
+                and (self.__pydantic_fields_set__.intersection({n}))  # pylint: disable=no-member
+            )
+
+            if val != UNSET_SENTINEL:
+                if (
+                    val is not None
+                    or k not in optional_fields
+                    or is_nullable_and_explicitly_set
+                ):
+                    m[k] = val
 
         return m
 
@@ -2444,6 +2853,22 @@ class DeploymentGetConfigFunction(BaseModel):
     Omitting `parameters` defines a function with an empty parameter list.
     """
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["description", "parameters"])
+        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 DeploymentGetConfigToolsTypedDict(TypedDict):
     type: DeploymentGetConfigDeploymentsType
@@ -2503,3 +2928,19 @@ class DeploymentGetConfigResponseBody(BaseModel):
 
     tools: Optional[List[DeploymentGetConfigTools]] = None
     r"""A list of tools the model may call. Currently, only functions are supported as a tool. Use this to provide a list of functions the model may generate JSON inputs for."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["type", "tools"])
+        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