vellum-ai 0.7.11__py3-none-any.whl → 0.8.4__py3-none-any.whl

vellum/resources/workflow_deployments/client.py

@@ -1,17 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from json.decoder import JSONDecodeError
-
-from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import pydantic_v1
+from ...core.client_wrapper import SyncClientWrapper
+from .types.workflow_deployments_list_request_status import WorkflowDeploymentsListRequestStatus
 from ...core.request_options import RequestOptions
 from ...types.paginated_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploymentList
+from ...core.pydantic_utilities import parse_obj_as
+from json.decoder import JSONDecodeError
+from ...core.api_error import ApiError
 from ...types.workflow_deployment_read import WorkflowDeploymentRead
+from ...core.jsonable_encoder import jsonable_encoder
 from ...types.workflow_release_tag_read import WorkflowReleaseTagRead
-from .types.workflow_deployments_list_request_status import WorkflowDeploymentsListRequestStatus
+from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -57,7 +57,7 @@ class WorkflowDeploymentsClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -68,12 +68,23 @@ class WorkflowDeploymentsClient:
             "v1/workflow-deployments",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
-            params={"limit": limit, "offset": offset, "ordering": ordering, "status": status},
+            params={
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+                "status": status,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PaginatedSlimWorkflowDeploymentList, _response.json())  # type: ignore
+                return typing.cast(
+                    PaginatedSlimWorkflowDeploymentList,
+                    parse_obj_as(
+                        type_=PaginatedSlimWorkflowDeploymentList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -98,7 +109,7 @@ class WorkflowDeploymentsClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -115,7 +126,13 @@ class WorkflowDeploymentsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(WorkflowDeploymentRead, _response.json())  # type: ignore
+                return typing.cast(
+                    WorkflowDeploymentRead,
+                    parse_obj_as(
+                        type_=WorkflowDeploymentRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -145,7 +162,7 @@ class WorkflowDeploymentsClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -163,7 +180,13 @@ class WorkflowDeploymentsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(WorkflowReleaseTagRead, _response.json())  # type: ignore
+                return typing.cast(
+                    WorkflowReleaseTagRead,
+                    parse_obj_as(
+                        type_=WorkflowReleaseTagRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -201,7 +224,7 @@ class WorkflowDeploymentsClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -215,13 +238,21 @@ class WorkflowDeploymentsClient:
             f"v1/workflow-deployments/{jsonable_encoder(id)}/release-tags/{jsonable_encoder(name)}",
             base_url=self._client_wrapper.get_environment().default,
             method="PATCH",
-            json={"history_item_id": history_item_id},
+            json={
+                "history_item_id": history_item_id,
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(WorkflowReleaseTagRead, _response.json())  # type: ignore
+                return typing.cast(
+                    WorkflowReleaseTagRead,
+                    parse_obj_as(
+                        type_=WorkflowReleaseTagRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -270,7 +301,7 @@ class AsyncWorkflowDeploymentsClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -287,12 +318,23 @@ class AsyncWorkflowDeploymentsClient:
             "v1/workflow-deployments",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
-            params={"limit": limit, "offset": offset, "ordering": ordering, "status": status},
+            params={
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+                "status": status,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PaginatedSlimWorkflowDeploymentList, _response.json())  # type: ignore
+                return typing.cast(
+                    PaginatedSlimWorkflowDeploymentList,
+                    parse_obj_as(
+                        type_=PaginatedSlimWorkflowDeploymentList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -321,7 +363,7 @@ class AsyncWorkflowDeploymentsClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -344,7 +386,13 @@ class AsyncWorkflowDeploymentsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(WorkflowDeploymentRead, _response.json())  # type: ignore
+                return typing.cast(
+                    WorkflowDeploymentRead,
+                    parse_obj_as(
+                        type_=WorkflowDeploymentRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -376,7 +424,7 @@ class AsyncWorkflowDeploymentsClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -400,7 +448,13 @@ class AsyncWorkflowDeploymentsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(WorkflowReleaseTagRead, _response.json())  # type: ignore
+                return typing.cast(
+                    WorkflowReleaseTagRead,
+                    parse_obj_as(
+                        type_=WorkflowReleaseTagRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -440,7 +494,7 @@ class AsyncWorkflowDeploymentsClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -460,13 +514,21 @@ class AsyncWorkflowDeploymentsClient:
             f"v1/workflow-deployments/{jsonable_encoder(id)}/release-tags/{jsonable_encoder(name)}",
             base_url=self._client_wrapper.get_environment().default,
             method="PATCH",
-            json={"history_item_id": history_item_id},
+            json={
+                "history_item_id": history_item_id,
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(WorkflowReleaseTagRead, _response.json())  # type: ignore
+                return typing.cast(
+                    WorkflowReleaseTagRead,
+                    parse_obj_as(
+                        type_=WorkflowReleaseTagRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)

vellum/resources/workflow_sandboxes/client.py

@@ -1,14 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from json.decoder import JSONDecodeError
-
-from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import pydantic_v1
+from ...core.client_wrapper import SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.workflow_deployment_read import WorkflowDeploymentRead
+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
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -60,7 +60,7 @@ class WorkflowSandboxesClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -85,7 +85,13 @@ class WorkflowSandboxesClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(WorkflowDeploymentRead, _response.json())  # type: ignore
+                return typing.cast(
+                    WorkflowDeploymentRead,
+                    parse_obj_as(
+                        type_=WorkflowDeploymentRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -140,7 +146,7 @@ class AsyncWorkflowSandboxesClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -171,7 +177,13 @@ class AsyncWorkflowSandboxesClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(WorkflowDeploymentRead, _response.json())  # type: ignore
+                return typing.cast(
+                    WorkflowDeploymentRead,
+                    parse_obj_as(
+                        type_=WorkflowDeploymentRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)