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

vellum/client/resources/workflow_sandboxes/__init__.py

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

vellum/client/resources/workflow_sandboxes/client.py

@@ -1,14 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .raw_client import RawWorkflowSandboxesClient
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
+from ...types.paginated_workflow_sandbox_example_list import PaginatedWorkflowSandboxExampleList
 from ...types.workflow_deployment_read import WorkflowDeploymentRead
+from .raw_client import AsyncRawWorkflowSandboxesClient, RawWorkflowSandboxesClient
 from .types.list_workflow_sandbox_examples_request_tag import ListWorkflowSandboxExamplesRequestTag
-from ...types.paginated_workflow_sandbox_example_list import PaginatedWorkflowSandboxExampleList
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawWorkflowSandboxesClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -90,7 +89,7 @@ class WorkflowSandboxesClient:
             workflow_id="workflow_id",
         )
         """
-        response = self._raw_client.deploy_workflow(
+        _response = self._raw_client.deploy_workflow(
             id,
             workflow_id,
             workflow_deployment_id=workflow_deployment_id,
@@ -100,7 +99,7 @@ class WorkflowSandboxesClient:
             release_description=release_description,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     def list_workflow_sandbox_examples(
         self,
@@ -145,14 +144,10 @@ 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,
+        _response = self._raw_client.list_workflow_sandbox_examples(
+            limit=limit, offset=offset, ordering=ordering, tag=tag, request_options=request_options
         )
-        return response.data
+        return _response.data
 
 
 class AsyncWorkflowSandboxesClient:
@@ -235,7 +230,7 @@ class AsyncWorkflowSandboxesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.deploy_workflow(
+        _response = await self._raw_client.deploy_workflow(
             id,
             workflow_id,
             workflow_deployment_id=workflow_deployment_id,
@@ -245,7 +240,7 @@ class AsyncWorkflowSandboxesClient:
             release_description=release_description,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     async def list_workflow_sandbox_examples(
         self,
@@ -298,11 +293,7 @@ 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,
+        _response = await self._raw_client.list_workflow_sandbox_examples(
+            limit=limit, offset=offset, ordering=ordering, tag=tag, request_options=request_options
         )
-        return response.data
+        return _response.data

vellum/client/resources/workflow_sandboxes/raw_client.py

@@ -1,18 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...types.workflow_deployment_read import WorkflowDeploymentRead
-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 .types.list_workflow_sandbox_examples_request_tag import ListWorkflowSandboxExamplesRequestTag
+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 ...types.paginated_workflow_sandbox_example_list import PaginatedWorkflowSandboxExampleList
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
+from ...types.workflow_deployment_read import WorkflowDeploymentRead
+from .types.list_workflow_sandbox_examples_request_tag import ListWorkflowSandboxExamplesRequestTag
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -95,8 +94,8 @@ class RawWorkflowSandboxesClient:
                 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 list_workflow_sandbox_examples(
         self,
@@ -155,8 +154,8 @@ class RawWorkflowSandboxesClient:
                 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)
 
 
 class AsyncRawWorkflowSandboxesClient:
@@ -236,8 +235,8 @@ class AsyncRawWorkflowSandboxesClient:
                 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 list_workflow_sandbox_examples(
         self,
@@ -296,5 +295,5 @@ class AsyncRawWorkflowSandboxesClient:
                 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)

vellum/client/resources/workflow_sandboxes/types/__init__.py

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

vellum/client/resources/workflows/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+

vellum/client/resources/workflows/client.py

@@ -1,15 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .raw_client import RawWorkflowsClient
+
+from ... import core
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
-from ...types.workflow_push_exec_config import WorkflowPushExecConfig
 from ...types.workflow_push_deployment_config_request import WorkflowPushDeploymentConfigRequest
-from ... import core
+from ...types.workflow_push_exec_config import WorkflowPushExecConfig
 from ...types.workflow_push_response import WorkflowPushResponse
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawWorkflowsClient
+from .raw_client import AsyncRawWorkflowsClient, RawWorkflowsClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -69,7 +68,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 +76,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,
@@ -116,7 +115,7 @@ class WorkflowsClient:
         WorkflowPushResponse
 
         """
