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

vellum/client/resources/workflow_deployments/raw_client.py

@@ -1,25 +1,24 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .types.workflow_deployments_list_request_status import WorkflowDeploymentsListRequestStatus
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...types.paginated_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploymentList
-from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
+
 from ...core.api_error import ApiError
-from ...types.workflow_deployment_read import WorkflowDeploymentRead
+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_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploymentList
+from ...types.paginated_workflow_release_tag_read_list import PaginatedWorkflowReleaseTagReadList
 from ...types.workflow_deployment_event_executions_response import WorkflowDeploymentEventExecutionsResponse
-from ...types.workflow_event_execution_read import WorkflowEventExecutionRead
 from ...types.workflow_deployment_history_item import WorkflowDeploymentHistoryItem
-from .types.list_workflow_release_tags_request_source import ListWorkflowReleaseTagsRequestSource
-from ...types.paginated_workflow_release_tag_read_list import PaginatedWorkflowReleaseTagReadList
-from ...types.workflow_release_tag_read import WorkflowReleaseTagRead
+from ...types.workflow_deployment_read import WorkflowDeploymentRead
 from ...types.workflow_deployment_release import WorkflowDeploymentRelease
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
+from ...types.workflow_event_execution_read import WorkflowEventExecutionRead
+from ...types.workflow_release_tag_read import WorkflowReleaseTagRead
+from .types.list_workflow_release_tags_request_source import ListWorkflowReleaseTagsRequestSource
+from .types.workflow_deployments_list_request_status import WorkflowDeploymentsListRequestStatus
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -87,8 +86,8 @@ class RawWorkflowDeploymentsClient:
                 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 retrieve(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -127,8 +126,8 @@ class RawWorkflowDeploymentsClient:
                 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_deployment_event_executions(
         self,
@@ -183,8 +182,8 @@ class RawWorkflowDeploymentsClient:
                 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 workflow_deployment_event_execution(
         self, execution_id: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -222,8 +221,8 @@ class RawWorkflowDeploymentsClient:
                 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 workflow_deployment_history_item_retrieve(
         self, history_id_or_release_tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -266,8 +265,8 @@ class RawWorkflowDeploymentsClient:
                 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_release_tags(
         self,
@@ -330,8 +329,8 @@ class RawWorkflowDeploymentsClient:
                 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 retrieve_workflow_release_tag(
         self, id: str, name: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -373,8 +372,8 @@ class RawWorkflowDeploymentsClient:
                 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 update_workflow_release_tag(
         self,
@@ -431,8 +430,8 @@ class RawWorkflowDeploymentsClient:
                 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 retrieve_workflow_deployment_release(
         self, id: str, release_id_or_release_tag: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -474,8 +473,8 @@ class RawWorkflowDeploymentsClient:
                 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 AsyncRawWorkflowDeploymentsClient:
@@ -540,8 +539,8 @@ class AsyncRawWorkflowDeploymentsClient:
                 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 retrieve(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -580,8 +579,8 @@ class AsyncRawWorkflowDeploymentsClient:
                 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_deployment_event_executions(
         self,
@@ -636,8 +635,8 @@ class AsyncRawWorkflowDeploymentsClient:
                 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 workflow_deployment_event_execution(
         self, execution_id: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -675,8 +674,8 @@ class AsyncRawWorkflowDeploymentsClient:
                 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 workflow_deployment_history_item_retrieve(
         self, history_id_or_release_tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -719,8 +718,8 @@ class AsyncRawWorkflowDeploymentsClient:
                 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_release_tags(
         self,
@@ -783,8 +782,8 @@ class AsyncRawWorkflowDeploymentsClient:
                 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 retrieve_workflow_release_tag(
         self, id: str, name: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -826,8 +825,8 @@ class AsyncRawWorkflowDeploymentsClient:
                 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 update_workflow_release_tag(
         self,
@@ -884,8 +883,8 @@ class AsyncRawWorkflowDeploymentsClient:
                 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 retrieve_workflow_deployment_release(
         self, id: str, release_id_or_release_tag: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -927,5 +926,5 @@ class AsyncRawWorkflowDeploymentsClient:
                 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_deployments/types/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from .list_workflow_release_tags_request_source import ListWorkflowReleaseTagsRequestSource
 from .workflow_deployments_list_request_status import WorkflowDeploymentsListRequestStatus
 

vellum/client/resources/workflow_executions/__init__.py

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

vellum/client/resources/workflow_executions/client.py

@@ -1,12 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ...core.client_wrapper import SyncClientWrapper
-from .raw_client import RawWorkflowExecutionsClient
 import typing
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.workflow_execution_detail import WorkflowExecutionDetail
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawWorkflowExecutionsClient
+from .raw_client import AsyncRawWorkflowExecutionsClient, RawWorkflowExecutionsClient
 
 
 class WorkflowExecutionsClient:
@@ -56,8 +55,8 @@ class WorkflowExecutionsClient:
             execution_id="execution_id",
         )
         """
-        response = self._raw_client.retrieve_workflow_execution_detail(execution_id, request_options=request_options)
-        return response.data
+        _response = self._raw_client.retrieve_workflow_execution_detail(execution_id, request_options=request_options)
+        return _response.data
 
 
 class AsyncWorkflowExecutionsClient:
@@ -111,7 +110,7 @@ class AsyncWorkflowExecutionsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve_workflow_execution_detail(
+        _response = await self._raw_client.retrieve_workflow_execution_detail(
             execution_id, request_options=request_options
         )
-        return response.data
+        return _response.data

vellum/client/resources/workflow_executions/raw_client.py

@@ -1,16 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ...core.client_wrapper import SyncClientWrapper
 import typing
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...types.workflow_execution_detail import WorkflowExecutionDetail
-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.workflow_execution_detail import WorkflowExecutionDetail
 
 
 class RawWorkflowExecutionsClient:
@@ -51,8 +50,8 @@ class RawWorkflowExecutionsClient:
                 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 AsyncRawWorkflowExecutionsClient:
@@ -93,5 +92,5 @@ class AsyncRawWorkflowExecutionsClient:
                 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/__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,10 +144,10 @@ class WorkflowSandboxesClient:
         )
         client.workflow_sandboxes.list_workflow_sandbox_examples()
         """
-        response = self._raw_client.list_workflow_sandbox_examples(
+        _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:
@@ -231,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,
@@ -241,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,
@@ -294,7 +293,7 @@ class AsyncWorkflowSandboxesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.list_workflow_sandbox_examples(
+        _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, ...)
@@ -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:
@@ -187,8 +186,8 @@ class AsyncWorkflowsClient:
             strict=strict,
             request_options=request_options,
         ) as r:
-            async for data in r.data:
-                yield data
+            async for _chunk in r.data:
+                yield _chunk
 
     async def push(
         self,
@@ -226,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,
@@ -235,4 +234,4 @@ class AsyncWorkflowsClient:
             strict=strict,
             request_options=request_options,
         )
-        return response.data
+        return _response.data