vellum-ai 0.6.9__py3-none-any.whl → 0.7.2__py3-none-any.whl

vellum/types/json_vellum_value.py

@@ -12,7 +12,7 @@ class JsonVellumValue(pydantic_v1.BaseModel):
     A value representing a JSON object.
     """
 
-    value: typing.Optional[typing.Dict[str, typing.Any]] = None
+    value: typing.Any
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/json_vellum_value_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
+from ..core.pydantic_utilities import pydantic_v1
+
+
+class JsonVellumValueRequest(pydantic_v1.BaseModel):
+    """
+    A value representing a JSON object.
+    """
+
+    value: 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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/merge_node_result.py

@@ -5,6 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
+from .merge_node_result_data import MergeNodeResultData
 
 
 class MergeNodeResult(pydantic_v1.BaseModel):
@@ -12,6 +13,8 @@ class MergeNodeResult(pydantic_v1.BaseModel):
     A Node Result Event emitted from a Merge Node.
     """
 
+    data: MergeNodeResultData
+
     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)

vellum/types/merge_node_result_data.py

@@ -0,0 +1,25 @@
+# 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 ..core.pydantic_utilities import pydantic_v1
+
+
+class MergeNodeResultData(pydantic_v1.BaseModel):
+    paused_node_data: 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}
+        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_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_array_variable_value.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 ..core.pydantic_utilities import pydantic_v1
+from .array_vellum_value_item import ArrayVellumValueItem
+
+
+class NamedTestCaseArrayVariableValue(pydantic_v1.BaseModel):
+    """
+    Named Test Case value that is of type ARRAY
+    """
+
+    value: typing.Optional[typing.List[ArrayVellumValueItem]] = None
+    name: 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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_array_variable_value_request.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 ..core.pydantic_utilities import pydantic_v1
+from .array_vellum_value_item_request import ArrayVellumValueItemRequest
+
+
+class NamedTestCaseArrayVariableValueRequest(pydantic_v1.BaseModel):
+    """
+    Named Test Case value that is of type ARRAY
+    """
+
+    value: typing.Optional[typing.List[ArrayVellumValueItemRequest]] = None
+    name: 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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_json_variable_value.py

@@ -12,7 +12,7 @@ class NamedTestCaseJsonVariableValue(pydantic_v1.BaseModel):
     Named Test Case value that is of type JSON
     """
 
-    value: typing.Optional[typing.Dict[str, typing.Any]] = None
+    value: typing.Any
     name: str
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/named_test_case_json_variable_value_request.py

@@ -12,7 +12,7 @@ class NamedTestCaseJsonVariableValueRequest(pydantic_v1.BaseModel):
     Named Test Case value that is of type JSON
     """
 
-    value: typing.Optional[typing.Dict[str, typing.Any]] = None
+    value: typing.Any
     name: str
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/named_test_case_variable_value.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import typing
 
+from .named_test_case_array_variable_value import NamedTestCaseArrayVariableValue
 from .named_test_case_chat_history_variable_value import NamedTestCaseChatHistoryVariableValue
 from .named_test_case_error_variable_value import NamedTestCaseErrorVariableValue
 from .named_test_case_function_call_variable_value import NamedTestCaseFunctionCallVariableValue
