vellum-ai 0.1.13__py3-none-any.whl → 0.2.1__py3-none-any.whl

vellum/types/__init__.py

@@ -2,9 +2,41 @@
 
 from .api_node_result import ApiNodeResult
 from .api_node_result_data import ApiNodeResultData
+from .array_chat_message_content import ArrayChatMessageContent
+from .array_chat_message_content_item import (
+    ArrayChatMessageContentItem,
+    ArrayChatMessageContentItem_FunctionCall,
+    ArrayChatMessageContentItem_Image,
+    ArrayChatMessageContentItem_String,
+)
+from .array_chat_message_content_item_request import (
+    ArrayChatMessageContentItemRequest,
+    ArrayChatMessageContentItemRequest_FunctionCall,
+    ArrayChatMessageContentItemRequest_Image,
+    ArrayChatMessageContentItemRequest_String,
+)
+from .array_chat_message_content_request import ArrayChatMessageContentRequest
+from .array_enum import ArrayEnum
+from .array_variable_value import ArrayVariableValue
 from .block_type_enum import BlockTypeEnum
+from .chat_history_enum import ChatHistoryEnum
 from .chat_history_input_request import ChatHistoryInputRequest
+from .chat_history_variable_value import ChatHistoryVariableValue
 from .chat_message import ChatMessage
+from .chat_message_content import (
+    ChatMessageContent,
+    ChatMessageContent_Array,
+    ChatMessageContent_FunctionCall,
+    ChatMessageContent_Image,
+    ChatMessageContent_String,
+)
+from .chat_message_content_request import (
+    ChatMessageContentRequest,
+    ChatMessageContentRequest_Array,
+    ChatMessageContentRequest_FunctionCall,
+    ChatMessageContentRequest_Image,
+    ChatMessageContentRequest_String,
+)
 from .chat_message_request import ChatMessageRequest
 from .chat_message_role import ChatMessageRole
 from .code_execution_node_chat_history_result import CodeExecutionNodeChatHistoryResult
@@ -28,14 +60,14 @@ from .conditional_node_result import ConditionalNodeResult
 from .conditional_node_result_data import ConditionalNodeResultData
 from .deployment_provider_payload_response import DeploymentProviderPayloadResponse
 from .deployment_read import DeploymentRead
-from .deployment_status import DeploymentStatus
 from .document_document_to_document_index import DocumentDocumentToDocumentIndex
 from .document_index_read import DocumentIndexRead
-from .document_index_status import DocumentIndexStatus
 from .document_read import DocumentRead
 from .document_status import DocumentStatus
 from .enriched_normalized_completion import EnrichedNormalizedCompletion
+from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
+from .error_enum import ErrorEnum
 from .error_variable_value import ErrorVariableValue
 from .evaluation_params import EvaluationParams
 from .evaluation_params_request import EvaluationParamsRequest
@@ -52,14 +84,27 @@ from .execute_prompt_response import (
     ExecutePromptResponse_Fulfilled,
     ExecutePromptResponse_Rejected,
 )
+from .execute_workflow_error_response import ExecuteWorkflowErrorResponse
+from .execute_workflow_response import ExecuteWorkflowResponse
 from .execute_workflow_stream_error_response import ExecuteWorkflowStreamErrorResponse
+from .execute_workflow_workflow_result_event import (
+    ExecuteWorkflowWorkflowResultEvent,
+    ExecuteWorkflowWorkflowResultEvent_Fulfilled,
+    ExecuteWorkflowWorkflowResultEvent_Rejected,
+)
 from .finish_reason_enum import FinishReasonEnum
 from .fulfilled_enum import FulfilledEnum
 from .fulfilled_execute_prompt_event import FulfilledExecutePromptEvent
 from .fulfilled_execute_prompt_response import FulfilledExecutePromptResponse
+from .fulfilled_execute_workflow_workflow_result_event import FulfilledExecuteWorkflowWorkflowResultEvent
 from .fulfilled_function_call import FulfilledFunctionCall
 from .fulfilled_prompt_execution_meta import FulfilledPromptExecutionMeta
 from .function_call import FunctionCall, FunctionCall_Fulfilled, FunctionCall_Rejected
+from .function_call_chat_message_content import FunctionCallChatMessageContent
+from .function_call_chat_message_content_request import FunctionCallChatMessageContentRequest
+from .function_call_chat_message_content_value import FunctionCallChatMessageContentValue
+from .function_call_chat_message_content_value_request import FunctionCallChatMessageContentValueRequest
+from .function_call_enum import FunctionCallEnum
 from .function_call_variable_value import FunctionCallVariableValue
 from .generate_error_response import GenerateErrorResponse
 from .generate_options_request import GenerateOptionsRequest
@@ -71,10 +116,14 @@ from .generate_result_error import GenerateResultError
 from .generate_stream_response import GenerateStreamResponse
 from .generate_stream_result import GenerateStreamResult
 from .generate_stream_result_data import GenerateStreamResultData
+from .image_chat_message_content import ImageChatMessageContent
+from .image_chat_message_content_request import ImageChatMessageContentRequest
+from .image_enum import ImageEnum
 from .indexing_state_enum import IndexingStateEnum
 from .initiated_enum import InitiatedEnum
 from .initiated_execute_prompt_event import InitiatedExecutePromptEvent
 from .initiated_prompt_execution_meta import InitiatedPromptExecutionMeta
+from .json_enum import JsonEnum
 from .json_input_request import JsonInputRequest
 from .json_variable_value import JsonVariableValue
 from .logical_operator import LogicalOperator
@@ -120,7 +169,11 @@ from .node_input_variable_compiled_value import (
 )
 from .normalized_log_probs import NormalizedLogProbs
 from .normalized_token_log_probs import NormalizedTokenLogProbs
+from .number_enum import NumberEnum
+from .number_variable_value import NumberVariableValue
+from .paginated_slim_deployment_read_list import PaginatedSlimDeploymentReadList
 from .paginated_slim_document_list import PaginatedSlimDocumentList
+from .paginated_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploymentList
 from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .processing_state_enum import ProcessingStateEnum
 from .prompt_deployment_expand_meta_request_request import PromptDeploymentExpandMetaRequestRequest
@@ -161,6 +214,7 @@ from .registered_prompt_sandbox_snapshot import RegisteredPromptSandboxSnapshot
 from .rejected_enum import RejectedEnum
 from .rejected_execute_prompt_event import RejectedExecutePromptEvent
 from .rejected_execute_prompt_response import RejectedExecutePromptResponse
+from .rejected_execute_workflow_workflow_result_event import RejectedExecuteWorkflowWorkflowResultEvent
 from .rejected_function_call import RejectedFunctionCall
 from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
 from .sandbox_metric_input_params import SandboxMetricInputParams
@@ -180,11 +234,18 @@ from .search_result_document import SearchResultDocument
 from .search_result_document_request import SearchResultDocumentRequest
 from .search_result_merging_request import SearchResultMergingRequest
 from .search_result_request import SearchResultRequest
+from .search_results_enum import SearchResultsEnum
+from .search_results_variable_value import SearchResultsVariableValue
 from .search_weights_request import SearchWeightsRequest
+from .slim_deployment_read import SlimDeploymentRead
 from .slim_document import SlimDocument
+from .slim_workflow_deployment import SlimWorkflowDeployment
 from .streaming_enum import StreamingEnum
 from .streaming_execute_prompt_event import StreamingExecutePromptEvent
 from .streaming_prompt_execution_meta import StreamingPromptExecutionMeta
+from .string_chat_message_content import StringChatMessageContent
+from .string_chat_message_content_request import StringChatMessageContentRequest
+from .string_enum import StringEnum
 from .string_input_request import StringInputRequest
 from .string_variable_value import StringVariableValue
 from .submit_completion_actual_request import SubmitCompletionActualRequest
@@ -247,9 +308,22 @@ from .test_case_variable_value import (
 from .test_suite_test_case import TestSuiteTestCase
 from .upload_document_error_response import UploadDocumentErrorResponse
 from .upload_document_response import UploadDocumentResponse
+from .variable_value import (
+    VariableValue,
+    VariableValue_Array,
+    VariableValue_ChatHistory,
+    VariableValue_Error,
+    VariableValue_FunctionCall,
+    VariableValue_Json,
+    VariableValue_Number,
+    VariableValue_SearchResults,
+    VariableValue_String,
+)
 from .vellum_error import VellumError
 from .vellum_error_code_enum import VellumErrorCodeEnum
 from .vellum_error_request import VellumErrorRequest
+from .vellum_image import VellumImage
+from .vellum_image_request import VellumImageRequest
 from .vellum_variable import VellumVariable
 from .vellum_variable_type import VellumVariableType
 from .workflow_event_error import WorkflowEventError
@@ -272,6 +346,27 @@ from .workflow_node_result_data import (
 )
 from .workflow_node_result_event import WorkflowNodeResultEvent
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
+from .workflow_output import (
+    WorkflowOutput,
+    WorkflowOutput_Array,
+    WorkflowOutput_ChatHistory,
+    WorkflowOutput_Error,
+    WorkflowOutput_FunctionCall,
+    WorkflowOutput_Image,
+    WorkflowOutput_Json,
+    WorkflowOutput_Number,
+    WorkflowOutput_SearchResults,
+    WorkflowOutput_String,
+)
+from .workflow_output_array import WorkflowOutputArray
+from .workflow_output_chat_history import WorkflowOutputChatHistory
+from .workflow_output_error import WorkflowOutputError
+from .workflow_output_function_call import WorkflowOutputFunctionCall
+from .workflow_output_image import WorkflowOutputImage
+from .workflow_output_json import WorkflowOutputJson
+from .workflow_output_number import WorkflowOutputNumber
+from .workflow_output_search_results import WorkflowOutputSearchResults
+from .workflow_output_string import WorkflowOutputString
 from .workflow_request_chat_history_input_request import WorkflowRequestChatHistoryInputRequest
 from .workflow_request_input_request import (
     WorkflowRequestInputRequest,
@@ -304,9 +399,33 @@ from .workflow_stream_event import WorkflowStreamEvent, WorkflowStreamEvent_Node
 __all__ = [
     "ApiNodeResult",
     "ApiNodeResultData",
+    "ArrayChatMessageContent",
+    "ArrayChatMessageContentItem",
+    "ArrayChatMessageContentItemRequest",
+    "ArrayChatMessageContentItemRequest_FunctionCall",
+    "ArrayChatMessageContentItemRequest_Image",
+    "ArrayChatMessageContentItemRequest_String",
+    "ArrayChatMessageContentItem_FunctionCall",
+    "ArrayChatMessageContentItem_Image",
+    "ArrayChatMessageContentItem_String",
+    "ArrayChatMessageContentRequest",
+    "ArrayEnum",
+    "ArrayVariableValue",
     "BlockTypeEnum",
+    "ChatHistoryEnum",
     "ChatHistoryInputRequest",
+    "ChatHistoryVariableValue",
     "ChatMessage",
+    "ChatMessageContent",
+    "ChatMessageContentRequest",
+    "ChatMessageContentRequest_Array",
+    "ChatMessageContentRequest_FunctionCall",
+    "ChatMessageContentRequest_Image",
+    "ChatMessageContentRequest_String",
+    "ChatMessageContent_Array",
+    "ChatMessageContent_FunctionCall",
+    "ChatMessageContent_Image",
+    "ChatMessageContent_String",
     "ChatMessageRequest",
     "ChatMessageRole",
     "CodeExecutionNodeChatHistoryResult",
@@ -328,14 +447,14 @@ __all__ = [
     "ConditionalNodeResultData",
     "DeploymentProviderPayloadResponse",
     "DeploymentRead",
-    "DeploymentStatus",
     "DocumentDocumentToDocumentIndex",
     "DocumentIndexRead",
-    "DocumentIndexStatus",
     "DocumentRead",
     "DocumentStatus",
     "EnrichedNormalizedCompletion",
+    "EntityStatus",
     "EnvironmentEnum",
+    "ErrorEnum",
     "ErrorVariableValue",
     "EvaluationParams",
     "EvaluationParamsRequest",
@@ -348,14 +467,25 @@ __all__ = [
     "ExecutePromptResponse",
     "ExecutePromptResponse_Fulfilled",
     "ExecutePromptResponse_Rejected",
+    "ExecuteWorkflowErrorResponse",
+    "ExecuteWorkflowResponse",
     "ExecuteWorkflowStreamErrorResponse",
+    "ExecuteWorkflowWorkflowResultEvent",
+    "ExecuteWorkflowWorkflowResultEvent_Fulfilled",
+    "ExecuteWorkflowWorkflowResultEvent_Rejected",
     "FinishReasonEnum",
     "FulfilledEnum",
     "FulfilledExecutePromptEvent",
     "FulfilledExecutePromptResponse",
+    "FulfilledExecuteWorkflowWorkflowResultEvent",
     "FulfilledFunctionCall",
     "FulfilledPromptExecutionMeta",
     "FunctionCall",
+    "FunctionCallChatMessageContent",
+    "FunctionCallChatMessageContentRequest",
+    "FunctionCallChatMessageContentValue",
+    "FunctionCallChatMessageContentValueRequest",
+    "FunctionCallEnum",
     "FunctionCallVariableValue",
     "FunctionCall_Fulfilled",
     "FunctionCall_Rejected",
@@ -369,10 +499,14 @@ __all__ = [
     "GenerateStreamResponse",
     "GenerateStreamResult",
     "GenerateStreamResultData",
+    "ImageChatMessageContent",
+    "ImageChatMessageContentRequest",
+    "ImageEnum",
     "IndexingStateEnum",
     "InitiatedEnum",
     "InitiatedExecutePromptEvent",
     "InitiatedPromptExecutionMeta",
+    "JsonEnum",
     "JsonInputRequest",
     "JsonVariableValue",
     "LogicalOperator",
@@ -414,7 +548,11 @@ __all__ = [
     "NodeInputVariableCompiledValue_String",
     "NormalizedLogProbs",
     "NormalizedTokenLogProbs",
+    "NumberEnum",
+    "NumberVariableValue",
+    "PaginatedSlimDeploymentReadList",
     "PaginatedSlimDocumentList",
+    "PaginatedSlimWorkflowDeploymentList",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
     "PromptDeploymentExpandMetaRequestRequest",
@@ -451,6 +589,7 @@ __all__ = [
     "RejectedEnum",
     "RejectedExecutePromptEvent",
     "RejectedExecutePromptResponse",
+    "RejectedExecuteWorkflowWorkflowResultEvent",
     "RejectedFunctionCall",
     "RejectedPromptExecutionMeta",
     "SandboxMetricInputParams",
@@ -470,11 +609,18 @@ __all__ = [
     "SearchResultDocumentRequest",
     "SearchResultMergingRequest",
     "SearchResultRequest",
+    "SearchResultsEnum",
+    "SearchResultsVariableValue",
     "SearchWeightsRequest",
+    "SlimDeploymentRead",
     "SlimDocument",
+    "SlimWorkflowDeployment",
     "StreamingEnum",
     "StreamingExecutePromptEvent",
     "StreamingPromptExecutionMeta",
+    "StringChatMessageContent",
+    "StringChatMessageContentRequest",
+    "StringEnum",
     "StringInputRequest",
     "StringVariableValue",
     "SubmitCompletionActualRequest",
@@ -529,9 +675,20 @@ __all__ = [
     "TestSuiteTestCase",
     "UploadDocumentErrorResponse",
     "UploadDocumentResponse",
+    "VariableValue",
+    "VariableValue_Array",
+    "VariableValue_ChatHistory",
+    "VariableValue_Error",
+    "VariableValue_FunctionCall",
+    "VariableValue_Json",
+    "VariableValue_Number",
+    "VariableValue_SearchResults",
+    "VariableValue_String",
     "VellumError",
     "VellumErrorCodeEnum",
     "VellumErrorRequest",
+    "VellumImage",
+    "VellumImageRequest",
     "VellumVariable",
     "VellumVariableType",
     "WorkflowEventError",
@@ -552,6 +709,25 @@ __all__ = [
     "WorkflowNodeResultData_Terminal",
     "WorkflowNodeResultEvent",
     "WorkflowNodeResultEventState",
+    "WorkflowOutput",
+    "WorkflowOutputArray",
+    "WorkflowOutputChatHistory",
+    "WorkflowOutputError",
+    "WorkflowOutputFunctionCall",
+    "WorkflowOutputImage",
+    "WorkflowOutputJson",
+    "WorkflowOutputNumber",
+    "WorkflowOutputSearchResults",
+    "WorkflowOutputString",
+    "WorkflowOutput_Array",
+    "WorkflowOutput_ChatHistory",
+    "WorkflowOutput_Error",
+    "WorkflowOutput_FunctionCall",
+    "WorkflowOutput_Image",
+    "WorkflowOutput_Json",
+    "WorkflowOutput_Number",
+    "WorkflowOutput_SearchResults",
+    "WorkflowOutput_String",
     "WorkflowRequestChatHistoryInputRequest",
     "WorkflowRequestInputRequest",
     "WorkflowRequestInputRequest_ChatHistory",

vellum/types/array_chat_message_content.py

@@ -0,0 +1,33 @@
+# 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_chat_message_content_item import ArrayChatMessageContentItem
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ArrayChatMessageContent(pydantic.BaseModel):
+    """
+    A list of chat message content items.
+    """
+
+    value: typing.List[ArrayChatMessageContentItem]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/array_chat_message_content_item.py

@@ -0,0 +1,43 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ArrayChatMessageContentItem_FunctionCall(FunctionCallChatMessageContent):
+    type: typing_extensions.Literal["FUNCTION_CALL"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ArrayChatMessageContentItem_Image(ImageChatMessageContent):
+    type: typing_extensions.Literal["IMAGE"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+ArrayChatMessageContentItem = typing.Union[
+    ArrayChatMessageContentItem_String, ArrayChatMessageContentItem_FunctionCall, ArrayChatMessageContentItem_Image
+]

vellum/types/array_chat_message_content_item_request.py

@@ -0,0 +1,45 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ArrayChatMessageContentItemRequest_FunctionCall(FunctionCallChatMessageContentRequest):
+    type: typing_extensions.Literal["FUNCTION_CALL"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ArrayChatMessageContentItemRequest_Image(ImageChatMessageContentRequest):
+    type: typing_extensions.Literal["IMAGE"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+ArrayChatMessageContentItemRequest = typing.Union[
+    ArrayChatMessageContentItemRequest_String,
+    ArrayChatMessageContentItemRequest_FunctionCall,
+    ArrayChatMessageContentItemRequest_Image,
+]

vellum/types/array_chat_message_content_request.py

@@ -0,0 +1,33 @@
+# 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_chat_message_content_item_request import ArrayChatMessageContentItemRequest
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ArrayChatMessageContentRequest(pydantic.BaseModel):
+    """
+    A list of chat message content items.
+    """
+
+    value: typing.List[ArrayChatMessageContentItemRequest]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/array_enum.py

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

vellum/types/array_variable_value.py

@@ -0,0 +1,35 @@
+# 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
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ArrayVariableValue(pydantic.BaseModel):
+    value: typing.List[VariableValue]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}
+
+
+from .variable_value import VariableValue  # noqa: E402
+
+ArrayVariableValue.update_forward_refs()

vellum/types/chat_history_enum.py

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

vellum/types/chat_history_input_request.py

@@ -13,6 +13,10 @@ except ImportError:
 
 
 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.")
     value: typing.List[ChatMessageRequest]
 

vellum/types/chat_history_variable_value.py

@@ -0,0 +1,29 @@
+# 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 .chat_message import ChatMessage
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ChatHistoryVariableValue(pydantic.BaseModel):
+    value: typing.Optional[typing.List[ChatMessage]]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/chat_message.py

@@ -4,6 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .chat_message_content import ChatMessageContent
 from .chat_message_role import ChatMessageRole
 
 try:
@@ -13,8 +14,9 @@ except ImportError:
 
 
 class ChatMessage(pydantic.BaseModel):
-    text: str
+    text: typing.Optional[str]
     role: ChatMessageRole
+    content: typing.Optional[ChatMessageContent]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/chat_message_content.py

@@ -0,0 +1,53 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from .array_chat_message_content import ArrayChatMessageContent
+from .function_call_chat_message_content import FunctionCallChatMessageContent
+from .image_chat_message_content import ImageChatMessageContent
+from .string_chat_message_content import StringChatMessageContent
+
+
+class ChatMessageContent_String(StringChatMessageContent):
+    type: typing_extensions.Literal["STRING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ChatMessageContent_FunctionCall(FunctionCallChatMessageContent):
+    type: typing_extensions.Literal["FUNCTION_CALL"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ChatMessageContent_Array(ArrayChatMessageContent):
+    type: typing_extensions.Literal["ARRAY"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ChatMessageContent_Image(ImageChatMessageContent):
+    type: typing_extensions.Literal["IMAGE"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+ChatMessageContent = typing.Union[
+    ChatMessageContent_String, ChatMessageContent_FunctionCall, ChatMessageContent_Array, ChatMessageContent_Image
+]