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

vellum/types/search_result_request.py

@@ -1,50 +1,40 @@
 # 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 .search_result_document_request import SearchResultDocumentRequest
 from .search_result_meta_request import SearchResultMetaRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class SearchResultRequest(pydantic_v1.BaseModel):
-    text: str = pydantic_v1.Field()
+class SearchResultRequest(UniversalBaseModel):
+    text: str = pydantic.Field()
     """
     The text of the chunk that matched the search query.
     """
 
-    score: float = pydantic_v1.Field()
+    score: float = pydantic.Field()
     """
     A score representing how well the chunk matches the search query.
     """
 
     keywords: typing.List[str]
-    document: SearchResultDocumentRequest = pydantic_v1.Field()
+    document: SearchResultDocumentRequest = pydantic.Field()
     """
     The document that contains the chunk that matched the search query.
     """
 
-    meta: typing.Optional[SearchResultMetaRequest] = pydantic_v1.Field(default=None)
+    meta: typing.Optional[SearchResultMetaRequest] = pydantic.Field(default=None)
     """
     Additional information about the search result.
     """
 
-    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/search_weights_request.py

@@ -1,37 +1,27 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class SearchWeightsRequest(pydantic_v1.BaseModel):
-    semantic_similarity: typing.Optional[float] = pydantic_v1.Field(default=None)
+class SearchWeightsRequest(UniversalBaseModel):
+    semantic_similarity: typing.Optional[float] = pydantic.Field(default=None)
     """
     The relative weight to give to semantic similarity
     """
 
-    keywords: typing.Optional[float] = pydantic_v1.Field(default=None)
+    keywords: typing.Optional[float] = pydantic.Field(default=None)
     """
     The relative weight to give to keywords
     """
 
-    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/sentence_chunker_config.py

@@ -1,13 +1,12 @@
 # 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.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class SentenceChunkerConfig(pydantic_v1.BaseModel):
+class SentenceChunkerConfig(UniversalBaseModel):
     """
     Configuration for sentence chunking
     """
@@ -15,20 +14,11 @@ class SentenceChunkerConfig(pydantic_v1.BaseModel):
     character_limit: typing.Optional[int] = None
     min_overlap_ratio: typing.Optional[float] = 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/sentence_chunker_config_request.py

@@ -1,13 +1,12 @@
 # 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.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class SentenceChunkerConfigRequest(pydantic_v1.BaseModel):
+class SentenceChunkerConfigRequest(UniversalBaseModel):
     """
     Configuration for sentence chunking
     """
@@ -15,20 +14,11 @@ class SentenceChunkerConfigRequest(pydantic_v1.BaseModel):
     character_limit: typing.Optional[int] = None
     min_overlap_ratio: typing.Optional[float] = 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/sentence_chunking.py

@@ -1,34 +1,25 @@
 # 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 .sentence_chunker_config import SentenceChunkerConfig
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class SentenceChunking(pydantic_v1.BaseModel):
+class SentenceChunking(UniversalBaseModel):
     """
     Sentence chunking
     """
 
+    chunker_name: typing.Literal["sentence-chunker"] = "sentence-chunker"
     chunker_config: typing.Optional[SentenceChunkerConfig] = 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/sentence_chunking_request.py

@@ -1,34 +1,25 @@
 # 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 .sentence_chunker_config_request import SentenceChunkerConfigRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class SentenceChunkingRequest(pydantic_v1.BaseModel):
+class SentenceChunkingRequest(UniversalBaseModel):
     """
     Sentence chunking
     """
 
+    chunker_name: typing.Literal["sentence-chunker"] = "sentence-chunker"
     chunker_config: typing.Optional[SentenceChunkerConfigRequest] = 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/slim_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 SlimDeploymentRead(pydantic_v1.BaseModel):
