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

vellum/client/resources/test_suite_runs/client.py

@@ -120,10 +120,7 @@ class TestSuiteRunsClient:
             id="id",
         )
         """
-        response = self._raw_client.retrieve(
-            id,
-            request_options=request_options,
-        )
+        response = self._raw_client.retrieve(id, request_options=request_options)
         return response.data
 
     def list_executions(
@@ -174,11 +171,7 @@ class TestSuiteRunsClient:
         )
         """
         response = self._raw_client.list_executions(
-            id,
-            expand=expand,
-            limit=limit,
-            offset=offset,
-            request_options=request_options,
+            id, expand=expand, limit=limit, offset=offset, request_options=request_options
         )
         return response.data
 
@@ -301,10 +294,7 @@ class AsyncTestSuiteRunsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve(
-            id,
-            request_options=request_options,
-        )
+        response = await self._raw_client.retrieve(id, request_options=request_options)
         return response.data
 
     async def list_executions(
@@ -363,10 +353,6 @@ class AsyncTestSuiteRunsClient:
         asyncio.run(main())
         """
         response = await self._raw_client.list_executions(
-            id,
-            expand=expand,
-            limit=limit,
-            offset=offset,
-            request_options=request_options,
+            id, expand=expand, limit=limit, offset=offset, request_options=request_options
         )
         return response.data

vellum/client/resources/test_suites/client.py

@@ -9,12 +9,6 @@ from ...types.named_test_case_variable_value_request import NamedTestCaseVariabl
 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
 
@@ -84,10 +78,7 @@ class TestSuitesClient:
         )
         """
         response = self._raw_client.list_test_suite_test_cases(
-            id,
-            limit=limit,
-            offset=offset,
-            request_options=request_options,
+            id, limit=limit, offset=offset, request_options=request_options
         )
         return response.data
 
@@ -266,37 +257,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,11 +294,7 @@ 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,
-        )
+        response = self._raw_client.delete_test_suite_test_case(id, test_case_id, request_options=request_options)
         return response.data
 
 
@@ -406,10 +364,7 @@ class AsyncTestSuitesClient:
         asyncio.run(main())
         """
         response = await self._raw_client.list_test_suite_test_cases(
-            id,
-            limit=limit,
-            offset=offset,
-            request_options=request_options,
+            id, limit=limit, offset=offset, request_options=request_options
         )
         return response.data
 
@@ -604,37 +559,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 data in r.data:
+                yield data
 
     async def delete_test_suite_test_case(
         self, id: str, test_case_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -678,9 +607,5 @@ 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

vellum/client/resources/test_suites/raw_client.py

@@ -12,6 +12,10 @@ from ...core.api_error import ApiError
 from ...types.named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
 from ...types.test_suite_test_case import TestSuiteTestCase
 from ...core.serialization import convert_and_respect_annotation_metadata
+from ...types.test_suite_test_case_bulk_operation_request import TestSuiteTestCaseBulkOperationRequest
+from ...types.test_suite_test_case_bulk_result import TestSuiteTestCaseBulkResult
+import json
+import contextlib
 from ...core.client_wrapper import AsyncClientWrapper
 from ...core.http_response import AsyncHttpResponse
 
@@ -163,6 +167,72 @@ class RawTestSuitesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, 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"
+            ),
+            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, body=_response.text)
+                raise ApiError(status_code=_response.status_code, body=_response_json)
+
+            yield stream()
+
     def delete_test_suite_test_case(
         self, id: str, test_case_id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[None]:
@@ -343,6 +413,72 @@ class AsyncRawTestSuitesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, 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"
+            ),
+            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, body=_response.text)
+                raise ApiError(status_code=_response.status_code, 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
     ) -> AsyncHttpResponse[None]:

vellum/client/resources/workflow_deployments/client.py

@@ -40,7 +40,7 @@ class WorkflowDeploymentsClient:
         """
         return self._raw_client
 
