pangea-sdk 6.4.0__tar.gz → 6.5.0__tar.gz

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/PKG-INFO

@@ -1,24 +1,25 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: pangea-sdk
-Version: 6.4.0
+Version: 6.5.0
 Summary: Pangea API SDK
-License: MIT
 Keywords: Pangea,SDK,Audit
 Author: Glenn Gallien
-Author-email: glenn.gallien@pangea.cloud
-Requires-Python: >=3.9.2,<4.0.0
+Author-email: Glenn Gallien <glenn.gallien@pangea.cloud>
+License-Expression: MIT
 Classifier: Topic :: Software Development
 Classifier: Topic :: Software Development :: Libraries
-Requires-Dist: aiohttp (>=3.12.14,<4.0.0)
-Requires-Dist: cryptography (>=45.0.5,<46.0.0)
-Requires-Dist: deprecated (>=1.2.18,<2.0.0)
-Requires-Dist: google-crc32c (>=1.7.1,<2.0.0)
-Requires-Dist: pydantic (>=2.11.7,<3.0.0)
-Requires-Dist: python-dateutil (>=2.9.0.post0,<3.0.0)
-Requires-Dist: requests (>=2.32.4,<3.0.0)
-Requires-Dist: requests-toolbelt (>=1.0.0,<2.0.0)
-Requires-Dist: typing-extensions (>=4.14.1,<5.0.0)
-Requires-Dist: yarl (>=1.20.1,<2.0.0)
+Requires-Dist: aiohttp>=3.12.15,<4.0.0
+Requires-Dist: cryptography>=45.0.6,<46.0.0
+Requires-Dist: deprecated>=1.2.18,<2.0.0
+Requires-Dist: google-crc32c>=1.7.1,<2.0.0
+Requires-Dist: pydantic>=2.11.7,<3.0.0
+Requires-Dist: python-dateutil>=2.9.0.post0,<3.0.0
+Requires-Dist: requests>=2.32.4,<3.0.0
+Requires-Dist: requests-toolbelt>=1.0.0,<2.0.0
+Requires-Dist: typing-extensions>=4.14.1,<5.0.0
+Requires-Dist: yarl>=1.20.1,<2.0.0
+Requires-Python: >=3.9.2, <4.0.0
+Project-URL: repository, https://github.com/pangeacyber/pangea-python
 Description-Content-Type: text/markdown
 
 <a href="https://pangea.cloud?utm_source=github&utm_medium=python-sdk" target="_blank" rel="noopener noreferrer">
@@ -116,8 +117,7 @@ The SDK supports the following configuration options via `PangeaConfig`:
   Use `base_url_template` for more control over the URL, such as setting
   service-specific paths. Defaults to `aws.us.pangea.cloud`.
 - `request_retries` — Number of retries on the initial request.
-- `request_backoff` — Backoff strategy passed to 'requests'.
-- `request_timeout` — Timeout used on initial request attempts.
+- `request_backoff` — A backoff factor to apply between request attempts.
 - `poll_result_timeout` — Timeout used to poll results after 202 (in secs).
 - `queued_retry_enabled` — Enable queued request retry support.
 - `custom_user_agent` — Custom user agent to be used in the request headers.
@@ -243,4 +243,3 @@ It accepts multiple file formats:
   [Beta Examples]: https://github.com/pangeacyber/pangea-python/tree/beta/examples
   [Pangea Console]: https://console.pangea.cloud/
   [Secure Audit Log]: https://pangea.cloud/docs/audit
-

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/README.md

@@ -93,8 +93,7 @@ The SDK supports the following configuration options via `PangeaConfig`:
   Use `base_url_template` for more control over the URL, such as setting
   service-specific paths. Defaults to `aws.us.pangea.cloud`.
 - `request_retries` — Number of retries on the initial request.
