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

vellum/client/resources/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from . import (
     ad_hoc,
     container_images,

vellum/client/resources/ad_hoc/__init__.py

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

vellum/client/resources/ad_hoc/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 .raw_client import RawAdHocClient
-from ...types.prompt_request_input import PromptRequestInput
-from ...types.vellum_variable import VellumVariable
-from ...types.prompt_parameters import PromptParameters
-from ...types.prompt_block import PromptBlock
-from ...types.prompt_settings import PromptSettings
-from ...types.function_definition import FunctionDefinition
-from ...types.ad_hoc_expand_meta import AdHocExpandMeta
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.ad_hoc_execute_prompt_event import AdHocExecutePromptEvent
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawAdHocClient
+from ...types.ad_hoc_expand_meta import AdHocExpandMeta
+from ...types.function_definition import FunctionDefinition
+from ...types.prompt_block import PromptBlock
+from ...types.prompt_parameters import PromptParameters
+from ...types.prompt_request_input import PromptRequestInput
+from ...types.prompt_settings import PromptSettings
+from ...types.vellum_variable import VellumVariable
+from .raw_client import AsyncRawAdHocClient, RawAdHocClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -127,7 +126,7 @@ class AdHocClient:
             ],
         )
         """
-        response = self._raw_client.adhoc_execute_prompt(
+        _response = self._raw_client.adhoc_execute_prompt(
             ml_model=ml_model,
             input_values=input_values,
             input_variables=input_variables,
@@ -138,7 +137,7 @@ class AdHocClient:
             expand_meta=expand_meta,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     def adhoc_execute_prompt_stream(
         self,
@@ -357,7 +356,7 @@ class AsyncAdHocClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.adhoc_execute_prompt(
+        _response = await self._raw_client.adhoc_execute_prompt(
             ml_model=ml_model,
             input_values=input_values,
             input_variables=input_variables,
@@ -368,7 +367,7 @@ class AsyncAdHocClient:
             expand_meta=expand_meta,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     async def adhoc_execute_prompt_stream(
         self,
@@ -480,5 +479,5 @@ class AsyncAdHocClient:
             expand_meta=expand_meta,
             request_options=request_options,
         ) as r:
-            async for data in r.data:
-                yield data
+            async for _chunk in r.data:
+                yield _chunk

vellum/client/resources/ad_hoc/raw_client.py

@@ -1,28 +1,27 @@
 # 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.prompt_request_input import PromptRequestInput
-from ...types.vellum_variable import VellumVariable
-from ...types.prompt_parameters import PromptParameters
-from ...types.prompt_block import PromptBlock
-from ...types.prompt_settings import PromptSettings
-from ...types.function_definition import FunctionDefinition
-from ...types.ad_hoc_expand_meta import AdHocExpandMeta
+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.pydantic_utilities import parse_obj_as
 from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...types.ad_hoc_execute_prompt_event import AdHocExecutePromptEvent
 from ...core.serialization import convert_and_respect_annotation_metadata
-from ...core.pydantic_utilities import parse_obj_as
 from ...errors.bad_request_error import BadRequestError
 from ...errors.forbidden_error import ForbiddenError
 from ...errors.internal_server_error import InternalServerError
-from json.decoder import JSONDecodeError
-from ...core.api_error import ApiError
-import json
-import contextlib
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
+from ...types.ad_hoc_execute_prompt_event import AdHocExecutePromptEvent
+from ...types.ad_hoc_expand_meta import AdHocExpandMeta
+from ...types.function_definition import FunctionDefinition
+from ...types.prompt_block import PromptBlock
+from ...types.prompt_parameters import PromptParameters
+from ...types.prompt_request_input import PromptRequestInput
+from ...types.prompt_settings import PromptSettings
+from ...types.vellum_variable import VellumVariable
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -120,38 +119,41 @@ class RawAdHocClient:
                 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 == 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 adhoc_execute_prompt_stream(
@@ -231,7 +233,7 @@ class RawAdHocClient:
             omit=OMIT,
         ) as _response:
 
-            def stream() -> HttpResponse[typing.Iterator[AdHocExecutePromptEvent]]:
+            def _stream() -> HttpResponse[typing.Iterator[AdHocExecutePromptEvent]]:
                 try:
                     if 200 <= _response.status_code < 300:
 
@@ -255,40 +257,45 @@ class RawAdHocClient:
                     _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 == 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()
 
 
 class AsyncRawAdHocClient:
@@ -383,38 +390,41 @@ class AsyncRawAdHocClient:
                 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 == 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 adhoc_execute_prompt_stream(
@@ -494,7 +504,7 @@ class AsyncRawAdHocClient:
             omit=OMIT,
         ) as _response:
 
-            async def stream() -> AsyncHttpResponse[typing.AsyncIterator[AdHocExecutePromptEvent]]:
+            async def _stream() -> AsyncHttpResponse[typing.AsyncIterator[AdHocExecutePromptEvent]]:
                 try:
                     if 200 <= _response.status_code < 300:
 
@@ -518,37 +528,42 @@ class AsyncRawAdHocClient:
                     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 == 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()

vellum/client/resources/container_images/__init__.py

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

vellum/client/resources/container_images/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 RawContainerImagesClient
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
-from ...types.paginated_container_image_read_list import PaginatedContainerImageReadList
 from ...types.container_image_read import ContainerImageRead
 from ...types.docker_service_token import DockerServiceToken
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawContainerImagesClient
+from ...types.paginated_container_image_read_list import PaginatedContainerImageReadList
+from .raw_client import AsyncRawContainerImagesClient, RawContainerImagesClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -73,8 +72,10 @@ class ContainerImagesClient:
         )
         client.container_images.list()
         """
