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

vellum/client/raw_client.py

@@ -1,49 +1,48 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import contextlib
+import json
 import typing
-from .core.client_wrapper import SyncClientWrapper
-from .types.method_enum import MethodEnum
-from .types.execute_api_request_body import ExecuteApiRequestBody
-from .types.execute_api_request_headers_value import ExecuteApiRequestHeadersValue
-from .types.execute_api_request_bearer_token import ExecuteApiRequestBearerToken
-from .core.request_options import RequestOptions
-from .core.http_response import HttpResponse
-from .types.execute_api_response import ExecuteApiResponse
-from .core.serialization import convert_and_respect_annotation_metadata
-from .core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
+
 from .core.api_error import ApiError
-from .types.code_execution_runtime import CodeExecutionRuntime
-from .types.code_executor_input import CodeExecutorInput
-from .types.code_execution_package import CodeExecutionPackage
-from .types.vellum_variable_type import VellumVariableType
-from .types.code_executor_response import CodeExecutorResponse
+from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from .core.http_response import AsyncHttpResponse, HttpResponse
+from .core.pydantic_utilities import parse_obj_as
+from .core.request_options import RequestOptions
+from .core.serialization import convert_and_respect_annotation_metadata
 from .errors.bad_request_error import BadRequestError
-from .types.prompt_deployment_input_request import PromptDeploymentInputRequest
-from .types.prompt_deployment_expand_meta_request import PromptDeploymentExpandMetaRequest
-from .types.raw_prompt_execution_overrides_request import RawPromptExecutionOverridesRequest
-from .types.execute_prompt_response import ExecutePromptResponse
 from .errors.forbidden_error import ForbiddenError
-from .errors.not_found_error import NotFoundError
 from .errors.internal_server_error import InternalServerError
+from .errors.not_found_error import NotFoundError
+from .types.code_execution_package import CodeExecutionPackage
+from .types.code_execution_runtime import CodeExecutionRuntime
+from .types.code_executor_input import CodeExecutorInput
+from .types.code_executor_response import CodeExecutorResponse
+from .types.execute_api_request_bearer_token import ExecuteApiRequestBearerToken
+from .types.execute_api_request_body import ExecuteApiRequestBody
+from .types.execute_api_request_headers_value import ExecuteApiRequestHeadersValue
+from .types.execute_api_response import ExecuteApiResponse
 from .types.execute_prompt_event import ExecutePromptEvent
-import json
-import contextlib
-from .types.workflow_request_input_request import WorkflowRequestInputRequest
-from .types.workflow_expand_meta_request import WorkflowExpandMetaRequest
+from .types.execute_prompt_response import ExecutePromptResponse
 from .types.execute_workflow_response import ExecuteWorkflowResponse
-from .types.workflow_execution_event_type import WorkflowExecutionEventType
-from .types.workflow_stream_event import WorkflowStreamEvent
-from .types.generate_request import GenerateRequest
 from .types.generate_options_request import GenerateOptionsRequest
+from .types.generate_request import GenerateRequest
 from .types.generate_response import GenerateResponse
 from .types.generate_stream_response import GenerateStreamResponse
+from .types.method_enum import MethodEnum
+from .types.prompt_deployment_expand_meta_request import PromptDeploymentExpandMetaRequest
+from .types.prompt_deployment_input_request import PromptDeploymentInputRequest
+from .types.raw_prompt_execution_overrides_request import RawPromptExecutionOverridesRequest
 from .types.search_request_options_request import SearchRequestOptionsRequest
 from .types.search_response import SearchResponse
 from .types.submit_completion_actual_request import SubmitCompletionActualRequest
 from .types.submit_workflow_execution_actual_request import SubmitWorkflowExecutionActualRequest
