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

vellum/client/resources/workflows/raw_client.py

@@ -0,0 +1,307 @@
+# 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 ...core.jsonable_encoder import jsonable_encoder
+from ...errors.bad_request_error import BadRequestError
+from ...core.pydantic_utilities import parse_obj_as
+from json.decoder import JSONDecodeError
+from ...core.api_error import ApiError
+from ...types.workflow_push_exec_config import WorkflowPushExecConfig
+from ...types.workflow_push_deployment_config_request import WorkflowPushDeploymentConfigRequest
+from ... import core
+from ...types.workflow_push_response import WorkflowPushResponse
+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, ...)
+
+
+class RawWorkflowsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def pull(
+        self,
+        id: str,
+        *,
+        exclude_code: typing.Optional[bool] = None,
+        exclude_display: typing.Optional[bool] = None,
+        include_json: typing.Optional[bool] = None,
+        include_sandbox: typing.Optional[bool] = None,
+        strict: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[typing.Iterator[bytes]]:
+        """
+        Parameters
+        ----------
+        id : str
+            The ID of the Workflow to pull from
+
+        exclude_code : typing.Optional[bool]
+
+        exclude_display : typing.Optional[bool]
+
+        include_json : typing.Optional[bool]
+
+        include_sandbox : typing.Optional[bool]
+
+        strict : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
+
+        Returns
+        -------
+        HttpResponse[typing.Iterator[bytes]]
+
+        """
+        with self._client_wrapper.httpx_client.stream(
+            f"v1/workflows/{jsonable_encoder(id)}/pull",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "exclude_code": exclude_code,
+                "exclude_display": exclude_display,
+                "include_json": include_json,
+                "include_sandbox": include_sandbox,
+                "strict": strict,
+            },
+            request_options=request_options,
+        ) as _response:
+            try:
+                if 200 <= _response.status_code < 300:
+                    _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
+                    return HttpResponse(
+                        response=_response, data=(_chunk for _chunk in _response.iter_bytes(chunk_size=_chunk_size))
+                    )
+                _response.read()
+                if _response.status_code == 400:
+                    raise BadRequestError(
+                        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)
+
+    def push(
+        self,
+        *,
+        exec_config: WorkflowPushExecConfig,
+        workflow_sandbox_id: typing.Optional[str] = OMIT,
+        deployment_config: typing.Optional[WorkflowPushDeploymentConfigRequest] = OMIT,
+        artifact: typing.Optional[core.File] = OMIT,
+        dry_run: typing.Optional[bool] = OMIT,
+        strict: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[WorkflowPushResponse]:
+        """
+        Parameters
+        ----------
+        exec_config : WorkflowPushExecConfig
+            The execution configuration of the workflow.
+
+        workflow_sandbox_id : typing.Optional[str]
+
+        deployment_config : typing.Optional[WorkflowPushDeploymentConfigRequest]
+
+        artifact : typing.Optional[core.File]
+            See core.File for more documentation
+
+        dry_run : typing.Optional[bool]
+
+        strict : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[WorkflowPushResponse]
+
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/workflows/push",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            data={
+                "exec_config": exec_config,
+                "workflow_sandbox_id": workflow_sandbox_id,
+                "deployment_config": deployment_config,
+                "dry_run": dry_run,
+                "strict": strict,
+            },
+            files={
+                "artifact": artifact,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    WorkflowPushResponse,
+                    parse_obj_as(
+                        type_=WorkflowPushResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                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)
+
+
+class AsyncRawWorkflowsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def pull(
+        self,
+        id: str,
+        *,
+        exclude_code: typing.Optional[bool] = None,
+        exclude_display: typing.Optional[bool] = None,
+        include_json: typing.Optional[bool] = None,
+        include_sandbox: typing.Optional[bool] = None,
+        strict: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[typing.AsyncIterator[bytes]]:
+        """
+        Parameters
+        ----------
+        id : str
+            The ID of the Workflow to pull from
+
+        exclude_code : typing.Optional[bool]
+
+        exclude_display : typing.Optional[bool]
+
+        include_json : typing.Optional[bool]
+
+        include_sandbox : typing.Optional[bool]
+
+        strict : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
+
+        Returns
+        -------
+        AsyncHttpResponse[typing.AsyncIterator[bytes]]
+
+        """
+        async with self._client_wrapper.httpx_client.stream(
+            f"v1/workflows/{jsonable_encoder(id)}/pull",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "exclude_code": exclude_code,
+                "exclude_display": exclude_display,
+                "include_json": include_json,
+                "include_sandbox": include_sandbox,
+                "strict": strict,
+            },
+            request_options=request_options,
+        ) as _response:
+            try:
+                if 200 <= _response.status_code < 300:
+                    _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
+                    return AsyncHttpResponse(
+                        response=_response,
+                        data=(_chunk async for _chunk in _response.aiter_bytes(chunk_size=_chunk_size)),
+                    )
+                await _response.aread()
+                if _response.status_code == 400:
+                    raise BadRequestError(
+                        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)
+
+    async def push(
+        self,
+        *,
+        exec_config: WorkflowPushExecConfig,
+        workflow_sandbox_id: typing.Optional[str] = OMIT,
+        deployment_config: typing.Optional[WorkflowPushDeploymentConfigRequest] = OMIT,
+        artifact: typing.Optional[core.File] = OMIT,
+        dry_run: typing.Optional[bool] = OMIT,
+        strict: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[WorkflowPushResponse]:
+        """
+        Parameters
+        ----------
+        exec_config : WorkflowPushExecConfig
+            The execution configuration of the workflow.
+
+        workflow_sandbox_id : typing.Optional[str]
+
+        deployment_config : typing.Optional[WorkflowPushDeploymentConfigRequest]
+
+        artifact : typing.Optional[core.File]
+            See core.File for more documentation
+
+        dry_run : typing.Optional[bool]
+
+        strict : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[WorkflowPushResponse]
+
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/workflows/push",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            data={
+                "exec_config": exec_config,
+                "workflow_sandbox_id": workflow_sandbox_id,
+                "deployment_config": deployment_config,
+                "dry_run": dry_run,
+                "strict": strict,
+            },
+            files={
+                "artifact": artifact,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    WorkflowPushResponse,
+                    parse_obj_as(
+                        type_=WorkflowPushResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                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)

vellum/client/resources/workspace_secrets/client.py

@@ -2,13 +2,11 @@
 
 import typing
 from ...core.client_wrapper import SyncClientWrapper
+from .raw_client import RawWorkspaceSecretsClient
 from ...core.request_options import RequestOptions
 from ...types.workspace_secret_read import WorkspaceSecretRead
-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 .raw_client import AsyncRawWorkspaceSecretsClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -16,7 +14,22 @@ OMIT = typing.cast(typing.Any, ...)
 
 class WorkspaceSecretsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawWorkspaceSecretsClient(client_wrapper=client_wrapper)
+
+    @property
+    def _client_wrapper(self) -> SyncClientWrapper:
+        return self._raw_client._client_wrapper
+
+    @property
+    def with_raw_response(self) -> RawWorkspaceSecretsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawWorkspaceSecretsClient
+        """
+        return self._raw_client
 
     def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> WorkspaceSecretRead:
         """
@@ -47,25 +60,11 @@ class WorkspaceSecretsClient:
             id="id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/workspace-secrets/{jsonable_encoder(id)}",
-            base_url=self._client_wrapper.get_environment().default,
-            method="GET",
+        response = self._raw_client.retrieve(
+            id,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    WorkspaceSecretRead,
-                    parse_obj_as(
-                        type_=WorkspaceSecretRead,  # 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)
+        return response.data
 
     def partial_update(
         self,
@@ -107,38 +106,29 @@ class WorkspaceSecretsClient:
             id="id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/workspace-secrets/{jsonable_encoder(id)}",
-            base_url=self._client_wrapper.get_environment().default,
-            method="PATCH",
-            json={
-                "label": label,
-                "value": value,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        response = self._raw_client.partial_update(
+            id,
+            label=label,
+            value=value,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    WorkspaceSecretRead,
-                    parse_obj_as(
-                        type_=WorkspaceSecretRead,  # 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)
+        return response.data
 
 
 class AsyncWorkspaceSecretsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = AsyncRawWorkspaceSecretsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawWorkspaceSecretsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawWorkspaceSecretsClient
+        """
+        return self._raw_client
 
     async def retrieve(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -179,25 +169,11 @@ class AsyncWorkspaceSecretsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/workspace-secrets/{jsonable_encoder(id)}",
-            base_url=self._client_wrapper.get_environment().default,
-            method="GET",
+        response = await self._raw_client.retrieve(
+            id,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    WorkspaceSecretRead,
-                    parse_obj_as(
-                        type_=WorkspaceSecretRead,  # 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)
+        return response.data
 
     async def partial_update(
         self,
@@ -247,30 +223,10 @@ class AsyncWorkspaceSecretsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/workspace-secrets/{jsonable_encoder(id)}",
-            base_url=self._client_wrapper.get_environment().default,
-            method="PATCH",
-            json={
-                "label": label,
-                "value": value,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        response = await self._raw_client.partial_update(
+            id,
+            label=label,
+            value=value,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    WorkspaceSecretRead,
-                    parse_obj_as(
-                        type_=WorkspaceSecretRead,  # 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)
+        return response.data