-        response = self._raw_client.list(limit=limit, offset=offset, ordering=ordering, request_options=request_options)
-        return response.data
+        _response = self._raw_client.list(
+            limit=limit, offset=offset, ordering=ordering, request_options=request_options
+        )
+        return _response.data
 
     def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> ContainerImageRead:
         """
@@ -105,8 +106,8 @@ class ContainerImagesClient:
             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 docker_service_token(self, *, request_options: typing.Optional[RequestOptions] = None) -> DockerServiceToken:
         """
@@ -130,8 +131,8 @@ class ContainerImagesClient:
         )
         client.container_images.docker_service_token()
         """
-        response = self._raw_client.docker_service_token(request_options=request_options)
-        return response.data
+        _response = self._raw_client.docker_service_token(request_options=request_options)
+        return _response.data
 
     def push_container_image(
         self,
@@ -172,8 +173,10 @@ class ContainerImagesClient:
             tags=["tags", "tags"],
         )
         """
-        response = self._raw_client.push_container_image(name=name, sha=sha, tags=tags, request_options=request_options)
-        return response.data
+        _response = self._raw_client.push_container_image(
+            name=name, sha=sha, tags=tags, request_options=request_options
+        )
+        return _response.data
 
 
 class AsyncContainerImagesClient:
@@ -239,10 +242,10 @@ class AsyncContainerImagesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.list(
+        _response = await self._raw_client.list(
             limit=limit, offset=offset, ordering=ordering, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> ContainerImageRead:
         """
@@ -281,8 +284,8 @@ class AsyncContainerImagesClient:
 
         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 docker_service_token(
         self, *, request_options: typing.Optional[RequestOptions] = None
@@ -316,8 +319,8 @@ class AsyncContainerImagesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.docker_service_token(request_options=request_options)
-        return response.data
+        _response = await self._raw_client.docker_service_token(request_options=request_options)
+        return _response.data
 
     async def push_container_image(
         self,
@@ -366,7 +369,7 @@ class AsyncContainerImagesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.push_container_image(
+        _response = await self._raw_client.push_container_image(
             name=name, sha=sha, tags=tags, request_options=request_options
         )
-        return response.data
+        return _response.data

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"]