vellum-ai 0.3.14__py3-none-any.whl → 0.3.16__py3-none-any.whl

vellum/types/submit_workflow_execution_actual_request.py

@@ -4,38 +4,39 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .workflow_execution_actual_chat_history_request import WorkflowExecutionActualChatHistoryRequest
 from .workflow_execution_actual_json_request import WorkflowExecutionActualJsonRequest
 from .workflow_execution_actual_string_request import WorkflowExecutionActualStringRequest
 
 
 class SubmitWorkflowExecutionActualRequest_String(WorkflowExecutionActualStringRequest):
-    output_type: typing_extensions.Literal["STRING"]
+    output_type: typing.Literal["STRING"] = "STRING"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class SubmitWorkflowExecutionActualRequest_Json(WorkflowExecutionActualJsonRequest):
-    output_type: typing_extensions.Literal["JSON"]
+    output_type: typing.Literal["JSON"] = "JSON"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class SubmitWorkflowExecutionActualRequest_ChatHistory(WorkflowExecutionActualChatHistoryRequest):
-    output_type: typing_extensions.Literal["CHAT_HISTORY"]
+    output_type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 SubmitWorkflowExecutionActualRequest = typing.Union[

vellum/types/subworkflow_enum.py

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing_extensions
+import typing
 
-SubworkflowEnum = typing_extensions.Literal["SUBWORKFLOW"]
+SubworkflowEnum = typing.Literal["SUBWORKFLOW"]

vellum/types/subworkflow_node_result.py

@@ -27,4 +27,5 @@ class SubworkflowNodeResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_array_result.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .array_variable_value_item import ArrayVariableValueItem
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class TemplatingNodeArrayResult(pydantic.BaseModel):
+    id: str
+    value: typing.Optional[typing.List[ArrayVariableValueItem]] = 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: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_chat_history_result.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class TemplatingNodeChatHistoryResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[typing.List[ChatMessage]]
+    value: typing.Optional[typing.List[ChatMessage]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +27,5 @@ class TemplatingNodeChatHistoryResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_error_result.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class TemplatingNodeErrorResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[VellumError]
+    value: typing.Optional[VellumError] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +27,5 @@ class TemplatingNodeErrorResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_function_call_result.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .function_call import FunctionCall
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class TemplatingNodeFunctionCallResult(pydantic.BaseModel):
+    id: str
+    value: typing.Optional[FunctionCall] = 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: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_json_result.py

@@ -13,7 +13,7 @@ except ImportError:
 
 class TemplatingNodeJsonResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[typing.Dict[str, typing.Any]]
+    value: typing.Optional[typing.Dict[str, typing.Any]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -26,4 +26,5 @@ class TemplatingNodeJsonResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_number_result.py

@@ -13,7 +13,7 @@ except ImportError:
 
 class TemplatingNodeNumberResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[float]
+    value: typing.Optional[float] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -26,4 +26,5 @@ class TemplatingNodeNumberResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_result.py

@@ -30,4 +30,5 @@ class TemplatingNodeResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_result_data.py

@@ -26,4 +26,5 @@ class TemplatingNodeResultData(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_result_output.py

@@ -4,10 +4,10 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
+from .templating_node_array_result import TemplatingNodeArrayResult
 from .templating_node_chat_history_result import TemplatingNodeChatHistoryResult
 from .templating_node_error_result import TemplatingNodeErrorResult
+from .templating_node_function_call_result import TemplatingNodeFunctionCallResult
 from .templating_node_json_result import TemplatingNodeJsonResult
 from .templating_node_number_result import TemplatingNodeNumberResult
 from .templating_node_search_results_result import TemplatingNodeSearchResultsResult
@@ -15,57 +15,83 @@ from .templating_node_string_result import TemplatingNodeStringResult
 
 
 class TemplatingNodeResultOutput_String(TemplatingNodeStringResult):
-    type: typing_extensions.Literal["STRING"]
+    type: typing.Literal["STRING"] = "STRING"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TemplatingNodeResultOutput_Number(TemplatingNodeNumberResult):
-    type: typing_extensions.Literal["NUMBER"]
+    type: typing.Literal["NUMBER"] = "NUMBER"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TemplatingNodeResultOutput_Json(TemplatingNodeJsonResult):
-    type: typing_extensions.Literal["JSON"]
+    type: typing.Literal["JSON"] = "JSON"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TemplatingNodeResultOutput_ChatHistory(TemplatingNodeChatHistoryResult):
-    type: typing_extensions.Literal["CHAT_HISTORY"]
+    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TemplatingNodeResultOutput_SearchResults(TemplatingNodeSearchResultsResult):
-    type: typing_extensions.Literal["SEARCH_RESULTS"]
+    type: typing.Literal["SEARCH_RESULTS"] = "SEARCH_RESULTS"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TemplatingNodeResultOutput_Error(TemplatingNodeErrorResult):
-    type: typing_extensions.Literal["ERROR"]
+    type: typing.Literal["ERROR"] = "ERROR"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class TemplatingNodeResultOutput_Array(TemplatingNodeArrayResult):
+    type: typing.Literal["ARRAY"] = "ARRAY"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class TemplatingNodeResultOutput_FunctionCall(TemplatingNodeFunctionCallResult):
+    type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 TemplatingNodeResultOutput = typing.Union[
@@ -75,4 +101,6 @@ TemplatingNodeResultOutput = typing.Union[
     TemplatingNodeResultOutput_ChatHistory,
     TemplatingNodeResultOutput_SearchResults,
     TemplatingNodeResultOutput_Error,
+    TemplatingNodeResultOutput_Array,
+    TemplatingNodeResultOutput_FunctionCall,
 ]

vellum/types/templating_node_search_results_result.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class TemplatingNodeSearchResultsResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[typing.List[SearchResult]]
+    value: typing.Optional[typing.List[SearchResult]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +27,5 @@ class TemplatingNodeSearchResultsResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_string_result.py

@@ -13,7 +13,7 @@ except ImportError:
 
 class TemplatingNodeStringResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[str]
+    value: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -26,4 +26,5 @@ class TemplatingNodeStringResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/terminal_node_array_result.py

@@ -13,9 +13,13 @@ except ImportError:
 
 
 class TerminalNodeArrayResult(pydantic.BaseModel):
-    id: typing.Optional[str]
-    name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: typing.Optional[typing.List[ArrayVariableValueItem]]
+    id: typing.Optional[str] = None
+    name: str = pydantic.Field()
+    """
+    The unique name given to the terminal node that produced this output.
+    """
+
+    value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,4 +32,5 @@ class TerminalNodeArrayResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/terminal_node_chat_history_result.py

@@ -13,9 +13,13 @@ except ImportError:
 
 
 class TerminalNodeChatHistoryResult(pydantic.BaseModel):
-    id: typing.Optional[str]
-    name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: typing.Optional[typing.List[ChatMessage]]
+    id: typing.Optional[str] = None
+    name: str = pydantic.Field()
+    """
+    The unique name given to the terminal node that produced this output.
+    """
+
+    value: typing.Optional[typing.List[ChatMessage]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,4 +32,5 @@ class TerminalNodeChatHistoryResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/terminal_node_error_result.py

@@ -13,9 +13,13 @@ except ImportError:
 
 
 class TerminalNodeErrorResult(pydantic.BaseModel):
-    id: typing.Optional[str]
-    name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: typing.Optional[VellumError]
+    id: typing.Optional[str] = None
+    name: str = pydantic.Field()
+    """
+    The unique name given to the terminal node that produced this output.
+    """
+
+    value: typing.Optional[VellumError] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,4 +32,5 @@ class TerminalNodeErrorResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/terminal_node_function_call_result.py

@@ -13,9 +13,13 @@ except ImportError:
 
 
 class TerminalNodeFunctionCallResult(pydantic.BaseModel):
-    id: typing.Optional[str]
-    name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: typing.Optional[FunctionCall]
+    id: typing.Optional[str] = None
+    name: str = pydantic.Field()
+    """
+    The unique name given to the terminal node that produced this output.
+    """
+
+    value: typing.Optional[FunctionCall] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,4 +32,5 @@ class TerminalNodeFunctionCallResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/terminal_node_json_result.py

@@ -12,9 +12,13 @@ except ImportError:
 
 
 class TerminalNodeJsonResult(pydantic.BaseModel):
-    id: typing.Optional[str]
-    name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: typing.Optional[typing.Dict[str, typing.Any]]
+    id: typing.Optional[str] = None
+    name: str = pydantic.Field()
+    """
+    The unique name given to the terminal node that produced this output.
+    """
+
+    value: typing.Optional[typing.Dict[str, typing.Any]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +31,5 @@ class TerminalNodeJsonResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/terminal_node_number_result.py

@@ -12,9 +12,13 @@ except ImportError:
 
 
 class TerminalNodeNumberResult(pydantic.BaseModel):
-    id: typing.Optional[str]
-    name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: typing.Optional[float]
+    id: typing.Optional[str] = None
+    name: str = pydantic.Field()
+    """
+    The unique name given to the terminal node that produced this output.
+    """
+
+    value: typing.Optional[float] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +31,5 @@ class TerminalNodeNumberResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/terminal_node_result.py

@@ -30,4 +30,5 @@ class TerminalNodeResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/terminal_node_result_data.py

@@ -26,4 +26,5 @@ class TerminalNodeResultData(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/terminal_node_result_output.py

@@ -4,8 +4,6 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .terminal_node_array_result import TerminalNodeArrayResult
 from .terminal_node_chat_history_result import TerminalNodeChatHistoryResult
 from .terminal_node_error_result import TerminalNodeErrorResult
@@ -17,75 +15,83 @@ from .terminal_node_string_result import TerminalNodeStringResult
 
 
 class TerminalNodeResultOutput_String(TerminalNodeStringResult):
-    type: typing_extensions.Literal["STRING"]
+    type: typing.Literal["STRING"] = "STRING"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TerminalNodeResultOutput_Number(TerminalNodeNumberResult):
-    type: typing_extensions.Literal["NUMBER"]
+    type: typing.Literal["NUMBER"] = "NUMBER"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TerminalNodeResultOutput_Json(TerminalNodeJsonResult):
-    type: typing_extensions.Literal["JSON"]
+    type: typing.Literal["JSON"] = "JSON"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TerminalNodeResultOutput_ChatHistory(TerminalNodeChatHistoryResult):
-    type: typing_extensions.Literal["CHAT_HISTORY"]
+    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TerminalNodeResultOutput_SearchResults(TerminalNodeSearchResultsResult):
-    type: typing_extensions.Literal["SEARCH_RESULTS"]
+    type: typing.Literal["SEARCH_RESULTS"] = "SEARCH_RESULTS"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TerminalNodeResultOutput_Array(TerminalNodeArrayResult):
-    type: typing_extensions.Literal["ARRAY"]
+    type: typing.Literal["ARRAY"] = "ARRAY"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TerminalNodeResultOutput_FunctionCall(TerminalNodeFunctionCallResult):
-    type: typing_extensions.Literal["FUNCTION_CALL"]
+    type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class TerminalNodeResultOutput_Error(TerminalNodeErrorResult):
-    type: typing_extensions.Literal["ERROR"]
+    type: typing.Literal["ERROR"] = "ERROR"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 TerminalNodeResultOutput = typing.Union[

vellum/types/terminal_node_search_results_result.py

@@ -13,9 +13,13 @@ except ImportError:
 
 
 class TerminalNodeSearchResultsResult(pydantic.BaseModel):
-    id: typing.Optional[str]
-    name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: typing.Optional[typing.List[SearchResult]]
+    id: typing.Optional[str] = None
+    name: str = pydantic.Field()
+    """
+    The unique name given to the terminal node that produced this output.
+    """
+
+    value: typing.Optional[typing.List[SearchResult]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,4 +32,5 @@ class TerminalNodeSearchResultsResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/terminal_node_string_result.py

@@ -12,9 +12,13 @@ except ImportError:
 
 
 class TerminalNodeStringResult(pydantic.BaseModel):
-    id: typing.Optional[str]
-    name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: typing.Optional[str]
+    id: typing.Optional[str] = None
+    name: str = pydantic.Field()
+    """
+    The unique name given to the terminal node that produced this output.
+    """
+
+    value: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +31,5 @@ class TerminalNodeStringResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_case_chat_history_variable_value.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class TestCaseChatHistoryVariableValue(pydantic.BaseModel):
     variable_id: str
-    value: typing.Optional[typing.List[ChatMessage]]
+    value: typing.Optional[typing.List[ChatMessage]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +27,5 @@ class TestCaseChatHistoryVariableValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}