-- `request_backoff` — Backoff strategy passed to 'requests'.
-- `request_timeout` — Timeout used on initial request attempts.
+- `request_backoff` — A backoff factor to apply between request attempts.
 - `poll_result_timeout` — Timeout used to poll results after 202 (in secs).
 - `queued_retry_enabled` — Enable queued request retry support.
 - `custom_user_agent` — Custom user agent to be used in the request headers.

pangea_sdk-6.5.0/pangea/__init__.py

@@ -0,0 +1,7 @@
+__version__ = "6.5.0"
+
+from pangea.config import PangeaConfig
+from pangea.file_uploader import FileUploader
+from pangea.response import PangeaResponse, PangeaResponseResult, TransferMethod
+
+__all__ = ("FileUploader", "PangeaConfig", "PangeaResponse", "PangeaResponseResult", "TransferMethod")

pangea_sdk-6.5.0/pangea/_constants.py

@@ -0,0 +1,4 @@
+from __future__ import annotations
+
+MAX_RETRY_DELAY = 8.0
+RETRYABLE_HTTP_CODES = frozenset({500, 502, 503, 504})

pangea_sdk-6.5.0/pangea/_typing.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+from typing import TypeVar
+
+T = TypeVar("T")

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/asyncio/__init__.py

@@ -1,2 +1,3 @@
-# ruff: noqa: F401
 from .file_uploader import FileUploaderAsync
+
+__all__ = ("FileUploaderAsync",)

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/asyncio/file_uploader.py

@@ -6,9 +6,10 @@ from __future__ import annotations
 import io
 import logging
 
+from pangea import PangeaConfig, TransferMethod
 from pangea.asyncio.request import PangeaRequestAsync
-from pangea.request import PangeaConfig
-from pangea.response import TransferMethod
+
+__all__ = ("FileUploaderAsync",)
 
 
 class FileUploaderAsync:

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/asyncio/request.py

@@ -10,6 +10,7 @@ import asyncio
 import json
 import time
 from collections.abc import Iterable, Mapping
+from random import random
 from typing import Dict, List, Optional, Sequence, Tuple, Type, Union, cast
 
 import aiohttp
@@ -19,10 +20,13 @@ from pydantic_core import to_jsonable_python
 from typing_extensions import Any, TypeAlias, TypeVar, override
 
 import pangea.exceptions as pe
+from pangea._constants import MAX_RETRY_DELAY, RETRYABLE_HTTP_CODES
 from pangea.request import MultipartResponse, PangeaRequestBase
 from pangea.response import AttachedFile, PangeaResponse, PangeaResponseResult, ResponseStatus, TransferMethod
 from pangea.utils import default_encoder
 
+__all__ = ("PangeaRequestAsync",)
+
 _FileName: TypeAlias = Union[str, None]
 _FileContent: TypeAlias = Union[str, bytes]
 _FileContentType: TypeAlias = str
@@ -462,13 +466,46 @@ class PangeaRequestAsync(PangeaRequestBase):
 
     @override
     def _init_session(self) -> aiohttp.ClientSession:
-        # retry_config = Retry(
-        #     total=self.config.request_retries,
-        #     backoff_factor=self.config.request_backoff,
-        #     status_forcelist=[500, 502, 503, 504],
-        # )
-        # adapter = HTTPAdapter(max_retries=retry_config)
-        # TODO: Add retry config
-
-        session = aiohttp.ClientSession()
-        return session
+        return aiohttp.ClientSession(middlewares=[self._retry_middleware])
+
+    def _calculate_retry_timeout(self, remaining_retries: int) -> float:
+        max_retries = self.config.request_retries
+        nb_retries = min(max_retries - remaining_retries, 1000)
+        sleep_seconds = min(self.config.request_backoff * pow(2.0, nb_retries), MAX_RETRY_DELAY)
+        jitter = 1 - 0.25 * random()
+        timeout = sleep_seconds * jitter
+        return max(timeout, 0)
+
+    async def _retry_middleware(
+        self, request: aiohttp.ClientRequest, handler: aiohttp.ClientHandlerType
+    ) -> aiohttp.ClientResponse:
+        max_retries = self.config.request_retries
+        request_ids = set[str]()
+        retries_taken = 0
+        for retries_taken in range(max_retries + 1):
+            remaining_retries = max_retries - retries_taken
+
+            if len(request_ids) > 0:
+                request.headers["X-Pangea-Retried-Request-Ids"] = ",".join(request_ids)
+
+            response = await handler(request)
+
+            request_id = response.headers.get("x-request-id")
+            if request_id:
+                request_ids.add(request_id)
+
+            if not response.ok and remaining_retries > 0 and self._should_retry(response):
+                await self._sleep_for_retry(retries_taken=retries_taken, max_retries=max_retries)
+                continue
+
+            break
+
+        return response
+
+    def _should_retry(self, response: aiohttp.ClientResponse) -> bool:
+        return response.status in RETRYABLE_HTTP_CODES
+
+    async def _sleep_for_retry(self, *, retries_taken: int, max_retries: int) -> None:
+        remaining_retries = max_retries - retries_taken
+        timeout = self._calculate_retry_timeout(remaining_retries)
+        await asyncio.sleep(timeout)

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/asyncio/services/__init__.py

@@ -1,5 +1,3 @@
-# ruff: noqa: F401
-
 from .ai_guard import AIGuardAsync
 from .audit import AuditAsync
 from .authn import AuthNAsync
@@ -12,3 +10,22 @@ from .redact import RedactAsync
 from .sanitize import SanitizeAsync
 from .share import ShareAsync
 from .vault import VaultAsync
+
+__all__ = (
+    "AIGuardAsync",
+    "AuditAsync",
+    "AuthNAsync",
+    "AuthZAsync",
+    "DomainIntelAsync",
+    "EmbargoAsync",
+    "FileIntelAsync",
+    "FileScanAsync",
+    "IpIntelAsync",
+    "PromptGuardAsync",
+    "RedactAsync",
+    "SanitizeAsync",
+    "ShareAsync",
+    "UrlIntelAsync",
+    "UserIntelAsync",
+    "VaultAsync",
+)

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/asyncio/services/authn.py

@@ -10,9 +10,9 @@ from collections.abc import Mapping
 from typing import Dict, List, Literal, Optional, Union
 
 import pangea.services.authn.models as m
+from pangea import PangeaResponse, PangeaResponseResult
 from pangea.asyncio.services.base import ServiceBaseAsync
 from pangea.config import PangeaConfig
-from pangea.response import PangeaResponse, PangeaResponseResult
 
 __all__ = ["AuthNAsync"]
 
@@ -66,11 +66,18 @@ class AuthNAsync(ServiceBaseAsync):
              authn = AuthNAsync(token="pangea_token", config=config)
         """
         super().__init__(token, config, logger_name=logger_name)
-        self.user = AuthNAsync.UserAsync(token, config, logger_name=logger_name)
-        self.flow = AuthNAsync.FlowAsync(token, config, logger_name=logger_name)
+        self.agreements = AuthNAsync.AgreementsAsync(token, config, logger_name=logger_name)
         self.client = AuthNAsync.ClientAsync(token, config, logger_name=logger_name)
+        self.flow = AuthNAsync.FlowAsync(token, config, logger_name=logger_name)
         self.session = AuthNAsync.SessionAsync(token, config, logger_name=logger_name)
-        self.agreements = AuthNAsync.AgreementsAsync(token, config, logger_name=logger_name)
+        self.user = AuthNAsync.UserAsync(token, config, logger_name=logger_name)
+
+    async def close(self) -> None:
+        await self.agreements.close()
+        await self.client.close()
+        await self.flow.close()
+        await self.session.close()
+        await self.user.close()
 
     class SessionAsync(ServiceBaseAsync):
         service_name = _SERVICE_NAME
@@ -179,10 +186,15 @@ class AuthNAsync(ServiceBaseAsync):
             logger_name: str = "pangea",
         ) -> None:
             super().__init__(token, config, logger_name=logger_name)
-            self.session = AuthNAsync.ClientAsync.SessionAsync(token, config, logger_name=logger_name)
             self.password = AuthNAsync.ClientAsync.PasswordAsync(token, config, logger_name=logger_name)
+            self.session = AuthNAsync.ClientAsync.SessionAsync(token, config, logger_name=logger_name)
             self.token_endpoints = AuthNAsync.ClientAsync.TokenAsync(token, config, logger_name=logger_name)
 
+        async def close(self) -> None:
+            await self.password.close()
+            await self.session.close()
+            await self.token_endpoints.close()
+
         async def userinfo(self, code: str) -> PangeaResponse[m.ClientUserinfoResult]:
             """
             Get User (client token)