-from .core.client_wrapper import AsyncClientWrapper
-from .core.http_response import AsyncHttpResponse
+from .types.vellum_variable_type import VellumVariableType
+from .types.workflow_execution_event_type import WorkflowExecutionEventType
+from .types.workflow_expand_meta_request import WorkflowExpandMetaRequest
+from .types.workflow_request_input_request import WorkflowRequestInputRequest
+from .types.workflow_stream_event import WorkflowStreamEvent
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -119,8 +118,8 @@ class RawVellum:
                 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 execute_code(
         self,
@@ -186,18 +185,19 @@ class RawVellum:
                 return HttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    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, 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 execute_prompt(
         self,
@@ -296,48 +296,52 @@ class RawVellum:
                 return HttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 403:
                 raise ForbiddenError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 404:
                 raise NotFoundError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    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, 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 execute_prompt_stream(
@@ -426,7 +430,7 @@ class RawVellum:
             omit=OMIT,
         ) as _response:
 
-            def stream() -> HttpResponse[typing.Iterator[ExecutePromptEvent]]:
+            def _stream() -> HttpResponse[typing.Iterator[ExecutePromptEvent]]:
                 try:
                     if 200 <= _response.status_code < 300:
 
@@ -450,50 +454,56 @@ class RawVellum:
                     _response.read()
                     if _response.status_code == 400:
                         raise BadRequestError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 403:
                         raise ForbiddenError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 404:
                         raise NotFoundError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 500:
                         raise InternalServerError(
-                            typing.cast(
+                            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, 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)
 
-            yield stream()
+            yield _stream()
 
     def execute_workflow(
         self,
@@ -576,38 +586,41 @@ class RawVellum:
                 return HttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 404:
                 raise NotFoundError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    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, 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 execute_workflow_stream(
@@ -685,7 +698,7 @@ class RawVellum:
             omit=OMIT,
         ) as _response:
 
-            def stream() -> HttpResponse[typing.Iterator[WorkflowStreamEvent]]:
+            def _stream() -> HttpResponse[typing.Iterator[WorkflowStreamEvent]]:
                 try:
                     if 200 <= _response.status_code < 300:
 
@@ -709,40 +722,45 @@ class RawVellum:
                     _response.read()
                     if _response.status_code == 400:
                         raise BadRequestError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 404:
                         raise NotFoundError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 500:
                         raise InternalServerError(
-                            typing.cast(
+                            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, 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)
 
-            yield stream()
+            yield _stream()
 
     def generate(
         self,
@@ -813,48 +831,52 @@ class RawVellum:
                 return HttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 403:
                 raise ForbiddenError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 404:
                 raise NotFoundError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    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, 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 generate_stream(
@@ -915,7 +937,7 @@ class RawVellum:
             omit=OMIT,
         ) as _response:
 
-            def stream() -> HttpResponse[typing.Iterator[GenerateStreamResponse]]:
+            def _stream() -> HttpResponse[typing.Iterator[GenerateStreamResponse]]:
                 try:
                     if 200 <= _response.status_code < 300:
 
@@ -939,50 +961,56 @@ class RawVellum:
                     _response.read()
                     if _response.status_code == 400:
                         raise BadRequestError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 403:
                         raise ForbiddenError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 404:
                         raise NotFoundError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 500:
                         raise InternalServerError(
-                            typing.cast(
+                            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, 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)
 
-            yield stream()
+            yield _stream()
 
     def search(
         self,
@@ -1053,38 +1081,41 @@ class RawVellum:
                 return HttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 404:
                 raise NotFoundError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    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, 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 submit_completion_actuals(
         self,
@@ -1137,38 +1168,41 @@ class RawVellum:
                 return HttpResponse(response=_response, data=None)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 404:
                 raise NotFoundError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    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, 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 submit_workflow_execution_actuals(
         self,
@@ -1223,8 +1257,8 @@ class RawVellum:
                 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 AsyncRawVellum:
@@ -1297,8 +1331,8 @@ class AsyncRawVellum:
                 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 execute_code(
         self,
@@ -1364,18 +1398,19 @@ class AsyncRawVellum:
                 return AsyncHttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    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, 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 execute_prompt(
         self,
@@ -1474,48 +1509,52 @@ class AsyncRawVellum:
                 return AsyncHttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 403:
                 raise ForbiddenError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 404:
                 raise NotFoundError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    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, 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 execute_prompt_stream(
@@ -1604,7 +1643,7 @@ class AsyncRawVellum:
             omit=OMIT,
         ) as _response:
 
-            async def stream() -> AsyncHttpResponse[typing.AsyncIterator[ExecutePromptEvent]]:
+            async def _stream() -> AsyncHttpResponse[typing.AsyncIterator[ExecutePromptEvent]]:
                 try:
                     if 200 <= _response.status_code < 300:
 
@@ -1628,50 +1667,56 @@ class AsyncRawVellum:
                     await _response.aread()
                     if _response.status_code == 400:
                         raise BadRequestError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 403:
                         raise ForbiddenError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 404:
                         raise NotFoundError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 500:
                         raise InternalServerError(
-                            typing.cast(
+                            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, 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)
 
-            yield await stream()
+            yield await _stream()
 
     async def execute_workflow(
         self,
@@ -1754,38 +1799,41 @@ class AsyncRawVellum:
                 return AsyncHttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 404:
                 raise NotFoundError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    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, 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 execute_workflow_stream(
@@ -1863,7 +1911,7 @@ class AsyncRawVellum:
             omit=OMIT,
         ) as _response:
 
-            async def stream() -> AsyncHttpResponse[typing.AsyncIterator[WorkflowStreamEvent]]:
+            async def _stream() -> AsyncHttpResponse[typing.AsyncIterator[WorkflowStreamEvent]]:
                 try:
                     if 200 <= _response.status_code < 300:
 
@@ -1887,40 +1935,45 @@ class AsyncRawVellum:
                     await _response.aread()
                     if _response.status_code == 400:
                         raise BadRequestError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 404:
                         raise NotFoundError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 500:
                         raise InternalServerError(
-                            typing.cast(
+                            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, 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)
 
-            yield await stream()
+            yield await _stream()
 
     async def generate(
         self,
@@ -1991,48 +2044,52 @@ class AsyncRawVellum:
                 return AsyncHttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 403:
                 raise ForbiddenError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 404:
                 raise NotFoundError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    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, 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 generate_stream(
@@ -2093,7 +2150,7 @@ class AsyncRawVellum:
             omit=OMIT,
         ) as _response:
 
-            async def stream() -> AsyncHttpResponse[typing.AsyncIterator[GenerateStreamResponse]]:
+            async def _stream() -> AsyncHttpResponse[typing.AsyncIterator[GenerateStreamResponse]]:
                 try:
                     if 200 <= _response.status_code < 300:
 
@@ -2117,50 +2174,56 @@ class AsyncRawVellum:
                     await _response.aread()
                     if _response.status_code == 400:
                         raise BadRequestError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 403:
                         raise ForbiddenError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 404:
                         raise NotFoundError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     if _response.status_code == 500:
                         raise InternalServerError(
-                            typing.cast(
+                            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, 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)
 
-            yield await stream()
+            yield await _stream()
 
     async def search(
         self,
@@ -2231,38 +2294,41 @@ class AsyncRawVellum:
                 return AsyncHttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 404:
                 raise NotFoundError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    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, 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 submit_completion_actuals(
         self,
@@ -2315,38 +2381,41 @@ class AsyncRawVellum:
                 return AsyncHttpResponse(response=_response, data=None)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 404:
                 raise NotFoundError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    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, 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 submit_workflow_execution_actuals(
         self,
@@ -2401,5 +2470,5 @@ class AsyncRawVellum:
                 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)