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

vellum/types/document_index_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 .document_index_indexing_config import DocumentIndexIndexingConfig
 from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
+from .document_index_indexing_config import DocumentIndexIndexingConfig
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class DocumentIndexRead(pydantic_v1.BaseModel):
+class DocumentIndexRead(UniversalBaseModel):
     id: str
     created: dt.datetime
-    label: str = pydantic_v1.Field()
+    label: str = pydantic.Field()
     """
     A human-readable label for the document index
     """
 
-    name: str = pydantic_v1.Field()
+    name: str = pydantic.Field()
     """
     A name that uniquely identifies this index 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 document index
     
@@ -31,7 +31,7 @@ class DocumentIndexRead(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 document index is used in
     
@@ -42,20 +42,11 @@ class DocumentIndexRead(pydantic_v1.BaseModel):
 
     indexing_config: DocumentIndexIndexingConfig
 
-    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_read.py

@@ -1,29 +1,29 @@
 # 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 .document_document_to_document_index import DocumentDocumentToDocumentIndex
-from .document_status import DocumentStatus
+import pydantic
+import datetime as dt
 from .processing_state_enum import ProcessingStateEnum
+from .document_status import DocumentStatus
+from .document_document_to_document_index import DocumentDocumentToDocumentIndex
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class DocumentRead(pydantic_v1.BaseModel):
+class DocumentRead(UniversalBaseModel):
     id: str
-    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    external_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     The unique id of this document as it exists in the user's system.
     """
 
     last_uploaded_at: dt.datetime
-    label: str = pydantic_v1.Field()
+    label: str = pydantic.Field()
     """
     A human-readable label for the document. Defaults to the originally uploaded file's file name.
     """
 
-    processing_state: typing.Optional[ProcessingStateEnum] = pydantic_v1.Field(default=None)
+    processing_state: typing.Optional[ProcessingStateEnum] = pydantic.Field(default=None)
     """
     The current processing state of the document
     
@@ -33,7 +33,7 @@ class DocumentRead(pydantic_v1.BaseModel):
     - `FAILED` - Failed
     """
 
-    status: typing.Optional[DocumentStatus] = pydantic_v1.Field(default=None)
+    status: typing.Optional[DocumentStatus] = pydantic.Field(default=None)
     """
     The current status of the document
     
@@ -43,25 +43,16 @@ class DocumentRead(pydantic_v1.BaseModel):
     original_file_url: typing.Optional[str] = None
     processed_file_url: typing.Optional[str] = None
     document_to_document_indexes: typing.List[DocumentDocumentToDocumentIndex]
-    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.
     """
 
-    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/enriched_normalized_completion.py

@@ -1,32 +1,31 @@
 # 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 .finish_reason_enum import FinishReasonEnum
 from .normalized_log_probs import NormalizedLogProbs
 from .vellum_variable_type import VellumVariableType
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class EnrichedNormalizedCompletion(pydantic_v1.BaseModel):
-    id: str = pydantic_v1.Field()
+class EnrichedNormalizedCompletion(UniversalBaseModel):
+    id: str = pydantic.Field()
     """
     The Vellum-generated ID of the completion.
     """
 
-    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 along with the generation request, which uniquely identifies this generation in an external system.
     """
 
-    text: str = pydantic_v1.Field()
+    text: str = pydantic.Field()
     """
     The text generated by the LLM.
     """
 
-    finish_reason: typing.Optional[FinishReasonEnum] = pydantic_v1.Field(default=None)
+    finish_reason: typing.Optional[FinishReasonEnum] = pydantic.Field(default=None)
     """
     The reason the generation finished.
     