@@ -470,9 +482,14 @@ class AuthNAsync(ServiceBaseAsync):
             logger_name: str = "pangea",
         ) -> None:
             super().__init__(token, config, logger_name=logger_name)
-            self.profile = AuthNAsync.UserAsync.ProfileAsync(token, config, logger_name=logger_name)
             self.authenticators = AuthNAsync.UserAsync.AuthenticatorsAsync(token, config, logger_name=logger_name)
             self.invites = AuthNAsync.UserAsync.InvitesAsync(token, config, logger_name=logger_name)
+            self.profile = AuthNAsync.UserAsync.ProfileAsync(token, config, logger_name=logger_name)
+
+        async def close(self) -> None:
+            await self.authenticators.close()
+            await self.invites.close()
+            await self.profile.close()
 
         async def create(
             self,
@@ -932,7 +949,7 @@ class AuthNAsync(ServiceBaseAsync):
                 return await self.request.post(
                     "v2/user/group/assign",
                     data={"id": user_id, "group_ids": group_ids},
-                    result_class=m.PangeaResponseResult,
+                    result_class=PangeaResponseResult,
                 )
 
             async def remove(self, user_id: str, group_id: str) -> PangeaResponse[PangeaResponseResult]:
@@ -946,7 +963,7 @@ class AuthNAsync(ServiceBaseAsync):
                 return await self.request.post(
                     "v2/user/group/remove",
                     data={"id": user_id, "group_id": group_id},
-                    result_class=m.PangeaResponseResult,
+                    result_class=PangeaResponseResult,
                 )
 
             async def list(self, user_id: str) -> PangeaResponse[m.GroupList]:

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/asyncio/services/base.py

@@ -6,14 +6,20 @@
 
 from __future__ import annotations
 
+from types import TracebackType
 from typing import Dict, Optional, Type, Union
 
 from typing_extensions import override
 
+from pangea import PangeaResponse, PangeaResponseResult
+from pangea._typing import T
 from pangea.asyncio.request import PangeaRequestAsync
 from pangea.exceptions import AcceptedRequestException
-from pangea.response import AttachedFile, PangeaResponse, PangeaResponseResult
-from pangea.services.base import PangeaRequest, ServiceBase
+from pangea.request import PangeaRequest
+from pangea.response import AttachedFile
+from pangea.services.base import ServiceBase
+
+__all__ = ("ServiceBaseAsync",)
 
 
 class ServiceBaseAsync(ServiceBase):
@@ -85,8 +91,17 @@ class ServiceBaseAsync(ServiceBase):
         return await self.request.download_file(url=url, filename=filename)
 
     async def close(self):
+        """Close the underlying aiohttp client."""
+
         await self.request.session.close()
-        # Loop over all attributes to check if they are derived from ServiceBaseAsync and close them
-        for _, value in self.__dict__.items():
-            if issubclass(type(value), ServiceBaseAsync):
-                await value.close()
+
+    async def __aenter__(self: T) -> T:
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc: BaseException | None,
+        exc_tb: TracebackType | None,
+    ) -> None:
+        await self.close()

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/asyncio/services/file_scan.py

