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

vellum/types/create_test_suite_test_case_request.py

@@ -1,52 +1,42 @@
 # 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
+import pydantic
 from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class CreateTestSuiteTestCaseRequest(pydantic_v1.BaseModel):
+class CreateTestSuiteTestCaseRequest(UniversalBaseModel):
     """
     Information about the Test Case to create
     """
 
-    label: typing.Optional[str] = pydantic_v1.Field(default=None)
+    label: typing.Optional[str] = pydantic.Field(default=None)
     """
     A human-readable label used to convey the intention of this Test Case
     """
 
-    input_values: typing.List[NamedTestCaseVariableValueRequest] = pydantic_v1.Field()
+    input_values: typing.List[NamedTestCaseVariableValueRequest] = pydantic.Field()
     """
     Values for each of the Test Case's input variables
     """
 
-    evaluation_values: typing.List[NamedTestCaseVariableValueRequest] = pydantic_v1.Field()
+    evaluation_values: typing.List[NamedTestCaseVariableValueRequest] = pydantic.Field()
     """
     Values for each of the Test Case's evaluation variables
     """
 
-    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    external_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     Optionally provide an ID that uniquely identifies this Test Case in your system. Useful for updating this Test Cases data after initial creation. Cannot be changed later.
     """
 
-    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/deployment_provider_payload_response.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
+from .deployment_provider_payload_response_payload import DeploymentProviderPayloadResponsePayload
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .compile_prompt_meta import CompilePromptMeta
-from .deployment_provider_payload_response_payload import DeploymentProviderPayloadResponsePayload
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class DeploymentProviderPayloadResponse(pydantic_v1.BaseModel):
+class DeploymentProviderPayloadResponse(UniversalBaseModel):
     payload: DeploymentProviderPayloadResponsePayload
     meta: typing.Optional[CompilePromptMeta] = 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/deployment_provider_payload_response_payload.py

@@ -2,4 +2,4 @@
 
 import typing
 
-DeploymentProviderPayloadResponsePayload = typing.Union[typing.Dict[str, typing.Any], str]
+DeploymentProviderPayloadResponsePayload = typing.Union[typing.Dict[str, typing.Optional[typing.Any]], str]

vellum/types/deployment_read.py

@@ -1,29 +1,29 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from ..core.pydantic_utilities import UniversalBaseModel
 import datetime as dt
+import pydantic
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .vellum_variable import VellumVariable
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class DeploymentRead(pydantic_v1.BaseModel):
+class DeploymentRead(UniversalBaseModel):
     id: str
     created: dt.datetime
-    label: str = pydantic_v1.Field()
+    label: str = pydantic.Field()
     """
     A human-readable label for the deployment
     """
 
-    name: str = pydantic_v1.Field()
+    name: str = pydantic.Field()
     """
     A name that uniquely identifies this deployment within its workspace
     """
 
-    status: typing.Optional[EntityStatus] = pydantic_v1.Field(default=None)
+    status: typing.Optional[EntityStatus] = pydantic.Field(default=None)
     """
     The current status of the deployment
     
@@ -31,7 +31,7 @@ class DeploymentRead(pydantic_v1.BaseModel):
     - `ARCHIVED` - Archived
     """
 
-    environment: typing.Optional[EnvironmentEnum] = pydantic_v1.Field(default=None)
+    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
     The environment this deployment is used in
     
@@ -42,35 +42,26 @@ class DeploymentRead(pydantic_v1.BaseModel):
 
     last_deployed_on: dt.datetime
     input_variables: typing.List[VellumVariable]
-    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    description: typing.Optional[str] = pydantic.Field(default=None)
     """
     A human-readable description of the deployment
     """
 
-    active_model_version_ids: typing.List[str] = pydantic_v1.Field()
+    active_model_version_ids: typing.List[str] = pydantic.Field()
     """
     Deprecated. The Prompt execution endpoints return a `prompt_version_id` that could be used instead.
     """
 
-    last_deployed_history_item_id: str = pydantic_v1.Field()
+    last_deployed_history_item_id: str = pydantic.Field()
     """
     The ID of the history item associated with this Deployment's LATEST Release Tag
     """
 
-    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/deployment_release_tag_deployment_history_item.py

@@ -1,30 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from ..core.pydantic_utilities import UniversalBaseModel
 import datetime as dt
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+import pydantic
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class DeploymentReleaseTagDeploymentHistoryItem(pydantic_v1.BaseModel):
+class DeploymentReleaseTagDeploymentHistoryItem(UniversalBaseModel):
     id: str
     timestamp: dt.datetime
 
-    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/deployment_release_tag_read.py

@@ -1,21 +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 .deployment_release_tag_deployment_history_item import DeploymentReleaseTagDeploymentHistoryItem
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
 from .release_tag_source import ReleaseTagSource
+from .deployment_release_tag_deployment_history_item import DeploymentReleaseTagDeploymentHistoryItem
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
 
 
-class DeploymentReleaseTagRead(pydantic_v1.BaseModel):
-    name: str = pydantic_v1.Field()
+class DeploymentReleaseTagRead(UniversalBaseModel):
+    name: str = pydantic.Field()
     """
     The name of the Release Tag
     """
 
-    source: ReleaseTagSource = pydantic_v1.Field()
+    source: ReleaseTagSource = pydantic.Field()
     """
     The source of how the Release Tag was originally created
     
