vellum-ai 0.7.2__py3-none-any.whl → 0.7.5__py3-none-any.whl

vellum/types/ml_model_exec_config_request.py

@@ -0,0 +1,40 @@
+# 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 .ml_model_feature import MlModelFeature
+from .ml_model_request_config_request import MlModelRequestConfigRequest
+from .ml_model_response_config_request import MlModelResponseConfigRequest
+from .ml_model_tokenizer_config_request import MlModelTokenizerConfigRequest
+
+
+class MlModelExecConfigRequest(pydantic_v1.BaseModel):
+    model_identifier: str
+    base_url: str
+    metadata: typing.Dict[str, typing.Any]
+    features: typing.List[MlModelFeature]
+    force_system_credentials: typing.Optional[bool] = None
+    tokenizer_config: typing.Optional[MlModelTokenizerConfigRequest] = None
+    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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_family.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+MlModelFamily = typing.Union[
+    typing.Literal[
+        "CAPYBARA",
+        "CHAT_GPT",
+        "CLAUDE",
+        "COHERE",
+        "FALCON",
+        "GEMINI",
+        "GRANITE",
+        "GPT3",
+        "FIREWORKS",
+        "LLAMA2",
+        "LLAMA3",
+        "MISTRAL",
+        "MPT",
+        "OPENCHAT",
+        "PALM",
+        "SOLAR",
+        "TITAN",
+        "WIZARD",
+        "YI",
+        "ZEPHYR",
+    ],
+    typing.Any,
+]

vellum/types/ml_model_family_enum_value_label.py

@@ -0,0 +1,31 @@
+# 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 .ml_model_family import MlModelFamily
+
+
+class MlModelFamilyEnumValueLabel(pydantic_v1.BaseModel):
+    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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_feature.py

@@ -0,0 +1,18 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+MlModelFeature = typing.Union[
+    typing.Literal[
+        "TEXT",
+        "CHAT_MESSAGE_SYSTEM",
+        "CHAT_MESSAGE_USER",
+        "CHAT_MESSAGE_ASSISTANT",
+        "CHAT_MESSAGE_ASSISTANT_UNTERMINATED",
+        "CHAT_MESSAGE_FUNCTION_CALL",
+        "CHAT_MESSAGE_IMAGE",
+        "FUNCTION_DEFINITION",
+        "STREAMING_SUPPORT",
+    ],
+    typing.Any,
+]

vellum/types/ml_model_parameter_config.py

@@ -0,0 +1,42 @@
+# 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 .array_parameter_config import ArrayParameterConfig
+from .integer_parameter_config import IntegerParameterConfig
+from .number_parameter_config import NumberParameterConfig
+from .object_parameter_config import ObjectParameterConfig
+from .parameter_config import ParameterConfig
+
+
+class MlModelParameterConfig(pydantic_v1.BaseModel):
+    temperature: typing.Optional[NumberParameterConfig] = None
+    max_tokens: typing.Optional[IntegerParameterConfig] = None
+    stop: typing.Optional[ArrayParameterConfig] = None
+    top_p: typing.Optional[NumberParameterConfig] = None
+    top_k: typing.Optional[IntegerParameterConfig] = None
+    frequency_penalty: typing.Optional[NumberParameterConfig] = None
+    presence_penalty: typing.Optional[NumberParameterConfig] = None
+    logit_bias: typing.Optional[ObjectParameterConfig] = None
+    custom_parameters: typing.Optional[typing.Dict[str, typing.Optional[ParameterConfig]]] = 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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_parameter_config_request.py

@@ -0,0 +1,42 @@
+# 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 .array_parameter_config_request import ArrayParameterConfigRequest
+from .integer_parameter_config_request import IntegerParameterConfigRequest
+from .number_parameter_config_request import NumberParameterConfigRequest
+from .object_parameter_config_request import ObjectParameterConfigRequest
+from .parameter_config_request import ParameterConfigRequest
+
+
+class MlModelParameterConfigRequest(pydantic_v1.BaseModel):
+    temperature: typing.Optional[NumberParameterConfigRequest] = None
+    max_tokens: typing.Optional[IntegerParameterConfigRequest] = None
+    stop: typing.Optional[ArrayParameterConfigRequest] = None
+    top_p: typing.Optional[NumberParameterConfigRequest] = None
+    top_k: typing.Optional[IntegerParameterConfigRequest] = None
+    frequency_penalty: typing.Optional[NumberParameterConfigRequest] = None
+    presence_penalty: typing.Optional[NumberParameterConfigRequest] = None
+    logit_bias: typing.Optional[ObjectParameterConfigRequest] = None
+    custom_parameters: typing.Optional[typing.Dict[str, typing.Optional[ParameterConfigRequest]]] = 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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_read.py

