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

vellum/types/reducto_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 .reducto_chunker_config_request import ReductoChunkerConfigRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class ReductoChunkingRequest(pydantic_v1.BaseModel):
+class ReductoChunkingRequest(UniversalBaseModel):
     """
     Reducto chunking
     """
 
+    chunker_name: typing.Literal["reducto-chunker"] = "reducto-chunker"
     chunker_config: typing.Optional[ReductoChunkerConfigRequest] = 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/rejected_execute_prompt_event.py

@@ -1,37 +1,28 @@
 # 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 .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
 from .vellum_error import VellumError
+from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class RejectedExecutePromptEvent(pydantic_v1.BaseModel):
+class RejectedExecutePromptEvent(UniversalBaseModel):
     """
     The final data returned indicating an error occurred during the stream.
     """
 
+    state: typing.Literal["REJECTED"] = "REJECTED"
     error: VellumError
     execution_id: str
     meta: typing.Optional[RejectedPromptExecutionMeta] = 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/rejected_execute_prompt_response.py

@@ -1,46 +1,37 @@
 # 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_execution_meta import PromptExecutionMeta
+import pydantic
 from .vellum_error import VellumError
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class RejectedExecutePromptResponse(pydantic_v1.BaseModel):
+class RejectedExecutePromptResponse(UniversalBaseModel):
     """
     The unsuccessful response from the model containing an error of what went wrong.
     """
 
     meta: typing.Optional[PromptExecutionMeta] = 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`.
     """
 
-    execution_id: str = pydantic_v1.Field()
+    execution_id: str = pydantic.Field()
     """
     The ID of the execution.
     """
 
+    state: typing.Literal["REJECTED"] = "REJECTED"
     error: VellumError
 
-    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/rejected_execute_workflow_workflow_result_event.py

@@ -1,36 +1,28 @@
 # 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 datetime as dt
 from .workflow_event_error import WorkflowEventError
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class RejectedExecuteWorkflowWorkflowResultEvent(pydantic_v1.BaseModel):
+class RejectedExecuteWorkflowWorkflowResultEvent(UniversalBaseModel):
     """
     The unsuccessful response from the Workflow execution containing an error specifying what went wrong.
     """
 
     id: str
+    state: typing.Literal["REJECTED"] = "REJECTED"
     ts: dt.datetime
     error: WorkflowEventError
 
-    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/rejected_prompt_execution_meta.py

@@ -1,14 +1,13 @@
 # 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 .finish_reason_enum import FinishReasonEnum
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class RejectedPromptExecutionMeta(pydantic_v1.BaseModel):
+class RejectedPromptExecutionMeta(UniversalBaseModel):
     """
     The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
     """
@@ -16,20 +15,11 @@ class RejectedPromptExecutionMeta(pydantic_v1.BaseModel):
     latency: typing.Optional[int] = None
     finish_reason: typing.Optional[FinishReasonEnum] = 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/rejected_workflow_node_result_event.py

@@ -1,15 +1,15 @@
 # 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 .workflow_event_error import WorkflowEventError
+import datetime as dt
 from .workflow_node_result_data import WorkflowNodeResultData
+from .workflow_event_error import WorkflowEventError
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class RejectedWorkflowNodeResultEvent(pydantic_v1.BaseModel):
+class RejectedWorkflowNodeResultEvent(UniversalBaseModel):
     """
     An event that indicates that the node has rejected its execution.
     """
@@ -17,25 +17,17 @@ class RejectedWorkflowNodeResultEvent(pydantic_v1.BaseModel):
     id: str
     node_id: str
     node_result_id: str
+    state: typing.Literal["REJECTED"] = "REJECTED"
     ts: typing.Optional[dt.datetime] = None
     data: typing.Optional[WorkflowNodeResultData] = None
     source_execution_id: typing.Optional[str] = None
     error: WorkflowEventError
 
-    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/replace_test_suite_test_case_request.py

@@ -1,57 +1,47 @@
 # 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 ReplaceTestSuiteTestCaseRequest(pydantic_v1.BaseModel):
+class ReplaceTestSuiteTestCaseRequest(UniversalBaseModel):
     """
     Information about the Test Case to replace
     """
 
-    id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    id: typing.Optional[str] = pydantic.Field(default=None)
     """
     The Vellum-generated ID of the Test Case whose data you'd like to replace. Must specify either this or external_id.
     """
 
-    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    external_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     The ID that was originally provided upon Test Case creation that uniquely identifies the Test Case whose data you'd like to replace. Must specify either this of id.
     """
 
-    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
     """
 
-    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/sandbox_scenario.py

@@ -1,43 +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 .scenario_input import ScenarioInput
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class SandboxScenario(pydantic_v1.BaseModel):
+class SandboxScenario(UniversalBaseModel):
     """
     Sandbox Scenario
     """
 
     label: typing.Optional[str] = None
-    inputs: typing.List[ScenarioInput] = pydantic_v1.Field()
+    inputs: typing.List[ScenarioInput] = pydantic.Field()
     """
     The inputs for the scenario
     """
 
-    id: str = pydantic_v1.Field()
+    id: str = pydantic.Field()
     """
     The id of the scenario
     """
 
-    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/scenario_input.py

@@ -1,85 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
-
-import datetime as dt
 import typing
+from .scenario_input_string_variable_value import ScenarioInputStringVariableValue
+from .scenario_input_json_variable_value import ScenarioInputJsonVariableValue
+from .scenario_input_chat_history_variable_value import ScenarioInputChatHistoryVariableValue
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .chat_message import ChatMessage
-
-
-class ScenarioInput_String(pydantic_v1.BaseModel):
-    value: typing.Optional[str] = None
-    input_variable_id: str
-    type: typing.Literal["STRING"] = "STRING"
-
-    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 ScenarioInput_Json(pydantic_v1.BaseModel):
-    value: typing.Any
-    input_variable_id: str
-    type: typing.Literal["JSON"] = "JSON"
-
-    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 ScenarioInput_ChatHistory(pydantic_v1.BaseModel):
-    value: typing.Optional[typing.List[ChatMessage]] = None
-    input_variable_id: str
-    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
-
-    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}
-
-
-ScenarioInput = typing.Union[ScenarioInput_String, ScenarioInput_Json, ScenarioInput_ChatHistory]
+ScenarioInput = typing.Union[
+    ScenarioInputStringVariableValue, ScenarioInputJsonVariableValue, ScenarioInputChatHistoryVariableValue
+]

vellum/types/scenario_input_chat_history_variable_value.py

@@ -1,35 +1,26 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .chat_message import ChatMessage
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class ScenarioInputChatHistoryVariableValue(pydantic_v1.BaseModel):
+class ScenarioInputChatHistoryVariableValue(UniversalBaseModel):
     """
     Prompt Sandbox Scenario input value that is of type CHAT_HISTORY
     """
 
+    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
     value: typing.Optional[typing.List[ChatMessage]] = None
     input_variable_id: 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/scenario_input_json_variable_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.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 ScenarioInputJsonVariableValue(pydantic_v1.BaseModel):
+class ScenarioInputJsonVariableValue(UniversalBaseModel):
     """
     Prompt Sandbox Scenario input value that is of type JSON
     """
 
-    value: typing.Any
+    type: typing.Literal["JSON"] = "JSON"
+    value: typing.Optional[typing.Any] = None
     input_variable_id: 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