-        response = self._raw_client.push(
+        _response = self._raw_client.push(
             exec_config=exec_config,
             workflow_sandbox_id=workflow_sandbox_id,
             deployment_config=deployment_config,
@@ -125,7 +124,7 @@ class WorkflowsClient:
             strict=strict,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
 
 class AsyncWorkflowsClient:
@@ -178,7 +177,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 +185,9 @@ class AsyncWorkflowsClient:
             include_sandbox=include_sandbox,
             strict=strict,
             request_options=request_options,
-        )
-        return response.data
+        ) as r:
+            async for _chunk in r.data:
+                yield _chunk
 
     async def push(
         self,
@@ -225,7 +225,7 @@ class AsyncWorkflowsClient:
         WorkflowPushResponse
 
         """
-        response = await self._raw_client.push(
+        _response = await self._raw_client.push(
             exec_config=exec_config,
             workflow_sandbox_id=workflow_sandbox_id,
             deployment_config=deployment_config,
@@ -234,4 +234,4 @@ class AsyncWorkflowsClient:
             strict=strict,
             request_options=request_options,
         )
-        return response.data
+        return _response.data

vellum/client/resources/workflows/raw_client.py

@@ -1,20 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import contextlib
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...core.jsonable_encoder import jsonable_encoder
-from ...errors.bad_request_error import BadRequestError
-from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
+
+from ... import core
 from ...core.api_error import ApiError
-from ...types.workflow_push_exec_config import WorkflowPushExecConfig
+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 ...errors.bad_request_error import BadRequestError
 from ...types.workflow_push_deployment_config_request import WorkflowPushDeploymentConfigRequest
-from ... import core
+from ...types.workflow_push_exec_config import WorkflowPushExecConfig
 from ...types.workflow_push_response import WorkflowPushResponse
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -24,6 +24,7 @@ class RawWorkflowsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    @contextlib.contextmanager
     def pull(
         self,
         id: str,
@@ -34,7 +35,7 @@ class RawWorkflowsClient:
         include_sandbox: typing.Optional[bool] = None,
         strict: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[typing.Iterator[bytes]]:
+    ) -> typing.Iterator[HttpResponse[typing.Iterator[bytes]]]:
         """
         Parameters
         ----------
@@ -56,7 +57,7 @@ class RawWorkflowsClient:
 
         Returns
         -------
-        HttpResponse[typing.Iterator[bytes]]
+        typing.Iterator[HttpResponse[typing.Iterator[bytes]]]
 
         """
         with self._client_wrapper.httpx_client.stream(
@@ -72,27 +73,34 @@ class RawWorkflowsClient:
             },
             request_options=request_options,
         ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
-                    return HttpResponse(
-                        response=_response, data=(_chunk for _chunk in _response.iter_bytes(chunk_size=_chunk_size))
-                    )
-                _response.read()
-                if _response.status_code == 400:
-                    raise BadRequestError(
-                        typing.cast(
-                            typing.Optional[typing.Any],
-                            parse_obj_as(
-                                type_=typing.Optional[typing.Any],  # type: ignore
-                                object_=_response.json(),
+
+            def _stream() -> HttpResponse[typing.Iterator[bytes]]:
+                try:
+                    if 200 <= _response.status_code < 300:
+                        _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
+                        return HttpResponse(
+                            response=_response, data=(_chunk for _chunk in _response.iter_bytes(chunk_size=_chunk_size))
+                        )
+                    _response.read()
+                    if _response.status_code == 400:
+                        raise BadRequestError(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
+                                typing.Optional[typing.Any],
+                                parse_obj_as(
+                                    type_=typing.Optional[typing.Any],  # type: ignore
+                                    object_=_response.json(),
+                                ),
                             ),
                         )
+                    _response_json = _response.json()
+                except JSONDecodeError:
+                    raise ApiError(
+                        status_code=_response.status_code, headers=dict(_response.headers), body=_response.text
                     )
-                _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_json)
+
+            yield _stream()
 
     def push(
         self,
@@ -142,10 +150,11 @@ class RawWorkflowsClient:
                 "strict": strict,
             },
             files={
-                "artifact": artifact,
+                **({"artifact": artifact} if artifact is not None else {}),
             },
             request_options=request_options,
             omit=OMIT,
+            force_multipart=True,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -159,14 +168,15 @@ class RawWorkflowsClient:
                 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)
 
 
 class AsyncRawWorkflowsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    @contextlib.asynccontextmanager
     async def pull(
         self,
         id: str,
@@ -177,7 +187,7 @@ class AsyncRawWorkflowsClient:
         include_sandbox: typing.Optional[bool] = None,
         strict: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[typing.AsyncIterator[bytes]]:
+    ) -> typing.AsyncIterator[AsyncHttpResponse[typing.AsyncIterator[bytes]]]:
         """
         Parameters
         ----------
@@ -199,7 +209,7 @@ class AsyncRawWorkflowsClient:
 
         Returns
         -------
-        AsyncHttpResponse[typing.AsyncIterator[bytes]]
+        typing.AsyncIterator[AsyncHttpResponse[typing.AsyncIterator[bytes]]]
 
         """
         async with self._client_wrapper.httpx_client.stream(
@@ -215,28 +225,35 @@ class AsyncRawWorkflowsClient:
             },
             request_options=request_options,
         ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
-                    return AsyncHttpResponse(
-                        response=_response,
-                        data=(_chunk async for _chunk in _response.aiter_bytes(chunk_size=_chunk_size)),
-                    )
-                await _response.aread()
-                if _response.status_code == 400:
-                    raise BadRequestError(
-                        typing.cast(
-                            typing.Optional[typing.Any],
-                            parse_obj_as(
-                                type_=typing.Optional[typing.Any],  # type: ignore
-                                object_=_response.json(),
+
+            async def _stream() -> AsyncHttpResponse[typing.AsyncIterator[bytes]]:
+                try:
+                    if 200 <= _response.status_code < 300:
+                        _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
+                        return AsyncHttpResponse(
+                            response=_response,
+                            data=(_chunk async for _chunk in _response.aiter_bytes(chunk_size=_chunk_size)),
+                        )
+                    await _response.aread()
+                    if _response.status_code == 400:
+                        raise BadRequestError(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
+                                typing.Optional[typing.Any],
+                                parse_obj_as(
+                                    type_=typing.Optional[typing.Any],  # type: ignore
+                                    object_=_response.json(),
+                                ),
                             ),
                         )
+                    _response_json = _response.json()
+                except JSONDecodeError:
+                    raise ApiError(
+                        status_code=_response.status_code, headers=dict(_response.headers), body=_response.text
                     )
-                _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_json)
+
+            yield await _stream()
 
     async def push(
         self,
@@ -286,10 +303,11 @@ class AsyncRawWorkflowsClient:
                 "strict": strict,
             },
             files={
-                "artifact": artifact,
+                **({"artifact": artifact} if artifact is not None else {}),
             },
             request_options=request_options,
             omit=OMIT,
+            force_multipart=True,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -303,5 +321,5 @@ class AsyncRawWorkflowsClient:
                 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)

vellum/client/resources/workspace_secrets/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+

vellum/client/resources/workspace_secrets/client.py

@@ -1,12 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .raw_client import RawWorkspaceSecretsClient
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.workspace_secret_read import WorkspaceSecretRead
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawWorkspaceSecretsClient
+from .raw_client import AsyncRawWorkspaceSecretsClient, RawWorkspaceSecretsClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -60,11 +59,8 @@ class WorkspaceSecretsClient:
             id="id",
         )
         """
-        response = self._raw_client.retrieve(
-            id,
-            request_options=request_options,
-        )
-        return response.data
+        _response = self._raw_client.retrieve(id, request_options=request_options)
+        return _response.data
 
     def partial_update(
         self,
@@ -106,13 +102,8 @@ class WorkspaceSecretsClient:
             id="id",
         )
         """
-        response = self._raw_client.partial_update(
-            id,
-            label=label,
-            value=value,
-            request_options=request_options,
-        )
-        return response.data
+        _response = self._raw_client.partial_update(id, label=label, value=value, request_options=request_options)
+        return _response.data
 
 
 class AsyncWorkspaceSecretsClient:
@@ -169,11 +160,8 @@ class AsyncWorkspaceSecretsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve(
-            id,
-            request_options=request_options,
-        )
-        return response.data
+        _response = await self._raw_client.retrieve(id, request_options=request_options)
+        return _response.data
 
     async def partial_update(
         self,
@@ -223,10 +211,5 @@ class AsyncWorkspaceSecretsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.partial_update(
-            id,
-            label=label,
-            value=value,
-            request_options=request_options,
-        )
-        return response.data
+        _response = await self._raw_client.partial_update(id, label=label, value=value, request_options=request_options)
+        return _response.data

vellum/client/resources/workspace_secrets/raw_client.py

@@ -1,16 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...types.workspace_secret_read import WorkspaceSecretRead
-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
-from ...core.http_response import AsyncHttpResponse
+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 ...types.workspace_secret_read import WorkspaceSecretRead
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -57,8 +56,8 @@ class RawWorkspaceSecretsClient:
                 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 partial_update(
         self,
@@ -114,8 +113,8 @@ class RawWorkspaceSecretsClient:
                 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)
 
 
 class AsyncRawWorkspaceSecretsClient:
@@ -159,8 +158,8 @@ class AsyncRawWorkspaceSecretsClient:
                 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 partial_update(
         self,
@@ -216,5 +215,5 @@ class AsyncRawWorkspaceSecretsClient:
                 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)

vellum/client/resources/workspaces/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+