@@ -83,6 +84,16 @@ class NamedTestCaseVariableValue_FunctionCall(NamedTestCaseFunctionCallVariableV
         populate_by_name = True
 
 
+class NamedTestCaseVariableValue_Array(NamedTestCaseArrayVariableValue):
+    type: typing.Literal["ARRAY"] = "ARRAY"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
 NamedTestCaseVariableValue = typing.Union[
     NamedTestCaseVariableValue_String,
     NamedTestCaseVariableValue_Number,
@@ -91,4 +102,5 @@ NamedTestCaseVariableValue = typing.Union[
     NamedTestCaseVariableValue_SearchResults,
     NamedTestCaseVariableValue_Error,
     NamedTestCaseVariableValue_FunctionCall,
+    NamedTestCaseVariableValue_Array,
 ]

vellum/types/named_test_case_variable_value_request.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import typing
 
+from .named_test_case_array_variable_value_request import NamedTestCaseArrayVariableValueRequest
 from .named_test_case_chat_history_variable_value_request import NamedTestCaseChatHistoryVariableValueRequest
 from .named_test_case_error_variable_value_request import NamedTestCaseErrorVariableValueRequest
 from .named_test_case_function_call_variable_value_request import NamedTestCaseFunctionCallVariableValueRequest
@@ -83,6 +84,16 @@ class NamedTestCaseVariableValueRequest_FunctionCall(NamedTestCaseFunctionCallVa
         populate_by_name = True
 
 
+class NamedTestCaseVariableValueRequest_Array(NamedTestCaseArrayVariableValueRequest):
+    type: typing.Literal["ARRAY"] = "ARRAY"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
 NamedTestCaseVariableValueRequest = typing.Union[
     NamedTestCaseVariableValueRequest_String,
     NamedTestCaseVariableValueRequest_Number,
@@ -91,4 +102,5 @@ NamedTestCaseVariableValueRequest = typing.Union[
     NamedTestCaseVariableValueRequest_SearchResults,
     NamedTestCaseVariableValueRequest_Error,
     NamedTestCaseVariableValueRequest_FunctionCall,
+    NamedTestCaseVariableValueRequest_Array,
 ]

vellum/types/node_input_compiled_json_value.py

@@ -10,7 +10,7 @@ from ..core.pydantic_utilities import pydantic_v1
 class NodeInputCompiledJsonValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
-    value: typing.Optional[typing.Dict[str, typing.Any]] = None
+    value: typing.Any
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/node_output_compiled_json_value.py

@@ -13,7 +13,7 @@ class NodeOutputCompiledJsonValue(pydantic_v1.BaseModel):
     An output returned by a node that is of type JSON.
     """
 
-    value: typing.Optional[typing.Dict[str, typing.Any]] = None
+    value: typing.Any
     node_output_id: str
     state: typing.Optional[WorkflowNodeResultEventState] = None
 

vellum/types/number_vellum_value_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
+from ..core.pydantic_utilities import pydantic_v1
+
+
+class NumberVellumValueRequest(pydantic_v1.BaseModel):
+    """
+    A value representing a number.
+    """
+
+    value: typing.Optional[float] = 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_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/prompt_deployment_expand_meta_request_request.py

@@ -10,32 +10,32 @@ from ..core.pydantic_utilities import pydantic_v1
 class PromptDeploymentExpandMetaRequestRequest(pydantic_v1.BaseModel):
     model_name: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
-    If enabled, the response will include the model identifier representing the ML Model invoked by the Prompt Deployment.
+    If enabled, the response will include the model identifier representing the ML Model invoked by the Prompt.
     """
 
-    latency: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    usage: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
-    If enabled, the response will include the time in nanoseconds it took to execute the Prompt Deployment.
+    If enabled, the response will include model host usage tracking. This may increase latency for some model hosts.
     """
 
-    deployment_release_tag: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    finish_reason: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
-    If enabled, the response will include the release tag of the Prompt Deployment.
+    If enabled, the response will include the reason provided by the model for why the execution finished.
     """
 
-    prompt_version_id: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    latency: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
-    If enabled, the response will include the ID of the Prompt Version backing the deployment.
+    If enabled, the response will include the time in nanoseconds it took to execute the Prompt Deployment.
     """
 
-    finish_reason: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    deployment_release_tag: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
-    If enabled, the response will include the reason provided by the model for why the execution finished.
+    If enabled, the response will include the release tag of the Prompt Deployment.
     """
 
-    usage: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    prompt_version_id: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
-    If enabled, the response will include model host usage tracking. This may increase latency for some model hosts.
+    If enabled, the response will include the ID of the Prompt Version backing the deployment.
     """
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/prompt_execution_meta.py

@@ -14,12 +14,12 @@ class PromptExecutionMeta(pydantic_v1.BaseModel):
     The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
     """
 
-    usage: typing.Optional[MlModelUsage] = None
     model_name: typing.Optional[str] = None
     latency: typing.Optional[int] = None
     deployment_release_tag: typing.Optional[str] = None
     prompt_version_id: typing.Optional[str] = None
     finish_reason: typing.Optional[FinishReasonEnum] = None
+    usage: typing.Optional[MlModelUsage] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/prompt_node_execution_meta.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 ..core.pydantic_utilities import pydantic_v1
+from .ml_model_usage import MlModelUsage
+
+
+class PromptNodeExecutionMeta(pydantic_v1.BaseModel):
+    """
+    The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
+    """
+
+    usage: typing.Optional[MlModelUsage] = 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_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/prompt_node_result_data.py

@@ -5,11 +5,14 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
+from .prompt_node_execution_meta import PromptNodeExecutionMeta
 
 
 class PromptNodeResultData(pydantic_v1.BaseModel):
+    execution_meta: typing.Optional[PromptNodeExecutionMeta] = None
     output_id: str
     array_output_id: typing.Optional[str] = None
+    execution_id: typing.Optional[str] = None
     text: typing.Optional[str] = None
     delta: typing.Optional[str] = None
 

vellum/types/string_vellum_value_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
+from ..core.pydantic_utilities import pydantic_v1
+
+
+class StringVellumValueRequest(pydantic_v1.BaseModel):
+    """
+    A value representing a string.
+    """
+
+    value: typing.Optional[str] = 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_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/templating_node_json_result.py

@@ -9,7 +9,7 @@ from ..core.pydantic_utilities import pydantic_v1
 
 class TemplatingNodeJsonResult(pydantic_v1.BaseModel):
     id: str
-    value: typing.Optional[typing.Dict[str, typing.Any]] = None
+    value: typing.Any
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/terminal_node_json_result.py

@@ -14,7 +14,7 @@ class TerminalNodeJsonResult(pydantic_v1.BaseModel):
     The unique name given to the terminal node that produced this output.
     """
 
-    value: typing.Optional[typing.Dict[str, typing.Any]] = None
+    value: typing.Any
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/test_case_array_variable_value.py

@@ -0,0 +1,32 @@
+# 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 ..core.pydantic_utilities import pydantic_v1
+from .array_vellum_value_item import ArrayVellumValueItem
+
+
+class TestCaseArrayVariableValue(pydantic_v1.BaseModel):
+    """
+    An Array value for a variable in a Test Case.
+    """
+
+    variable_id: str
+    name: str
+    value: typing.Optional[typing.List[ArrayVellumValueItem]] = 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_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_case_variable_value.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import typing
 
+from .test_case_array_variable_value import TestCaseArrayVariableValue
 from .test_case_chat_history_variable_value import TestCaseChatHistoryVariableValue
 from .test_case_error_variable_value import TestCaseErrorVariableValue
 from .test_case_function_call_variable_value import TestCaseFunctionCallVariableValue
@@ -83,6 +84,16 @@ class TestCaseVariableValue_FunctionCall(TestCaseFunctionCallVariableValue):
         populate_by_name = True
 
 
+class TestCaseVariableValue_Array(TestCaseArrayVariableValue):
+    type: typing.Literal["ARRAY"] = "ARRAY"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
 TestCaseVariableValue = typing.Union[
     TestCaseVariableValue_String,
     TestCaseVariableValue_Number,
@@ -91,4 +102,5 @@ TestCaseVariableValue = typing.Union[
     TestCaseVariableValue_SearchResults,
     TestCaseVariableValue_Error,
     TestCaseVariableValue_FunctionCall,
+    TestCaseVariableValue_Array,
 ]

vellum/types/test_suite_run_execution_array_output.py

@@ -0,0 +1,32 @@
+# 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 ..core.pydantic_utilities import pydantic_v1
+from .array_vellum_value_item import ArrayVellumValueItem
+
+
+class TestSuiteRunExecutionArrayOutput(pydantic_v1.BaseModel):
+    """
+    Execution output of an entity evaluated during a Test Suite Run that is of type ARRAY
+    """
+
+    name: str
+    value: typing.Optional[typing.List[ArrayVellumValueItem]] = None
+    output_variable_id: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_execution_json_output.py

@@ -13,7 +13,7 @@ class TestSuiteRunExecutionJsonOutput(pydantic_v1.BaseModel):
     """
 
     name: str
-    value: typing.Optional[typing.Dict[str, typing.Any]] = None
+    value: typing.Any
     output_variable_id: str
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/test_suite_run_execution_output.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import typing
 
+from .test_suite_run_execution_array_output import TestSuiteRunExecutionArrayOutput
 from .test_suite_run_execution_chat_history_output import TestSuiteRunExecutionChatHistoryOutput
 from .test_suite_run_execution_error_output import TestSuiteRunExecutionErrorOutput
 from .test_suite_run_execution_function_call_output import TestSuiteRunExecutionFunctionCallOutput
@@ -83,6 +84,16 @@ class TestSuiteRunExecutionOutput_FunctionCall(TestSuiteRunExecutionFunctionCall
         populate_by_name = True
 
 
+class TestSuiteRunExecutionOutput_Array(TestSuiteRunExecutionArrayOutput):
+    type: typing.Literal["ARRAY"] = "ARRAY"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
 TestSuiteRunExecutionOutput = typing.Union[
     TestSuiteRunExecutionOutput_String,
     TestSuiteRunExecutionOutput_Number,
@@ -91,4 +102,5 @@ TestSuiteRunExecutionOutput = typing.Union[
     TestSuiteRunExecutionOutput_SearchResults,
     TestSuiteRunExecutionOutput_Error,
     TestSuiteRunExecutionOutput_FunctionCall,
+    TestSuiteRunExecutionOutput_Array,
 ]

vellum/types/test_suite_run_metric_number_output.py

@@ -12,7 +12,7 @@ class TestSuiteRunMetricNumberOutput(pydantic_v1.BaseModel):
     Output for a test suite run metric that is of type NUMBER
     """
 
-    value: float
+    value: typing.Optional[float] = None
     name: str
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/test_suite_run_metric_string_output.py

@@ -12,7 +12,7 @@ class TestSuiteRunMetricStringOutput(pydantic_v1.BaseModel):
     Output for a test suite run metric that is of type STRING
     """
 
-    value: str
+    value: typing.Optional[str] = None
     name: str
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/test_suite_test_case_bulk_operation_request.py

@@ -7,6 +7,7 @@ import typing
 from .test_suite_test_case_create_bulk_operation_request import TestSuiteTestCaseCreateBulkOperationRequest
 from .test_suite_test_case_delete_bulk_operation_request import TestSuiteTestCaseDeleteBulkOperationRequest
 from .test_suite_test_case_replace_bulk_operation_request import TestSuiteTestCaseReplaceBulkOperationRequest
+from .test_suite_test_case_upsert_bulk_operation_request import TestSuiteTestCaseUpsertBulkOperationRequest
 
 
 class TestSuiteTestCaseBulkOperationRequest_Create(TestSuiteTestCaseCreateBulkOperationRequest):
@@ -29,6 +30,16 @@ class TestSuiteTestCaseBulkOperationRequest_Replace(TestSuiteTestCaseReplaceBulk
         populate_by_name = True
 
 
+class TestSuiteTestCaseBulkOperationRequest_Upsert(TestSuiteTestCaseUpsertBulkOperationRequest):
+    type: typing.Literal["UPSERT"] = "UPSERT"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
 class TestSuiteTestCaseBulkOperationRequest_Delete(TestSuiteTestCaseDeleteBulkOperationRequest):
     type: typing.Literal["DELETE"] = "DELETE"
 
@@ -42,5 +53,6 @@ class TestSuiteTestCaseBulkOperationRequest_Delete(TestSuiteTestCaseDeleteBulkOp
 TestSuiteTestCaseBulkOperationRequest = typing.Union[
     TestSuiteTestCaseBulkOperationRequest_Create,
     TestSuiteTestCaseBulkOperationRequest_Replace,
+    TestSuiteTestCaseBulkOperationRequest_Upsert,
     TestSuiteTestCaseBulkOperationRequest_Delete,
 ]

vellum/types/test_suite_test_case_rejected_bulk_result.py

@@ -12,7 +12,7 @@ class TestSuiteTestCaseRejectedBulkResult(pydantic_v1.BaseModel):
     The result of a bulk operation that failed to operate on a Test Case.
     """
 
-    id: str = pydantic_v1.Field()
+    id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     An ID that maps back to one of the initially supplied operations. Can be used to determine the result of a given operation.
     """