+class SlimDeploymentRead(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 SlimDeploymentRead(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,25 +42,16 @@ class SlimDeploymentRead(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
     """
 
-    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/slim_document.py

@@ -1,38 +1,38 @@
 # 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 .document_document_to_document_index import DocumentDocumentToDocumentIndex
-from .document_status import DocumentStatus
-from .processing_failure_reason_enum import ProcessingFailureReasonEnum
+import datetime as dt
 from .processing_state_enum import ProcessingStateEnum
+from .processing_failure_reason_enum import ProcessingFailureReasonEnum
+from .document_status import DocumentStatus
+from .document_document_to_document_index import DocumentDocumentToDocumentIndex
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class SlimDocument(pydantic_v1.BaseModel):
-    id: str = pydantic_v1.Field()
+class SlimDocument(UniversalBaseModel):
+    id: str = pydantic.Field()
     """
     Vellum-generated ID that uniquely identifies this document.
     """
 
-    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    external_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     The external ID that was originally provided when uploading the document.
     """
 
-    last_uploaded_at: dt.datetime = pydantic_v1.Field()
+    last_uploaded_at: dt.datetime = pydantic.Field()
     """
     A timestamp representing when this document was most recently uploaded.
     """
 
-    label: str = pydantic_v1.Field()
+    label: str = pydantic.Field()
     """
     Human-friendly name for this document.
     """
 
-    processing_state: typing.Optional[ProcessingStateEnum] = pydantic_v1.Field(default=None)
+    processing_state: typing.Optional[ProcessingStateEnum] = pydantic.Field(default=None)
     """
     An enum value representing where this document is along its processing lifecycle. Note that this is different than its indexing lifecycle.
     
@@ -42,7 +42,7 @@ class SlimDocument(pydantic_v1.BaseModel):
     - `FAILED` - Failed
     """
 
-    processing_failure_reason: typing.Optional[ProcessingFailureReasonEnum] = pydantic_v1.Field(default=None)
+    processing_failure_reason: typing.Optional[ProcessingFailureReasonEnum] = pydantic.Field(default=None)
     """
     An enum value representing why the document could not be processed. Is null unless processing_state is FAILED.
     
@@ -50,39 +50,30 @@ class SlimDocument(pydantic_v1.BaseModel):
     - `INVALID_FILE` - Invalid File
     """
 
-    status: typing.Optional[DocumentStatus] = pydantic_v1.Field(default=None)
+    status: typing.Optional[DocumentStatus] = pydantic.Field(default=None)
     """
     The document's current status.
     
     - `ACTIVE` - Active
     """
 
-    keywords: typing.Optional[typing.List[str]] = pydantic_v1.Field(default=None)
+    keywords: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
     """
     A list of keywords associated with this document. Originally provided when uploading the document.
     """
 
-    metadata: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     A previously supplied JSON object containing metadata that can be filtered on when searching.
     """
 
     document_to_document_indexes: typing.List[DocumentDocumentToDocumentIndex]
 
-    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/slim_workflow_deployment.py

@@ -1,28 +1,28 @@
 # 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 .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
+import datetime as dt
 from .vellum_variable import VellumVariable
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class SlimWorkflowDeployment(pydantic_v1.BaseModel):
+class SlimWorkflowDeployment(UniversalBaseModel):
     id: str
-    name: str = pydantic_v1.Field()
+    name: str = pydantic.Field()
     """
     A name that uniquely identifies this workflow deployment within its workspace
     """
 
-    label: str = pydantic_v1.Field()
+    label: str = pydantic.Field()
     """
     A human-readable label for the workflow deployment
     """
 
-    status: typing.Optional[EntityStatus] = pydantic_v1.Field(default=None)
+    status: typing.Optional[EntityStatus] = pydantic.Field(default=None)
     """
     The current status of the workflow deployment
     
@@ -30,7 +30,7 @@ class SlimWorkflowDeployment(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 workflow deployment is used in
     
@@ -41,35 +41,26 @@ class SlimWorkflowDeployment(pydantic_v1.BaseModel):
 
     created: dt.datetime
     last_deployed_on: dt.datetime
-    input_variables: typing.List[VellumVariable] = pydantic_v1.Field()
+    input_variables: typing.List[VellumVariable] = pydantic.Field()
     """
     The input variables this Workflow Deployment expects to receive values for when it is executed.
     """
 
-    output_variables: typing.List[VellumVariable] = pydantic_v1.Field()
+    output_variables: typing.List[VellumVariable] = pydantic.Field()
     """
     The output variables this Workflow Deployment will produce when it is executed.
     """
 
-    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    description: typing.Optional[str] = pydantic.Field(default=None)
     """
     A human-readable description of the workflow deployment
     """
 
-    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/streaming_execute_prompt_event.py

@@ -1,42 +1,33 @@
 # 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 .prompt_output import PromptOutput
 from .streaming_prompt_execution_meta import StreamingPromptExecutionMeta
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class StreamingExecutePromptEvent(pydantic_v1.BaseModel):
+class StreamingExecutePromptEvent(UniversalBaseModel):
     """
     The data returned for each delta during the prompt execution stream.
     """
 
+    state: typing.Literal["STREAMING"] = "STREAMING"
     output: PromptOutput
     output_index: int
     execution_id: str
     meta: typing.Optional[StreamingPromptExecutionMeta] = None
-    raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
+    raw: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     The subset of the raw response from the model that the request opted into with `expand_raw`.
     """
 
-    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