@@ -11,9 +11,9 @@ import logging
 from typing import Dict, List, Optional, Tuple
 
 import pangea.services.file_scan as m
+from pangea import PangeaConfig
 from pangea.asyncio.request import PangeaRequestAsync
 from pangea.asyncio.services.base import ServiceBaseAsync
-from pangea.request import PangeaConfig
 from pangea.response import PangeaResponse, TransferMethod
 from pangea.utils import FileUploadParams, get_file_upload_params
 

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/asyncio/services/prompt_guard.py

@@ -12,6 +12,9 @@ if TYPE_CHECKING:
     from pangea.response import PangeaResponse
 
 
+__all__ = ("Message", "PromptGuardAsync")
+
+
 class PromptGuardAsync(ServiceBaseAsync):
     """Prompt Guard service client.
 

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/config.py

@@ -6,6 +6,8 @@ from typing import Any, Optional
 
 from pydantic import BaseModel, model_validator
 
+__all__ = ("PangeaConfig",)
+
 
 class PangeaConfig(BaseModel):
     """Holds run time configuration information used by SDK components."""
@@ -34,12 +36,12 @@ class PangeaConfig(BaseModel):
 
     request_backoff: float = 0.5
     """
-    Backoff strategy passed to 'requests'
+    A backoff factor to apply between request attempts.
     """
 
     request_timeout: int = 5
     """