@@ -35,12 +34,12 @@ class EnrichedNormalizedCompletion(pydantic_v1.BaseModel):
     - `UNKNOWN` - UNKNOWN
     """
 
-    logprobs: typing.Optional[NormalizedLogProbs] = pydantic_v1.Field(default=None)
+    logprobs: typing.Optional[NormalizedLogProbs] = pydantic.Field(default=None)
     """
     The logprobs of the completion. Only present if specified in the original request options.
     """
 
-    model_version_id: str = pydantic_v1.Field()
+    model_version_id: str = pydantic.Field()
     """
     The ID of the model version used to generate this completion.
     """
@@ -50,20 +49,11 @@ class EnrichedNormalizedCompletion(pydantic_v1.BaseModel):
     deployment_release_tag: str
     model_name: str
 
-    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/error_variable_value.py

@@ -1,30 +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 .vellum_error import VellumError
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class ErrorVariableValue(pydantic_v1.BaseModel):
+class ErrorVariableValue(UniversalBaseModel):
+    type: typing.Literal["ERROR"] = "ERROR"
     value: typing.Optional[VellumError] = 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/error_vellum_value.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 .vellum_error import VellumError
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class ErrorVellumValue(pydantic_v1.BaseModel):
+class ErrorVellumValue(UniversalBaseModel):
     """
     A value representing an Error.
     """
 
+    type: typing.Literal["ERROR"] = "ERROR"
     value: typing.Optional[VellumError] = 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/error_vellum_value_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 .vellum_error_request import VellumErrorRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class ErrorVellumValueRequest(pydantic_v1.BaseModel):
+class ErrorVellumValueRequest(UniversalBaseModel):
     """
     A value representing an Error.
     """
 
+    type: typing.Literal["ERROR"] = "ERROR"
     value: typing.Optional[VellumErrorRequest] = 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/execute_prompt_event.py

@@ -1,124 +1,11 @@
 # 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 .fulfilled_prompt_execution_meta import FulfilledPromptExecutionMeta
-from .initiated_prompt_execution_meta import InitiatedPromptExecutionMeta
-from .prompt_output import PromptOutput
-from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
-from .streaming_prompt_execution_meta import StreamingPromptExecutionMeta
-from .vellum_error import VellumError
-
-
-class ExecutePromptEvent_Initiated(pydantic_v1.BaseModel):
-    meta: typing.Optional[InitiatedPromptExecutionMeta] = None
-    execution_id: str
-    state: typing.Literal["INITIATED"] = "INITIATED"
-
-    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 ExecutePromptEvent_Streaming(pydantic_v1.BaseModel):
-    output: PromptOutput
-    output_index: int
-    execution_id: str
-    meta: typing.Optional[StreamingPromptExecutionMeta] = None
-    raw: typing.Optional[typing.Dict[str, typing.Any]] = None
-    state: typing.Literal["STREAMING"] = "STREAMING"
-
-    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 ExecutePromptEvent_Fulfilled(pydantic_v1.BaseModel):
-    outputs: typing.List[PromptOutput]
-    execution_id: str
-    meta: typing.Optional[FulfilledPromptExecutionMeta] = None
-    state: typing.Literal["FULFILLED"] = "FULFILLED"
-
-    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 ExecutePromptEvent_Rejected(pydantic_v1.BaseModel):
-    error: VellumError
-    execution_id: str
-    meta: typing.Optional[RejectedPromptExecutionMeta] = None
-    state: typing.Literal["REJECTED"] = "REJECTED"
-
-    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 .initiated_execute_prompt_event import InitiatedExecutePromptEvent
+from .streaming_execute_prompt_event import StreamingExecutePromptEvent
+from .fulfilled_execute_prompt_event import FulfilledExecutePromptEvent
+from .rejected_execute_prompt_event import RejectedExecutePromptEvent
 
 ExecutePromptEvent = typing.Union[
-    ExecutePromptEvent_Initiated,
-    ExecutePromptEvent_Streaming,
-    ExecutePromptEvent_Fulfilled,
-    ExecutePromptEvent_Rejected,
+    InitiatedExecutePromptEvent, StreamingExecutePromptEvent, FulfilledExecutePromptEvent, RejectedExecutePromptEvent
 ]

vellum/types/execute_prompt_response.py

@@ -1,67 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
-
-import datetime as dt
 import typing
+from .fulfilled_execute_prompt_response import FulfilledExecutePromptResponse
+from .rejected_execute_prompt_response import RejectedExecutePromptResponse
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .prompt_execution_meta import PromptExecutionMeta
-from .prompt_output import PromptOutput
-from .vellum_error import VellumError
-
-
-class ExecutePromptResponse_Fulfilled(pydantic_v1.BaseModel):
-    meta: typing.Optional[PromptExecutionMeta] = None
-    raw: typing.Optional[typing.Dict[str, typing.Any]] = None
-    execution_id: str
-    outputs: typing.List[PromptOutput]
-    state: typing.Literal["FULFILLED"] = "FULFILLED"
-
-    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 ExecutePromptResponse_Rejected(pydantic_v1.BaseModel):
-    meta: typing.Optional[PromptExecutionMeta] = None
-    raw: typing.Optional[typing.Dict[str, typing.Any]] = None
-    execution_id: str
-    error: VellumError
-    state: typing.Literal["REJECTED"] = "REJECTED"
-
-    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}
-
-
-ExecutePromptResponse = typing.Union[ExecutePromptResponse_Fulfilled, ExecutePromptResponse_Rejected]
+ExecutePromptResponse = typing.Union[FulfilledExecutePromptResponse, RejectedExecutePromptResponse]

vellum/types/execute_workflow_response.py

@@ -1,33 +1,23 @@
 # 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 .execute_workflow_workflow_result_event import ExecuteWorkflowWorkflowResultEvent
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class ExecuteWorkflowResponse(pydantic_v1.BaseModel):
+class ExecuteWorkflowResponse(UniversalBaseModel):
     execution_id: str
     run_id: typing.Optional[str] = None
     external_id: typing.Optional[str] = None
     data: ExecuteWorkflowWorkflowResultEvent
 
-    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