vellum-ai 1.1.2__py3-none-any.whl → 1.1.4__py3-none-any.whl

vellum/client/resources/test_suites/client.py

@@ -1,22 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .raw_client import RawTestSuitesClient
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
-from ...types.paginated_test_suite_test_case_list import PaginatedTestSuiteTestCaseList
 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 ...core.jsonable_encoder import jsonable_encoder
-from ...core.serialization import convert_and_respect_annotation_metadata
-from ...core.pydantic_utilities import parse_obj_as
-import json
-from json.decoder import JSONDecodeError
-from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawTestSuitesClient
+from .raw_client import AsyncRawTestSuitesClient, RawTestSuitesClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -83,13 +76,10 @@ class TestSuitesClient:
             id="id",
         )
         """
-        response = self._raw_client.list_test_suite_test_cases(
-            id,
-            limit=limit,
-            offset=offset,
-            request_options=request_options,
+        _response = self._raw_client.list_test_suite_test_cases(
+            id, limit=limit, offset=offset, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     def upsert_test_suite_test_case(
         self,
@@ -167,7 +157,7 @@ class TestSuitesClient:
             ],
         )
         """
-        response = self._raw_client.upsert_test_suite_test_case(
+        _response = self._raw_client.upsert_test_suite_test_case(
             id_,
             input_values=input_values,
             evaluation_values=evaluation_values,
@@ -176,7 +166,7 @@ class TestSuitesClient:
             label=label,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     def test_suite_test_cases_bulk(
         self,
@@ -266,37 +256,8 @@ class TestSuitesClient:
         for chunk in response:
             yield chunk
         """
-        with self._raw_client._client_wrapper.httpx_client.stream(
-            f"v1/test-suites/{jsonable_encoder(id)}/test-cases-bulk",
-            base_url=self._raw_client._client_wrapper.get_environment().default,
-            method="POST",
-            json=convert_and_respect_annotation_metadata(
-                object_=request, annotation=typing.Sequence[TestSuiteTestCaseBulkOperationRequest], direction="write"
-            ),
-            request_options=request_options,
-            omit=OMIT,
-        ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    for _text in _response.iter_lines():
-                        try:
-                            if len(_text) == 0:
-                                continue
-                            yield typing.cast(
-                                typing.List[TestSuiteTestCaseBulkResult],
-                                parse_obj_as(
-                                    type_=typing.List[TestSuiteTestCaseBulkResult],  # type: ignore
-                                    object_=json.loads(_text),
-                                ),
-                            )
-                        except Exception:
-                            pass
-                    return
-                _response.read()
-                _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)
+        with self._raw_client.test_suite_test_cases_bulk(id, request=request, request_options=request_options) as r:
+            yield from r.data
 
     def delete_test_suite_test_case(
         self, id: str, test_case_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -332,12 +293,8 @@ class TestSuitesClient:
             test_case_id="test_case_id",
         )
         """
-        response = self._raw_client.delete_test_suite_test_case(
-            id,
-            test_case_id,
-            request_options=request_options,
-        )
-        return response.data
+        _response = self._raw_client.delete_test_suite_test_case(id, test_case_id, request_options=request_options)
+        return _response.data
 
 
 class AsyncTestSuitesClient:
@@ -405,13 +362,10 @@ class AsyncTestSuitesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.list_test_suite_test_cases(
-            id,
-            limit=limit,
-            offset=offset,
-            request_options=request_options,
+        _response = await self._raw_client.list_test_suite_test_cases(
+            id, limit=limit, offset=offset, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def upsert_test_suite_test_case(
         self,
@@ -497,7 +451,7 @@ class AsyncTestSuitesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.upsert_test_suite_test_case(
+        _response = await self._raw_client.upsert_test_suite_test_case(
             id_,
             input_values=input_values,
             evaluation_values=evaluation_values,
@@ -506,7 +460,7 @@ class AsyncTestSuitesClient:
             label=label,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     async def test_suite_test_cases_bulk(
         self,
@@ -604,37 +558,11 @@ class AsyncTestSuitesClient:
 
         asyncio.run(main())
         """
-        async with self._raw_client._client_wrapper.httpx_client.stream(
-            f"v1/test-suites/{jsonable_encoder(id)}/test-cases-bulk",
-            base_url=self._raw_client._client_wrapper.get_environment().default,
-            method="POST",
-            json=convert_and_respect_annotation_metadata(
-                object_=request, annotation=typing.Sequence[TestSuiteTestCaseBulkOperationRequest], direction="write"
-            ),
-            request_options=request_options,
-            omit=OMIT,
-        ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    async for _text in _response.aiter_lines():
-                        try:
-                            if len(_text) == 0:
-                                continue
-                            yield typing.cast(
-                                typing.List[TestSuiteTestCaseBulkResult],
-                                parse_obj_as(
-                                    type_=typing.List[TestSuiteTestCaseBulkResult],  # type: ignore
-                                    object_=json.loads(_text),
-                                ),
-                            )
-                        except Exception:
-                            pass
-                    return
-                await _response.aread()
-                _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 with self._raw_client.test_suite_test_cases_bulk(
+            id, request=request, request_options=request_options
+        ) as r:
+            async for _chunk in r.data:
+                yield _chunk
 
     async def delete_test_suite_test_case(
         self, id: str, test_case_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -678,9 +606,7 @@ class AsyncTestSuitesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.delete_test_suite_test_case(
-            id,
-            test_case_id,
-            request_options=request_options,
+        _response = await self._raw_client.delete_test_suite_test_case(
+            id, test_case_id, request_options=request_options
         )
-        return response.data
+        return _response.data

vellum/client/resources/test_suites/raw_client.py

@@ -1,19 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import contextlib
+import json
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-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.http_response import AsyncHttpResponse, HttpResponse
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...core.serialization import convert_and_respect_annotation_metadata
 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 ...core.serialization import convert_and_respect_annotation_metadata
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
+from ...types.test_suite_test_case_bulk_operation_request import TestSuiteTestCaseBulkOperationRequest
+from ...types.test_suite_test_case_bulk_result import TestSuiteTestCaseBulkResult
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -75,8 +78,8 @@ class RawTestSuitesClient:
                 return HttpResponse(response=_response, data=_data)
             _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)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     def upsert_test_suite_test_case(
         self,
@@ -145,6 +148,9 @@ class RawTestSuitesClient:
                     direction="write",
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -160,8 +166,79 @@ class RawTestSuitesClient:
                 return HttpResponse(response=_response, data=_data)
             _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)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    @contextlib.contextmanager
+    def test_suite_test_cases_bulk(
+        self,
+        id: str,
+        *,
+        request: typing.Sequence[TestSuiteTestCaseBulkOperationRequest],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Iterator[HttpResponse[typing.Iterator[typing.List[TestSuiteTestCaseBulkResult]]]]:
+        """
+        Created, replace, and delete Test Cases within the specified Test Suite in bulk
+
+        Parameters
+        ----------
+        id : str
+            Either the Test Suites' ID or its unique name
+
+        request : typing.Sequence[TestSuiteTestCaseBulkOperationRequest]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Yields
+        ------
+        typing.Iterator[HttpResponse[typing.Iterator[typing.List[TestSuiteTestCaseBulkResult]]]]
+
+        """
+        with self._client_wrapper.httpx_client.stream(
+            f"v1/test-suites/{jsonable_encoder(id)}/test-cases-bulk",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=typing.Sequence[TestSuiteTestCaseBulkOperationRequest], direction="write"
+            ),
+            headers={
+                "content-type": "application/x-ndjson",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        ) as _response:
+
+            def _stream() -> HttpResponse[typing.Iterator[typing.List[TestSuiteTestCaseBulkResult]]]:
+                try:
+                    if 200 <= _response.status_code < 300:
+
+                        def _iter():
+                            for _text in _response.iter_lines():
+                                try:
+                                    if len(_text) == 0:
+                                        continue
+                                    yield typing.cast(
+                                        typing.List[TestSuiteTestCaseBulkResult],
+                                        parse_obj_as(
+                                            type_=typing.List[TestSuiteTestCaseBulkResult],  # type: ignore
+                                            object_=json.loads(_text),
+                                        ),
+                                    )
+                                except Exception:
+                                    pass
+                            return
+
+                        return HttpResponse(response=_response, data=_iter())
+                    _response.read()
+                    _response_json = _response.json()
+                except JSONDecodeError:
+                    raise ApiError(
+                        status_code=_response.status_code, headers=dict(_response.headers), body=_response.text
+                    )
+                raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+            yield _stream()
 
     def delete_test_suite_test_case(
         self, id: str, test_case_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -195,8 +272,8 @@ class RawTestSuitesClient:
                 return HttpResponse(response=_response, data=None)
             _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)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
 
 class AsyncRawTestSuitesClient:
@@ -255,8 +332,8 @@ class AsyncRawTestSuitesClient:
                 return AsyncHttpResponse(response=_response, data=_data)
             _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)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     async def upsert_test_suite_test_case(
         self,
@@ -325,6 +402,9 @@ class AsyncRawTestSuitesClient:
                     direction="write",
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -340,8 +420,79 @@ class AsyncRawTestSuitesClient:
                 return AsyncHttpResponse(response=_response, data=_data)
             _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)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    @contextlib.asynccontextmanager
+    async def test_suite_test_cases_bulk(
+        self,
+        id: str,
+        *,
+        request: typing.Sequence[TestSuiteTestCaseBulkOperationRequest],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.AsyncIterator[AsyncHttpResponse[typing.AsyncIterator[typing.List[TestSuiteTestCaseBulkResult]]]]:
+        """
+        Created, replace, and delete Test Cases within the specified Test Suite in bulk
+
+        Parameters
+        ----------
+        id : str
+            Either the Test Suites' ID or its unique name
+
+        request : typing.Sequence[TestSuiteTestCaseBulkOperationRequest]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Yields
+        ------
+        typing.AsyncIterator[AsyncHttpResponse[typing.AsyncIterator[typing.List[TestSuiteTestCaseBulkResult]]]]
+
+        """
+        async with self._client_wrapper.httpx_client.stream(
+            f"v1/test-suites/{jsonable_encoder(id)}/test-cases-bulk",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=typing.Sequence[TestSuiteTestCaseBulkOperationRequest], direction="write"
+            ),
+            headers={
+                "content-type": "application/x-ndjson",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        ) as _response:
+
+            async def _stream() -> AsyncHttpResponse[typing.AsyncIterator[typing.List[TestSuiteTestCaseBulkResult]]]:
+                try:
+                    if 200 <= _response.status_code < 300:
+
+                        async def _iter():
+                            async for _text in _response.aiter_lines():
+                                try:
+                                    if len(_text) == 0:
+                                        continue
+                                    yield typing.cast(
+                                        typing.List[TestSuiteTestCaseBulkResult],
+                                        parse_obj_as(
+                                            type_=typing.List[TestSuiteTestCaseBulkResult],  # type: ignore
+                                            object_=json.loads(_text),
+                                        ),
+                                    )
+                                except Exception:
+                                    pass
+                            return
+
+                        return AsyncHttpResponse(response=_response, data=_iter())
+                    await _response.aread()
+                    _response_json = _response.json()
+                except JSONDecodeError:
+                    raise ApiError(
+                        status_code=_response.status_code, headers=dict(_response.headers), body=_response.text
+                    )
+                raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+            yield await _stream()
 
     async def delete_test_suite_test_case(
         self, id: str, test_case_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -375,5 +526,5 @@ class AsyncRawTestSuitesClient:
                 return AsyncHttpResponse(response=_response, data=None)
             _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)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

vellum/client/resources/workflow_deployments/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from .types import ListWorkflowReleaseTagsRequestSource, WorkflowDeploymentsListRequestStatus
 
 __all__ = ["ListWorkflowReleaseTagsRequestSource", "WorkflowDeploymentsListRequestStatus"]