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

vellum/client/resources/documents/raw_client.py

@@ -1,23 +1,22 @@
 # 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.paginated_slim_document_list import PaginatedSlimDocumentList
-from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
+
+from ... import core
 from ...core.api_error import ApiError
-from ...types.document_read import DocumentRead
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.http_response import AsyncHttpResponse, HttpResponse
 from ...core.jsonable_encoder import jsonable_encoder
-from ...types.document_status import DocumentStatus
-from ... import core
-from ...types.upload_document_response import UploadDocumentResponse
+from ...core.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
 from ...errors.bad_request_error import BadRequestError
-from ...errors.not_found_error import NotFoundError
 from ...errors.internal_server_error import InternalServerError
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
+from ...errors.not_found_error import NotFoundError
+from ...types.document_read import DocumentRead
+from ...types.document_status import DocumentStatus
+from ...types.paginated_slim_document_list import PaginatedSlimDocumentList
+from ...types.upload_document_response import UploadDocumentResponse
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -90,8 +89,8 @@ class RawDocumentsClient:
                 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
@@ -129,8 +128,8 @@ class RawDocumentsClient:
                 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 destroy(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> HttpResponse[None]:
         """
@@ -158,8 +157,8 @@ class RawDocumentsClient:
                 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)
 
     def partial_update(
         self,
@@ -228,8 +227,8 @@ class RawDocumentsClient:
                 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 upload(
         self,
@@ -291,6 +290,7 @@ class RawDocumentsClient:
             },
             request_options=request_options,
             omit=OMIT,
+            force_multipart=True,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -304,38 +304,41 @@ class RawDocumentsClient:
                 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)
 
 
 class AsyncRawDocumentsClient:
@@ -405,8 +408,8 @@ class AsyncRawDocumentsClient:
                 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
@@ -444,8 +447,8 @@ class AsyncRawDocumentsClient:
                 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 destroy(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -475,8 +478,8 @@ class AsyncRawDocumentsClient:
                 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)
 
     async def partial_update(
         self,
@@ -545,8 +548,8 @@ class AsyncRawDocumentsClient:
                 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 upload(
         self,
@@ -608,6 +611,7 @@ class AsyncRawDocumentsClient:
             },
             request_options=request_options,
             omit=OMIT,
+            force_multipart=True,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -621,35 +625,38 @@ class AsyncRawDocumentsClient:
                 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)

vellum/client/resources/folder_entities/__init__.py

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

vellum/client/resources/folder_entities/client.py

@@ -1,13 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .raw_client import RawFolderEntitiesClient
-from .types.folder_entities_list_request_entity_status import FolderEntitiesListRequestEntityStatus
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.paginated_folder_entity_list import PaginatedFolderEntityList
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawFolderEntitiesClient
+from .raw_client import AsyncRawFolderEntitiesClient, RawFolderEntitiesClient
+from .types.folder_entities_list_request_entity_status import FolderEntitiesListRequestEntityStatus
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -92,7 +91,7 @@ class FolderEntitiesClient:
             parent_folder_id="parent_folder_id",
         )
         """
-        response = self._raw_client.list(
+        _response = self._raw_client.list(
             parent_folder_id=parent_folder_id,
             entity_status=entity_status,
             limit=limit,
@@ -100,7 +99,7 @@ class FolderEntitiesClient:
             ordering=ordering,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     def add_entity_to_folder(
         self, folder_id: str, *, entity_id: str, request_options: typing.Optional[RequestOptions] = None
@@ -143,10 +142,10 @@ class FolderEntitiesClient:
             entity_id="entity_id",
         )
         """
-        response = self._raw_client.add_entity_to_folder(
+        _response = self._raw_client.add_entity_to_folder(
             folder_id, entity_id=entity_id, request_options=request_options
         )
-        return response.data
+        return _response.data
 
 
 class AsyncFolderEntitiesClient:
@@ -232,7 +231,7 @@ class AsyncFolderEntitiesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.list(
+        _response = await self._raw_client.list(
             parent_folder_id=parent_folder_id,
             entity_status=entity_status,
             limit=limit,
@@ -240,7 +239,7 @@ class AsyncFolderEntitiesClient:
             ordering=ordering,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     async def add_entity_to_folder(
         self, folder_id: str, *, entity_id: str, request_options: typing.Optional[RequestOptions] = None
@@ -291,7 +290,7 @@ class AsyncFolderEntitiesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.add_entity_to_folder(
+        _response = await self._raw_client.add_entity_to_folder(
             folder_id, entity_id=entity_id, request_options=request_options
         )
-        return response.data
+        return _response.data

vellum/client/resources/folder_entities/raw_client.py

@@ -1,17 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .types.folder_entities_list_request_entity_status import FolderEntitiesListRequestEntityStatus
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...types.paginated_folder_entity_list import PaginatedFolderEntityList
-from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
+
 from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.http_response import AsyncHttpResponse, HttpResponse
 from ...core.jsonable_encoder import jsonable_encoder
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
+from ...core.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...types.paginated_folder_entity_list import PaginatedFolderEntityList
+from .types.folder_entities_list_request_entity_status import FolderEntitiesListRequestEntityStatus
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -94,8 +93,8 @@ class RawFolderEntitiesClient:
                 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 add_entity_to_folder(
         self, folder_id: str, *, entity_id: str, request_options: typing.Optional[RequestOptions] = None
@@ -143,8 +142,8 @@ class RawFolderEntitiesClient:
                 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 AsyncRawFolderEntitiesClient:
@@ -224,8 +223,8 @@ class AsyncRawFolderEntitiesClient:
                 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 add_entity_to_folder(
         self, folder_id: str, *, entity_id: str, request_options: typing.Optional[RequestOptions] = None
@@ -273,5 +272,5 @@ class AsyncRawFolderEntitiesClient:
                 return AsyncHttpResponse(response=_response, data=None)
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

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

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

vellum/client/resources/metric_definitions/__init__.py

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

vellum/client/resources/metric_definitions/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 RawMetricDefinitionsClient
-from ...types.metric_definition_input import MetricDefinitionInput
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.metric_definition_execution import MetricDefinitionExecution
 from ...types.metric_definition_history_item import MetricDefinitionHistoryItem
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawMetricDefinitionsClient
+from ...types.metric_definition_input import MetricDefinitionInput
+from .raw_client import AsyncRawMetricDefinitionsClient, RawMetricDefinitionsClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -81,10 +80,10 @@ class MetricDefinitionsClient:
             ],
         )
         """
-        response = self._raw_client.execute_metric_definition(
+        _response = self._raw_client.execute_metric_definition(
             id, inputs=inputs, release_tag=release_tag, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     def metric_definition_history_item_retrieve(
         self, history_id_or_release_tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -119,10 +118,10 @@ class MetricDefinitionsClient:
             id="id",
         )
         """
-        response = self._raw_client.metric_definition_history_item_retrieve(
+        _response = self._raw_client.metric_definition_history_item_retrieve(
             history_id_or_release_tag, id, request_options=request_options
         )
-        return response.data
+        return _response.data
 
 
 class AsyncMetricDefinitionsClient:
@@ -196,10 +195,10 @@ class AsyncMetricDefinitionsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.execute_metric_definition(
+        _response = await self._raw_client.execute_metric_definition(
             id, inputs=inputs, release_tag=release_tag, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def metric_definition_history_item_retrieve(
         self, history_id_or_release_tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -242,7 +241,7 @@ class AsyncMetricDefinitionsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.metric_definition_history_item_retrieve(
+        _response = await self._raw_client.metric_definition_history_item_retrieve(
             history_id_or_release_tag, id, request_options=request_options
         )
-        return response.data
+        return _response.data

vellum/client/resources/metric_definitions/raw_client.py

@@ -1,19 +1,18 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from ...types.metric_definition_input import MetricDefinitionInput
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...types.metric_definition_execution import MetricDefinitionExecution
-from ...core.jsonable_encoder import jsonable_encoder
-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 ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.http_response import AsyncHttpResponse, HttpResponse
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...core.serialization import convert_and_respect_annotation_metadata
+from ...types.metric_definition_execution import MetricDefinitionExecution
 from ...types.metric_definition_history_item import MetricDefinitionHistoryItem
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
+from ...types.metric_definition_input import MetricDefinitionInput
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -77,8 +76,8 @@ class RawMetricDefinitionsClient:
                 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 metric_definition_history_item_retrieve(
         self, history_id_or_release_tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -118,8 +117,8 @@ class RawMetricDefinitionsClient:
                 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 AsyncRawMetricDefinitionsClient:
@@ -180,8 +179,8 @@ class AsyncRawMetricDefinitionsClient:
                 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 metric_definition_history_item_retrieve(
         self, history_id_or_release_tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -221,5 +220,5 @@ class AsyncRawMetricDefinitionsClient:
                 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/ml_models/__init__.py

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

vellum/client/resources/ml_models/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 RawMlModelsClient
 import typing
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.ml_model_read import MlModelRead
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawMlModelsClient
+from .raw_client import AsyncRawMlModelsClient, RawMlModelsClient
 
 
 class MlModelsClient:
@@ -57,8 +56,8 @@ class MlModelsClient:
             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
 
 
 class AsyncMlModelsClient:
@@ -113,5 +112,5 @@ class AsyncMlModelsClient:
 
         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

vellum/client/resources/ml_models/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.ml_model_read import MlModelRead
-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.ml_model_read import MlModelRead
 
 
 class RawMlModelsClient:
@@ -54,8 +53,8 @@ class RawMlModelsClient:
                 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 AsyncRawMlModelsClient:
@@ -99,5 +98,5 @@ class AsyncRawMlModelsClient:
                 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/organizations/__init__.py

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

vellum/client/resources/organizations/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 RawOrganizationsClient
 import typing
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.organization_read import OrganizationRead
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawOrganizationsClient
+from .raw_client import AsyncRawOrganizationsClient, RawOrganizationsClient
 
 
 class OrganizationsClient:
@@ -52,8 +51,8 @@ class OrganizationsClient:
         )
         client.organizations.organization_identity()
         """
-        response = self._raw_client.organization_identity(request_options=request_options)
-        return response.data
+        _response = self._raw_client.organization_identity(request_options=request_options)
+        return _response.data
 
 
 class AsyncOrganizationsClient:
@@ -105,5 +104,5 @@ class AsyncOrganizationsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.organization_identity(request_options=request_options)
-        return response.data
+        _response = await self._raw_client.organization_identity(request_options=request_options)
+        return _response.data