vellum-ai 0.7.11__py3-none-any.whl → 0.8.5__py3-none-any.whl

vellum/resources/test_suite_runs/client.py

@@ -1,16 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
+from ...core.client_wrapper import SyncClientWrapper
+from ...types.test_suite_run_exec_config_request import TestSuiteRunExecConfigRequest
+from ...core.request_options import RequestOptions
+from ...types.test_suite_run_read import TestSuiteRunRead
+from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
-
 from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import pydantic_v1
-from ...core.request_options import RequestOptions
 from ...types.paginated_test_suite_run_execution_list import PaginatedTestSuiteRunExecutionList
-from ...types.test_suite_run_exec_config_request import TestSuiteRunExecConfigRequest
-from ...types.test_suite_run_read import TestSuiteRunRead
+from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -50,21 +50,19 @@ class TestSuiteRunsClient:
         --------
         from vellum import (
             TestSuiteRunDeploymentReleaseTagExecConfigDataRequest,
-            TestSuiteRunExecConfigRequest_DeploymentReleaseTag,
+            TestSuiteRunDeploymentReleaseTagExecConfigRequest,
+            Vellum,
         )
-        from vellum.client import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
         )
         client.test_suite_runs.create(
-            test_suite_id="string",
-            exec_config=TestSuiteRunExecConfigRequest_DeploymentReleaseTag(
+            test_suite_id="test_suite_id",
+            exec_config=TestSuiteRunDeploymentReleaseTagExecConfigRequest(
                 data=TestSuiteRunDeploymentReleaseTagExecConfigDataRequest(
-                    deployment_id="string",
-                    tag="string",
+                    deployment_id="deployment_id",
                 ),
-                test_case_ids=["string"],
             ),
         )
         """
@@ -72,13 +70,22 @@ class TestSuiteRunsClient:
             "v1/test-suite-runs",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
-            json={"test_suite_id": test_suite_id, "exec_config": exec_config},
+            json={
+                "test_suite_id": test_suite_id,
+                "exec_config": exec_config,
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(TestSuiteRunRead, _response.json())  # type: ignore
+                return typing.cast(
+                    TestSuiteRunRead,
+                    parse_obj_as(
+                        type_=TestSuiteRunRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -103,7 +110,7 @@ class TestSuiteRunsClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -120,7 +127,13 @@ class TestSuiteRunsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(TestSuiteRunRead, _response.json())  # type: ignore
+                return typing.cast(
+                    TestSuiteRunRead,
+                    parse_obj_as(
+                        type_=TestSuiteRunRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -164,7 +177,7 @@ class TestSuiteRunsClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -177,12 +190,22 @@ class TestSuiteRunsClient:
             f"v1/test-suite-runs/{jsonable_encoder(id)}/executions",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
-            params={"expand": expand, "limit": limit, "offset": offset},
+            params={
+                "expand": expand,
+                "limit": limit,
+                "offset": offset,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PaginatedTestSuiteRunExecutionList, _response.json())  # type: ignore
+                return typing.cast(
+                    PaginatedTestSuiteRunExecutionList,
+                    parse_obj_as(
+                        type_=PaginatedTestSuiteRunExecutionList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -224,10 +247,10 @@ class AsyncTestSuiteRunsClient:
         import asyncio
 
         from vellum import (
+            AsyncVellum,
             TestSuiteRunDeploymentReleaseTagExecConfigDataRequest,
-            TestSuiteRunExecConfigRequest_DeploymentReleaseTag,
+            TestSuiteRunDeploymentReleaseTagExecConfigRequest,
         )
-        from vellum.client import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -236,13 +259,11 @@ class AsyncTestSuiteRunsClient:
 
         async def main() -> None:
             await client.test_suite_runs.create(
-                test_suite_id="string",
-                exec_config=TestSuiteRunExecConfigRequest_DeploymentReleaseTag(
+                test_suite_id="test_suite_id",
+                exec_config=TestSuiteRunDeploymentReleaseTagExecConfigRequest(
                     data=TestSuiteRunDeploymentReleaseTagExecConfigDataRequest(
-                        deployment_id="string",
-                        tag="string",
+                        deployment_id="deployment_id",
                     ),
-                    test_case_ids=["string"],
                 ),
             )
 
@@ -253,13 +274,22 @@ class AsyncTestSuiteRunsClient:
             "v1/test-suite-runs",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
-            json={"test_suite_id": test_suite_id, "exec_config": exec_config},
+            json={
+                "test_suite_id": test_suite_id,
+                "exec_config": exec_config,
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(TestSuiteRunRead, _response.json())  # type: ignore
+                return typing.cast(
+                    TestSuiteRunRead,
+                    parse_obj_as(
+                        type_=TestSuiteRunRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -286,7 +316,7 @@ class AsyncTestSuiteRunsClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -309,7 +339,13 @@ class AsyncTestSuiteRunsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(TestSuiteRunRead, _response.json())  # type: ignore
+                return typing.cast(
+                    TestSuiteRunRead,
+                    parse_obj_as(
+                        type_=TestSuiteRunRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -355,7 +391,7 @@ class AsyncTestSuiteRunsClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -374,12 +410,22 @@ class AsyncTestSuiteRunsClient:
             f"v1/test-suite-runs/{jsonable_encoder(id)}/executions",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
-            params={"expand": expand, "limit": limit, "offset": offset},
+            params={
+                "expand": expand,
+                "limit": limit,
+                "offset": offset,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PaginatedTestSuiteRunExecutionList, _response.json())  # type: ignore
+                return typing.cast(
+                    PaginatedTestSuiteRunExecutionList,
+                    parse_obj_as(
+                        type_=PaginatedTestSuiteRunExecutionList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)

vellum/resources/test_suites/client.py

@@ -1,19 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import json
 import typing
+from ...core.client_wrapper import SyncClientWrapper
+from ...core.request_options import RequestOptions
+from ...types.paginated_test_suite_test_case_list import PaginatedTestSuiteTestCaseList
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
-
 from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import pydantic_v1
-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
+import json
+from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -55,7 +55,7 @@ class TestSuitesClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -68,12 +68,21 @@ class TestSuitesClient:
             f"v1/test-suites/{jsonable_encoder(id)}/test-cases",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
-            params={"limit": limit, "offset": offset},
+            params={
+                "limit": limit,
+                "offset": offset,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PaginatedTestSuiteTestCaseList, _response.json())  # type: ignore
+                return typing.cast(
+                    PaginatedTestSuiteTestCaseList,
+                    parse_obj_as(
+                        type_=PaginatedTestSuiteTestCaseList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -129,15 +138,23 @@ class TestSuitesClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import NamedTestCaseStringVariableValueRequest, Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
         )
         client.test_suites.upsert_test_suite_test_case(
             id_="id",
-            input_values=[],
-            evaluation_values=[],
+            input_values=[
+                NamedTestCaseStringVariableValueRequest(
+                    name="name",
+                )
+            ],
+            evaluation_values=[
+                NamedTestCaseStringVariableValueRequest(
+                    name="name",
+                )
+            ],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -156,7 +173,13 @@ class TestSuitesClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(TestSuiteTestCase, _response.json())  # type: ignore
+                return typing.cast(
+                    TestSuiteTestCase,
+                    parse_obj_as(
+                        type_=TestSuiteTestCase,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -191,10 +214,10 @@ class TestSuitesClient:
         --------
         from vellum import (
             CreateTestSuiteTestCaseRequest,
-            NamedTestCaseVariableValueRequest_String,
-            TestSuiteTestCaseBulkOperationRequest_Create,
+            NamedTestCaseStringVariableValueRequest,
+            TestSuiteTestCaseCreateBulkOperationRequest,
+            Vellum,
         )
-        from vellum.client import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -202,12 +225,12 @@ class TestSuitesClient:
         response = client.test_suites.test_suite_test_cases_bulk(
             id="string",
             request=[
-                TestSuiteTestCaseBulkOperationRequest_Create(
+                TestSuiteTestCaseCreateBulkOperationRequest(
                     id="string",
                     data=CreateTestSuiteTestCaseRequest(
                         label="string",
-                        input_values=[NamedTestCaseVariableValueRequest_String()],
-                        evaluation_values=[NamedTestCaseVariableValueRequest_String()],
+                        input_values=[NamedTestCaseStringVariableValueRequest()],
+                        evaluation_values=[NamedTestCaseStringVariableValueRequest()],
                         external_id="string",
                     ),
                 )
@@ -230,7 +253,13 @@ class TestSuitesClient:
                         try:
                             if len(_text) == 0:
                                 continue
-                            yield pydantic_v1.parse_obj_as(typing.List[TestSuiteTestCaseBulkResult], json.loads(_text))  # type: ignore
+                            yield typing.cast(
+                                typing.List[TestSuiteTestCaseBulkResult],
+                                parse_obj_as(
+                                    type_=typing.List[TestSuiteTestCaseBulkResult],  # type: ignore
+                                    object_=json.loads(_text),
+                                ),
+                            )
                         except:
                             pass
                     return
@@ -263,7 +292,7 @@ class TestSuitesClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -326,7 +355,7 @@ class AsyncTestSuitesClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -345,12 +374,21 @@ class AsyncTestSuitesClient:
             f"v1/test-suites/{jsonable_encoder(id)}/test-cases",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
-            params={"limit": limit, "offset": offset},
+            params={
+                "limit": limit,
+                "offset": offset,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PaginatedTestSuiteTestCaseList, _response.json())  # type: ignore
+                return typing.cast(
+                    PaginatedTestSuiteTestCaseList,
+                    parse_obj_as(
+                        type_=PaginatedTestSuiteTestCaseList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -408,7 +446,7 @@ class AsyncTestSuitesClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum, NamedTestCaseStringVariableValueRequest
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -418,8 +456,16 @@ class AsyncTestSuitesClient:
         async def main() -> None:
             await client.test_suites.upsert_test_suite_test_case(
                 id_="id",
-                input_values=[],
-                evaluation_values=[],
+                input_values=[
+                    NamedTestCaseStringVariableValueRequest(
+                        name="name",
+                    )
+                ],
+                evaluation_values=[
+                    NamedTestCaseStringVariableValueRequest(
+                        name="name",
+                    )
+                ],
             )
 
 
@@ -441,7 +487,13 @@ class AsyncTestSuitesClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(TestSuiteTestCase, _response.json())  # type: ignore
+                return typing.cast(
+                    TestSuiteTestCase,
+                    parse_obj_as(
+                        type_=TestSuiteTestCase,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -477,11 +529,11 @@ class AsyncTestSuitesClient:
         import asyncio
 
         from vellum import (
+            AsyncVellum,
             CreateTestSuiteTestCaseRequest,
-            NamedTestCaseVariableValueRequest_String,
-            TestSuiteTestCaseBulkOperationRequest_Create,
+            NamedTestCaseStringVariableValueRequest,
+            TestSuiteTestCaseCreateBulkOperationRequest,
         )
-        from vellum.client import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -492,13 +544,13 @@ class AsyncTestSuitesClient:
             response = await client.test_suites.test_suite_test_cases_bulk(
                 id="string",
                 request=[
-                    TestSuiteTestCaseBulkOperationRequest_Create(
+                    TestSuiteTestCaseCreateBulkOperationRequest(
                         id="string",
                         data=CreateTestSuiteTestCaseRequest(
                             label="string",
-                            input_values=[NamedTestCaseVariableValueRequest_String()],
+                            input_values=[NamedTestCaseStringVariableValueRequest()],
                             evaluation_values=[
-                                NamedTestCaseVariableValueRequest_String()
+                                NamedTestCaseStringVariableValueRequest()
                             ],
                             external_id="string",
                         ),
@@ -525,7 +577,13 @@ class AsyncTestSuitesClient:
                         try:
                             if len(_text) == 0:
                                 continue
-                            yield pydantic_v1.parse_obj_as(typing.List[TestSuiteTestCaseBulkResult], json.loads(_text))  # type: ignore
+                            yield typing.cast(
+                                typing.List[TestSuiteTestCaseBulkResult],
+                                parse_obj_as(
+                                    type_=typing.List[TestSuiteTestCaseBulkResult],  # type: ignore
+                                    object_=json.loads(_text),
+                                ),
+                            )
                         except:
                             pass
                     return
@@ -560,7 +618,7 @@ class AsyncTestSuitesClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",