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

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,8 +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,
@@ -146,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:
@@ -217,8 +216,10 @@ class AsyncPromptsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.pull(id, prompt_variant_id=prompt_variant_id, request_options=request_options)
-        return response.data
+        _response = await self._raw_client.pull(
+            id, prompt_variant_id=prompt_variant_id, request_options=request_options
+        )
+        return _response.data
 
     async def push(
         self,
@@ -301,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

vellum/client/resources/prompts/raw_client.py

@@ -1,20 +1,19 @@
 # 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.prompt_exec_config import PromptExecConfig
+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 ...errors.bad_request_error import BadRequestError
 from ...errors.not_found_error import NotFoundError
-from json.decoder import JSONDecodeError
-from ...core.api_error import ApiError
+from ...types.prompt_exec_config import PromptExecConfig
 from ...types.prompt_push_response import PromptPushResponse
-from ...core.serialization import convert_and_respect_annotation_metadata
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -74,28 +73,30 @@ class RawPromptsClient:
                 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(),
                         ),
-                    )
+                    ),
                 )
             _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(
         self,
@@ -159,28 +160,30 @@ class RawPromptsClient:
                 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(),
                         ),
-                    )
+                    ),
                 )
             _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 AsyncRawPromptsClient:
@@ -237,28 +240,30 @@ class AsyncRawPromptsClient:
                 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(),
                         ),
-                    )
+                    ),
                 )
             _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(
         self,
@@ -322,25 +327,27 @@ class AsyncRawPromptsClient:
                 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(),
                         ),
-                    )
+                    ),
                 )
             _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/sandboxes/__init__.py

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

vellum/client/resources/sandboxes/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 RawSandboxesClient
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.deployment_read import DeploymentRead
 from ...types.named_scenario_input_request import NamedScenarioInputRequest
 from ...types.sandbox_scenario import SandboxScenario
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawSandboxesClient
+from .raw_client import AsyncRawSandboxesClient, RawSandboxesClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -90,7 +89,7 @@ class SandboxesClient:
             prompt_variant_id="prompt_variant_id",
         )
         """
-        response = self._raw_client.deploy_prompt(
+        _response = self._raw_client.deploy_prompt(
             id,
             prompt_variant_id,
             prompt_deployment_id=prompt_deployment_id,
@@ -100,7 +99,7 @@ class SandboxesClient:
             release_description=release_description,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     def upsert_sandbox_scenario(
         self,
@@ -161,10 +160,10 @@ class SandboxesClient:
             ],
         )
         """
-        response = self._raw_client.upsert_sandbox_scenario(
+        _response = self._raw_client.upsert_sandbox_scenario(
             id, inputs=inputs, label=label, scenario_id=scenario_id, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     def delete_sandbox_scenario(
         self, id: str, scenario_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -200,8 +199,8 @@ class SandboxesClient:
             scenario_id="scenario_id",
         )
         """
-        response = self._raw_client.delete_sandbox_scenario(id, scenario_id, request_options=request_options)
-        return response.data
+        _response = self._raw_client.delete_sandbox_scenario(id, scenario_id, request_options=request_options)
+        return _response.data
 
 
 class AsyncSandboxesClient:
@@ -284,7 +283,7 @@ class AsyncSandboxesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.deploy_prompt(
+        _response = await self._raw_client.deploy_prompt(
             id,
             prompt_variant_id,
             prompt_deployment_id=prompt_deployment_id,
@@ -294,7 +293,7 @@ class AsyncSandboxesClient:
             release_description=release_description,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     async def upsert_sandbox_scenario(
         self,
@@ -363,10 +362,10 @@ class AsyncSandboxesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.upsert_sandbox_scenario(
+        _response = await self._raw_client.upsert_sandbox_scenario(
             id, inputs=inputs, label=label, scenario_id=scenario_id, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def delete_sandbox_scenario(
         self, id: str, scenario_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -410,5 +409,5 @@ class AsyncSandboxesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.delete_sandbox_scenario(id, scenario_id, request_options=request_options)
-        return response.data
+        _response = await self._raw_client.delete_sandbox_scenario(id, scenario_id, request_options=request_options)
+        return _response.data

vellum/client/resources/sandboxes/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 ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...types.deployment_read import DeploymentRead
-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, 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.deployment_read import DeploymentRead
 from ...types.named_scenario_input_request import NamedScenarioInputRequest
 from ...types.sandbox_scenario import SandboxScenario
-from ...core.serialization import convert_and_respect_annotation_metadata
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -96,8 +95,8 @@ class RawSandboxesClient:
                 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 upsert_sandbox_scenario(
         self,
@@ -167,8 +166,8 @@ class RawSandboxesClient:
                 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 delete_sandbox_scenario(
         self, id: str, scenario_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -202,8 +201,8 @@ class RawSandboxesClient:
                 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 AsyncRawSandboxesClient:
@@ -283,8 +282,8 @@ class AsyncRawSandboxesClient:
                 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 upsert_sandbox_scenario(
         self,
@@ -354,8 +353,8 @@ class AsyncRawSandboxesClient:
                 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 delete_sandbox_scenario(
         self, id: str, scenario_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -389,5 +388,5 @@ class AsyncRawSandboxesClient:
                 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/test_suite_runs/__init__.py

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

vellum/client/resources/test_suite_runs/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 RawTestSuiteRunsClient
-from ...types.test_suite_run_exec_config_request import TestSuiteRunExecConfigRequest
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
-from ...types.test_suite_run_read import TestSuiteRunRead
 from ...types.paginated_test_suite_run_execution_list import PaginatedTestSuiteRunExecutionList
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawTestSuiteRunsClient
+from ...types.test_suite_run_exec_config_request import TestSuiteRunExecConfigRequest
+from ...types.test_suite_run_read import TestSuiteRunRead
+from .raw_client import AsyncRawTestSuiteRunsClient, RawTestSuiteRunsClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -83,13 +82,13 @@ class TestSuiteRunsClient:
             ),
         )
         """
-        response = self._raw_client.create(
+        _response = self._raw_client.create(
             exec_config=exec_config,
             test_suite_id=test_suite_id,
             test_suite_name=test_suite_name,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> TestSuiteRunRead:
         """
@@ -120,8 +119,8 @@ class TestSuiteRunsClient:
             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 list_executions(
         self,
@@ -170,10 +169,10 @@ class TestSuiteRunsClient:
             id="id",
         )
         """
-        response = self._raw_client.list_executions(
+        _response = self._raw_client.list_executions(
             id, expand=expand, limit=limit, offset=offset, request_options=request_options
         )
-        return response.data
+        return _response.data
 
 
 class AsyncTestSuiteRunsClient:
@@ -249,13 +248,13 @@ class AsyncTestSuiteRunsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.create(
+        _response = await self._raw_client.create(
             exec_config=exec_config,
             test_suite_id=test_suite_id,
             test_suite_name=test_suite_name,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
     async def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> TestSuiteRunRead:
         """
@@ -294,8 +293,8 @@ class AsyncTestSuiteRunsClient:
 
         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 list_executions(
         self,
@@ -352,7 +351,7 @@ class AsyncTestSuiteRunsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.list_executions(
+        _response = await self._raw_client.list_executions(
             id, expand=expand, limit=limit, offset=offset, request_options=request_options
         )
-        return response.data
+        return _response.data