vellum-ai 0.3.13__py3-none-any.whl → 0.3.15__py3-none-any.whl

vellum/resources/workflow_deployments/client.py

@@ -6,7 +6,9 @@ from json.decoder import JSONDecodeError
 
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.jsonable_encoder import jsonable_encoder
 from ...core.remove_none_from_dict import remove_none_from_dict
+from ...core.request_options import RequestOptions
 from ...types.paginated_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploymentList
 from ...types.workflow_deployment_read import WorkflowDeploymentRead
 from .types.workflow_deployments_list_request_status import WorkflowDeploymentsListRequestStatus
@@ -28,6 +30,7 @@ class WorkflowDeploymentsClient:
         offset: typing.Optional[int] = None,
         ordering: typing.Optional[str] = None,
         status: typing.Optional[WorkflowDeploymentsListRequestStatus] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> PaginatedSlimWorkflowDeploymentList:
         """
         Parameters:
@@ -38,6 +41,8 @@ class WorkflowDeploymentsClient:
             - ordering: typing.Optional[str]. Which field to use when ordering the results.
 
             - status: typing.Optional[WorkflowDeploymentsListRequestStatus]. status
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from vellum.client import Vellum
 
@@ -49,9 +54,34 @@ class WorkflowDeploymentsClient:
         _response = self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/workflow-deployments"),
-            params=remove_none_from_dict({"limit": limit, "offset": offset, "ordering": ordering, "status": status}),
-            headers=self._client_wrapper.get_headers(),
-            timeout=None,
+            params=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        "limit": limit,
+                        "offset": offset,
+                        "ordering": ordering,
+                        "status": status,
+                        **(
+                            request_options.get("additional_query_parameters", {})
+                            if request_options is not None
+                            else {}
+                        ),
+                    }
+                )
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
             return pydantic.parse_obj_as(PaginatedSlimWorkflowDeploymentList, _response.json())  # type: ignore
@@ -61,12 +91,14 @@ class WorkflowDeploymentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def retrieve(self, id: str) -> WorkflowDeploymentRead:
+    def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> WorkflowDeploymentRead:
         """
         Used to retrieve a workflow deployment given its ID or name.
 
         Parameters:
             - id: str. Either the Workflow Deployment's ID or its unique name
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from vellum.client import Vellum
 
@@ -79,9 +111,25 @@ class WorkflowDeploymentsClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", f"v1/workflow-deployments/{id}"),
-            headers=self._client_wrapper.get_headers(),
-            timeout=None,
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_environment().default}/", f"v1/workflow-deployments/{jsonable_encoder(id)}"
+            ),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
             return pydantic.parse_obj_as(WorkflowDeploymentRead, _response.json())  # type: ignore
@@ -103,6 +151,7 @@ class AsyncWorkflowDeploymentsClient:
         offset: typing.Optional[int] = None,
         ordering: typing.Optional[str] = None,
         status: typing.Optional[WorkflowDeploymentsListRequestStatus] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> PaginatedSlimWorkflowDeploymentList:
         """
         Parameters:
@@ -113,6 +162,8 @@ class AsyncWorkflowDeploymentsClient:
             - ordering: typing.Optional[str]. Which field to use when ordering the results.
 
             - status: typing.Optional[WorkflowDeploymentsListRequestStatus]. status
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from vellum.client import AsyncVellum
 
@@ -124,9 +175,34 @@ class AsyncWorkflowDeploymentsClient:
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/workflow-deployments"),
-            params=remove_none_from_dict({"limit": limit, "offset": offset, "ordering": ordering, "status": status}),
-            headers=self._client_wrapper.get_headers(),
-            timeout=None,
+            params=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        "limit": limit,
+                        "offset": offset,
+                        "ordering": ordering,
+                        "status": status,
+                        **(
+                            request_options.get("additional_query_parameters", {})
+                            if request_options is not None
+                            else {}
+                        ),
+                    }
+                )
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
             return pydantic.parse_obj_as(PaginatedSlimWorkflowDeploymentList, _response.json())  # type: ignore
@@ -136,12 +212,16 @@ class AsyncWorkflowDeploymentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def retrieve(self, id: str) -> WorkflowDeploymentRead:
+    async def retrieve(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> WorkflowDeploymentRead:
         """
         Used to retrieve a workflow deployment given its ID or name.
 
         Parameters:
             - id: str. Either the Workflow Deployment's ID or its unique name
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from vellum.client import AsyncVellum
 
