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

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, ...)
@@ -32,7 +31,7 @@ class FolderEntitiesClient:
         """
         return self._raw_client
 
-    def list_(
+    def list(
         self,
         *,
         parent_folder_id: str,
@@ -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,12 +142,10 @@ class FolderEntitiesClient:
             entity_id="entity_id",
         )
         """
-        response = self._raw_client.add_entity_to_folder(
-            folder_id,
-            entity_id=entity_id,
-            request_options=request_options,
+        _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:
@@ -166,7 +163,7 @@ class AsyncFolderEntitiesClient:
         """
         return self._raw_client
 
-    async def list_(
+    async def list(
         self,
         *,
         parent_folder_id: str,
@@ -234,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,
@@ -242,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
@@ -293,9 +290,7 @@ class AsyncFolderEntitiesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.add_entity_to_folder(
-            folder_id,
-            entity_id=entity_id,
-            request_options=request_options,
+        _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,13 +80,10 @@ class MetricDefinitionsClient:
             ],
         )
         """
-        response = self._raw_client.execute_metric_definition(
-            id,
-            inputs=inputs,
-            release_tag=release_tag,
-            request_options=request_options,
+        _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
@@ -122,12 +118,10 @@ class MetricDefinitionsClient:
             id="id",
         )
         """
-        response = self._raw_client.metric_definition_history_item_retrieve(
-            history_id_or_release_tag,
-            id,
-            request_options=request_options,
+        _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:
@@ -201,13 +195,10 @@ class AsyncMetricDefinitionsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.execute_metric_definition(
-            id,
-            inputs=inputs,
-            release_tag=release_tag,
-            request_options=request_options,
+        _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
@@ -250,9 +241,7 @@ class AsyncMetricDefinitionsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.metric_definition_history_item_retrieve(
-            history_id_or_release_tag,
-            id,
-            request_options=request_options,
+        _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,11 +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:
@@ -116,8 +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,10 +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:
@@ -107,7 +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

vellum/client/resources/organizations/raw_client.py

@@ -1,15 +1,14 @@
 # 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.organization_read import OrganizationRead
-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.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...types.organization_read import OrganizationRead
 
 
 class RawOrganizationsClient:
@@ -50,8 +49,8 @@ class RawOrganizationsClient:
                 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 AsyncRawOrganizationsClient:
@@ -92,5 +91,5 @@ class AsyncRawOrganizationsClient:
                 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/prompts/__init__.py

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

vellum/client/resources/prompts/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 RawPromptsClient
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.prompt_exec_config import PromptExecConfig
 from ...types.prompt_push_response import PromptPushResponse
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawPromptsClient
+from .raw_client import AsyncRawPromptsClient, RawPromptsClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -70,12 +69,8 @@ class PromptsClient:
             id="id",
         )
         """
-        response = self._raw_client.pull(
-            id,
-            prompt_variant_id=prompt_variant_id,
-            request_options=request_options,
-        )
-        return response.data
+        _response = self._raw_client.pull(id, prompt_variant_id=prompt_variant_id, request_options=request_options)
+        return _response.data
 
     def push(
         self,
@@ -150,14 +145,14 @@ class PromptsClient:
             ),
         )
         """
-        response = self._raw_client.push(
+        _response = self._raw_client.push(
             exec_config=exec_config,
             prompt_variant_id=prompt_variant_id,
             prompt_variant_label=prompt_variant_label,
             prompt_sandbox_id=prompt_sandbox_id,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
 
 class AsyncPromptsClient:
@@ -221,12 +216,10 @@ class AsyncPromptsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.pull(
-            id,
-            prompt_variant_id=prompt_variant_id,
-            request_options=request_options,
+        _response = await self._raw_client.pull(
+            id, prompt_variant_id=prompt_variant_id, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def push(
         self,
@@ -309,11 +302,11 @@ class AsyncPromptsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.push(
+        _response = await self._raw_client.push(
             exec_config=exec_config,
             prompt_variant_id=prompt_variant_id,
             prompt_variant_label=prompt_variant_label,
             prompt_sandbox_id=prompt_sandbox_id,
             request_options=request_options,
         )
-        return response.data
+        return _response.data