vellum-ai 0.7.11__py3-none-any.whl → 0.8.4__py3-none-any.whl

vellum/types/ml_model_display_config_request.py

@@ -1,36 +1,26 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .ml_model_display_tag import MlModelDisplayTag
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class MlModelDisplayConfigRequest(pydantic_v1.BaseModel):
+class MlModelDisplayConfigRequest(UniversalBaseModel):
     label: str
     description: str
     tags: typing.List[MlModelDisplayTag]
-    default_display_priority: typing.Optional[float] = pydantic_v1.Field(default=None)
+    default_display_priority: typing.Optional[float] = pydantic.Field(default=None)
     """
     Can only be set when using an internal service token.
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_display_tag_enum_value_label.py

@@ -1,31 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from ..core.pydantic_utilities import UniversalBaseModel
 from .ml_model_display_tag import MlModelDisplayTag
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
 
 
-class MlModelDisplayTagEnumValueLabel(pydantic_v1.BaseModel):
+class MlModelDisplayTagEnumValueLabel(UniversalBaseModel):
     label: str
     value: MlModelDisplayTag
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_exec_config.py

@@ -1,22 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .ml_model_feature import MlModelFeature
+import pydantic
+from .ml_model_tokenizer_config import MlModelTokenizerConfig
 from .ml_model_request_config import MlModelRequestConfig
 from .ml_model_response_config import MlModelResponseConfig
-from .ml_model_tokenizer_config import MlModelTokenizerConfig
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class MlModelExecConfig(pydantic_v1.BaseModel):
+class MlModelExecConfig(UniversalBaseModel):
     model_identifier: str
     base_url: str
-    metadata: typing.Dict[str, typing.Any]
+    metadata: typing.Dict[str, typing.Optional[typing.Any]]
     features: typing.List[MlModelFeature]
-    force_system_credentials: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    force_system_credentials: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Can only be set when using an internal service token.
     """
@@ -25,20 +24,11 @@ class MlModelExecConfig(pydantic_v1.BaseModel):
     request_config: typing.Optional[MlModelRequestConfig] = None
     response_config: typing.Optional[MlModelResponseConfig] = None
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_exec_config_request.py

@@ -1,22 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .ml_model_feature import MlModelFeature
+import pydantic
+from .ml_model_tokenizer_config_request import MlModelTokenizerConfigRequest
 from .ml_model_request_config_request import MlModelRequestConfigRequest
 from .ml_model_response_config_request import MlModelResponseConfigRequest
-from .ml_model_tokenizer_config_request import MlModelTokenizerConfigRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class MlModelExecConfigRequest(pydantic_v1.BaseModel):
+class MlModelExecConfigRequest(UniversalBaseModel):
     model_identifier: str
     base_url: str
-    metadata: typing.Dict[str, typing.Any]
+    metadata: typing.Dict[str, typing.Optional[typing.Any]]
     features: typing.List[MlModelFeature]
-    force_system_credentials: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    force_system_credentials: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Can only be set when using an internal service token.
     """
@@ -25,20 +24,11 @@ class MlModelExecConfigRequest(pydantic_v1.BaseModel):
     request_config: typing.Optional[MlModelRequestConfigRequest] = None
     response_config: typing.Optional[MlModelResponseConfigRequest] = None
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_family_enum_value_label.py

@@ -1,31 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from ..core.pydantic_utilities import UniversalBaseModel
 from .ml_model_family import MlModelFamily
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
 
 
-class MlModelFamilyEnumValueLabel(pydantic_v1.BaseModel):
+class MlModelFamilyEnumValueLabel(UniversalBaseModel):
     label: str
     value: MlModelFamily
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_parameter_config.py

@@ -1,18 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .open_api_array_property import OpenApiArrayProperty
-from .open_api_integer_property import OpenApiIntegerProperty
 from .open_api_number_property import OpenApiNumberProperty
+from .open_api_integer_property import OpenApiIntegerProperty
+from .open_api_array_property import OpenApiArrayProperty
 from .open_api_object_property import OpenApiObjectProperty
 from .open_api_property import OpenApiProperty
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class MlModelParameterConfig(pydantic_v1.BaseModel):
+class MlModelParameterConfig(UniversalBaseModel):
     temperature: typing.Optional[OpenApiNumberProperty] = None
     max_tokens: typing.Optional[OpenApiIntegerProperty] = None
     stop: typing.Optional[OpenApiArrayProperty] = None
@@ -23,20 +22,11 @@ class MlModelParameterConfig(pydantic_v1.BaseModel):
     logit_bias: typing.Optional[OpenApiObjectProperty] = None
     custom_parameters: typing.Optional[typing.Dict[str, typing.Optional[OpenApiProperty]]] = None
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_parameter_config_request.py

@@ -1,18 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .open_api_array_property_request import OpenApiArrayPropertyRequest
-from .open_api_integer_property_request import OpenApiIntegerPropertyRequest
 from .open_api_number_property_request import OpenApiNumberPropertyRequest
+from .open_api_integer_property_request import OpenApiIntegerPropertyRequest
+from .open_api_array_property_request import OpenApiArrayPropertyRequest
 from .open_api_object_property_request import OpenApiObjectPropertyRequest
 from .open_api_property_request import OpenApiPropertyRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class MlModelParameterConfigRequest(pydantic_v1.BaseModel):
+class MlModelParameterConfigRequest(UniversalBaseModel):
     temperature: typing.Optional[OpenApiNumberPropertyRequest] = None
     max_tokens: typing.Optional[OpenApiIntegerPropertyRequest] = None
     stop: typing.Optional[OpenApiArrayPropertyRequest] = None
@@ -23,20 +22,11 @@ class MlModelParameterConfigRequest(pydantic_v1.BaseModel):
     logit_bias: typing.Optional[OpenApiObjectPropertyRequest] = None
     custom_parameters: typing.Optional[typing.Dict[str, typing.Optional[OpenApiPropertyRequest]]] = None
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_read.py

@@ -1,36 +1,35 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+from .ml_model_family_enum_value_label import MlModelFamilyEnumValueLabel
 from .hosted_by_enum import HostedByEnum
 from .ml_model_developer_enum_value_label import MlModelDeveloperEnumValueLabel
-from .ml_model_display_config_labelled import MlModelDisplayConfigLabelled
+import typing
+from .visibility_enum import VisibilityEnum
 from .ml_model_exec_config import MlModelExecConfig
-from .ml_model_family_enum_value_label import MlModelFamilyEnumValueLabel
 from .ml_model_parameter_config import MlModelParameterConfig
-from .visibility_enum import VisibilityEnum
+from .ml_model_display_config_labelled import MlModelDisplayConfigLabelled
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class MlModelRead(pydantic_v1.BaseModel):
+class MlModelRead(UniversalBaseModel):
     """
     An ML Model that your Workspace has access to.
     """
 
     id: str
-    name: str = pydantic_v1.Field()
+    name: str = pydantic.Field()
     """
     The unique name of the ML Model.
     """
 
-    family: MlModelFamilyEnumValueLabel = pydantic_v1.Field()
+    family: MlModelFamilyEnumValueLabel = pydantic.Field()
     """
     The family of the ML Model.
     """
 
-    hosted_by: HostedByEnum = pydantic_v1.Field()
+    hosted_by: HostedByEnum = pydantic.Field()
     """
     The organization hosting the ML Model.
     
