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

vellum/types/code_execution_node_json_result.py

@@ -9,7 +9,7 @@ from ..core.pydantic_utilities import pydantic_v1
 
 class CodeExecutionNodeJsonResult(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/execution_json_vellum_value.py

@@ -18,7 +18,7 @@ class ExecutionJsonVellumValue(pydantic_v1.BaseModel):
     """
 
     name: 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/json_variable_value.py

@@ -8,7 +8,7 @@ from ..core.pydantic_utilities import pydantic_v1
 
 
 class JsonVariableValue(pydantic_v1.BaseModel):
-    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.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/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_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/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/prompt_node_result_data.py

@@ -10,6 +10,7 @@ from ..core.pydantic_utilities import pydantic_v1
 class PromptNodeResultData(pydantic_v1.BaseModel):
     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/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_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.
     """

vellum/types/test_suite_test_case_upsert_bulk_operation_request.py

@@ -0,0 +1,35 @@
+# 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 .upsert_test_suite_test_case_request import UpsertTestSuiteTestCaseRequest
+
+
+class TestSuiteTestCaseUpsertBulkOperationRequest(pydantic_v1.BaseModel):
+    """
+    A bulk operation that represents the upserting of a Test Case.
+    """
+
+    id: str = pydantic_v1.Field()
+    """
+    An ID representing this specific operation. Can later be used to look up information about the operation's success in the response.
+    """
+
+    data: UpsertTestSuiteTestCaseRequest
+
+    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/upsert_enum.py

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

vellum/types/upsert_test_suite_test_case_request.py

@@ -0,0 +1,49 @@
+# 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 .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
+
+
+class UpsertTestSuiteTestCaseRequest(pydantic_v1.BaseModel):
+    id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    The Vellum-generated ID of an existing Test Case whose data you'd like to replace. If specified and no Test Case exists with this ID, a 404 will be returned.
+    """
+
+    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    An ID external to Vellum that uniquely identifies the Test Case that you'd like to create/update. If there's a match on a Test Case that was previously created with the same external_id, it will be updated. Otherwise, a new Test Case will be created with this value as its external_id. If no external_id is specified, then a new Test Case will always be created.
+    """
+
+    label: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    A human-readable label used to convey the intention of this Test Case
+    """
+
+    input_values: typing.List[NamedTestCaseVariableValueRequest] = pydantic_v1.Field()
+    """
+    Values for each of the Test Case's input variables
+    """
+
+    evaluation_values: typing.List[NamedTestCaseVariableValueRequest] = pydantic_v1.Field()
+    """
+    Values for each of the Test Case's evaluation variables
+    """
+
+    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/workflow_output_json.py

@@ -18,7 +18,7 @@ class WorkflowOutputJson(pydantic_v1.BaseModel):
     The output's name, as defined in the workflow
     """
 
-    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/workflow_request_json_input_request.py

@@ -17,7 +17,7 @@ class WorkflowRequestJsonInputRequest(pydantic_v1.BaseModel):
     The variable's name, as defined in the Workflow.
     """
 
-    value: typing.Dict[str, typing.Any]
+    value: typing.Any
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/workflow_result_event_output_data_json.py

@@ -22,7 +22,7 @@ class WorkflowResultEventOutputDataJson(pydantic_v1.BaseModel):
     The newly output string value. Only relevant for string outputs with a state of STREAMING.
     """
 
-    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_ai-0.6.9.dist-info → vellum_ai-0.7.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.6.9
+Version: 0.7.1
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Programming Language :: Python :: 3