-    def list_(
+    def list(
         self,
         *,
         limit: typing.Optional[int] = None,
@@ -85,11 +85,7 @@ class WorkflowDeploymentsClient:
         client.workflow_deployments.list()
         """
         response = self._raw_client.list(
-            limit=limit,
-            offset=offset,
-            ordering=ordering,
-            status=status,
-            request_options=request_options,
+            limit=limit, offset=offset, ordering=ordering, status=status, request_options=request_options
         )
         return response.data
 
@@ -122,10 +118,7 @@ class WorkflowDeploymentsClient:
             id="id",
         )
         """
-        response = self._raw_client.retrieve(
-            id,
-            request_options=request_options,
-        )
+        response = self._raw_client.retrieve(id, request_options=request_options)
         return response.data
 
     def list_workflow_deployment_event_executions(
@@ -171,11 +164,7 @@ class WorkflowDeploymentsClient:
         )
         """
         response = self._raw_client.list_workflow_deployment_event_executions(
-            id,
-            filters=filters,
-            limit=limit,
-            offset=offset,
-            request_options=request_options,
+            id, filters=filters, limit=limit, offset=offset, request_options=request_options
         )
         return response.data
 
@@ -211,9 +200,7 @@ class WorkflowDeploymentsClient:
         )
         """
         response = self._raw_client.workflow_deployment_event_execution(
-            execution_id,
-            id,
-            request_options=request_options,
+            execution_id, id, request_options=request_options
         )
         return response.data
 
@@ -254,9 +241,7 @@ class WorkflowDeploymentsClient:
         )
         """
         response = self._raw_client.workflow_deployment_history_item_retrieve(
-            history_id_or_release_tag,
-            id,
-            request_options=request_options,
+            history_id_or_release_tag, id, request_options=request_options
         )
         return response.data
 
@@ -310,12 +295,7 @@ class WorkflowDeploymentsClient:
         )
         """
         response = self._raw_client.list_workflow_release_tags(
-            id,
-            limit=limit,
-            offset=offset,
-            ordering=ordering,
-            source=source,
-            request_options=request_options,
+            id, limit=limit, offset=offset, ordering=ordering, source=source, request_options=request_options
         )
         return response.data
 
@@ -354,11 +334,7 @@ class WorkflowDeploymentsClient:
             name="name",
         )
         """
-        response = self._raw_client.retrieve_workflow_release_tag(
-            id,
-            name,
-            request_options=request_options,
-        )
+        response = self._raw_client.retrieve_workflow_release_tag(id, name, request_options=request_options)
         return response.data
 
     def update_workflow_release_tag(
@@ -405,10 +381,7 @@ class WorkflowDeploymentsClient:
         )
         """
         response = self._raw_client.update_workflow_release_tag(
-            id,
-            name,
-            history_item_id=history_item_id,
-            request_options=request_options,
+            id, name, history_item_id=history_item_id, request_options=request_options
         )
         return response.data
 
@@ -448,9 +421,7 @@ class WorkflowDeploymentsClient:
         )
         """
         response = self._raw_client.retrieve_workflow_deployment_release(
-            id,
-            release_id_or_release_tag,
-            request_options=request_options,
+            id, release_id_or_release_tag, request_options=request_options
         )
         return response.data
 
@@ -470,7 +441,7 @@ class AsyncWorkflowDeploymentsClient:
         """
         return self._raw_client
 
