vellum-ai 0.6.6__tar.gz → 0.6.7__tar.gz

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/PKG-INFO

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

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "0.6.6"
+version = "0.6.7"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/__init__.py

@@ -35,8 +35,6 @@ from .types import (
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request,
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1,
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
-    BulkCreateTestSuiteTestCaseDataRequest,
-    BulkReplaceTestSuiteTestCaseDataRequest,
     ChatHistoryEnum,
     ChatHistoryInputRequest,
     ChatMessage,
@@ -74,10 +72,12 @@ from .types import (
     ConditionalNodeResult,
     ConditionalNodeResultData,
     CreateEnum,
+    CreateTestSuiteTestCaseRequest,
     CreatedEnum,
     DeleteEnum,
     DeletedEnum,
     DeploymentProviderPayloadResponse,
+    DeploymentProviderPayloadResponsePayload,
     DeploymentRead,
     DeploymentReleaseTagDeploymentHistoryItem,
     DeploymentReleaseTagRead,
@@ -199,6 +199,9 @@ from .types import (
     JsonVellumValue,
     LogicalOperator,
     LogprobsEnum,
+    MapEnum,
+    MapNodeResult,
+    MapNodeResultData,
     MergeEnum,
     MergeNodeResult,
     MetadataFilterConfigRequest,
@@ -327,6 +330,7 @@ from .types import (
     RejectedWorkflowNodeResultEvent,
     ReleaseTagSource,
     ReplaceEnum,
+    ReplaceTestSuiteTestCaseRequest,
     ReplacedEnum,
     SandboxScenario,
     ScenarioInput,
@@ -540,6 +544,7 @@ from .types import (
     WorkflowNodeResultData_Api,
     WorkflowNodeResultData_CodeExecution,
     WorkflowNodeResultData_Conditional,
+    WorkflowNodeResultData_Map,
     WorkflowNodeResultData_Merge,
     WorkflowNodeResultData_Metric,
     WorkflowNodeResultData_Prompt,
@@ -659,8 +664,6 @@ __all__ = [
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
-    "BulkCreateTestSuiteTestCaseDataRequest",
-    "BulkReplaceTestSuiteTestCaseDataRequest",
     "ChatHistoryEnum",
     "ChatHistoryInputRequest",
     "ChatMessage",
@@ -698,10 +701,12 @@ __all__ = [
     "ConditionalNodeResult",
     "ConditionalNodeResultData",
     "CreateEnum",
+    "CreateTestSuiteTestCaseRequest",
     "CreatedEnum",
     "DeleteEnum",
     "DeletedEnum",
     "DeploymentProviderPayloadResponse",
+    "DeploymentProviderPayloadResponsePayload",
     "DeploymentRead",
     "DeploymentReleaseTagDeploymentHistoryItem",
     "DeploymentReleaseTagRead",
@@ -827,6 +832,9 @@ __all__ = [
     "JsonVellumValue",
     "LogicalOperator",
     "LogprobsEnum",
+    "MapEnum",
+    "MapNodeResult",
+    "MapNodeResultData",
     "MergeEnum",
     "MergeNodeResult",
     "MetadataFilterConfigRequest",
@@ -956,6 +964,7 @@ __all__ = [
     "RejectedWorkflowNodeResultEvent",
     "ReleaseTagSource",
     "ReplaceEnum",
+    "ReplaceTestSuiteTestCaseRequest",
     "ReplacedEnum",
     "SandboxScenario",
     "ScenarioInput",
@@ -1171,6 +1180,7 @@ __all__ = [
     "WorkflowNodeResultData_Api",
     "WorkflowNodeResultData_CodeExecution",
     "WorkflowNodeResultData_Conditional",
+    "WorkflowNodeResultData_Map",
     "WorkflowNodeResultData_Merge",
     "WorkflowNodeResultData_Metric",
     "WorkflowNodeResultData_Prompt",

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.6.6",
+            "X-Fern-SDK-Version": "0.6.7",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/resources/test_suites/client.py

@@ -100,6 +100,7 @@ class TestSuitesClient:
         id: str,
         *,
         upsert_test_suite_test_case_request_id: typing.Optional[str] = OMIT,
+        external_id: typing.Optional[str] = OMIT,
         label: typing.Optional[str] = OMIT,
         input_values: typing.Sequence[NamedTestCaseVariableValueRequest],
         evaluation_values: typing.Sequence[NamedTestCaseVariableValueRequest],
@@ -117,9 +118,11 @@ class TestSuitesClient:
         Parameters:
             - id: str. A UUID string identifying this test suite.
 
-            - upsert_test_suite_test_case_request_id: typing.Optional[str]. The ID of the Test Case to upsert. If specified and a match is found, the existing Test Case will be updated. If specified and no match is found, a Test Case will be created with the provided ID. If not provided, a new Test Case will be created with an auto-generated ID.
+            - upsert_test_suite_test_case_request_id: typing.Optional[str]. 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.
 
-            - label: typing.Optional[str].
+            - external_id: typing.Optional[str]. 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]. A human-readable label used to convey the intention of this Test Case
 
             - input_values: typing.Sequence[NamedTestCaseVariableValueRequest]. Values for each of the Test Case's input variables
 
@@ -141,6 +144,8 @@ class TestSuitesClient:
         _request: typing.Dict[str, typing.Any] = {"input_values": input_values, "evaluation_values": evaluation_values}
         if upsert_test_suite_test_case_request_id is not OMIT:
             _request["id"] = upsert_test_suite_test_case_request_id
+        if external_id is not OMIT:
+            _request["external_id"] = external_id
         if label is not OMIT:
             _request["label"] = label
         _response = self._client_wrapper.httpx_client.request(
@@ -198,7 +203,7 @@ class TestSuitesClient:
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from vellum import (
-            BulkCreateTestSuiteTestCaseDataRequest,
+            CreateTestSuiteTestCaseRequest,
             NamedTestCaseVariableValueRequest_String,
             TestSuiteTestCaseBulkOperationRequest_Create,
         )
@@ -212,10 +217,11 @@ class TestSuitesClient:
             request=[
                 TestSuiteTestCaseBulkOperationRequest_Create(
                     id="string",
-                    data=BulkCreateTestSuiteTestCaseDataRequest(
+                    data=CreateTestSuiteTestCaseRequest(
                         label="string",
                         input_values=[NamedTestCaseVariableValueRequest_String()],
                         evaluation_values=[NamedTestCaseVariableValueRequest_String()],
+                        external_id="string",
                     ),
                 )
             ],
@@ -397,6 +403,7 @@ class AsyncTestSuitesClient:
         id: str,
         *,
         upsert_test_suite_test_case_request_id: typing.Optional[str] = OMIT,
+        external_id: typing.Optional[str] = OMIT,
         label: typing.Optional[str] = OMIT,
         input_values: typing.Sequence[NamedTestCaseVariableValueRequest],
         evaluation_values: typing.Sequence[NamedTestCaseVariableValueRequest],
@@ -414,9 +421,11 @@ class AsyncTestSuitesClient:
         Parameters:
             - id: str. A UUID string identifying this test suite.
 
-            - upsert_test_suite_test_case_request_id: typing.Optional[str]. The ID of the Test Case to upsert. If specified and a match is found, the existing Test Case will be updated. If specified and no match is found, a Test Case will be created with the provided ID. If not provided, a new Test Case will be created with an auto-generated ID.
+            - upsert_test_suite_test_case_request_id: typing.Optional[str]. 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]. 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].
+            - label: typing.Optional[str]. A human-readable label used to convey the intention of this Test Case
 
             - input_values: typing.Sequence[NamedTestCaseVariableValueRequest]. Values for each of the Test Case's input variables
 
@@ -438,6 +447,8 @@ class AsyncTestSuitesClient:
         _request: typing.Dict[str, typing.Any] = {"input_values": input_values, "evaluation_values": evaluation_values}
         if upsert_test_suite_test_case_request_id is not OMIT:
             _request["id"] = upsert_test_suite_test_case_request_id
+        if external_id is not OMIT:
+            _request["external_id"] = external_id
         if label is not OMIT:
             _request["label"] = label
         _response = await self._client_wrapper.httpx_client.request(
@@ -495,7 +506,7 @@ class AsyncTestSuitesClient:
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from vellum import (
-            BulkCreateTestSuiteTestCaseDataRequest,
+            CreateTestSuiteTestCaseRequest,
             NamedTestCaseVariableValueRequest_String,
             TestSuiteTestCaseBulkOperationRequest_Create,
         )
@@ -509,10 +520,11 @@ class AsyncTestSuitesClient:
             request=[
                 TestSuiteTestCaseBulkOperationRequest_Create(
                     id="string",
-                    data=BulkCreateTestSuiteTestCaseDataRequest(
+                    data=CreateTestSuiteTestCaseRequest(
                         label="string",
                         input_values=[NamedTestCaseVariableValueRequest_String()],
                         evaluation_values=[NamedTestCaseVariableValueRequest_String()],
+                        external_id="string",
                     ),
                 )
             ],

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/__init__.py

@@ -50,8 +50,6 @@ from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1 import
 from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request import (
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
 )
-from .bulk_create_test_suite_test_case_data_request import BulkCreateTestSuiteTestCaseDataRequest
-from .bulk_replace_test_suite_test_case_data_request import BulkReplaceTestSuiteTestCaseDataRequest
 from .chat_history_enum import ChatHistoryEnum
 from .chat_history_input_request import ChatHistoryInputRequest
 from .chat_message import ChatMessage
@@ -95,10 +93,12 @@ from .code_execution_node_string_result import CodeExecutionNodeStringResult
 from .conditional_node_result import ConditionalNodeResult
 from .conditional_node_result_data import ConditionalNodeResultData
 from .create_enum import CreateEnum
+from .create_test_suite_test_case_request import CreateTestSuiteTestCaseRequest
 from .created_enum import CreatedEnum
 from .delete_enum import DeleteEnum
 from .deleted_enum import DeletedEnum
 from .deployment_provider_payload_response import DeploymentProviderPayloadResponse
+from .deployment_provider_payload_response_payload import DeploymentProviderPayloadResponsePayload
 from .deployment_read import DeploymentRead
 from .deployment_release_tag_deployment_history_item import DeploymentReleaseTagDeploymentHistoryItem
 from .deployment_release_tag_read import DeploymentReleaseTagRead
@@ -236,6 +236,9 @@ from .json_variable_value import JsonVariableValue
 from .json_vellum_value import JsonVellumValue
 from .logical_operator import LogicalOperator
 from .logprobs_enum import LogprobsEnum
+from .map_enum import MapEnum
+from .map_node_result import MapNodeResult
+from .map_node_result_data import MapNodeResultData
 from .merge_enum import MergeEnum
 from .merge_node_result import MergeNodeResult
 from .metadata_filter_config_request import MetadataFilterConfigRequest
@@ -378,6 +381,7 @@ from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
 from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
 from .release_tag_source import ReleaseTagSource
 from .replace_enum import ReplaceEnum
+from .replace_test_suite_test_case_request import ReplaceTestSuiteTestCaseRequest
 from .replaced_enum import ReplacedEnum
 from .sandbox_scenario import SandboxScenario
 from .scenario_input import ScenarioInput, ScenarioInput_ChatHistory, ScenarioInput_String
@@ -614,6 +618,7 @@ from .workflow_node_result_data import (
     WorkflowNodeResultData_Api,
     WorkflowNodeResultData_CodeExecution,
     WorkflowNodeResultData_Conditional,
+    WorkflowNodeResultData_Map,
     WorkflowNodeResultData_Merge,
     WorkflowNodeResultData_Metric,
     WorkflowNodeResultData_Prompt,
@@ -721,8 +726,6 @@ __all__ = [
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
-    "BulkCreateTestSuiteTestCaseDataRequest",
-    "BulkReplaceTestSuiteTestCaseDataRequest",
     "ChatHistoryEnum",
     "ChatHistoryInputRequest",
     "ChatMessage",
@@ -760,10 +763,12 @@ __all__ = [
     "ConditionalNodeResult",
     "ConditionalNodeResultData",
     "CreateEnum",
+    "CreateTestSuiteTestCaseRequest",
     "CreatedEnum",
     "DeleteEnum",
     "DeletedEnum",
     "DeploymentProviderPayloadResponse",
+    "DeploymentProviderPayloadResponsePayload",
     "DeploymentRead",
     "DeploymentReleaseTagDeploymentHistoryItem",
     "DeploymentReleaseTagRead",
@@ -885,6 +890,9 @@ __all__ = [
     "JsonVellumValue",
     "LogicalOperator",
     "LogprobsEnum",
+    "MapEnum",
+    "MapNodeResult",
+    "MapNodeResultData",
     "MergeEnum",
     "MergeNodeResult",
     "MetadataFilterConfigRequest",
@@ -1013,6 +1021,7 @@ __all__ = [
     "RejectedWorkflowNodeResultEvent",
     "ReleaseTagSource",
     "ReplaceEnum",
+    "ReplaceTestSuiteTestCaseRequest",
     "ReplacedEnum",
     "SandboxScenario",
     "ScenarioInput",
@@ -1226,6 +1235,7 @@ __all__ = [
     "WorkflowNodeResultData_Api",
     "WorkflowNodeResultData_CodeExecution",
     "WorkflowNodeResultData_Conditional",
+    "WorkflowNodeResultData_Map",
     "WorkflowNodeResultData_Merge",
     "WorkflowNodeResultData_Metric",
     "WorkflowNodeResultData_Prompt",

vellum_ai-0.6.6/src/vellum/types/bulk_create_test_suite_test_case_data_request.py → vellum_ai-0.6.7/src/vellum/types/create_test_suite_test_case_request.py

@@ -8,12 +8,16 @@ from ..core.pydantic_utilities import pydantic_v1
 from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
 
 
-class BulkCreateTestSuiteTestCaseDataRequest(pydantic_v1.BaseModel):
+class CreateTestSuiteTestCaseRequest(pydantic_v1.BaseModel):
     """
     Information about the Test Case to create
     """
 
-    label: typing.Optional[str] = None
+    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
@@ -24,6 +28,11 @@ class BulkCreateTestSuiteTestCaseDataRequest(pydantic_v1.BaseModel):
     Values for each of the Test Case's evaluation variables
     """
 
+    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Optionally provide an ID that uniquely identifies this Test Case in your system. Useful for updating this Test Cases data after initial creation. Cannot be changed later.
+    """
+
     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_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/deployment_provider_payload_response.py

@@ -5,10 +5,11 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
+from .deployment_provider_payload_response_payload import DeploymentProviderPayloadResponsePayload
 
 
 class DeploymentProviderPayloadResponse(pydantic_v1.BaseModel):
-    payload: typing.Dict[str, typing.Any]
+    payload: DeploymentProviderPayloadResponsePayload
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum_ai-0.6.7/src/vellum/types/deployment_provider_payload_response_payload.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+DeploymentProviderPayloadResponsePayload = typing.Union[typing.Dict[str, typing.Any], str]

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/execution_array_vellum_value.py

@@ -5,17 +5,21 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
-from .array_variable_value_item import ArrayVariableValueItem
+from .array_vellum_value_item import ArrayVellumValueItem
 
 
 class ExecutionArrayVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing an array of Vellum variable values.
+    """
+
     id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.
     """
 
     name: str
-    value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
+    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}

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/execution_chat_history_vellum_value.py

@@ -9,6 +9,10 @@ from .chat_message import ChatMessage
 
 
 class ExecutionChatHistoryVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing Chat History.
+    """
+
     id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/execution_error_vellum_value.py

@@ -9,6 +9,10 @@ from .vellum_error import VellumError
 
 
 class ExecutionErrorVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing an Error.
+    """
+
     id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/execution_function_call_vellum_value.py

@@ -9,6 +9,10 @@ from .function_call import FunctionCall
 
 
 class ExecutionFunctionCallVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing a Function Call.
+    """
+
     id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/execution_json_vellum_value.py

@@ -8,6 +8,10 @@ from ..core.pydantic_utilities import pydantic_v1
 
 
 class ExecutionJsonVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing a JSON object.
+    """
+
     id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/execution_number_vellum_value.py

@@ -8,6 +8,10 @@ from ..core.pydantic_utilities import pydantic_v1
 
 
 class ExecutionNumberVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing a number.
+    """
+
     id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/execution_search_results_vellum_value.py

@@ -9,6 +9,10 @@ from .search_result import SearchResult
 
 
 class ExecutionSearchResultsVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing Search Results.
+    """
+
     id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/execution_string_vellum_value.py

@@ -8,6 +8,10 @@ from ..core.pydantic_utilities import pydantic_v1
 
 
 class ExecutionStringVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing a string.
+    """
+
     id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.

vellum_ai-0.6.7/src/vellum/types/map_enum.py

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

vellum_ai-0.6.7/src/vellum/types/map_node_result.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 .map_node_result_data import MapNodeResultData
+
+
+class MapNodeResult(pydantic_v1.BaseModel):
+    """
+    A Node Result Event emitted from a Map Node.
+    """
+
+    data: typing.Optional[MapNodeResultData] = 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_ai-0.6.7/src/vellum/types/map_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 MapNodeResultData(pydantic_v1.BaseModel):
+    execution_ids: typing.List[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_ai-0.6.6/src/vellum/types/bulk_replace_test_suite_test_case_data_request.py → vellum_ai-0.6.7/src/vellum/types/replace_test_suite_test_case_request.py

@@ -8,17 +8,26 @@ from ..core.pydantic_utilities import pydantic_v1
 from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
 
 
-class BulkReplaceTestSuiteTestCaseDataRequest(pydantic_v1.BaseModel):
+class ReplaceTestSuiteTestCaseRequest(pydantic_v1.BaseModel):
     """
     Information about the Test Case to replace
     """
 
-    id: str = pydantic_v1.Field()
+    id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
-    The ID of the Test Case to replace.
+    The Vellum-generated ID of the Test Case whose data you'd like to replace. Must specify either this or external_id.
+    """
+
+    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    The ID that was originally provided upon Test Case creation that uniquely identifies the Test Case whose data you'd like to replace. Must specify either this of id.
+    """
+
+    label: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    A human-readable label used to convey the intention of this Test Case
     """
 
-    label: typing.Optional[str] = None
     input_values: typing.List[NamedTestCaseVariableValueRequest] = pydantic_v1.Field()
     """
     Values for each of the Test Case's input variables

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/search_result_meta.py

@@ -9,7 +9,7 @@ from .search_result_meta_source import SearchResultMetaSource
 
 
 class SearchResultMeta(pydantic_v1.BaseModel):
-    source: SearchResultMetaSource
+    source: typing.Optional[SearchResultMetaSource] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/search_result_meta_request.py

@@ -9,7 +9,7 @@ from .search_result_meta_source_request import SearchResultMetaSourceRequest
 
 
 class SearchResultMetaRequest(pydantic_v1.BaseModel):
-    source: SearchResultMetaSourceRequest
+    source: typing.Optional[SearchResultMetaSourceRequest] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/test_suite_test_case.py

@@ -10,6 +10,7 @@ from .test_case_variable_value import TestCaseVariableValue
 
 class TestSuiteTestCase(pydantic_v1.BaseModel):
     id: typing.Optional[str] = None
+    external_id: typing.Optional[str] = None
     label: typing.Optional[str] = None
     input_values: typing.List[TestCaseVariableValue]
     evaluation_values: typing.List[TestCaseVariableValue]

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/test_suite_test_case_create_bulk_operation_request.py

@@ -5,7 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
-from .bulk_create_test_suite_test_case_data_request import BulkCreateTestSuiteTestCaseDataRequest
+from .create_test_suite_test_case_request import CreateTestSuiteTestCaseRequest
 
 
 class TestSuiteTestCaseCreateBulkOperationRequest(pydantic_v1.BaseModel):
@@ -18,7 +18,7 @@ class TestSuiteTestCaseCreateBulkOperationRequest(pydantic_v1.BaseModel):
     An ID representing this specific operation. Can later be used to look up information about the operation's success in the response.
     """
 
-    data: BulkCreateTestSuiteTestCaseDataRequest
+    data: CreateTestSuiteTestCaseRequest
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{vellum_ai-0.6.6 → vellum_ai-0.6.7}/src/vellum/types/test_suite_test_case_replace_bulk_operation_request.py

@@ -5,7 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
-from .bulk_replace_test_suite_test_case_data_request import BulkReplaceTestSuiteTestCaseDataRequest
+from .replace_test_suite_test_case_request import ReplaceTestSuiteTestCaseRequest
 
 
 class TestSuiteTestCaseReplaceBulkOperationRequest(pydantic_v1.BaseModel):
@@ -18,7 +18,7 @@ class TestSuiteTestCaseReplaceBulkOperationRequest(pydantic_v1.BaseModel):
     An ID representing this specific operation. Can later be used to look up information about the operation's success in the response.
     """
 
-    data: BulkReplaceTestSuiteTestCaseDataRequest
+    data: ReplaceTestSuiteTestCaseRequest
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}