@@ -154,9 +234,25 @@ class AsyncWorkflowDeploymentsClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", f"v1/workflow-deployments/{id}"),
-            headers=self._client_wrapper.get_headers(),
-            timeout=None,
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_environment().default}/", f"v1/workflow-deployments/{jsonable_encoder(id)}"
+            ),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
             return pydantic.parse_obj_as(WorkflowDeploymentRead, _response.json())  # type: ignore

vellum/types/__init__.py

@@ -49,16 +49,20 @@ from .chat_message_content_request import (
 )
 from .chat_message_request import ChatMessageRequest
 from .chat_message_role import ChatMessageRole
+from .code_execution_node_array_result import CodeExecutionNodeArrayResult
 from .code_execution_node_chat_history_result import CodeExecutionNodeChatHistoryResult
 from .code_execution_node_error_result import CodeExecutionNodeErrorResult
+from .code_execution_node_function_call_result import CodeExecutionNodeFunctionCallResult
 from .code_execution_node_json_result import CodeExecutionNodeJsonResult
 from .code_execution_node_number_result import CodeExecutionNodeNumberResult
 from .code_execution_node_result import CodeExecutionNodeResult
 from .code_execution_node_result_data import CodeExecutionNodeResultData
 from .code_execution_node_result_output import (
     CodeExecutionNodeResultOutput,
+    CodeExecutionNodeResultOutput_Array,
     CodeExecutionNodeResultOutput_ChatHistory,
     CodeExecutionNodeResultOutput_Error,
+    CodeExecutionNodeResultOutput_FunctionCall,
     CodeExecutionNodeResultOutput_Json,
     CodeExecutionNodeResultOutput_Number,
     CodeExecutionNodeResultOutput_SearchResults,
@@ -185,6 +189,7 @@ from .named_test_case_variable_value_request import (
 from .node_input_compiled_array_value import NodeInputCompiledArrayValue
 from .node_input_compiled_chat_history_value import NodeInputCompiledChatHistoryValue
 from .node_input_compiled_error_value import NodeInputCompiledErrorValue
+from .node_input_compiled_function_call import NodeInputCompiledFunctionCall
 from .node_input_compiled_json_value import NodeInputCompiledJsonValue
 from .node_input_compiled_number_value import NodeInputCompiledNumberValue
 from .node_input_compiled_search_results_value import NodeInputCompiledSearchResultsValue
@@ -194,6 +199,7 @@ from .node_input_variable_compiled_value import (
     NodeInputVariableCompiledValue_Array,
     NodeInputVariableCompiledValue_ChatHistory,
     NodeInputVariableCompiledValue_Error,
+    NodeInputVariableCompiledValue_FunctionCall,
     NodeInputVariableCompiledValue_Json,
     NodeInputVariableCompiledValue_Number,
     NodeInputVariableCompiledValue_SearchResults,
@@ -311,16 +317,20 @@ from .submit_workflow_execution_actual_request import (
 )
 from .subworkflow_enum import SubworkflowEnum
 from .subworkflow_node_result import SubworkflowNodeResult
+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_result import TemplatingNodeResult
 from .templating_node_result_data import TemplatingNodeResultData
 from .templating_node_result_output import (
     TemplatingNodeResultOutput,
+    TemplatingNodeResultOutput_Array,
     TemplatingNodeResultOutput_ChatHistory,
     TemplatingNodeResultOutput_Error,
+    TemplatingNodeResultOutput_FunctionCall,
     TemplatingNodeResultOutput_Json,
     TemplatingNodeResultOutput_Number,
     TemplatingNodeResultOutput_SearchResults,
@@ -551,15 +561,19 @@ __all__ = [
     "ChatMessageContent_String",
     "ChatMessageRequest",
     "ChatMessageRole",
+    "CodeExecutionNodeArrayResult",
     "CodeExecutionNodeChatHistoryResult",
     "CodeExecutionNodeErrorResult",
+    "CodeExecutionNodeFunctionCallResult",
     "CodeExecutionNodeJsonResult",
     "CodeExecutionNodeNumberResult",
     "CodeExecutionNodeResult",
     "CodeExecutionNodeResultData",
     "CodeExecutionNodeResultOutput",
+    "CodeExecutionNodeResultOutput_Array",
     "CodeExecutionNodeResultOutput_ChatHistory",
     "CodeExecutionNodeResultOutput_Error",
+    "CodeExecutionNodeResultOutput_FunctionCall",
     "CodeExecutionNodeResultOutput_Json",
     "CodeExecutionNodeResultOutput_Number",
     "CodeExecutionNodeResultOutput_SearchResults",
@@ -677,6 +691,7 @@ __all__ = [
     "NodeInputCompiledArrayValue",
     "NodeInputCompiledChatHistoryValue",
     "NodeInputCompiledErrorValue",
+    "NodeInputCompiledFunctionCall",
     "NodeInputCompiledJsonValue",
     "NodeInputCompiledNumberValue",
     "NodeInputCompiledSearchResultsValue",
@@ -685,6 +700,7 @@ __all__ = [
     "NodeInputVariableCompiledValue_Array",
     "NodeInputVariableCompiledValue_ChatHistory",
     "NodeInputVariableCompiledValue_Error",
+    "NodeInputVariableCompiledValue_FunctionCall",
     "NodeInputVariableCompiledValue_Json",
     "NodeInputVariableCompiledValue_Number",
     "NodeInputVariableCompiledValue_SearchResults",
@@ -793,15 +809,19 @@ __all__ = [
     "SubmitWorkflowExecutionActualRequest_String",
     "SubworkflowEnum",
     "SubworkflowNodeResult",
+    "TemplatingNodeArrayResult",
     "TemplatingNodeChatHistoryResult",
     "TemplatingNodeErrorResult",
+    "TemplatingNodeFunctionCallResult",
     "TemplatingNodeJsonResult",
     "TemplatingNodeNumberResult",
     "TemplatingNodeResult",
     "TemplatingNodeResultData",
     "TemplatingNodeResultOutput",
+    "TemplatingNodeResultOutput_Array",
     "TemplatingNodeResultOutput_ChatHistory",
     "TemplatingNodeResultOutput_Error",
+    "TemplatingNodeResultOutput_FunctionCall",
     "TemplatingNodeResultOutput_Json",
     "TemplatingNodeResultOutput_Number",
     "TemplatingNodeResultOutput_SearchResults",

vellum/types/api_node_result.py

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

vellum/types/api_node_result_data.py

@@ -13,9 +13,9 @@ except ImportError:
 
 class ApiNodeResultData(pydantic.BaseModel):
     text_output_id: str
-    text: typing.Optional[str]
+    text: typing.Optional[str] = None
     json_output_id: str
-    json_: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(alias="json")
+    json_: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(alias="json", default=None)
     status_code_output_id: str
     status_code: int
 
@@ -31,4 +31,6 @@ class ApiNodeResultData(pydantic.BaseModel):
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/array_chat_message_content.py

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

vellum/types/array_chat_message_content_item.py

@@ -4,38 +4,39 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .function_call_chat_message_content import FunctionCallChatMessageContent
 from .image_chat_message_content import ImageChatMessageContent
 from .string_chat_message_content import StringChatMessageContent
 
 
 class ArrayChatMessageContentItem_String(StringChatMessageContent):
-    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 ArrayChatMessageContentItem_FunctionCall(FunctionCallChatMessageContent):
-    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 ArrayChatMessageContentItem_Image(ImageChatMessageContent):
-    type: typing_extensions.Literal["IMAGE"]
+    type: typing.Literal["IMAGE"] = "IMAGE"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 ArrayChatMessageContentItem = typing.Union[

vellum/types/array_chat_message_content_item_request.py

@@ -4,38 +4,39 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .function_call_chat_message_content_request import FunctionCallChatMessageContentRequest
 from .image_chat_message_content_request import ImageChatMessageContentRequest
 from .string_chat_message_content_request import StringChatMessageContentRequest
 
 
 class ArrayChatMessageContentItemRequest_String(StringChatMessageContentRequest):
-    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 ArrayChatMessageContentItemRequest_FunctionCall(FunctionCallChatMessageContentRequest):
-    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 ArrayChatMessageContentItemRequest_Image(ImageChatMessageContentRequest):
-    type: typing_extensions.Literal["IMAGE"]
+    type: typing.Literal["IMAGE"] = "IMAGE"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 ArrayChatMessageContentItemRequest = typing.Union[

vellum/types/array_chat_message_content_request.py

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

vellum/types/array_enum.py

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

vellum/types/array_variable_value_item.py

@@ -4,8 +4,6 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .chat_history_variable_value import ChatHistoryVariableValue
 from .error_variable_value import ErrorVariableValue
 from .function_call_variable_value import FunctionCallVariableValue
@@ -17,75 +15,83 @@ from .string_variable_value import StringVariableValue
 
 
 class ArrayVariableValueItem_String(StringVariableValue):
-    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 ArrayVariableValueItem_Number(NumberVariableValue):
-    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 ArrayVariableValueItem_Json(JsonVariableValue):
-    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 ArrayVariableValueItem_ChatHistory(ChatHistoryVariableValue):
-    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 ArrayVariableValueItem_SearchResults(SearchResultsVariableValue):
-    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 ArrayVariableValueItem_Error(ErrorVariableValue):
-    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 ArrayVariableValueItem_FunctionCall(FunctionCallVariableValue):
-    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 ArrayVariableValueItem_Image(ImageVariableValue):
-    type: typing_extensions.Literal["IMAGE"]
+    type: typing.Literal["IMAGE"] = "IMAGE"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 ArrayVariableValueItem = typing.Union[

vellum/types/chat_history_enum.py

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

vellum/types/chat_history_input_request.py

@@ -17,7 +17,11 @@ class ChatHistoryInputRequest(pydantic.BaseModel):
     A user input representing a list of chat messages
     """
 
-    name: str = pydantic.Field(description="The variable's name, as defined in the deployment.")
+    name: str = pydantic.Field()
+    """
+    The variable's name, as defined in the deployment.
+    """
+
     value: typing.List[ChatMessageRequest]
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -31,4 +35,5 @@ class ChatHistoryInputRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/chat_history_variable_value.py

@@ -13,7 +13,7 @@ except ImportError:
 
 
 class ChatHistoryVariableValue(pydantic.BaseModel):
-    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}
@@ -26,4 +26,5 @@ class ChatHistoryVariableValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/chat_message.py

@@ -14,12 +14,13 @@ except ImportError:
 
 
 class ChatMessage(pydantic.BaseModel):
-    text: typing.Optional[str]
+    text: typing.Optional[str] = None
     role: ChatMessageRole
-    content: typing.Optional[ChatMessageContent]
-    source: typing.Optional[str] = pydantic.Field(
-        description="An optional identifier representing who or what generated this message."
-    )
+    content: typing.Optional[ChatMessageContent] = None
+    source: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    An optional identifier representing who or what generated this message.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -32,4 +33,5 @@ class ChatMessage(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}