-    async def list_(
+    async def list(
         self,
         *,
         limit: typing.Optional[int] = None,
@@ -523,11 +494,7 @@ class AsyncWorkflowDeploymentsClient:
         asyncio.run(main())
         """
         response = await self._raw_client.list(
-            limit=limit,
-            offset=offset,
-            ordering=ordering,
-            status=status,
-            request_options=request_options,
+            limit=limit, offset=offset, ordering=ordering, status=status, request_options=request_options
         )
         return response.data
 
@@ -570,10 +537,7 @@ class AsyncWorkflowDeploymentsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve(
-            id,
-            request_options=request_options,
-        )
+        response = await self._raw_client.retrieve(id, request_options=request_options)
         return response.data
 
     async def list_workflow_deployment_event_executions(
@@ -627,11 +591,7 @@ class AsyncWorkflowDeploymentsClient:
         asyncio.run(main())
         """
         response = await self._raw_client.list_workflow_deployment_event_executions(
-            id,
-            filters=filters,
-            limit=limit,
-            offset=offset,
-            request_options=request_options,
+            id, filters=filters, limit=limit, offset=offset, request_options=request_options
         )
         return response.data
 
@@ -675,9 +635,7 @@ class AsyncWorkflowDeploymentsClient:
         asyncio.run(main())
         """
         response = await self._raw_client.workflow_deployment_event_execution(
-            execution_id,
-            id,
-            request_options=request_options,
+            execution_id, id, request_options=request_options
         )
         return response.data
 
@@ -726,9 +684,7 @@ class AsyncWorkflowDeploymentsClient:
         asyncio.run(main())
         """
         response = await self._raw_client.workflow_deployment_history_item_retrieve(
-            history_id_or_release_tag,
-            id,
-            request_options=request_options,
+            history_id_or_release_tag, id, request_options=request_options
         )
         return response.data
 
@@ -790,12 +746,7 @@ class AsyncWorkflowDeploymentsClient:
         asyncio.run(main())
         """
         response = await self._raw_client.list_workflow_release_tags(
-            id,
-            limit=limit,
-            offset=offset,
-            ordering=ordering,
-            source=source,
-            request_options=request_options,
+            id, limit=limit, offset=offset, ordering=ordering, source=source, request_options=request_options
         )
         return response.data
 
@@ -842,11 +793,7 @@ class AsyncWorkflowDeploymentsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve_workflow_release_tag(
-            id,
-            name,
-            request_options=request_options,
-        )
+        response = await self._raw_client.retrieve_workflow_release_tag(id, name, request_options=request_options)
         return response.data
 
     async def update_workflow_release_tag(
@@ -901,10 +848,7 @@ class AsyncWorkflowDeploymentsClient:
         asyncio.run(main())
         """
         response = await self._raw_client.update_workflow_release_tag(
-            id,
-            name,
-            history_item_id=history_item_id,
-            request_options=request_options,
+            id, name, history_item_id=history_item_id, request_options=request_options
         )
         return response.data
 
@@ -952,8 +896,6 @@ class AsyncWorkflowDeploymentsClient:
         asyncio.run(main())
         """
         response = await self._raw_client.retrieve_workflow_deployment_release(
-            id,
-            release_id_or_release_tag,
-            request_options=request_options,
+            id, release_id_or_release_tag, request_options=request_options
         )
         return response.data

vellum/client/resources/workflow_executions/client.py

@@ -56,10 +56,7 @@ class WorkflowExecutionsClient:
             execution_id="execution_id",
         )
         """
-        response = self._raw_client.retrieve_workflow_execution_detail(
-            execution_id,
-            request_options=request_options,
-        )
+        response = self._raw_client.retrieve_workflow_execution_detail(execution_id, request_options=request_options)
         return response.data
 
 
@@ -115,7 +112,6 @@ class AsyncWorkflowExecutionsClient:
         asyncio.run(main())
         """
         response = await self._raw_client.retrieve_workflow_execution_detail(
-            execution_id,
-            request_options=request_options,
+            execution_id, request_options=request_options
         )
         return response.data

vellum/client/resources/workflow_sandboxes/client.py

@@ -146,11 +146,7 @@ class WorkflowSandboxesClient:
         client.workflow_sandboxes.list_workflow_sandbox_examples()
         """
         response = self._raw_client.list_workflow_sandbox_examples(
-            limit=limit,
-            offset=offset,
-            ordering=ordering,
-            tag=tag,
-            request_options=request_options,
+            limit=limit, offset=offset, ordering=ordering, tag=tag, request_options=request_options
         )
         return response.data
 
@@ -299,10 +295,6 @@ class AsyncWorkflowSandboxesClient:
         asyncio.run(main())
         """
         response = await self._raw_client.list_workflow_sandbox_examples(
-            limit=limit,
-            offset=offset,
-            ordering=ordering,
-            tag=tag,
-            request_options=request_options,
+            limit=limit, offset=offset, ordering=ordering, tag=tag, request_options=request_options
         )
         return response.data

vellum/client/resources/workflows/client.py

@@ -69,7 +69,7 @@ class WorkflowsClient:
         typing.Iterator[bytes]
 
         """
-        response = self._raw_client.pull(
+        with self._raw_client.pull(
             id,
             exclude_code=exclude_code,
             exclude_display=exclude_display,
@@ -77,8 +77,8 @@ class WorkflowsClient:
             include_sandbox=include_sandbox,
             strict=strict,
             request_options=request_options,
-        )
-        return response.data
+        ) as r:
+            yield from r.data
 
     def push(
         self,
@@ -178,7 +178,7 @@ class AsyncWorkflowsClient:
         typing.AsyncIterator[bytes]
 
         """
-        response = await self._raw_client.pull(
+        async with self._raw_client.pull(
             id,
             exclude_code=exclude_code,
             exclude_display=exclude_display,
@@ -186,8 +186,9 @@ class AsyncWorkflowsClient:
             include_sandbox=include_sandbox,
             strict=strict,
             request_options=request_options,
-        )
-        return response.data
+        ) as r:
+            async for data in r.data:
+                yield data
 
     async def push(
         self,