@@ -53,12 +52,12 @@ class MlModelRead(pydantic_v1.BaseModel):
     - `REPLICATE` - Replicate
     """
 
-    developed_by: MlModelDeveloperEnumValueLabel = pydantic_v1.Field()
+    developed_by: MlModelDeveloperEnumValueLabel = pydantic.Field()
     """
     The organization that developed the ML Model.
     """
 
-    visibility: typing.Optional[VisibilityEnum] = pydantic_v1.Field(default=None)
+    visibility: typing.Optional[VisibilityEnum] = pydantic.Field(default=None)
     """
     The visibility of the ML Model.
     
@@ -68,35 +67,26 @@ class MlModelRead(pydantic_v1.BaseModel):
     - `DISABLED` - Disabled
     """
 
-    exec_config: MlModelExecConfig = pydantic_v1.Field()
+    exec_config: MlModelExecConfig = pydantic.Field()
     """
     Configuration for how to execute the ML Model.
     """
 
-    parameter_config: MlModelParameterConfig = pydantic_v1.Field()
+    parameter_config: MlModelParameterConfig = pydantic.Field()
     """
     Configuration for the ML Model's parameters.
     """
 
-    display_config: MlModelDisplayConfigLabelled = pydantic_v1.Field()
+    display_config: MlModelDisplayConfigLabelled = pydantic.Field()
     """
     Configuration for how to display the ML Model.
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_request_authorization_config.py

@@ -1,30 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from ..core.pydantic_utilities import UniversalBaseModel
 from .ml_model_request_authorization_config_type_enum import MlModelRequestAuthorizationConfigTypeEnum
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
 
 
-class MlModelRequestAuthorizationConfig(pydantic_v1.BaseModel):
+class MlModelRequestAuthorizationConfig(UniversalBaseModel):
     type: MlModelRequestAuthorizationConfigTypeEnum
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_request_authorization_config_request.py

@@ -1,30 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from ..core.pydantic_utilities import UniversalBaseModel
 from .ml_model_request_authorization_config_type_enum import MlModelRequestAuthorizationConfigTypeEnum
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
 
 
-class MlModelRequestAuthorizationConfigRequest(pydantic_v1.BaseModel):
+class MlModelRequestAuthorizationConfigRequest(UniversalBaseModel):
     type: MlModelRequestAuthorizationConfigTypeEnum
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_request_config.py

@@ -1,32 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .ml_model_request_authorization_config import MlModelRequestAuthorizationConfig
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class MlModelRequestConfig(pydantic_v1.BaseModel):
+class MlModelRequestConfig(UniversalBaseModel):
     headers: typing.Optional[typing.Dict[str, typing.Optional[str]]] = None
     authorization: typing.Optional[MlModelRequestAuthorizationConfig] = None
     body_template: typing.Optional[str] = None
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/ml_model_request_config_request.py

@@ -1,32 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .ml_model_request_authorization_config_request import MlModelRequestAuthorizationConfigRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class MlModelRequestConfigRequest(pydantic_v1.BaseModel):
+class MlModelRequestConfigRequest(UniversalBaseModel):
     headers: typing.Optional[typing.Dict[str, typing.Optional[str]]] = None
     authorization: typing.Optional[MlModelRequestAuthorizationConfigRequest] = None
     body_template: typing.Optional[str] = None
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow