vellum-ai 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

vellum/types/__init__.py

@@ -3,6 +3,7 @@
 from .api_node_result import ApiNodeResult
 from .api_node_result_data import ApiNodeResultData
 from .block_type_enum import BlockTypeEnum
+from .chat_history_input_request import ChatHistoryInputRequest
 from .chat_message import ChatMessage
 from .chat_message_request import ChatMessageRequest
 from .chat_message_role import ChatMessageRole
@@ -17,8 +18,16 @@ from .document_read import DocumentRead
 from .document_status import DocumentStatus
 from .enriched_normalized_completion import EnrichedNormalizedCompletion
 from .environment_enum import EnvironmentEnum
+from .error_execute_prompt_response import ErrorExecutePromptResponse
 from .evaluation_params import EvaluationParams
 from .evaluation_params_request import EvaluationParamsRequest
+from .execute_prompt_api_error_response import ExecutePromptApiErrorResponse
+from .execute_prompt_response import (
+    ExecutePromptResponse,
+    ExecutePromptResponse_Error,
+    ExecutePromptResponse_Json,
+    ExecutePromptResponse_String,
+)
 from .execute_workflow_stream_error_response import ExecuteWorkflowStreamErrorResponse
 from .finish_reason_enum import FinishReasonEnum
 from .generate_error_response import GenerateErrorResponse
@@ -32,6 +41,8 @@ from .generate_stream_response import GenerateStreamResponse
 from .generate_stream_result import GenerateStreamResult
 from .generate_stream_result_data import GenerateStreamResultData
 from .indexing_state_enum import IndexingStateEnum
+from .json_execute_prompt_response import JsonExecutePromptResponse
+from .json_input_request import JsonInputRequest
 from .logical_operator import LogicalOperator
 from .logprobs_enum import LogprobsEnum
 from .metadata_filter_config_request import MetadataFilterConfigRequest
@@ -63,6 +74,12 @@ from .normalized_token_log_probs import NormalizedTokenLogProbs
 from .paginated_slim_document_list import PaginatedSlimDocumentList
 from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .processing_state_enum import ProcessingStateEnum
+from .prompt_deployment_input_request import (
+    PromptDeploymentInputRequest,
+    PromptDeploymentInputRequest_ChatHistory,
+    PromptDeploymentInputRequest_Json,
+    PromptDeploymentInputRequest_String,
+)
 from .prompt_node_result import PromptNodeResult
 from .prompt_node_result_data import PromptNodeResultData
 from .prompt_template_block import PromptTemplateBlock
@@ -96,9 +113,13 @@ from .search_request_options_request import SearchRequestOptionsRequest
 from .search_response import SearchResponse
 from .search_result import SearchResult
 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_weights_request import SearchWeightsRequest
 from .slim_document import SlimDocument
+from .string_execute_prompt_response import StringExecutePromptResponse
+from .string_input_request import StringInputRequest
 from .submit_completion_actual_request import SubmitCompletionActualRequest
 from .submit_completion_actuals_error_response import SubmitCompletionActualsErrorResponse
 from .submit_workflow_execution_actual_request import (
@@ -141,11 +162,42 @@ from .terminal_node_result_output import (
 )
 from .terminal_node_search_results_result import TerminalNodeSearchResultsResult
 from .terminal_node_string_result import TerminalNodeStringResult
+from .test_case_chat_history_variable_value import TestCaseChatHistoryVariableValue
+from .test_case_chat_history_variable_value_request import TestCaseChatHistoryVariableValueRequest
+from .test_case_error_variable_value import TestCaseErrorVariableValue
+from .test_case_error_variable_value_request import TestCaseErrorVariableValueRequest
+from .test_case_json_variable_value import TestCaseJsonVariableValue
+from .test_case_json_variable_value_request import TestCaseJsonVariableValueRequest
+from .test_case_number_variable_value import TestCaseNumberVariableValue
+from .test_case_number_variable_value_request import TestCaseNumberVariableValueRequest
+from .test_case_search_results_variable_value import TestCaseSearchResultsVariableValue
+from .test_case_search_results_variable_value_request import TestCaseSearchResultsVariableValueRequest
+from .test_case_string_variable_value import TestCaseStringVariableValue
+from .test_case_string_variable_value_request import TestCaseStringVariableValueRequest
+from .test_case_variable_value import (
+    TestCaseVariableValue,
+    TestCaseVariableValue_ChatHistory,
+    TestCaseVariableValue_Error,
+    TestCaseVariableValue_Json,
+    TestCaseVariableValue_Number,
+    TestCaseVariableValue_SearchResults,
+    TestCaseVariableValue_String,
+)
+from .test_case_variable_value_request import (
+    TestCaseVariableValueRequest,
+    TestCaseVariableValueRequest_ChatHistory,
+    TestCaseVariableValueRequest_Error,
+    TestCaseVariableValueRequest_Json,
+    TestCaseVariableValueRequest_Number,
+    TestCaseVariableValueRequest_SearchResults,
+    TestCaseVariableValueRequest_String,
+)
 from .test_suite_test_case import TestSuiteTestCase
 from .upload_document_error_response import UploadDocumentErrorResponse
 from .upload_document_response import UploadDocumentResponse
 from .vellum_error import VellumError
 from .vellum_error_code_enum import VellumErrorCodeEnum
+from .vellum_error_request import VellumErrorRequest
 from .vellum_variable import VellumVariable
 from .vellum_variable_type import VellumVariableType
 from .workflow_event_error import WorkflowEventError
@@ -198,6 +250,7 @@ __all__ = [
     "ApiNodeResult",
     "ApiNodeResultData",
     "BlockTypeEnum",
+    "ChatHistoryInputRequest",
     "ChatMessage",
     "ChatMessageRequest",
     "ChatMessageRole",
@@ -212,8 +265,14 @@ __all__ = [
     "DocumentStatus",
     "EnrichedNormalizedCompletion",
     "EnvironmentEnum",
+    "ErrorExecutePromptResponse",
     "EvaluationParams",
     "EvaluationParamsRequest",
+    "ExecutePromptApiErrorResponse",
+    "ExecutePromptResponse",
+    "ExecutePromptResponse_Error",
+    "ExecutePromptResponse_Json",
+    "ExecutePromptResponse_String",
     "ExecuteWorkflowStreamErrorResponse",
     "FinishReasonEnum",
     "GenerateErrorResponse",
@@ -227,6 +286,8 @@ __all__ = [
     "GenerateStreamResult",
     "GenerateStreamResultData",
     "IndexingStateEnum",
+    "JsonExecutePromptResponse",
+    "JsonInputRequest",
     "LogicalOperator",
     "LogprobsEnum",
     "MetadataFilterConfigRequest",
@@ -256,6 +317,10 @@ __all__ = [
     "PaginatedSlimDocumentList",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
+    "PromptDeploymentInputRequest",
+    "PromptDeploymentInputRequest_ChatHistory",
+    "PromptDeploymentInputRequest_Json",
+    "PromptDeploymentInputRequest_String",
     "PromptNodeResult",
     "PromptNodeResultData",
     "PromptTemplateBlock",
@@ -289,9 +354,13 @@ __all__ = [
     "SearchResponse",
     "SearchResult",
     "SearchResultDocument",
+    "SearchResultDocumentRequest",
     "SearchResultMergingRequest",
+    "SearchResultRequest",
     "SearchWeightsRequest",
     "SlimDocument",
+    "StringExecutePromptResponse",
+    "StringInputRequest",
     "SubmitCompletionActualRequest",
     "SubmitCompletionActualsErrorResponse",
     "SubmitWorkflowExecutionActualRequest",
@@ -328,11 +397,38 @@ __all__ = [
     "TerminalNodeResultOutput_String",
     "TerminalNodeSearchResultsResult",
     "TerminalNodeStringResult",
+    "TestCaseChatHistoryVariableValue",
+    "TestCaseChatHistoryVariableValueRequest",
+    "TestCaseErrorVariableValue",
+    "TestCaseErrorVariableValueRequest",
+    "TestCaseJsonVariableValue",
+    "TestCaseJsonVariableValueRequest",
+    "TestCaseNumberVariableValue",
+    "TestCaseNumberVariableValueRequest",
+    "TestCaseSearchResultsVariableValue",
+    "TestCaseSearchResultsVariableValueRequest",
+    "TestCaseStringVariableValue",
+    "TestCaseStringVariableValueRequest",
+    "TestCaseVariableValue",
+    "TestCaseVariableValueRequest",
+    "TestCaseVariableValueRequest_ChatHistory",
+    "TestCaseVariableValueRequest_Error",
+    "TestCaseVariableValueRequest_Json",
+    "TestCaseVariableValueRequest_Number",
+    "TestCaseVariableValueRequest_SearchResults",
+    "TestCaseVariableValueRequest_String",
+    "TestCaseVariableValue_ChatHistory",
+    "TestCaseVariableValue_Error",
+    "TestCaseVariableValue_Json",
+    "TestCaseVariableValue_Number",
+    "TestCaseVariableValue_SearchResults",
+    "TestCaseVariableValue_String",
     "TestSuiteTestCase",
     "UploadDocumentErrorResponse",
     "UploadDocumentResponse",
     "VellumError",
     "VellumErrorCodeEnum",
+    "VellumErrorRequest",
     "VellumVariable",
     "VellumVariableType",
     "WorkflowEventError",

vellum/types/chat_history_input_request.py

@@ -0,0 +1,30 @@
+# 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_request import ChatMessageRequest
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ChatHistoryInputRequest(pydantic.BaseModel):
+    name: str = pydantic.Field(description="The variable's name, as defined in the deployment.")
+    value: typing.List[ChatMessageRequest]
+
+    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/error_execute_prompt_response.py

@@ -0,0 +1,30 @@
+# 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 .vellum_error import VellumError
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ErrorExecutePromptResponse(pydantic.BaseModel):
+    value: VellumError
+    execution_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: 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/execute_prompt_api_error_response.py

@@ -0,0 +1,28 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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 ExecutePromptApiErrorResponse(pydantic.BaseModel):
+    detail: str = pydantic.Field(description="Details about why the request failed.")
+
+    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/execute_prompt_response.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 .error_execute_prompt_response import ErrorExecutePromptResponse
+from .json_execute_prompt_response import JsonExecutePromptResponse
+from .string_execute_prompt_response import StringExecutePromptResponse
+
+
+class ExecutePromptResponse_Error(ErrorExecutePromptResponse):
+    type: typing_extensions.Literal["ERROR"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ExecutePromptResponse_Json(JsonExecutePromptResponse):
+    type: typing_extensions.Literal["JSON"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ExecutePromptResponse_String(StringExecutePromptResponse):
+    type: typing_extensions.Literal["STRING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+ExecutePromptResponse = typing.Union[
+    ExecutePromptResponse_Error, ExecutePromptResponse_Json, ExecutePromptResponse_String
+]

vellum/types/json_execute_prompt_response.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
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class JsonExecutePromptResponse(pydantic.BaseModel):
+    value: typing.Dict[str, typing.Any]
+    execution_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: 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/json_input_request.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
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class JsonInputRequest(pydantic.BaseModel):
+    name: str = pydantic.Field(description="The variable's name, as defined in the deployment.")
+    value: typing.Dict[str, typing.Any]
+
+    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/prompt_deployment_input_request.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 .chat_history_input_request import ChatHistoryInputRequest
+from .json_input_request import JsonInputRequest
+from .string_input_request import StringInputRequest
+
+
+class PromptDeploymentInputRequest_String(StringInputRequest):
+    type: typing_extensions.Literal["STRING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class PromptDeploymentInputRequest_Json(JsonInputRequest):
+    type: typing_extensions.Literal["JSON"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class PromptDeploymentInputRequest_ChatHistory(ChatHistoryInputRequest):
+    type: typing_extensions.Literal["CHAT_HISTORY"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+PromptDeploymentInputRequest = typing.Union[
+    PromptDeploymentInputRequest_String, PromptDeploymentInputRequest_Json, PromptDeploymentInputRequest_ChatHistory
+]

vellum/types/register_prompt_model_parameters_request.py

@@ -16,7 +16,7 @@ class RegisterPromptModelParametersRequest(pydantic.BaseModel):
     max_tokens: int
     stop: typing.Optional[typing.List[str]]
     top_p: float
-    top_k: typing.Optional[float]
+    top_k: typing.Optional[int]
     frequency_penalty: float
     presence_penalty: float
     logit_bias: typing.Optional[typing.Dict[str, typing.Optional[float]]]

vellum/types/search_result_document_request.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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 SearchResultDocumentRequest(pydantic.BaseModel):
+    label: str = pydantic.Field(description="The human-readable name for the document.")
+    external_id: typing.Optional[str] = pydantic.Field(
+        description="The unique ID of the document as represented in an external system and specified when it was originally uploaded."
+    )
+    metadata: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(
+        description="A previously supplied JSON object containing metadata that can be filtered on when searching."
+    )
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: 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/search_result_request.py

@@ -0,0 +1,34 @@
+# 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 .search_result_document_request import SearchResultDocumentRequest
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class SearchResultRequest(pydantic.BaseModel):
+    text: str = pydantic.Field(description="The text of the chunk that matched the search query.")
+    score: float = pydantic.Field(description="A score representing how well the chunk matches the search query.")
+    keywords: typing.List[str]
+    document: SearchResultDocumentRequest = pydantic.Field(
+        description="The document that contains the chunk that matched the search query."
+    )
+
+    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/string_execute_prompt_response.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
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class StringExecutePromptResponse(pydantic.BaseModel):
+    value: str
+    execution_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: 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/string_input_request.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
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class StringInputRequest(pydantic.BaseModel):
+    name: str = pydantic.Field(description="The variable's name, as defined in the deployment.")
+    value: 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: 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/test_case_chat_history_variable_value.py

@@ -0,0 +1,30 @@
+# 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 TestCaseChatHistoryVariableValue(pydantic.BaseModel):
+    variable_id: str
+    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/test_case_chat_history_variable_value_request.py

@@ -0,0 +1,30 @@
+# 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_request import ChatMessageRequest
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class TestCaseChatHistoryVariableValueRequest(pydantic.BaseModel):
+    variable_id: str
+    value: typing.Optional[typing.List[ChatMessageRequest]]
+
+    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/test_case_error_variable_value.py

@@ -0,0 +1,30 @@
+# 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 .vellum_error import VellumError
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class TestCaseErrorVariableValue(pydantic.BaseModel):
+    variable_id: str
+    value: typing.Optional[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: 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}