vellum-ai 0.6.8__tar.gz → 0.7.1__tar.gz

{vellum_ai-0.6.8 → vellum_ai-0.7.1}/PKG-INFO

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

{vellum_ai-0.6.8 → vellum_ai-0.7.1}/pyproject.toml

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

{vellum_ai-0.6.8 → vellum_ai-0.7.1}/src/vellum/__init__.py

@@ -193,6 +193,7 @@ from .types import (
     InstructorVectorizerConfig,
     InstructorVectorizerConfigRequest,
     IntfloatMultilingualE5LargeEnum,
+    IterationStateEnum,
     JsonEnum,
     JsonInputRequest,
     JsonVariableValue,
@@ -204,6 +205,7 @@ from .types import (
     MapNodeResultData,
     MergeEnum,
     MergeNodeResult,
+    MergeNodeResultData,
     MetadataFilterConfigRequest,
     MetadataFilterRuleCombinator,
     MetadataFilterRuleRequest,
@@ -453,6 +455,7 @@ from .types import (
     TestSuiteRunExecConfig_External,
     TestSuiteRunExecConfig_WorkflowReleaseTag,
     TestSuiteRunExecution,
+    TestSuiteRunExecutionArrayOutput,
     TestSuiteRunExecutionChatHistoryOutput,
     TestSuiteRunExecutionErrorOutput,
     TestSuiteRunExecutionFunctionCallOutput,
@@ -461,6 +464,7 @@ from .types import (
     TestSuiteRunExecutionMetricResult,
     TestSuiteRunExecutionNumberOutput,
     TestSuiteRunExecutionOutput,
+    TestSuiteRunExecutionOutput_Array,
     TestSuiteRunExecutionOutput_ChatHistory,
     TestSuiteRunExecutionOutput_Error,
     TestSuiteRunExecutionOutput_FunctionCall,
@@ -498,6 +502,7 @@ from .types import (
     TestSuiteTestCaseBulkOperationRequest_Create,
     TestSuiteTestCaseBulkOperationRequest_Delete,
     TestSuiteTestCaseBulkOperationRequest_Replace,
+    TestSuiteTestCaseBulkOperationRequest_Upsert,
     TestSuiteTestCaseBulkResult,
     TestSuiteTestCaseBulkResult_Created,
     TestSuiteTestCaseBulkResult_Deleted,
@@ -514,6 +519,7 @@ from .types import (
     TestSuiteTestCaseReplaceBulkOperationRequest,
     TestSuiteTestCaseReplacedBulkResult,
     TestSuiteTestCaseReplacedBulkResultData,
+    TestSuiteTestCaseUpsertBulkOperationRequest,
     TextEmbedding3LargeEnum,
     TextEmbedding3SmallEnum,
     TextEmbeddingAda002Enum,
@@ -524,6 +530,8 @@ from .types import (
     TokenOverlappingWindowChunkingRequest,
     UploadDocumentErrorResponse,
     UploadDocumentResponse,
+    UpsertEnum,
+    UpsertTestSuiteTestCaseRequest,
     VellumError,
     VellumErrorCodeEnum,
     VellumErrorRequest,
@@ -826,6 +834,7 @@ __all__ = [
     "InstructorVectorizerConfigRequest",
     "InternalServerError",
     "IntfloatMultilingualE5LargeEnum",
+    "IterationStateEnum",
     "JsonEnum",
     "JsonInputRequest",
     "JsonVariableValue",
@@ -837,6 +846,7 @@ __all__ = [
     "MapNodeResultData",
     "MergeEnum",
     "MergeNodeResult",
+    "MergeNodeResultData",
     "MetadataFilterConfigRequest",
     "MetadataFilterRuleCombinator",
     "MetadataFilterRuleRequest",
@@ -1087,6 +1097,7 @@ __all__ = [
     "TestSuiteRunExecConfig_External",
     "TestSuiteRunExecConfig_WorkflowReleaseTag",
     "TestSuiteRunExecution",
+    "TestSuiteRunExecutionArrayOutput",
     "TestSuiteRunExecutionChatHistoryOutput",
     "TestSuiteRunExecutionErrorOutput",
     "TestSuiteRunExecutionFunctionCallOutput",
@@ -1095,6 +1106,7 @@ __all__ = [
     "TestSuiteRunExecutionMetricResult",
     "TestSuiteRunExecutionNumberOutput",
     "TestSuiteRunExecutionOutput",
+    "TestSuiteRunExecutionOutput_Array",
     "TestSuiteRunExecutionOutput_ChatHistory",
     "TestSuiteRunExecutionOutput_Error",
     "TestSuiteRunExecutionOutput_FunctionCall",
@@ -1132,6 +1144,7 @@ __all__ = [
     "TestSuiteTestCaseBulkOperationRequest_Create",
     "TestSuiteTestCaseBulkOperationRequest_Delete",
     "TestSuiteTestCaseBulkOperationRequest_Replace",
+    "TestSuiteTestCaseBulkOperationRequest_Upsert",
     "TestSuiteTestCaseBulkResult",
     "TestSuiteTestCaseBulkResult_Created",
     "TestSuiteTestCaseBulkResult_Deleted",
@@ -1148,6 +1161,7 @@ __all__ = [
     "TestSuiteTestCaseReplaceBulkOperationRequest",
     "TestSuiteTestCaseReplacedBulkResult",
     "TestSuiteTestCaseReplacedBulkResultData",
+    "TestSuiteTestCaseUpsertBulkOperationRequest",
     "TextEmbedding3LargeEnum",
     "TextEmbedding3SmallEnum",
     "TextEmbeddingAda002Enum",
@@ -1158,6 +1172,8 @@ __all__ = [
     "TokenOverlappingWindowChunkingRequest",
     "UploadDocumentErrorResponse",
     "UploadDocumentResponse",
+    "UpsertEnum",
+    "UpsertTestSuiteTestCaseRequest",
     "VellumEnvironment",
     "VellumError",
     "VellumErrorCodeEnum",

{vellum_ai-0.6.8 → vellum_ai-0.7.1}/src/vellum/client.py

@@ -121,23 +121,23 @@ class Vellum:
         Executes a deployed Prompt and returns the result.
 
         Parameters:
-            - inputs: typing.Sequence[PromptDeploymentInputRequest]. The list of inputs defined in the Prompt's deployment with their corresponding values.
+            - inputs: typing.Sequence[PromptDeploymentInputRequest]. A list consisting of the Prompt Deployment's input variables and their values.
 
             - prompt_deployment_id: typing.Optional[str]. The ID of the Prompt Deployment. Must provide either this or prompt_deployment_name.
 
-            - prompt_deployment_name: typing.Optional[str]. The name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
+            - prompt_deployment_name: typing.Optional[str]. The unique name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
 
             - release_tag: typing.Optional[str]. Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
-            - external_id: typing.Optional[str]. "Optionally include a unique identifier for tracking purposes. Must be unique for a given prompt deployment.
+            - external_id: typing.Optional[str]. Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
 
-            - expand_meta: typing.Optional[PromptDeploymentExpandMetaRequestRequest]. The name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
+            - expand_meta: typing.Optional[PromptDeploymentExpandMetaRequestRequest]. An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
 
-            - raw_overrides: typing.Optional[RawPromptExecutionOverridesRequest].
+            - raw_overrides: typing.Optional[RawPromptExecutionOverridesRequest]. Overrides for the raw API request sent to the model host. Combined with `expand_raw`, it can be used to access new features from models.
 
-            - expand_raw: typing.Optional[typing.Sequence[str]]. Returns the raw API response data sent from the model host. Combined with `raw_overrides`, it can be used to access new features from models.
+            - expand_raw: typing.Optional[typing.Sequence[str]]. A list of keys whose values you'd like to directly return from the JSON response of the model provider. Useful if you need lower-level info returned by model providers that Vellum would otherwise omit. Corresponding key/value pairs will be returned under the `raw` key of the API response.
 
-            - metadata: typing.Optional[typing.Dict[str, typing.Any]].
+            - metadata: typing.Optional[typing.Dict[str, typing.Any]]. Arbitrary JSON metadata associated with this request. Can be used to capture additional monitoring data such as user id, session id, etc. for future analysis.
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
@@ -256,23 +256,23 @@ class Vellum:
         Executes a deployed Prompt and streams back the results.
 
         Parameters:
-            - inputs: typing.Sequence[PromptDeploymentInputRequest]. The list of inputs defined in the Prompt's deployment with their corresponding values.
+            - inputs: typing.Sequence[PromptDeploymentInputRequest]. A list consisting of the Prompt Deployment's input variables and their values.
 
             - prompt_deployment_id: typing.Optional[str]. The ID of the Prompt Deployment. Must provide either this or prompt_deployment_name.
 
-            - prompt_deployment_name: typing.Optional[str]. The name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
+            - prompt_deployment_name: typing.Optional[str]. The unique name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
 
             - release_tag: typing.Optional[str]. Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
-            - external_id: typing.Optional[str]. "Optionally include a unique identifier for tracking purposes. Must be unique for a given prompt deployment.
+            - external_id: typing.Optional[str]. Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
 
-            - expand_meta: typing.Optional[PromptDeploymentExpandMetaRequestRequest]. The name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
+            - expand_meta: typing.Optional[PromptDeploymentExpandMetaRequestRequest]. An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
 
-            - raw_overrides: typing.Optional[RawPromptExecutionOverridesRequest].
+            - raw_overrides: typing.Optional[RawPromptExecutionOverridesRequest]. Overrides for the raw API request sent to the model host. Combined with `expand_raw`, it can be used to access new features from models.
 
-            - expand_raw: typing.Optional[typing.Sequence[str]]. Returns the raw API response data sent from the model host. Combined with `raw_overrides`, it can be used to access new features from models.
+            - expand_raw: typing.Optional[typing.Sequence[str]]. A list of keys whose values you'd like to directly return from the JSON response of the model provider. Useful if you need lower-level info returned by model providers that Vellum would otherwise omit. Corresponding key/value pairs will be returned under the `raw` key of the API response.
 
-            - metadata: typing.Optional[typing.Dict[str, typing.Any]].
+            - metadata: typing.Optional[typing.Dict[str, typing.Any]]. Arbitrary JSON metadata associated with this request. Can be used to capture additional monitoring data such as user id, session id, etc. for future analysis.
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
@@ -1082,23 +1082,23 @@ class AsyncVellum:
         Executes a deployed Prompt and returns the result.
 
         Parameters:
-            - inputs: typing.Sequence[PromptDeploymentInputRequest]. The list of inputs defined in the Prompt's deployment with their corresponding values.
+            - inputs: typing.Sequence[PromptDeploymentInputRequest]. A list consisting of the Prompt Deployment's input variables and their values.
 
             - prompt_deployment_id: typing.Optional[str]. The ID of the Prompt Deployment. Must provide either this or prompt_deployment_name.
 
-            - prompt_deployment_name: typing.Optional[str]. The name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
+            - prompt_deployment_name: typing.Optional[str]. The unique name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
 
             - release_tag: typing.Optional[str]. Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
-            - external_id: typing.Optional[str]. "Optionally include a unique identifier for tracking purposes. Must be unique for a given prompt deployment.
+            - external_id: typing.Optional[str]. Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
 
-            - expand_meta: typing.Optional[PromptDeploymentExpandMetaRequestRequest]. The name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
+            - expand_meta: typing.Optional[PromptDeploymentExpandMetaRequestRequest]. An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
 
-            - raw_overrides: typing.Optional[RawPromptExecutionOverridesRequest].
+            - raw_overrides: typing.Optional[RawPromptExecutionOverridesRequest]. Overrides for the raw API request sent to the model host. Combined with `expand_raw`, it can be used to access new features from models.
 
-            - expand_raw: typing.Optional[typing.Sequence[str]]. Returns the raw API response data sent from the model host. Combined with `raw_overrides`, it can be used to access new features from models.
+            - expand_raw: typing.Optional[typing.Sequence[str]]. A list of keys whose values you'd like to directly return from the JSON response of the model provider. Useful if you need lower-level info returned by model providers that Vellum would otherwise omit. Corresponding key/value pairs will be returned under the `raw` key of the API response.
 
-            - metadata: typing.Optional[typing.Dict[str, typing.Any]].
+            - metadata: typing.Optional[typing.Dict[str, typing.Any]]. Arbitrary JSON metadata associated with this request. Can be used to capture additional monitoring data such as user id, session id, etc. for future analysis.
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
@@ -1217,23 +1217,23 @@ class AsyncVellum:
         Executes a deployed Prompt and streams back the results.
 
         Parameters:
-            - inputs: typing.Sequence[PromptDeploymentInputRequest]. The list of inputs defined in the Prompt's deployment with their corresponding values.
+            - inputs: typing.Sequence[PromptDeploymentInputRequest]. A list consisting of the Prompt Deployment's input variables and their values.
 
             - prompt_deployment_id: typing.Optional[str]. The ID of the Prompt Deployment. Must provide either this or prompt_deployment_name.
 
-            - prompt_deployment_name: typing.Optional[str]. The name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
+            - prompt_deployment_name: typing.Optional[str]. The unique name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
 
             - release_tag: typing.Optional[str]. Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
-            - external_id: typing.Optional[str]. "Optionally include a unique identifier for tracking purposes. Must be unique for a given prompt deployment.
+            - external_id: typing.Optional[str]. Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
 
-            - expand_meta: typing.Optional[PromptDeploymentExpandMetaRequestRequest]. The name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
+            - expand_meta: typing.Optional[PromptDeploymentExpandMetaRequestRequest]. An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
 
-            - raw_overrides: typing.Optional[RawPromptExecutionOverridesRequest].
+            - raw_overrides: typing.Optional[RawPromptExecutionOverridesRequest]. Overrides for the raw API request sent to the model host. Combined with `expand_raw`, it can be used to access new features from models.
 
-            - expand_raw: typing.Optional[typing.Sequence[str]]. Returns the raw API response data sent from the model host. Combined with `raw_overrides`, it can be used to access new features from models.
+            - expand_raw: typing.Optional[typing.Sequence[str]]. A list of keys whose values you'd like to directly return from the JSON response of the model provider. Useful if you need lower-level info returned by model providers that Vellum would otherwise omit. Corresponding key/value pairs will be returned under the `raw` key of the API response.
 
-            - metadata: typing.Optional[typing.Dict[str, typing.Any]].
+            - metadata: typing.Optional[typing.Dict[str, typing.Any]]. Arbitrary JSON metadata associated with this request. Can be used to capture additional monitoring data such as user id, session id, etc. for future analysis.
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---

{vellum_ai-0.6.8 → vellum_ai-0.7.1}/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.8",
+            "X-Fern-SDK-Version": "0.7.1",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.6.8 → vellum_ai-0.7.1}/src/vellum/lib/test_suites/resources.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import logging
 import time
 from functools import cached_property
-from typing import Callable, Generator, List, Any
+from typing import Callable, Generator, List, cast, Iterable
 
 from vellum import TestSuiteRunRead, TestSuiteRunMetricOutput_Number
 from vellum.client import Vellum
@@ -174,7 +174,7 @@ class VellumTestSuiteRunResults:
         self,
         metric_identifier: str | None = None,
         output_identifier: str | None = None,
-    ) -> List[float]:
+    ) -> List[float | None]:
         """Returns the values of a numeric metric output that match the given criteria."""
 
         metric_outputs: list[TestSuiteRunMetricOutput_Number] = []
@@ -198,7 +198,7 @@ class VellumTestSuiteRunResults:
         output_values = self.get_numeric_metric_output_values(
             metric_identifier=metric_identifier, output_identifier=output_identifier
         )
-        return sum(output_values) / len(output_values)
+        return sum(cast(Iterable[float], filter(lambda o: isinstance(o, float), output_values))) / len(output_values)
 
     def get_min_metric_output(
         self, metric_identifier: str | None = None, output_identifier: str | None = None
@@ -207,7 +207,7 @@ class VellumTestSuiteRunResults:
         output_values = self.get_numeric_metric_output_values(
             metric_identifier=metric_identifier, output_identifier=output_identifier
         )
-        return min(output_values)
+        return min(cast(Iterable[float], filter(lambda o: isinstance(o, float), output_values)))
 
     def get_max_metric_output(
         self, metric_identifier: str | None = None, output_identifier: str | None = None
@@ -216,7 +216,7 @@ class VellumTestSuiteRunResults:
         output_values = self.get_numeric_metric_output_values(
             metric_identifier=metric_identifier, output_identifier=output_identifier
         )
-        return max(output_values)
+        return max(cast(Iterable[float], filter(lambda o: isinstance(o, float), output_values)))
 
     def wait_until_complete(self) -> None:
         """Wait until the Test Suite Run is no longer in a QUEUED or RUNNING state."""

{vellum_ai-0.6.8 → vellum_ai-0.7.1}/src/vellum/resources/document_indexes/client.py

@@ -458,6 +458,63 @@ class DocumentIndexesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def add_document(
+        self, document_id: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Adds a previously uploaded Document to the specified Document Index.
+
+        Parameters:
+            - document_id: str. Either the Vellum-generated ID or the originally supplied external_id that uniquely identifies the Document you'd like to add.
+
+            - id: str. Either the Vellum-generated ID or the originally specified name that uniquely identifies the Document Index to which you'd like to add the Document.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from vellum.client import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.document_indexes.add_document(
+            document_id="document_id",
+            id="id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            method="POST",
+            url=urllib.parse.urljoin(
+                f"{self._client_wrapper.get_environment().default}/",
+                f"v1/document-indexes/{jsonable_encoder(id)}/documents/{jsonable_encoder(document_id)}",
+            ),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))
+            if request_options is not None
+            else None,
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     def remove_document(
         self, document_id: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> None:
@@ -950,6 +1007,63 @@ class AsyncDocumentIndexesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def add_document(
+        self, document_id: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Adds a previously uploaded Document to the specified Document Index.
+
+        Parameters:
+            - document_id: str. Either the Vellum-generated ID or the originally supplied external_id that uniquely identifies the Document you'd like to add.
+
+            - id: str. Either the Vellum-generated ID or the originally specified name that uniquely identifies the Document Index to which you'd like to add the Document.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from vellum.client import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+        await client.document_indexes.add_document(
+            document_id="document_id",
+            id="id",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            method="POST",
+            url=urllib.parse.urljoin(
+                f"{self._client_wrapper.get_environment().default}/",
+                f"v1/document-indexes/{jsonable_encoder(id)}/documents/{jsonable_encoder(document_id)}",
+            ),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))
+            if request_options is not None
+            else None,
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     async def remove_document(
         self, document_id: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> None:

{vellum_ai-0.6.8 → vellum_ai-0.7.1}/src/vellum/resources/test_suites/client.py

@@ -11,11 +11,11 @@ from ...core.jsonable_encoder import jsonable_encoder
 from ...core.pydantic_utilities import pydantic_v1
 from ...core.remove_none_from_dict import remove_none_from_dict
 from ...core.request_options import RequestOptions
-from ...types.named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
 from ...types.paginated_test_suite_test_case_list import PaginatedTestSuiteTestCaseList
 from ...types.test_suite_test_case import TestSuiteTestCase
 from ...types.test_suite_test_case_bulk_operation_request import TestSuiteTestCaseBulkOperationRequest
 from ...types.test_suite_test_case_bulk_result import TestSuiteTestCaseBulkResult
+from ...types.upsert_test_suite_test_case_request import UpsertTestSuiteTestCaseRequest
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -99,11 +99,7 @@ class TestSuitesClient:
         self,
         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],
+        request: UpsertTestSuiteTestCaseRequest,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSuiteTestCase:
         """
@@ -118,18 +114,11 @@ class TestSuitesClient:
         Parameters:
             - id: str. A UUID string identifying this test suite.
 
-            - 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]. 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
-
-            - evaluation_values: typing.Sequence[NamedTestCaseVariableValueRequest]. Values for each of the Test Case's evaluation variables
+            - request: UpsertTestSuiteTestCaseRequest.
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
+        from vellum import UpsertTestSuiteTestCaseRequest
         from vellum.client import Vellum
 
         client = Vellum(
@@ -137,17 +126,12 @@ class TestSuitesClient:
         )
         client.test_suites.upsert_test_suite_test_case(
             id="id",
-            input_values=[],
-            evaluation_values=[],
+            request=UpsertTestSuiteTestCaseRequest(
+                input_values=[],
+                evaluation_values=[],
+            ),
         )
         """
-        _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(
             method="POST",
             url=urllib.parse.urljoin(
@@ -157,10 +141,10 @@ class TestSuitesClient:
             params=jsonable_encoder(
                 request_options.get("additional_query_parameters") if request_options is not None else None
             ),
-            json=jsonable_encoder(_request)
+            json=jsonable_encoder(request)
             if request_options is None or request_options.get("additional_body_parameters") is None
             else {
-                **jsonable_encoder(_request),
+                **jsonable_encoder(request),
                 **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
             },
             headers=jsonable_encoder(
@@ -402,11 +386,7 @@ class AsyncTestSuitesClient:
         self,
         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],
+        request: UpsertTestSuiteTestCaseRequest,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSuiteTestCase:
         """
@@ -421,18 +401,11 @@ class AsyncTestSuitesClient:
         Parameters:
             - id: str. A UUID string identifying this test suite.
 
-            - 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]. 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
-
-            - evaluation_values: typing.Sequence[NamedTestCaseVariableValueRequest]. Values for each of the Test Case's evaluation variables
+            - request: UpsertTestSuiteTestCaseRequest.
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
+        from vellum import UpsertTestSuiteTestCaseRequest
         from vellum.client import AsyncVellum
 
         client = AsyncVellum(
@@ -440,17 +413,12 @@ class AsyncTestSuitesClient:
         )
         await client.test_suites.upsert_test_suite_test_case(
             id="id",
-            input_values=[],
-            evaluation_values=[],
+            request=UpsertTestSuiteTestCaseRequest(
+                input_values=[],
+                evaluation_values=[],
+            ),
         )
         """
-        _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(
             method="POST",
             url=urllib.parse.urljoin(
@@ -460,10 +428,10 @@ class AsyncTestSuitesClient:
             params=jsonable_encoder(
                 request_options.get("additional_query_parameters") if request_options is not None else None
             ),
-            json=jsonable_encoder(_request)
+            json=jsonable_encoder(request)
             if request_options is None or request_options.get("additional_body_parameters") is None
             else {
-                **jsonable_encoder(_request),
+                **jsonable_encoder(request),
                 **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
             },
             headers=jsonable_encoder(