pangea-sdk 6.2.0b2__py3-none-any.whl → 6.3.0__py3-none-any.whl

pangea/asyncio/services/vault.py

@@ -59,7 +59,6 @@ from pangea.services.vault.models.common import (
     JWTVerifyResult,
     ListRequest,
     ListResult,
-    Metadata,
     PangeaToken,
     PangeaTokenRotateRequest,
     RequestManualRotationState,
@@ -319,7 +318,7 @@ class VaultAsync(ServiceBaseAsync):
         *,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         disabled_at: str | None = None,
         enabled: bool | None = None,
@@ -395,7 +394,7 @@ class VaultAsync(ServiceBaseAsync):
         result_class: type[TResult] = SecretStoreResult,  # type: ignore[assignment]
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         disabled_at: datetime.datetime | None = None,
         **kwargs: Any,
@@ -420,22 +419,22 @@ class VaultAsync(ServiceBaseAsync):
         *,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         disabled_at: datetime.datetime | None = None,
     ) -> PangeaResponse[Secret]:
         """
         Store secret
 
-        Store a secret.
+        Store a secret as a Vault item.
 
         Args:
-            secret: The secret value.
-            name: The name of this item.
-            folder: The folder where this item is stored.
-            metadata: User-provided metadata.
-            tags: A list of user-defined tags.
-            disabled_at: Timestamp indicating when the item will be disabled.
+            secret: Secret value
+            name: Name of the item
+            folder: Folder where the item is stored
+            metadata: Metadata provided by the user
+            tags: List of user-defined tags
+            disabled_at: Timestamp indicating when the item will be disabled
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -461,25 +460,38 @@ class VaultAsync(ServiceBaseAsync):
         *,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         disabled_at: datetime.datetime | None = None,
         rotation_frequency: str | None = None,
-        rotation_state: RotationState | None = None,
+        rotation_state: Literal["deactivated", "destroyed", "inherited"] | None = None,
         rotation_grace_period: str | None = None,
     ) -> PangeaResponse[PangeaToken]:
         """
         Store secret
 
-        Store a Pangea token.
+        Store a Pangea token as a Vault item.
 
         Args:
-            token: The Pangea token value.
-            name: The name of this item.
-            folder: The folder where this item is stored.
-            metadata: User-provided metadata.
-            tags: A list of user-defined tags.
-            disabled_at: Timestamp indicating when the item will be disabled.
+            token: Pangea token value
+            name: Name of the item
+            folder: Folder where the item is stored
+            metadata: Metadata provided by the user
+            tags: List of user-defined tags
+            disabled_at: Timestamp indicating when the item will be disabled
+            rotation_frequency: Time interval between item rotations, provided
+              as a positive number followed by a time unit: `secs`, `mins`, `hrs`,
+              `days`, `weeks`, `months`, or `years`. You can use abbreviations
+              like `1d`. Omit to inherit from the parent folder or default
+              settings. Set to `never` to disable rotation.
+            rotation_state: Target state for the previous version after
+                rotation. Set to `inherited` to inherit from the parent folder
+                or default settings.
+            rotation_grace_period: Grace period for the previous version,
+              provided as a positive number followed by a time unit: `secs`,
+              `mins`, `hrs`, `days`, `weeks`, `months`, or `years`. You can use
+              abbreviations like `1d`. Omit to inherit from the parent folder or
+              default settings.
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -510,7 +522,7 @@ class VaultAsync(ServiceBaseAsync):
         *,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         disabled_at: datetime.datetime | None = None,
         rotation_frequency: str | None = None,
@@ -672,7 +684,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeySigningAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -718,7 +730,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeyEncryptionAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -764,7 +776,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeyJwtAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -810,7 +822,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeyPkiAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -856,7 +868,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeyAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -902,7 +914,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: SymmetricKeyEncryptionAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -948,7 +960,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: SymmetricKeyJwtAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -994,7 +1006,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: SymmetricKeyFpeAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1040,7 +1052,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: SymmetricKeyAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1085,7 +1097,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeyAlgorithm | SymmetricKeyAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1151,7 +1163,7 @@ class VaultAsync(ServiceBaseAsync):
         private_key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1201,7 +1213,7 @@ class VaultAsync(ServiceBaseAsync):
         private_key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1251,7 +1263,7 @@ class VaultAsync(ServiceBaseAsync):
         private_key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1301,7 +1313,7 @@ class VaultAsync(ServiceBaseAsync):
         private_key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1350,7 +1362,7 @@ class VaultAsync(ServiceBaseAsync):
         key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1398,7 +1410,7 @@ class VaultAsync(ServiceBaseAsync):
         key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1446,7 +1458,7 @@ class VaultAsync(ServiceBaseAsync):
         key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1503,7 +1515,7 @@ class VaultAsync(ServiceBaseAsync):
         key: str | None = None,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1941,7 +1953,7 @@ class VaultAsync(ServiceBaseAsync):
         name: str,
         folder: str,
         *,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState = RequestRotationState.INHERITED,

