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

vellum/client/resources/container_images/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.paginated_container_image_read_list import PaginatedContainerImageReadList
-from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
+
 from ...core.api_error import ApiError
-from ...types.container_image_read import ContainerImageRead
+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.container_image_read import ContainerImageRead
 from ...types.docker_service_token import DockerServiceToken
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
+from ...types.paginated_container_image_read_list import PaginatedContainerImageReadList
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -75,8 +74,8 @@ class RawContainerImagesClient:
                 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
@@ -115,8 +114,8 @@ class RawContainerImagesClient:
                 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 docker_service_token(
         self, *, request_options: typing.Optional[RequestOptions] = None
@@ -150,8 +149,8 @@ class RawContainerImagesClient:
                 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 push_container_image(
         self,
@@ -205,8 +204,8 @@ class RawContainerImagesClient:
                 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 AsyncRawContainerImagesClient:
@@ -266,8 +265,8 @@ class AsyncRawContainerImagesClient:
                 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
@@ -306,8 +305,8 @@ class AsyncRawContainerImagesClient:
                 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 docker_service_token(
         self, *, request_options: typing.Optional[RequestOptions] = None
@@ -341,8 +340,8 @@ class AsyncRawContainerImagesClient:
                 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 push_container_image(
         self,
@@ -396,5 +395,5 @@ class AsyncRawContainerImagesClient:
                 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/deployments/__init__.py

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

vellum/client/resources/deployments/client.py

@@ -1,22 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .raw_client import RawDeploymentsClient
-from .types.deployments_list_request_status import DeploymentsListRequestStatus
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
-from ...types.paginated_slim_deployment_read_list import PaginatedSlimDeploymentReadList
-from ...types.deployment_read import DeploymentRead
+from ...types.compile_prompt_deployment_expand_meta_request import CompilePromptDeploymentExpandMetaRequest
 from ...types.deployment_history_item import DeploymentHistoryItem
-from .types.list_deployment_release_tags_request_source import ListDeploymentReleaseTagsRequestSource
-from ...types.paginated_deployment_release_tag_read_list import PaginatedDeploymentReleaseTagReadList
+from ...types.deployment_provider_payload_response import DeploymentProviderPayloadResponse
+from ...types.deployment_read import DeploymentRead
 from ...types.deployment_release_tag_read import DeploymentReleaseTagRead
-from ...types.prompt_deployment_release import PromptDeploymentRelease
+from ...types.paginated_deployment_release_tag_read_list import PaginatedDeploymentReleaseTagReadList
+from ...types.paginated_slim_deployment_read_list import PaginatedSlimDeploymentReadList
 from ...types.prompt_deployment_input_request import PromptDeploymentInputRequest
-from ...types.compile_prompt_deployment_expand_meta_request import CompilePromptDeploymentExpandMetaRequest
-from ...types.deployment_provider_payload_response import DeploymentProviderPayloadResponse
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawDeploymentsClient
+from ...types.prompt_deployment_release import PromptDeploymentRelease
+from .raw_client import AsyncRawDeploymentsClient, RawDeploymentsClient
+from .types.deployments_list_request_status import DeploymentsListRequestStatus
+from .types.list_deployment_release_tags_request_source import ListDeploymentReleaseTagsRequestSource
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -41,7 +40,7 @@ class DeploymentsClient:
         """
         return self._raw_client
 
-    def list_(
+    def list(
         self,
         *,
         limit: typing.Optional[int] = None,
@@ -85,14 +84,10 @@ class DeploymentsClient:
         )
         client.deployments.list()
         """
-        response = self._raw_client.list(
-            limit=limit,
-            offset=offset,
-            ordering=ordering,
-            status=status,
-            request_options=request_options,
+        _response = self._raw_client.list(
+            limit=limit, offset=offset, ordering=ordering, status=status, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> DeploymentRead:
         """
@@ -123,11 +118,8 @@ class DeploymentsClient:
             id="id",
         )
         """
-        response = self._raw_client.retrieve(
-            id,
-            request_options=request_options,
-        )
-        return response.data
+        _response = self._raw_client.retrieve(id, request_options=request_options)
+        return _response.data
 
     def deployment_history_item_retrieve(
         self, history_id_or_release_tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -165,12 +157,10 @@ class DeploymentsClient:
             id="id",
         )
         """
-        response = self._raw_client.deployment_history_item_retrieve(
-            history_id_or_release_tag,
-            id,
-            request_options=request_options,
+        _response = self._raw_client.deployment_history_item_retrieve(
+            history_id_or_release_tag, id, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     def list_deployment_release_tags(
         self,
@@ -221,15 +211,10 @@ class DeploymentsClient:
             id="id",
         )
         """
-        response = self._raw_client.list_deployment_release_tags(
-            id,
-            limit=limit,
-            offset=offset,
-            ordering=ordering,
-            source=source,
-            request_options=request_options,
+        _response = self._raw_client.list_deployment_release_tags(
+            id, limit=limit, offset=offset, ordering=ordering, source=source, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     def retrieve_deployment_release_tag(
         self, id: str, name: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -266,12 +251,8 @@ class DeploymentsClient:
             name="name",
         )
         """
-        response = self._raw_client.retrieve_deployment_release_tag(
-            id,
-            name,
-            request_options=request_options,
-        )
-        return response.data
+        _response = self._raw_client.retrieve_deployment_release_tag(id, name, request_options=request_options)
+        return _response.data
 
     def update_deployment_release_tag(
         self,
@@ -316,13 +297,10 @@ class DeploymentsClient:
             name="name",
         )
         """
-        response = self._raw_client.update_deployment_release_tag(
-            id,
-            name,
-            history_item_id=history_item_id,
-            request_options=request_options,
+        _response = self._raw_client.update_deployment_release_tag(
+            id, name, history_item_id=history_item_id, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     def retrieve_prompt_deployment_release(
         self, id: str, release_id_or_release_tag: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -359,12 +337,10 @@ class DeploymentsClient:
             release_id_or_release_tag="release_id_or_release_tag",
         )
         """
-        response = self._raw_client.retrieve_prompt_deployment_release(
-            id,
-            release_id_or_release_tag,
-            request_options=request_options,
+        _response = self._raw_client.retrieve_prompt_deployment_release(
+            id, release_id_or_release_tag, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     def retrieve_provider_payload(
         self,
@@ -433,7 +409,7 @@ class DeploymentsClient:
             ],
         )
         """
-        response = self._raw_client.retrieve_provider_payload(
+        _response = self._raw_client.retrieve_provider_payload(
             inputs=inputs,
             deployment_id=deployment_id,
             deployment_name=deployment_name,
@@ -441,7 +417,7 @@ class DeploymentsClient:
             expand_meta=expand_meta,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
 
 class AsyncDeploymentsClient:
@@ -459,7 +435,7 @@ class AsyncDeploymentsClient:
         """
         return self._raw_client
 
-    async def list_(
+    async def list(
         self,
         *,
         limit: typing.Optional[int] = None,
@@ -511,14 +487,10 @@ class AsyncDeploymentsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.list(
-            limit=limit,
-            offset=offset,
-            ordering=ordering,
-            status=status,
-            request_options=request_options,
+        _response = await self._raw_client.list(
+            limit=limit, offset=offset, ordering=ordering, status=status, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> DeploymentRead:
         """
@@ -557,11 +529,8 @@ class AsyncDeploymentsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve(
-            id,
-            request_options=request_options,
-        )
-        return response.data
+        _response = await self._raw_client.retrieve(id, request_options=request_options)
+        return _response.data
 
     async def deployment_history_item_retrieve(
         self, history_id_or_release_tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -607,12 +576,10 @@ class AsyncDeploymentsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.deployment_history_item_retrieve(
-            history_id_or_release_tag,
-            id,
-            request_options=request_options,
+        _response = await self._raw_client.deployment_history_item_retrieve(
+            history_id_or_release_tag, id, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def list_deployment_release_tags(
         self,
@@ -671,15 +638,10 @@ class AsyncDeploymentsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.list_deployment_release_tags(
-            id,
-            limit=limit,
-            offset=offset,
-            ordering=ordering,
-            source=source,
-            request_options=request_options,
+        _response = await self._raw_client.list_deployment_release_tags(
+            id, limit=limit, offset=offset, ordering=ordering, source=source, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def retrieve_deployment_release_tag(
         self, id: str, name: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -724,12 +686,8 @@ class AsyncDeploymentsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve_deployment_release_tag(
-            id,
-            name,
-            request_options=request_options,
-        )
-        return response.data
+        _response = await self._raw_client.retrieve_deployment_release_tag(id, name, request_options=request_options)
+        return _response.data
 
     async def update_deployment_release_tag(
         self,
@@ -782,13 +740,10 @@ class AsyncDeploymentsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.update_deployment_release_tag(
-            id,
-            name,
-            history_item_id=history_item_id,
-            request_options=request_options,
+        _response = await self._raw_client.update_deployment_release_tag(
+            id, name, history_item_id=history_item_id, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def retrieve_prompt_deployment_release(
         self, id: str, release_id_or_release_tag: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -833,12 +788,10 @@ class AsyncDeploymentsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve_prompt_deployment_release(
-            id,
-            release_id_or_release_tag,
-            request_options=request_options,
+        _response = await self._raw_client.retrieve_prompt_deployment_release(
+            id, release_id_or_release_tag, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def retrieve_provider_payload(
         self,
@@ -915,7 +868,7 @@ class AsyncDeploymentsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve_provider_payload(
+        _response = await self._raw_client.retrieve_provider_payload(
             inputs=inputs,
             deployment_id=deployment_id,
             deployment_name=deployment_name,
@@ -923,4 +876,4 @@ class AsyncDeploymentsClient:
             expand_meta=expand_meta,
             request_options=request_options,
         )
-        return response.data
+        return _response.data