-    Timeout used on initial request attempts
+    Unused.
     """
 
     poll_result_timeout: int = 30

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/file_uploader.py

@@ -8,9 +8,12 @@ import io
 import logging
 from typing import Dict, Optional
 
-from pangea.request import PangeaConfig, PangeaRequest
+from pangea.config import PangeaConfig
+from pangea.request import PangeaRequest
 from pangea.response import TransferMethod
 
+__all__ = ("FileUploader",)
+
 
 class FileUploader:
     def __init__(self):

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/request.py

@@ -11,18 +11,20 @@ import json
 import logging
 import time
 from collections.abc import Iterable, Mapping
+from random import random
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Type, Union, cast
 
 import requests
 from pydantic import BaseModel
 from pydantic_core import to_jsonable_python
-from requests.adapters import HTTPAdapter, Retry
+from requests.adapters import HTTPAdapter
 from requests_toolbelt import MultipartDecoder  # type: ignore[import-untyped]
 from typing_extensions import TypeAlias, TypeVar, override
 from yarl import URL
 
 import pangea
 import pangea.exceptions as pe
+from pangea._constants import MAX_RETRY_DELAY, RETRYABLE_HTTP_CODES
 from pangea.config import PangeaConfig
 from pangea.response import AttachedFile, PangeaResponse, PangeaResponseResult, ResponseStatus, TransferMethod
 from pangea.utils import default_encoder
@@ -46,6 +48,70 @@ _Files: TypeAlias = Union[Mapping[str, _FileSpec], Iterable[tuple[str, _FileSpec
 _HeadersUpdateMapping: TypeAlias = Mapping[str, str]
 
 
+class PangeaHTTPAdapter(HTTPAdapter):
+    """Custom HTTP adapter that keeps track of retried request IDs."""
+
+    @override
+    def __init__(self, config: PangeaConfig, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.config = config
+
+    @override
+    def send(
+        self,
+        request: requests.PreparedRequest,
+        stream: bool = False,
+        timeout: None | float | tuple[float, float] | tuple[float, None] = None,
+        verify: bool | str = True,
+        cert: None | bytes | str | tuple[bytes | str, bytes | str] = None,
+        proxies: Mapping[str, str] | None = None,
+    ) -> requests.Response:
+        max_retries = self.config.request_retries
+        request_ids = set[str]()
+        retries_taken = 0
+        for retries_taken in range(max_retries + 1):
+            remaining_retries = max_retries - retries_taken
+
+            if len(request_ids) > 0:
+                request.headers["X-Pangea-Retried-Request-Ids"] = ",".join(request_ids)
+
+            response = super().send(request, stream, timeout, verify, cert, proxies)
+
+            request_id = response.headers.get("x-request-id")
+            if request_id:
+                request_ids.add(request_id)
+
+            try:
+                response.raise_for_status()
+            except requests.HTTPError as error:
+                if remaining_retries > 0 and self._should_retry(error.response):
+                    error.response.close()
+                    self._sleep_for_retry(retries_taken=retries_taken, max_retries=max_retries, config=self.config)
+                    continue
+
+                break
+
+            break
+
+        return response
+
+    def _calculate_retry_timeout(self, remaining_retries: int, config: PangeaConfig) -> float:
+        max_retries = config.request_retries
+        nb_retries = min(max_retries - remaining_retries, 1000)
+        sleep_seconds = min(config.request_backoff * pow(2.0, nb_retries), MAX_RETRY_DELAY)
+        jitter = 1 - 0.25 * random()
+        timeout = sleep_seconds * jitter
+        return max(timeout, 0)
+
+    def _sleep_for_retry(self, *, retries_taken: int, max_retries: int, config: PangeaConfig) -> None:
+        remaining_retries = max_retries - retries_taken
+        timeout = self._calculate_retry_timeout(remaining_retries, config)
+        time.sleep(timeout)
+
+    def _should_retry(self, response: requests.Response) -> bool:
+        return response.status_code in RETRYABLE_HTTP_CODES
+
+
 class MultipartResponse:
     pangea_json: Dict[str, str]
     attached_files: List = []
@@ -111,8 +177,8 @@ class PangeaRequestBase:
 
         return self._queued_retry_enabled
 
-    def _get_delay(self, retry_count, start):
-        delay = retry_count * retry_count
+    def _get_delay(self, retry_count: int, start: float) -> float:
+        delay: float = retry_count * retry_count
         now = time.time()
         # if with this delay exceed timeout, reduce delay
         if now - start + delay >= self.config.poll_result_timeout:
@@ -120,10 +186,10 @@ class PangeaRequestBase:
 
         return delay
 
-    def _reach_timeout(self, start):
+    def _reach_timeout(self, start: float) -> bool:
         return time.time() - start >= self.config.poll_result_timeout
 
-    def _get_poll_path(self, request_id: str):
+    def _get_poll_path(self, request_id: str) -> str:
         return f"request/{request_id}"
 
     def _url(self, path: str) -> str:
@@ -628,13 +694,7 @@ class PangeaRequest(PangeaRequestBase):
 
     @override
     def _init_session(self) -> requests.Session:
-        retry_config = Retry(
-            total=self.config.request_retries,
-            backoff_factor=self.config.request_backoff,
-            status_forcelist=[500, 502, 503, 504],
-        )
-
-        adapter = HTTPAdapter(max_retries=retry_config)
+        adapter = PangeaHTTPAdapter(config=self.config)
         session = requests.Session()
 
         session.mount("http://", adapter)

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/response.py

@@ -18,6 +18,8 @@ from typing_extensions import TypeVar
 
 from pangea.utils import format_datetime
 
+__all__ = ("PangeaResponse", "PangeaResponseResult", "TransferMethod")
+
 
 class AttachedFile:
     filename: str
@@ -243,7 +245,9 @@ class PangeaResponse(ResponseHeader, Generic[T]):
 
     @property
     def http_status(self) -> int:  # type: ignore[return]
-        if self.raw_response:
+        # Must be an explicit None check because Response's boolean
+        # representation is equal to whether or not the response is OK.
+        if self.raw_response is not None:
             if isinstance(self.raw_response, aiohttp.ClientResponse):
                 return self.raw_response.status
             else:

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/services/__init__.py

@@ -1,5 +1,3 @@
-# ruff: noqa: F401
-
 from .ai_guard import AIGuard
 from .audit.audit import Audit
 from .authn.authn import AuthN
@@ -12,3 +10,22 @@ from .redact import Redact
 from .sanitize import Sanitize
 from .share.share import Share
 from .vault.vault import Vault
+
+__all__ = (
+    "AIGuard",
+    "Audit",
+    "AuthN",
+    "AuthZ",
+    "DomainIntel",
+    "Embargo",
+    "FileIntel",
+    "FileScan",
+    "IpIntel",
+    "PromptGuard",
+    "Redact",
+    "Sanitize",
+    "Share",
+    "UrlIntel",
+    "UserIntel",
+    "Vault",
+)

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/services/audit/audit.py

@@ -57,6 +57,8 @@ from pangea.services.audit.util import (
 from pangea.services.base import ServiceBase
 from pangea.utils import canonicalize_nested_json
 
+__all__ = ("Audit", "SearchOutput", "SearchResultOutput")
+
 
 class AuditBase:
     def __init__(

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/services/audit/util.py

@@ -20,6 +20,8 @@ import requests
 from pangea.services.audit.models import Event, EventEnvelope, PublishedRoot
 from pangea.utils import default_encoder, format_datetime
 
+__all__ = ("format_datetime",)
+
 Hash = bytes
 
 

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/services/authn/authn.py

@@ -10,11 +10,10 @@ from collections.abc import Mapping
 from typing import Dict, List, Literal, Optional, Union
 
 import pangea.services.authn.models as m
-from pangea.config import PangeaConfig
-from pangea.response import PangeaResponse, PangeaResponseResult
+from pangea import PangeaConfig, PangeaResponse, PangeaResponseResult
 from pangea.services.base import ServiceBase
 
-__all__ = ["AuthN"]
+__all__ = ("AuthN",)
 
 
 _SERVICE_NAME = "authn"
@@ -930,7 +929,7 @@ class AuthN(ServiceBase):
                 return self.request.post(
                     "v2/user/group/assign",
                     data={"id": user_id, "group_ids": group_ids},
-                    result_class=m.PangeaResponseResult,
+                    result_class=PangeaResponseResult,
                 )
 
             def remove(self, user_id: str, group_id: str) -> PangeaResponse[PangeaResponseResult]:
@@ -944,7 +943,7 @@ class AuthN(ServiceBase):
                 return self.request.post(
                     "v2/user/group/remove",
                     data={"id": user_id, "group_id": group_id},
-                    result_class=m.PangeaResponseResult,
+                    result_class=PangeaResponseResult,
                 )
 
             def list(self, user_id: str) -> PangeaResponse[m.GroupList]:

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/services/base.py

@@ -18,6 +18,9 @@ from pangea.exceptions import AcceptedRequestException
 from pangea.request import PangeaRequest
 from pangea.response import AttachedFile, PangeaResponse, PangeaResponseResult
 
+__all__ = ("ServiceBase",)
+
+
 TResult = TypeVar("TResult", bound=PangeaResponseResult, default=PangeaResponseResult)
 
 

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/services/file_scan.py

@@ -10,8 +10,9 @@ import io
 import logging
 from typing import Dict, List, Optional, Tuple
 
-from pangea.request import PangeaConfig, PangeaRequest
-from pangea.response import APIRequestModel, PangeaResponse, PangeaResponseResult, TransferMethod
+from pangea import PangeaConfig, PangeaResponse, PangeaResponseResult, TransferMethod
+from pangea.request import PangeaRequest
+from pangea.response import APIRequestModel
 from pangea.services.base import ServiceBase
 from pangea.utils import FileUploadParams, get_file_upload_params
 

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/services/prompt_guard.py

@@ -10,6 +10,9 @@ if TYPE_CHECKING:
     from collections.abc import Iterable
 
 
+__all__ = ("Message", "PromptGuard")
+
+
 class Message(APIRequestModel):
     role: str
     content: str

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pangea/services/vault/vault.py

@@ -88,6 +88,9 @@ if TYPE_CHECKING:
     from pangea.request import TResult
 
 
+__all__ = ("ExportEncryptionAlgorithm", "ItemType", "ItemVersionState", "TransformAlphabet", "Vault")
+
+
 VaultItem = Annotated[
     Union[AsymmetricKey, SymmetricKey, Secret, ClientSecret, Folder, PangeaToken], Field(discriminator="type")
 ]

{pangea_sdk-6.4.0 → pangea_sdk-6.5.0}/pyproject.toml

@@ -1,14 +1,12 @@
 [project]
 name = "pangea-sdk"
-version = "6.4.0"
+version = "6.5.0"
 description = "Pangea API SDK"
 authors = [
     {name = "Glenn Gallien", email = "glenn.gallien@pangea.cloud"}
 ]
 license = "MIT"
 readme = "README.md"
-homepage = "https://pangea.cloud/docs/sdk/python/"
-repository = "https://github.com/pangeacyber/pangea-python/tree/main/packages/pangea-sdk"
 keywords = ["Pangea", "SDK", "Audit"]
 classifiers = [
     "Topic :: Software Development",
@@ -16,8 +14,8 @@ classifiers = [
 ]
 requires-python = ">=3.9.2,<4.0.0"
 dependencies = [
-  "aiohttp (>=3.12.14,<4.0.0)",
-  "cryptography (>=45.0.5,<46.0.0)",
+  "aiohttp (>=3.12.15,<4.0.0)",
+  "cryptography (>=45.0.6,<46.0.0)",
   "deprecated (>=1.2.18,<2.0.0)",
   "google-crc32c (>=1.7.1,<2.0.0)",
   "pydantic (>=2.11.7,<3.0.0)",
@@ -28,24 +26,28 @@ dependencies = [
   "yarl (>=1.20.1,<2.0.0)"
 ]
 
-[tool.poetry]
-packages = [
-    { include = "pangea" }
+[project.urls]
+repository = "https://github.com/pangeacyber/pangea-python"
+
+[dependency-groups]
+dev = [
+  "docstring-parser ==0.17.0",
+  "pytest-asyncio ==1.1.0",
+  "pytest_httpserver ==1.1.3",
+  "types-Deprecated ==1.2.15.20250304",
+  "types-python-dateutil ==2.9.0.20250809",
+  "types-requests ==2.32.4.20250809",
 ]
 
-[tool.poetry.group.dev.dependencies]
-docstring-parser = "^0.15"
-mypy = "1.17.0"
-pytest = "8.3.5"
-pytest-asyncio = "1.1.0"
-types-Deprecated = "^1.2.9.3"
-types-python-dateutil = "^2.8.19.14"
-types-requests = "2.32.4.20250611"
+[build-system]
+requires = ["uv_build==0.8.11"]
+build-backend = "uv_build"
 
 [tool.mypy]
 python_version = "3.9"
 color_output = true
 error_summary = true
+implicit_reexport = false
 pretty = true
 show_column_numbers = true
 warn_unused_ignores = true
@@ -64,6 +66,6 @@ asyncio_mode = "auto"
 asyncio_default_fixture_loop_scope = "session"
 filterwarnings = ["error"]
 
-[build-system]
-requires = ["poetry-core>=2.1.3"]
-build-backend = "poetry.core.masonry.api"
+[tool.uv.build-backend]
+module-name = "pangea"
+module-root = ""

pangea_sdk-6.4.0/pangea/__init__.py

@@ -1,15 +0,0 @@
-__version__ = "6.4.0"
-
-from pangea.asyncio.request import PangeaRequestAsync
-from pangea.config import PangeaConfig
-from pangea.file_uploader import FileUploader
-from pangea.request import PangeaRequest
-from pangea.response import PangeaResponse
-
-__all__ = (
-    "FileUploader",
-    "PangeaConfig",
-    "PangeaRequest",
-    "PangeaRequestAsync",
-    "PangeaResponse",
-)