pangea/request.py

@@ -11,14 +11,14 @@ import json
 import logging
 import time
 from collections.abc import Iterable, Mapping
-from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Type, Union, cast, overload
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Type, Union, cast
 
 import requests
-from pydantic import BaseModel, TypeAdapter
+from pydantic import BaseModel
 from pydantic_core import to_jsonable_python
 from requests.adapters import HTTPAdapter, Retry
 from requests_toolbelt import MultipartDecoder  # type: ignore[import-untyped]
-from typing_extensions import Literal, TypeAlias, TypeVar, override
+from typing_extensions import TypeAlias, TypeVar, override
 from yarl import URL
 
 import pangea
@@ -43,6 +43,8 @@ _FileSpecTuple4: TypeAlias = tuple[_FileName, _FileContent, _FileContentType, _F
 _FileSpec: TypeAlias = Union[_FileContent, _FileSpecTuple2, _FileSpecTuple3, _FileSpecTuple4]
 _Files: TypeAlias = Union[Mapping[str, _FileSpec], Iterable[tuple[str, _FileSpec]]]
 
+_HeadersUpdateMapping: TypeAlias = Mapping[str, str]
+
 
 class MultipartResponse:
     pangea_json: Dict[str, str]
@@ -66,7 +68,7 @@ class PangeaRequestBase:
         self._queued_retry_enabled = config.queued_retry_enabled
 
         # Custom headers
-        self._extra_headers: Dict = {}
+        self._extra_headers: _HeadersUpdateMapping = {}
         self._user_agent = ""
 
         self.set_custom_user_agent(config.custom_user_agent)
@@ -81,18 +83,17 @@ class PangeaRequestBase:
 
         return self._session
 
-    def set_extra_headers(self, headers: dict):
+    def set_extra_headers(self, headers: _HeadersUpdateMapping):
         """Sets any additional headers in the request.
 
         Args:
-            headers (dict): key-value pair containing extra headers to et
+            headers: key-value pairs containing extra headers to set
 
         Example:
             set_extra_headers({ "My-Header" : "foobar" })
         """
 
-        if isinstance(headers, dict):
-            self._extra_headers = headers
+        self._extra_headers = headers
 
     def set_custom_user_agent(self, user_agent: Optional[str]):
         self.config.custom_user_agent = user_agent
@@ -129,17 +130,14 @@ class PangeaRequestBase:
         url = URL(self.config.base_url_template.format(SERVICE_NAME=self.service))
         return str(url / path)
 
-    def _headers(self) -> dict:
-        headers = {
-            "User-Agent": self._user_agent,
+    def _headers(self) -> dict[str, str]:
+        return {
+            **self._extra_headers,
             "Authorization": f"Bearer {self.token}",
             "Content-Type": "application/json",
+            "User-Agent": self._user_agent,
         }
 
-        # We want to ignore previous headers if user tried to set them, so we will overwrite them.
-        self._extra_headers.update(headers)
-        return self._extra_headers
-
     def _get_filename_from_content_disposition(self, content_disposition: str) -> Optional[str]:
         filename_parts = content_disposition.split("name=")
         if len(filename_parts) > 1:
@@ -222,24 +220,6 @@ class PangeaRequest(PangeaRequestBase):
     def __del__(self) -> None:
         self.session.close()
 
-    def delete(self, endpoint: str) -> None:
-        """
-        Makes a DELETE call to a Pangea endpoint.
-
-        Args:
-            endpoint: The Pangea API endpoint.
-        """
-
-        url = self._url(endpoint)
-
-        self.logger.debug(
-            json.dumps({"service": self.service, "action": "delete", "url": url}, default=default_encoder)
-        )
-
-        requests_response = self._http_delete(url, headers=self._headers())
-        self._check_http_errors(requests_response)
-
-    @overload
     def post(
         self,
         endpoint: str,
@@ -248,62 +228,18 @@ class PangeaRequest(PangeaRequestBase):
         files: Optional[list[Tuple]] = None,
         poll_result: bool = True,
         url: Optional[str] = None,
-        *,
-        pangea_response: Literal[True] = True,
     ) -> PangeaResponse[TResult]:
-        """
-        Makes the POST call to a Pangea Service endpoint.
+        """Makes the POST call to a Pangea Service endpoint.
 
         Args:
-            endpoint: The Pangea Service API endpoint.
-            data: The POST body payload object
+            endpoint(str): The Pangea Service API endpoint.
+            data(dict): The POST body payload object
 
         Returns:
             PangeaResponse which contains the response in its entirety and
                various properties to retrieve individual fields
         """
 
-    @overload
-    def post(
-        self,
-        endpoint: str,
-        result_class: Type[TResult],
-        data: str | BaseModel | Mapping[str, Any] | None = None,
-        files: Optional[list[Tuple]] = None,
-        poll_result: bool = True,
-        url: Optional[str] = None,
-        *,
-        pangea_response: Literal[False],
-    ) -> TResult:
-        """
-        Makes the POST call to a Pangea Service endpoint.
-
-        Args:
-            endpoint: The Pangea Service API endpoint.
-            data: The POST body payload object
-        """
-
-    def post(
-        self,
-        endpoint: str,
-        result_class: Type[TResult],
-        data: str | BaseModel | Mapping[str, Any] | None = None,
-        files: Optional[list[Tuple]] = None,
-        poll_result: bool = True,
-        url: Optional[str] = None,
-        *,
-        pangea_response: bool = True,
-    ) -> PangeaResponse[TResult] | TResult:
-        """
-        Makes a POST call to a Pangea Service endpoint.
-
-        Args:
-            endpoint: The Pangea Service API endpoint.
-            data: The POST body payload object
-            pangea_response: Whether or not the response body follows Pangea's
-              standard response schema
-        """
-
         if isinstance(data, BaseModel):
             data = data.model_dump(exclude_none=True)
 
@@ -341,13 +277,9 @@ class PangeaRequest(PangeaRequestBase):
 
         self._check_http_errors(requests_response)
 
-        if not pangea_response:
-            type_adapter = TypeAdapter(result_class)
-            return type_adapter.validate_python(requests_response.json())
-
         if "multipart/form-data" in requests_response.headers.get("content-type", ""):
             multipart_response = self._process_multipart_response(requests_response)
-            pangea_response_obj: PangeaResponse = PangeaResponse(
+            pangea_response: PangeaResponse = PangeaResponse(
                 requests_response,
                 result_class=result_class,
                 json=multipart_response.pangea_json,
@@ -360,14 +292,14 @@ class PangeaRequest(PangeaRequestBase):
                     json.dumps({"service": self.service, "action": "post", "url": url, "response": json_resp})
                 )
 
-                pangea_response_obj = PangeaResponse(requests_response, result_class=result_class, json=json_resp)
+                pangea_response = PangeaResponse(requests_response, result_class=result_class, json=json_resp)
             except requests.exceptions.JSONDecodeError as e:
                 raise pe.PangeaException(f"Failed to decode json response. {e}. Body: {requests_response.text}") from e
 
         if poll_result:
-            pangea_response_obj = self._handle_queued_result(pangea_response_obj)
+            pangea_response = self._handle_queued_result(pangea_response)
 
-        return self._check_response(pangea_response_obj)
+        return self._check_response(pangea_response)
 
     def _get_pangea_json(self, decoder: MultipartDecoder) -> Optional[Dict]:
         # Iterate through parts
@@ -410,18 +342,10 @@ class PangeaRequest(PangeaRequestBase):
         if resp.status_code == 503:
             raise pe.ServiceTemporarilyUnavailable(resp.json())
 
-    def _http_delete(
-        self,
-        url: str,
-        *,
-        headers: Mapping[str, str | bytes | None] = {},
-    ) -> requests.Response:
-        return self.session.delete(url, headers=headers)
-
     def _http_post(
         self,
         url: str,
-        headers: Mapping[str, str] = {},
+        headers: _HeadersUpdateMapping = {},  # noqa: B006
         data: str | dict[Any, Any] = {},  # noqa: B006
         files: _Files | None = None,
         multipart_post: bool = True,
@@ -462,98 +386,37 @@ class PangeaRequest(PangeaRequestBase):
 
         return response
 
-    @overload
-    def get(
-        self,
-        path: str,
-        result_class: Type[TResult],
-        check_response: bool = True,
-        *,
-        params: (
-            Mapping[str | bytes | int | float, str | bytes | int | float | Iterable[str | bytes | int | float] | None]
-            | None
-        ) = None,
-        pangea_response: Literal[True] = True,
-    ) -> PangeaResponse[TResult]:
-        """
-        Makes the GET call to a Pangea Service endpoint.
+    def get(self, path: str, result_class: Type[TResult], check_response: bool = True) -> PangeaResponse[TResult]:
+        """Makes the GET call to a Pangea Service endpoint.
 
         Args:
-            path: Additional URL path
-            params: Dictionary of querystring data to attach to the request
+            endpoint(str): The Pangea Service API endpoint.
+            path(str): Additional URL path
 
         Returns:
             PangeaResponse which contains the response in its entirety and
                various properties to retrieve individual fields
         """
 
-    @overload
-    def get(
-        self,
-        path: str,
-        result_class: Type[TResult],
-        check_response: bool = True,
-        *,
-        params: (
-            Mapping[str | bytes | int | float, str | bytes | int | float | Iterable[str | bytes | int | float] | None]
-            | None
-        ) = None,
-        pangea_response: Literal[False] = False,
-    ) -> TResult:
-        """
-        Makes the GET call to a Pangea Service endpoint.
-
-        Args:
-            path: Additional URL path
-            params: Dictionary of querystring data to attach to the request
-        """
-
-    def get(
-        self,
-        path: str,
-        result_class: Type[TResult],
-        check_response: bool = True,
-        *,
-        params: (
-            Mapping[str | bytes | int | float, str | bytes | int | float | Iterable[str | bytes | int | float] | None]
-            | None
-        ) = None,
-        pangea_response: bool = True,
-    ) -> PangeaResponse[TResult] | TResult:
-        """
-        Makes the GET call to a Pangea Service endpoint.
-
-        Args:
-            path: Additional URL path
-            params: Dictionary of querystring data to attach to the request
-            pangea_response: Whether or not the response body follows Pangea's
-              standard response schema
-        """
-
         url = self._url(path)
         self.logger.debug(json.dumps({"service": self.service, "action": "get", "url": url}))
-        requests_response = self.session.get(url, params=params, headers=self._headers())
+        requests_response = self.session.get(url, headers=self._headers())
         self._check_http_errors(requests_response)
-
-        if not pangea_response:
-            type_adapter = TypeAdapter(result_class)
-            return type_adapter.validate_python(requests_response.json())
-
-        pangea_response_obj: PangeaResponse = PangeaResponse(
+        pangea_response: PangeaResponse = PangeaResponse(
             requests_response, result_class=result_class, json=requests_response.json()
         )
 
         self.logger.debug(
             json.dumps(
-                {"service": self.service, "action": "get", "url": url, "response": pangea_response_obj.json},
+                {"service": self.service, "action": "get", "url": url, "response": pangea_response.json},
                 default=default_encoder,
             )
         )
 
         if check_response is False:
-            return pangea_response_obj
+            return pangea_response
 
-        return self._check_response(pangea_response_obj)
+        return self._check_response(pangea_response)
 
     def download_file(self, url: str, filename: str | None = None) -> AttachedFile:
         """

pangea/response.py

@@ -82,14 +82,12 @@ class TransferMethod(str, enum.Enum):
 PangeaDateTime = Annotated[datetime.datetime, PlainSerializer(format_datetime)]
 
 
-# API response should accept arbitrary fields to make them accept possible new parameters
-class APIResponseModel(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True, extra="allow")
+class APIRequestModel(BaseModel):
+    model_config = ConfigDict(extra="forbid")
 
 
-# API request models doesn't not allow arbitrary fields
-class APIRequestModel(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True, extra="allow")
+class APIResponseModel(BaseModel):
+    model_config = ConfigDict(extra="allow")
 
 
 class PangeaResponseResult(APIResponseModel):
@@ -195,6 +193,8 @@ T = TypeVar("T", bound=PangeaResponseResult)
 
 
 class PangeaResponse(ResponseHeader, Generic[T]):
+    model_config = ConfigDict(arbitrary_types_allowed=True, extra="forbid")
+
     raw_result: Optional[Dict[str, Any]] = None
     raw_response: Optional[Union[requests.Response, aiohttp.ClientResponse]] = None
     result: Optional[T] = None

pangea/services/__init__.py

@@ -7,7 +7,6 @@ from .authz import AuthZ
 from .embargo import Embargo
 from .file_scan import FileScan
 from .intel import DomainIntel, FileIntel, IpIntel, UrlIntel, UserIntel
-from .management import Management
 from .prompt_guard import PromptGuard
 from .redact import Redact
 from .sanitize import Sanitize