@@ -23,25 +22,16 @@ class DeploymentReleaseTagRead(pydantic_v1.BaseModel):
     - `USER` - User
     """
 
-    history_item: DeploymentReleaseTagDeploymentHistoryItem = pydantic_v1.Field()
+    history_item: DeploymentReleaseTagDeploymentHistoryItem = pydantic.Field()
     """
     The Deployment History Item that this Release Tag is associated with
     """
 
-    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/document_document_to_document_index.py

@@ -1,25 +1,24 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .indexing_state_enum import IndexingStateEnum
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class DocumentDocumentToDocumentIndex(pydantic_v1.BaseModel):
-    id: str = pydantic_v1.Field()
+class DocumentDocumentToDocumentIndex(UniversalBaseModel):
+    id: str = pydantic.Field()
     """
     Vellum-generated ID that uniquely identifies this link.
     """
 
-    document_index_id: str = pydantic_v1.Field()
+    document_index_id: str = pydantic.Field()
     """
     Vellum-generated ID that uniquely identifies the index this document is included in.
     """
 
-    indexing_state: typing.Optional[IndexingStateEnum] = pydantic_v1.Field(default=None)
+    indexing_state: typing.Optional[IndexingStateEnum] = pydantic.Field(default=None)
     """
     An enum value representing where this document is along its indexing lifecycle for this index.
     
@@ -30,20 +29,11 @@ class DocumentDocumentToDocumentIndex(pydantic_v1.BaseModel):
     - `FAILED` - Failed
     """
 
-    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/document_index_chunking.py

@@ -1,88 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
-
-import datetime as dt
 import typing
+from .reducto_chunking import ReductoChunking
+from .sentence_chunking import SentenceChunking
+from .token_overlapping_window_chunking import TokenOverlappingWindowChunking
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .reducto_chunker_config import ReductoChunkerConfig
-from .sentence_chunker_config import SentenceChunkerConfig
-from .token_overlapping_window_chunker_config import TokenOverlappingWindowChunkerConfig
-
-
-class DocumentIndexChunking_ReductoChunker(pydantic_v1.BaseModel):
-    chunker_config: typing.Optional[ReductoChunkerConfig] = None
-    chunker_name: typing.Literal["reducto-chunker"] = "reducto-chunker"
-
-    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 DocumentIndexChunking_SentenceChunker(pydantic_v1.BaseModel):
-    chunker_config: typing.Optional[SentenceChunkerConfig] = None
-    chunker_name: typing.Literal["sentence-chunker"] = "sentence-chunker"
-
-    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 DocumentIndexChunking_TokenOverlappingWindowChunker(pydantic_v1.BaseModel):
-    chunker_config: typing.Optional[TokenOverlappingWindowChunkerConfig] = None
-    chunker_name: typing.Literal["token-overlapping-window-chunker"] = "token-overlapping-window-chunker"
-
-    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}
-
-
-DocumentIndexChunking = typing.Union[
-    DocumentIndexChunking_ReductoChunker,
-    DocumentIndexChunking_SentenceChunker,
-    DocumentIndexChunking_TokenOverlappingWindowChunker,
-]
+DocumentIndexChunking = typing.Union[ReductoChunking, SentenceChunking, TokenOverlappingWindowChunking]

vellum/types/document_index_chunking_request.py

@@ -1,88 +1,10 @@
 # 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
-from .reducto_chunker_config_request import ReductoChunkerConfigRequest
-from .sentence_chunker_config_request import SentenceChunkerConfigRequest
-from .token_overlapping_window_chunker_config_request import TokenOverlappingWindowChunkerConfigRequest
-
-
-class DocumentIndexChunkingRequest_ReductoChunker(pydantic_v1.BaseModel):
-    chunker_config: typing.Optional[ReductoChunkerConfigRequest] = None
-    chunker_name: typing.Literal["reducto-chunker"] = "reducto-chunker"
-
-    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 DocumentIndexChunkingRequest_SentenceChunker(pydantic_v1.BaseModel):
-    chunker_config: typing.Optional[SentenceChunkerConfigRequest] = None
-    chunker_name: typing.Literal["sentence-chunker"] = "sentence-chunker"
-
-    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 DocumentIndexChunkingRequest_TokenOverlappingWindowChunker(pydantic_v1.BaseModel):
-    chunker_config: typing.Optional[TokenOverlappingWindowChunkerConfigRequest] = None
-    chunker_name: typing.Literal["token-overlapping-window-chunker"] = "token-overlapping-window-chunker"
-
-    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}
-
+from .reducto_chunking_request import ReductoChunkingRequest
+from .sentence_chunking_request import SentenceChunkingRequest
+from .token_overlapping_window_chunking_request import TokenOverlappingWindowChunkingRequest
 
 DocumentIndexChunkingRequest = typing.Union[
-    DocumentIndexChunkingRequest_ReductoChunker,
-    DocumentIndexChunkingRequest_SentenceChunker,
-    DocumentIndexChunkingRequest_TokenOverlappingWindowChunker,
+    ReductoChunkingRequest, SentenceChunkingRequest, TokenOverlappingWindowChunkingRequest
 ]

vellum/types/document_index_indexing_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
+from .indexing_config_vectorizer import IndexingConfigVectorizer
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .document_index_chunking import DocumentIndexChunking
-from .indexing_config_vectorizer import IndexingConfigVectorizer
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class DocumentIndexIndexingConfig(pydantic_v1.BaseModel):
+class DocumentIndexIndexingConfig(UniversalBaseModel):
     vectorizer: IndexingConfigVectorizer
     chunking: typing.Optional[DocumentIndexChunking] = 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/document_index_indexing_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
+from .indexing_config_vectorizer_request import IndexingConfigVectorizerRequest
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .document_index_chunking_request import DocumentIndexChunkingRequest
-from .indexing_config_vectorizer_request import IndexingConfigVectorizerRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class DocumentIndexIndexingConfigRequest(pydantic_v1.BaseModel):
+class DocumentIndexIndexingConfigRequest(UniversalBaseModel):
     vectorizer: IndexingConfigVectorizerRequest
     chunking: typing.Optional[DocumentIndexChunkingRequest] = 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