@@ -0,0 +1,108 @@
+# 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 .hosted_by_enum import HostedByEnum
+from .ml_model_build_config import MlModelBuildConfig
+from .ml_model_developer_enum_value_label import MlModelDeveloperEnumValueLabel
+from .ml_model_display_config_labelled import MlModelDisplayConfigLabelled
+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
+
+
+class MlModelRead(pydantic_v1.BaseModel):
+    """
+    An ML Model that your Workspace has access to.
+    """
+
+    id: str
+    name: str = pydantic_v1.Field()
+    """
+    The unique name of the ML Model.
+    """
+
+    family: MlModelFamilyEnumValueLabel = pydantic_v1.Field()
+    """
+    The family of the ML Model.
+    """
+
+    hosted_by: HostedByEnum = pydantic_v1.Field()
+    """
+    The organization hosting the ML Model.
+    
+    - `ANTHROPIC` - Anthropic
+    - `AWS_BEDROCK` - AWS Bedrock
+    - `AZURE_OPENAI` - Azure OpenAI
+    - `COHERE` - Cohere
+    - `CUSTOM` - Custom
+    - `FIREWORKS_AI` - Fireworks AI
+    - `GOOGLE` - Google
+    - `GOOGLE_VERTEX_AI` - Google Vertex AI
+    - `GROQ` - Groq
+    - `HUGGINGFACE` - HuggingFace
+    - `IBM_WATSONX` - IBM WatsonX
+    - `MOSAICML` - MosaicML
+    - `MYSTIC` - Mystic
+    - `OPENAI` - OpenAI
+    - `OPENPIPE` - OpenPipe
+    - `PYQ` - Pyq
+    - `REPLICATE` - Replicate
+    """
+
+    developed_by: MlModelDeveloperEnumValueLabel = pydantic_v1.Field()
+    """
+    The organization that developed the ML Model.
+    """
+
+    visibility: typing.Optional[VisibilityEnum] = pydantic_v1.Field(default=None)
+    """
+    The visibility of the ML Model.
+    
+    - `DEFAULT` - Default
+    - `PUBLIC` - Public
+    - `PRIVATE` - Private
+    - `DISABLED` - Disabled
+    """
+
+    build_config: MlModelBuildConfig = pydantic_v1.Field()
+    """
+    Configuration for how the ML Model was built.
+    """
+
+    exec_config: MlModelExecConfig = pydantic_v1.Field()
+    """
+    Configuration for how to execute the ML Model.
+    """
+
+    parameter_config: MlModelParameterConfig = pydantic_v1.Field()
+    """
+    Configuration for the ML Model's parameters.
+    """
+
+    display_config: MlModelDisplayConfigLabelled = pydantic_v1.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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_request_authorization_config.py

@@ -0,0 +1,30 @@
+# 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 .ml_model_request_authorization_config_type_enum import MlModelRequestAuthorizationConfigTypeEnum
+
+
+class MlModelRequestAuthorizationConfig(pydantic_v1.BaseModel):
+    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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_request_authorization_config_request.py

@@ -0,0 +1,30 @@
+# 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 .ml_model_request_authorization_config_type_enum import MlModelRequestAuthorizationConfigTypeEnum
+
+
+class MlModelRequestAuthorizationConfigRequest(pydantic_v1.BaseModel):
+    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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_request_authorization_config_type_enum.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+MlModelRequestAuthorizationConfigTypeEnum = typing.Union[typing.Literal["BEARER_TOKEN", "API_KEY"], typing.Any]

vellum/types/ml_model_request_config.py

@@ -0,0 +1,32 @@
+# 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 .ml_model_request_authorization_config import MlModelRequestAuthorizationConfig
+
+
+class MlModelRequestConfig(pydantic_v1.BaseModel):
+    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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_request_config_request.py

@@ -0,0 +1,32 @@
+# 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 .ml_model_request_authorization_config_request import MlModelRequestAuthorizationConfigRequest
+
+
+class MlModelRequestConfigRequest(pydantic_v1.BaseModel):
+    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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_response_config.py

@@ -0,0 +1,31 @@
+# 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
+
+
+class MlModelResponseConfig(pydantic_v1.BaseModel):
+    result_path: typing.Optional[str] = None
+    result_extraction_regex: typing.Optional[str] = None
+    result_substitution_regexes: typing.Optional[typing.Dict[str, 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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_response_config_request.py

@@ -0,0 +1,31 @@
+# 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
+
+
+class MlModelResponseConfigRequest(pydantic_v1.BaseModel):
+    result_path: typing.Optional[str] = None
+    result_extraction_regex: typing.Optional[str] = None
+    result_substitution_regexes: typing.Optional[typing.Dict[str, 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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/ml_model_tokenizer_config.py

@@ -0,0 +1,59 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class MlModelTokenizerConfig_HuggingFace(pydantic_v1.BaseModel):
+    name: str
+    path: typing.Optional[str] = None
+    type: typing.Literal["HUGGING_FACE"] = "HUGGING_FACE"
+
+    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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}
+
+
+class MlModelTokenizerConfig_Tiktoken(pydantic_v1.BaseModel):
+    name: str
+    type: typing.Literal["TIKTOKEN"] = "TIKTOKEN"
+
+    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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}
+
+
+MlModelTokenizerConfig = typing.Union[MlModelTokenizerConfig_HuggingFace, MlModelTokenizerConfig_Tiktoken]

vellum/types/ml_model_tokenizer_config_request.py

@@ -0,0 +1,61 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class MlModelTokenizerConfigRequest_HuggingFace(pydantic_v1.BaseModel):
+    name: str
+    path: typing.Optional[str] = None
+    type: typing.Literal["HUGGING_FACE"] = "HUGGING_FACE"
+
+    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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}
+
+
+class MlModelTokenizerConfigRequest_Tiktoken(pydantic_v1.BaseModel):
+    name: str
+    type: typing.Literal["TIKTOKEN"] = "TIKTOKEN"
+
+    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)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}
+
+
+MlModelTokenizerConfigRequest = typing.Union[
+    MlModelTokenizerConfigRequest_HuggingFace, MlModelTokenizerConfigRequest_Tiktoken
+]

vellum/types/ml_model_usage.py

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import pydantic_v1
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
 
 class MlModelUsage(pydantic_v1.BaseModel):
@@ -19,8 +19,12 @@ class MlModelUsage(pydantic_v1.BaseModel):
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
+        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)
+        )
 
     class Config:
         frozen = True